from matplotlib.figure import Figure

import numpy as np

from pathlib import Path

from datetime import datetime

from .utils import gitrev

def plotdiff(A: np.ndarray, B: np.ndarray, name: str, time: datetime, outdir: Path, refdir: Path):

A = A.squeeze()

B = B.squeeze()

if A.ndim != 2 or B.ndim != 2:

return None

fg = Figure(tight_layout=True, figsize=(12, 5))

axs = fg.subplots(1, 2)

if name.startswith(("J", "v")):

cmap = "bwr"

else:

cmap = None

hi = axs[0].pcolormesh(A, cmap=cmap)

fg.colorbar(hi, ax=axs[0])

axs[0].set_title(str(outdir))

hi = axs[1].pcolormesh(B, cmap=cmap)

fg.colorbar(hi, ax=axs[1])

axs[1].set_title(str(refdir))

ttxt = f"{name} {time.isoformat()} Git: {gitrev()}"

fg.suptitle(ttxt)

fn = outdir / f"{name}-diff-{time.isoformat().replace(':','')}.png"

print("writing", fn)

fg.savefig(fn)