from matplotlib import pyplot as plt
from astropy.time import Time
from astropy.visualization import time_support

SCALE = "tai" # or "utc" etc

intervals = [
    (">> Years\n(2022–2077)",   "2022-03-22T12:30:30.9", "2077-03-22T12:30:32.1"),
    ("> Year\n(2022–2017)",    "2022-03-22T12:30:30.9", "2023-03-22T12:30:32.1"),
    ("< Year>\n(Mar–Jan)",   "2022-03-22T12:30:30.9", "2023-01-22T12:30:32.1"),
    ("Hours\n(12:30–13:31)", "2022-03-22T12:30:30.9", "2022-03-22T13:31:30.9"),
    ("Seconds\n(12:30:30–40)", "2022-03-22T12:30:30.9", "2022-03-22T12:30:40.9"),
]

MODES = [
    (False,     "simplify=False\n(full strings)"),
    (True,      "simplify=True\n(strip trailing zeros)"),
    ("concise", "simplify='concise'\n(vary-prat + offset)"),
]

fig, axes = plt.subplots(
    nrows=len(INTERVALS),
    ncols=len(MODES),
    figsize=(14, 10),
    constrained_layout=True,
)

fig.suptitle("time_support simplify modes", fontsize=14, fontweight="bold")

for col, (mode, mode_label) in enumerate(MODES):
    axes[0, col].set_title(mode_label, fontsize=10)

for row, (interval_label, t1, t2) in enumerate(INTERVALS):
    axes[row, 0].set_ylabel(interval_label, fontsize=9, rotation=0,
                            labelpad=110, va="center")

    for col, (mode, _) in enumerate(MODES):
        ax = axes[row, col]

        with time_support(simplify=mode, scale=SCALE):
            tstart = Time(t1, scale=SCALE)
            tend   = Time(t2, scale=SCALE)
            tmid   = Time((tstart.mjd + tend.mjd) / 2, format="mjd", scale=SCALE)
            # Plot a simple diagonal line so the axis is populated
            ax.plot(Time([t1, t2], scale=SCALE), [0, 1], color="steelblue", lw=1.5)
            ax.set_xlim(tstart, tend)

        ax.set_yticks([])
        ax.tick_params(axis="x", labelsize=7, rotation=30)