Changeset 1941 in lmdz_wrf

Timestamp:

Jul 17, 2018, 9:23:44 PM (7 years ago)

Author:

lfita

Message:

Update:

• Consecutive time-steps
• New individual 1D variables

File:

• trunk/tools/UWyoming_snd_nc.py (modified) (7 diffs)

trunk/tools/UWyoming_snd_nc.py

@@ -54 +54 @@
 txtTXTvals = ['Station identifier', 'Observation time']
 
+# Float values for global attributes:
+floatTXTvals = ['Station elevation', 'Station longitude', 'Station latitude']
+
 # Not lower pressure variables
 NOTlowpres = ['DWPT', 'RELH', 'MIXR', 'THTE']
@@ -73 +76 @@
 ## MAIN
     #######
+
+# Global attributes for the station
+stngattrk = intTXTvals + txtTXTvals + floatTXTvals
 
 if opts.debug is None:
@@ -213 +219 @@
         print ic, ':', txtvals[ic]
 
+Tsoundings.append(stationdateS)
 soundings[stationdateS] = [pvals, txtvals]
 if len(soundings.keys()) == 1:
@@ -263 +270 @@
     print sndvals
 
-# Removing not comoputed values
+# Removing not computed values
+statglobalattr = {}
 txtn = list(txtvals.keys())
+snditS = Tsoundings[0]
+sndvs = soundings[snditS]
+sndc = sndvs[1]
+
 for Sn in intTXTvals:
-    if gen.searchInlist(txtn, Sn): txtn.remove(Sn)
+    SnS = Sn.replace(' ','_')
+    if gen.searchInlist(txtn, Sn):
+        if sndc.has_key(Sn): statglobalattr[SnS] = int(sndc[Sn])
+        else: statglobalattr[SnS] = '-'
+        txtn.remove(Sn)
+    else:
+        if sndc.has_key(Sn): statglobalattr[SnS] = int(sndc[Sn])
+        else: statglobalattr[SnS] = '-'
+
 for Sn in txtTXTvals:
-    if gen.searchInlist(txtn, Sn): txtn.remove(Sn)
+    SnS = Sn.replace(' ','_')
+    if gen.searchInlist(txtn, Sn):
+        if sndc.has_key(Sn): statglobalattr[SnS] = sndc[Sn]
+        else: statglobalattr[SnS] = '-'
+        txtn.remove(Sn)
+    else:
+        if sndc.has_key(Sn): statglobalattr[SnS] = int(sndc[Sn])
+        else: statglobalattr[SnS] = '-'
+
+for Sn in floatTXTvals:
+    SnS = Sn.replace(' ','_')
+    if gen.searchInlist(txtn, Sn): 
+        if sndc.has_key(Sn): statglobalattr[SnS] = np.float(sndc[Sn])
+        else: statglobalattr[Sn] = '-'
+        txtn.remove(Sn)
+    else:
+        if sndc.has_key(Sn): statglobalattr[SnS] = int(sndc[Sn])
+        else: statglobalattr[SnS] = '-'
 
 Ncomp = len(txtn)
@@ -291 +328 @@
 # Creation of dimensions
 onewnc.createDimension('pres', Npres)
-onewnc.createDimension('cvals', Ncomp)
 onewnc.createDimension('time', None)
-onewnc.createDimension('Lstring', Lstring)
+#onewnc.createDimension('cvals', Ncomp)
+#onewnc.createDimension('Lstring', Lstring)
 
 # Creation of variable-dimensions 
@@ -300 +337 @@
 ncvar.basicvardef(newvar, 'pres', 'pressure', varu['PRES'])
 
-newvar = onewnc.createVariable('cvals', 'c', ('cvals', 'Lstring'))
-ncvar.writing_str_nc(newvar, txtn, Lstring)
-ncvar.basicvardef(newvar, 'cvals', 'computed values from sounding data', 'hPa')
+# No more computedvalues matrix !
+#newvar = onewnc.createVariable('cvals', 'c', ('cvals', 'Lstring'))
+#ncvar.writing_str_nc(newvar, txtn, Lstring)
+#ncvar.basicvardef(newvar, 'cvals', 'computed values from sounding data', 'hPa')
 
 newvar = onewnc.createVariable('time', 'f8', ('time'))
@@ -331 +369 @@
 onewnc.sync()
 
-newvar = onewnc.createVariable('computedvals', 'f', ('time', 'cvals'),               \
-  fill_value=gen.fillValueF)
-newvar[:] = compvals
-ncvar.basicvardef(newvar, 'computedvals', 'values computed from sounding data', '-')
-onewnc.sync()
+# No more computedvalues matrix !
+#newvar = onewnc.createVariable('computedvals', 'f', ('time', 'cvals'),               \
+#  fill_value=gen.fillValueF)
+#newvar[:] = compvals
+#ncvar.basicvardef(newvar, 'computedvals', 'values computed from sounding data', '-')
+#onewnc.sync()
+
+# Getting specific 1D values
+for ic in range(len(txtn)):
+    Sn = txtn[ic]
+    CFvalues = gen.variables_values(Sn)
+    newvar=onewnc.createVariable(CFvalues[0],'f', ('time'), fill_value=gen.fillValueF)
+    newvar[:] = compvals[:,ic]
+    ncvar.basicvardef(newvar, CFvalues[1], CFvalues[4].replace('|',' '), CFvalues[5])
+    onewnc.sync()
 
 # Global attributes
-ncvar.add_global_PyNCplot(onewnc, main, '', '0.1')
+ncvar.add_global_PyNCplot(onewnc, main, '', '0.2')
+for atn in stngattrk:
+    atnS = atn.replace(' ','_')
+    onewnc.setncattr(atnS, statglobalattr[atnS])
 
 onewnc.close()