# pyright: reportMissingImports=false
import os
import pygmt
from gplately.auxiliary import get_gplot, get_pygmt_basemap_figure
from gplately.mapping.pygmt_plot import PygmtPlotEngine
from gplately import Raster
from plate_model_manager import PresentDayRasterManager

if __name__ == "__main__":
    reconstruction_time = 55
    # tell PlotTopologies object to use the PygmtPlotEngine
    gplot = get_gplot(
        model="Muller2025",
        model_repo_dir="plate-model-repo",
        time=reconstruction_time,
        plot_engine=PygmtPlotEngine(),
    )
    # you need to know how to specify projection and region in GMT way
    fig = get_pygmt_basemap_figure(
        projection="N180/10c", region="d", frame=["xafg30", "yafg30"]
    )

    # reconstruct the topography grid for the specified reconstruction time,
    # and use it to create an illumination grid for shading
    data_dir = "./unittest-data"
    topo_file = PresentDayRasterManager(data_dir=data_dir).get_raster("topography")
    topo_grid = Raster(
        data=topo_file, plate_reconstruction=gplot.plate_reconstruction
    ).reconstruct(time=reconstruction_time)

    # download age grid CPT file from https://raw.githubusercontent.com/GPlates/gplately/refs/heads/master/tests-dir/unittest/create-age-grids-video/agegrid.cpt
    cpt_file = f"{data_dir}/agegrid.cpt"
    if not os.path.isfile(cpt_file):
        import urllib.request

        urllib.request.urlretrieve(
            "https://raw.githubusercontent.com/GPlates/gplately/refs/heads/master/tests-dir/unittest/create-age-grids-video/agegrid.cpt",
            cpt_file,
        )

    gplot.plot_grid(
        fig,
        "AgeGrids",
        cmap=cpt_file,
        nan_transparent=True,
        # shading=True,
        # shading="+a315+ne0.6",
        shading=topo_grid.to_data_array(),
    )

    # now you can plot some features with the PlotTopologies object
    gplot.plot_coastlines(
        fig,
        edgecolor="none",
        facecolor="gray",
        linewidth=0.1,
        central_meridian=180,
        gmtlabel="Coastlines",
    )

    gplot.plot_all_topological_sections(
        fig,
        plot_subduction_teeth=True,
        other_kwargs={
            "color": "black",
            "linewidth": 0.5,
            "gmtlabel": "Miscellaneous",
        },
        ridge_kwargs={"color": "red", "linewidth": 0.7, "gmtlabel": "Ridges"},
        transform_kwargs={"color": "green", "linewidth": 0.7, "gmtlabel": "Transforms"},
        trench_kwargs={
            "color": "blue",
            "linewidth": 0.7,
            "gmtlabel": "Subduction Zones",
        },
    )

    gplot.plot_plate_motion_vectors(fig, normalise=True, color="red")

    # use pygmt directly to plot title and legend
    fig.text(
        text=f"{reconstruction_time} Ma (Muller2025)",
        position="TC",
        no_clip=True,
        font="12p,Helvetica,black",
        offset="j0/-0.5c",
    )
    with pygmt.config(FONT_ANNOT_PRIMARY=4):
        fig.legend(position="jBL+o-1.0/0", box="+gwhite+p0.25p")

    fig.show(width=1200, crop="+m0.4c")

    out_f = "test-pygmt-plot.pdf"
    fig.savefig(out_f, crop="+m0.4c")
    print(f"The map has been saved to {out_f}.")

/home/runner/micromamba/envs/doc-env/lib/python3.14/site-packages/pygmt/clib/session.py:1648: RuntimeWarning: Grid may have irregular spacing in the 'lon' dimension, but GMT only supports regular spacing. Calculated regular spacing 0.09997222222222223 is assumed in the 'lon' dimension.

/home/runner/micromamba/envs/doc-env/lib/python3.14/site-packages/gplately/plot.py:1876: UserWarning: The 'plot_ridges' function has been changed since GPlately 1.3.0. You need to check your workflow to make sure the new 'plot_ridges' function still suits your purpose. In earlier releases of GPlately, we used an algorithm to identify the 'ridges' and 'transforms' within the gpml:MidOceanRidge features. Unfortunately, the algorithm did not work very well. So we have removed the algorithm and now the 'plot_ridges' function plots all the features which are labelled as gpml:MidOceanRidge in the reconstruction model.
/home/runner/micromamba/envs/doc-env/lib/python3.14/site-packages/gplately/decorators.py:67: UserWarning: The 'plot_transforms' function has been changed since GPlately 1.3.0. You need to check your workflow to make sure the new 'plot_transforms' function still suits your purpose. In earlier releases of GPlately, we used an algorithm to identify the 'ridges' and 'transforms' within the gpml:MidOceanRidge features. Unfortunately, the algorithm did not work very well. So we have removed the algorithm and now the 'plot_transforms' function plots all the features which are labelled as gpml:Transform in the reconstruction model.

/home/runner/micromamba/envs/doc-env/lib/python3.14/site-packages/gplately/mapping/pygmt_plot.py:382: FutureWarning: The 'uncertaintyfill' parameter has been deprecated since v0.18.0 and will be removed in v0.20.0. Please use 'uncertainty_fill' instead.

The map has been saved to test-pygmt-plot.pdf.