Mercurial > repos > public > sbplib
diff +scheme/Hypsyst3dCurve.m @ 350:5d5652fe826a feature/hypsyst
A commit before I try resolving the performance issues
author | Ylva Rydin <ylva.rydin@telia.com> |
---|---|
date | Wed, 02 Nov 2016 00:02:01 +0100 |
parents | |
children | 7cc3d5bd3692 |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/+scheme/Hypsyst3dCurve.m Wed Nov 02 00:02:01 2016 +0100 @@ -0,0 +1,479 @@ +classdef Hypsyst3dCurve < scheme.Scheme + properties + m % Number of points in each direction, possibly a vector + n %size of system + h % Grid spacing + X, Y, Z% Values of x and y for each grid point + Yx, Zx, Xy, Zy, Xz, Yz %Grid values for boundary surfaces + + xi,eta,zeta + Xi, Eta, Zeta + + Eta_xi, Zeta_xi, Xi_eta, Zeta_eta, Xi_zeta, Eta_zeta + + X_xi, X_eta, X_zeta,Y_xi,Y_eta,Y_zeta,Z_xi,Z_eta,Z_zeta + Aev + + metric_terms + + order % Order accuracy for the approximation + + D % non-stabalized scheme operator + Aevaluated, Bevaluated, Cevaluated, Eevaluated + Ahat, Bhat, Chat, E + A,B,C + + J, Ji + + H % Discrete norm + % Norms in the x, y and z directions + Hxii,Hetai,Hzetai, Hzi % Kroneckerd norms. 1'*Hx*v corresponds to integration in the x dir. + I_xi,I_eta,I_zeta, I_N,onesN + e_w, e_e, e_s, e_n, e_b, e_t + index_w, index_e,index_s,index_n, index_b, index_t + params %parameters for the coeficient matrice + end + + + methods + function obj = Hypsyst3dCurve(m, order, A, B,C, E, params,ti) + xilim ={0 1}; + etalim = {0 1}; + zetalim = {0 1}; + + if length(m) == 1 + m = [m m m]; + end + m_xi = m(1); + m_eta = m(2); + m_zeta=m(3); + m_tot=m_xi*m_eta*m_zeta; + obj.params = params; + obj.n = length(A(obj,0,0,0)); + + obj.m=m; + + obj.order=order; + obj.onesN=ones(obj.n); + ops_xi = sbp.D2Standard(m_xi,xilim,order); + ops_eta = sbp.D2Standard(m_eta,etalim,order); + ops_zeta = sbp.D2Standard(m_zeta,zetalim,order); + + obj.xi = ops_xi.x; + obj.eta = ops_eta.x; + obj.zeta = ops_zeta.x; + + obj.Xi = kr(obj.xi,ones(m_eta,1),ones(m_zeta,1));%% Que pasa? + obj.Eta = kr(ones(m_xi,1),obj.eta,ones(m_zeta,1)); + obj.Zeta = kr(ones(m_xi,1),ones(m_eta,1),obj.zeta); + + obj.Eta_xi=kr(obj.eta,ones(m_xi,1)); + obj.Zeta_xi=kr(ones(m_eta,1),obj.zeta); + + obj.Xi_eta=kr(obj.xi,ones(m_zeta,1)); + obj.Zeta_eta=kr(ones(m_xi,1),obj.zeta); + + obj.Xi_zeta=kr(obj.xi,ones(m_eta,1)); + obj.Eta_zeta=kr(ones(m_zeta,1),obj.eta); + + [X,Y,Z] = ti.map(obj.Xi,obj.Eta,obj.Zeta); + obj.X=X; + obj.Y=Y; + obj.Z=Z; + + I_n = eye(obj.n); + I_xi = speye(m_xi); + obj.I_xi = I_xi; + I_eta = speye(m_eta); + obj.I_eta = I_eta; + I_zeta = speye(m_zeta); + obj.I_zeta = I_zeta; + + + O_xi=ones(m_xi,1); + O_eta=ones(m_eta,1); + O_zeta=ones(m_zeta,1); + + D1_xi = kr(ops_xi.D1, I_eta,I_zeta); + obj.Hxii = kr(I_n, ops_xi.HI, I_eta,I_zeta); + D1_eta = kr(I_xi, ops_eta.D1,I_zeta); + obj.Hetai = kr(I_n, I_xi, ops_eta.HI,I_zeta); + D1_zeta = kr(I_xi, I_eta,ops_zeta.D1); + obj.Hzetai = kr(I_n, I_xi,I_eta, ops_zeta.HI); + obj.h=[ops_xi.h ops_eta.h ops_zeta.h]; + + obj.e_w = kr(I_n, ops_xi.e_l, I_eta,I_zeta); + obj.e_e = kr(I_n, ops_xi.e_r, I_eta,I_zeta); + obj.e_s = kr(I_n, I_xi, ops_eta.e_l,I_zeta); + obj.e_n = kr(I_n, I_xi, ops_eta.e_r,I_zeta); + obj.e_b = kr(I_n, I_xi, I_eta, ops_zeta.e_l); + obj.e_t = kr(I_n, I_xi, I_eta, ops_zeta.e_r); + + obj.A=A; + obj.B=B; + obj.C=C; + + obj.X_xi=D1_xi*X; + obj.X_eta=D1_eta*X; + obj.X_zeta=D1_zeta*X; + obj.Y_xi=D1_xi*Y; + obj.Y_eta=D1_eta*Y; + obj.Y_zeta=D1_zeta*Y; + obj.Z_xi=D1_xi*Z; + obj.Z_eta=D1_eta*Z; + obj.Z_zeta=D1_zeta*Z; + + D1_xi=kr(I_n,D1_xi); + D1_eta=kr(I_n,D1_eta); + D1_zeta=kr(I_n,D1_zeta); + + obj.index_w=(kr(ops_xi.e_l, O_eta,O_zeta)==1); + obj.index_e=(kr(ops_xi.e_r, O_eta,O_zeta)==1); + obj.index_s=(kr(O_xi, ops_eta.e_l,O_zeta)==1); + obj.index_n=(kr(O_xi, ops_eta.e_r,O_zeta)==1); + obj.index_b=(kr(O_xi, O_eta, ops_zeta.e_l)==1); + obj.index_t=(kr(O_xi, O_eta, ops_zeta.e_r)==1); + + + obj.Ahat=@transform_coefficient_matrix; + obj.Bhat=@transform_coefficient_matrix; + obj.Chat=@transform_coefficient_matrix; + obj.E=@(obj,x,y,z,~,~,~,~,~,~)E(obj,x,y,z); + + obj.Aevaluated = obj.evaluateCoefficientMatrix(obj.Ahat,obj.X, obj.Y,obj.Z, obj.X_eta,obj.X_zeta,obj.Y_eta,obj.Y_zeta,obj.Z_eta,obj.Z_zeta); + obj.Bevaluated = obj.evaluateCoefficientMatrix(obj.Bhat,obj.X, obj.Y,obj.Z, obj.X_zeta,obj.X_xi,obj.Y_zeta,obj.Y_xi,obj.Z_zeta,obj.Z_xi); + obj.Cevaluated = obj.evaluateCoefficientMatrix(obj.Chat,obj.X,obj.Y,obj.Z, obj.X_xi,obj.X_eta,obj.Y_xi,obj.Y_eta,obj.Z_xi,obj.Z_eta); + obj.Eevaluated = obj.evaluateCoefficientMatrix(obj.E, obj.X, obj.Y,obj.Z,[],[],[],[],[],[]); + + obj.J=obj.X_xi.*obj.Y_eta.*obj.Z_zeta... + +obj.X_zeta.*obj.Y_xi.*obj.Z_eta... + +obj.X_eta.*obj.Y_zeta.*obj.Z_xi... + -obj.X_xi.*obj.Y_zeta.*obj.Z_eta... + -obj.X_eta.*obj.Y_xi.*obj.Z_zeta... + -obj.X_zeta.*obj.Y_eta.*obj.Z_xi; + + obj.Ji =kr(I_n,spdiags(1./obj.J,0,m_tot,m_tot)); + + obj.D=obj.Ji*(-obj.Aevaluated*D1_xi-obj.Bevaluated*D1_eta -obj.Cevaluated*D1_zeta)-obj.Eevaluated; + end + + function [ret]=transform_coefficient_matrix(obj,x,y,z,x_1,x_2,y_1,y_2,z_1,z_2) + ret=obj.A(obj,x,y,z).*(y_1.*z_2-z_1.*y_2); + ret=ret+obj.B(obj,x,y,z).*(x_2.*z_1-x_1.*z_2); + ret=ret+obj.C(obj,x,y,z).*(x_1.*y_2-x_2.*y_1); + end + + + % Closure functions return the opertors applied to the own doamin to close the boundary + % Penalty functions return the opertors to force the solution. In the case of an interface it returns the operator applied to the other doamin. + % boundary is a string specifying the boundary e.g. 'l','r' or 'e','w','n','s'. + % type is a string specifying the type of boundary condition if there are several. + % data is a function returning the data that should be applied at the boundary. + function [closure, penalty] = boundary_condition(obj,boundary,type,L) + default_arg('type','char'); + BM=boundary_matrices(obj,boundary); + + switch type + case{'c','char'} + [closure,penalty]=boundary_condition_char(obj,BM); + case{'general'} + [closure,penalty]=boundary_condition_general(obj,BM,boundary,L); + otherwise + error('No such boundary condition') + end + end + + function [closure, penalty] = interface(obj,boundary,neighbour_scheme,neighbour_boundary) + error('An interface function does not exist yet'); + end + + function N = size(obj) + N = obj.m; + end + + function [ret] = evaluateCoefficientMatrix(obj,mat, X, Y, Z , x_1 , x_2 , y_1 , y_2 , z_1 , z_2) + params=obj.params; + side=max(length(X),length(Y)); + if isa(mat,'function_handle') + [rows,cols]=size(mat(obj,0,0,0,0,0,0,0,0,0)); + x_1=kr(obj.onesN,x_1); + x_2=kr(obj.onesN,x_2); + y_1=kr(obj.onesN,y_1); + y_2=kr(obj.onesN,y_2); + z_1=kr(obj.onesN,z_1); + z_2=kr(obj.onesN,z_2); + matVec=mat(obj,X',Y',Z',x_1',x_2',y_1',y_2',z_1',z_2'); + matVec=sparse(matVec); + else + matVec=mat; + [rows,cols]=size(matVec); + side=max(length(X),length(Y)); + cols=cols/side; + end + ret=kron(ones(rows,cols),speye(side)); + + for ii=1:rows + for jj=1:cols + ret((ii-1)*side+1:ii*side,(jj-1)*side+1:jj*side)=diag(matVec(ii,(jj-1)*side+1:jj*side)); + end + end + end + + + function [BM]=boundary_matrices(obj,boundary) + params=obj.params; + BM.boundary=boundary; + switch boundary + case {'w','W','west'} + BM.e_=obj.e_w; + mat=obj.Ahat; + BM.boundpos='l'; + BM.Hi=obj.Hxii; + BM.index=obj.index_w; + BM.x_1=obj.X_eta(BM.index); + BM.x_2=obj.X_zeta(BM.index); + BM.y_1=obj.Y_eta(BM.index); + BM.y_2=obj.Y_zeta(BM.index); + BM.z_1=obj.Z_eta(BM.index); + BM.z_2=obj.Z_zeta(BM.index); + case {'e','E','east'} + BM.e_=obj.e_e; + mat=obj.Ahat; + BM.boundpos='r'; + BM.Hi=obj.Hxii; + BM.index=obj.index_e; + BM.x_1=obj.X_eta(BM.index); + BM.x_2=obj.X_zeta(BM.index); + BM.y_1=obj.Y_eta(BM.index); + BM.y_2=obj.Y_zeta(BM.index); + BM.z_1=obj.Z_eta(BM.index); + BM.z_2=obj.Z_zeta(BM.index); + case {'s','S','south'} + BM.e_=obj.e_s; + mat=obj.Bhat; + BM.boundpos='l'; + BM.Hi=obj.Hetai; + BM.index=obj.index_s; + BM.x_1=obj.X_zeta(BM.index); + BM.x_2=obj.X_xi(BM.index); + BM.y_1=obj.Y_zeta(BM.index); + BM.y_2=obj.Y_xi(BM.index); + BM.z_1=obj.Z_zeta(BM.index); + BM.z_2=obj.Z_xi(BM.index); + case {'n','N','north'} + BM.e_=obj.e_n; + mat=obj.Bhat; + BM.boundpos='r'; + BM.Hi=obj.Hetai; + BM.index=obj.index_n; + BM.x_1=obj.X_zeta(BM.index); + BM.x_2=obj.X_xi(BM.index); + BM.y_1=obj.Y_zeta(BM.index); + BM.y_2=obj.Y_xi(BM.index); + BM.z_1=obj.Z_zeta(BM.index); + BM.z_2=obj.Z_xi(BM.index); + case{'b','B','Bottom'} + BM.e_=obj.e_b; + mat=obj.Chat; + BM.boundpos='l'; + BM.Hi=obj.Hzetai; + BM.index=obj.index_b; + BM.x_1=obj.X_xi(BM.index); + BM.x_2=obj.X_eta(BM.index); + BM.y_1=obj.Y_xi(BM.index); + BM.y_2=obj.Y_eta(BM.index); + BM.z_1=obj.Z_xi(BM.index); + BM.z_2=obj.Z_eta(BM.index); + case{'t','T','Top'} + BM.e_=obj.e_t; + mat=obj.Chat; + BM.boundpos='r'; + BM.Hi=obj.Hzetai; + BM.index=obj.index_t; + BM.x_1=obj.X_xi(BM.index); + BM.x_2=obj.X_eta(BM.index); + BM.y_1=obj.Y_xi(BM.index); + BM.y_2=obj.Y_eta(BM.index); + BM.z_1=obj.Z_xi(BM.index); + BM.z_2=obj.Z_eta(BM.index); + end + [BM.V,BM.Vi,BM.D,signVec]=obj.matrixDiag(mat,obj.X(BM.index),obj.Y(BM.index),obj.Z(BM.index),... + BM.x_1,BM.x_2,BM.y_1,BM.y_2,BM.z_1,BM.z_2); + BM.side=sum(BM.index); + BM.pos=signVec(1); BM.zeroval=signVec(2); BM.neg=signVec(3); + end + + + function [closure, penalty]=boundary_condition_char(obj,BM) + side = BM.side; + pos = BM.pos; + neg = BM.neg; + zeroval=BM.zeroval; + V = BM.V; + Vi = BM.Vi; + Hi=BM.Hi; + D=BM.D; + e_=BM.e_; + + switch BM.boundpos + case {'l'} + tau=sparse(obj.n*side,pos); + Vi_plus=Vi(1:pos,:); + tau(1:pos,:)=-abs(D(1:pos,1:pos)); + closure=Hi*e_*V*tau*Vi_plus*e_'; + penalty=-Hi*e_*V*tau*Vi_plus; + case {'r'} + tau=sparse(obj.n*side,neg); + tau((pos+zeroval)+1:obj.n*side,:)=-abs(D((pos+zeroval)+1:obj.n*side,(pos+zeroval)+1:obj.n*side)); + Vi_minus=Vi((pos+zeroval)+1:obj.n*side,:); + closure=Hi*e_*V*tau*Vi_minus*e_'; + penalty=-Hi*e_*V*tau*Vi_minus; + end + end + + + function [closure,penalty]=boundary_condition_general(obj,BM,boundary,L) + side = BM.side; + pos = BM.pos; + neg = BM.neg; + zeroval=BM.zeroval; + V = BM.V; + Vi = BM.Vi; + Hi=BM.Hi; + D=BM.D; + e_=BM.e_; + index=BM.index; + + switch BM.boundary + case{'b','B','bottom'} + Ji_vec=diag(obj.Ji); + Ji=diag(Ji_vec(index)); + Zeta_x=Ji*(obj.Y_xi(index).*obj.Z_eta(index)-obj.Z_xi(index).*obj.Y_eta(index)); + Zeta_y=Ji*(obj.X_eta(index).*obj.Z_xi(index)-obj.X_xi(index).*obj.Z_eta(index)); + Zeta_z=Ji*(obj.X_xi(index).*obj.Y_eta(index)-obj.Y_xi(index).*obj.X_eta(index)); + + L=obj.evaluateCoefficientMatrix(L,Zeta_x,Zeta_y,Zeta_z,[],[],[],[],[],[]); + end + + switch BM.boundpos + case {'l'} + tau=sparse(obj.n*side,pos); + Vi_plus=Vi(1:pos,:); + Vi_minus=Vi(pos+zeroval+1:obj.n*side,:); + V_plus=V(:,1:pos); + V_minus=V(:,(pos+zeroval)+1:obj.n*side); + + tau(1:pos,:)=-abs(D(1:pos,1:pos)); + R=-inv(L*V_plus)*(L*V_minus); + closure=Hi*e_*V*tau*(Vi_plus-R*Vi_minus)*e_'; + penalty=-Hi*e_*V*tau*inv(L*V_plus)*L; + case {'r'} + tau=sparse(obj.n*side,neg); + tau((pos+zeroval)+1:obj.n*side,:)=-abs(D((pos+zeroval)+1:obj.n*side,(pos+zeroval)+1:obj.n*side)); + Vi_plus=Vi(1:pos,:); + Vi_minus=Vi((pos+zeroval)+1:obj.n*side,:); + + V_plus=V(:,1:pos); + V_minus=V(:,(pos+zeroval)+1:obj.n*side); + R=-inv(L*V_minus)*(L*V_plus); + closure=Hi*e_*V*tau*(Vi_minus-R*Vi_plus)*e_'; + penalty=-Hi*e_*V*tau*inv(L*V_minus)*L; + end + end + + + function [V,Vi, D,signVec]=matrixDiag(obj,mat,x,y,z,x_1,x_2,y_1,y_2,z_1,z_2) + params=obj.params; + eps=10^(-10); + if(sum(abs(x_1))>eps) + syms x_1s + else + x_1s=0; + end + + if(sum(abs(x_2))>eps) + syms x_2s; + else + x_2s=0; + end + + + if(sum(abs(y_1))>eps) + syms y_1s + else + y_1s=0; + end + + if(sum(abs(y_2))>eps) + syms y_2s; + else + y_2s=0; + end + + + if(sum(abs(z_1))>eps) + syms z_1s + else + z_1s=0; + end + + if(sum(abs(z_2))>eps) + syms z_2s; + else + z_2s=0; + end + + syms xs ys zs + [V, D]=eig(mat(obj,xs,ys,zs,x_1s,x_2s,y_1s,y_2s,z_1s,z_2s)); + Vi=inv(V); + + syms x_1s x_2s y_1s y_2s z_1s z_2s +% V= matlabFunction(V); +% D= matlabFunction(D); +% Vi= matlabFunction(Vi); +% +% xs=x; +% ys=y; +% zs=z; +% x_1s=x_1; +% x_2s=x_2; +% y_1s=y_1; +% y_2s=y_2; +% z_1s=z_1; +% z_2s=z_2; + + side=max(length(x),length(y)); + Dret=zeros(obj.n,side*obj.n); + Vret=zeros(obj.n,side*obj.n); + Viret=zeros(obj.n,side*obj.n); + + for ii=1:obj.n + for jj=1:obj.n + Dpart=matlabFunction(D(jj,ii),'Vars',[xs ys zs x_1s x_2s y_1s y_2s z_1s z_2s]); + Vpart=matlabFunction(V(jj,ii),'Vars',[xs ys zs x_1s x_2s y_1s y_2s z_1s z_2s]); + Vipart=matlabFunction(V(jj,ii),'Vars',[xs ys zs x_1s x_2s y_1s y_2s z_1s z_2s]); + Dret(jj,(ii-1)*side+1:side*ii)=sparse(Dpart(x,y,z,x_1,x_2,y_1,y_2,z_1,z_2)); + Vret(jj,(ii-1)*side+1:side*ii)=sparse(Vpart(x,y,z,x_1,x_2,y_1,y_2,z_1,z_2)); + Viret(jj,(ii-1)*side+1:side*ii)=sparse(Vipart(x,y,z,x_1,x_2,y_1,y_2,z_1,z_2)); + end + end + + D=sparse(Dret); + V=sparse(Vret); + Vi=sparse(Viret); + V=obj.evaluateCoefficientMatrix(V,x,y,z,x_1,x_2,y_1,y_2,z_1,z_2); + D=obj.evaluateCoefficientMatrix(D,x,y,z,x_1,x_2,y_1,y_2,z_1,z_2); + Vi=obj.evaluateCoefficientMatrix(Vi,x,y,z,x_1,x_2,y_1,y_2,z_1,z_2); + DD=diag(D); + + poseig=(DD>0); + zeroeig=(DD==0); + negeig=(DD<0); + + D=diag([DD(poseig); DD(zeroeig); DD(negeig)]); + V=[V(:,poseig) V(:,zeroeig) V(:,negeig)]; + %Vi=inv(V); + signVec=[sum(poseig),sum(zeroeig),sum(negeig)]; + end + end + end