1# https://pyrocko.org - GPLv3
2#
3# The Pyrocko Developers, 21st Century
4# ---|P------/S----------~Lg----------
6import copy
7import logging
8try:
9 from kite import Scene
10except ImportError as e:
11 print(e)
12 Scene = None
14import numpy as num
16from pyrocko import geometry
17from pyrocko.guts import Bool, String, List
18from pyrocko.gui.qt_compat import qw
19from pyrocko.gui.vtk_util import TrimeshPipe, faces_to_cells
21from .. import common
23from .base import Element, ElementState, CPTHandler, CPTState
25logger = logging.getLogger('kite_scene')
26guts_prefix = 'sparrow'
28km = 1e3
29d2r = num.pi/180.
32class SceneTileAdapter(object):
34 def __init__(self, scene):
35 self._scene = scene
37 def x(self):
38 # TODO how to handle E given in m
39 frame = self._scene.frame
40 x = num.zeros(frame.cols + 1)
41 x[0] = frame.E[0] - 0.5 * frame.dE
42 x[1:] = frame.E + 0.5 * frame.dE
43 x += frame.llLon
44 return x
46 def y(self):
47 # TODO how to handle N given in m
48 frame = self._scene.frame
49 y = num.zeros(frame.rows + 1)
50 y[0] = frame.N[0] - 0.5 * frame.dN
51 y[1:] = frame.N + 0.5 * frame.dN
52 y += frame.llLat
53 return y
55 @property
56 def data(self):
57 disp = self._scene.displacement
58 disp[num.isnan(disp)] = None
59 return disp
62class KiteMeshPipe(TrimeshPipe):
63 def __init__(self, tile, cells_cache=None, **kwargs):
64 lat_edge = tile.y()
65 lon_edge = tile.x()
66 data_center = tile.data
68 nlat = lat_edge.size
69 nlon = lon_edge.size
70 nvertices = nlat * nlon
72 assert nlat > 1 and nlon > 1
73 assert data_center.shape == (nlat-1, nlon-1)
75 rtp = num.empty((nvertices, 3))
76 rtp[:, 0] = 1.0
77 rtp[:, 1] = (num.repeat(lat_edge, nlon) + 90.) * d2r
78 rtp[:, 2] = num.tile(lon_edge, nlat) * d2r
79 vertices = geometry.rtp2xyz(rtp)
81 faces = geometry.topo_to_faces_quad(nlat, nlon)
83 self._tile = tile
84 self._raw_vertices = vertices
86 if cells_cache is not None:
87 if id(faces) not in cells_cache:
88 cells_cache[id(faces)] = faces_to_cells(faces)
90 cells = cells_cache[id(faces)]
91 else:
92 cells = faces_to_cells(faces)
94 data_center = data_center.flatten()
96 TrimeshPipe.__init__(
97 self, self._raw_vertices,
98 cells=cells, values=data_center, **kwargs)
101class KiteSceneElement(ElementState):
102 visible = Bool.T(default=True)
103 filename = String.T()
104 scene = None
107class KiteState(ElementState):
108 visible = Bool.T(default=True)
109 scenes = List.T(KiteSceneElement.T(), default=[])
110 cpt = CPTState.T(default=CPTState.D(cpt_name='seismic_r'))
112 def create(self):
113 element = KiteElement()
114 return element
116 def add_scene(self, scene):
117 self.scenes.append(scene)
119 def remove_scene(self, scene):
120 if scene in self.scenes:
121 self.scenes.remove(scene)
124class KiteElement(Element):
126 def __init__(self):
127 Element.__init__(self)
128 self._controls = None
129 self._meshes = {}
130 self._cells = {}
131 self.cpt_handler = CPTHandler()
133 def bind_state(self, state):
134 Element.bind_state(self, state)
135 self.talkie_connect(state, ['visible', 'scenes'], self.update)
137 self.cpt_handler.bind_state(state.cpt, self.update)
139 def unbind_state(self):
140 self.cpt_handler.unbind_state()
141 Element.unbind_state(self)
143 def get_name(self):
144 return 'Kite InSAR Scenes'
146 def set_parent(self, parent):
147 if Scene is None:
148 qw.QMessageBox.warning(
149 parent, 'Import Error',
150 'Software package Kite is needed to display InSAR scenes!')
151 return
153 self._parent = parent
154 self._parent.add_panel(
155 self.get_name(),
156 self._get_controls(),
157 visible=True,
158 title_controls=[
159 self.get_title_control_remove(),
160 self.get_title_control_visible()])
162 self.update()
164 def unset_parent(self):
165 self.unbind_state()
166 if self._parent:
167 for mesh in self._meshes:
168 self._parent.remove_actor(mesh.actor)
170 self._meshes.clear()
171 self._cells.clear()
173 if self._controls:
174 self._parent.remove_panel(self._controls)
175 self._controls = None
177 self._parent.update_view()
178 self._parent = None
180 def _load_scene_from_fn(self, fn):
181 try:
182 scene = Scene.load(fn)
183 except ImportError:
184 qw.QMessageBox.warning(
185 self._parent, 'Import Error',
186 'Could not load Kite scene from %s' % fn)
187 return
189 if scene.frame.spacing != 'degree':
190 logger.warning(
191 'Sparrow requires Scene spacing in degrees. '
192 'Skipped %s', fn)
194 return
196 return scene
198 def open_load_scene_dialog(self, *args):
199 caption = 'Select one or more Kite scenes to open'
201 fns, _ = qw.QFileDialog.getOpenFileNames(
202 self._parent, caption,
203 filter='YAML file (*.yml *.yaml)',
204 options=common.qfiledialog_options)
206 for fname in fns:
207 scene = self._load_scene_from_fn(fname)
209 if scene is None:
210 continue
212 logger.info('Adding Kite scene %s', fname)
214 scene_element = KiteSceneElement(filename=fname)
215 scene_element.scene = scene
216 self._state.add_scene(scene_element)
218 self.update()
220 def clear_scenes(self, *args):
221 logger.info('Clearing all loaded Kite scenes')
223 for mesh in self._meshes.values():
224 self._parent.remove_actor(mesh.actor)
226 self._meshes.clear()
227 self._state.scenes = []
229 self.update()
231 def update(self, *args):
232 state = self._state
234 for mesh in self._meshes.values():
235 self._parent.remove_actor(mesh.actor)
237 if self._state.visible:
238 for scene_element in state.scenes:
239 logger.info('Drawing Kite scene')
241 if scene_element.scene is None:
242 scene_element.scene = self._load_scene_from_fn(
243 scene_element.filename)
245 scene = scene_element.scene
247 scene_tile = SceneTileAdapter(scene)
249 k = (scene_tile, state.cpt.cpt_name)
251 if k not in self._meshes:
252 cpt = copy.deepcopy(
253 self.cpt_handler._cpts[state.cpt.cpt_name])
255 mesh = KiteMeshPipe(
256 scene_tile,
257 cells_cache=None,
258 cpt=cpt,
259 backface_culling=False)
261 values = scene_tile.data.flatten()
262 self.cpt_handler._values = values
263 self.cpt_handler.update_cpt()
265 mesh.set_shading('phong')
266 mesh.set_lookuptable(self.cpt_handler._lookuptable)
268 self._meshes[k] = mesh
269 else:
270 mesh = self._meshes[k]
271 self.cpt_handler.update_cpt()
273 if scene_element.visible:
274 self._parent.add_actor(mesh.actor)
276 self._parent.update_view()
278 def _get_controls(self):
279 if not self._controls:
280 frame = qw.QFrame()
281 layout = qw.QGridLayout()
282 frame.setLayout(layout)
284 pb_load = qw.QPushButton('Add Scene')
285 pb_load.clicked.connect(self.open_load_scene_dialog)
286 layout.addWidget(pb_load, 0, 1)
288 pb_clear = qw.QPushButton('Clear Scenes')
289 pb_clear.clicked.connect(self.clear_scenes)
290 layout.addWidget(pb_clear, 0, 2)
292 self.cpt_handler.cpt_controls(
293 self._parent, self._state.cpt, layout)
295 layout.addWidget(qw.QFrame(), 4, 0, 1, 3)
297 self._controls = frame
299 self._update_controls()
301 return self._controls
303 def _update_controls(self):
304 self.cpt_handler._update_cpt_combobox()
305 self.cpt_handler._update_cptscale_lineedit()
308__all__ = [
309 'KiteState',
310 'KiteElement'
311]