Changeset aa275fc in flex_extract.git for source/python/classes/GribTools.py
- Timestamp:
- Oct 9, 2018, 9:13:15 PM (5 years ago)
- Branches:
- master, ctbto, dev
- Children:
- efba13f
- Parents:
- c5074d2
- File:
-
- 1 edited
Legend:
- Unmodified
- Added
- Removed
-
source/python/classes/GribTools.py
r25b14be raa275fc 44 44 # ------------------------------------------------------------------------------ 45 45 import os 46 from gribapi import grib_new_from_file, grib_is_defined, grib_get, \ 47 grib_release, grib_set, grib_write, grib_index_read, \ 48 grib_index_new_from_file, grib_index_add_file, \ 49 grib_index_write 46 47 from eccodes import (codes_grib_new_from_file, codes_is_defined, codes_get, 48 codes_release, codes_set, codes_write, codes_index_read, 49 codes_index_new_from_file, codes_index_add_file, 50 codes_index_write) 50 51 51 52 # ------------------------------------------------------------------------------ … … 104 105 105 106 while 1: 106 gid _in = grib_new_from_file(fileid)107 108 if gid _inis None:107 gid = codes_new_from_file(fileid) 108 109 if gid is None: 109 110 break 110 111 … … 116 117 i = 0 117 118 for wherekey in wherekeynames: 118 if not grib_is_defined(gid_in, wherekey):119 if not codes_is_defined(gid, wherekey): 119 120 raise Exception("where key was not defined") 120 121 121 122 select = (select and (str(wherekeyvalues[i]) == 122 str( grib_get(gid_in, wherekey))))123 str(codes_get(gid, wherekey)))) 123 124 i += 1 124 125 … … 126 127 llist = [] 127 128 for key in keynames: 128 llist.extend([str( grib_get(gid_in, key))])129 llist.extend([str(codes_get(gid, key))]) 129 130 return_list.append(llist) 130 131 131 grib_release(gid_in)132 codes_release(gid) 132 133 133 134 fileid.close() … … 178 179 179 180 while 1: 180 gid _in = grib_new_from_file(fin)181 182 if gid _inis None:181 gid = codes_new_from_file(fin) 182 183 if gid is None: 183 184 break 184 185 … … 189 190 i = 0 190 191 for wherekey in wherekeynames: 191 if not grib_is_defined(gid_in, wherekey):192 if not codes_is_defined(gid, wherekey): 192 193 raise Exception("where Key was not defined") 193 194 194 195 select = (select and (str(wherekeyvalues[i]) == 195 str( grib_get(gid_in, wherekey))))196 str(codes_get(gid, wherekey)))) 196 197 i += 1 197 198 … … 199 200 i = 0 200 201 for key in keynames: 201 grib_set(gid_in, key, keyvalues[i])202 codes_set(gid, key, keyvalues[i]) 202 203 i += 1 203 204 204 grib_write(gid_in, fout)205 206 grib_release(gid_in)205 codes_write(gid, fout) 206 207 codes_release(gid) 207 208 208 209 fin.close() … … 245 246 246 247 while 1: 247 gid _in = grib_new_from_file(fin)248 249 if gid _inis None:248 gid = codes_new_from_file(fin) 249 250 if gid is None: 250 251 break 251 252 … … 256 257 i = 0 257 258 for key in keynames: 258 if not grib_is_defined(gid_in, key):259 if not codes_is_defined(gid, key): 259 260 raise Exception("Key was not defined") 260 261 261 262 if selectWhere: 262 263 select = (select and (str(keyvalues[i]) == 263 str( grib_get(gid_in, key))))264 str(codes_get(gid, key)))) 264 265 else: 265 266 select = (select and (str(keyvalues[i]) != 266 str( grib_get(gid_in, key))))267 str(codes_get(gid, key)))) 267 268 i += 1 268 269 269 270 if select: 270 grib_write(gid_in, fout)271 272 grib_release(gid_in)271 codes_write(gid, fout) 272 273 codes_release(gid) 273 274 274 275 fin.close() … … 301 302 302 303 if os.path.exists(index_file): 303 iid = grib_index_read(index_file)304 iid = codes_index_read(index_file) 304 305 print("Use existing index file: %s " % (index_file)) 305 306 else: … … 307 308 print("Inputfile: %s " % (filename)) 308 309 if iid is None: 309 iid = grib_index_new_from_file(filename, index_keys)310 iid = codes_index_new_from_file(filename, index_keys) 310 311 else: 311 grib_index_add_file(iid, filename)312 codes_index_add_file(iid, filename) 312 313 313 314 if iid is not None: 314 grib_index_write(iid, index_file)315 codes_index_write(iid, index_file) 315 316 316 317 print('... index done')
Note: See TracChangeset
for help on using the changeset viewer.