source: CMIP6dreqbuild/trunk/src/framework/dreqPy/scope_utils.py @ 1254

Subversion URL: http://proj.badc.rl.ac.uk/svn/exarch/CMIP6dreqbuild/trunk/src/framework/dreqPy/scope_utils.py@1254
Revision 1254, 2.4 KB checked in by mjuckes, 3 years ago (diff)

fixed --txt option

Line 
1import collections, string
2try:
3  import table_utils
4except:
5  import dreqPy.table_utils as table_utils
6
7##NT_txtopts = collections.namedtuple( 'txtopts', ['mode'] )
8
9class c1(object):
10  def __init__(self):
11    self.a = collections.defaultdict( int )
12
13class xlsTabs(object):
14  """used in scope.py; uses table_utils.py"""
15  def __init__(self,sc,tiermax=1,pmax=1,xls=True, txt=False, txtOpts=None, odir='xls'):
16    self.pmax=pmax
17    self.tiermax=tiermax
18    self.sc = sc
19    sc.setTierMax( tiermax )
20    self.cc = collections.defaultdict( c1 )
21    self.dq = sc.dq
22    self.doXls = xls
23    self.doTxt = txt
24
25    self.mips = ['AerChemMIP', 'C4MIP', 'CFMIP', 'DAMIP', 'DCPP', 'FAFMIP', 'GeoMIP', 'GMMIP', 'HighResMIP', 'ISMIP6', 'LS3MIP', 'LUMIP', 'OMIP', 'PAMIP', 'PMIP', 'RFMIP', 'ScenarioMIP', 'VolMIP', 'CORDEX', 'DynVar', 'SIMIP', 'VIACSAB']
26    self.mipsp = ['DECK','CMIP6',] + self.mips[:-4]
27
28    print 'xlsTabs ',txt
29    self.tabs = table_utils.tables( sc, xls=xls, txt=txt, txtOpts=txtOpts, odir=odir )
30
31  def run(self,m,colCallback=None,verb=False,mlab=None,exid=None):
32      if m == 'TOTAL':
33        l1 = self.sc.rqiByMip( set( self.mips ) )
34      else:
35        l1 = self.sc.rqiByMip( m )
36
37      if mlab == None:
38        mlab = m
39
40      verb = False
41      if verb:
42        print ( 'r1: m=%s, len(l1)=%s' % (mlab,len(l1)) )
43
44      self.cc[mlab].dd = {}
45      self.cc[mlab].ee = {}
46      self.tabs.accReset()
47      vcc = collections.defaultdict( int )
48      for m2 in self.mipsp + ['TOTAL',]:
49        if m2 == 'TOTAL':
50          xx = self.dq.coll['experiment'].items
51        else:
52          xx = [i for i in self.dq.coll['experiment'].items if i.mip == m2]
53        if exid != None:
54          xxx = [i for i in xx if i.uid == exid]
55          if len(xxx) == 0:
56            break
57          xx = xxx
58        self.cc[mlab].ee[m2] = xx
59        xxi = set( [i.label for i in xx] )
60##
61## need to check this option, and add a template for a view summarising the experiments for each mip-mip combinations
62##
63        if m2 != 'TOTAL':
64          for i in xx:
65            self.tabs.doTable(m,l1,i.uid,self.pmax,self.cc,acc=False,cc=vcc,exptids=xxi,mlab=None)
66
67        self.tabs.doTable(m,l1,m2,self.pmax,self.cc,cc=vcc,exptids=xxi,mlab=None)
68
69        if verb:
70          print ( 'r1: mlab=%s,m2=%s, len(l1)=%s, len(xxi)=%s' % (mlab,m2,len(l1),len(xxi)) )
71
72        if colCallback != None:
73          colCallback( m,m2,mlab=mlab )
74
75
Note: See TracBrowser for help on using the repository browser.