1 | from fielddisplay import fielddisplay
|
---|
2 | from EnumDefinitions import *
|
---|
3 | from checkfield import *
|
---|
4 | from WriteData import *
|
---|
5 |
|
---|
6 | class mask(object):
|
---|
7 | """
|
---|
8 | MASK class definition
|
---|
9 |
|
---|
10 | Usage:
|
---|
11 | mask=mask();
|
---|
12 | """
|
---|
13 |
|
---|
14 | def __init__(self): # {{{
|
---|
15 | self.elementonfloatingice = float('NaN')
|
---|
16 | self.elementongroundedice = float('NaN')
|
---|
17 | self.vertexonfloatingice = float('NaN')
|
---|
18 | self.vertexongroundedice = float('NaN')
|
---|
19 | self.icelevelset = float('NaN')
|
---|
20 |
|
---|
21 | #set defaults
|
---|
22 | self.setdefaultparameters()
|
---|
23 |
|
---|
24 | #}}}
|
---|
25 | def __repr__(self): # {{{
|
---|
26 | string=" masks:"
|
---|
27 |
|
---|
28 | string="%s\n%s"%(string,fielddisplay(self,"elementonfloatingice","element on floating ice flags list"))
|
---|
29 | string="%s\n%s"%(string,fielddisplay(self,"vertexonfloatingice","vertex on floating ice flags list"))
|
---|
30 | string="%s\n%s"%(string,fielddisplay(self,"elementongroundedice","element on grounded ice list"))
|
---|
31 | string="%s\n%s"%(string,fielddisplay(self,"vertexongroundedice","vertex on grounded ice flags list"))
|
---|
32 | string="%s\n%s"%(string,fielddisplay(self,"icelevelset","presence of ice if > 0, icefront position if = 0, no ice if < 0"))
|
---|
33 | return string
|
---|
34 | #}}}
|
---|
35 | def setdefaultparameters(self): # {{{
|
---|
36 | return self
|
---|
37 | #}}}
|
---|
38 | def checkconsistency(self,md,solution,analyses): # {{{
|
---|
39 |
|
---|
40 | md = checkfield(md,'mask.elementonfloatingice','size',[md.mesh.numberofelements],'values',[0,1])
|
---|
41 | md = checkfield(md,'mask.elementongroundedice','size',[md.mesh.numberofelements],'values',[0,1])
|
---|
42 | md = checkfield(md,'mask.vertexonfloatingice' ,'size',[md.mesh.numberofvertices],'values',[0,1])
|
---|
43 | md = checkfield(md,'mask.vertexongroundedice' ,'size',[md.mesh.numberofvertices],'values',[0,1])
|
---|
44 | md = checkfield(md,'mask.icelevelset' ,'size',[md.mesh.numberofvertices])
|
---|
45 | isice=numpy.array(md.mask.icelevelset>0,int)
|
---|
46 | totallyicefree=(numpy.sum(isice[md.mesh.elements-1],axis=1)==0).astype(int)
|
---|
47 | if any(totallyicefree):
|
---|
48 | raise TypeError("elements with no ice not implemented yet, each element should have at least one vertex with md.mask.icelevelset > 0")
|
---|
49 |
|
---|
50 | return md
|
---|
51 | # }}}
|
---|
52 | def marshall(self,md,fid): # {{{
|
---|
53 | WriteData(fid,'object',self,'fieldname','elementonfloatingice','format','BooleanMat','mattype',2)
|
---|
54 | WriteData(fid,'object',self,'fieldname','elementongroundedice','format','BooleanMat','mattype',2)
|
---|
55 | WriteData(fid,'object',self,'fieldname','vertexonfloatingice','format','DoubleMat','mattype',1)
|
---|
56 | WriteData(fid,'object',self,'fieldname','vertexongroundedice','format','DoubleMat','mattype',1)
|
---|
57 | WriteData(fid,'object',self,'fieldname','icelevelset','format','DoubleMat','mattype',1)
|
---|
58 | # }}}
|
---|