Changeset 2511 in lmdz_wrf for trunk/tools

Timestamp:

May 6, 2019, 9:21:50 PM (7 years ago)

Author:

lfita

Message:

Adding execption when year < 1900

File:

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

trunk/tools/nc_var_tools.py

@@ -2614 +2614 @@
                 self.firstTt = gen.timeref_datetime(self.refdate, self.firstTu,      \
                   self.units, cal)
-                self.firstTS = self.firstTt.strftime("%Y%m%d%H%M%S")
-                self.firstTm = gen.timeref_datetime_mat(self.refdate, self.firstTu,  \
-                  self.units)
+                if self.firsTt.year >= 1900:
+                    self.firstTS = self.firstTt.strftime("%Y%m%d%H%M%S")
+                    self.firstTm = gen.timeref_datetime_mat(self.refdate,            \
+                      self.firstTu, self.units)
+                else:
+                    print warnmsg
+                    print '    first time step:', self.firsTt, 'below 1900 !!'
+                    self.firstTs = str(self.firstTt.year) + str(self.firstTt.month)+ \
+                       str(self.firstTt.day) + str(self.firstTt.hour) +              \
+                       str(self.firstTt.minute) + str(self.firstTt.second)
                 self.lastTu = timev[self.dimt-1]
                 self.lastTt = gen.timeref_datetime(self.refdate, self.lastTu,        \
                   self.units, cal)
-                self.lastTS = self.lastTt.strftime("%Y%m%d%H%M%S")
-                self.lastTm = gen.timeref_datetime_mat(self.refdate, self.lastTu,    \
-                  self.units)
+                if self.lasTt.year >= 1900:
+                    self.lastTS = self.lastTt.strftime("%Y%m%d%H%M%S")
+                    self.lastTm = gen.timeref_datetime_mat(self.refdate, self.lastTu,\
+                      self.units)
+                else:
+                    print warnmsg
+                    print '    last time step:', self.lastTt, 'below 1900 !!'
+                    self.lastTs = str(self.lastTt.year) + str(self.lastTt.month) +  \
+                       str(self.lastTt.day) + str(self.lastTt.hour) +                \
+                       str(self.lastTt.minute) + str(self.lastTt.second)
 
         return
@@ -23186 +23200 @@
     onc2 = NetCDFFile(filen2, 'r')
 
-    dimn1 = onc1.dimensions
-    dimn2 = onc2.dimensions
+    dimn1 = onc1.dimensions.keys()
+    dimn2 = onc2.dimensions.keys()
     varn1 = onc1.variables.keys()
     varn2 = onc2.variables.keys()
-
-    # Getting dimension variables
-    dimvar = {}
-    
-    for dimvarn in dimvarns:
-        dimn = dimvarn.split('|')[0]
-        dvn = dimvarn.split('|')[1]
-
-        if not gen.searchInlist(dimn1, dimn):
-            print errormsg
-            print '  ' +fname+ ": file '" + filen1 + "' does not have dimension: '"+ \
-              dimn
-            print '    available ones:', dimn1
-            quit(-1)
-
-        if not gen.searchInlist(dimn2, dimn):
-            print errormsg
-            print '  ' +fname+ ": file '" + filen2 + "' does not have dimension: '"+ \
-              dimn
-            print '    available ones:', dimn2
-            quit(-1)
-
-        if not gen.searchInlist(varn1, dvn):
-            print errormsg
-            print '  ' +fname+ ": file '" + filen1 + "' does not have variable: '" + \
-              dimn
-            print '    available ones:', varn1
-            quit(-1)
-
-        if not gen.searchInlist(varn2, dvn):
-            print errormsg
-            print '  ' +fname+ ": file '" + filen2 + "' does not have variable: '" + \
-              dimn
-            print '    available ones:', varn2
-            quit(-1)
-
-        dimvar[dimn] = dvn
 
     if variable == 'all':
@@ -23233 +23210 @@
         setvarn2 = set(varn2)
         coincidentvarn = list(setvarn1.intersection(setvarn2))
+        notcoincidentvarn = list(setvarn1.difference(setvarn2))
+        notcoincidentvarn = notcoincidentvarn + list(setvarn2.difference(setvarn1))
     else:
-        coincidentvarn = gen.str_list(variable, ',')
+        lvariable = gen.str_list(variable, ',')
+        coincidentvarn = []
+        notcoincidentvarn = []
+        for lvarn in lvariable: 
+            if gen.searchInlist(varn2, lvarn): coincidentvarn.append(lvarn)
+            else: notcoincidentvarn.append(lvarn)
+
+    print 'coincident variables:', coincidentvarn
+    print 'not coincident variables:', notcoincidentvarn
 
     # Creation of merged file
     onewnc = NetCDFFile(mergedfilen, 'w')
+
+    # Getting dimension variables
+    dimvar = {}
+    
+    if len(coincidentvarn) != 0:
+        for dimvarn in dimvarns:
+            dimn = dimvarn.split('|')[0]
+            dvn = dimvarn.split('|')[1]
+
+            if not gen.searchInlist(dimn1, dimn):
+                print errormsg
+                print '  ' +fname+ ": file '" + filen1 + "' does not have dimension: '"+ \
+                  dimn
+                print '    available ones:', dimn1
+                #quit(-1)
+
+            if not gen.searchInlist(dimn2, dimn):
+                print errormsg
+                print '  ' +fname+ ": file '" + filen2 + "' does not have dimension: '"+ \
+                  dimn
+                print '    available ones:', dimn2
+                #quit(-1)
+
+            if not gen.searchInlist(varn1, dvn):
+                print errormsg
+                print '  ' +fname+ ": file '" + filen1 + "' does not have variable: '" + \
+                  dimn
+                print '    available ones:', varn1
+                #quit(-1)
+
+            if not gen.searchInlist(varn2, dvn):
+                print errormsg
+                print '  ' +fname+ ": file '" + filen2 + "' does not have variable: '" + \
+                  dimn
+                print '    available ones:', varn2
+                #quit(-1)
+
+            dimvar[dimn] = dvn
 
     # Looping on coincident variables
@@ -23368 +23393 @@
         onewnc.sync()
 
+    # Looping on not coincident variables
+    for varn in notcoincidentvarn:
+        print '  ' + fname + ": merging on variable '" + varn + "' ...."
+        if onc1.variables.has_key(varn):
+            add_vars(onc1,onewnc,[varn])
+            ovar = onc1.variables[varn]
+        elif onc2.variables.has_key(varn):
+            add_vars(onc2,onewnc,[varn])
+        else:
+            print errormsg
+            print '  ' + fname + ": any file has '" + varn + "' variable !!"
+            print "    available ones from file 1 '" + filen1 + "': ", varn1
+            print "    available ones from file 2 '" + filen2 + "': ", varn2
+            quit(-1)
+
+        onewnc.sync()
+
     # Global attributes
     gattrs = onc1.ncattrs()
@@ -23376 +23418 @@
 
     # Adding PyNCplot
-    add_global_PyNCplot(onewnc, main, fname, '0.1')
+    add_global_PyNCplot(onewnc, main, fname, '0.2')
     onewnc.sync()