source: issm/trunk/src/m/classes/mesh2dvertical.py@ 23189

import numpy as np
from fielddisplay import fielddisplay
from checkfield import checkfield
import MatlabFuncs as m
from WriteData import WriteData

class mesh2dvertical(object):
        """
        MESH2DVERTICAL class definition

           Usage:
              mesh2dvertical=mesh2dvertical();
        """

        def __init__(self): # {{{
                self.x                           = float('NaN')
                self.y                           = float('NaN')
                self.elements                    = float('NaN')
                self.numberofelements            = 0
                self.numberofvertices            = 0
                self.numberofedges               = 0
                
                self.lat                         = float('NaN')
                self.long                        = float('NaN')
                self.epsg                        = float('NaN')
                self.scale_factor                = float('NaN');

                self.vertexonboundary            = float('NaN')
                self.vertexonbase                = float('NaN')
                self.vertexonsurface             = float('NaN')

                self.edges                       = float('NaN')
                self.segments                    = float('NaN')
                self.segmentmarkers              = float('NaN')
                self.vertexconnectivity          = float('NaN')
                self.elementconnectivity         = float('NaN')
                self.average_vertex_connectivity = 0

                #set defaults
                self.setdefaultparameters()

                #}}}
        def __repr__(self): # {{{
                string="   2D tria Mesh (vertical):" 

                string="%s\n%s"%(string,"\n      Elements and vertices:")
                string="%s\n%s"%(string,fielddisplay(self,"numberofelements","number of elements"))
                string="%s\n%s"%(string,fielddisplay(self,"numberofvertices","number of vertices"))
                string="%s\n%s"%(string,fielddisplay(self,"elements","vertex indices of the mesh elements"))
                string="%s\n%s"%(string,fielddisplay(self,"x","vertices x coordinate [m]"))
                string="%s\n%s"%(string,fielddisplay(self,"y","vertices y coordinate [m]"))
                string="%s\n%s"%(string,fielddisplay(self,"edges","edges of the 2d mesh (vertex1 vertex2 element1 element2)"))
                string="%s\n%s"%(string,fielddisplay(self,"numberofedges","number of edges of the 2d mesh"))

                string="%s%s"%(string,"\n\n      Properties:")
                string="%s\n%s"%(string,fielddisplay(self,"vertexonboundary","vertices on the boundary of the domain flag list"))
                string="%s\n%s"%(string,fielddisplay(self,'vertexonbase','vertices on the bed of the domain flag list'))
                string="%s\n%s"%(string,fielddisplay(self,'vertexonsurface','vertices on the surface of the domain flag list'))
                string="%s\n%s"%(string,fielddisplay(self,"segments","edges on domain boundary (vertex1 vertex2 element)"))
                string="%s\n%s"%(string,fielddisplay(self,"segmentmarkers","number associated to each segment"))
                string="%s\n%s"%(string,fielddisplay(self,"vertexconnectivity","list of elements connected to vertex_i"))
                string="%s\n%s"%(string,fielddisplay(self,"elementconnectivity","list of elements adjacent to element_i"))
                string="%s\n%s"%(string,fielddisplay(self,"average_vertex_connectivity","average number of vertices connected to one vertex"))

                string="%s%s"%(string,"\n\n      Projection:")
                string="%s\n%s"%(string,fielddisplay(self,"lat","vertices latitude [degrees]"))
                string="%s\n%s"%(string,fielddisplay(self,"long","vertices longitude [degrees]"))
                string="%s\n%s"%(string,fielddisplay(self,"epsg","EPSG code (ex: 3413 for UPS Greenland, 3031 for UPS Antarctica)"))
                string="%s\n%s"%(string,fielddisplay(self,"scale_factor","Projection correction for volume, area, etc. computation"))
                return string
                #}}}
        def setdefaultparameters(self): # {{{
                
                #the connectivity is the averaged number of nodes linked to a
                #given node through an edge. This connectivity is used to initially
                #allocate memory to the stiffness matrix. A value of 16 seems to
                #give a good memory/time ration. This value can be checked in
                #trunk/test/Miscellaneous/runme.m
                self.average_vertex_connectivity=25.

                return self
        #}}}
        def checkconsistency(self,md,solution,analyses):    # {{{
                if(solution=='LoveSolution'):
                        return

                md = checkfield(md,'fieldname','mesh.x','NaN',1,'Inf',1,'size',[md.mesh.numberofvertices])
                md = checkfield(md,'fieldname','mesh.y','NaN',1,'Inf',1,'size',[md.mesh.numberofvertices])
                md = checkfield(md,'fieldname','mesh.elements','NaN',1,'Inf',1,'>',0,'values',np.arange(1,md.mesh.numberofvertices+1))
                md = checkfield(md,'fieldname','mesh.elements','size',[md.mesh.numberofelements,3])
                if np.any(np.logical_not(m.ismember(np.arange(1,md.mesh.numberofvertices+1),md.mesh.elements))):
                        md.checkmessage("orphan nodes have been found. Check the mesh outline")
                md = checkfield(md,'fieldname','mesh.numberofelements','>',0)
                md = checkfield(md,'fieldname','mesh.numberofvertices','>',0)
                md = checkfield(md,'fieldname','mesh.vertexonbase','size',[md.mesh.numberofvertices],'values',[0,1])
                md = checkfield(md,'fieldname','mesh.vertexonsurface','size',[md.mesh.numberofvertices],'values',[0,1])
                md = checkfield(md,'fieldname','mesh.average_vertex_connectivity','>=',9,'message',"'mesh.average_vertex_connectivity' should be at least 9 in 2d")
                if(np.size(self.scale_factor)>1):
                        md = checkfield(md,'fieldname','mesh.scale_factor','NaN',1,'Inf',1,'size',[md.mesh.numberofvertices])

                if solution=='ThermalSolution':
                        md.checkmessage("thermal not supported for 2d mesh")

                return md
        # }}}
        def domaintype(self): # {{{
                return "2Dvertical"
        #}}}
        def dimension(self): # {{{
                return 2
        #}}}
        def elementtype(self): # {{{
                return "Tria"
        #}}}
        def vertexflags(self,value): # {{{
                flags = np.zeros((self.numberofvertices,))
                pos   = self.segments[np.where(self.segmentmarkers==value),0:2]-1
                flags[pos] = 1
                return flags
        #}}}
        def marshall(self,prefix,md,fid):    # {{{
                WriteData(fid,prefix,'name','md.mesh.domain_type','data',"Domain"+self.domaintype(),'format','String');
                WriteData(fid,prefix,'name','md.mesh.domain_dimension','data',self.dimension(),'format','Integer');
                WriteData(fid,prefix,'name','md.mesh.elementtype','data',self.elementtype(),'format','String');
                WriteData(fid,prefix,'object',self,'class','mesh','fieldname','x','format','DoubleMat','mattype',1)
                WriteData(fid,prefix,'object',self,'class','mesh','fieldname','y','format','DoubleMat','mattype',1)
                WriteData(fid,prefix,'name','md.mesh.z','data',np.zeros(self.numberofvertices),'format','DoubleMat','mattype',1);
                WriteData(fid,prefix,'object',self,'class','mesh','fieldname','elements','format','DoubleMat','mattype',2)
                WriteData(fid,prefix,'object',self,'class','mesh','fieldname','numberofelements','format','Integer')
                WriteData(fid,prefix,'object',self,'class','mesh','fieldname','numberofvertices','format','Integer')
                WriteData(fid,prefix,'object',self,'class','mesh','fieldname','vertexonbase','format','BooleanMat','mattype',1)
                WriteData(fid,prefix,'object',self,'class','mesh','fieldname','vertexonsurface','format','BooleanMat','mattype',1)
                WriteData(fid,prefix,'object',self,'class','mesh','fieldname','average_vertex_connectivity','format','Integer')
                WriteData(fid,prefix,'object',self,'class','mesh','fieldname','scale_factor','format','DoubleMat','mattype',1)
        # }}}