tree.py 86 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829183018311832183318341835183618371838183918401841184218431844184518461847184818491850185118521853185418551856185718581859186018611862186318641865186618671868186918701871187218731874187518761877187818791880188118821883188418851886188718881889189018911892189318941895189618971898189919001901190219031904190519061907190819091910191119121913191419151916191719181919192019211922192319241925192619271928192919301931193219331934193519361937193819391940194119421943194419451946194719481949195019511952195319541955195619571958195919601961196219631964196519661967196819691970197119721973197419751976197719781979198019811982198319841985198619871988198919901991199219931994199519961997199819992000200120022003200420052006200720082009201020112012201320142015201620172018201920202021202220232024202520262027202820292030203120322033203420352036203720382039204020412042204320442045204620472048204920502051205220532054205520562057205820592060206120622063206420652066206720682069207020712072207320742075207620772078207920802081208220832084208520862087208820892090209120922093209420952096209720982099210021012102210321042105210621072108210921102111211221132114211521162117211821192120212121222123212421252126212721282129213021312132213321342135213621372138213921402141214221432144214521462147214821492150215121522153215421552156215721582159216021612162216321642165216621672168216921702171217221732174217521762177217821792180218121822183218421852186218721882189219021912192219321942195219621972198219922002201220222032204220522062207220822092210221122122213221422152216221722182219222022212222222322242225222622272228222922302231223222332234223522362237223822392240224122422243224422452246224722482249225022512252225322542255225622572258225922602261226222632264226522662267226822692270227122722273227422752276227722782279228022812282228322842285228622872288228922902291229222932294229522962297229822992300230123022303230423052306230723082309231023112312231323142315231623172318231923202321232223232324232523262327232823292330
  1. """
  2. @package datacatalog::tree
  3. @brief Data catalog tree classes
  4. Classes:
  5. - datacatalog::NameEntryDialog
  6. - datacatalog::DataCatalogNode
  7. - datacatalog::DataCatalogTree
  8. (C) 2014-2018 by Tereza Fiedlerova, and the GRASS Development Team
  9. This program is free software under the GNU General Public
  10. License (>=v2). Read the file COPYING that comes with GRASS
  11. for details.
  12. @author Tereza Fiedlerova
  13. @author Anna Petrasova (kratochanna gmail com)
  14. @author Linda Kladivova (l.kladivova@seznam.cz)
  15. """
  16. import os
  17. import re
  18. import copy
  19. from multiprocessing import Process, Queue, cpu_count
  20. watchdog_used = True
  21. try:
  22. from watchdog.observers import Observer
  23. from watchdog.events import PatternMatchingEventHandler
  24. except ImportError:
  25. watchdog_used = False
  26. PatternMatchingEventHandler = object
  27. import wx
  28. from wx.lib.newevent import NewEvent
  29. from core.gcmd import RunCommand, GError, GMessage
  30. from core.utils import GetListOfLocations
  31. from core.debug import Debug
  32. from core.gthread import gThread
  33. from gui_core.dialogs import TextEntryDialog
  34. from core.giface import StandaloneGrassInterface
  35. from core.treemodel import TreeModel, DictNode
  36. from gui_core.treeview import TreeView
  37. from gui_core.wrap import Menu
  38. from datacatalog.dialogs import CatalogReprojectionDialog
  39. from icons.icon import MetaIcon
  40. from core.settings import UserSettings
  41. from startup.guiutils import (
  42. create_mapset_interactively,
  43. create_location_interactively,
  44. rename_mapset_interactively,
  45. rename_location_interactively,
  46. delete_mapsets_interactively,
  47. delete_locations_interactively,
  48. download_location_interactively,
  49. delete_grassdb_interactively,
  50. can_switch_mapset_interactive,
  51. switch_mapset_interactively,
  52. get_reason_mapset_not_removable,
  53. get_reasons_location_not_removable,
  54. get_mapset_name_invalid_reason,
  55. get_location_name_invalid_reason,
  56. )
  57. from grass.grassdb.manage import rename_mapset, rename_location
  58. from grass.pydispatch.signal import Signal
  59. import grass.script as gscript
  60. from grass.script import gisenv
  61. from grass.grassdb.data import map_exists
  62. from grass.grassdb.checks import (
  63. get_mapset_owner,
  64. is_mapset_locked,
  65. is_different_mapset_owner,
  66. is_first_time_user,
  67. )
  68. from grass.exceptions import CalledModuleError
  69. updateMapset, EVT_UPDATE_MAPSET = NewEvent()
  70. def getLocationTree(gisdbase, location, queue, mapsets=None, lazy=False):
  71. """Creates dictionary with mapsets, elements, layers for given location.
  72. Returns tuple with the dictionary and error (or None)"""
  73. tmp_gisrc_file, env = gscript.create_environment(gisdbase, location, "PERMANENT")
  74. env["GRASS_SKIP_MAPSET_OWNER_CHECK"] = "1"
  75. maps_dict = {}
  76. elements = ["raster", "raster_3d", "vector"]
  77. try:
  78. if not mapsets:
  79. mapsets = gscript.read_command(
  80. "g.mapsets", flags="l", separator="comma", quiet=True, env=env
  81. ).strip()
  82. except CalledModuleError:
  83. queue.put(
  84. (
  85. maps_dict,
  86. _("Failed to read mapsets from location <{l}>.").format(l=location),
  87. )
  88. )
  89. gscript.try_remove(tmp_gisrc_file)
  90. return
  91. else:
  92. mapsets = mapsets.split(",")
  93. Debug.msg(4, "Location <{0}>: {1} mapsets found".format(location, len(mapsets)))
  94. for each in mapsets:
  95. maps_dict[each] = []
  96. if lazy:
  97. queue.put((maps_dict, None))
  98. return
  99. try:
  100. maplist = gscript.read_command(
  101. "g.list",
  102. flags="mt",
  103. type=elements,
  104. mapset=",".join(mapsets),
  105. quiet=True,
  106. env=env,
  107. ).strip()
  108. except CalledModuleError:
  109. queue.put(
  110. (
  111. maps_dict,
  112. _("Failed to read maps from location <{l}>.").format(l=location),
  113. )
  114. )
  115. gscript.try_remove(tmp_gisrc_file)
  116. return
  117. else:
  118. # fill dictionary
  119. listOfMaps = maplist.splitlines()
  120. Debug.msg(4, "Location <{0}>: {1} maps found".format(location, len(listOfMaps)))
  121. for each in listOfMaps:
  122. ltype, wholename = each.split("/")
  123. name, mapset = wholename.split("@", maxsplit=1)
  124. maps_dict[mapset].append({"name": name, "type": ltype})
  125. queue.put((maps_dict, None))
  126. gscript.try_remove(tmp_gisrc_file)
  127. class MapWatch(PatternMatchingEventHandler):
  128. """Monitors file events (create, delete, move files) using watchdog
  129. to inform about changes in current mapset. One instance monitors
  130. only one element (raster, vector, raster_3d).
  131. Patterns are not used/needed in this case, use just '*' for matching
  132. everything. When file/directory change is detected, wx event is dispatched
  133. to event handler (can't use Signals because this is different thread),
  134. containing info about the change."""
  135. def __init__(self, patterns, element, event_handler):
  136. PatternMatchingEventHandler.__init__(self, patterns=patterns)
  137. self.element = element
  138. self.event_handler = event_handler
  139. def on_created(self, event):
  140. if (
  141. self.element == "vector" or self.element == "raster_3d"
  142. ) and not event.is_directory:
  143. return
  144. evt = updateMapset(
  145. src_path=event.src_path,
  146. event_type=event.event_type,
  147. is_directory=event.is_directory,
  148. dest_path=None,
  149. )
  150. wx.PostEvent(self.event_handler, evt)
  151. def on_deleted(self, event):
  152. if (
  153. self.element == "vector" or self.element == "raster_3d"
  154. ) and not event.is_directory:
  155. return
  156. evt = updateMapset(
  157. src_path=event.src_path,
  158. event_type=event.event_type,
  159. is_directory=event.is_directory,
  160. dest_path=None,
  161. )
  162. wx.PostEvent(self.event_handler, evt)
  163. def on_moved(self, event):
  164. if (
  165. self.element == "vector" or self.element == "raster_3d"
  166. ) and not event.is_directory:
  167. return
  168. evt = updateMapset(
  169. src_path=event.src_path,
  170. event_type=event.event_type,
  171. is_directory=event.is_directory,
  172. dest_path=event.dest_path,
  173. )
  174. wx.PostEvent(self.event_handler, evt)
  175. class NameEntryDialog(TextEntryDialog):
  176. def __init__(self, element, mapset, env, **kwargs):
  177. TextEntryDialog.__init__(self, **kwargs)
  178. self._element = element
  179. self._mapset = mapset
  180. self._env = env
  181. id_OK = self.GetAffirmativeId()
  182. self.Bind(wx.EVT_BUTTON, self.OnOK, self.FindWindowById(id_OK))
  183. def OnOK(self, event):
  184. new = self.GetValue()
  185. if not new:
  186. return
  187. if map_exists(new, self._element, env=self._env, mapset=self._mapset):
  188. dlg = wx.MessageDialog(
  189. self,
  190. message=_(
  191. "Map of type {elem} <{name}> already exists in mapset <{mapset}>. "
  192. "Do you want to overwrite it?"
  193. ).format(elem=self._element, name=new, mapset=self._mapset),
  194. caption=_("Overwrite?"),
  195. style=wx.YES_NO,
  196. )
  197. if dlg.ShowModal() == wx.ID_YES:
  198. dlg.Destroy()
  199. self._env["GRASS_OVERWRITE"] = "1"
  200. self.EndModal(wx.ID_OK)
  201. else:
  202. dlg.Destroy()
  203. return
  204. else:
  205. self.EndModal(wx.ID_OK)
  206. class DataCatalogNode(DictNode):
  207. """Node representing item in datacatalog."""
  208. def __init__(self, data=None):
  209. super(DataCatalogNode, self).__init__(data=data)
  210. @property
  211. def label(self):
  212. data = self.data
  213. if data["type"] == "mapset":
  214. owner = data["owner"] if data["owner"] else _("name unknown")
  215. if data["current"]:
  216. return _("{name} (current)").format(**data)
  217. elif data["is_different_owner"] and data["lock"]:
  218. return _("{name} (in use, owner: {owner})").format(
  219. name=data["name"], owner=owner
  220. )
  221. elif data["lock"]:
  222. return _("{name} (in use)").format(**data)
  223. elif data["is_different_owner"]:
  224. return _("{name} (owner: {owner})").format(
  225. name=data["name"], owner=owner
  226. )
  227. return _("{name}").format(**data)
  228. def match(self, method="exact", **kwargs):
  229. """Method used for searching according to given parameters.
  230. :param method: 'exact' for exact match or 'filtering' for filtering by type/name
  231. :param kwargs key-value to be matched, filtering method uses 'type' and 'name'
  232. where 'name' is compiled regex
  233. """
  234. if not kwargs:
  235. return False
  236. if method == "exact":
  237. for key, value in kwargs.items():
  238. if not (key in self.data and self.data[key] == value):
  239. return False
  240. return True
  241. # for filtering
  242. if (
  243. "type" in kwargs
  244. and "type" in self.data
  245. and kwargs["type"] != self.data["type"]
  246. ):
  247. return False
  248. if (
  249. "name" in kwargs
  250. and "name" in self.data
  251. and not kwargs["name"].search(self.data["name"])
  252. ):
  253. return False
  254. return True
  255. class DataCatalogTree(TreeView):
  256. """Tree structure visualizing and managing grass database.
  257. Uses virtual tree and model defined in core/treemodel.py.
  258. When changes to data are initiated from inside, the model
  259. and the tree are not changed directly, rather a grassdbChanged
  260. signal needs to be emitted and the handler of the signal
  261. takes care of the refresh. At the same time, watchdog (if installed)
  262. monitors changes in current mapset and refreshes the tree.
  263. """
  264. def __init__(
  265. self,
  266. parent,
  267. model=None,
  268. giface=None,
  269. style=wx.TR_HIDE_ROOT
  270. | wx.TR_EDIT_LABELS
  271. | wx.TR_LINES_AT_ROOT
  272. | wx.TR_HAS_BUTTONS
  273. | wx.TR_FULL_ROW_HIGHLIGHT
  274. | wx.TR_MULTIPLE,
  275. ):
  276. """Location Map Tree constructor."""
  277. self._model = TreeModel(DataCatalogNode)
  278. self._orig_model = self._model
  279. super(DataCatalogTree, self).__init__(
  280. parent=parent, model=self._model, id=wx.ID_ANY, style=style
  281. )
  282. self._giface = giface
  283. self._restricted = True
  284. self.showNotification = Signal("Tree.showNotification")
  285. self.showImportDataInfo = Signal("Tree.showImportDataInfo")
  286. self.loadingDone = Signal("Tree.loadingDone")
  287. self.parent = parent
  288. self.contextMenu.connect(self.OnRightClick)
  289. self.itemActivated.connect(self.OnDoubleClick)
  290. self._giface.currentMapsetChanged.connect(self._updateAfterMapsetChanged)
  291. self._giface.grassdbChanged.connect(self._updateAfterGrassdbChanged)
  292. self._iconTypes = [
  293. "grassdb",
  294. "location",
  295. "mapset",
  296. "raster",
  297. "vector",
  298. "raster_3d",
  299. ]
  300. self._initImages()
  301. self.thread = gThread()
  302. self._resetSelectVariables()
  303. self._resetCopyVariables()
  304. self.current_grassdb_node = None
  305. self.current_location_node = None
  306. self.current_mapset_node = None
  307. self.UpdateCurrentDbLocationMapsetNode()
  308. self._lastWatchdogUpdate = gscript.clock()
  309. self._updateMapsetWhenIdle = None
  310. # Get databases from settings
  311. # add current to settings if it's not included
  312. self.grassdatabases = self._getValidSavedGrassDBs()
  313. currentDB = gisenv()["GISDBASE"]
  314. if currentDB not in self.grassdatabases:
  315. self.grassdatabases.append(currentDB)
  316. self._saveGrassDBs()
  317. self.beginDrag = Signal("DataCatalogTree.beginDrag")
  318. self.endDrag = Signal("DataCatalogTree.endDrag")
  319. self.startEdit = Signal("DataCatalogTree.startEdit")
  320. self.endEdit = Signal("DataCatalogTree.endEdit")
  321. self.Bind(
  322. wx.EVT_TREE_BEGIN_DRAG,
  323. lambda evt: self._emitSignal(evt.GetItem(), self.beginDrag, event=evt),
  324. )
  325. self.Bind(
  326. wx.EVT_TREE_END_DRAG,
  327. lambda evt: self._emitSignal(evt.GetItem(), self.endDrag, event=evt),
  328. )
  329. self.beginDrag.connect(self.OnBeginDrag)
  330. self.endDrag.connect(self.OnEndDrag)
  331. self.Bind(
  332. wx.EVT_TREE_BEGIN_LABEL_EDIT,
  333. lambda evt: self._emitSignal(evt.GetItem(), self.startEdit, event=evt),
  334. )
  335. self.Bind(
  336. wx.EVT_TREE_END_LABEL_EDIT,
  337. lambda evt: self._emitSignal(evt.GetItem(), self.endEdit, event=evt),
  338. )
  339. self.startEdit.connect(self.OnStartEditLabel)
  340. self.endEdit.connect(self.OnEditLabel)
  341. self.Bind(
  342. EVT_UPDATE_MAPSET, lambda evt: self._onWatchdogMapsetReload(evt.src_path)
  343. )
  344. self.Bind(wx.EVT_IDLE, self._onUpdateMapsetWhenIdle)
  345. self.observer = None
  346. def _resetSelectVariables(self):
  347. """Reset variables related to item selection."""
  348. self.selected_grassdb = []
  349. self.selected_layer = []
  350. self.selected_mapset = []
  351. self.selected_location = []
  352. self.mixed = False
  353. def _resetCopyVariables(self):
  354. """Reset copy related variables."""
  355. self.copy_mode = False
  356. self.copy_layer = None
  357. self.copy_mapset = None
  358. self.copy_location = None
  359. self.copy_grassdb = None
  360. def _useLazyLoading(self):
  361. settings = UserSettings.Get(group="datacatalog")
  362. # workaround defining new settings in datacatalog group
  363. # force writing new settings in the wx.json file during start
  364. # can be removed later on
  365. if "lazyLoading" not in settings:
  366. lazySettings = UserSettings.Get(
  367. group="datacatalog", key="lazyLoading", settings_type="default"
  368. )
  369. # update local settings
  370. for subkey, value in lazySettings.items():
  371. UserSettings.Append(
  372. UserSettings.userSettings,
  373. group="datacatalog",
  374. key="lazyLoading",
  375. subkey=subkey,
  376. value=value,
  377. overwrite=False,
  378. )
  379. # update settings file
  380. jsonSettings = {}
  381. UserSettings.ReadSettingsFile(settings=jsonSettings)
  382. jsonSettings["datacatalog"]["lazyLoading"] = lazySettings
  383. UserSettings.SaveToFile(jsonSettings)
  384. return UserSettings.Get(
  385. group="datacatalog", key="lazyLoading", subkey="enabled"
  386. )
  387. def _getValidSavedGrassDBs(self):
  388. """Returns list of GRASS databases from settings.
  389. Returns only existing directories."""
  390. dbs = UserSettings.Get(
  391. group="datacatalog", key="grassdbs", subkey="listAsString"
  392. )
  393. dbs = [db for db in dbs.split(",") if os.path.isdir(db)]
  394. return dbs
  395. def _saveGrassDBs(self):
  396. """Save current grass dbs in tree to settings"""
  397. UserSettings.Set(
  398. group="datacatalog",
  399. key="grassdbs",
  400. subkey="listAsString",
  401. value=",".join(self.grassdatabases),
  402. )
  403. grassdbSettings = {}
  404. UserSettings.ReadSettingsFile(settings=grassdbSettings)
  405. if "datacatalog" not in grassdbSettings:
  406. grassdbSettings["datacatalog"] = UserSettings.Get(group="datacatalog")
  407. # update only dbs
  408. grassdbSettings["datacatalog"]["grassdbs"] = UserSettings.Get(
  409. group="datacatalog", key="grassdbs"
  410. )
  411. UserSettings.SaveToFile(grassdbSettings)
  412. def _reloadMapsetNode(self, mapset_node):
  413. """Recursively reload the model of a specific mapset node"""
  414. if mapset_node.children:
  415. del mapset_node.children[:]
  416. q = Queue()
  417. p = Process(
  418. target=getLocationTree,
  419. args=(
  420. mapset_node.parent.parent.data["name"],
  421. mapset_node.parent.data["name"],
  422. q,
  423. mapset_node.data["name"],
  424. ),
  425. )
  426. p.start()
  427. maps, error = q.get()
  428. self._populateMapsetItem(mapset_node, maps[mapset_node.data["name"]])
  429. self._orig_model = copy.deepcopy(self._model)
  430. return error
  431. def _reloadLocationNode(self, location_node):
  432. """Recursively reload the model of a specific location node"""
  433. if location_node.children:
  434. del location_node.children[:]
  435. q = Queue()
  436. p = Process(
  437. target=getLocationTree,
  438. args=(
  439. location_node.parent.data["name"],
  440. location_node.data["name"],
  441. q,
  442. None,
  443. ),
  444. )
  445. p.start()
  446. maps, error = q.get()
  447. for mapset in maps:
  448. mapset_path = os.path.join(
  449. location_node.parent.data["name"], location_node.data["name"], mapset
  450. )
  451. mapset_node = self._model.AppendNode(
  452. parent=location_node,
  453. data=dict(
  454. type="mapset",
  455. name=mapset,
  456. current=False,
  457. lock=is_mapset_locked(mapset_path),
  458. is_different_owner=is_different_mapset_owner(mapset_path),
  459. owner=get_mapset_owner(mapset_path),
  460. ),
  461. )
  462. self._populateMapsetItem(mapset_node, maps[mapset])
  463. self._model.SortChildren(location_node)
  464. self._orig_model = copy.deepcopy(self._model)
  465. return error
  466. def _lazyReloadGrassDBNode(self, grassdb_node):
  467. genv = gisenv()
  468. if grassdb_node.children:
  469. del grassdb_node.children[:]
  470. all_location_nodes = []
  471. errors = []
  472. current_mapset_node = None
  473. locations = GetListOfLocations(grassdb_node.data["name"])
  474. for location in locations:
  475. loc_node = self._model.AppendNode(
  476. parent=grassdb_node, data=dict(type="location", name=location)
  477. )
  478. all_location_nodes.append(loc_node)
  479. q = Queue()
  480. getLocationTree(grassdb_node.data["name"], location, q, lazy=True)
  481. maps, error = q.get()
  482. if error:
  483. errors.append(error)
  484. for key in sorted(maps.keys()):
  485. mapset_path = os.path.join(
  486. loc_node.parent.data["name"], loc_node.data["name"], key
  487. )
  488. mapset_node = self._model.AppendNode(
  489. parent=loc_node,
  490. data=dict(
  491. type="mapset",
  492. name=key,
  493. lock=is_mapset_locked(mapset_path),
  494. current=False,
  495. is_different_owner=is_different_mapset_owner(mapset_path),
  496. owner=get_mapset_owner(mapset_path),
  497. ),
  498. )
  499. if (
  500. grassdb_node.data["name"] == genv["GISDBASE"]
  501. and location == genv["LOCATION_NAME"]
  502. and key == genv["MAPSET"]
  503. ):
  504. current_mapset_node = mapset_node
  505. if current_mapset_node:
  506. self._reloadMapsetNode(current_mapset_node)
  507. for node in all_location_nodes:
  508. self._model.SortChildren(node)
  509. self._model.SortChildren(grassdb_node)
  510. self._orig_model = copy.deepcopy(self._model)
  511. return errors
  512. def _reloadGrassDBNode(self, grassdb_node):
  513. """Recursively reload the model of a specific grassdb node.
  514. Runs reloading locations in parallel."""
  515. if grassdb_node.children:
  516. del grassdb_node.children[:]
  517. locations = GetListOfLocations(grassdb_node.data["name"])
  518. loc_count = proc_count = 0
  519. queue_list = []
  520. proc_list = []
  521. loc_list = []
  522. try:
  523. nprocs = max(1, cpu_count() - 1)
  524. except NotImplementedError:
  525. nprocs = 1
  526. results = dict()
  527. errors = []
  528. location_nodes = []
  529. all_location_nodes = []
  530. nlocations = len(locations)
  531. for location in locations:
  532. results[location] = dict()
  533. varloc = self._model.AppendNode(
  534. parent=grassdb_node, data=dict(type="location", name=location)
  535. )
  536. location_nodes.append(varloc)
  537. all_location_nodes.append(varloc)
  538. loc_count += 1
  539. Debug.msg(
  540. 3,
  541. "Scanning location <{0}> ({1}/{2})".format(
  542. location, loc_count, nlocations
  543. ),
  544. )
  545. q = Queue()
  546. p = Process(
  547. target=getLocationTree, args=(grassdb_node.data["name"], location, q)
  548. )
  549. p.start()
  550. queue_list.append(q)
  551. proc_list.append(p)
  552. loc_list.append(location)
  553. proc_count += 1
  554. # Wait for all running processes
  555. if proc_count == nprocs or loc_count == nlocations:
  556. Debug.msg(4, "Process subresults")
  557. for i in range(len(loc_list)):
  558. maps, error = queue_list[i].get()
  559. proc_list[i].join()
  560. if error:
  561. errors.append(error)
  562. for key in sorted(maps.keys()):
  563. mapset_path = os.path.join(
  564. location_nodes[i].parent.data["name"],
  565. location_nodes[i].data["name"],
  566. key,
  567. )
  568. mapset_node = self._model.AppendNode(
  569. parent=location_nodes[i],
  570. data=dict(
  571. type="mapset",
  572. name=key,
  573. lock=is_mapset_locked(mapset_path),
  574. current=False,
  575. is_different_owner=is_different_mapset_owner(
  576. mapset_path
  577. ),
  578. owner=get_mapset_owner(mapset_path),
  579. ),
  580. )
  581. self._populateMapsetItem(mapset_node, maps[key])
  582. proc_count = 0
  583. proc_list = []
  584. queue_list = []
  585. loc_list = []
  586. location_nodes = []
  587. for node in all_location_nodes:
  588. self._model.SortChildren(node)
  589. self._model.SortChildren(grassdb_node)
  590. self._orig_model = copy.deepcopy(self._model)
  591. return errors
  592. def _reloadTreeItems(self, full=False):
  593. """Updates grass databases, locations, mapsets and layers in the tree.
  594. It runs in thread, so it should not directly interact with GUI.
  595. In case of any errors it returns the errors as a list of strings, otherwise None.
  596. Option full=True forces full reload, full=False will behave based on user settings.
  597. """
  598. errors = []
  599. for grassdatabase in self.grassdatabases:
  600. grassdb_nodes = self._model.SearchNodes(name=grassdatabase, type="grassdb")
  601. if not grassdb_nodes:
  602. grassdb_node = self._model.AppendNode(
  603. parent=self._model.root,
  604. data=dict(type="grassdb", name=grassdatabase),
  605. )
  606. else:
  607. grassdb_node = grassdb_nodes[0]
  608. if full or not self._useLazyLoading():
  609. error = self._reloadGrassDBNode(grassdb_node)
  610. else:
  611. error = self._lazyReloadGrassDBNode(grassdb_node)
  612. if error:
  613. errors += error
  614. if errors:
  615. return errors
  616. return None
  617. def ScheduleWatchCurrentMapset(self):
  618. """Using watchdog library, sets up watching of current mapset folder
  619. to detect changes not captured by other means (e.g. from command line).
  620. Schedules 3 watches (raster, vector, 3D raster).
  621. If watchdog observers are active, it restarts the observers in current mapset.
  622. """
  623. global watchdog_used
  624. if not watchdog_used:
  625. return
  626. if self.observer and self.observer.is_alive():
  627. self.observer.stop()
  628. self.observer.join()
  629. self.observer.unschedule_all()
  630. self.observer = Observer()
  631. gisenv = gscript.gisenv()
  632. for element, directory in (
  633. ("raster", "cell"),
  634. ("vector", "vector"),
  635. ("raster_3d", "grid3"),
  636. ):
  637. path = os.path.join(
  638. gisenv["GISDBASE"], gisenv["LOCATION_NAME"], gisenv["MAPSET"], directory
  639. )
  640. if not os.path.exists(path):
  641. try:
  642. os.mkdir(path)
  643. except OSError:
  644. pass
  645. if os.path.exists(path):
  646. self.observer.schedule(
  647. MapWatch("*", element, self), path=path, recursive=False
  648. )
  649. try:
  650. self.observer.start()
  651. except OSError:
  652. # in case inotify on linux exceeds limits
  653. watchdog_used = False
  654. return
  655. def _onUpdateMapsetWhenIdle(self, event):
  656. """When idle, check if current mapset should be reloaded
  657. because there are skipped update events."""
  658. if self._updateMapsetWhenIdle:
  659. self._lastWatchdogUpdate = 0
  660. self._onWatchdogMapsetReload(self._updateMapsetWhenIdle)
  661. self._updateMapsetWhenIdle = None
  662. def _onWatchdogMapsetReload(self, event_path):
  663. """Reload mapset node associated with watchdog event.
  664. Check if events come to quickly and skip them."""
  665. time = gscript.clock()
  666. time_diff = time - self._lastWatchdogUpdate
  667. self._lastWatchdogUpdate = time
  668. if (time_diff) < 0.5:
  669. self._updateMapsetWhenIdle = event_path
  670. return
  671. mapset_path = os.path.dirname(os.path.dirname(os.path.abspath(event_path)))
  672. location_path = os.path.dirname(os.path.abspath(mapset_path))
  673. db = os.path.dirname(os.path.abspath(location_path))
  674. node = self.GetDbNode(
  675. grassdb=db,
  676. location=os.path.basename(location_path),
  677. mapset=os.path.basename(mapset_path),
  678. )
  679. if node:
  680. self._reloadMapsetNode(node)
  681. self.RefreshNode(node, recursive=True)
  682. def GetDbNode(self, grassdb, location=None, mapset=None, map=None, map_type=None):
  683. """Returns node representing db/location/mapset/map or None if not found."""
  684. grassdb_nodes = self._model.SearchNodes(name=grassdb, type="grassdb")
  685. if grassdb_nodes:
  686. if not location:
  687. return grassdb_nodes[0]
  688. location_nodes = self._model.SearchNodes(
  689. parent=grassdb_nodes[0], name=location, type="location"
  690. )
  691. if location_nodes:
  692. if not mapset:
  693. return location_nodes[0]
  694. mapset_nodes = self._model.SearchNodes(
  695. parent=location_nodes[0], name=mapset, type="mapset"
  696. )
  697. if mapset_nodes:
  698. if not map:
  699. return mapset_nodes[0]
  700. map_nodes = self._model.SearchNodes(
  701. parent=mapset_nodes[0], name=map, type=map_type
  702. )
  703. if map_nodes:
  704. return map_nodes[0]
  705. return None
  706. def _renameNode(self, node, name):
  707. """Rename node (map, mapset, location), sort and refresh.
  708. Should be called after actual renaming of a map, mapset, location."""
  709. node.data["name"] = name
  710. self._model.SortChildren(node.parent)
  711. self.RefreshNode(node.parent, recursive=True)
  712. def UpdateCurrentDbLocationMapsetNode(self):
  713. """Update variables storing current mapset/location/grassdb node.
  714. Updates associated mapset node data ('lock' and 'current').
  715. """
  716. def is_current_mapset_node_locked():
  717. mapset_path = os.path.join(
  718. self.current_grassdb_node.data["name"],
  719. self.current_location_node.data["name"],
  720. self.current_mapset_node.data["name"],
  721. )
  722. return is_mapset_locked(mapset_path)
  723. if self.current_mapset_node:
  724. self.current_mapset_node.data["current"] = False
  725. self.current_mapset_node.data["lock"] = is_current_mapset_node_locked()
  726. (
  727. self.current_grassdb_node,
  728. self.current_location_node,
  729. self.current_mapset_node,
  730. ) = self.GetCurrentDbLocationMapsetNode()
  731. if self.current_mapset_node:
  732. self.current_mapset_node.data["current"] = True
  733. self.current_mapset_node.data["lock"] = is_current_mapset_node_locked()
  734. def ReloadTreeItems(self, full=False):
  735. """Reload dbs, locations, mapsets and layers in the tree."""
  736. self.busy = wx.BusyCursor()
  737. if full or not self._useLazyLoading():
  738. self._quickLoading()
  739. self.thread.Run(
  740. callable=self._reloadTreeItems, full=full, ondone=self._loadItemsDone
  741. )
  742. def _quickLoading(self):
  743. """Quick loading of locations to show
  744. something when loading for the first time"""
  745. if self._model.root.children:
  746. return
  747. gisenv = gscript.gisenv()
  748. for grassdatabase in self.grassdatabases:
  749. grassdb_node = self._model.AppendNode(
  750. parent=self._model.root, data=dict(type="grassdb", name=grassdatabase)
  751. )
  752. for location in GetListOfLocations(grassdatabase):
  753. self._model.AppendNode(
  754. parent=grassdb_node, data=dict(type="location", name=location)
  755. )
  756. self.RefreshItems()
  757. if grassdatabase == gisenv["GISDBASE"]:
  758. self.ExpandNode(grassdb_node, recursive=False)
  759. def _loadItemsDone(self, event):
  760. Debug.msg(1, "Tree filled")
  761. del self.busy
  762. if event.ret is not None:
  763. self._giface.WriteWarning("\n".join(event.ret))
  764. self.UpdateCurrentDbLocationMapsetNode()
  765. self.ScheduleWatchCurrentMapset()
  766. self.RefreshItems()
  767. self.ExpandCurrentMapset()
  768. self.loadingDone.emit()
  769. def ReloadCurrentMapset(self):
  770. """Reload current mapset tree only."""
  771. self.UpdateCurrentDbLocationMapsetNode()
  772. if (
  773. not self.current_grassdb_node
  774. or not self.current_location_node
  775. or not self.current_mapset_node
  776. ):
  777. return
  778. self._reloadMapsetNode(self.current_mapset_node)
  779. self.RefreshNode(self.current_mapset_node, recursive=True)
  780. def _populateMapsetItem(self, mapset_node, data):
  781. for item in data:
  782. self._model.AppendNode(parent=mapset_node, data=dict(**item))
  783. self._model.SortChildren(mapset_node)
  784. def _initImages(self):
  785. bmpsize = (16, 16)
  786. icons = {
  787. "grassdb": MetaIcon(img="grassdb").GetBitmap(bmpsize),
  788. "location": MetaIcon(img="location").GetBitmap(bmpsize),
  789. "mapset": MetaIcon(img="mapset").GetBitmap(bmpsize),
  790. "raster": MetaIcon(img="raster").GetBitmap(bmpsize),
  791. "vector": MetaIcon(img="vector").GetBitmap(bmpsize),
  792. "raster_3d": MetaIcon(img="raster3d").GetBitmap(bmpsize),
  793. }
  794. il = wx.ImageList(bmpsize[0], bmpsize[1], mask=False)
  795. for each in self._iconTypes:
  796. il.Add(icons[each])
  797. self.AssignImageList(il)
  798. def GetControl(self):
  799. """Returns control itself."""
  800. return self
  801. def DefineItems(self, selected):
  802. """Set selected items."""
  803. self._resetSelectVariables()
  804. mixed = []
  805. for item in selected:
  806. type = item.data["type"]
  807. if type in ("raster", "raster_3d", "vector"):
  808. self.selected_layer.append(item)
  809. self.selected_mapset.append(item.parent)
  810. self.selected_location.append(item.parent.parent)
  811. self.selected_grassdb.append(item.parent.parent.parent)
  812. mixed.append("layer")
  813. elif type == "mapset":
  814. self.selected_layer.append(None)
  815. self.selected_mapset.append(item)
  816. self.selected_location.append(item.parent)
  817. self.selected_grassdb.append(item.parent.parent)
  818. mixed.append("mapset")
  819. elif type == "location":
  820. self.selected_layer.append(None)
  821. self.selected_mapset.append(None)
  822. self.selected_location.append(item)
  823. self.selected_grassdb.append(item.parent)
  824. mixed.append("location")
  825. elif type == "grassdb":
  826. self.selected_layer.append(None)
  827. self.selected_mapset.append(None)
  828. self.selected_location.append(None)
  829. self.selected_grassdb.append(item)
  830. mixed.append("grassdb")
  831. self.mixed = False
  832. if len(set(mixed)) > 1:
  833. self.mixed = True
  834. def OnSelChanged(self, event):
  835. self.selected_layer = None
  836. def OnRightClick(self, node):
  837. """Display popup menu."""
  838. self.DefineItems(self.GetSelected())
  839. if self.mixed:
  840. self._popupMenuEmpty()
  841. return
  842. if not self.selected_layer:
  843. self._popupMenuEmpty()
  844. elif self.selected_layer[0]:
  845. self._popupMenuLayer()
  846. elif self.selected_mapset[0] and len(self.selected_mapset) == 1:
  847. self._popupMenuMapset()
  848. elif (
  849. self.selected_location[0]
  850. and not self.selected_mapset[0]
  851. and len(self.selected_location) == 1
  852. ):
  853. self._popupMenuLocation()
  854. elif (
  855. self.selected_grassdb[0]
  856. and not self.selected_location[0]
  857. and len(self.selected_grassdb) == 1
  858. ):
  859. self._popupMenuGrassDb()
  860. elif len(self.selected_grassdb) > 1 and not self.selected_location[0]:
  861. self._popupMenuEmpty()
  862. elif len(self.selected_location) > 1 and not self.selected_mapset[0]:
  863. self._popupMenuMultipleLocations()
  864. elif len(self.selected_mapset) > 1:
  865. self._popupMenuMultipleMapsets()
  866. else:
  867. self._popupMenuEmpty()
  868. def OnDoubleClick(self, node):
  869. """Double click on item/node.
  870. Display selected layer if node is a map layer otherwise
  871. expand/collapse node.
  872. """
  873. if not isinstance(self._giface, StandaloneGrassInterface):
  874. self.DefineItems([node])
  875. selected_layer = self.selected_layer[0]
  876. selected_mapset = self.selected_mapset[0]
  877. selected_loc = self.selected_location[0]
  878. if selected_layer is not None:
  879. genv = gisenv()
  880. # Check if the layer is in different location
  881. if selected_loc.data["name"] != genv["LOCATION_NAME"]:
  882. dlg = wx.MessageDialog(
  883. parent=self,
  884. message=_(
  885. "Map <{0}@{1}> is not in the current location"
  886. " and therefore cannot be displayed."
  887. "\n\n"
  888. "To display this map switch to mapset <{1}> first."
  889. ).format(
  890. selected_layer.data["name"], selected_mapset.data["name"]
  891. ),
  892. caption=_("Unable to display the map"),
  893. style=wx.OK | wx.ICON_WARNING,
  894. )
  895. dlg.ShowModal()
  896. dlg.Destroy()
  897. else:
  898. self.DisplayLayer()
  899. return
  900. if node.data["type"] == "mapset" and not node.children:
  901. self._reloadMapsetNode(node)
  902. self.RefreshNode(node, recursive=True)
  903. # expand/collapse location/mapset...
  904. if self.IsNodeExpanded(node):
  905. self.CollapseNode(node, recursive=False)
  906. else:
  907. self.ExpandNode(node, recursive=False)
  908. def ExpandCurrentLocation(self):
  909. """Expand current location"""
  910. location = gscript.gisenv()["LOCATION_NAME"]
  911. item = self._model.SearchNodes(name=location, type="location")
  912. if item:
  913. self.Select(item[0], select=True)
  914. self.ExpandNode(item[0], recursive=False)
  915. else:
  916. Debug.msg(1, "Location <%s> not found" % location)
  917. def GetCurrentDbLocationMapsetNode(self):
  918. """Get current mapset node"""
  919. genv = gisenv()
  920. gisdbase = genv["GISDBASE"]
  921. location = genv["LOCATION_NAME"]
  922. mapset = genv["MAPSET"]
  923. grassdbItem = self._model.SearchNodes(name=gisdbase, type="grassdb")
  924. if not grassdbItem:
  925. return None, None, None
  926. locationItem = self._model.SearchNodes(
  927. parent=grassdbItem[0], name=location, type="location"
  928. )
  929. if not locationItem:
  930. return grassdbItem[0], None, None
  931. mapsetItem = self._model.SearchNodes(
  932. parent=locationItem[0], name=mapset, type="mapset"
  933. )
  934. if not mapsetItem:
  935. return grassdbItem[0], locationItem[0], None
  936. return grassdbItem[0], locationItem[0], mapsetItem[0]
  937. def OnGetItemImage(self, index, which=wx.TreeItemIcon_Normal, column=0):
  938. """Overridden method to return image for each item."""
  939. node = self._model.GetNodeByIndex(index)
  940. try:
  941. return self._iconTypes.index(node.data["type"])
  942. except ValueError:
  943. return 0
  944. def OnGetItemTextColour(self, index):
  945. """Overridden method to return colour for each item.
  946. Used to distinguish lock and ownership on mapsets."""
  947. node = self._model.GetNodeByIndex(index)
  948. if node.data["type"] == "mapset":
  949. if node.data["current"]:
  950. return wx.SystemSettings.GetColour(wx.SYS_COLOUR_WINDOWTEXT)
  951. elif node.data["lock"] or node.data["is_different_owner"]:
  952. return wx.SystemSettings.GetColour(wx.SYS_COLOUR_GRAYTEXT)
  953. return wx.SystemSettings.GetColour(wx.SYS_COLOUR_WINDOWTEXT)
  954. def OnGetItemFont(self, index):
  955. """Overridden method to return font for each item.
  956. Used to highlight current db/loc/mapset."""
  957. node = self._model.GetNodeByIndex(index)
  958. font = self.GetFont()
  959. if node.data["type"] in ("grassdb", "location", "mapset"):
  960. if node in (
  961. self.current_grassdb_node,
  962. self.current_location_node,
  963. self.current_mapset_node,
  964. ):
  965. font.SetWeight(wx.FONTWEIGHT_BOLD)
  966. else:
  967. font.SetWeight(wx.FONTWEIGHT_NORMAL)
  968. return font
  969. def ExpandCurrentMapset(self):
  970. """Expand current mapset"""
  971. if self.current_mapset_node:
  972. self.Select(self.current_mapset_node, select=True)
  973. self.ExpandNode(self.current_mapset_node, recursive=True)
  974. def SetRestriction(self, restrict):
  975. self._restricted = restrict
  976. def _runCommand(self, prog, **kwargs):
  977. cmdString = " ".join(gscript.make_command(prog, **kwargs))
  978. ret = RunCommand(prog, parent=self, **kwargs)
  979. return ret, cmdString
  980. def OnMoveMap(self, event):
  981. """Move layer or mapset (just save it temporarily, copying is done by paste)"""
  982. self.copy_mode = False
  983. self.copy_layer = self.selected_layer[:]
  984. self.copy_mapset = self.selected_mapset[:]
  985. self.copy_location = self.selected_location[:]
  986. self.copy_grassdb = self.selected_grassdb[:]
  987. if len(self.copy_layer) > 1:
  988. label = _("{c} maps marked for moving.").format(c=len(self.selected_layer))
  989. else:
  990. label = _("Map <{layer}> marked for moving.").format(
  991. layer=self.copy_layer[0].data["name"]
  992. )
  993. self.showNotification.emit(message=label)
  994. def OnCopyMap(self, event):
  995. """Copy layer or mapset (just save it temporarily, copying is done by paste)"""
  996. self.copy_mode = True
  997. self.copy_layer = self.selected_layer[:]
  998. self.copy_mapset = self.selected_mapset[:]
  999. self.copy_location = self.selected_location[:]
  1000. self.copy_grassdb = self.selected_grassdb[:]
  1001. if len(self.copy_layer) > 1:
  1002. label = _("{c} maps marked for copying.").format(c=len(self.selected_layer))
  1003. else:
  1004. label = _("Map <{layer}> marked for copying.").format(
  1005. layer=self.copy_layer[0].data["name"]
  1006. )
  1007. self.showNotification.emit(message=label)
  1008. def OnRenameMap(self, event):
  1009. """Rename layer with dialog"""
  1010. old_name = self.selected_layer[0].data["name"]
  1011. gisrc, env = gscript.create_environment(
  1012. self.selected_grassdb[0].data["name"],
  1013. self.selected_location[0].data["name"],
  1014. self.selected_mapset[0].data["name"],
  1015. )
  1016. new_name = self._getNewMapName(
  1017. _("New name"),
  1018. _("Rename map"),
  1019. old_name,
  1020. env=env,
  1021. mapset=self.selected_mapset[0].data["name"],
  1022. element=self.selected_layer[0].data["type"],
  1023. )
  1024. if new_name:
  1025. self.Rename(old_name, new_name)
  1026. def CreateMapset(self, grassdb_node, location_node):
  1027. """Creates new mapset interactively and adds it to the tree."""
  1028. mapset = create_mapset_interactively(
  1029. self, grassdb_node.data["name"], location_node.data["name"]
  1030. )
  1031. if mapset:
  1032. self._giface.grassdbChanged.emit(
  1033. grassdb=grassdb_node.data["name"],
  1034. location=location_node.data["name"],
  1035. mapset=mapset,
  1036. element="mapset",
  1037. action="new",
  1038. )
  1039. self.SwitchMapset(
  1040. grassdb_node.data["name"],
  1041. location_node.data["name"],
  1042. mapset,
  1043. show_confirmation=True,
  1044. )
  1045. def OnCreateMapset(self, event):
  1046. """Create new mapset"""
  1047. self.CreateMapset(self.selected_grassdb[0], self.selected_location[0])
  1048. def CreateLocation(self, grassdb_node):
  1049. """
  1050. Creates new location interactively and adds it to the tree and switch
  1051. to its new PERMANENT mapset.
  1052. If a user is a first-time user, it shows data import infobar.
  1053. """
  1054. grassdatabase, location, mapset = create_location_interactively(
  1055. self, grassdb_node.data["name"]
  1056. )
  1057. if location:
  1058. self._giface.grassdbChanged.emit(
  1059. grassdb=grassdatabase,
  1060. location=location,
  1061. element="location",
  1062. action="new",
  1063. )
  1064. # switch to PERMANENT mapset in newly created location
  1065. self.SwitchMapset(grassdatabase, location, mapset, show_confirmation=True)
  1066. # show data import infobar for first-time user with proper layout
  1067. if is_first_time_user():
  1068. self.showImportDataInfo.emit()
  1069. def OnCreateLocation(self, event):
  1070. """Create new location"""
  1071. self.CreateLocation(self.selected_grassdb[0])
  1072. def OnRenameMapset(self, event):
  1073. """
  1074. Rename selected mapset
  1075. """
  1076. newmapset = rename_mapset_interactively(
  1077. self,
  1078. self.selected_grassdb[0].data["name"],
  1079. self.selected_location[0].data["name"],
  1080. self.selected_mapset[0].data["name"],
  1081. )
  1082. if newmapset:
  1083. self._giface.grassdbChanged.emit(
  1084. grassdb=self.selected_grassdb[0].data["name"],
  1085. location=self.selected_location[0].data["name"],
  1086. mapset=self.selected_mapset[0].data["name"],
  1087. element="mapset",
  1088. action="rename",
  1089. newname=newmapset,
  1090. )
  1091. def OnRenameLocation(self, event):
  1092. """
  1093. Rename selected location
  1094. """
  1095. newlocation = rename_location_interactively(
  1096. self,
  1097. self.selected_grassdb[0].data["name"],
  1098. self.selected_location[0].data["name"],
  1099. )
  1100. if newlocation:
  1101. self._giface.grassdbChanged.emit(
  1102. grassdb=self.selected_grassdb[0].data["name"],
  1103. location=self.selected_location[0].data["name"],
  1104. element="location",
  1105. action="rename",
  1106. newname=newlocation,
  1107. )
  1108. def OnStartEditLabel(self, node, event):
  1109. """Start label editing"""
  1110. self.DefineItems([node])
  1111. # Not allowed for grassdb node
  1112. if node.data["type"] == "grassdb":
  1113. event.Veto()
  1114. # Check selected mapset
  1115. elif node.data["type"] == "mapset":
  1116. if self._restricted or get_reason_mapset_not_removable(
  1117. self.selected_grassdb[0].data["name"],
  1118. self.selected_location[0].data["name"],
  1119. self.selected_mapset[0].data["name"],
  1120. check_permanent=True,
  1121. ):
  1122. event.Veto()
  1123. # Check selected location
  1124. elif node.data["type"] == "location":
  1125. if self._restricted or get_reasons_location_not_removable(
  1126. self.selected_grassdb[0].data["name"],
  1127. self.selected_location[0].data["name"],
  1128. ):
  1129. event.Veto()
  1130. elif node.data["type"] in ("raster", "raster_3d", "vector"):
  1131. currentGrassDb, currentLocation, currentMapset = self._isCurrent(gisenv())
  1132. if not currentMapset:
  1133. event.Veto()
  1134. def OnEditLabel(self, node, event):
  1135. """End label editing"""
  1136. if event.IsEditCancelled():
  1137. return
  1138. old_name = node.data["name"]
  1139. Debug.msg(1, "End label edit {name}".format(name=old_name))
  1140. new_name = event.GetLabel()
  1141. if node.data["type"] in ("raster", "raster_3d", "vector"):
  1142. self.Rename(old_name, new_name)
  1143. elif node.data["type"] == "mapset":
  1144. message = get_mapset_name_invalid_reason(
  1145. self.selected_grassdb[0].data["name"],
  1146. self.selected_location[0].data["name"],
  1147. new_name,
  1148. )
  1149. if message:
  1150. GError(
  1151. parent=self,
  1152. message=message,
  1153. caption=_("Cannot rename mapset"),
  1154. showTraceback=False,
  1155. )
  1156. event.Veto()
  1157. return
  1158. rename_mapset(
  1159. self.selected_grassdb[0].data["name"],
  1160. self.selected_location[0].data["name"],
  1161. self.selected_mapset[0].data["name"],
  1162. new_name,
  1163. )
  1164. self._renameNode(self.selected_mapset[0], new_name)
  1165. label = _(
  1166. "Renaming mapset <{oldmapset}> to <{newmapset}> completed"
  1167. ).format(oldmapset=old_name, newmapset=new_name)
  1168. self.showNotification.emit(message=label)
  1169. elif node.data["type"] == "location":
  1170. message = get_location_name_invalid_reason(
  1171. self.selected_grassdb[0].data["name"], new_name
  1172. )
  1173. if message:
  1174. GError(
  1175. parent=self,
  1176. message=message,
  1177. caption=_("Cannot rename location"),
  1178. showTraceback=False,
  1179. )
  1180. event.Veto()
  1181. return
  1182. rename_location(
  1183. self.selected_grassdb[0].data["name"],
  1184. self.selected_location[0].data["name"],
  1185. new_name,
  1186. )
  1187. self._renameNode(self.selected_location[0], new_name)
  1188. label = _(
  1189. "Renaming location <{oldlocation}> to <{newlocation}> completed"
  1190. ).format(oldlocation=old_name, newlocation=new_name)
  1191. self.showNotification.emit(message=label)
  1192. def Rename(self, old, new):
  1193. """Rename layer"""
  1194. string = old + "," + new
  1195. gisrc, env = gscript.create_environment(
  1196. self.selected_grassdb[0].data["name"],
  1197. self.selected_location[0].data["name"],
  1198. self.selected_mapset[0].data["name"],
  1199. )
  1200. label = _("Renaming map <{name}>...").format(name=string)
  1201. self.showNotification.emit(message=label)
  1202. if self.selected_layer[0].data["type"] == "vector":
  1203. renamed, cmd = self._runCommand("g.rename", vector=string, env=env)
  1204. elif self.selected_layer[0].data["type"] == "raster":
  1205. renamed, cmd = self._runCommand("g.rename", raster=string, env=env)
  1206. else:
  1207. renamed, cmd = self._runCommand("g.rename", raster3d=string, env=env)
  1208. gscript.try_remove(gisrc)
  1209. if renamed == 0:
  1210. self.showNotification.emit(message=_("{cmd} -- completed").format(cmd=cmd))
  1211. Debug.msg(1, "LAYER RENAMED TO: " + new)
  1212. self._giface.grassdbChanged.emit(
  1213. grassdb=self.selected_grassdb[0].data["name"],
  1214. location=self.selected_location[0].data["name"],
  1215. mapset=self.selected_mapset[0].data["name"],
  1216. map=old,
  1217. element=self.selected_layer[0].data["type"],
  1218. newname=new,
  1219. action="rename",
  1220. )
  1221. def OnPasteMap(self, event):
  1222. # copying between mapsets of one location
  1223. if not self.copy_layer:
  1224. if self.copy_mode:
  1225. GMessage(_("No map selected for copying."), parent=self)
  1226. else:
  1227. GMessage(_("No map selected for moving."), parent=self)
  1228. return
  1229. for i in range(len(self.copy_layer)):
  1230. gisrc, env = gscript.create_environment(
  1231. self.selected_grassdb[0].data["name"],
  1232. self.selected_location[0].data["name"],
  1233. self.selected_mapset[0].data["name"],
  1234. )
  1235. gisrc2, env2 = gscript.create_environment(
  1236. self.copy_grassdb[i].data["name"],
  1237. self.copy_location[i].data["name"],
  1238. self.copy_mapset[i].data["name"],
  1239. )
  1240. new_name = self.copy_layer[i].data["name"]
  1241. if self.selected_location[0] == self.copy_location[i]:
  1242. # within one mapset
  1243. if self.selected_mapset[0] == self.copy_mapset[i]:
  1244. # ignore when just moves map
  1245. if self.copy_mode is False:
  1246. return
  1247. new_name = self._getNewMapName(
  1248. _("New name for <{n}>").format(
  1249. n=self.copy_layer[i].data["name"]
  1250. ),
  1251. _("Select new name"),
  1252. self.copy_layer[i].data["name"],
  1253. env=env,
  1254. mapset=self.selected_mapset[0].data["name"],
  1255. element=self.copy_layer[i].data["type"],
  1256. )
  1257. if not new_name:
  1258. return
  1259. # within one location, different mapsets
  1260. else:
  1261. if map_exists(
  1262. new_name,
  1263. element=self.copy_layer[i].data["type"],
  1264. env=env,
  1265. mapset=self.selected_mapset[0].data["name"],
  1266. ):
  1267. new_name = self._getNewMapName(
  1268. _("New name for <{n}>").format(
  1269. n=self.copy_layer[i].data["name"]
  1270. ),
  1271. _("Select new name"),
  1272. self.copy_layer[i].data["name"],
  1273. env=env,
  1274. mapset=self.selected_mapset[0].data["name"],
  1275. element=self.copy_layer[i].data["type"],
  1276. )
  1277. if not new_name:
  1278. return
  1279. string = (
  1280. self.copy_layer[i].data["name"]
  1281. + "@"
  1282. + self.copy_mapset[i].data["name"]
  1283. + ","
  1284. + new_name
  1285. )
  1286. pasted = 0
  1287. if self.copy_mode:
  1288. label = _("Copying <{name}>...").format(name=string)
  1289. else:
  1290. label = _("Moving <{name}>...").format(name=string)
  1291. self.showNotification.emit(message=label)
  1292. if self.copy_layer[i].data["type"] == "vector":
  1293. pasted, cmd = self._runCommand("g.copy", vector=string, env=env)
  1294. node = "vector"
  1295. elif self.copy_layer[i].data["type"] == "raster":
  1296. pasted, cmd = self._runCommand("g.copy", raster=string, env=env)
  1297. node = "raster"
  1298. else:
  1299. pasted, cmd = self._runCommand("g.copy", raster_3d=string, env=env)
  1300. node = "raster_3d"
  1301. if pasted == 0:
  1302. Debug.msg(1, "COPIED TO: " + new_name)
  1303. if self.copy_mode:
  1304. self.showNotification.emit(message=_("g.copy completed"))
  1305. else:
  1306. self.showNotification.emit(message=_("g.copy completed"))
  1307. self._giface.grassdbChanged.emit(
  1308. grassdb=self.selected_grassdb[0].data["name"],
  1309. location=self.selected_location[0].data["name"],
  1310. mapset=self.selected_mapset[0].data["name"],
  1311. map=new_name,
  1312. element=node,
  1313. action="new",
  1314. )
  1315. # remove old
  1316. if not self.copy_mode:
  1317. self._removeMapAfterCopy(
  1318. self.copy_layer[i], self.copy_mapset[i], env2
  1319. )
  1320. gscript.try_remove(gisrc)
  1321. gscript.try_remove(gisrc2)
  1322. # expand selected mapset
  1323. else:
  1324. if self.copy_layer[i].data["type"] == "raster_3d":
  1325. GError(
  1326. _("Reprojection is not implemented for 3D rasters"), parent=self
  1327. )
  1328. return
  1329. if map_exists(
  1330. new_name,
  1331. element=self.copy_layer[i].data["type"],
  1332. env=env,
  1333. mapset=self.selected_mapset[0].data["name"],
  1334. ):
  1335. new_name = self._getNewMapName(
  1336. _("New name"),
  1337. _("Select new name"),
  1338. self.copy_layer[i].data["name"],
  1339. env=env,
  1340. mapset=self.selected_mapset[0].data["name"],
  1341. element=self.copy_layer[i].data["type"],
  1342. )
  1343. if not new_name:
  1344. continue
  1345. callback = lambda gisrc2=gisrc2, gisrc=gisrc, cLayer=self.copy_layer[
  1346. i
  1347. ], cMapset=self.copy_mapset[
  1348. i
  1349. ], cMode=self.copy_mode, sMapset=self.selected_mapset[
  1350. 0
  1351. ], name=new_name: self._onDoneReprojection(
  1352. env2, gisrc2, gisrc, cLayer, cMapset, cMode, sMapset, name
  1353. )
  1354. dlg = CatalogReprojectionDialog(
  1355. self,
  1356. self._giface,
  1357. self.copy_grassdb[i].data["name"],
  1358. self.copy_location[i].data["name"],
  1359. self.copy_mapset[i].data["name"],
  1360. self.copy_layer[i].data["name"],
  1361. env2,
  1362. self.selected_grassdb[0].data["name"],
  1363. self.selected_location[0].data["name"],
  1364. self.selected_mapset[0].data["name"],
  1365. new_name,
  1366. self.copy_layer[i].data["type"],
  1367. env,
  1368. callback,
  1369. )
  1370. if dlg.ShowModal() == wx.ID_CANCEL:
  1371. return
  1372. self.ExpandNode(self.selected_mapset[0], recursive=True)
  1373. self._resetCopyVariables()
  1374. def _onDoneReprojection(
  1375. self, iEnv, iGisrc, oGisrc, cLayer, cMapset, cMode, sMapset, name
  1376. ):
  1377. self._giface.grassdbChanged.emit(
  1378. grassdb=sMapset.parent.parent.data["name"],
  1379. location=sMapset.parent.data["name"],
  1380. mapset=sMapset.data["name"],
  1381. element=cLayer.data["type"],
  1382. map=name,
  1383. action="new",
  1384. )
  1385. if not cMode:
  1386. self._removeMapAfterCopy(cLayer, cMapset, iEnv)
  1387. gscript.try_remove(iGisrc)
  1388. gscript.try_remove(oGisrc)
  1389. self.ExpandNode(sMapset, recursive=True)
  1390. def _removeMapAfterCopy(self, cLayer, cMapset, env):
  1391. removed, cmd = self._runCommand(
  1392. "g.remove",
  1393. type=cLayer.data["type"],
  1394. name=cLayer.data["name"],
  1395. flags="f",
  1396. env=env,
  1397. )
  1398. if removed == 0:
  1399. Debug.msg(1, "LAYER " + cLayer.data["name"] + " DELETED")
  1400. self.showNotification.emit(message=_("g.remove completed"))
  1401. self._giface.grassdbChanged.emit(
  1402. grassdb=cMapset.parent.parent.data["name"],
  1403. location=cMapset.parent.data["name"],
  1404. mapset=cMapset.data["name"],
  1405. map=cLayer.data["name"],
  1406. element=cLayer.data["type"],
  1407. action="delete",
  1408. )
  1409. def InsertLayer(self, name, mapset_node, element_name):
  1410. """Insert layer into model and refresh tree"""
  1411. self._model.AppendNode(
  1412. parent=mapset_node, data=dict(type=element_name, name=name)
  1413. )
  1414. self._model.SortChildren(mapset_node)
  1415. self.RefreshNode(mapset_node, recursive=True)
  1416. def InsertMapset(self, name, location_node):
  1417. """Insert new mapset into model and refresh tree.
  1418. Assumes mapset is empty."""
  1419. mapset_path = os.path.join(
  1420. location_node.parent.data["name"], location_node.data["name"], name
  1421. )
  1422. mapset_node = self._model.AppendNode(
  1423. parent=location_node,
  1424. data=dict(
  1425. type="mapset",
  1426. name=name,
  1427. current=False,
  1428. lock=is_mapset_locked(mapset_path),
  1429. is_different_owner=is_different_mapset_owner(mapset_path),
  1430. owner=get_mapset_owner(mapset_path),
  1431. ),
  1432. )
  1433. self._model.SortChildren(location_node)
  1434. self.RefreshNode(location_node, recursive=True)
  1435. return mapset_node
  1436. def InsertLocation(self, name, grassdb_node):
  1437. """Insert new location into model and refresh tree"""
  1438. location_node = self._model.AppendNode(
  1439. parent=grassdb_node, data=dict(type="location", name=name)
  1440. )
  1441. # reload new location since it has a mapset
  1442. self._reloadLocationNode(location_node)
  1443. self._model.SortChildren(grassdb_node)
  1444. self.RefreshNode(grassdb_node, recursive=True)
  1445. return location_node
  1446. def InsertGrassDb(self, name):
  1447. """
  1448. Insert new grass db into model, update user setting and refresh tree.
  1449. Check if not already added.
  1450. """
  1451. grassdb_node = self._model.SearchNodes(name=name, type="grassdb")
  1452. if not grassdb_node:
  1453. grassdb_node = self._model.AppendNode(
  1454. parent=self._model.root, data=dict(type="grassdb", name=name)
  1455. )
  1456. if self._useLazyLoading():
  1457. self._lazyReloadGrassDBNode(grassdb_node)
  1458. else:
  1459. self._reloadGrassDBNode(grassdb_node)
  1460. self.RefreshItems()
  1461. # Update user's settings
  1462. self.grassdatabases.append(name)
  1463. self._saveGrassDBs()
  1464. return grassdb_node
  1465. def OnDeleteMap(self, event):
  1466. """Delete layer or mapset"""
  1467. names = [
  1468. self.selected_layer[i].data["name"]
  1469. + "@"
  1470. + self.selected_mapset[i].data["name"]
  1471. for i in range(len(self.selected_layer))
  1472. ]
  1473. if len(names) < 10:
  1474. question = _("Do you really want to delete map(s) <{m}>?").format(
  1475. m=", ".join(names)
  1476. )
  1477. else:
  1478. question = _("Do you really want to delete {n} maps?").format(n=len(names))
  1479. if self._confirmDialog(question, title=_("Delete map")) == wx.ID_YES:
  1480. label = _("Deleting {name}...").format(name=names)
  1481. self.showNotification.emit(message=label)
  1482. for i in range(len(self.selected_layer)):
  1483. gisrc, env = gscript.create_environment(
  1484. self.selected_grassdb[i].data["name"],
  1485. self.selected_location[i].data["name"],
  1486. self.selected_mapset[i].data["name"],
  1487. )
  1488. removed, cmd = self._runCommand(
  1489. "g.remove",
  1490. flags="f",
  1491. type=self.selected_layer[i].data["type"],
  1492. name=self.selected_layer[i].data["name"],
  1493. env=env,
  1494. )
  1495. gscript.try_remove(gisrc)
  1496. if removed == 0:
  1497. self._giface.grassdbChanged.emit(
  1498. grassdb=self.selected_grassdb[i].data["name"],
  1499. location=self.selected_location[i].data["name"],
  1500. mapset=self.selected_mapset[i].data["name"],
  1501. element=self.selected_layer[i].data["type"],
  1502. map=self.selected_layer[i].data["name"],
  1503. action="delete",
  1504. )
  1505. Debug.msg(
  1506. 1, "LAYER " + self.selected_layer[i].data["name"] + " DELETED"
  1507. )
  1508. self.UnselectAll()
  1509. self.showNotification.emit(message=_("g.remove completed"))
  1510. def OnDeleteMapset(self, event):
  1511. """
  1512. Delete selected mapset or mapsets
  1513. """
  1514. mapsets = []
  1515. changes = []
  1516. for i in range(len(self.selected_mapset)):
  1517. # Append to the list of tuples
  1518. mapsets.append(
  1519. (
  1520. self.selected_grassdb[i].data["name"],
  1521. self.selected_location[i].data["name"],
  1522. self.selected_mapset[i].data["name"],
  1523. )
  1524. )
  1525. changes.append(
  1526. dict(
  1527. grassdb=self.selected_grassdb[i].data["name"],
  1528. location=self.selected_location[i].data["name"],
  1529. mapset=self.selected_mapset[i].data["name"],
  1530. action="delete",
  1531. element="mapset",
  1532. )
  1533. )
  1534. if delete_mapsets_interactively(self, mapsets):
  1535. for change in changes:
  1536. self._giface.grassdbChanged.emit(**change)
  1537. def OnDeleteLocation(self, event):
  1538. """
  1539. Delete selected location or locations
  1540. """
  1541. locations = []
  1542. changes = []
  1543. for i in range(len(self.selected_location)):
  1544. # Append to the list of tuples
  1545. locations.append(
  1546. (
  1547. self.selected_grassdb[i].data["name"],
  1548. self.selected_location[i].data["name"],
  1549. )
  1550. )
  1551. changes.append(
  1552. dict(
  1553. grassdb=self.selected_grassdb[i].data["name"],
  1554. location=self.selected_location[i].data["name"],
  1555. action="delete",
  1556. element="location",
  1557. )
  1558. )
  1559. if delete_locations_interactively(self, locations):
  1560. for change in changes:
  1561. self._giface.grassdbChanged.emit(**change)
  1562. def DownloadLocation(self, grassdb_node):
  1563. """
  1564. Download new location interactively.
  1565. """
  1566. grassdatabase, location, mapset = download_location_interactively(
  1567. self, grassdb_node.data["name"]
  1568. )
  1569. if location:
  1570. self._reloadGrassDBNode(grassdb_node)
  1571. self.UpdateCurrentDbLocationMapsetNode()
  1572. self.RefreshItems()
  1573. def OnDownloadLocation(self, event):
  1574. """
  1575. Download location online
  1576. """
  1577. self.DownloadLocation(self.selected_grassdb[0])
  1578. def DeleteGrassDb(self, grassdb_node):
  1579. """
  1580. Delete grassdb from disk.
  1581. """
  1582. grassdb = grassdb_node.data["name"]
  1583. if delete_grassdb_interactively(self, grassdb):
  1584. self.RemoveGrassDB(grassdb_node)
  1585. def OnDeleteGrassDb(self, event):
  1586. """
  1587. Delete grassdb from disk.
  1588. """
  1589. self.DeleteGrassDb(self.selected_grassdb[0])
  1590. def OnRemoveGrassDb(self, event):
  1591. """
  1592. Remove grassdb node from data catalogue.
  1593. """
  1594. self.RemoveGrassDB(self.selected_grassdb[0])
  1595. def RemoveGrassDB(self, grassdb_node):
  1596. """
  1597. Remove grassdb node from tree
  1598. and updates settings. Doesn't check if it's current db.
  1599. """
  1600. self.grassdatabases.remove(grassdb_node.data["name"])
  1601. self._model.RemoveNode(grassdb_node)
  1602. self.RefreshItems()
  1603. # Update user's settings
  1604. self._saveGrassDBs()
  1605. def OnDisplayLayer(self, event):
  1606. """
  1607. Display layer in current graphics view
  1608. """
  1609. self.DisplayLayer()
  1610. def DisplayLayer(self):
  1611. """Display selected layer in current graphics view"""
  1612. all_names = []
  1613. names = {"raster": [], "vector": [], "raster_3d": []}
  1614. for i in range(len(self.selected_layer)):
  1615. name = (
  1616. self.selected_layer[i].data["name"]
  1617. + "@"
  1618. + self.selected_mapset[i].data["name"]
  1619. )
  1620. names[self.selected_layer[i].data["type"]].append(name)
  1621. all_names.append(name)
  1622. # if self.selected_location[0].data['name'] == gisenv()['LOCATION_NAME'] and self.selected_mapset[0]:
  1623. for ltype in names:
  1624. if names[ltype]:
  1625. self._giface.lmgr.AddMaps(list(reversed(names[ltype])), ltype, True)
  1626. if len(self._giface.GetLayerList()) == 1:
  1627. # zoom to map if there is only one map layer
  1628. self._giface.GetMapWindow().ZoomToMap()
  1629. Debug.msg(1, "Displayed layer(s): " + str(all_names))
  1630. def OnBeginDrag(self, node, event):
  1631. """Just copy necessary data"""
  1632. self.DefineItems(self.GetSelected())
  1633. if None not in self.selected_layer and not (
  1634. self._restricted
  1635. and gisenv()["LOCATION_NAME"] != self.selected_location[0].data["name"]
  1636. ):
  1637. event.Allow()
  1638. self.OnCopyMap(event)
  1639. Debug.msg(1, "DRAG")
  1640. else:
  1641. event.Veto()
  1642. def OnEndDrag(self, node, event):
  1643. """Copy layer into target"""
  1644. self.copy_mode = wx.GetMouseState().ControlDown()
  1645. if node:
  1646. self.DefineItems([node])
  1647. if None not in self.selected_mapset:
  1648. if (
  1649. self._restricted
  1650. and gisenv()["MAPSET"] != self.selected_mapset[0].data["name"]
  1651. ):
  1652. GMessage(
  1653. _(
  1654. "To move or copy maps to other mapsets, unlock editing of other mapsets"
  1655. ),
  1656. parent=self,
  1657. )
  1658. event.Veto()
  1659. return
  1660. event.Allow()
  1661. Debug.msg(1, "DROP DONE")
  1662. self.OnPasteMap(event)
  1663. else:
  1664. GMessage(
  1665. _(
  1666. "To move or copy maps to other location, "
  1667. "please drag them to a mapset in the "
  1668. "destination location"
  1669. ),
  1670. parent=self,
  1671. )
  1672. event.Veto()
  1673. return
  1674. def SwitchMapset(self, grassdb, location, mapset, show_confirmation=False):
  1675. """
  1676. Switch to location and mapset interactively.
  1677. """
  1678. if can_switch_mapset_interactive(self, grassdb, location, mapset):
  1679. genv = gisenv()
  1680. # Switch to mapset in the same location
  1681. if grassdb == genv["GISDBASE"] and location == genv["LOCATION_NAME"]:
  1682. switch_mapset_interactively(
  1683. self, self._giface, None, None, mapset, show_confirmation
  1684. )
  1685. # Switch to mapset in the same grassdb
  1686. elif grassdb == genv["GISDBASE"]:
  1687. switch_mapset_interactively(
  1688. self, self._giface, None, location, mapset, show_confirmation
  1689. )
  1690. # Switch to mapset in a different grassdb
  1691. else:
  1692. switch_mapset_interactively(
  1693. self, self._giface, grassdb, location, mapset, show_confirmation
  1694. )
  1695. def OnSwitchMapset(self, event):
  1696. """Switch to location and mapset"""
  1697. grassdb = self.selected_grassdb[0].data["name"]
  1698. location = self.selected_location[0].data["name"]
  1699. mapset = self.selected_mapset[0].data["name"]
  1700. self.SwitchMapset(grassdb, location, mapset)
  1701. def _updateAfterGrassdbChanged(
  1702. self, action, element, grassdb, location, mapset=None, map=None, newname=None
  1703. ):
  1704. """Update tree after grassdata changed"""
  1705. if element == "mapset":
  1706. if action == "new":
  1707. node = self.GetDbNode(grassdb=grassdb, location=location)
  1708. if node:
  1709. self.InsertMapset(name=mapset, location_node=node)
  1710. elif action == "delete":
  1711. node = self.GetDbNode(grassdb=grassdb, location=location)
  1712. if node:
  1713. self._reloadLocationNode(node)
  1714. self.UpdateCurrentDbLocationMapsetNode()
  1715. self.RefreshNode(node, recursive=True)
  1716. elif action == "rename":
  1717. node = self.GetDbNode(grassdb=grassdb, location=location, mapset=mapset)
  1718. if node:
  1719. self._renameNode(node, newname)
  1720. elif element == "location":
  1721. if action == "new":
  1722. node = self.GetDbNode(grassdb=grassdb)
  1723. if not node:
  1724. node = self.InsertGrassDb(name=grassdb)
  1725. if node:
  1726. self.InsertLocation(location, node)
  1727. elif action == "delete":
  1728. node = self.GetDbNode(grassdb=grassdb)
  1729. if node:
  1730. self._reloadGrassDBNode(node)
  1731. self.UpdateCurrentDbLocationMapsetNode()
  1732. self.RefreshNode(node, recursive=True)
  1733. elif action == "rename":
  1734. node = self.GetDbNode(grassdb=grassdb, location=location)
  1735. if node:
  1736. self._renameNode(node, newname)
  1737. elif element == "grassdb":
  1738. if action == "delete":
  1739. node = self.GetDbNode(grassdb=grassdb)
  1740. if node:
  1741. self.RemoveGrassDB(node)
  1742. elif element in ("raster", "vector", "raster_3d"):
  1743. # when watchdog is used, it watches current mapset,
  1744. # so we don't process any signals here,
  1745. # instead the watchdog handler takes care of refreshing tree
  1746. if (
  1747. watchdog_used
  1748. and grassdb == self.current_grassdb_node.data["name"]
  1749. and location == self.current_location_node.data["name"]
  1750. and mapset == self.current_mapset_node.data["name"]
  1751. ):
  1752. return
  1753. if action == "new":
  1754. node = self.GetDbNode(grassdb=grassdb, location=location, mapset=mapset)
  1755. if node:
  1756. if map:
  1757. # reload entire mapset when using lazy loading
  1758. if not node.children and self._useLazyLoading():
  1759. self._reloadMapsetNode(node)
  1760. self.RefreshNode(node.parent, recursive=True)
  1761. # check if map already exists
  1762. elif not self._model.SearchNodes(
  1763. parent=node, name=map, type=element
  1764. ):
  1765. self.InsertLayer(
  1766. name=map, mapset_node=node, element_name=element
  1767. )
  1768. else:
  1769. # we know some maps created
  1770. self._reloadMapsetNode(node)
  1771. self.RefreshNode(node)
  1772. elif action == "delete":
  1773. node = self.GetDbNode(
  1774. grassdb=grassdb,
  1775. location=location,
  1776. mapset=mapset,
  1777. map=map,
  1778. map_type=element,
  1779. )
  1780. if node:
  1781. self._model.RemoveNode(node)
  1782. self.RefreshNode(node.parent, recursive=True)
  1783. elif action == "rename":
  1784. node = self.GetDbNode(
  1785. grassdb=grassdb,
  1786. location=location,
  1787. mapset=mapset,
  1788. map=map,
  1789. map_type=element,
  1790. )
  1791. if node:
  1792. self._renameNode(node, newname)
  1793. def _updateAfterMapsetChanged(self):
  1794. """Update tree after current mapset has changed"""
  1795. self.UpdateCurrentDbLocationMapsetNode()
  1796. self._reloadMapsetNode(self.current_mapset_node)
  1797. self.RefreshNode(self.current_mapset_node, recursive=True)
  1798. self.ExpandCurrentMapset()
  1799. self.RefreshItems()
  1800. self.ScheduleWatchCurrentMapset()
  1801. def OnMetadata(self, event):
  1802. """Show metadata of any raster/vector/3draster"""
  1803. def done(event):
  1804. gscript.try_remove(event.userData)
  1805. for i in range(len(self.selected_layer)):
  1806. if self.selected_layer[i].data["type"] == "raster":
  1807. cmd = ["r.info"]
  1808. elif self.selected_layer[i].data["type"] == "vector":
  1809. cmd = ["v.info"]
  1810. elif self.selected_layer[i].data["type"] == "raster_3d":
  1811. cmd = ["r3.info"]
  1812. cmd.append(
  1813. "map=%s@%s"
  1814. % (
  1815. self.selected_layer[i].data["name"],
  1816. self.selected_mapset[i].data["name"],
  1817. )
  1818. )
  1819. gisrc, env = gscript.create_environment(
  1820. self.selected_grassdb[i].data["name"],
  1821. self.selected_location[i].data["name"],
  1822. self.selected_mapset[i].data["name"],
  1823. )
  1824. # print output to command log area
  1825. # temp gisrc file must be deleted onDone
  1826. self._giface.RunCmd(cmd, env=env, onDone=done, userData=gisrc)
  1827. def OnCopyName(self, event):
  1828. """Copy layer name to clipboard"""
  1829. if wx.TheClipboard.Open():
  1830. do = wx.TextDataObject()
  1831. text = []
  1832. for i in range(len(self.selected_layer)):
  1833. text.append(
  1834. "%s@%s"
  1835. % (
  1836. self.selected_layer[i].data["name"],
  1837. self.selected_mapset[i].data["name"],
  1838. )
  1839. )
  1840. do.SetText(",".join(text))
  1841. wx.TheClipboard.SetData(do)
  1842. wx.TheClipboard.Close()
  1843. def Filter(self, text, element=None):
  1844. """Filter tree based on name and type."""
  1845. name = text.strip()
  1846. if not name:
  1847. self._model = self._orig_model
  1848. else:
  1849. try:
  1850. compiled = re.compile(name)
  1851. except re.error:
  1852. return
  1853. if element:
  1854. self._model = self._orig_model.Filtered(
  1855. method="filtering", name=compiled, type=element
  1856. )
  1857. else:
  1858. self._model = self._orig_model.Filtered(
  1859. method="filtering", name=compiled
  1860. )
  1861. self.UpdateCurrentDbLocationMapsetNode()
  1862. self.RefreshItems()
  1863. self.ExpandCurrentMapset()
  1864. if self._model.GetLeafCount(self._model.root) <= 50:
  1865. self.ExpandAll()
  1866. def OnReloadMapset(self, event):
  1867. """Reload selected mapset"""
  1868. node = self.selected_mapset[0]
  1869. self._reloadMapsetNode(node)
  1870. self.RefreshNode(node, recursive=True)
  1871. self.ExpandNode(node, recursive=False)
  1872. def OnReloadLocation(self, event):
  1873. """Reload all mapsets in selected location"""
  1874. node = self.selected_location[0]
  1875. self._reloadLocationNode(node)
  1876. self.UpdateCurrentDbLocationMapsetNode()
  1877. self.RefreshNode(node, recursive=True)
  1878. self.ExpandNode(node, recursive=False)
  1879. def OnReloadGrassdb(self, event):
  1880. """Reload all mapsets in selected grass db"""
  1881. node = self.selected_grassdb[0]
  1882. self.busy = wx.BusyCursor()
  1883. self.thread.Run(
  1884. callable=self._reloadGrassDBNode,
  1885. grassdb_node=node,
  1886. ondone=self._onDoneReloadingGrassdb,
  1887. userdata={"node": node},
  1888. )
  1889. def _onDoneReloadingGrassdb(self, event):
  1890. del self.busy
  1891. self.UpdateCurrentDbLocationMapsetNode()
  1892. self.RefreshNode(event.userdata["node"], recursive=True)
  1893. self.ExpandNode(event.userdata["node"], recursive=False)
  1894. def _getNewMapName(self, message, title, value, element, mapset, env):
  1895. """Dialog for simple text entry"""
  1896. dlg = NameEntryDialog(
  1897. parent=self,
  1898. message=message,
  1899. caption=title,
  1900. element=element,
  1901. env=env,
  1902. mapset=mapset,
  1903. )
  1904. dlg.SetValue(value)
  1905. if dlg.ShowModal() == wx.ID_OK:
  1906. name = dlg.GetValue()
  1907. else:
  1908. name = None
  1909. dlg.Destroy()
  1910. return name
  1911. def _confirmDialog(self, question, title):
  1912. """Confirm dialog"""
  1913. dlg = wx.MessageDialog(self, question, title, wx.YES_NO)
  1914. res = dlg.ShowModal()
  1915. dlg.Destroy()
  1916. return res
  1917. def _isCurrent(self, genv):
  1918. if self._restricted:
  1919. currentMapset = currentLocation = currentGrassDb = True
  1920. for i in range(len(self.selected_grassdb)):
  1921. if self.selected_grassdb[i].data["name"] != genv["GISDBASE"]:
  1922. currentGrassDb = False
  1923. currentLocation = False
  1924. currentMapset = False
  1925. break
  1926. if currentLocation and self.selected_location[0]:
  1927. for i in range(len(self.selected_location)):
  1928. if self.selected_location[i].data["name"] != genv["LOCATION_NAME"]:
  1929. currentLocation = False
  1930. currentMapset = False
  1931. break
  1932. if currentMapset and self.selected_mapset[0]:
  1933. for i in range(len(self.selected_mapset)):
  1934. if self.selected_mapset[i].data["name"] != genv["MAPSET"]:
  1935. currentMapset = False
  1936. break
  1937. return currentGrassDb, currentLocation, currentMapset
  1938. else:
  1939. return True, True, True
  1940. def _popupMenuLayer(self):
  1941. """Create popup menu for layers"""
  1942. menu = Menu()
  1943. genv = gisenv()
  1944. currentGrassDb, currentLocation, currentMapset = self._isCurrent(genv)
  1945. item = wx.MenuItem(menu, wx.ID_ANY, _("&Cut"))
  1946. menu.AppendItem(item)
  1947. self.Bind(wx.EVT_MENU, self.OnMoveMap, item)
  1948. if not currentMapset:
  1949. item.Enable(False)
  1950. item = wx.MenuItem(menu, wx.ID_ANY, _("&Copy"))
  1951. menu.AppendItem(item)
  1952. self.Bind(wx.EVT_MENU, self.OnCopyMap, item)
  1953. item = wx.MenuItem(menu, wx.ID_ANY, _("Copy &name"))
  1954. menu.AppendItem(item)
  1955. self.Bind(wx.EVT_MENU, self.OnCopyName, item)
  1956. item = wx.MenuItem(menu, wx.ID_ANY, _("&Paste"))
  1957. menu.AppendItem(item)
  1958. self.Bind(wx.EVT_MENU, self.OnPasteMap, item)
  1959. if not (currentMapset and self.copy_layer):
  1960. item.Enable(False)
  1961. item = wx.MenuItem(menu, wx.ID_ANY, _("&Delete"))
  1962. menu.AppendItem(item)
  1963. self.Bind(wx.EVT_MENU, self.OnDeleteMap, item)
  1964. item.Enable(currentMapset)
  1965. item = wx.MenuItem(menu, wx.ID_ANY, _("&Rename"))
  1966. menu.AppendItem(item)
  1967. self.Bind(wx.EVT_MENU, self.OnRenameMap, item)
  1968. item.Enable(currentMapset and len(self.selected_layer) == 1)
  1969. menu.AppendSeparator()
  1970. if not isinstance(self._giface, StandaloneGrassInterface):
  1971. if all(
  1972. [
  1973. each.data["name"] == genv["LOCATION_NAME"]
  1974. for each in self.selected_location
  1975. ]
  1976. ):
  1977. if len(self.selected_layer) > 1:
  1978. item = wx.MenuItem(menu, wx.ID_ANY, _("&Display layers"))
  1979. else:
  1980. item = wx.MenuItem(menu, wx.ID_ANY, _("&Display layer"))
  1981. menu.AppendItem(item)
  1982. self.Bind(wx.EVT_MENU, self.OnDisplayLayer, item)
  1983. item = wx.MenuItem(menu, wx.ID_ANY, _("Show &metadata"))
  1984. menu.AppendItem(item)
  1985. self.Bind(wx.EVT_MENU, self.OnMetadata, item)
  1986. self.PopupMenu(menu)
  1987. menu.Destroy()
  1988. def _popupMenuMapset(self):
  1989. """Create popup menu for mapsets"""
  1990. menu = Menu()
  1991. genv = gisenv()
  1992. currentGrassDb, currentLocation, currentMapset = self._isCurrent(genv)
  1993. item = wx.MenuItem(menu, wx.ID_ANY, _("&Paste"))
  1994. menu.AppendItem(item)
  1995. self.Bind(wx.EVT_MENU, self.OnPasteMap, item)
  1996. if not (currentMapset and self.copy_layer):
  1997. item.Enable(False)
  1998. item = wx.MenuItem(menu, wx.ID_ANY, _("&Switch mapset"))
  1999. menu.AppendItem(item)
  2000. self.Bind(wx.EVT_MENU, self.OnSwitchMapset, item)
  2001. if (
  2002. self.selected_grassdb[0].data["name"] == genv["GISDBASE"]
  2003. and self.selected_location[0].data["name"] == genv["LOCATION_NAME"]
  2004. and self.selected_mapset[0].data["name"] == genv["MAPSET"]
  2005. ):
  2006. item.Enable(False)
  2007. item = wx.MenuItem(menu, wx.ID_ANY, _("&Delete mapset"))
  2008. menu.AppendItem(item)
  2009. self.Bind(wx.EVT_MENU, self.OnDeleteMapset, item)
  2010. if self._restricted:
  2011. item.Enable(False)
  2012. item = wx.MenuItem(menu, wx.ID_ANY, _("&Rename mapset"))
  2013. menu.AppendItem(item)
  2014. self.Bind(wx.EVT_MENU, self.OnRenameMapset, item)
  2015. if self._restricted:
  2016. item.Enable(False)
  2017. item = wx.MenuItem(menu, wx.ID_ANY, _("Re&load maps"))
  2018. menu.AppendItem(item)
  2019. self.Bind(wx.EVT_MENU, self.OnReloadMapset, item)
  2020. self.PopupMenu(menu)
  2021. menu.Destroy()
  2022. def _popupMenuLocation(self):
  2023. """Create popup menu for locations"""
  2024. menu = Menu()
  2025. item = wx.MenuItem(menu, wx.ID_ANY, _("&Create mapset"))
  2026. menu.AppendItem(item)
  2027. self.Bind(wx.EVT_MENU, self.OnCreateMapset, item)
  2028. item = wx.MenuItem(menu, wx.ID_ANY, _("&Delete location"))
  2029. menu.AppendItem(item)
  2030. self.Bind(wx.EVT_MENU, self.OnDeleteLocation, item)
  2031. if self._restricted:
  2032. item.Enable(False)
  2033. item = wx.MenuItem(menu, wx.ID_ANY, _("&Rename location"))
  2034. menu.AppendItem(item)
  2035. self.Bind(wx.EVT_MENU, self.OnRenameLocation, item)
  2036. if self._restricted:
  2037. item.Enable(False)
  2038. item = wx.MenuItem(menu, wx.ID_ANY, _("Re&load maps"))
  2039. menu.AppendItem(item)
  2040. self.Bind(wx.EVT_MENU, self.OnReloadLocation, item)
  2041. self.PopupMenu(menu)
  2042. menu.Destroy()
  2043. def _popupMenuGrassDb(self):
  2044. """Create popup menu for grass db"""
  2045. menu = Menu()
  2046. genv = gisenv()
  2047. currentGrassDb, currentLocation, currentMapset = self._isCurrent(genv)
  2048. item = wx.MenuItem(menu, wx.ID_ANY, _("&Create new location"))
  2049. menu.AppendItem(item)
  2050. self.Bind(wx.EVT_MENU, self.OnCreateLocation, item)
  2051. item = wx.MenuItem(menu, wx.ID_ANY, _("&Download sample location"))
  2052. menu.AppendItem(item)
  2053. self.Bind(wx.EVT_MENU, self.OnDownloadLocation, item)
  2054. item = wx.MenuItem(
  2055. menu, wx.ID_ANY, _("&Remove GRASS database from data catalog")
  2056. )
  2057. menu.AppendItem(item)
  2058. self.Bind(wx.EVT_MENU, self.OnRemoveGrassDb, item)
  2059. if self.selected_grassdb[0].data["name"] == genv["GISDBASE"]:
  2060. item.Enable(False)
  2061. item = wx.MenuItem(menu, wx.ID_ANY, _("&Delete GRASS database from disk"))
  2062. menu.AppendItem(item)
  2063. self.Bind(wx.EVT_MENU, self.OnDeleteGrassDb, item)
  2064. if self._restricted:
  2065. item.Enable(False)
  2066. item = wx.MenuItem(menu, wx.ID_ANY, _("Re&load maps"))
  2067. menu.AppendItem(item)
  2068. self.Bind(wx.EVT_MENU, self.OnReloadGrassdb, item)
  2069. self.PopupMenu(menu)
  2070. menu.Destroy()
  2071. def _popupMenuElement(self):
  2072. """Create popup menu for elements"""
  2073. menu = Menu()
  2074. item = wx.MenuItem(menu, wx.ID_ANY, _("&Paste"))
  2075. menu.AppendItem(item)
  2076. self.Bind(wx.EVT_MENU, self.OnPasteMap, item)
  2077. genv = gisenv()
  2078. currentGrassDb, currentLocation, currentMapset = self._isCurrent(genv)
  2079. if not (currentMapset and self.copy_layer):
  2080. item.Enable(False)
  2081. self.PopupMenu(menu)
  2082. menu.Destroy()
  2083. def _popupMenuMultipleLocations(self):
  2084. """Create popup menu for multiple selected locations"""
  2085. menu = Menu()
  2086. item = wx.MenuItem(menu, wx.ID_ANY, _("&Delete locations"))
  2087. menu.AppendItem(item)
  2088. self.Bind(wx.EVT_MENU, self.OnDeleteLocation, item)
  2089. if self._restricted:
  2090. item.Enable(False)
  2091. self.PopupMenu(menu)
  2092. menu.Destroy()
  2093. def _popupMenuMultipleMapsets(self):
  2094. """Create popup menu for multiple selected mapsets"""
  2095. menu = Menu()
  2096. item = wx.MenuItem(menu, wx.ID_ANY, _("&Delete mapsets"))
  2097. menu.AppendItem(item)
  2098. self.Bind(wx.EVT_MENU, self.OnDeleteMapset, item)
  2099. if self._restricted:
  2100. item.Enable(False)
  2101. self.PopupMenu(menu)
  2102. menu.Destroy()
  2103. def _popupMenuEmpty(self):
  2104. """Create empty popup when multiple different types of items are selected"""
  2105. menu = Menu()
  2106. item = wx.MenuItem(menu, wx.ID_ANY, _("No available options"))
  2107. menu.AppendItem(item)
  2108. item.Enable(False)
  2109. self.PopupMenu(menu)
  2110. menu.Destroy()