Changeset 0934db1 in flex_extract.git for source/python/classes/GribTools.py
- Timestamp:
- Oct 12, 2018, 7:23:41 PM (5 years ago)
- Branches:
- master, ctbto, dev
- Children:
- c97d8ec
- Parents:
- 40408cb
- File:
-
- 1 edited
Legend:
- Unmodified
- Added
- Removed
-
source/python/classes/GribTools.py
raa275fc r0934db1 44 44 # ------------------------------------------------------------------------------ 45 45 import os 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) 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 50 51 52 # from eccodes import (codes_grib_new_from_file, codes_is_defined, codes_get, 53 # codes_release, codes_set, codes_write, codes_index_read, 54 # codes_index_new_from_file, codes_index_add_file, 55 # codes_index_write) 51 56 52 57 # ------------------------------------------------------------------------------ … … 105 110 106 111 while 1: 107 gid = codes_new_from_file(fileid)112 gid = grib_new_from_file(fileid) 108 113 109 114 if gid is None: … … 117 122 i = 0 118 123 for wherekey in wherekeynames: 119 if not codes_is_defined(gid, wherekey):124 if not grib_is_defined(gid, wherekey): 120 125 raise Exception("where key was not defined") 121 126 122 127 select = (select and (str(wherekeyvalues[i]) == 123 str( codes_get(gid, wherekey))))128 str(grib_get(gid, wherekey)))) 124 129 i += 1 125 130 … … 127 132 llist = [] 128 133 for key in keynames: 129 llist.extend([str( codes_get(gid, key))])134 llist.extend([str(grib_get(gid, key))]) 130 135 return_list.append(llist) 131 136 132 codes_release(gid)137 grib_release(gid) 133 138 134 139 fileid.close() … … 179 184 180 185 while 1: 181 gid = codes_new_from_file(fin)186 gid = grib_new_from_file(fin) 182 187 183 188 if gid is None: … … 190 195 i = 0 191 196 for wherekey in wherekeynames: 192 if not codes_is_defined(gid, wherekey):197 if not grib_is_defined(gid, wherekey): 193 198 raise Exception("where Key was not defined") 194 199 195 200 select = (select and (str(wherekeyvalues[i]) == 196 str( codes_get(gid, wherekey))))201 str(grib_get(gid, wherekey)))) 197 202 i += 1 198 203 … … 200 205 i = 0 201 206 for key in keynames: 202 codes_set(gid, key, keyvalues[i])207 grib_set(gid, key, keyvalues[i]) 203 208 i += 1 204 209 205 codes_write(gid, fout)206 207 codes_release(gid)210 grib_write(gid, fout) 211 212 grib_release(gid) 208 213 209 214 fin.close() … … 246 251 247 252 while 1: 248 gid = codes_new_from_file(fin)253 gid = grib_new_from_file(fin) 249 254 250 255 if gid is None: … … 257 262 i = 0 258 263 for key in keynames: 259 if not codes_is_defined(gid, key):264 if not grib_is_defined(gid, key): 260 265 raise Exception("Key was not defined") 261 266 262 267 if selectWhere: 263 268 select = (select and (str(keyvalues[i]) == 264 str( codes_get(gid, key))))269 str(grib_get(gid, key)))) 265 270 else: 266 271 select = (select and (str(keyvalues[i]) != 267 str( codes_get(gid, key))))272 str(grib_get(gid, key)))) 268 273 i += 1 269 274 270 275 if select: 271 codes_write(gid, fout)272 273 codes_release(gid)276 grib_write(gid, fout) 277 278 grib_release(gid) 274 279 275 280 fin.close() … … 302 307 303 308 if os.path.exists(index_file): 304 iid = codes_index_read(index_file)309 iid = grib_index_read(index_file) 305 310 print("Use existing index file: %s " % (index_file)) 306 311 else: … … 308 313 print("Inputfile: %s " % (filename)) 309 314 if iid is None: 310 iid = codes_index_new_from_file(filename, index_keys)315 iid = grib_index_new_from_file(filename, index_keys) 311 316 else: 312 codes_index_add_file(iid, filename)317 grib_index_add_file(iid, filename) 313 318 314 319 if iid is not None: 315 codes_index_write(iid, index_file)320 grib_index_write(iid, index_file) 316 321 317 322 print('... index done')
Note: See TracChangeset
for help on using the changeset viewer.