Changeset 442 in lmdz_wrf

Timestamp:

May 28, 2015, 1:20:19 PM (10 years ago)

Author:

lfita

Message:

Adding WRFtime' on RAINTOT' calculation

File:

• trunk/tools/diagnostics.py (modified) (2 diffs)

trunk/tools/diagnostics.py

@@ -12 +12 @@
 import re
 import nc_var_tools as ncvar
+import datetime as dt
 
 main = 'diagnostics.py'
@@ -979 +980 @@
 
 # Transforming to a flux
-        dtimeunits = var2.getncattr('units')
-        tunits = dtimeunits.split(' ')[0]
-
-        dtime = (var2[1] - var2[0])*timeunits_seconds(tunits)
+        if depvarsp[2] != 'WRFtime':
+            dtimeunits = var2.getncattr('units')
+            tunits = dtimeunits.split(' ')[0]
+
+            dtime = (var2[1] - var2[0])*timeunits_seconds(tunits)
+        else:
+            time1 = var2[0,:]
+            time2 = var2[1,:]
+            dtdate1 = dt.datetime.strptime(time1,"%Y-%m-%d_%H:%M:%S")
+            dtdate2 = dt.datetime.strptime(time2,"%Y-%m-%d_%H:%M:%S")
+            diffdate12 = ddate2 - ddate1
+            dtime = diffdate12.total_seconds()
+
         ncvar.insert_variable(ncobj, 'pr', diagout/dtime, diagoutd, diagoutvd, newnc)