source: issm/trunk-jpl/src/m/classes/mask.py@ 13984

#module imports
from fielddisplay import fielddisplay
from EnumDefinitions import *
from checkfield import *
from WriteData import *

class mask(object):
        """
        MASK class definition

           Usage:
              mask=mask();
        """

        #properties
        def __init__(self):
                # {{{ Properties
                self.elementonfloatingice = float('NaN')
                self.elementongroundedice = float('NaN')
                self.elementonwater       = float('NaN')
                self.vertexonfloatingice  = float('NaN')
                self.vertexongroundedice  = float('NaN')
                self.vertexonwater        = float('NaN')

                #set defaults
                self.setdefaultparameters()

                #}}}
        def __repr__(self):
                # {{{ Display

                string="";
                string="%s\n%s"%(string,fielddisplay(self,"elementonfloatingice","element on floating ice flags list"))
                string="%s\n%s"%(string,fielddisplay(self,"vertexonfloatingice","vertex on floating ice flags list"))
                string="%s\n%s"%(string,fielddisplay(self,"elementongroundedice","element on grounded ice list"))
                string="%s\n%s"%(string,fielddisplay(self,"vertexongroundedice","vertex on grounded ice flags list"))
                string="%s\n%s"%(string,fielddisplay(self,"elementonwater","element on water flags list"))
                string="%s\n%s"%(string,fielddisplay(self,"vertexonwater","vertex on water flags list"))
                return string
                #}}}
                
        def setdefaultparameters(self):
                # {{{setdefaultparameters
                return self
        #}}}

        def checkconsistency(self,md,solution,analyses):    # {{{

                md = checkfield(md,'mask.elementonfloatingice','size',[md.mesh.numberofelements],'values',[0,1])
                md = checkfield(md,'mask.elementongroundedice','size',[md.mesh.numberofelements],'values',[0,1])
                md = checkfield(md,'mask.elementonwater'      ,'size',[md.mesh.numberofelements],'values',[0,1])
                md = checkfield(md,'mask.vertexonfloatingice' ,'size',[md.mesh.numberofvertices],'values',[0,1])
                md = checkfield(md,'mask.vertexongroundedice' ,'size',[md.mesh.numberofvertices],'values',[0,1])
                md = checkfield(md,'mask.vertexonwater'       ,'size',[md.mesh.numberofvertices],'values',[0,1])

                return md
        # }}}

        def marshall(self,fid):    # {{{
                WriteData(fid,'object',self,'fieldname','elementonfloatingice','format','BooleanMat','mattype',2)
                WriteData(fid,'object',self,'fieldname','elementongroundedice','format','BooleanMat','mattype',2)
                WriteData(fid,'object',self,'fieldname','elementonwater','format','BooleanMat','mattype',2)
                WriteData(fid,'object',self,'fieldname','vertexonfloatingice','format','DoubleMat','mattype',1)
                WriteData(fid,'object',self,'fieldname','vertexongroundedice','format','DoubleMat','mattype',1)
                WriteData(fid,'object',self,'fieldname','vertexonwater','format','DoubleMat','mattype',1)
        # }}}