Changeset 870 for cp4cds-tmp/simChk1.py


Ignore:
Timestamp:
15/11/16 17:51:02 (3 years ago)
Author:
mjuckes
Message:

added code to generate list of files for models with basic availability

File:
1 edited

Legend:

Unmodified
Added
Removed
  • cp4cds-tmp/simChk1.py

    r868 r870  
    1616    l10 = glob.glob( '%s/*' % base ) 
    1717    l1 = [i for i in l10 if os.path.isdir( i ) ] 
     18    self.files = dict() 
    1819 
    1920    self.im = [] 
     
    2122    for i in sorted( l1 ): 
    2223      self.addModels( i ) 
     24 
     25    si = set() 
     26    sm = set() 
     27    for i,m in self.im: 
     28      si.add(i) 
     29      sm.add(m) 
     30    print sorted( list( si ) ) 
     31    print sorted( list( sm ) ) 
    2332    print self.im 
    2433    self.filterModels(['piControl','historical','rcp45']) 
     
    5463    print sok 
    5564    print sokk 
     65    oo = open( 'fileList.txt', 'w' ) 
     66    for e,i,m,rip in self.files: 
     67       if (i,m) in sok: 
     68         for f in self.files[ (e,i,m,rip) ]: 
     69           oo.write( f + '\n' ) 
     70    oo.close() 
    5671 
    5772    print 'Models: %s; with historical, rcp45 and piControl: %s; with top variables in 1 simulation: %s; in 3: %s' % (n0,n1,n2,n3) 
     
    7085      eok = set() 
    7186      for e in ens: 
     87        efl = [] 
    7288        missing = set() 
    7389        for t,v in vars1: 
     
    7591          if os.path.isdir( '%s%s/%s/%s/%s/%s/latest/%s' % (base,inst,model,expt,p,e,v) ): 
    7692            fl =  glob.glob( '%s%s/%s/%s/%s/%s/latest/%s/*.nc' % (base,inst,model,expt,p,e,v) ) 
     93            for f in fl: 
     94              efl.append(f) 
    7795          else: 
    7896            missing.add( (t,v) ) 
    7997        if len(missing) == 0: 
    8098          eok.add(e) 
     99          self.files[ (expt,inst,model,e) ] = efl 
    81100      if len( eok ) == 0: 
    82101        return -1 
Note: See TracChangeset for help on using the changeset viewer.