#!/usr/bin/env pytest
# -*- coding: utf-8 -*-
###############################################################################
# $Id$
#
# Project:  GDAL/OGR Test Suite
# Purpose:  Test GRIB driver.
# Author:   Frank Warmerdam <warmerdam@pobox.com>
#
###############################################################################
# Copyright (c) 2008, Frank Warmerdam <warmerdam@pobox.com>
# Copyright (c) 2008-2012, Even Rouault <even dot rouault at spatialys.com>
#
# Permission is hereby granted, free of charge, to any person obtaining a
# copy of this software and associated documentation files (the "Software"),
# to deal in the Software without restriction, including without limitation
# the rights to use, copy, modify, merge, publish, distribute, sublicense,
# and/or sell copies of the Software, and to permit persons to whom the
# Software is furnished to do so, subject to the following conditions:
#
# The above copyright notice and this permission notice shall be included
# in all copies or substantial portions of the Software.
#
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
# OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
# THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
# FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
# DEALINGS IN THE SOFTWARE.
###############################################################################

import os
import shutil
import struct

import gdaltest
import pytest

from osgeo import gdal, osr

pytestmark = pytest.mark.require_driver("GRIB")


def has_jp2kdrv():
    for i in range(gdal.GetDriverCount()):
        if gdal.GetDriver(i).ShortName.startswith("JP2"):
            return True
    return False


###############################################################################
# Do a simple checksum on our test file


def test_grib_1():

    tst = gdaltest.GDALTest("GRIB", "grib/ds.mint.bin", 2, 46927)
    return tst.testOpen()


###############################################################################
# Test a small GRIB 1 sample file.


def test_grib_2():

    tst = gdaltest.GDALTest("GRIB", "grib/Sample_QuikSCAT.grb", 4, 50714)
    tst.testOpen()

    ds = gdal.Open("data/grib/Sample_QuikSCAT.grb")
    assert ds.GetRasterBand(1).GetNoDataValue() == 9999.0
    assert (
        ds.GetRasterBand(1).GetNoDataValue() == 9999.0
    )  # do it again to test correct caching


###############################################################################
# This file has different raster sizes for some of the products, which
# we sort-of-support per ticket Test a small GRIB 1 sample file.


def test_grib_read_different_sizes_messages():

    tst = gdaltest.GDALTest("GRIB", "grib/bug3246.grb", 4, 4081)
    gdal.PushErrorHandler("CPLQuietErrorHandler")
    result = tst.testOpen()
    gdal.PopErrorHandler()

    msg = gdal.GetLastErrorMsg()
    if msg.find("data access may be incomplete") == -1 or gdal.GetLastErrorType() != 2:
        gdaltest.post_reason("did not get expected warning.")

    return result


###############################################################################
# Check nodata


def test_grib_grib2_read_nodata():

    ds = gdal.Open("data/grib/ds.mint.bin")
    assert ds.GetRasterBand(1).GetNoDataValue() == 9999
    assert ds.GetRasterBand(2).GetNoDataValue() == 9999
    md = ds.GetRasterBand(1).GetMetadata()
    expected_md = {
        "GRIB_REF_TIME": "1203613200",
        "GRIB_PDS_TEMPLATE_ASSEMBLED_VALUES": "0 5 2 0 0 255 255 1 19 1 0 0 255 -1 -2147483647 2008 2 22 12 0 0 1 0 3 255 1 12 1 0",
        "GRIB_VALID_TIME": "1203681600",
        "GRIB_FORECAST_SECONDS": "68400",
        "GRIB_UNIT": "[C]",
        "GRIB_PDS_TEMPLATE_NUMBERS": "0 5 2 0 0 0 255 255 1 0 0 0 19 1 0 0 0 0 0 255 129 255 255 255 255 7 216 2 22 12 0 0 1 0 0 0 0 3 255 1 0 0 0 12 1 0 0 0 0",
        "GRIB_DISCIPLINE": "0(Meteorological)",
        "GRIB_PDS_PDTN": "8",
        "GRIB_COMMENT": "Minimum temperature [C]",
        "GRIB_SHORT_NAME": "0-SFC",
        "GRIB_ELEMENT": "MinT",
    }
    for k in expected_md:
        assert k in md and md[k] == expected_md[k], "Did not get expected metadata"


###############################################################################
# Check grib units (#3606)


def test_grib_read_units():

    gdal.Unlink("tmp/ds.mint.bin.aux.xml")

    shutil.copy("data/grib/ds.mint.bin", "tmp/ds.mint.bin")
    ds = gdal.Open("tmp/ds.mint.bin")
    md = ds.GetRasterBand(1).GetMetadata()
    assert md["GRIB_UNIT"] == "[C]"
    assert md["GRIB_COMMENT"] == "Minimum temperature [C]"
    ds.GetRasterBand(1).ComputeStatistics(False)
    assert ds.GetRasterBand(1).GetMinimum() == pytest.approx(13, abs=1)
    ds = None

    os.unlink("tmp/ds.mint.bin.aux.xml")

    with gdaltest.config_option("GRIB_NORMALIZE_UNITS", "NO"):
        ds = gdal.Open("tmp/ds.mint.bin")
        ds.GetRasterBand(1).ComputeStatistics(False)
    md = ds.GetRasterBand(1).GetMetadata()
    assert md["GRIB_UNIT"] == "[K]"
    assert md["GRIB_COMMENT"] == "Minimum temperature [K]"
    assert ds.GetRasterBand(1).GetMinimum() == pytest.approx(286, abs=1)
    ds = None

    gdal.GetDriverByName("GRIB").Delete("tmp/ds.mint.bin")


###############################################################################
# Handle geotransform for 1xn or nx1 grids.  The geotransform was faulty when
# grib files had one cell in either direction for geographic projections.  See
# ticket #5532


def test_grib_read_geotransform_one_n_or_n_one():

    ds = gdal.Open("data/grib/one_one.grib2")
    egt = (-114.25, 0.5, 0.0, 47.250, 0.0, -0.5)
    gt = ds.GetGeoTransform()
    ds = None
    assert gt == egt


###############################################################################
# This is more a /vsizip/ file test than a GRIB one, but could not easily
# come up with a pure /vsizip/ test case, so here's a real world use
# case (#5530).


def test_grib_read_vsizip():

    ds = gdal.Open(
        "/vsizip/data/grib/gfs.t00z.mastergrb2f03.zip/gfs.t00z.mastergrb2f03"
    )
    assert ds is not None


###############################################################################
# Write PDS numbers to all bands


def test_grib_grib2_test_grib_pds_all_bands():
    ds = gdal.Open(
        "/vsizip/data/grib/gfs.t00z.mastergrb2f03.zip/gfs.t00z.mastergrb2f03"
    )
    assert ds is not None
    band = ds.GetRasterBand(2)
    md = band.GetMetadataItem("GRIB_PDS_TEMPLATE_NUMBERS")
    ds = None
    assert md is not None, "Failed to fetch pds numbers (#5144)"

    with gdaltest.config_option("GRIB_PDS_ALL_BANDS", "OFF"):
        ds = gdal.Open(
            "/vsizip/data/grib/gfs.t00z.mastergrb2f03.zip/gfs.t00z.mastergrb2f03"
        )
        assert ds is not None
        band = ds.GetRasterBand(2)
        md = band.GetMetadataItem("GRIB_PDS_TEMPLATE_NUMBERS")
        ds = None
        assert md is None, "Got pds numbers, when disabled (#5144)"


###############################################################################
# Test support for template 4.15 (#5768)


def test_grib_grib2_read_template_4_15():

    import test_cli_utilities

    if test_cli_utilities.get_gdalinfo_path() is None:
        pytest.skip()

    ret, err = gdaltest.runexternal_out_and_err(
        test_cli_utilities.get_gdalinfo_path()
        + " data/grib/template_4_15.grb2 -checksum"
    )

    # This is a JPEG2000 compressed file, so just check we can open it or that we get a message saying there's no JPEG2000 driver available
    assert (
        "Checksum=" in ret or "Is the JPEG2000 driver available?" in err
    ), "Could not open file"


###############################################################################
# Test support for PNG compressed


def test_grib_grib2_read_png():

    if gdal.GetDriverByName("PNG") is None:
        pytest.skip()

    ds = gdal.Open("data/grib/MRMS_EchoTop_18_00.50_20161015-133230.grib2")
    cs = ds.GetRasterBand(1).Checksum()
    assert cs == 41854, "Could not open file"


###############################################################################
# Test support for GRIB2 Section 4 Template 32, Analysis or forecast at a horizontal level or in a horizontal layer at a point in time for synthetic satellite data.


def test_grib_grib2_read_template_4_32():

    # First band extracted from http://nomads.ncep.noaa.gov/pub/data/nccf/com/hur/prod/hwrf.2017102006/twenty-se27w.2017102006.hwrfsat.core.0p02.f000.grb2
    ds = gdal.Open(
        "data/grib/twenty-se27w.2017102006.hwrfsat.core.0p02.f000_truncated.grb2"
    )
    cs = ds.GetRasterBand(1).Checksum()
    assert cs == 48230, "Could not open file"
    assert ds.GetRasterBand(1).ComputeRasterMinMax() == pytest.approx(
        (-9.765, 2.415), 1e-3
    )  # Reasonable range for Celcius
    md = ds.GetRasterBand(1).GetMetadata()
    expected_md = {
        "GRIB_REF_TIME": "1508479200",
        "GRIB_VALID_TIME": "1508479200",
        "GRIB_FORECAST_SECONDS": "0",
        "GRIB_UNIT": "[C]",
        "GRIB_PDS_TEMPLATE_NUMBERS": "5 7 2 0 0 0 0 0 1 0 0 0 0 1 0 31 1 29 67 140 2 0 0 238 217",
        "GRIB_PDS_PDTN": "32",
        "GRIB_COMMENT": "Brightness Temperature [C]",
        "GRIB_SHORT_NAME": "0 undefined",
        "GRIB_ELEMENT": "BRTEMP",
        "GRIB_PDS_TEMPLATE_ASSEMBLED_VALUES": "5 7 2 0 0 0 0 1 0 1 31 285 17292 2 61145",
    }
    for k in expected_md:
        assert k in md and md[k] == expected_md[k], "Did not get expected metadata"


###############################################################################
# GRIB2 file with all 0 data


def test_grib_grib2_read_all_zero_data():

    # From http://dd.weather.gc.ca/model_wave/great_lakes/erie/grib2/00/CMC_rdwps_lake-erie_ICEC_SFC_0_latlon0.05x0.05_2017111800_P000.grib2
    ds = gdal.Open(
        "data/grib/CMC_rdwps_lake-erie_ICEC_SFC_0_latlon0.05x0.05_2017111800_P000.grib2"
    )
    cs = ds.GetRasterBand(1).Checksum()
    assert cs == 0, "Could not open file"
    md = ds.GetRasterBand(1).GetMetadata()
    expected_md = {
        "GRIB_REF_TIME": "1510963200",
        "GRIB_VALID_TIME": "1510963200",
        "GRIB_FORECAST_SECONDS": "0",
        "GRIB_UNIT": "[Proportion]",
        "GRIB_PDS_TEMPLATE_NUMBERS": "2 0 0 0 0 0 0 0 1 0 0 0 0 1 0 0 0 0 0 255 255 255 255 255 255",
        "GRIB_PDS_PDTN": "0",
        "GRIB_COMMENT": "Ice cover [Proportion]",
        "GRIB_SHORT_NAME": "0-SFC",
        "GRIB_ELEMENT": "ICEC",
    }
    for k in expected_md:
        assert k in md and md[k] == expected_md[k], "Did not get expected metadata"


###############################################################################
# GRIB1 file with rotate pole lonlat


def test_grib_grib1_read_rotated_pole_lonlat():

    ds = gdal.Open("/vsisparse/data/grib/rotated_pole.grb.xml")

    assert (
        ds.RasterXSize == 726 and ds.RasterYSize == 550
    ), "Did not get expected dimensions"

    assert ds.GetRasterBand(1).GetNoDataValue() is None
    assert (
        ds.GetRasterBand(1).GetNoDataValue() is None
    )  # do it again to test correct caching

    projection = ds.GetProjectionRef()
    expected_projection_proj_7 = 'GEOGCRS["Coordinate System imported from GRIB file",BASEGEOGCRS["Coordinate System imported from GRIB file",DATUM["unnamed",ELLIPSOID["Sphere",6367470,0,LENGTHUNIT["metre",1,ID["EPSG",9001]]]],PRIMEM["Greenwich",0,ANGLEUNIT["degree",0.0174532925199433,ID["EPSG",9122]]]],DERIVINGCONVERSION["Pole rotation (GRIB convention)",METHOD["Pole rotation (GRIB convention)"],PARAMETER["Latitude of the southern pole (GRIB convention)",-30,ANGLEUNIT["degree",0.0174532925199433,ID["EPSG",9122]]],PARAMETER["Longitude of the southern pole (GRIB convention)",-15,ANGLEUNIT["degree",0.0174532925199433,ID["EPSG",9122]]],PARAMETER["Axis rotation (GRIB convention)",0,ANGLEUNIT["degree",0.0174532925199433,ID["EPSG",9122]]]],CS[ellipsoidal,2],AXIS["latitude",north,ORDER[1],ANGLEUNIT["degree",0.0174532925199433,ID["EPSG",9122]]],AXIS["longitude",east,ORDER[2],ANGLEUNIT["degree",0.0174532925199433,ID["EPSG",9122]]]]'
    expected_projection_before_proj_7 = 'PROJCS["unnamed",GEOGCS["Coordinate System imported from GRIB file",DATUM["unnamed",SPHEROID["Sphere",6367470,0]],PRIMEM["Greenwich",0],UNIT["degree",0.0174532925199433,AUTHORITY["EPSG","9122"]]],PROJECTION["Rotated_pole"],UNIT["metre",1,AUTHORITY["EPSG","9001"]],AXIS["Easting",EAST],AXIS["Northing",NORTH],EXTENSION["PROJ4","+proj=ob_tran +lon_0=-15 +o_proj=longlat +o_lon_p=0 +o_lat_p=30 +a=6367470 +b=6367470 +to_meter=0.0174532925199 +wktext"]]'
    assert projection in (
        expected_projection_proj_7,
        expected_projection_before_proj_7,
    ), projection

    if projection == expected_projection_proj_7:
        assert ds.GetSpatialRef().IsDerivedGeographic()

    gt = ds.GetGeoTransform()
    expected_gt = (-30.25, 0.1, 0.0, 24.15, 0.0, -0.1)
    assert (
        max([abs(gt[i] - expected_gt[i]) for i in range(6)]) <= 1e-3
    ), "Did not get expected geotransform"

    md = ds.GetRasterBand(1).GetMetadata()
    expected_md = {
        "GRIB_REF_TIME": "1503295200",
        "GRIB_VALID_TIME": "1503295200",
        "GRIB_FORECAST_SECONDS": "0",
        "GRIB_UNIT": "[m^2/s^2]",
        "GRIB_COMMENT": "Geopotential [m^2/s^2]",
        "GRIB_SHORT_NAME": "0-HTGL",
        "GRIB_ELEMENT": "GP",
    }
    for k in expected_md:
        assert k in md and md[k] == expected_md[k], "Did not get expected metadata"


###############################################################################
# GRIB2 file with rotate pole lonlat


def test_grib_grib2_read_rotated_pole_lonlat():

    ds = gdal.Open("/vsisparse/data/grib/rotated_pole.grb2.xml")

    assert ds.RasterXSize == 1102 and ds.RasterYSize == 1076

    projection = ds.GetProjectionRef()
    expected_projection_proj_7 = 'GEOGCRS["Coordinate System imported from GRIB file",BASEGEOGCRS["Coordinate System imported from GRIB file",DATUM["unnamed",ELLIPSOID["Sphere",6371229,0,LENGTHUNIT["metre",1,ID["EPSG",9001]]]],PRIMEM["Greenwich",0,ANGLEUNIT["degree",0.0174532925199433,ID["EPSG",9122]]]],DERIVINGCONVERSION["Pole rotation (GRIB convention)",METHOD["Pole rotation (GRIB convention)"],PARAMETER["Latitude of the southern pole (GRIB convention)",-31.758312,ANGLEUNIT["degree",0.0174532925199433,ID["EPSG",9122]]],PARAMETER["Longitude of the southern pole (GRIB convention)",-92.402969,ANGLEUNIT["degree",0.0174532925199433,ID["EPSG",9122]]],PARAMETER["Axis rotation (GRIB convention)",0,ANGLEUNIT["degree",0.0174532925199433,ID["EPSG",9122]]]],CS[ellipsoidal,2],AXIS["latitude",north,ORDER[1],ANGLEUNIT["degree",0.0174532925199433,ID["EPSG",9122]]],AXIS["longitude",east,ORDER[2],ANGLEUNIT["degree",0.0174532925199433,ID["EPSG",9122]]]]'
    expected_projection_before_proj_7 = 'PROJCS["unnamed",GEOGCS["Coordinate System imported from GRIB file",DATUM["unnamed",SPHEROID["Sphere",6371229,0]],PRIMEM["Greenwich",0],UNIT["degree",0.0174532925199433,AUTHORITY["EPSG","9122"]]],PROJECTION["Rotated_pole"],UNIT["metre",1,AUTHORITY["EPSG","9001"]],AXIS["Easting",EAST],AXIS["Northing",NORTH],EXTENSION["PROJ4","+proj=ob_tran +lon_0=-92.4029689999999846 +o_proj=longlat +o_lon_p=0 +o_lat_p=31.7583120000000001 +a=6371229 +b=6371229 +to_meter=0.0174532925199 +wktext"]]'
    assert projection in (
        expected_projection_proj_7,
        expected_projection_before_proj_7,
    ), projection

    gt = ds.GetGeoTransform()
    expected_gt = (
        -62.6222310049955,
        0.09000000999091741,
        0.0,
        48.28500200186046,
        0.0,
        -0.09000000372093023,
    )
    assert gt == pytest.approx(expected_gt, 1e-3)


###############################################################################
# Test support for GRIB2 Section 4 Template 40, Analysis or forecast at a horizontal level or in a horizontal layer at a point in time for atmospheric chemical constituents


def test_grib_grib2_read_template_4_40():

    # We could use some other encoding that JP2K...
    if not has_jp2kdrv():
        pytest.skip()

    # First band extracted from https://download.regional.atmosphere.copernicus.eu/services/CAMS50?token=__M0bChV6QsoOFqHz31VRqnpr4GhWPtcpaRy3oeZjBNSg__&grid=0.1&model=ENSEMBLE&package=ANALYSIS_PM10_SURFACE&time=-24H-1H&referencetime=2017-09-12T00:00:00Z&format=GRIB2&licence=yes
    # with data nullified
    ds = gdal.Open("data/grib/template_4_40.grb2")
    md = ds.GetRasterBand(1).GetMetadata()
    expected_md = {
        "GRIB_REF_TIME": "1505088000",
        "GRIB_PDS_TEMPLATE_ASSEMBLED_VALUES": "20 0 40008 0 255 99 0 0 1 0 1 -127 -2147483647 255 -127 -2147483647",
        "GRIB_VALID_TIME": "1505088000",
        "GRIB_FORECAST_SECONDS": "0",
        "GRIB_UNIT": "[kg/(m^3)]",
        "GRIB_PDS_TEMPLATE_NUMBERS": "20 0 156 72 0 255 99 0 0 0 1 0 0 0 0 1 255 255 255 255 255 255 255 255 255 255 255",
        "GRIB_PDS_PDTN": "40",
        "GRIB_COMMENT": "Mass Density (Concentration) [kg/(m^3)]",
        "GRIB_SHORT_NAME": "0-SFC",
        "GRIB_ELEMENT": "MASSDEN",
    }
    for k in expected_md:
        assert k in md and md[k] == expected_md[k], "Did not get expected metadata"


###############################################################################
# Test support for a unhandled GRIB2 Section 4 Template


def test_grib_grib2_read_template_4_unhandled():

    with gdaltest.error_handler():
        ds = gdal.Open("data/grib/template_4_65535.grb2")
    md = ds.GetRasterBand(1).GetMetadata()
    expected_md = {"GRIB_PDS_TEMPLATE_NUMBERS": "0 1 2 3 4 5", "GRIB_PDS_PDTN": "65535"}
    for k in expected_md:
        assert k in md and md[k] == expected_md[k], "Did not get expected metadata"


###############################################################################
# Test reading GRIB2 Transverse Mercator grid


def test_grib_grib2_read_transverse_mercator():

    ds = gdal.Open("data/grib/transverse_mercator.grb2")

    projection = ds.GetProjectionRef()
    expected_projection = """PROJCS["unnamed",GEOGCS["Coordinate System imported from GRIB file",DATUM["unnamed",SPHEROID["Sphere",6367470,0]],PRIMEM["Greenwich",0],UNIT["degree",0.0174532925199433,AUTHORITY["EPSG","9122"]]],PROJECTION["Transverse_Mercator"],PARAMETER["latitude_of_origin",0],PARAMETER["central_meridian",-117],PARAMETER["scale_factor",0.9996],PARAMETER["false_easting",500000],PARAMETER["false_northing",0],UNIT["Metre",1],AXIS["Easting",EAST],AXIS["Northing",NORTH]]"""
    assert projection == expected_projection, "Did not get expected projection"

    gt = ds.GetGeoTransform()
    expected_gt = (440720.0, 60.0, 0.0, 3751320.0, 0.0, -60.0)
    assert (
        max([abs(gt[i] - expected_gt[i]) for i in range(6)]) <= 1e-3
    ), "Did not get expected geotransform"


###############################################################################
# Test reading GRIB2 Mercator grid


def test_grib_grib2_read_mercator():

    ds = gdal.Open("data/grib/mercator.grb2")

    projection = ds.GetProjectionRef()
    expected_projection = """PROJCS["unnamed",GEOGCS["Coordinate System imported from GRIB file",DATUM["WGS_1984",SPHEROID["WGS 84",6378137,298.257223563]],PRIMEM["Greenwich",0],UNIT["degree",0.0174532925199433,AUTHORITY["EPSG","9122"]]],PROJECTION["Mercator_1SP"],PARAMETER["central_meridian",0],PARAMETER["scale_factor",1],PARAMETER["false_easting",0],PARAMETER["false_northing",0],UNIT["Metre",1],AXIS["Easting",EAST],AXIS["Northing",NORTH]]"""
    assert projection == expected_projection, "Did not get expected projection"

    gt = ds.GetGeoTransform()
    expected_gt = (-13095853.598139772, 72.237, 0.0, 3991876.4600486886, 0.0, -72.237)
    assert (
        max([abs(gt[i] - expected_gt[i]) for i in range(6)]) <= 1e-3
    ), "Did not get expected geotransform"


###############################################################################
# Test reading GRIB2 Mercator grid


def test_grib_grib2_read_mercator_2sp():

    ds = gdal.Open("data/grib/mercator_2sp.grb2")

    projection = ds.GetProjectionRef()
    expected_projection = """PROJCS["unnamed",GEOGCS["Coordinate System imported from GRIB file",DATUM["WGS_1984",SPHEROID["WGS 84",6378137,298.257223563]],PRIMEM["Greenwich",0],UNIT["degree",0.0174532925199433,AUTHORITY["EPSG","9122"]]],PROJECTION["Mercator_2SP"],PARAMETER["standard_parallel_1",33.5],PARAMETER["central_meridian",0],PARAMETER["false_easting",0],PARAMETER["false_northing",0],UNIT["Metre",1],AXIS["Easting",EAST],AXIS["Northing",NORTH]]"""
    assert projection == expected_projection, "Did not get expected projection"

    gt = ds.GetGeoTransform()
    expected_gt = (-10931598.94836207, 60.299, 0.0, 3332168.629121481, 0.0, -60.299)
    assert (
        max([abs(gt[i] - expected_gt[i]) for i in range(6)]) <= 1e-3
    ), "Did not get expected geotransform"


###############################################################################
# Test reading GRIB2 Lambert Conformal Conic grid


def test_grib_grib2_read_lcc():

    ds = gdal.Open("data/grib/lambert_conformal_conic.grb2")

    projection = ds.GetProjectionRef()
    expected_projection = """PROJCS["unnamed",GEOGCS["Coordinate System imported from GRIB file",DATUM["WGS_1984",SPHEROID["WGS 84",6378137,298.257223563]],PRIMEM["Greenwich",0],UNIT["degree",0.0174532925199433,AUTHORITY["EPSG","9122"]]],PROJECTION["Lambert_Conformal_Conic_2SP"],PARAMETER["latitude_of_origin",33.5],PARAMETER["central_meridian",117],PARAMETER["standard_parallel_1",33],PARAMETER["standard_parallel_2",34],PARAMETER["false_easting",0],PARAMETER["false_northing",0],UNIT["Metre",1],AXIS["Easting",EAST],AXIS["Northing",NORTH]]"""
    assert projection == expected_projection, "Did not get expected projection"

    gt = ds.GetGeoTransform()
    expected_gt = (8974734.737685828, 60.021, 0.0, 6235918.9698001575, 0.0, -60.021)
    assert (
        max([abs(gt[i] - expected_gt[i]) for i in range(6)]) <= 1e-3
    ), "Did not get expected geotransform"


###############################################################################
# Test reading GRIB2 Polar Stereographic grid


def test_grib_grib2_read_polar_stereo():

    ds = gdal.Open("data/grib/polar_stereographic.grb2")

    projection = ds.GetProjectionRef()
    expected_projection = """PROJCS["unnamed",GEOGCS["Coordinate System imported from GRIB file",DATUM["WGS_1984",SPHEROID["WGS 84",6378137,298.257223563]],PRIMEM["Greenwich",0],UNIT["degree",0.0174532925199433,AUTHORITY["EPSG","9122"]]],PROJECTION["Polar_Stereographic"],PARAMETER["latitude_of_origin",60],PARAMETER["central_meridian",0],PARAMETER["false_easting",0],PARAMETER["false_northing",0],UNIT["Metre",1],AXIS["Easting",SOUTH],AXIS["Northing",SOUTH]]"""
    assert projection == expected_projection, "Did not get expected projection"

    gt = ds.GetGeoTransform()
    expected_gt = (-5621962.072511509, 71.86, 0.0, 2943991.8007649644, 0.0, -71.86)
    assert (
        max([abs(gt[i] - expected_gt[i]) for i in range(6)]) <= 1e-3
    ), "Did not get expected geotransform"


###############################################################################
# Test reading GRIB2 Albers Equal Area grid


def test_grib_grib2_read_aea():

    ds = gdal.Open("data/grib/albers_equal_area.grb2")

    projection = ds.GetProjectionRef()
    expected_projection = """PROJCS["unnamed",GEOGCS["Coordinate System imported from GRIB file",DATUM["WGS_1984",SPHEROID["WGS 84",6378137,298.257223563]],PRIMEM["Greenwich",0],UNIT["degree",0.0174532925199433,AUTHORITY["EPSG","9122"]]],PROJECTION["Albers_Conic_Equal_Area"],PARAMETER["latitude_of_center",33.5],PARAMETER["longitude_of_center",117],PARAMETER["standard_parallel_1",33],PARAMETER["standard_parallel_2",34],PARAMETER["false_easting",0],PARAMETER["false_northing",0],UNIT["Metre",1],AXIS["Easting",EAST],AXIS["Northing",NORTH]]"""
    assert projection == expected_projection, "Did not get expected projection"

    gt = ds.GetGeoTransform()
    expected_gt = (8974979.714292033, 60.022, 0.0, 6235686.52464211, 0.0, -60.022)
    assert (
        max([abs(gt[i] - expected_gt[i]) for i in range(6)]) <= 1e-3
    ), "Did not get expected geotransform"


###############################################################################
# Test reading GRIB2 Lambert Azimuthal Equal Area grid


def test_grib_grib2_read_laea():

    ds = gdal.Open("data/grib/lambert_azimuthal_equal_area.grb2")

    projection = ds.GetProjectionRef()
    expected_projection = """PROJCS["unnamed",GEOGCS["Coordinate System imported from GRIB file",DATUM["WGS_1984",SPHEROID["WGS 84",6378137,298.257223563]],PRIMEM["Greenwich",0],UNIT["degree",0.0174532925199433,AUTHORITY["EPSG","9122"]]],PROJECTION["Lambert_Azimuthal_Equal_Area"],PARAMETER["latitude_of_center",33.5],PARAMETER["longitude_of_center",-117],PARAMETER["false_easting",0],PARAMETER["false_northing",0],UNIT["Metre",1],AXIS["Easting",EAST],AXIS["Northing",NORTH]]"""
    assert projection == expected_projection, "Did not get expected projection"

    gt = ds.GetGeoTransform()
    expected_gt = (-59384.01063035424, 60.021, 0.0, 44812.5792223211, 0.0, -60.021)
    assert (
        max([abs(gt[i] - expected_gt[i]) for i in range(6)]) <= 1e-3
    ), "Did not get expected geotransform"


###############################################################################
# Test reading GRIB2 with Grid point data - IEEE Floating Point Data (template 5.4)


def test_grib_grib2_read_template_5_4_grid_point_ieee_floating_point():

    ds = gdal.Open("data/grib/ieee754_single.grb2")
    cs = ds.GetRasterBand(1).Checksum()
    assert cs == 4727, "Did not get expected checksum"

    ds = gdal.Open("data/grib/ieee754_double.grb2")
    cs = ds.GetRasterBand(1).Checksum()
    assert cs == 4727, "Did not get expected checksum"


###############################################################################
# Test reading GRIB2 with NBITS=0 and DECIMAL_SCALE !=0


def test_grib_grib2_read_section_5_nbits_zero_decimal_scaled():

    ds = gdal.Open("data/grib/simple_packing_nbits_zero_decimal_scaled.grb2")
    cs = ds.GetRasterBand(1).Checksum()
    assert cs == 5, "Did not get expected checksum"

    if gdal.GetDriverByName("PNG") is not None:
        ds = gdal.Open("data/grib/png_nbits_zero_decimal_scaled.grb2")
        cs = ds.GetRasterBand(1).Checksum()
        assert cs == 5, "Did not get expected checksum"

    if has_jp2kdrv():
        ds = gdal.Open("data/grib/jpeg2000_nbits_zero_decimal_scaled.grb2")
        cs = ds.GetRasterBand(1).Checksum()
        assert cs == 5, "Did not get expected checksum"


###############################################################################
# Test reading GRIB2 with complex packing and spatial differencing of order 1


def test_grib_grib2_read_spatial_differencing_order_1():

    ds = gdal.Open("data/grib/spatial_differencing_order_1.grb2")
    cs = ds.GetRasterBand(1).Checksum()
    if cs != 46650:
        gdaltest.post_reason("Did not get expected checksum")
        print(cs)


###############################################################################
# Test GRIB2 creation options


def test_grib_grib2_write_creation_options():

    tmpfilename = "/vsimem/out.grb2"
    gdal.Translate(
        tmpfilename,
        "data/byte.tif",
        format="GRIB",
        creationOptions=[
            "DISCIPLINE=1",
            "IDS=CENTER=85(Toulouse) SUBCENTER=2 MASTER_TABLE=5 LOCAL_TABLE=0 SIGNF_REF_TIME=0(Analysis) REF_TIME=2017-09-11T12:34:56Z PROD_STATUS=2(Research) TYPE=0(Analysis)",
            "IDS_SUBCENTER=3",  # Test that it overrides IDS
            "PDS_PDTN=30",
            "BAND_1_PDS_PDTN=40",  # Test that it overrides PDS_PDTN
            "PDS_TEMPLATE_NUMBERS=20 0 156 72 0 255 99 0 0 0 1 0 0 0 0 1 255 255 255 255 255 255 255 255 255 255 255",
        ],
    )
    ds = gdal.Open(tmpfilename)
    md = ds.GetRasterBand(1).GetMetadata()
    expected_md = {
        "GRIB_IDS": "CENTER=85(Toulouse) SUBCENTER=3 MASTER_TABLE=5 LOCAL_TABLE=0 SIGNF_REF_TIME=0(Analysis) REF_TIME=2017-09-11T12:34:56Z PROD_STATUS=2(Research) TYPE=0(Analysis)",
        "GRIB_PDS_TEMPLATE_NUMBERS": "20 0 156 72 0 255 99 0 0 0 1 0 0 0 0 1 255 255 255 255 255 255 255 255 255 255 255",
        "GRIB_DISCIPLINE": "1(Hydrological)",
        "GRIB_PDS_PDTN": "40",
    }
    for k in expected_md:
        assert k in md and md[k] == expected_md[k], "Did not get expected metadata"
    ds = None
    gdal.Unlink(tmpfilename)

    # Test with PDS_TEMPLATE_NUMBERS and more elements than needed (warning)
    with gdaltest.error_handler():
        out_ds = gdal.Translate(
            tmpfilename,
            "data/byte.tif",
            format="GRIB",
            creationOptions=[
                "PDS_PDTN=40",
                "PDS_TEMPLATE_NUMBERS=20 0 156 72 0 255 99 0 0 0 1 0 0 0 0 1 255 255 255 255 255 255 255 255 255 255 255 0extra",
            ],
        )
    assert out_ds is not None
    out_ds = None
    ds = gdal.Open(tmpfilename)
    md = ds.GetRasterBand(1).GetMetadata()
    expected_md = {
        "GRIB_PDS_PDTN": "40",
        "GRIB_PDS_TEMPLATE_NUMBERS": "20 0 156 72 0 255 99 0 0 0 1 0 0 0 0 1 255 255 255 255 255 255 255 255 255 255 255 0",
    }
    for k in expected_md:
        assert k in md and md[k] == expected_md[k], "Did not get expected metadata"
    ds = None
    gdal.Unlink(tmpfilename)

    # Test with PDS_TEMPLATE_ASSEMBLED_VALUES and insufficient number of elements
    with gdaltest.error_handler():
        out_ds = gdal.Translate(
            tmpfilename,
            "data/byte.tif",
            format="GRIB",
            creationOptions=[
                "PDS_PDTN=40",
                "PDS_TEMPLATE_NUMBERS=20 0 156 72 0 255 99 0 0 0 1 0 0 0 0 1 255 255 255 255 255 255 255 255 255 255",
            ],
        )
    assert out_ds is None
    gdal.Unlink(tmpfilename)

    # Test with PDS_TEMPLATE_ASSEMBLED_VALUES
    gdal.Translate(
        tmpfilename,
        "data/byte.tif",
        format="GRIB",
        creationOptions=[
            "PDS_PDTN=40",
            "PDS_TEMPLATE_ASSEMBLED_VALUES=20 0 40008 0 255 99 0 0 1 0 1 -127 -2147483647 255 -127 -2147483647",
        ],
    )
    ds = gdal.Open(tmpfilename)
    md = ds.GetRasterBand(1).GetMetadata()
    expected_md = {
        "GRIB_PDS_PDTN": "40",
        "GRIB_PDS_TEMPLATE_ASSEMBLED_VALUES": "20 0 40008 0 255 99 0 0 1 0 1 -127 -2147483647 255 -127 -2147483647",
    }
    for k in expected_md:
        assert k in md and md[k] == expected_md[k], "Did not get expected metadata"
    ds = None
    gdal.Unlink(tmpfilename)

    # Test with PDS_TEMPLATE_ASSEMBLED_VALUES and more elements than needed (warning)
    with gdaltest.error_handler():
        out_ds = gdal.Translate(
            tmpfilename,
            "data/byte.tif",
            format="GRIB",
            creationOptions=[
                "PDS_PDTN=40",
                "PDS_TEMPLATE_ASSEMBLED_VALUES=20 0 40008 0 255 99 0 0 1 0 1 -127 -2147483647 255 -127 -2147483647 0extra",
            ],
        )
    assert out_ds is not None
    out_ds = None
    ds = gdal.Open(tmpfilename)
    md = ds.GetRasterBand(1).GetMetadata()
    expected_md = {
        "GRIB_PDS_PDTN": "40",
        "GRIB_PDS_TEMPLATE_ASSEMBLED_VALUES": "20 0 40008 0 255 99 0 0 1 0 1 -127 -2147483647 255 -127 -2147483647",
    }
    for k in expected_md:
        assert k in md and md[k] == expected_md[k], "Did not get expected metadata"
    ds = None
    gdal.Unlink(tmpfilename)

    # Test with PDS_TEMPLATE_ASSEMBLED_VALUES and insufficient number of elements
    with gdaltest.error_handler():
        out_ds = gdal.Translate(
            tmpfilename,
            "data/byte.tif",
            format="GRIB",
            creationOptions=[
                "PDS_PDTN=40",
                "PDS_TEMPLATE_ASSEMBLED_VALUES=20 0 40008 0 255 99 0 0 1 0 1 -127 -2147483647 255 -127",
            ],
        )
    assert out_ds is None
    gdal.Unlink(tmpfilename)

    # Test with PDS_TEMPLATE_ASSEMBLED_VALUES with variable number of elements
    gdal.Translate(
        tmpfilename,
        "data/byte.tif",
        format="GRIB",
        creationOptions=[
            "PDS_PDTN=32",
            "PDS_TEMPLATE_ASSEMBLED_VALUES=5 7 2 0 0 0 0 1 0 2 31 285 17292 2 61145 31 285 17292 2 61145",
        ],
    )
    ds = gdal.Open(tmpfilename)
    md = ds.GetRasterBand(1).GetMetadata()
    expected_md = {
        "GRIB_PDS_PDTN": "32",
        "GRIB_PDS_TEMPLATE_ASSEMBLED_VALUES": "5 7 2 0 0 0 0 1 0 2 31 285 17292 2 61145 31 285 17292 2 61145",
    }
    for k in expected_md:
        assert k in md and md[k] == expected_md[k], "Did not get expected metadata"
    ds = None
    gdal.Unlink(tmpfilename)

    # Test with PDS_TEMPLATE_ASSEMBLED_VALUES with variable number of elements, and insufficient number of elements in the variable section
    with gdaltest.error_handler():
        out_ds = gdal.Translate(
            tmpfilename,
            "data/byte.tif",
            format="GRIB",
            creationOptions=[
                "PDS_PDTN=32",
                "PDS_TEMPLATE_ASSEMBLED_VALUES=5 7 2 0 0 0 0 1 0 2 31 285 17292 2 61145 31 285 17292 2",
            ],
        )
    assert out_ds is None
    gdal.Unlink(tmpfilename)

    # Test with PDS_TEMPLATE_ASSEMBLED_VALUES with variable number of elements, and extra elements
    with gdaltest.error_handler():
        gdal.Translate(
            tmpfilename,
            "data/byte.tif",
            format="GRIB",
            creationOptions=[
                "PDS_PDTN=32",
                "PDS_TEMPLATE_ASSEMBLED_VALUES=5 7 2 0 0 0 0 1 0 2 31 285 17292 2 61145 31 285 17292 2 61145 0extra",
            ],
        )
    ds = gdal.Open(tmpfilename)
    md = ds.GetRasterBand(1).GetMetadata()
    expected_md = {
        "GRIB_PDS_PDTN": "32",
        "GRIB_PDS_TEMPLATE_ASSEMBLED_VALUES": "5 7 2 0 0 0 0 1 0 2 31 285 17292 2 61145 31 285 17292 2 61145",
    }
    for k in expected_md:
        assert k in md and md[k] == expected_md[k], "Did not get expected metadata"
    ds = None
    gdal.Unlink(tmpfilename)

    # Test with PDS_TEMPLATE_NUMBERS with variable number of elements
    gdal.Translate(
        tmpfilename,
        "data/byte.tif",
        format="GRIB",
        creationOptions=[
            "PDS_PDTN=32",
            "PDS_TEMPLATE_NUMBERS=5 7 2 0 0 0 0 0 1 0 0 0 0 2 0 31 1 29 67 140 2 0 0 238 217 0 31 1 29 67 140 2 0 0 238 217",
        ],
    )
    ds = gdal.Open(tmpfilename)
    md = ds.GetRasterBand(1).GetMetadata()
    expected_md = {
        "GRIB_PDS_PDTN": "32",
        "GRIB_PDS_TEMPLATE_ASSEMBLED_VALUES": "5 7 2 0 0 0 0 1 0 2 31 285 17292 2 61145 31 285 17292 2 61145",
    }
    for k in expected_md:
        assert k in md and md[k] == expected_md[k], "Did not get expected metadata"
    ds = None
    gdal.Unlink(tmpfilename)

    # Test with unknown PDS_PDTN with PDS_TEMPLATE_NUMBERS
    with gdaltest.error_handler():
        out_ds = gdal.Translate(
            tmpfilename,
            "data/byte.tif",
            format="GRIB",
            creationOptions=["PDS_PDTN=65535", "PDS_TEMPLATE_NUMBERS=1 2 3 4 5"],
        )
    assert out_ds is not None
    out_ds = None
    with gdaltest.error_handler():
        ds = gdal.Open(tmpfilename)
    md = ds.GetRasterBand(1).GetMetadata()
    expected_md = {"GRIB_PDS_PDTN": "65535", "GRIB_PDS_TEMPLATE_NUMBERS": "1 2 3 4 5"}
    for k in expected_md:
        assert k in md and md[k] == expected_md[k], "Did not get expected metadata"
    ds = None
    gdal.Unlink(tmpfilename)

    # Test with unknown PDS_PDTN with PDS_TEMPLATE_ASSEMBLED_VALUES
    with gdaltest.error_handler():
        out_ds = gdal.Translate(
            tmpfilename,
            "data/byte.tif",
            format="GRIB",
            creationOptions=[
                "PDS_PDTN=65535",
                "PDS_TEMPLATE_ASSEMBLED_VALUES=1 2 3 4 5",
            ],
        )
    assert out_ds is None
    gdal.Unlink(tmpfilename)

    # Test with PDS_PDTN != 0 without template numbers
    with gdaltest.error_handler():
        out_ds = gdal.Translate(
            tmpfilename, "data/byte.tif", format="GRIB", creationOptions=["PDS_PDTN=32"]
        )
    assert out_ds is None
    gdal.Unlink(tmpfilename)

    # Test with invalid values in PDS_TEMPLATE_NUMBERS
    with gdaltest.error_handler():
        out_ds = gdal.Translate(
            tmpfilename,
            "data/byte.tif",
            format="GRIB",
            creationOptions=["PDS_PDTN=254", "PDS_TEMPLATE_NUMBERS=-1 256 0 0 0 0"],
        )
    assert out_ds is not None
    out_ds = None
    gdal.Unlink(tmpfilename)

    # Test with invalid values in PDS_TEMPLATE_ASSEMBLED_VALUES
    with gdaltest.error_handler():
        out_ds = gdal.Translate(
            tmpfilename,
            "data/byte.tif",
            format="GRIB",
            creationOptions=[
                "PDS_PDTN=44",
                # {44,21,0,                    {1,  1, 2,1,-1,      -4,   -1,-4,1,1,1,  2,  1,1,-2   ,1,-1,  -4,1,-1,-4} },
                "PDS_TEMPLATE_ASSEMBLED_VALUES=-1 256 -1 1 128 4000000000 -1 -4 1 1 1 65536 1 1 32768 1 -129 -4 1 -1 -4",
            ],
        )
    assert out_ds is not None
    out_ds = None
    gdal.Unlink(tmpfilename)

    # Test with both PDS_TEMPLATE_NUMBERS and PDS_TEMPLATE_ASSEMBLED_VALUES
    with gdaltest.error_handler():
        out_ds = gdal.Translate(
            tmpfilename,
            "data/byte.tif",
            format="GRIB",
            creationOptions=[
                "PDS_PDTN=40",
                "PDS_TEMPLATE_NUMBERS=20 0 156 72 0 255 99 0 0 0 1 0 0 0 0 1 255 255 255 255 255 255 255 255 255 255 255",
                "PDS_TEMPLATE_ASSEMBLED_VALUES=20 0 40008 0 255 99 0 0 1 0 1 -127 -2147483647 255 -127 -2147483647",
            ],
        )
    assert out_ds is None
    gdal.Unlink(tmpfilename)


###############################################################################
# Test GRIB2 write support for projections


def test_grib_grib2_write_projections():

    filenames = [
        "albers_equal_area.grb2",
        "lambert_azimuthal_equal_area.grb2",
        "lambert_conformal_conic.grb2",
        "mercator.grb2",
        "mercator_2sp.grb2",
        "polar_stereographic.grb2",
        "ieee754_single.grb2",  # Longitude latitude
    ]
    for filename in filenames:
        filename = "data/grib/" + filename
        src_ds = gdal.Open(filename)
        tmpfilename = "/vsimem/out.grb2"
        gdal.Translate(tmpfilename, filename, format="GRIB")
        out_ds = gdal.Open(tmpfilename)

        assert src_ds.GetProjectionRef() == out_ds.GetProjectionRef(), (
            "did not get expected projection for %s" % filename
        )

        expected_gt = src_ds.GetGeoTransform()
        got_gt = out_ds.GetGeoTransform()
        assert max([abs(expected_gt[i] - got_gt[i]) for i in range(6)]) <= 1e-5, (
            "did not get expected geotransform for %s" % filename
        )

        out_ds = None
        gdal.Unlink(tmpfilename)

    # Test writing GRS80
    src_ds = gdal.GetDriverByName("MEM").Create("", 2, 2, 1, gdal.GDT_Float32)
    src_ds.SetGeoTransform([2, 1, 0, 49, 0, -1])
    src_ds.SetProjection(
        """GEOGCS["GRS 1980(IUGG, 1980)",
    DATUM["unknown",
        SPHEROID["GRS80",6378137,298.257222101]],
    PRIMEM["Greenwich",0],
    UNIT["degree",0.0174532925199433]]"""
    )
    tmpfilename = "/vsimem/out.grb2"
    out_ds = gdaltest.grib_drv.CreateCopy(tmpfilename, src_ds)
    wkt = out_ds.GetProjectionRef()
    out_ds = None
    gdal.Unlink(tmpfilename)
    assert 'SPHEROID["GRS80",6378137,298.257222101]' in wkt

    # Test writing Mercator_1SP with scale != 1 (will be read as Mercator_2SP)
    src_ds = gdal.Warp(
        "",
        "data/byte.tif",
        format="MEM",
        dstSRS="""PROJCS["unnamed",
    GEOGCS["NAD27",
        DATUM["North_American_Datum_1927",
            SPHEROID["Clarke 1866",6378206.4,294.9786982138982,
                AUTHORITY["EPSG","7008"]],
            AUTHORITY["EPSG","6267"]],
        PRIMEM["Greenwich",0,
            AUTHORITY["EPSG","8901"]],
        UNIT["degree",0.0174532925199433,
            AUTHORITY["EPSG","9122"]],
        AUTHORITY["EPSG","4267"]],
    PROJECTION["Mercator_1SP"],
    PARAMETER["central_meridian",0],
    PARAMETER["scale_factor",0.8347374126019634],
    PARAMETER["false_easting",0],
    PARAMETER["false_northing",0],
    UNIT["metre",1,
        AUTHORITY["EPSG","9001"]]]""",
    )
    tmpfilename = "/vsimem/out.grb2"
    gdal.Translate(tmpfilename, src_ds, format="GRIB")
    out_ds = gdal.Open(tmpfilename)
    expected_wkt = 'PROJCS["unnamed",GEOGCS["Coordinate System imported from GRIB file",DATUM["unnamed",SPHEROID["Spheroid imported from GRIB file",6378206.4,294.978698213911]],PRIMEM["Greenwich",0],UNIT["degree",0.0174532925199433,AUTHORITY["EPSG","9122"]]],PROJECTION["Mercator_2SP"],PARAMETER["standard_parallel_1",33.500986],PARAMETER["central_meridian",0],PARAMETER["false_easting",0],PARAMETER["false_northing",0],UNIT["Metre",1],AXIS["Easting",EAST],AXIS["Northing",NORTH]]'
    got_sr = osr.SpatialReference()
    got_sr.SetFromUserInput(out_ds.GetProjectionRef())
    expected_sr = osr.SpatialReference()
    expected_sr.SetFromUserInput(expected_wkt)
    if got_sr.IsSame(expected_sr) == 0:
        print(out_ds.GetProjectionRef())
        pytest.fail("did not get expected projection for Mercator_1SP")
    expected_gt = (-10931635.565066436, 60.297, 0.0, 3331982.221608528, 0.0, -60.297)
    got_gt = out_ds.GetGeoTransform()
    assert (
        max([abs(expected_gt[i] - got_gt[i]) for i in range(6)]) <= 1e-5
    ), "did not get expected geotransform for Mercator_1SP"
    out_ds = None
    gdal.Unlink(tmpfilename)

    # Test writing LCC_1SP (will be read as LCC_2SP)
    src_ds = gdal.Warp(
        "",
        "data/byte.tif",
        format="MEM",
        dstSRS="""PROJCS["unnamed",
    GEOGCS["NAD27",
        DATUM["North_American_Datum_1927",
            SPHEROID["Clarke 1866",6378206.4,294.9786982138982,
                AUTHORITY["EPSG","7008"]],
            AUTHORITY["EPSG","6267"]],
        PRIMEM["Greenwich",0,
            AUTHORITY["EPSG","8901"]],
        UNIT["degree",0.0174532925199433,
            AUTHORITY["EPSG","9122"]],
        AUTHORITY["EPSG","4267"]],
    PROJECTION["Lambert_Conformal_Conic_1SP"],
    PARAMETER["latitude_of_origin",33.5],
    PARAMETER["central_meridian",117],
    PARAMETER["scale_factor",0.9999],
    PARAMETER["false_easting",0],
    PARAMETER["false_northing",0],
    UNIT["metre",1]]""",
    )
    tmpfilename = "/vsimem/out.grb2"
    gdal.Translate(tmpfilename, src_ds, format="GRIB")
    out_ds = gdal.Open(tmpfilename)
    expected_wkt = 'PROJCS["unnamed",GEOGCS["Coordinate System imported from GRIB file",DATUM["unnamed",SPHEROID["Spheroid imported from GRIB file",6378206.4,294.978698213911]],PRIMEM["Greenwich",0],UNIT["degree",0.0174532925199433,AUTHORITY["EPSG","9122"]]],PROJECTION["Lambert_Conformal_Conic_2SP"],PARAMETER["latitude_of_origin",33.5],PARAMETER["central_meridian",117],PARAMETER["standard_parallel_1",34.310911],PARAMETER["standard_parallel_2",32.686501],PARAMETER["false_easting",0],PARAMETER["false_northing",0],UNIT["Metre",1],AXIS["Easting",EAST],AXIS["Northing",NORTH]]'
    got_sr = osr.SpatialReference()
    got_sr.SetFromUserInput(out_ds.GetProjectionRef())
    expected_sr = osr.SpatialReference()
    expected_sr.SetFromUserInput(expected_wkt)
    if got_sr.IsSame(expected_sr) == 0:
        print(out_ds.GetProjectionRef())
        pytest.fail("did not get expected projection for LCC_1SP")
    expected_gt = (8974472.884926716, 60.017, 0.0, 6235685.688523474, 0.0, -60.017)
    got_gt = out_ds.GetGeoTransform()
    assert (
        max([abs(expected_gt[i] - got_gt[i]) for i in range(6)]) <= 1e-5
    ), "did not get expected geotransform for LCC_1SP"
    out_ds = None
    gdal.Unlink(tmpfilename)


###############################################################################


def _grib_read_section(filename, sect_num_to_read):

    f = gdal.VSIFOpenL(filename, "rb")
    if f is None:
        return None
    # Ignore Sect 0
    gdal.VSIFReadL(1, 16, f)
    ret = None
    while True:
        sect_size_bytes = gdal.VSIFReadL(1, 4, f)
        if not sect_size_bytes:
            break
        sect_size_num = struct.unpack(">I", sect_size_bytes)[0]
        sect_num_bytes = gdal.VSIFReadL(1, 1, f)
        sect_num = ord(sect_num_bytes)
        if sect_num == sect_num_to_read:
            ret = (
                sect_size_bytes
                + sect_num_bytes
                + gdal.VSIFReadL(1, sect_size_num - 5, f)
            )
            break
        gdal.VSIFSeekL(f, sect_size_num - 5, 1)

    gdal.VSIFCloseL(f)
    return ret


###############################################################################
# Test GRIB2 write support for data encodings


def test_grib_grib2_write_data_encodings():

    # Template 5 numbers
    GS5_SIMPLE = 0
    GS5_CMPLX = 2
    GS5_CMPLXSEC = 3
    GS5_IEEE = 4
    GS5_JPEG2000 = 40
    GS5_PNG = 41

    tests = []
    tests += [["data/byte.tif", [], 4672, GS5_SIMPLE]]
    tests += [["data/byte.tif", ["DATA_ENCODING=SIMPLE_PACKING"], 4672, GS5_SIMPLE]]

    tests += [["data/byte.tif", ["DATA_ENCODING=IEEE_FLOATING_POINT"], 4672, GS5_IEEE]]

    tests += [
        ["data/byte.tif", ["DATA_ENCODING=SIMPLE_PACKING", "NBITS=8"], 4672, GS5_SIMPLE]
    ]
    tests += [
        ["data/byte.tif", ["DATA_ENCODING=SIMPLE_PACKING", "NBITS=9"], 4672, GS5_SIMPLE]
    ]
    tests += [
        ["data/byte.tif", ["DATA_ENCODING=SIMPLE_PACKING", "NBITS=7"], 4484, GS5_SIMPLE]
    ]
    tests += [
        [
            "data/byte.tif",
            ["DATA_ENCODING=SIMPLE_PACKING", "DECIMAL_SCALE_FACTOR=-1"],
            4820,
            GS5_SIMPLE,
        ]
    ]
    tests += [
        [
            "data/byte.tif",
            ["DATA_ENCODING=SIMPLE_PACKING", "NBITS=5", "DECIMAL_SCALE_FACTOR=-1"],
            4820,
            GS5_SIMPLE,
        ]
    ]
    tests += [
        [
            "data/byte.tif",
            ["DATA_ENCODING=SIMPLE_PACKING", "NBITS=8", "DECIMAL_SCALE_FACTOR=-1"],
            4855,
            GS5_SIMPLE,
        ]
    ]

    tests += [
        [
            "data/grib/ds.mint.bin",
            [
                "PDS_PDTN=8",
                "PDS_TEMPLATE_ASSEMBLED_VALUES=0 5 2 0 0 255 255 1 19 1 0 0 255 -1 -2147483647 2008 2 22 12 0 0 1 0 3 255 1 12 1 0",
            ],
            46650,
            GS5_CMPLX,
        ]
    ]  # has nodata, hence complex packing
    tests += [["data/byte.tif", ["DATA_ENCODING=COMPLEX_PACKING"], 4672, GS5_CMPLX]]
    tests += [
        [
            "data/byte.tif",
            ["DATA_ENCODING=COMPLEX_PACKING", "SPATIAL_DIFFERENCING_ORDER=0"],
            4672,
            GS5_CMPLX,
        ]
    ]
    tests += [["data/byte.tif", ["SPATIAL_DIFFERENCING_ORDER=1"], 4672, GS5_CMPLXSEC]]
    tests += [
        [
            "data/byte.tif",
            ["DATA_ENCODING=COMPLEX_PACKING", "SPATIAL_DIFFERENCING_ORDER=1"],
            4672,
            GS5_CMPLXSEC,
        ]
    ]
    tests += [
        [
            "data/byte.tif",
            ["DATA_ENCODING=COMPLEX_PACKING", "SPATIAL_DIFFERENCING_ORDER=2"],
            4672,
            GS5_CMPLXSEC,
        ]
    ]
    tests += [
        [
            "data/byte.tif",
            ["DATA_ENCODING=COMPLEX_PACKING", "DECIMAL_SCALE_FACTOR=-1"],
            4820,
            GS5_CMPLX,
        ]
    ]
    tests += [
        ["data/byte.tif", ["DATA_ENCODING=COMPLEX_PACKING", "NBITS=7"], 4484, GS5_CMPLX]
    ]

    if gdal.GetDriverByName("PNG") is not None:
        tests += [["data/byte.tif", ["DATA_ENCODING=PNG"], 4672, GS5_PNG]]
        tests += [["data/byte.tif", ["DATA_ENCODING=PNG", "NBITS=8"], 4672, GS5_PNG]]
        tests += [
            [
                "data/byte.tif",
                ["DATA_ENCODING=PNG", "DECIMAL_SCALE_FACTOR=-1"],
                4820,
                GS5_PNG,
            ]
        ]
        tests += [
            [
                "data/byte.tif",
                ["DATA_ENCODING=PNG", "NBITS=8", "DECIMAL_SCALE_FACTOR=-1"],
                4855,
                GS5_PNG,
            ]
        ]
        tests += [
            ["data/byte.tif", ["DATA_ENCODING=PNG", "NBITS=9"], 4672, GS5_PNG]
        ]  # rounded to 16 bit
        tests += [
            ["data/byte.tif", ["DATA_ENCODING=PNG", "NBITS=7"], 4672, GS5_PNG]
        ]  # rounded to 8 bit
        tests += [["data/byte.tif", ["DATA_ENCODING=PNG", "NBITS=4"], 5103, GS5_PNG]]
        tests += [
            ["data/byte.tif", ["DATA_ENCODING=PNG", "NBITS=3"], 5103, GS5_PNG]
        ]  # rounded to 4 bit
        tests += [["data/byte.tif", ["DATA_ENCODING=PNG", "NBITS=2"], 5103, GS5_PNG]]
        tests += [["data/byte.tif", ["DATA_ENCODING=PNG", "NBITS=1"], 5103, GS5_PNG]]
        tests += [["../gcore/data/float32.tif", ["DATA_ENCODING=PNG"], 4672, GS5_PNG]]

    found_j2k_drivers = []
    for drvname in ["JP2KAK", "JP2OPENJPEG", "JPEG2000", "JP2ECW"]:
        if gdal.GetDriverByName(drvname) is not None:
            if drvname != "JP2ECW":
                found_j2k_drivers.append(drvname)
            else:
                import ecw

                if ecw.has_write_support():
                    found_j2k_drivers.append(drvname)

    if found_j2k_drivers:
        tests += [["data/byte.tif", ["DATA_ENCODING=JPEG2000"], 4672, GS5_JPEG2000]]
        tests += [
            [
                "data/byte.tif",
                ["DATA_ENCODING=JPEG2000", "COMPRESSION_RATIO=2"],
                4672,
                GS5_JPEG2000,
            ]
        ]  # COMPRESSION_RATIO ignored in that case
        tests += [
            ["data/byte.tif", ["DATA_ENCODING=JPEG2000", "NBITS=8"], 4672, GS5_JPEG2000]
        ]
        tests += [
            [
                "data/byte.tif",
                ["DATA_ENCODING=JPEG2000", "DECIMAL_SCALE_FACTOR=-1"],
                (4820, 4722),
                GS5_JPEG2000,
            ]
        ]
        tests += [
            [
                "data/byte.tif",
                ["DATA_ENCODING=JPEG2000", "NBITS=8", "DECIMAL_SCALE_FACTOR=-1"],
                (4855, 4795),
                GS5_JPEG2000,
            ]
        ]
        tests += [
            ["data/byte.tif", ["DATA_ENCODING=JPEG2000", "NBITS=9"], 4672, GS5_JPEG2000]
        ]
        # 4899 for JP2ECW, 4440 for JP2OPENJPEG
        tests += [
            [
                "data/byte.tif",
                ["DATA_ENCODING=JPEG2000", "NBITS=7"],
                (4484, 4899, 4440),
                GS5_JPEG2000,
            ]
        ]
        for drvname in found_j2k_drivers:
            tests += [
                ["data/byte.tif", ["JPEG2000_DRIVER=" + drvname], 4672, GS5_JPEG2000]
            ]
        tests += [
            [
                "../gcore/data/float32.tif",
                ["DATA_ENCODING=JPEG2000"],
                4672,
                GS5_JPEG2000,
            ]
        ]

    tests += [["../gcore/data/int16.tif", [], 4672, GS5_SIMPLE]]
    tests += [["../gcore/data/uint16.tif", [], 4672, GS5_SIMPLE]]
    tests += [["../gcore/data/int32.tif", [], 4672, GS5_SIMPLE]]
    tests += [["../gcore/data/uint32.tif", [], 4672, GS5_SIMPLE]]
    tests += [["../gcore/data/float32.tif", [], 4672, GS5_IEEE]]
    tests += [["../gcore/data/float64.tif", [], 4672, GS5_IEEE]]
    tests += [
        [
            "../gcore/data/float32.tif",
            ["DATA_ENCODING=IEEE_FLOATING_POINT"],
            4672,
            GS5_IEEE,
        ]
    ]
    tests += [
        [
            "../gcore/data/float64.tif",
            ["DATA_ENCODING=IEEE_FLOATING_POINT"],
            4672,
            GS5_IEEE,
        ]
    ]
    tests += [
        [
            "../gcore/data/float32.tif",
            ["DATA_ENCODING=COMPLEX_PACKING"],
            4672,
            GS5_CMPLX,
        ]
    ]

    one_ds = gdal.GetDriverByName("MEM").Create("", 1, 1)
    one_ds.SetGeoTransform([2, 1, 0, 49, 0, -1])
    sr = osr.SpatialReference()
    sr.SetFromUserInput("WGS84")
    one_ds.SetProjection(sr.ExportToWkt())
    one_ds.GetRasterBand(1).Fill(1)

    tests += [[one_ds, [], 1, GS5_SIMPLE]]
    tests += [[one_ds, ["DATA_ENCODING=COMPLEX_PACKING"], 1, GS5_CMPLX]]
    if gdal.GetDriverByName("PNG") is not None:
        tests += [[one_ds, ["DATA_ENCODING=PNG"], 1, GS5_PNG]]
    if found_j2k_drivers:
        tests += [[one_ds, ["DATA_ENCODING=JPEG2000"], 1, GS5_JPEG2000]]

    nodata_never_hit_ds = gdal.GetDriverByName("MEM").Create("", 1, 1)
    nodata_never_hit_ds.SetGeoTransform([2, 1, 0, 49, 0, -1])
    nodata_never_hit_ds.SetProjection(sr.ExportToWkt())
    nodata_never_hit_ds.GetRasterBand(1).SetNoDataValue(1)

    tests += [[nodata_never_hit_ds, [], 0, GS5_SIMPLE]]

    all_nodata_ds = gdal.GetDriverByName("MEM").Create("", 1, 1)
    all_nodata_ds.SetGeoTransform([2, 1, 0, 49, 0, -1])
    all_nodata_ds.SetProjection(sr.ExportToWkt())
    all_nodata_ds.GetRasterBand(1).SetNoDataValue(0)

    tests += [[all_nodata_ds, ["DATA_ENCODING=COMPLEX_PACKING"], 0, GS5_CMPLX]]

    for (filename, options, expected_cs, expected_section5_template_number) in tests:
        tmpfilename = "/vsimem/out.grb2"
        gdal.ErrorReset()
        gdal.Translate(tmpfilename, filename, format="GRIB", creationOptions=options)
        error_msg = gdal.GetLastErrorMsg()
        assert error_msg == "", "did not expect error for %s, %s" % (
            str(filename),
            str(options),
        )

        section5 = _grib_read_section(tmpfilename, 5)
        section5_template_number = struct.unpack(">h", section5[9:11])[0]
        assert (
            section5_template_number == expected_section5_template_number
        ), "did not get expected section 5 template number for %s, %s" % (
            str(filename),
            str(options),
        )

        out_ds = gdal.Open(tmpfilename)
        cs = out_ds.GetRasterBand(1).Checksum()
        nd = out_ds.GetRasterBand(1).GetNoDataValue()
        out_ds = None
        gdal.Unlink(tmpfilename)
        if not isinstance(expected_cs, tuple):
            expected_cs = (expected_cs,)
        assert cs in expected_cs, "did not get expected checksum for %s, %s" % (
            str(filename),
            str(options),
        )

        if section5_template_number in (GS5_CMPLX, GS5_CMPLXSEC):
            if isinstance(filename, str):
                ref_ds = gdal.Open(filename)
            else:
                ref_ds = filename
            expected_nd = ref_ds.GetRasterBand(1).GetNoDataValue()
            assert nd == expected_nd, "did not get expected nodata for %s, %s" % (
                str(filename),
                str(options),
            )

    # Test floating point data with dynamic < 1
    test_ds = gdal.GetDriverByName("MEM").Create("", 2, 2, 1, gdal.GDT_Float32)
    test_ds.SetGeoTransform([2, 1, 0, 49, 0, -1])
    test_ds.SetProjection(sr.ExportToWkt())
    test_ds.WriteRaster(0, 0, 2, 2, struct.pack(4 * "f", 1.23, 1.45, 1.56, 1.78))

    encodings = ["SIMPLE_PACKING", "COMPLEX_PACKING", "IEEE_FLOATING_POINT"]
    if gdal.GetDriverByName("PNG") is not None:
        encodings += ["PNG"]
    # JPEG2000 doesn't result in an appropriate result
    if (
        found_j2k_drivers
        and found_j2k_drivers != ["JPEG2000"]
        and found_j2k_drivers != ["JPEG2000", "JP2ECW"]
    ):
        encodings += ["JPEG2000"]

    for encoding in encodings:
        tmpfilename = "/vsimem/out.grb2"
        gdal.ErrorReset()
        options = ["DATA_ENCODING=" + encoding]
        if encoding == "COMPLEX_PACKING":
            with gdaltest.error_handler():
                success = gdal.Translate(
                    tmpfilename, test_ds, format="GRIB", creationOptions=options
                )
            assert not success, "expected error for %s, %s" % (
                "floating point data with dynamic < 1",
                str(options),
            )
        else:
            gdal.Translate(tmpfilename, test_ds, format="GRIB", creationOptions=options)
            error_msg = gdal.GetLastErrorMsg()
            assert error_msg == "", "did not expect error for %s, %s" % (
                "floating point data with dynamic < 1",
                str(options),
            )
            out_ds = gdal.Open(tmpfilename)
            got_vals = struct.unpack(4 * "d", out_ds.ReadRaster())
            out_ds = None
            if encoding == "IEEE_FLOATING_POINT":
                expected_vals = (1.23, 1.45, 1.56, 1.78)
            else:
                expected_vals = (
                    1.2300000190734863,
                    1.4487500190734863,
                    1.5581250190734863,
                    1.7807812690734863,
                )
            assert (
                max([abs(got_vals[i] - expected_vals[i]) for i in range(4)]) <= 1e-7
            ), "did not get expected values"
        gdal.Unlink(tmpfilename)

    test_ds = None

    # Test floating point data with very large dynamic
    test_ds = gdal.GetDriverByName("MEM").Create("", 2, 2, 1, gdal.GDT_Float32)
    test_ds.SetGeoTransform([2, 1, 0, 49, 0, -1])
    test_ds.SetProjection(sr.ExportToWkt())
    test_ds.WriteRaster(
        0, 0, 2, 2, struct.pack(4 * "f", 1.23e10, -2.45e10, 1.23e10, -2.45e10)
    )

    encodings = ["SIMPLE_PACKING"]
    if gdal.GetDriverByName("PNG") is not None:
        encodings += ["PNG"]
    # JP2ECW doesn't manage to compress such a small file
    if (
        found_j2k_drivers
        and found_j2k_drivers != ["JP2ECW"]
        and found_j2k_drivers != ["JPEG2000", "JP2ECW"]
    ):
        encodings += ["JPEG2000"]

    for encoding in encodings:
        tmpfilename = "/vsimem/out.grb2"
        if encoding != "SIMPLE_PACKING":
            gdal.PushErrorHandler()
        gdal.Translate(
            tmpfilename,
            test_ds,
            format="GRIB",
            creationOptions=["DATA_ENCODING=" + encoding],
        )
        if encoding != "SIMPLE_PACKING":
            gdal.PopErrorHandler()
        out_ds = gdal.Open(tmpfilename)
        assert out_ds is not None, "failed to re-open dataset for " + encoding
        got_vals = struct.unpack(4 * "d", out_ds.ReadRaster())
        out_ds = None
        gdal.Unlink(tmpfilename)
        expected_vals = (1.23e10, -2.45e10, 1.23e10, -2.45e10)
        assert (
            max(
                [
                    abs((got_vals[i] - expected_vals[i]) / expected_vals[i])
                    for i in range(4)
                ]
            )
            <= 1e-4
        ), ("did not get expected values for " + encoding)
    test_ds = None

    # Test lossy J2K compression
    for drvname in found_j2k_drivers:
        tmpfilename = "/vsimem/out.grb2"
        gdal.ErrorReset()
        gdal.Translate(
            tmpfilename,
            "data/utm.tif",
            format="GRIB",
            creationOptions=["JPEG2000_DRIVER=" + drvname, "COMPRESSION_RATIO=20"],
        )
        error_msg = gdal.GetLastErrorMsg()
        assert error_msg == "", "did not expect error for %s, %s" % (
            str(filename),
            str(options),
        )
        out_ds = gdal.Open(tmpfilename)
        cs = out_ds.GetRasterBand(1).Checksum()
        out_ds = None
        gdal.Unlink(tmpfilename)
        if cs == 0 or cs == 50235:  # 50235: lossless checksum
            gdaltest.post_reason(
                "did not get expected checksum for lossy JPEG2000 with " + drvname
            )
            print(cs)


###############################################################################
# Test GRIB2 write support with warnings/errors


def test_grib_grib2_write_data_encodings_warnings_and_errors():

    # Cases where warnings are expected
    tests = []
    tests += [
        [
            "data/byte.tif",
            ["DATA_ENCODING=SIMPLE_PACKING", "DECIMAL_SCALE_FACTOR=1"],
            4672,
        ]
    ]
    tests += [
        ["data/byte.tif", ["DATA_ENCODING=SIMPLE_PACKING", "JPEG2000_DRIVER=FOO"], 4672]
    ]
    tests += [
        ["data/byte.tif", ["DATA_ENCODING=SIMPLE_PACKING", "JPEG2000_DRIVER=FOO"], 4672]
    ]
    tests += [
        [
            "data/byte.tif",
            ["DATA_ENCODING=SIMPLE_PACKING", "SPATIAL_DIFFERENCING_ORDER=1"],
            4672,
        ]
    ]
    tests += [
        ["data/grib/ds.mint.bin", ["DATA_ENCODING=SIMPLE_PACKING"], 41640]
    ]  # should warn since simple packing doesn't support nodata
    tests += [["data/byte.tif", ["NBITS=32"], 4672]]
    tests += [["data/byte.tif", ["DATA_ENCODING=IEEE_FLOATING_POINT", "NBITS=8"], 4672]]
    tests += [
        [
            "data/byte.tif",
            ["DATA_ENCODING=IEEE_FLOATING_POINT", "DECIMAL_SCALE_FACTOR=-1"],
            4672,
        ]
    ]
    for (filename, options, expected_cs) in tests:
        tmpfilename = "/vsimem/out.grb2"
        src_ds = gdal.Open(filename)
        gdal.ErrorReset()
        with gdaltest.error_handler():
            out_ds = gdaltest.grib_drv.CreateCopy(tmpfilename, src_ds, options=options)

        error_msg = gdal.GetLastErrorMsg()
        assert error_msg != "", "expected warning for %s, %s" % (
            str(filename),
            str(options),
        )
        assert out_ds is not None, "did not expect null return for %s, %s" % (
            str(filename),
            str(options),
        )

        cs = out_ds.GetRasterBand(1).Checksum()

        out_ds = None
        gdal.Unlink(tmpfilename)
        if not isinstance(expected_cs, tuple):
            expected_cs = (expected_cs,)
        assert cs in expected_cs, "did not get expected checksum for %s, %s" % (
            str(filename),
            str(options),
        )

    # Cases where errors are expected
    tests = []
    tests += [["data/byte.tif", ["DATA_ENCODING=FOO"]]]  # invalid DATA_ENCODING
    tests += [
        ["data/byte.tif", ["JPEG2000_DRIVER=FOO", "SPATIAL_DIFFERENCING_ORDER=BAR"]]
    ]  # both cannot be used together
    tests += [["data/byte.tif", ["SPATIAL_DIFFERENCING_ORDER=3"]]]
    tests += [
        ["data/byte.tif", ["JPEG2000_DRIVER=THIS_IS_NOT_A_J2K_DRIVER"]]
    ]  # non-existing driver
    tests += [["data/byte.tif", ["JPEG2000_DRIVER=DERIVED"]]]  # Read-only driver
    tests += [["../gcore/data/cfloat32.tif", []]]  # complex data type
    tests += [["data/aaigrid/float64.asc", []]]  # no projection
    tests += [["data/test_nosrs.vrt", []]]  # no geotransform
    tests += [["data/envi/rotation.img", []]]  # geotransform with rotation terms
    gdal.GetDriverByName("GTiff").Create(
        "/vsimem/huge.tif", 65535, 65535, 1, options=["SPARSE_OK=YES"]
    )
    tests += [["/vsimem/huge.tif", []]]  # too many pixels

    for (
        filename,
        options,
    ) in tests:
        tmpfilename = "/vsimem/out.grb2"
        src_ds = gdal.Open(filename)
        gdal.ErrorReset()
        with gdaltest.error_handler():
            out_ds = gdaltest.grib_drv.CreateCopy(tmpfilename, src_ds, options=options)

        error_msg = gdal.GetLastErrorMsg()
        assert error_msg != "", "expected warning for %s, %s" % (
            str(filename),
            str(options),
        )
        assert out_ds is None, "expected null return for %s, %s" % (
            str(filename),
            str(options),
        )
        out_ds = None
        gdal.Unlink(tmpfilename)

    gdal.Unlink("/vsimem/huge.tif")

    with gdaltest.error_handler():
        out_ds = gdal.Translate("/i/do_not/exist.grb2", "data/byte.tif", format="GRIB")
    assert out_ds is None, "expected null return"


###############################################################################
# Test writing temperatures with automatic Celsius -> Kelvin conversion


def test_grib_grib2_write_temperatures():

    for (src_type, data_encoding, input_unit) in [
        (gdal.GDT_Float32, "IEEE_FLOATING_POINT", None),
        (gdal.GDT_Float32, "IEEE_FLOATING_POINT", "C"),
        (gdal.GDT_Float32, "IEEE_FLOATING_POINT", "K"),
        (gdal.GDT_Float64, "IEEE_FLOATING_POINT", None),
        (gdal.GDT_Float32, "SIMPLE_PACKING", None),
    ]:

        src_ds = gdal.GetDriverByName("MEM").Create("", 2, 2, 1, src_type)
        src_ds.SetGeoTransform([2, 1, 0, 49, 0, -1])
        sr = osr.SpatialReference()
        sr.SetFromUserInput("WGS84")
        src_ds.SetProjection(sr.ExportToWkt())
        src_ds.WriteRaster(
            0,
            0,
            2,
            2,
            struct.pack(4 * "f", 25.0, 25.1, 25.1, 25.2),
            buf_type=gdal.GDT_Float32,
        )

        tmpfilename = "/vsimem/out.grb2"
        options = [
            "DATA_ENCODING=" + data_encoding,
            "PDS_PDTN=8",
            "PDS_TEMPLATE_NUMBERS=0 5 2 0 0 0 255 255 1 0 0 0 43 1 0 0 0 0 0 255 129 255 255 255 255 7 216 2 23 12 0 0 1 0 0 0 0 3 255 1 0 0 0 12 1 0 0 0 0",
        ]
        if input_unit is not None:
            options += ["INPUT_UNIT=" + input_unit]
        gdaltest.grib_drv.CreateCopy(tmpfilename, src_ds, options=options)

        out_ds = gdal.Open(tmpfilename)
        got_vals = struct.unpack(4 * "d", out_ds.ReadRaster())
        out_ds = None
        gdal.Unlink(tmpfilename)
        if input_unit != "K":
            expected_vals = (25.0, 25.1, 25.1, 25.2)
        else:
            expected_vals = (25.0 - 273.15, 25.1 - 273.15, 25.1 - 273.15, 25.2 - 273.15)
        assert (
            max(
                [
                    abs((got_vals[i] - expected_vals[i]) / expected_vals[i])
                    for i in range(4)
                ]
            )
            <= 1e-4
        ), "fail with data_encoding = %s and type = %s" % (data_encoding, str(src_type))


###############################################################################


@pytest.mark.parametrize(
    "datatype", [gdal.GDT_Byte, gdal.GDT_Float32], ids=gdal.GetDataTypeName
)
def test_grib_grib2_write_nodata(datatype):

    src_ds = gdal.GetDriverByName("MEM").Create("", 2, 2, 1, datatype)
    src_ds.SetGeoTransform([2, 1, 0, 49, 0, -1])
    sr = osr.SpatialReference()
    sr.SetFromUserInput("WGS84")
    src_ds.SetProjection(sr.ExportToWkt())
    src_ds.GetRasterBand(1).SetNoDataValue(123)
    tmpfilename = "/vsimem/out.grb2"
    options = ["DATA_ENCODING=COMPLEX_PACKING"]
    gdaltest.grib_drv.CreateCopy(tmpfilename, src_ds, options=options)

    ds = gdal.Open(tmpfilename)
    assert ds.GetRasterBand(1).GetNoDataValue() == 123
    got_vals = struct.unpack(4 * "d", ds.ReadRaster())
    ds = None
    gdal.Unlink(tmpfilename)
    for i in range(4):
        assert got_vals[i] == 0.0


###############################################################################


@pytest.mark.parametrize(
    "datatype", [gdal.GDT_Byte, gdal.GDT_Float32], ids=gdal.GetDataTypeName
)
def test_grib_grib2_write_nodata_only(datatype):

    src_ds = gdal.GetDriverByName("MEM").Create("", 2, 2, 1, datatype)
    src_ds.SetGeoTransform([2, 1, 0, 49, 0, -1])
    sr = osr.SpatialReference()
    sr.SetFromUserInput("WGS84")
    src_ds.SetProjection(sr.ExportToWkt())
    src_ds.GetRasterBand(1).SetNoDataValue(12.3)
    src_ds.WriteRaster(
        0,
        0,
        2,
        2,
        struct.pack(4 * "f", 12.3, 12.3, 12.3, 12.3),
        buf_type=gdal.GDT_Float32,
    )
    tmpfilename = "/vsimem/out.grb2"
    options = ["DATA_ENCODING=COMPLEX_PACKING"]
    gdaltest.grib_drv.CreateCopy(tmpfilename, src_ds, options=options)

    ds = gdal.Open(tmpfilename)
    if datatype == gdal.GDT_Byte:
        assert ds.GetRasterBand(1).GetNoDataValue() == 12
    else:
        assert ds.GetRasterBand(1).GetNoDataValue() == pytest.approx(12.3, rel=1e-4)
    got_vals = struct.unpack(4 * "d", ds.ReadRaster())
    ds = None
    gdal.Unlink(tmpfilename)
    if datatype == gdal.GDT_Byte:
        expected_vals = (12, 12, 12, 12)
    else:
        expected_vals = (12.3, 12.3, 12.3, 12.3)
    assert got_vals == pytest.approx(expected_vals, rel=1e-4)


###############################################################################


@pytest.mark.parametrize(
    "datatype", [gdal.GDT_Byte, gdal.GDT_Float32], ids=gdal.GetDataTypeName
)
def test_grib_grib2_write_full_OneData(datatype):

    src_ds = gdal.GetDriverByName("MEM").Create("", 2, 2, 1, datatype)
    src_ds.SetGeoTransform([2, 1, 0, 49, 0, -1])
    sr = osr.SpatialReference()
    sr.SetFromUserInput("WGS84")
    src_ds.SetProjection(sr.ExportToWkt())
    src_ds.GetRasterBand(1).SetNoDataValue(123)
    src_ds.WriteRaster(
        0,
        0,
        2,
        2,
        struct.pack(4 * "f", 25.4, 25.4, 25.4, 25.4),
        buf_type=gdal.GDT_Float32,
    )
    tmpfilename = "/vsimem/out.grb2"
    options = ["DATA_ENCODING=COMPLEX_PACKING"]
    gdaltest.grib_drv.CreateCopy(tmpfilename, src_ds, options=options)

    ds = gdal.Open(tmpfilename)
    assert ds.GetRasterBand(1).GetNoDataValue() == 123
    got_vals = struct.unpack(4 * "d", ds.ReadRaster())
    ds = None
    gdal.Unlink(tmpfilename)
    if datatype == gdal.GDT_Byte:
        expected_vals = (25, 25, 25, 25)
    else:
        expected_vals = (25.4, 25.4, 25.4, 25.4)
    assert got_vals == pytest.approx(expected_vals, rel=1e-4)


###############################################################################


def test_grib_grib2_write_mix_nodata_and_a_single_data():
    src_ds = gdal.Open("data/grib/one_value_and_nodata_points.grb2")
    tmpfilename = "/vsimem/out.grb2"
    options = [
        "DATA_ENCODING=COMPLEX_PACKING",
    ]
    gdaltest.grib_drv.CreateCopy(tmpfilename, src_ds, options=options)
    src_ds = None

    ds = gdal.Open(tmpfilename)
    assert ds.GetRasterBand(1).GetNoDataValue() == 9999
    got_vals = struct.unpack(400 * "d", ds.ReadRaster())
    ds = None
    gdal.Unlink(tmpfilename)
    assert got_vals[0] == 9999
    assert got_vals[6] == pytest.approx(0.01, rel=1e-4)


###############################################################################
# Test GRIB2 file with JPEG2000 codestream on a single line (#6719)


def test_grib_online_grib2_jpeg2000_single_line():

    if not has_jp2kdrv():
        pytest.skip()

    filename = "CMC_hrdps_continental_PRATE_SFC_0_ps2.5km_2017111712_P001-00.grib2"
    if not gdaltest.download_file(
        "http://download.osgeo.org/gdal/data/grib/" + filename
    ):
        pytest.skip()

    ds = gdal.Open("tmp/cache/" + filename)
    cs = ds.GetRasterBand(1).Checksum()
    assert cs != 0, "Could not open file"
    nd = ds.GetRasterBand(1).GetNoDataValue()
    assert nd == 9999, "Bad nodata value"
    md = ds.GetRasterBand(1).GetMetadata()
    expected_md = {
        "GRIB_REF_TIME": "1510920000",
        "GRIB_VALID_TIME": "1510923600",
        "GRIB_FORECAST_SECONDS": "3600",
        "GRIB_UNIT": "[kg/(m^2 s)]",
        "GRIB_PDS_TEMPLATE_NUMBERS": "1 7 2 50 50 0 0 0 0 0 0 0 60 1 0 0 0 0 0 255 255 255 255 255 255",
        "GRIB_PDS_PDTN": "0",
        "GRIB_COMMENT": "Precipitation rate [kg/(m^2 s)]",
        "GRIB_SHORT_NAME": "0-SFC",
        "GRIB_ELEMENT": "PRATE",
    }
    for k in expected_md:
        assert k in md and md[k] == expected_md[k], "Did not get expected metadata"


###############################################################################
# Template 4.12 with Derived forecast = spread. Do not do unit conversion !


def test_grib_grib2_derived_forecast_spread():

    ds = gdal.Open("data/grib/template_4_12_spread.grb2")
    band = ds.GetRasterBand(1)
    assert band.GetMetadataItem("GRIB_UNIT") == "[spread]"
    assert band.ComputeRasterMinMax() == (0.24296024441719055, 0.24296024441719055)

    out_ds = gdaltest.grib_drv.CreateCopy("/vsimem/out.grb2", ds)
    band = out_ds.GetRasterBand(1)
    assert band.GetMetadataItem("GRIB_UNIT") == "[spread]"
    assert band.ComputeRasterMinMax() == (0.24296024441719055, 0.24296024441719055)
    out_ds = None

    gdal.Unlink("/vsimem/out.grb2")


###############################################################################
# Template 4.48 with Optical Properties of Aerosol


def test_grib_grib2_template_4_48():

    ds = gdal.Open("data/grib/template_4_48.grb2")
    band = ds.GetRasterBand(1)
    assert band.GetMetadataItem("GRIB_UNIT") == "[1/kg]"
    assert band.GetMetadataItem("GRIB_ELEMENT") == "ASNCON"
    assert band.GetMetadataItem("GRIB_SHORT_NAME") == "0-EATM"


###############################################################################
# Test reading product whose scan flag is not 64


def test_grib_grib2_scan_flag_not_64():

    ds = gdal.Open("/vsisparse/data/grib/blend.t17z.master.f001.co.grib2.sparse.xml")
    gt = ds.GetGeoTransform()
    expected_gt = (
        -3272421.457337171,
        2539.703,
        0.0,
        3790842.1060354356,
        0.0,
        -2539.703,
    )
    assert gt == pytest.approx(expected_gt, rel=1e-6)


###############################################################################
# Test reading message with subgrids


@pytest.mark.parametrize("use_idx", [True, False])
def test_grib_grib2_read_subgrids(use_idx):

    # data/grib/subgrids.grib2 generated with:
    # gdal_translate ../autotest/gcore/data/byte.tif band1.tif
    # gdal_translate ../autotest/gcore/data/byte.tif band2.tif -scale 0 255 255 0
    # gdalbuildvrt -separate tmp.vrt band1.tif band2.tif
    # gdal_translate tmp.vrt ../autotest/gdrivers/data/grib/subgrids.grib2 -co "BAND_1_PDS_TEMPLATE_ASSEMBLED_VALUES=2 2 2 0 84 0 0 1 0 220 0 0 255 0 0" -co "BAND_2_PDS_TEMPLATE_ASSEMBLED_VALUES=2 3 2 0 84 0 0 1 0 220 0 0 255 0 0" -co "IDS=CENTER=7(US-NCEP) SUBCENTER=0 MASTER_TABLE=2 LOCAL_TABLE=1 SIGNF_REF_TIME=1(Start_of_Forecast) REF_TIME=2020-09-26T00:00:00Z PROD_STATUS=0(Operational) TYPE=1(Forecast)" -co WRITE_SUBGRIDS=YES
    ds = gdal.OpenEx(
        "data/grib/subgrids.grib2",
        open_options=(["USE_IDX=YES"] if use_idx else ["USE_IDX=NO"]),
    )
    assert ds.GetRasterBand(1).Checksum() == 4672
    assert ds.GetRasterBand(2).Checksum() == 4563
    expected_ids = "CENTER=7(US-NCEP) SUBCENTER=0 MASTER_TABLE=2 LOCAL_TABLE=0 SIGNF_REF_TIME=1(Start_of_Forecast) REF_TIME=2020-09-26T00:00:00Z PROD_STATUS=0(Operational) TYPE=1(Forecast)"
    assert ds.GetRasterBand(1).GetMetadataItem("GRIB_IDS") == expected_ids
    assert (
        ds.GetRasterBand(1).GetMetadataItem("GRIB_COMMENT")
        == "u-component of wind [m/s]"
    )
    if use_idx:
        assert (
            ds.GetRasterBand(1).GetDescription() == "UGRD:planetary boundary layer:anl"
        )
        assert (
            ds.GetRasterBand(2).GetDescription() == "VGRD:planetary boundary layer:anl"
        )
    else:
        assert (
            ds.GetRasterBand(1).GetDescription()
            == "0[-] RESERVED(220) (Reserved for local use)"
        )
        assert (
            ds.GetRasterBand(2).GetDescription()
            == "0[-] RESERVED(220) (Reserved for local use)"
        )
    assert ds.GetRasterBand(2).GetMetadataItem("GRIB_IDS") == expected_ids
    assert (
        ds.GetRasterBand(2).GetMetadataItem("GRIB_COMMENT")
        == "v-component of wind [m/s]"
    )
    assert (
        ds.GetRasterBand(1).GetMetadataItem("GRIB_PDS_TEMPLATE_ASSEMBLED_VALUES")
        == "2 2 2 0 84 0 0 1 0 220 0 0 255 0 0"
    )
    assert (
        ds.GetRasterBand(2).GetMetadataItem("GRIB_PDS_TEMPLATE_ASSEMBLED_VALUES")
        == "2 3 2 0 84 0 0 1 0 220 0 0 255 0 0"
    )


###############################################################################
# Test reading message with subgrids with second subgrid reusing the bitmap from the first one
# Fixes https://github.com/OSGeo/gdal/issues/3099


def test_grib_grib2_read_subgrids_reuse_bitmap():

    # File generated with gdal_translate ../autotest/gdrivers/data/grib/subgrids.grib2 ../autotest/gdrivers/data/grib/subgrids_reuse_bitmap.grib2 --config GRIB_WRITE_BITMAP_TEST YES -co "BAND_1_PDS_TEMPLATE_ASSEMBLED_VALUES=2 2 2 0 84 0 0 1 0 220 0 0 255 0 0" -co "BAND_2_PDS_TEMPLATE_ASSEMBLED_VALUES=2 3 2 0 84 0 0 1 0 220 0 0 255 0 0" -co "IDS=CENTER=7(US-NCEP) SUBCENTER=0 MASTER_TABLE=2 LOCAL_TABLE=1 SIGNF_REF_TIME=1(Start_of_Forecast) REF_TIME=2020-09-26T00:00:00Z PROD_STATUS=0(Operational) TYPE=1(Forecast)" -co WRITE_SUBGRIDS=YES -co "DATA_ENCODING=SIMPLE_PACKING"
    ds = gdal.Open("data/grib/subgrids_reuse_bitmap.grib2")
    assert ds.GetRasterBand(1).Checksum() == 4672
    assert ds.GetRasterBand(2).Checksum() == 4563


###############################################################################
# Test reading and writing GRIBv2 with 0-360 longitudes
# Fixes https://github.com/OSGeo/gdal/issues/4524


@pytest.mark.parametrize(
    "test",
    [
        # Only the full globe can use split&swap
        {
            "file": "data/grib/gfs.t06z.pgrb2.1p0.grib2",
            "geo": (-185.0, 10.0, 0.0, 90.125, 0.0, -10.0),
            "band1csum": 7514,
        },
        # This one is very unorthodox and likely to trigger bugs
        {
            "file": "data/grib/gfs.t06z.pgrb2.1p0.partial_across_am.grib2",
            "geo": (24.875, 10.0, 0.0, 90.125, 0.0, -10.0),
            "band1csum": 5060,
        },
        # This one should have only the longitudes translation
        {
            "file": "data/grib/gfs.t06z.pgrb2.1p0.partial_east_of_am.grib2",
            "geo": (-60.125, 10.0, 0.0, 90.125, 0.0, -10.0),
            "band1csum": 698,
        },
        # This one should be identical with and without the translation
        {
            "file": "data/grib/gfs.t06z.pgrb2.1p0.partial_west_of_am.grib2",
            "geo": (24.875, 10.0, 0.0, 90.125, 0.0, -10.0),
            "band1csum": 601,
        },
        # This one should have split&swap enabled even if the longitude range slightly exceeds 360°
        {
            "file": "data/grib/gfswave-11.t00z.global.0p25.f000.grib2",
            "geo": (-180.125, 0.25, 0.0, 90.125, 0.0, -0.25),
            "band1csum": 52492,
        },
    ],
    ids=lambda x: x["file"],
)
def test_grib_grib2_split_and_swap(test):
    tmpfilename = "/vsimem/out.grb2"

    ds = gdal.Open(test["file"])
    gt = ds.GetGeoTransform()
    assert gt == pytest.approx(test["geo"], rel=1e-6)
    assert ds.GetRasterBand(1).Checksum() == test["band1csum"]

    out_ds = gdaltest.grib_drv.CreateCopy(tmpfilename, ds)
    gt = out_ds.GetGeoTransform()
    assert gt == pytest.approx(test["geo"], rel=1e-6)
    assert out_ds.GetRasterBand(1).Checksum() == test["band1csum"]

    out_ds = None
    gdal.Unlink(tmpfilename)
    ds = None


def test_grib_grib2_disable_split_and_swap():
    with gdaltest.config_option("GRIB_ADJUST_LONGITUDE_RANGE", "NO"):
        tmpfilename = "/vsimem/out.grb2"
        # This the untranslated range
        expected_gt = (-0.125, 10.0, 0.0, 90.125, 0.0, -10.0)

        ds = gdal.Open("data/grib/gfs.t06z.pgrb2.1p0.grib2")
        gt = ds.GetGeoTransform()
        assert gt == pytest.approx(expected_gt, rel=1e-6)
        assert ds.GetRasterBand(1).Checksum() == 7674

        out_ds = gdaltest.grib_drv.CreateCopy(tmpfilename, ds)
        gt = out_ds.GetGeoTransform()
        assert gt == pytest.approx(expected_gt, rel=1e-6)
        assert out_ds.GetRasterBand(1).Checksum() == 7674

        out_ds = None
        gdal.Unlink(tmpfilename)
        ds = None


# https://github.com/OSGeo/gdal/issues/4611
def test_grib_grib2_split_and_swap_offset_am():
    tempFilename = "/vsimem/temp.grb2"
    outFilename = "/vsimem/out.grb2"
    ds = gdal.Open("data/grib/gfs.t06z.pgrb2.1p0.grib2")

    geo = (119.975, 10.0, 0.0, 90.125, 0.0, -10.0)
    temp_ds = gdaltest.grib_drv.CreateCopy(tempFilename, ds)
    temp_ds.SetGeoTransform(geo)

    out_ds = gdaltest.grib_drv.CreateCopy(outFilename, temp_ds)
    gt = out_ds.GetGeoTransform()
    assert gt == pytest.approx((-185.0, 10.0, 0.0, 90.125, 0.0, -10.0), rel=1e-6)

    out_ds = None
    temp_ds = None
    gdal.Unlink(tempFilename)
    gdal.Unlink(outFilename)


# Test sidecar file support
# https://github.com/OSGeo/gdal/issues/3799


def test_grib_grib2_sidecar():

    ds_idx = gdal.Open("data/grib/gfs.t06z.pgrb2.10p0.f010.grib2")
    assert ds_idx.RasterCount == 6
    assert (
        ds_idx.GetRasterBand(6).GetDescription()
        == "VGRD:planetary boundary layer:10 hour fcst"
    ), "Description does not match, sidecar index is probably ignored"
    assert ds_idx.GetRasterBand(2).GetMetadataItem("GRIB_ELEMENT") == "REFD"
    assert ds_idx.GetRasterBand(3).GetMetadataItem("GRIB_ELEMENT") == "REFC"
    assert (
        ds_idx.GetRasterBand(6).GetMetadataItem("GRIB_PDS_TEMPLATE_ASSEMBLED_VALUES")
        == "2 3 2 0 96 0 0 1 10 220 0 0 255 0 0"
    )
    assert ds_idx.GetRasterBand(6).GetMetadataItem("GRIB_REF_TIME") == "1631944800"
    assert ds_idx.GetRasterBand(6).GetMetadataItem("GRIB_VALID_TIME") == "1631980800"
    assert ds_idx.GetRasterBand(6).GetMetadataItem("GRIB_FORECAST_SECONDS") == "36000"
    assert ds_idx.GetRasterBand(1).Checksum() == 59985
    assert ds_idx.GetRasterBand(2).Checksum() == 59986
    assert ds_idx.GetRasterBand(6).Checksum() == 206

    ds_no_idx = gdal.OpenEx(
        "data/grib/gfs.t06z.pgrb2.10p0.f010.grib2",
        gdal.GA_ReadOnly,
        open_options=["USE_IDX=NO"],
    )
    assert ds_no_idx.RasterCount == ds_idx.RasterCount
    assert (
        ds_no_idx.GetRasterBand(6).GetDescription()
        == "0[-] RESERVED(220) (Reserved for local use)"
    ), "Description does not match, sidecar index is probably loaded"
    for i in range(1, ds_no_idx.RasterCount):
        assert (
            ds_no_idx.GetRasterBand(i).Checksum() == ds_idx.GetRasterBand(i).Checksum()
        )
        assert (
            ds_no_idx.GetRasterBand(i).GetMetadata().keys()
            == ds_idx.GetRasterBand(i).GetMetadata().keys()
        )
        for key in ds_no_idx.GetRasterBand(i).GetMetadata().keys():
            assert ds_no_idx.GetRasterBand(i).GetMetadataItem(
                key
            ) == ds_idx.GetRasterBand(i).GetMetadataItem(key)


# Test reading a (broken) mix of GRIBv2/GRIBv1 bands


def test_grib_grib1_2_mix_sidecar():

    ds_idx = gdal.Open("data/grib/broken_combined_grib2_grib1.grb2")
    assert ds_idx.RasterCount == 18
    assert (
        ds_idx.GetRasterBand(6).GetDescription()
        == "VGRD:planetary boundary layer:10 hour fcst"
    ), "Description does not match, sidecar index is probably ignored"
    assert (
        ds_idx.GetRasterBand(18).GetDescription() == "DIRSW:Ground or water surface:anl"
    ), "Description does not match, sidecar index is probably ignored"
    assert ds_idx.GetRasterBand(2).GetMetadataItem("GRIB_ELEMENT") == "REFD"
    assert ds_idx.GetRasterBand(18).GetMetadataItem("GRIB_ELEMENT") == "DIRSW"
    assert ds_idx.GetRasterBand(1).Checksum() == 59985
    assert ds_idx.GetRasterBand(18).Checksum() == 4794

    ds_no_idx = gdal.OpenEx(
        "data/grib/broken_combined_grib2_grib1.grb2",
        gdal.GA_ReadOnly,
        open_options=["USE_IDX=NO"],
    )
    assert ds_no_idx.RasterCount == ds_idx.RasterCount
    assert (
        ds_no_idx.GetRasterBand(6).GetDescription()
        == "0[-] RESERVED(220) (Reserved for local use)"
    ), "Description does not match, sidecar index is probably loaded"
    assert (
        ds_no_idx.GetRasterBand(18).GetDescription()
        == "1[-] SFC (Ground or water surface)"
    ), "Description does not match, sidecar index is probably ignored"
    for i in range(1, ds_no_idx.RasterCount):
        assert (
            ds_no_idx.GetRasterBand(i).Checksum() == ds_idx.GetRasterBand(i).Checksum()
        )
        assert (
            ds_no_idx.GetRasterBand(i).GetMetadata().keys()
            == ds_idx.GetRasterBand(i).GetMetadata().keys()
        )
        for key in ds_no_idx.GetRasterBand(i).GetMetadata().keys():
            assert ds_no_idx.GetRasterBand(i).GetMetadataItem(
                key
            ) == ds_idx.GetRasterBand(i).GetMetadataItem(key)


# Test reading a parameter that is only in WMO tables and not DEGRIB ones
def test_grib_grib2_parameter_in_wmo_tables_only():

    ds = gdal.Open("data/grib/parameter_in_wmo_tables_only.grb2")
    assert (
        ds.GetRasterBand(1).GetMetadataItem("GRIB_COMMENT")
        == "Latent heat net flux due to evaporation [W m-2]"
    )
    assert (
        ds.GetRasterBand(1).GetMetadataItem("GRIB_ELEMENT")
        == "Latent heat net flux due to evaporation"
    )
