Function main

docs/plot/plot.py:313–354

Run main function of plotting script. Parses json-file with plot setups and runs the plotting for each plot setup.

(plotdefs, outdir, plots=[])

def main(plotdefs, outdir, plots=[]):
    """Run main function of plotting script.

    Parses json-file with plot setups and runs the plotting
    for each plot setup.
    """
    try:
        # Check 'proj' command, show version
        proc = run_proj_cmd(text=True)
        print(proc.stderr.splitlines()[0])
    except (FileNotFoundError, OSError) as e:
        sys.stderr.write(
            "'proj' binary not found, set the PROJ_EXE environment variable "
            "to point to your local 'proj' binary\n{}\n".format(e)
        )
        return 1

    outdir = Path(outdir)
    if not outdir.exists():
        outdir.mkdir(parents=True)
    if not outdir.is_dir():
        raise OSError("outdir is not a directory")

    plotdefs = json.loads(Path(plotdefs).read_text())

    data = {
        ("line", "low"): geojson2geom_array(LINE_LOW),
        ("line", "med"): geojson2geom_array(LINE_MED),
        ("poly", "low"): geojson2geom_array(POLY_LOW),
        ("poly", "med"): geojson2geom_array(POLY_MED),
    }

    for i, plotdef in enumerate(plotdefs):
        if plots != [] and plotdef["name"] not in plots:
            continue

        print("{}: {!r} {}".format(i, plotdef["filename"], plotdef["projstring"]))
        if "skip" in plotdef.keys():
            print("skipping")
            continue

        plotproj(plotdef, data[(plotdef["type"], plotdef["res"])], outdir)


if __name__ == "__main__":