frame.py 32 KB

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