Changeset 492 in lmdz_wrf for trunk/tools

Timestamp:

Jun 15, 2015, 12:14:14 PM (10 years ago)

Author:

lfita

Message:

Adding description of the station

File:

• trunk/tools/validation_sim.py (modified) (5 diffs)

trunk/tools/validation_sim.py

@@ -985 +985 @@
         return self.__values[elem]
 
+def adding_station_desc(onc,stdesc):
+    """ Function to add a station description in a netCDF file
+      onc= netCDF object
+      stdesc= station description lon, lat, height
+    """
+    fname = 'adding_station_desc'
+
+    newvar = objfile.createVariable( 'station', 'c', ('StrLength'))
+    newvar[0:len(stdesc[0])] = stdesc[0]
+
+    newdim = onc.createDimension('nst',1)
+
+    if onc.variables.has_key('lon'):
+        print warnmsg
+        print '  ' + fname + ": variable 'lon' already exist !!"
+        print "    renaming it as 'lonst'"
+        lonname = 'lonst'
+    else:
+        lonname = 'lon'
+
+    newvar = objfile.createVariable( lonname, 'f4', ('nst'))
+    basicvardef(newvar, lonname, 'longitude', 'degrees_West' )
+    newvar[:] = stdesc[1]
+
+    if onc.variables.has_key('lat'):
+        print warnmsg
+        print '  ' + fname + ": variable 'lat' already exist !!"
+        print "    renaming it as 'latst'"
+        latname = 'latst'
+    else:
+        latname = 'lat'
+
+    newvar = objfile.createVariable( latname, 'f4', ('nst'))
+    basicvardef(newvar, lonname, 'latitude', 'degrees_North' )
+    newvar[:] = stdesc[2]
+
+    if onc.variables.has_key('height'):
+        print warnmsg
+        print '  ' + fname + ": variable 'height' already exist !!"
+        print "    renaming it as 'heightst'"
+        heightname = 'heightst'
+    else:
+        heightname = 'height'
+
+    newvar = objfile.createVariable( heightname, 'f4', ('nst'))
+    basicvardef(newvar, heightname, 'height above sea level', 'm' )
+    newvar[:] = stdesc[3]
+
+    return
+
 ####### ###### ##### #### ### ## #
 
@@ -1029 +1079 @@
   help=strkObs, metavar="FILE")
 parser.add_option("-l", "--stationLocation", dest="stloc",  
-  help="longitude, latitude and height of the station (only for 'single-station')", 
+  help="name, longitude, latitude and height of the station (only for 'single-station')", 
   metavar="FILE")
 parser.add_option("-o", "--observation", dest="fobs",
@@ -1106 +1156 @@
             quit(-1)
         else:
-            stationdesc = [np.float(opts.stloc.split(',')[0]),                       \
-              np.float(opts.stloc.split(',')[1]), np.float(opts.stloc.split(',')[2])]
+            stationdesc = [opts.stloc.split(',')[0],                                 \
+              np.float(opts.stloc.split(',')[1]), np.float(opts.stloc.split(',')[2]),\
+              np.float(opts.stloc.split(',')[3])]
 
 if opts.fobs is None:
@@ -1176 +1227 @@
     else:
         if dn == 'X':
-            valdimobs[dn] = stationdesc[0]
+            valdimobs[dn] = stationdesc[1]
         elif dn == 'Y':
-            valdimobs[dn] = stationdesc[1]
+            valdimobs[dn] = stationdesc[2]
         elif dn == 'Z':
-            valdimobs[dn] = stationdesc[2]
+            valdimobs[dn] = stationdesc[3]
 
 osim = NetCDFFile(opts.fsim, 'r')
@@ -1728 +1779 @@
 newvar[:] = trjsim.transpose()
 
+# Adding three variables with the station location, longitude, latitude and height
+if obskind == 'single-station':
+    adding_station_desc(objfile,stationdesc)
+
 # Global attributes
 ##