Versions Compared

Key

  • This line was added.
  • This line was removed.
  • Formatting was changed.

...

Code Block
languagepy
linenumberstrue
# Copyright 2020 ECMWF.
#
# This software is licensed under the terms of the Apache Licence Version 2.0
# which can be obtained at http://www.apache.org/licenses/LICENSE-2.0.
#
# In applying this licence, ECMWF does not waive the privileges and immunities
# granted to it by virtue of its status as an intergovernmental organisation
# nor does it submit to any jurisdiction.
 
from __future__ import print_functionsys
import traceback
import sys

from eccodes import *
 
INPUT = "../../data/spherical_model_level.grib2"
VERBOSE = 1  # verbose error reporting
 
def example():
    f = open(INPUT, "rb")
 
    while 1:
        gid = codes_grib_new_from_file(f)
        if gid is None:
            break

        # Get the values. This will be a numpy array
        values = codes_get_array(gid,"values")
 
        # Store the real and imaginary parts of the coefficients in arrays a and b, respectively
        # The real parts 'a' stored in every second element starting at element 0
        a = values[0::2]
        # The imaginary parts 'b' stored in every second element starting at element 1
        b = values[1::2]
  
        codes_release(gid)

        # Loop through the values and print the m and n indices together with the corresponding
        # real and imaginary parts of the coefficients  
        m = 0
        n = 0
        for i in range(len(a)):
            print("%d\t n=%d\t m=%d %.10f\t%.10f" % (i, n, m, a[i], b[i]))
            n += 1
            if n > T:
                m += 1
                n = m 
 
    f.close()
 
def main():
    try:
        example()
    except CodesInternalError as err:
        if VERBOSE:
            traceback.print_exc(file=sys.stderr)
        else:
            sys.stderr.write(err.msg + "\n")
 
        return 1
 
 
if __name__ == "__main__":
    sys.exit(main())

...