1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
|
#!/usr/bin/env python3
############################################################################
#
# MODULE: Animation
# AUTHOR(S): Anna Kratochvilova
# PURPOSE: Tool for animating a series of GRASS raster and vector maps
# or a space time raster dataset
# COPYRIGHT: (C) 2012 by Anna Kratochvilova, 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: Tool for animating a series of raster and vector maps or a space time raster or vector dataset.
# % keyword: general
# % keyword: GUI
# % keyword: display
# % keyword: animation
# %end
# %option G_OPT_R_INPUTS
# % key: raster
# % description: Raster maps to animate
# % required: no
# % guisection: Input
# %end
# %option G_OPT_V_INPUTS
# % key: vector
# % label: Vector maps to animate
# % required: no
# % guisection: Input
# %end
# %option G_OPT_STRDS_INPUT
# % key: strds
# % description: Space time raster dataset to animate
# % required: no
# % guisection: Input
# %end
# %option G_OPT_STVDS_INPUT
# % key: stvds
# % description: Space time vector dataset to animate
# % required: no
# % guisection: Input
# %end
import grass.script as gscript
from grass.exceptions import FatalError
def main():
options, flags = gscript.parser()
# import wx only after running parser
# to avoid issues when only interface is needed
import grass.temporal as tgis
import wx
from grass.script.setup import set_gui_path
set_gui_path()
from core.giface import StandaloneGrassInterface
from core.layerlist import LayerList
from animation.frame import AnimationFrame, MAX_COUNT
from animation.data import AnimLayer
rast = options["raster"]
vect = options["vector"]
strds = options["strds"]
stvds = options["stvds"]
numInputs = 0
if rast:
numInputs += 1
if vect:
numInputs += 1
if strds:
numInputs += 1
if stvds:
numInputs += 1
if numInputs > 1:
gscript.fatal(
_("%s=, %s=, %s= and %s= are mutually exclusive.")
% ("raster", "vector", "strds", "stvds")
)
if numInputs > 0:
# We need to initialize the temporal framework in case
# a space time dataset was set on the command line so that
# the AnimLayer() class works correctly
try:
tgis.init()
except FatalError as e:
print(e)
layerList = LayerList()
if rast:
layer = AnimLayer()
layer.mapType = "raster"
layer.name = rast
layer.cmd = ["d.rast", "map={name}".format(name=rast.split(",")[0])]
layerList.AddLayer(layer)
if vect:
layer = AnimLayer()
layer.mapType = "vector"
layer.name = vect
layer.cmd = ["d.vect", "map={name}".format(name=vect.split(",")[0])]
layerList.AddLayer(layer)
if strds:
layer = AnimLayer()
layer.mapType = "strds"
layer.name = strds
layer.cmd = ["d.rast", "map="]
layerList.AddLayer(layer)
if stvds:
layer = AnimLayer()
layer.mapType = "stvds"
layer.name = stvds
layer.cmd = ["d.vect", "map="]
layerList.AddLayer(layer)
app = wx.App()
frame = AnimationFrame(
parent=None,
giface=StandaloneGrassInterface(),
title=_("Animation Tool - GRASS GIS"),
)
frame.CentreOnScreen()
frame.Show()
if len(layerList) >= 1:
# CallAfter added since it was crashing with wxPython 3 gtk
wx.CallAfter(frame.SetAnimations, [layerList] + [None] * (MAX_COUNT - 1))
app.MainLoop()
if __name__ == "__main__":
main()
|
