Geografic-Information-System/gui/wxpython/tplot/g.gui.tplot.py

#!/usr/bin/env python3
############################################################################
#
# MODULE:    g.gui.tplot.py
# AUTHOR(S): Luca Delucchi
# PURPOSE:   Temporal Plot Tool is a wxGUI component (based on matplotlib)
#            the user to see in a plot the values of one or more temporal
#            datasets for a queried point defined by a coordinate pair.
# COPYRIGHT: (C) 2014-2015 by Luca Delucchi, and the GRASS Development Team
#
#  This program is free software; you can redistribute it and/or modify
#  it under the terms of the GNU General Public License as published by
#  the Free Software Foundation; either version 2 of the License, or
#  (at your option) any later version.
#
#  This program is distributed in the hope that it will be useful,
#  but WITHOUT ANY WARRANTY; without even the implied warranty of
#  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
#  GNU General Public License for more details.
#
############################################################################

# %module
# % description: Plots the values of temporal datasets.
# % keywords: general
# % keywords: GUI
# % keywords: temporal
# % keywords: plot
# %end

# %flag
# % key: h
# % description: Set the header of CSV file, to be used with csv option
# %end

# %flag
# % key: l
# % description: Show simple linear regression model line
# %end

# %option G_OPT_STVDS_INPUTS
# % key: stvds
# % required: no
# %end

# %option G_OPT_STRDS_INPUTS
# % key: strds
# % required: no
# %end

# %option G_OPT_M_COORDS
# % required: no
# %end

# TODO use option G_OPT_V_CATS
# %option
# % key: cats
# % label: Categories of vectores features
# % description: To use only with stvds
# % required: no
# %end

# %option
# % key: attr
# % label: Name of attribute
# % description: Name of attribute which represent data for plotting
# % required: no
# %end

# %option G_OPT_F_OUTPUT
# % required: no
# % label: Name for output graphical file
# % description: Full path for output file containing the plot, ddd extension to specify format (.png, .pdf, .svg)
# %end

# %option G_OPT_F_OUTPUT
# % key: csv
# % required: no
# % label: Name for output CSV file
# % description: Full path for the CSV file containing the plotted data
# %end

# %option
# % key: title
# % label: Title for plot
# % description: The title for the output plot
# % required: no
# %end

# %option
# % key: xlabel
# % label: Label for x axis
# % description: The x axis label for the output plot
# % required: no
# %end

# %option
# % key: ylabel
# % label: Label for y axis
# % description: The y axis label for the output plot
# % required: no
# %end

# %option
# % key: size
# % type: string
# % label: The size for output image
# % description: It works only with output parameter
# % required: no
# %end

import grass.script as gscript


def main():
    options, flags = gscript.parser()

    import wx

    from grass.script.setup import set_gui_path

    set_gui_path()

    from core.giface import StandaloneGrassInterface

    try:
        from tplot.frame import TplotFrame
    except ImportError as e:
        gscript.fatal(e.message)
    rasters = None
    if options["strds"]:
        rasters = options["strds"].strip().split(",")
    coords = None
    if options["coordinates"]:
        coords = options["coordinates"].strip().split(",")
    cats = None
    if options["cats"]:
        cats = options["cats"]
    output = options["output"]
    vectors = None
    attr = None
    if options["stvds"]:
        vectors = options["stvds"].strip().split(",")
        if not options["attr"]:
            gscript.fatal(_("With stvds you have to set 'attr' option"))
        else:
            attr = options["attr"]
        if coords and cats:
            gscript.fatal(
                _(
                    "With stvds it is not possible to use 'coordinates' "
                    "and 'cats' options together"
                )
            )
        elif not coords and not cats:
            gscript.warning(
                _("With stvds you have to use 'coordinates' or " "'cats' option")
            )
    title = None
    if options["title"]:
        title = options["title"]
    xlabel = None
    if options["xlabel"]:
        xlabel = options["xlabel"]
    ylabel = None
    if options["ylabel"]:
        ylabel = options["ylabel"]
    csvfile = None
    if options["csv"]:
        csvfile = options["csv"]
    app = wx.App()
    frame = TplotFrame(
        parent=None,
        giface=StandaloneGrassInterface(),
        title=_("Temporal Plot Tool - GRASS GIS"),
    )
    if flags["l"]:
        frame.linRegRaster.SetValue(state=True)
        frame.linRegVector.SetValue(state=True)
    frame.SetDatasets(
        rasters,
        vectors,
        coords,
        cats,
        attr,
        title,
        xlabel,
        ylabel,
        csvfile,
        flags["h"],
        gscript.overwrite,
    )
    if output:
        frame.OnRedraw()
        if options["size"]:
            sizes = options["size"].strip().split(",")
            sizes = [int(s) for s in sizes]
            frame.canvas.SetSize(sizes)
        frame.canvas.figure.savefig(output)
    else:
        frame.Show()
        app.MainLoop()


if __name__ == "__main__":
    main()