Changeset 2e62398 in flex_extract.git for source/python/classes/GribTools.py

Timestamp:

Nov 16, 2018, 8:31:15 PM (5 years ago)

Author:

Anne Philipp <anne.philipp@…>

Branches:

master, ctbto, dev

Children:

403cbf1

Parents:

adb0b63

Message:

moved from grib_api to eccodes

File:

• source/python/classes/GribTools.py (modified) (11 diffs)

source/python/classes/GribTools.py

@@ -44 +44 @@
 # ------------------------------------------------------------------------------
 import os
-from gribapi import grib_new_from_file, grib_is_defined, grib_get, \
-     grib_release, grib_set, grib_write, grib_index_read, \
-     grib_index_new_from_file, grib_index_add_file,  \
-     grib_index_write
-
-
-# from eccodes import (codes_grib_new_from_file, codes_is_defined, codes_get,
-                     # codes_release, codes_set, codes_write, codes_index_read,
-                     # codes_index_new_from_file, codes_index_add_file,
-                     # codes_index_write)
+
+from eccodes import (codes_grib_new_from_file, codes_is_defined, codes_get,
+                     codes_release, codes_set, codes_write, codes_index_read,
+                     codes_index_new_from_file, codes_index_add_file,
+                     codes_index_write)
 
 # ------------------------------------------------------------------------------
@@ -109 +104 @@
 
         while 1:
-            gid = grib_new_from_file(fileid)
+            gid = codes_new_from_file(fileid)
 
             if gid is None:
@@ -121 +116 @@
             i = 0
             for wherekey in wherekeynames:
-                if not grib_is_defined(gid, wherekey):
+                if not codes_is_defined(gid, wherekey):
                     raise Exception("where key was not defined")
 
                 select = (select and (str(wherekeyvalues[i]) ==
-                                      str(grib_get(gid, wherekey))))
+                                      str(codes_get(gid, wherekey))))
                 i += 1
 
@@ -131 +126 @@
                 llist = []
                 for key in keynames:
-                    llist.extend([str(grib_get(gid, key))])
+                    llist.extend([str(codes_get(gid, key))])
                 return_list.append(llist)
 
-            grib_release(gid)
+            codes_release(gid)
 
         fileid.close()
@@ -182 +177 @@
 
         while 1:
-            gid = grib_new_from_file(fin)
+            gid = codes_new_from_file(fin)
 
             if gid is None:
@@ -193 +188 @@
             i = 0
             for wherekey in wherekeynames:
-                if not grib_is_defined(gid, wherekey):
+                if not codes_is_defined(gid, wherekey):
                     raise Exception("where Key was not defined")
 
                 select = (select and (str(wherekeyvalues[i]) ==
-                                      str(grib_get(gid, wherekey))))
+                                      str(codes_get(gid, wherekey))))
                 i += 1
 
@@ -203 +198 @@
                 i = 0
                 for key in keynames:
-                    grib_set(gid, key, keyvalues[i])
+                    codes_set(gid, key, keyvalues[i])
                     i += 1
 
-            grib_write(gid, fout)
-
-            grib_release(gid)
+            codes_write(gid, fout)
+
+            codes_release(gid)
 
         fin.close()
@@ -250 +245 @@
 
         while 1:
-            gid = grib_new_from_file(fin)
+            gid = codes_new_from_file(fin)
 
             if gid is None:
@@ -261 +256 @@
             i = 0
             for key in keynames:
-                if not grib_is_defined(gid, key):
+                if not codes_is_defined(gid, key):
                     raise Exception("Key was not defined")
 
                 if selectWhere:
                     select = (select and (str(keyvalues[i]) ==
-                                          str(grib_get(gid, key))))
+                                          str(codes_get(gid, key))))
                 else:
                     select = (select and (str(keyvalues[i]) !=
-                                          str(grib_get(gid, key))))
+                                          str(codes_get(gid, key))))
                 i += 1
 
             if select:
-                grib_write(gid, fout)
-
-            grib_release(gid)
+                codes_write(gid, fout)
+
+            codes_release(gid)
 
         fin.close()
@@ -306 +301 @@
 
         if os.path.exists(index_file):
-            iid = grib_index_read(index_file)
+            iid = codes_index_read(index_file)
             print("Use existing index file: %s " % (index_file))
         else:
@@ -312 +307 @@
                 print("Inputfile: %s " % (filename))
                 if iid is None:
-                    iid = grib_index_new_from_file(filename, index_keys)
+                    iid = codes_index_new_from_file(filename, index_keys)
                 else:
-                    grib_index_add_file(iid, filename)
+                    codes_index_add_file(iid, filename)
 
             if iid is not None:
-                grib_index_write(iid, index_file)
+                codes_index_write(iid, index_file)
 
         print('... index done')