frame.py 31 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863
  1. """
  2. @package mapswipe.frame
  3. @brief Map Swipe Frame
  4. Classes:
  5. - dialogs::SwipeMapDialog
  6. (C) 2012 by the GRASS Development Team
  7. This program is free software under the GNU General Public License
  8. (>=v2). Read the file COPYING that comes with GRASS for details.
  9. @author Anna Kratochvilova <kratochanna gmail.com>
  10. """
  11. import os
  12. import sys
  13. import wx
  14. import grass.script as grass
  15. from gui_core.mapdisp import DoubleMapFrame
  16. from gui_core.dialogs import GetImageHandlers
  17. from gui_core.wrap import Slider
  18. from mapwin.base import MapWindowProperties
  19. from core.render import Map
  20. from mapdisp import statusbar as sb
  21. from core.debug import Debug
  22. from core.gcmd import GError, GMessage
  23. from core.layerlist import LayerListToRendererConverter
  24. from gui_core.query import QueryDialog, PrepareQueryResults
  25. from mapswipe.toolbars import SwipeMapToolbar, SwipeMainToolbar, SwipeMiscToolbar
  26. from mapswipe.mapwindow import SwipeBufferedWindow
  27. from mapswipe.dialogs import SwipeMapDialog, PreferencesDialog
  28. class SwipeMapFrame(DoubleMapFrame):
  29. def __init__(self, parent=None, giface=None,
  30. title=_("GRASS GIS Map Swipe"), name="swipe", **kwargs):
  31. DoubleMapFrame.__init__(self, parent=parent, title=title, name=name,
  32. firstMap=Map(), secondMap=Map(), **kwargs)
  33. Debug.msg(1, "SwipeMapFrame.__init__()")
  34. #
  35. # Add toolbars
  36. #
  37. self.AddToolbars()
  38. self._giface = giface
  39. #
  40. # create widgets
  41. #
  42. self.splitter = MapSplitter(parent=self, id=wx.ID_ANY)
  43. self.sliderH = Slider(self, id=wx.ID_ANY, style=wx.SL_HORIZONTAL)
  44. self.sliderV = Slider(self, id=wx.ID_ANY, style=wx.SL_VERTICAL)
  45. self.mapWindowProperties = MapWindowProperties()
  46. self.mapWindowProperties.setValuesFromUserSettings()
  47. self.mapWindowProperties.autoRenderChanged.connect(
  48. self.OnAutoRenderChanged)
  49. self.firstMapWindow = SwipeBufferedWindow(
  50. parent=self.splitter, giface=self._giface,
  51. properties=self.mapWindowProperties, Map=self.firstMap)
  52. self.secondMapWindow = SwipeBufferedWindow(
  53. parent=self.splitter, giface=self._giface,
  54. properties=self.mapWindowProperties, Map=self.secondMap)
  55. # bind query signal
  56. self.firstMapWindow.mapQueried.connect(self.Query)
  57. self.secondMapWindow.mapQueried.connect(self.Query)
  58. # bind tracking cursosr to mirror it
  59. self.firstMapWindow.Bind(
  60. wx.EVT_MOTION,
  61. lambda evt: self.TrackCursor(evt))
  62. self.secondMapWindow.Bind(
  63. wx.EVT_MOTION,
  64. lambda evt: self.TrackCursor(evt))
  65. self.MapWindow = self.firstMapWindow # current by default
  66. self.firstMapWindow.zoomhistory = self.secondMapWindow.zoomhistory
  67. self.SetBindRegions(True)
  68. self._mode = 'swipe'
  69. self._addPanes()
  70. self._bindWindowsActivation()
  71. self._setUpMapWindow(self.firstMapWindow)
  72. self._setUpMapWindow(self.secondMapWindow)
  73. self._mgr.GetPane('sliderV').Hide()
  74. self._mgr.GetPane('sliderH').Show()
  75. self.slider = self.sliderH
  76. self.InitStatusbar()
  77. self.Bind(wx.EVT_SIZE, self.OnSize)
  78. self.Bind(wx.EVT_IDLE, self.OnIdle)
  79. self.Bind(wx.EVT_CLOSE, self.OnCloseWindow)
  80. self.SetSize((800, 600))
  81. self._mgr.Update()
  82. self.rasters = {'first': None, 'second': None}
  83. self._inputDialog = None
  84. self._preferencesDialog = None
  85. self._queryDialog = None
  86. # default action in map toolbar
  87. self.GetMapToolbar().SelectDefault()
  88. self.resize = False
  89. wx.CallAfter(self.CallAfterInit)
  90. def TrackCursor(self, event):
  91. """Track cursor in one window and show cross in the other.
  92. Only for mirror mode.
  93. """
  94. if self._mode == 'swipe':
  95. event.Skip()
  96. return
  97. coords = event.GetPosition()
  98. if event.GetId() == self.secondMapWindow.GetId():
  99. self.firstMapWindow.DrawMouseCursor(coords=coords)
  100. else:
  101. self.secondMapWindow.DrawMouseCursor(coords=coords)
  102. event.Skip()
  103. def ActivateFirstMap(self, event=None):
  104. """Switch tracking direction"""
  105. super(SwipeMapFrame, self).ActivateFirstMap(event)
  106. self.firstMapWindow.ClearLines()
  107. self.firstMapWindow.Refresh()
  108. def ActivateSecondMap(self, event=None):
  109. """Switch tracking direction"""
  110. super(SwipeMapFrame, self).ActivateSecondMap(event)
  111. self.secondMapWindow.ClearLines()
  112. self.secondMapWindow.Refresh()
  113. def CallAfterInit(self):
  114. self.InitSliderBindings()
  115. self.splitter.SplitVertically(
  116. self.firstMapWindow, self.secondMapWindow, 0)
  117. self.splitter.Init()
  118. if not (self.rasters['first'] and self.rasters['second']):
  119. self.OnSelectLayers(event=None)
  120. def InitStatusbar(self):
  121. """Init statusbar (default items)."""
  122. # items for choice
  123. self.statusbarItems = [sb.SbCoordinates,
  124. sb.SbRegionExtent,
  125. sb.SbCompRegionExtent,
  126. sb.SbShowRegion,
  127. sb.SbAlignExtent,
  128. sb.SbResolution,
  129. sb.SbDisplayGeometry,
  130. sb.SbMapScale,
  131. sb.SbGoTo,
  132. sb.SbProjection]
  133. # create statusbar and its manager
  134. statusbar = self.CreateStatusBar(number=4, style=0)
  135. statusbar.SetMinHeight(24)
  136. statusbar.SetStatusWidths([-5, -2, -1, -1])
  137. self.statusbarManager = sb.SbManager(
  138. mapframe=self, statusbar=statusbar)
  139. # fill statusbar manager
  140. self.statusbarManager.AddStatusbarItemsByClass(
  141. self.statusbarItems, mapframe=self, statusbar=statusbar)
  142. self.statusbarManager.AddStatusbarItem(
  143. sb.SbMask(self, statusbar=statusbar, position=2))
  144. sbRender = sb.SbRender(self, statusbar=statusbar, position=3)
  145. self.statusbarManager.AddStatusbarItem(sbRender)
  146. self.statusbarManager.Update()
  147. def ResetSlider(self):
  148. if self.splitter.GetSplitMode() == wx.SPLIT_VERTICAL:
  149. size = self.splitter.GetSize()[0]
  150. else:
  151. size = self.splitter.GetSize()[1]
  152. self.slider.SetRange(0, size)
  153. self.slider.SetValue(self.splitter.GetSashPosition())
  154. def InitSliderBindings(self):
  155. self.sliderH.Bind(wx.EVT_SPIN, self.OnSliderPositionChanging)
  156. self.sliderH.Bind(
  157. wx.EVT_SCROLL_THUMBRELEASE,
  158. self.OnSliderPositionChanged)
  159. self.sliderV.Bind(wx.EVT_SPIN, self.OnSliderPositionChanging)
  160. self.sliderV.Bind(
  161. wx.EVT_SCROLL_THUMBRELEASE,
  162. self.OnSliderPositionChanged)
  163. self.splitter.Bind(
  164. wx.EVT_SPLITTER_SASH_POS_CHANGING,
  165. self.OnSashChanging)
  166. self.splitter.Bind(
  167. wx.EVT_SPLITTER_SASH_POS_CHANGED,
  168. self.OnSashChanged)
  169. def OnSliderPositionChanging(self, event):
  170. """Slider changes its position, sash must be moved too."""
  171. Debug.msg(5, "SwipeMapFrame.OnSliderPositionChanging()")
  172. self.GetFirstWindow().movingSash = True
  173. self.GetSecondWindow().movingSash = True
  174. pos = event.GetPosition()
  175. if pos > 0:
  176. self.splitter.SetSashPosition(pos)
  177. self.splitter.OnSashChanging(None)
  178. def OnSliderPositionChanged(self, event):
  179. """Slider position changed, sash must be moved too."""
  180. Debug.msg(5, "SwipeMapFrame.OnSliderPositionChanged()")
  181. self.splitter.SetSashPosition(event.GetPosition())
  182. self.splitter.OnSashChanged(None)
  183. def OnSashChanging(self, event):
  184. """Sash position is changing, slider must be moved too."""
  185. Debug.msg(5, "SwipeMapFrame.OnSashChanging()")
  186. self.slider.SetValue(self.splitter.GetSashPosition())
  187. event.Skip()
  188. def OnSashChanged(self, event):
  189. """Sash position changed, slider must be moved too."""
  190. Debug.msg(5, "SwipeMapFrame.OnSashChanged()")
  191. self.OnSashChanging(event)
  192. event.Skip()
  193. def OnSize(self, event):
  194. Debug.msg(4, "SwipeMapFrame.OnSize()")
  195. self.resize = grass.clock()
  196. super(SwipeMapFrame, self).OnSize(event)
  197. def OnIdle(self, event):
  198. if self.resize and grass.clock() - self.resize > 0.2:
  199. w1 = self.GetFirstWindow()
  200. w2 = self.GetSecondWindow()
  201. sizeAll = self.splitter.GetSize()
  202. w1.SetClientSize(sizeAll)
  203. w2.SetClientSize(sizeAll)
  204. w1.OnSize(event)
  205. w2.OnSize(event)
  206. self.ResetSlider()
  207. self.resize = False
  208. def OnAutoRenderChanged(self, value):
  209. """Auto rendering state changed."""
  210. style = self.splitter.GetWindowStyle()
  211. style ^= wx.SP_LIVE_UPDATE
  212. self.splitter.SetWindowStyle(style)
  213. def AddToolbars(self):
  214. """Add defined toolbar to the window
  215. Currently known toolbars are:
  216. - 'swipeMap' - basic map toolbar
  217. - 'swipeMain' - swipe functionality
  218. - 'swipeMisc' - misc (settings, help)
  219. """
  220. self.toolbars["swipeMap"] = SwipeMapToolbar(self, self._toolSwitcher)
  221. self._mgr.AddPane(self.toolbars["swipeMap"],
  222. wx.aui.AuiPaneInfo().
  223. Name("swipeMap").Caption(_("Map Toolbar")).
  224. ToolbarPane().Top().
  225. LeftDockable(False).RightDockable(False).
  226. BottomDockable(False).TopDockable(True).
  227. CloseButton(False).Layer(2).Row(1).Position(1).
  228. BestSize((self.toolbars["swipeMap"].GetBestSize())))
  229. self.toolbars["swipeMain"] = SwipeMainToolbar(self)
  230. self._mgr.AddPane(self.toolbars["swipeMain"],
  231. wx.aui.AuiPaneInfo().
  232. Name("swipeMain").Caption(_("Main Toolbar")).
  233. ToolbarPane().Top().
  234. LeftDockable(False).RightDockable(False).
  235. BottomDockable(False).TopDockable(True).
  236. CloseButton(False).Layer(2).Row(1).Position(0).
  237. BestSize((self.toolbars["swipeMain"].GetBestSize())))
  238. self.toolbars["swipeMisc"] = SwipeMiscToolbar(self)
  239. self._mgr.AddPane(self.toolbars["swipeMisc"],
  240. wx.aui.AuiPaneInfo().
  241. Name("swipeMisc").Caption(_("Misc Toolbar")).
  242. ToolbarPane().Top().
  243. LeftDockable(False).RightDockable(False).
  244. BottomDockable(False).TopDockable(True).
  245. CloseButton(False).Layer(2).Row(1).Position(2).
  246. BestSize((self.toolbars["swipeMisc"].GetBestSize())))
  247. def _addPanes(self):
  248. """Add splitter window and sliders to aui manager"""
  249. # splitter window
  250. self._mgr.AddPane(self.splitter, wx.aui.AuiPaneInfo().
  251. Name('splitter').CaptionVisible(False).PaneBorder(True).
  252. Dockable(False).Floatable(False).CloseButton(False).
  253. Center().Layer(1).BestSize((self.splitter.GetBestSize())))
  254. # sliders
  255. self._mgr.AddPane(self.sliderH, wx.aui.AuiPaneInfo().
  256. Name('sliderH').CaptionVisible(False).PaneBorder(False).
  257. CloseButton(False).Gripper(True).GripperTop(False).
  258. BottomDockable(True).TopDockable(True).
  259. LeftDockable(False).RightDockable(False).
  260. Bottom().Layer(1).BestSize((self.sliderH.GetBestSize())))
  261. self._mgr.AddPane(self.sliderV, wx.aui.AuiPaneInfo().
  262. Name('sliderV').CaptionVisible(False).PaneBorder(False).
  263. CloseButton(False).Gripper(True).GripperTop(True).
  264. BottomDockable(False).TopDockable(False).
  265. LeftDockable(True).RightDockable(True).
  266. Right().Layer(1).BestSize((self.sliderV.GetBestSize())))
  267. def ZoomToMap(self):
  268. """
  269. Set display extents to match selected raster (including NULLs)
  270. or vector map.
  271. """
  272. layers = []
  273. if self.rasters['first']:
  274. layers += self.firstMap.GetListOfLayers()
  275. if self.rasters['second']:
  276. layers += self.secondMap.GetListOfLayers()
  277. if layers:
  278. self.GetFirstWindow().ZoomToMap(layers=layers)
  279. self.GetSecondWindow().ZoomToMap(layers=layers)
  280. def OnZoomToMap(self, event):
  281. """Zoom to map"""
  282. self.ZoomToMap()
  283. def OnZoomBack(self, event):
  284. self.GetFirstWindow().ZoomBack()
  285. self.secondMap.region = self.firstMap.region
  286. self.Render(self.GetSecondWindow())
  287. def OnSelectLayers(self, event):
  288. if self._inputDialog is None:
  289. dlg = SwipeMapDialog(self, first=self.rasters['first'],
  290. second=self.rasters['second'],
  291. firstLayerList=None, secondLayerList=None)
  292. dlg.applyChanges.connect(self.OnApplyInputChanges)
  293. # connect to convertor object to convert to Map
  294. # store reference to convertor is needed otherwise it would be
  295. # discarded
  296. self._firstConverter = self._connectSimpleLmgr(
  297. dlg.GetFirstSimpleLmgr(), self.GetFirstMap())
  298. self._secondConverter = self._connectSimpleLmgr(
  299. dlg.GetSecondSimpleLmgr(), self.GetSecondMap())
  300. self._inputDialog = dlg
  301. dlg.CentreOnParent()
  302. dlg.Show()
  303. else:
  304. if self._inputDialog.IsShown():
  305. self._inputDialog.Raise()
  306. self._inputDialog.SetFocus()
  307. else:
  308. self._inputDialog.Show()
  309. def _connectSimpleLmgr(self, lmgr, renderer):
  310. converter = LayerListToRendererConverter(renderer)
  311. lmgr.opacityChanged.connect(converter.ChangeLayerOpacity)
  312. lmgr.cmdChanged.connect(converter.ChangeLayerCmd)
  313. lmgr.layerAdded.connect(converter.AddLayer)
  314. lmgr.layerRemoved.connect(converter.RemoveLayer)
  315. lmgr.layerActivated.connect(converter.ChangeLayerActive)
  316. lmgr.layerMovedUp.connect(converter.MoveLayerUp)
  317. lmgr.layerMovedDown.connect(converter.MoveLayerDown)
  318. lmgr.anyChange.connect(self._simpleLmgrChanged)
  319. return converter
  320. def _simpleLmgrChanged(self):
  321. if self.IsAutoRendered():
  322. self.OnRender(event=None)
  323. def OnApplyInputChanges(self):
  324. first, second = self._inputDialog.GetValues()
  325. if self._inputDialog.IsSimpleMode():
  326. self.rasters['first'], self.rasters['second'] = first, second
  327. res1 = self.SetFirstRaster(name=self.rasters['first'])
  328. res2 = self.SetSecondRaster(name=self.rasters['second'])
  329. if not (res1 and res2) and (first or second):
  330. message = ''
  331. if first and not res1:
  332. message += _("Map <%s> not found. ") % self.rasters[
  333. 'first']
  334. if second and not res2:
  335. message += _("Map <%s> not found.") % self.rasters[
  336. 'second']
  337. if message:
  338. GError(parent=self, message=message)
  339. return
  340. self.ZoomToMap()
  341. else:
  342. LayerListToRendererConverter(self.GetFirstMap()).ConvertAll(first)
  343. LayerListToRendererConverter(
  344. self.GetSecondMap()).ConvertAll(second)
  345. self.SetRasterNames()
  346. if self.IsAutoRendered():
  347. self.OnRender(event=None)
  348. def SetFirstRaster(self, name):
  349. """Set raster map to first Map"""
  350. if name:
  351. raster = grass.find_file(name=name, element='cell')
  352. if raster.get('fullname'):
  353. self.rasters['first'] = raster['fullname']
  354. self.SetLayer(
  355. name=raster['fullname'],
  356. mapInstance=self.GetFirstMap())
  357. return True
  358. return False
  359. def SetSecondRaster(self, name):
  360. """Set raster map to second Map"""
  361. if name:
  362. raster = grass.find_file(name=name, element='cell')
  363. if raster.get('fullname'):
  364. self.rasters['second'] = raster['fullname']
  365. self.SetLayer(
  366. name=raster['fullname'],
  367. mapInstance=self.GetSecondMap())
  368. return True
  369. return False
  370. def SetLayer(self, name, mapInstance):
  371. """Sets layer in Map.
  372. :param name: layer (raster) name
  373. """
  374. Debug.msg(3, "SwipeMapFrame.SetLayer(): name=%s" % name)
  375. # this simple application enables to keep only one raster
  376. mapInstance.DeleteAllLayers()
  377. cmdlist = ['d.rast', 'map=%s' % name]
  378. # add layer to Map instance (core.render)
  379. newLayer = mapInstance.AddLayer(
  380. ltype='raster',
  381. command=cmdlist,
  382. active=True,
  383. name=name,
  384. hidden=False,
  385. opacity=1.0,
  386. render=True)
  387. def OnSwitchWindows(self, event):
  388. """Switch windows position."""
  389. Debug.msg(3, "SwipeMapFrame.OnSwitchWindows()")
  390. splitter = self.splitter
  391. w1, w2 = splitter.GetWindow1(), splitter.GetWindow2()
  392. splitter.ReplaceWindow(w1, w2)
  393. splitter.ReplaceWindow(w2, w1)
  394. # self.OnSize(None)
  395. splitter.OnSashChanged(None)
  396. def _saveToFile(self, fileName, fileType):
  397. """Creates composite image by rendering both images and
  398. pasting them into the new one.
  399. .. todo::
  400. specify size of the new image (problem is inaccurate scaling)
  401. .. todo::
  402. make dividing line width and color optional
  403. """
  404. w1 = self.splitter.GetWindow1()
  405. w2 = self.splitter.GetWindow2()
  406. lineWidth = 1
  407. # render to temporary files
  408. filename1 = grass.tempfile(False) + '1'
  409. filename2 = grass.tempfile(False) + '2'
  410. width, height = self.splitter.GetClientSize()
  411. if self._mode == 'swipe':
  412. x, y = w2.GetImageCoords()
  413. w1.SaveToFile(filename1, fileType, width, height)
  414. w2.SaveToFile(filename2, fileType, width, height)
  415. else:
  416. fw, fh = w1.GetClientSize()
  417. w1.SaveToFile(filename1, fileType, fw, fh)
  418. sw, sh = w2.GetClientSize()
  419. w2.SaveToFile(filename2, fileType, sw, sh)
  420. # create empty white image - needed for line
  421. im = wx.EmptyImage(width, height)
  422. im.Replace(0, 0, 0, 255, 255, 255)
  423. # paste images
  424. if self._mode == 'swipe':
  425. if self.splitter.GetSplitMode() == wx.SPLIT_HORIZONTAL:
  426. im1 = wx.Image(filename1).GetSubImage((0, 0, width, -y))
  427. im.Paste(im1, 0, 0)
  428. im.Paste(wx.Image(filename2), -x, -y + lineWidth)
  429. else:
  430. im1 = wx.Image(filename1).GetSubImage((0, 0, -x, height))
  431. im.Paste(im1, 0, 0)
  432. im.Paste(wx.Image(filename2), -x + lineWidth, -y)
  433. else:
  434. if self.splitter.GetSplitMode() == wx.SPLIT_HORIZONTAL:
  435. im1 = wx.Image(filename1)
  436. im.Paste(im1, 0, 0)
  437. im.Paste(wx.Image(filename2), 0, fh + lineWidth)
  438. else:
  439. im1 = wx.Image(filename1)
  440. im.Paste(im1, 0, 0)
  441. im.Paste(wx.Image(filename2), fw + lineWidth, 0)
  442. im.SaveFile(fileName, fileType)
  443. # remove temporary files
  444. grass.try_remove(filename1)
  445. grass.try_remove(filename2)
  446. def SaveToFile(self, event):
  447. """Save map to image
  448. """
  449. img = self.firstMapWindow.img or self.secondMapWindow.img
  450. if not img:
  451. GMessage(parent=self, message=_(
  452. "Nothing to render (empty map). Operation canceled."))
  453. return
  454. filetype, ltype = GetImageHandlers(img)
  455. # get filename
  456. dlg = wx.FileDialog(parent=self,
  457. message=_("Choose a file name to save the image "
  458. "(no need to add extension)"),
  459. wildcard=filetype,
  460. style=wx.FD_SAVE | wx.FD_OVERWRITE_PROMPT)
  461. if dlg.ShowModal() == wx.ID_OK:
  462. path = dlg.GetPath()
  463. if not path:
  464. dlg.Destroy()
  465. return
  466. base, ext = os.path.splitext(path)
  467. fileType = ltype[dlg.GetFilterIndex()]['type']
  468. extType = ltype[dlg.GetFilterIndex()]['ext']
  469. if ext != extType:
  470. path = base + '.' + extType
  471. self._saveToFile(path, fileType)
  472. dlg.Destroy()
  473. def OnSwitchOrientation(self, event):
  474. """Switch orientation of the sash."""
  475. Debug.msg(3, "SwipeMapFrame.OnSwitchOrientation()")
  476. splitter = self.splitter
  477. splitter.Unsplit()
  478. if splitter.GetSplitMode() == wx.SPLIT_HORIZONTAL:
  479. splitter.SplitVertically(
  480. self.firstMapWindow, self.secondMapWindow, 0)
  481. self.slider = self.sliderH
  482. if self._mode == 'swipe':
  483. self._mgr.GetPane('sliderH').Show()
  484. self._mgr.GetPane('sliderV').Hide()
  485. else:
  486. splitter.SplitHorizontally(
  487. self.firstMapWindow, self.secondMapWindow, 0)
  488. self.slider = self.sliderV
  489. if self._mode == 'swipe':
  490. self._mgr.GetPane('sliderV').Show()
  491. self._mgr.GetPane('sliderH').Hide()
  492. self._mgr.Update()
  493. splitter.OnSashChanged(None)
  494. self.OnSize(None)
  495. self.SetRasterNames()
  496. def OnAddText(self, event):
  497. """Double click on text overlay
  498. So far not implemented.
  499. """
  500. pass
  501. def SetViewMode(self, mode):
  502. """Sets view mode.
  503. :param mode: view mode ('swipe', 'mirror')
  504. """
  505. if self._mode == mode:
  506. return
  507. self._mode = mode
  508. self.toolbars['swipeMain'].SetMode(mode)
  509. # set window mode
  510. self.GetFirstWindow().SetMode(mode)
  511. self.GetSecondWindow().SetMode(mode)
  512. # hide/show slider
  513. if self.splitter.GetSplitMode() == wx.SPLIT_HORIZONTAL:
  514. self._mgr.GetPane("sliderV").Show(mode == "swipe")
  515. size = self.splitter.GetSize()[1] // 2
  516. else:
  517. self._mgr.GetPane("sliderH").Show(mode == "swipe")
  518. size = self.splitter.GetSize()[0] // 2
  519. # set sash in the middle
  520. self.splitter.SetSashPosition(size)
  521. self.slider.SetValue(size)
  522. self._mgr.Update()
  523. # enable / disable sash
  524. self.splitter.EnableSash(mode == 'swipe')
  525. # hack to make it work
  526. self.splitter.OnSashChanged(None)
  527. self.SendSizeEvent()
  528. def SetRasterNames(self):
  529. if not self._inputDialog or self._inputDialog.IsSimpleMode():
  530. if self.rasters['first']:
  531. self.GetFirstWindow().SetRasterNameText(
  532. self.rasters['first'], 101)
  533. if self.rasters['second']:
  534. self.GetSecondWindow().SetRasterNameText(
  535. self.rasters['second'], 102)
  536. else:
  537. self.GetFirstWindow().SetRasterNameText('', 101)
  538. self.GetSecondWindow().SetRasterNameText('', 102)
  539. def Query(self, x, y):
  540. """Query active layers from both mapwindows.
  541. :param x,y: coordinates
  542. """
  543. rasters = (
  544. [layer.GetName()
  545. for layer in self.GetFirstMap().GetListOfLayers(
  546. ltype='raster', active=True)],
  547. [layer.GetName()
  548. for layer in self.GetSecondMap().GetListOfLayers(
  549. ltype='raster', active=True)])
  550. vectors = (
  551. [layer.GetName()
  552. for layer in self.GetFirstMap().GetListOfLayers(
  553. ltype='vector', active=True)],
  554. [layer.GetName()
  555. for layer in self.GetSecondMap().GetListOfLayers(
  556. ltype='vector', active=True)])
  557. if not (rasters[0] + rasters[1] + vectors[0] + vectors[1]):
  558. GMessage(
  559. parent=self,
  560. message=_(
  561. 'No raster or vector map layer selected for querying.'))
  562. return
  563. # set query snap distance for v.what at map unit equivalent of 10
  564. # pixels
  565. qdist = 10.0 * (
  566. (self.GetFirstMap().region['e'] - self.GetFirstMap().region['w']) /
  567. self.GetFirstMap().width)
  568. east, north = self.GetFirstWindow().Pixel2Cell((x, y))
  569. # use display region settings instead of computation region settings
  570. self.tmpreg = os.getenv("GRASS_REGION")
  571. os.environ["GRASS_REGION"] = self.GetFirstMap(
  572. ).SetRegion(windres=False)
  573. result = []
  574. if rasters[0]:
  575. result.extend(
  576. grass.raster_what(
  577. map=rasters[0],
  578. coord=(east, north),
  579. localized=True))
  580. if vectors[0]:
  581. result.extend(
  582. grass.vector_what(
  583. map=vectors[0],
  584. coord=(east, north),
  585. distance=qdist))
  586. if rasters[1]:
  587. result.extend(
  588. grass.raster_what(
  589. map=rasters[1],
  590. coord=(east, north),
  591. localized=True))
  592. if vectors[1]:
  593. result.extend(
  594. grass.vector_what(
  595. map=vectors[1],
  596. coord=(east, north),
  597. distance=qdist))
  598. self._QueryMapDone()
  599. result = PrepareQueryResults(coordinates=(east, north), result=result)
  600. if self._queryDialog:
  601. self._queryDialog.Raise()
  602. self._queryDialog.SetData(result)
  603. else:
  604. self._queryDialog = QueryDialog(parent=self, data=result)
  605. self._queryDialog.Bind(wx.EVT_CLOSE, self._oncloseQueryDialog)
  606. self._queryDialog.redirectOutput.connect(
  607. lambda output: self._giface.WriteLog(output))
  608. self._queryDialog.Show()
  609. def _oncloseQueryDialog(self, event):
  610. self._queryDialog = None
  611. event.Skip()
  612. def _QueryMapDone(self):
  613. """Restore settings after querying (restore GRASS_REGION)
  614. """
  615. if hasattr(self, "tmpreg"):
  616. if self.tmpreg:
  617. os.environ["GRASS_REGION"] = self.tmpreg
  618. elif 'GRASS_REGION' in os.environ:
  619. del os.environ["GRASS_REGION"]
  620. elif 'GRASS_REGION' in os.environ:
  621. del os.environ["GRASS_REGION"]
  622. if hasattr(self, "tmpreg"):
  623. del self.tmpreg
  624. def GetMapToolbar(self):
  625. """Returns toolbar with zooming tools"""
  626. return self.toolbars['swipeMap']
  627. def IsStandalone(self):
  628. """Since we do not need layer manager, we are standalone"""
  629. return True
  630. def OnHelp(self, event):
  631. self._giface.Help(entry='wxGUI.mapswipe')
  632. def OnPreferences(self, event):
  633. if not self._preferencesDialog:
  634. dlg = PreferencesDialog(parent=self, giface=self._giface)
  635. self._preferencesDialog = dlg
  636. self._preferencesDialog.CenterOnParent()
  637. self._preferencesDialog.Show()
  638. def OnCloseWindow(self, event):
  639. self.GetFirstMap().Clean()
  640. self.GetSecondMap().Clean()
  641. self._mgr.UnInit()
  642. if self._inputDialog:
  643. self._inputDialog.UnInit()
  644. self.Destroy()
  645. class MapSplitter(wx.SplitterWindow):
  646. """Splitter window for displaying two maps"""
  647. def __init__(self, parent, id):
  648. wx.SplitterWindow.__init__(self, parent=parent, id=id,
  649. style=wx.SP_LIVE_UPDATE
  650. )
  651. Debug.msg(2, "MapSplitter.__init__()")
  652. self.sashWidthMin = 1
  653. self.sashWidthMax = 10
  654. self.Bind(wx.EVT_SPLITTER_SASH_POS_CHANGED, self.OnSashChanged)
  655. self.Bind(wx.EVT_SPLITTER_SASH_POS_CHANGING, self.OnSashChanging)
  656. self._moveSash = True
  657. def EnableSash(self, enable):
  658. self._moveSash = enable
  659. def Init(self):
  660. self.OnSashChanged(evt=None)
  661. self.SetMinimumPaneSize(0)
  662. # def OnMotion(self, event):
  663. # w = self.GetSashSize()
  664. # w1, w2 = self.GetWindow1(), self.GetWindow2()
  665. # if self.SashHitTest(event.GetX(), event.GetY(), tolerance = 20):
  666. # if w == self.sashWidthMin:
  667. # self.SetSashSize(self.sashWidthMax)
  668. # self.SetNeedUpdating(True)
  669. # w1.movingSash = True
  670. # w2.movingSash = True
  671. # else:
  672. # w1.movingSash = False
  673. # w1.movingSash = False
  674. # else:
  675. # if w == self.sashWidthMax:
  676. # self.SetSashSize(self.sashWidthMin)
  677. # self.SetNeedUpdating(True)
  678. # w1.movingSash = True
  679. # w2.movingSash = True
  680. # else:
  681. # w1.movingSash = False
  682. # w2.movingSash = False
  683. # event.Skip()
  684. def OnSashChanged(self, evt):
  685. Debug.msg(5, "MapSplitter.OnSashChanged()")
  686. if not self._moveSash:
  687. return
  688. w1, w2 = self.GetWindow1(), self.GetWindow2()
  689. w1.movingSash = False
  690. w2.movingSash = False
  691. wx.CallAfter(self.SashChanged)
  692. def SashChanged(self):
  693. Debug.msg(5, "MapSplitter.SashChanged()")
  694. w1, w2 = self.GetWindow1(), self.GetWindow2()
  695. w1.SetImageCoords((0, 0))
  696. if self.GetSplitMode() == wx.SPLIT_VERTICAL:
  697. w = w1.GetSize()[0]
  698. w2.SetImageCoords((-w, 0))
  699. else:
  700. h = w1.GetSize()[1]
  701. w2.SetImageCoords((0, -h))
  702. w1.UpdateMap(render=False, renderVector=False)
  703. w2.UpdateMap(render=False, renderVector=False)
  704. pos = self.GetSashPosition()
  705. self.last = pos
  706. def OnSashChanging(self, event):
  707. Debug.msg(5, "MapSplitter.OnSashChanging()")
  708. if not self._moveSash:
  709. event.SetSashPosition(-1)
  710. return
  711. if not (self.GetWindowStyle() & wx.SP_LIVE_UPDATE):
  712. if event:
  713. event.Skip()
  714. return
  715. pos = self.GetSashPosition()
  716. dpos = pos - self.last
  717. self.last = pos
  718. if self.GetSplitMode() == wx.SPLIT_VERTICAL:
  719. dx = -dpos
  720. dy = 0
  721. else:
  722. dx = 0
  723. dy = -dpos
  724. self.GetWindow2().TranslateImage(dx, dy)
  725. self.GetWindow1().movingSash = True
  726. self.GetWindow2().movingSash = True