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