source: issm/trunk/src/m/classes/regionaloutput.py@ 22758

from project3d import project3d
from fielddisplay import fielddisplay
from pairoptions import pairoptions
from checkfield import checkfield
from WriteData import WriteData
from MeshProfileIntersection import MeshProfileIntersection
from ContourToMesh import ContourToMesh
import numpy as np
import os

class regionaloutput(object):
        """
        REGIONALOUTPUT class definition
        
           Usage:
              regionaloutput=regionaloutput();
              regionaloutput=regionaloutput('name','Volume1','definitionstring','Outputdefinition1','outputnamestring','IceVolume','mask',mask);
              regionaloutput=regionaloutput('name','Volume1','definitionstring','Outputdefinition1','outputnamestring','IceVolume','maskexpstring','Exp/Mask.exp','model',md)
        
           where mask is a vectorial field of size md.mesh.numberofvertices,1 : where vertices with values > 1 are to be included in the calculated region.
           Alternatively, the user can pass in an Argus file and model object instead of a mask, and mask will be calculated for the user
        """

        def __init__(self,*args): # {{{

                self.name              = ''
                self.definitionstring  = ''
                self.outputnamestring  = ''
                self.mask              = float('NaN')
                self.maskexpstring     = ''

                #set defaults
                self.setdefaultparameters()

                #use provided options to change fields
                options=pairoptions(*args)

                #OK get other fields
                self=options.AssignObjectFields(self)

                #get name
                if options.getfieldvalue('model',0):
                        if options.getfieldvalue('maskexpstring',0):
                                modelname=options.getfieldvalue('model')
                                self.maskexpstring=options.getfieldvalue('maskexpstring')
                                self.setmaskfromexp(modelname)
                        
                if (len(self.mask)<=1 & np.any(np.isnan(self.mask))):
                        error('regionaloutput error message: ''mask'' field or ''maskexpstring'' and ''model'' fields should be defined!');

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

                string="   Regionaloutput:"
                string="%s\n%s"%(string,fielddisplay(self,'name','identifier for this regional response'))
                string="%s\n%s"%(string,fielddisplay(self,'definitionstring','string that identifies this output definition uniquely, from Outputdefinition[1-100]'))
                string="%s\n%s"%(string,fielddisplay(self,'outputnamestring','string that identifies the type of output you want, eg. IceVolume, TotalSmb, GroudedArea'))
                string="%s\n%s"%(string,fielddisplay(self,'mask','mask vectorial field which identifies the region of interest (value > 0 will be included)'))
                string="%s\n%s"%(string,fielddisplay(self,'maskexpstring','name of Argus file that can be passed in to define the regional mask'))
                return string
                #}}}
        def extrude(self,md): # {{{
                self.mask=project3d(md,'vector',self.mask,'type','node')
                return self
           #}}}
        def setdefaultparameters(self): # {{{
                return self
        #}}}
        def setmaskfromexp(self,md):    # {{{
                if len(self.maskexpstring) > 0:
                        self.mask=ContourToMesh(md.mesh.elements,md.mesh.x,md.mesh.y,self.maskexpstring,'node',1)
                        
                return self
         # }}}
        def checkconsistency(self,md,solution,analyses):    # {{{
                
                if  not isinstance(self.name, basestring):
                        raise RuntimeError("regionaloutput error message: 'name' field should be a string!")
                        
                if  not isinstance(self.outputnamestring, basestring):
                        raise RuntimeError("regionaloutput error message: 'outputnamestring' field should be a string!") 
                
                if len(self.maskexpstring) > 0:
                        if not os.path.isfile(self.maskexpstring):
                                raise RuntimeError("regionaloutput error message: file name for mask exp does not point to a legitimate file on disk!")
                        else:
                                self.setmaskfromexp(md)

                OutputdefinitionStringArray=[]
                for i in range(1,100):
                        x='Outputdefinition'+str(i)
                        OutputdefinitionStringArray.append(x)

                md = checkfield(md,'field',self.definitionstring,'values',OutputdefinitionStringArray)
                md = checkfield(md,'field',self.mask,'size',[md.mesh.numberofvertices],'NaN',1,'Inf',1)
                return md
        # }}}
        def marshall(self,prefix,md,fid):    # {{{

                #before marshalling, make sure mask is set: 
                self.setmaskfromexp(md)

                #ok, marshall strings and mask: 
                WriteData(fid,prefix,'data',self.name,'name','md.regionaloutput.name','format','String')
                WriteData(fid,prefix,'data',self.definitionstring,'name','md.regionaloutput.definitionstring','format','String')
                WriteData(fid,prefix,'data',self.outputnamestring,'name','md.regionaloutput.outputnamestring','format','String');
                WriteData(fid,prefix,'data',self.mask,'name','md.regionaloutput.mask','format','DoubleMat','mattype',1);

        # }}}