Changeset 648 in lmdz_wrf

Timestamp:

Sep 23, 2015, 11:23:57 AM (9 years ago)

Author:

lfita

Message:

Adding 'stations-map' as multiple stations at the same time-step

File:

• trunk/tools/create_OBSnetcdf.py (modified) (8 diffs)

trunk/tools/create_OBSnetcdf.py

@@ -216 +216 @@
             newval = None
         else:
-            print 'op: **' + op + '**'
             if op != '-':
                 opern = op.split(',')[0]
@@ -816 +815 @@
   " and time units: 'weeks', 'days', 'hours', 'miuntes', 'seconds')"
 
-kindobs=['multi-points', 'single-station', 'trajectory']
+kindobs=['stations-map','multi-points', 'single-station', 'trajectory']
 strkObs="kind of observations; 'multi-points': multiple individual punctual obs " +  \
   "(e.g., lightning strikes), 'single-station': single station on a fixed position,"+\
@@ -963 +962 @@
 # Total number of values
 Ntvalues = len(datavalues[description['varN'][0]])
-print main + ': total temporal values found:',Ntvalues
+if obskind == 'stations-map':
+    print main + ': total values found:',Ntvalues
+else:
+    print main + ': total temporal values found:',Ntvalues
 
 objfile = NetCDFFile(ofile, 'w')
@@ -969 +971 @@
 # Creation of dimensions
 ##
-objfile.createDimension('time',None)
+if obskind == 'stations-map':
+    rowsdim = 'Npoints'
+    dimlength = Ntvalues
+else:
+    rowsdim = 'time'
+    dimlength = None
+
+objfile.createDimension(rowsdim,dimlength)
 objfile.createDimension('StrLength',StringLength)
 
@@ -979 +988 @@
 
     if formats[ivar] == 'D':
-        newvar = objfile.createVariable(varn, 'f32', ('time'), fill_value=fillValueF)
+        newvar = objfile.createVariable(varn, 'f32', (rowsdim), fill_value=fillValueF)
         basicvardef(newvar, varn, description['varLN'][ivar],                        \
           description['varU'][ivar])
         newvar[:] = np.where(datavalues[varn] is None, fillValueF, datavalues[varn])
     elif formats[ivar] == 'F':
-        newvar = objfile.createVariable(varn, 'f', ('time'), fill_value=fillValueF)
+        newvar = objfile.createVariable(varn, 'f', (rowsdim), fill_value=fillValueF)
         basicvardef(newvar, varn, description['varLN'][ivar],                        \
           description['varU'][ivar])
         newvar[:] = np.where(datavalues[varn] is None, fillValueF, datavalues[varn])
     elif formats[ivar] == 'I':
-        newvar = objfile.createVariable(varn, 'i', ('time'), fill_value=fillValueI)
+        newvar = objfile.createVariable(varn, 'i', (rowsdim), fill_value=fillValueI)
         basicvardef(newvar, varn, description['varLN'][ivar],                        \
           description['varU'][ivar])
@@ -998 +1007 @@
         newvar[:] = vals
     elif formats[ivar] == 'S':
-        newvar = objfile.createVariable(varn, 'c', ('time','StrLength'))
+        newvar = objfile.createVariable(varn, 'c', (rowsdim,'StrLength'))
         basicvardef(newvar, varn, description['varLN'][ivar],                        \
           description['varU'][ivar])
@@ -1058 +1067 @@
     basicvardef(newvar, timeCFname, 'time', tunits )
 else:
+    newdim = objfile.createDimension('time',None)
     timeCFname = 'time'
     newvar = objfile.createVariable( timeCFname, 'f8', ('time'))
@@ -1063 +1073 @@
 
 set_attribute(newvar, 'calendar', 'standard')
-newvar[:] = timevals
+if obskind == 'stations-map':
+    newvar[:] = timevals[0]
+else:
+    newvar[:] = timevals
 
 # Global attributes