from issm.fielddisplay import fielddisplay
from issm.checkfield import *
from issm.project3d import *
from issm.WriteData import *
[docs]class SMBcomponents(object):
"""
SMBcomponents Class definition
Usage:
SMBcomponents=SMBcomponents();
"""
def __init__(self): # {{{
self.accumulation = float('NaN')
self.runoff = float('NaN')
self.evaporation = float('NaN')
self.requested_outputs = []
#}}}
def __repr__(self): # {{{
string=" surface forcings parameters (SMB=accumulation-runoff-evaporation) :"
string="%s\n%s"%(string,fielddisplay(self,'accumulation','accumulated snow [m/yr ice eq]'))
string="%s\n%s"%(string,fielddisplay(self,'runoff','amount of ice melt lost from the ice column [m/yr ice eq]'))
string="%s\n%s"%(string,fielddisplay(self,'evaporation','mount of ice lost to evaporative processes [m/yr ice eq]'))
string="%s\n%s"%(string,fielddisplay(self,'requested_outputs','additional outputs requested'))
return string
#}}}
[docs] def extrude(self,md): # {{{
self.mass_balance=project3d(md,'vector',self.accumulation,'type','node');
self.mass_balance=project3d(md,'vector',self.runoff,'type','node');
self.mass_balance=project3d(md,'vector',self.evaporation,'type','node');
return self
#}}}
[docs] def defaultoutputs(self,md): # {{{
return []
#}}}
[docs] def initialize(self,md): # {{{
if np.all(np.isnan(self.accumulation)):
self.accumulation=np.zeros((md.mesh.numberofvertices))
print " no SMB.accumulation specified: values set as zero"
if np.all(np.isnan(self.runoff)):
self.runoff=np.zeros((md.mesh.numberofvertices))
print " no SMB.runoff specified: values set as zero"
if np.all(np.isnan(self.evaporation)):
self.evaporation=np.zeros((md.mesh.numberofvertices))
print " no SMB.evaporation specified: values set as zero"
return self
#}}}
[docs] def checkconsistency(self,md,solution,analyses): # {{{
if 'MasstransportAnalysis' in analyses:
md = checkfield(md,'fieldname','smb.accumulation','timeseries',1,'NaN',1,'Inf',1)
if 'BalancethicknessAnalysis' in analyses:
md = checkfield(md,'fieldname','smb.accumulation','size',[md.mesh.numberofvertices],'NaN',1,'Inf',1)
if 'MasstransportAnalysis' in analyses:
md = checkfield(md,'fieldname','smb.runoff','timeseries',1,'NaN',1,'Inf',1)
if 'BalancethicknessAnalysis' in analyses:
md = checkfield(md,'fieldname','smb.runoff','size',[md.mesh.numberofvertices],'NaN',1,'Inf',1)
if 'MasstransportAnalysis' in analyses:
md = checkfield(md,'fieldname','smb.evaporation','timeseries',1,'NaN',1,'Inf',1)
if 'BalancethicknessAnalysis' in analyses:
md = checkfield(md,'fieldname','smb.evaporation','size',[md.mesh.numberofvertices],'NaN',1,'Inf',1)
md = checkfield(md,'fieldname','masstransport.requested_outputs','stringrow',1)
return md
# }}}
[docs] def marshall(self,prefix,md,fid): # {{{
yts=md.constants.yts
WriteData(fid,prefix,'name','md.smb.model','data',2,'format','Integer');
WriteData(fid,prefix,'object',self,'class','smb','fieldname','accumulation','format','DoubleMat','mattype',1,'scale',1./yts,'timeserieslength',md.mesh.numberofvertices+1,'yts',md.constants.yts)
WriteData(fid,prefix,'object',self,'class','smb','fieldname','runoff','format','DoubleMat','mattype',1,'scale',1./yts,'timeserieslength',md.mesh.numberofvertices+1,'yts',md.constants.yts)
WriteData(fid,prefix,'object',self,'class','smb','fieldname','evaporation','format','DoubleMat','mattype',1,'scale',1./yts,'timeserieslength',md.mesh.numberofvertices+1,'yts',md.constants.yts)
#process requested outputs
outputs = self.requested_outputs
indices = [i for i, x in enumerate(outputs) if x == 'default']
if len(indices) > 0:
outputscopy=outputs[0:max(0,indices[0]-1)]+self.defaultoutputs(md)+outputs[indices[0]+1:]
outputs =outputscopy
WriteData(fid,prefix,'data',outputs,'name','md.smb.requested_outputs','format','StringArray')
# }}}