"""Figure out the grid max/min values"""
import datetime
import os

import numpy as np
import pygrib

from matplotlib.colors import ListedColormap
from metpy.units import units
from pyiem.plot import MapPlot
from pyiem.plot.use_agg import plt


def plot():
    """Do plotting work"""
    cmap1 = plt.get_cmap("rainbow")
    colors = list(cmap1(np.arange(1, 7) / 7.0))
    # cmap2 = plt.get_cmap("Pastel1")
    # colors.extend(list(cmap2(np.arange(3) / 3.0)))
    cmap = ListedColormap(colors)

    cmap.set_over("yellow")
    cmap.set_under("white")
    prev = np.load("maxval.npy")
    today = np.load("maxval2.npy")
    lons = np.load("lons.npy")
    lats = np.load("lats.npy")
    mp = MapPlot(
        sector="custom",
        south=35,
        west=-105,
        north=49,
        east=-82,
        continentalcolor="tan",
        # statebordercolor="white",
        title=(
            "Max Air Temperature April 12 vs January 1 through April 11 2023"
        ),
        subtitle="based on hourly NCEP Real-Time Mesoscale Analysis",
    )

    # levels = [1, 4, 8, 12]
    # levels2 = list(range(24, 14 * 24 + 1, 48))
    # levels.extend(levels2)
    # levels = range(12, 145, 12)
    levels = range(1, 14, 2)
    mp.pcolormesh(
        lons,
        lats,
        today - prev,
        levels,
        cmap=cmap,
        clip_on=False,
        units=r"$^\circ$F",
        spacing="proportional",
        extend="both",
    )
    mp.drawcounties()
    mp.postprocess(filename="230413.png")


def process():
    """Go Main Go"""
    now = datetime.datetime(2022, 1, 1, 0)
    ets = datetime.datetime(2022, 9, 6, 1)
    interval = datetime.timedelta(hours=1)
    hours = None
    while now < ets:
        fn = now.strftime(
            "/mesonet/ARCHIVE/data/%Y/%m/%d/"
            "model/rtma/%H/rtma.t%Hz.awp2p5f000.grib2"
        )
        if not os.path.isfile(fn):
            print(f"missing {fn}")
        else:
            with pygrib.open(fn) as grbs:
                try:
                    t2 = grbs.select(shortName="2t")[0].values
                except Exception:
                    print(f"failed to get 2t in {fn}!")
                    now += interval
                    continue
                if hours is None:
                    hours = np.zeros(np.shape(t2))
                    lats, lons = grbs.select(shortName="2t")[0].latlons()
                    np.save("/tmp/lons", lons)
                    np.save("/tmp/lats", lats)
            t2 = (t2 * units("degK")).to(units("degF")).magnitude
            hours = np.where(t2 >= 95, hours + 1, hours)
            print(f"{now} max: {np.max(hours):.1f}")

        now += interval

    np.save("/tmp/hours2022", hours)


if __name__ == "__main__":
    process()
    # plot()