tree.py 63 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576
  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. import wx
  21. from core.gcmd import RunCommand, GError, GMessage, GWarning
  22. from core.utils import GetListOfLocations
  23. from core.debug import Debug
  24. from gui_core.dialogs import TextEntryDialog
  25. from core.giface import StandaloneGrassInterface
  26. from core.treemodel import TreeModel, DictNode
  27. from gui_core.treeview import TreeView
  28. from gui_core.wrap import Menu
  29. from datacatalog.dialogs import CatalogReprojectionDialog
  30. from icons.icon import MetaIcon
  31. from core.settings import UserSettings
  32. from startup.guiutils import (
  33. create_mapset_interactively,
  34. create_location_interactively,
  35. rename_mapset_interactively,
  36. rename_location_interactively,
  37. delete_mapsets_interactively,
  38. delete_locations_interactively,
  39. download_location_interactively,
  40. delete_grassdb_interactively
  41. )
  42. from grass.pydispatch.signal import Signal
  43. import grass.script as gscript
  44. from grass.script import gisenv
  45. from grass.grassdb.data import map_exists
  46. from grass.grassdb.checks import (get_mapset_owner, is_mapset_locked,
  47. is_different_mapset_owner)
  48. from grass.exceptions import CalledModuleError
  49. def filterModel(model, element=None, name=None):
  50. """Filter tree model based on type or name of map using regular expressions.
  51. Copies tree and remove nodes which don't match."""
  52. fmodel = copy.deepcopy(model)
  53. nodesToRemove = []
  54. if name:
  55. try:
  56. regex = re.compile(name)
  57. except:
  58. return fmodel
  59. for gisdbase in fmodel.root.children:
  60. for location in gisdbase.children:
  61. for mapset in location.children:
  62. for layer in mapset.children:
  63. if element and layer.data['type'] != element:
  64. nodesToRemove.append(layer)
  65. continue
  66. if name and regex.search(layer.data['name']) is None:
  67. nodesToRemove.append(layer)
  68. for node in reversed(nodesToRemove):
  69. fmodel.RemoveNode(node)
  70. cleanUpTree(fmodel)
  71. return fmodel
  72. def cleanUpTree(model):
  73. """Removes empty element/mapsets/locations nodes.
  74. It first removes empty elements, then mapsets, then locations"""
  75. # removes empty mapsets
  76. nodesToRemove = []
  77. for gisdbase in model.root.children:
  78. for location in gisdbase.children:
  79. for mapset in location.children:
  80. if not mapset.children:
  81. nodesToRemove.append(mapset)
  82. for node in reversed(nodesToRemove):
  83. model.RemoveNode(node)
  84. # removes empty locations
  85. nodesToRemove = []
  86. for gisdbase in model.root.children:
  87. for location in gisdbase.children:
  88. if not location.children:
  89. nodesToRemove.append(location)
  90. for node in reversed(nodesToRemove):
  91. model.RemoveNode(node)
  92. def getLocationTree(gisdbase, location, queue, mapsets=None):
  93. """Creates dictionary with mapsets, elements, layers for given location.
  94. Returns tuple with the dictionary and error (or None)"""
  95. tmp_gisrc_file, env = gscript.create_environment(gisdbase, location, 'PERMANENT')
  96. env['GRASS_SKIP_MAPSET_OWNER_CHECK'] = '1'
  97. maps_dict = {}
  98. elements = ['raster', 'raster_3d', 'vector']
  99. try:
  100. if not mapsets:
  101. mapsets = gscript.read_command(
  102. 'g.mapsets',
  103. flags='l',
  104. separator='comma',
  105. quiet=True,
  106. env=env).strip()
  107. except CalledModuleError:
  108. queue.put(
  109. (maps_dict,
  110. _("Failed to read mapsets from location <{l}>.").format(
  111. l=location)))
  112. gscript.try_remove(tmp_gisrc_file)
  113. return
  114. else:
  115. mapsets = mapsets.split(',')
  116. Debug.msg(
  117. 4, "Location <{0}>: {1} mapsets found".format(
  118. location, len(mapsets)))
  119. for each in mapsets:
  120. maps_dict[each] = []
  121. try:
  122. maplist = gscript.read_command(
  123. 'g.list', flags='mt', type=elements,
  124. mapset=','.join(mapsets),
  125. quiet=True, env=env).strip()
  126. except CalledModuleError:
  127. queue.put(
  128. (maps_dict,
  129. _("Failed to read maps from location <{l}>.").format(
  130. l=location)))
  131. gscript.try_remove(tmp_gisrc_file)
  132. return
  133. else:
  134. # fill dictionary
  135. listOfMaps = maplist.splitlines()
  136. Debug.msg(
  137. 4, "Location <{0}>: {1} maps found".format(
  138. location, len(listOfMaps)))
  139. for each in listOfMaps:
  140. ltype, wholename = each.split('/')
  141. name, mapset = wholename.split('@')
  142. maps_dict[mapset].append({'name': name, 'type': ltype})
  143. queue.put((maps_dict, None))
  144. gscript.try_remove(tmp_gisrc_file)
  145. class NameEntryDialog(TextEntryDialog):
  146. def __init__(self, element, mapset, env, **kwargs):
  147. TextEntryDialog.__init__(self, **kwargs)
  148. self._element = element
  149. self._mapset = mapset
  150. self._env = env
  151. id_OK = self.GetAffirmativeId()
  152. self.Bind(wx.EVT_BUTTON, self.OnOK, self.FindWindowById(id_OK))
  153. def OnOK(self, event):
  154. new = self.GetValue()
  155. if not new:
  156. return
  157. if map_exists(new, self._element, env=self._env, mapset=self._mapset):
  158. dlg = wx.MessageDialog(
  159. self,
  160. message=_(
  161. "Map of type {elem} <{name}> already exists in mapset <{mapset}>. "
  162. "Do you want to overwrite it?").format(
  163. elem=self._element,
  164. name=new,
  165. mapset=self._mapset),
  166. caption=_("Overwrite?"),
  167. style=wx.YES_NO)
  168. if dlg.ShowModal() == wx.ID_YES:
  169. dlg.Destroy()
  170. self._env['GRASS_OVERWRITE'] = '1'
  171. self.EndModal(wx.ID_OK)
  172. else:
  173. dlg.Destroy()
  174. return
  175. else:
  176. self.EndModal(wx.ID_OK)
  177. class DataCatalogNode(DictNode):
  178. """Node representing item in datacatalog."""
  179. def __init__(self, data=None):
  180. super(DataCatalogNode, self).__init__(data=data)
  181. @property
  182. def label(self):
  183. data = self.data
  184. if data['type'] == 'mapset':
  185. owner = data['owner'] if data['owner'] else _("name unknown")
  186. if data['current']:
  187. return _("{name} (current)").format(**data)
  188. elif data['is_different_owner'] and data['lock']:
  189. return _("{name} (in use, owner: {owner})").format(
  190. name=data["name"], owner=owner
  191. )
  192. elif data['lock']:
  193. return _("{name} (in use)").format(**data)
  194. elif data['is_different_owner']:
  195. return _("{name} (owner: {owner})").format(name=data["name"],
  196. owner=owner)
  197. return _("{name}").format(**data)
  198. def match(self, **kwargs):
  199. """Method used for searching according to given parameters.
  200. :param value: dictionary value to be matched
  201. :param key: data dictionary key
  202. """
  203. if not kwargs:
  204. return False
  205. for key in kwargs:
  206. if not (key in self.data and self.data[key] == kwargs[key]):
  207. return False
  208. return True
  209. class DataCatalogTree(TreeView):
  210. def __init__(
  211. self, parent, model=None, giface=None,
  212. style=wx.TR_HIDE_ROOT | wx.TR_EDIT_LABELS |
  213. wx.TR_LINES_AT_ROOT | wx.TR_HAS_BUTTONS |
  214. wx.TR_FULL_ROW_HIGHLIGHT | wx.TR_MULTIPLE):
  215. """Location Map Tree constructor."""
  216. self._model = TreeModel(DataCatalogNode)
  217. self._orig_model = self._model
  218. super(
  219. DataCatalogTree,
  220. self).__init__(
  221. parent=parent,
  222. model=self._model,
  223. id=wx.ID_ANY,
  224. style=style)
  225. self._giface = giface
  226. self._restricted = True
  227. self.showNotification = Signal('Tree.showNotification')
  228. self.changeMapset = Signal('Tree.changeMapset')
  229. self.changeLocation = Signal('Tree.changeLocation')
  230. self.parent = parent
  231. self.contextMenu.connect(self.OnRightClick)
  232. self.itemActivated.connect(self.OnDoubleClick)
  233. self._iconTypes = ['grassdb', 'location', 'mapset', 'raster',
  234. 'vector', 'raster_3d']
  235. self._initImages()
  236. self._resetSelectVariables()
  237. self._resetCopyVariables()
  238. self.current_grassdb_node = None
  239. self.current_location_node = None
  240. self.current_mapset_node = None
  241. self.UpdateCurrentDbLocationMapsetNode()
  242. # Get databases from settings
  243. # add current to settings if it's not included
  244. self.grassdatabases = self._getValidSavedGrassDBs()
  245. currentDB = gisenv()['GISDBASE']
  246. if currentDB not in self.grassdatabases:
  247. self.grassdatabases.append(currentDB)
  248. self._saveGrassDBs()
  249. self.beginDrag = Signal('DataCatalogTree.beginDrag')
  250. self.endDrag = Signal('DataCatalogTree.endDrag')
  251. self.startEdit = Signal('DataCatalogTree.startEdit')
  252. self.endEdit = Signal('DataCatalogTree.endEdit')
  253. self.Bind(wx.EVT_TREE_BEGIN_DRAG, lambda evt:
  254. self._emitSignal(evt.GetItem(), self.beginDrag, event=evt))
  255. self.Bind(wx.EVT_TREE_END_DRAG, lambda evt:
  256. self._emitSignal(evt.GetItem(), self.endDrag, event=evt))
  257. self.beginDrag.connect(self.OnBeginDrag)
  258. self.endDrag.connect(self.OnEndDrag)
  259. self.Bind(wx.EVT_TREE_BEGIN_LABEL_EDIT, lambda evt:
  260. self._emitSignal(evt.GetItem(), self.startEdit, event=evt))
  261. self.Bind(wx.EVT_TREE_END_LABEL_EDIT, lambda evt:
  262. self._emitSignal(evt.GetItem(), self.endEdit, event=evt))
  263. self.startEdit.connect(self.OnStartEditLabel)
  264. self.endEdit.connect(self.OnEditLabel)
  265. def _resetSelectVariables(self):
  266. """Reset variables related to item selection."""
  267. self.selected_grassdb = []
  268. self.selected_layer = []
  269. self.selected_mapset = []
  270. self.selected_location = []
  271. self.mixed = False
  272. def _resetCopyVariables(self):
  273. """Reset copy related variables."""
  274. self.copy_mode = False
  275. self.copy_layer = None
  276. self.copy_mapset = None
  277. self.copy_location = None
  278. self.copy_grassdb = None
  279. def _getValidSavedGrassDBs(self):
  280. """Returns list of GRASS databases from settings.
  281. Returns only existing directories."""
  282. dbs = UserSettings.Get(group='datacatalog',
  283. key='grassdbs',
  284. subkey='listAsString')
  285. dbs = [db for db in dbs.split(',') if os.path.isdir(db)]
  286. return dbs
  287. def _saveGrassDBs(self):
  288. """Save current grass dbs in tree to settings"""
  289. UserSettings.Set(group='datacatalog',
  290. key='grassdbs',
  291. subkey='listAsString',
  292. value=",".join(self.grassdatabases))
  293. grassdbSettings = {}
  294. UserSettings.ReadSettingsFile(settings=grassdbSettings)
  295. if 'datacatalog' not in grassdbSettings:
  296. grassdbSettings['datacatalog'] = UserSettings.Get(group='datacatalog')
  297. # update only dbs
  298. grassdbSettings['datacatalog']['grassdbs'] = UserSettings.Get(group='datacatalog', key='grassdbs')
  299. UserSettings.SaveToFile(grassdbSettings)
  300. def _reloadMapsetNode(self, mapset_node):
  301. """Recursively reload the model of a specific mapset node"""
  302. if mapset_node.children:
  303. del mapset_node.children[:]
  304. q = Queue()
  305. p = Process(
  306. target=getLocationTree,
  307. args=(
  308. mapset_node.parent.parent.data['name'],
  309. mapset_node.parent.data['name'],
  310. q,
  311. mapset_node.data['name']))
  312. p.start()
  313. maps, error = q.get()
  314. self._populateMapsetItem(mapset_node,
  315. maps[mapset_node.data['name']])
  316. self._orig_model = copy.deepcopy(self._model)
  317. return error
  318. def _reloadLocationNode(self, location_node):
  319. """Recursively reload the model of a specific location node"""
  320. if location_node.children:
  321. del location_node.children[:]
  322. q = Queue()
  323. p = Process(
  324. target=getLocationTree,
  325. args=(
  326. location_node.parent.data['name'],
  327. location_node.data['name'],
  328. q,
  329. None))
  330. p.start()
  331. maps, error = q.get()
  332. for mapset in maps:
  333. mapset_path = os.path.join(location_node.parent.data['name'],
  334. location_node.data['name'],
  335. mapset)
  336. mapset_node = self._model.AppendNode(
  337. parent=location_node,
  338. data=dict(type='mapset',
  339. name=mapset,
  340. current=False,
  341. lock=is_mapset_locked(mapset_path),
  342. is_different_owner=is_different_mapset_owner(mapset_path),
  343. owner=get_mapset_owner(mapset_path)))
  344. self._populateMapsetItem(mapset_node,
  345. maps[mapset])
  346. self._model.SortChildren(location_node)
  347. self._orig_model = copy.deepcopy(self._model)
  348. return error
  349. def _reloadGrassDBNode(self, grassdb_node):
  350. """Recursively reload the model of a specific grassdb node.
  351. Runs reloading locations in parallel."""
  352. if grassdb_node.children:
  353. del grassdb_node.children[:]
  354. locations = GetListOfLocations(grassdb_node.data['name'])
  355. loc_count = proc_count = 0
  356. queue_list = []
  357. proc_list = []
  358. loc_list = []
  359. try:
  360. nprocs = cpu_count()
  361. except NotImplementedError:
  362. nprocs = 4
  363. results = dict()
  364. errors = []
  365. location_nodes = []
  366. all_location_nodes = []
  367. nlocations = len(locations)
  368. for location in locations:
  369. results[location] = dict()
  370. varloc = self._model.AppendNode(parent=grassdb_node,
  371. data=dict(type='location',
  372. name=location))
  373. location_nodes.append(varloc)
  374. all_location_nodes.append(varloc)
  375. loc_count += 1
  376. Debug.msg(
  377. 3, "Scanning location <{0}> ({1}/{2})".format(location, loc_count, nlocations))
  378. q = Queue()
  379. p = Process(target=getLocationTree,
  380. args=(grassdb_node.data['name'], location, q))
  381. p.start()
  382. queue_list.append(q)
  383. proc_list.append(p)
  384. loc_list.append(location)
  385. proc_count += 1
  386. # Wait for all running processes
  387. if proc_count == nprocs or loc_count == nlocations:
  388. Debug.msg(4, "Process subresults")
  389. for i in range(len(loc_list)):
  390. maps, error = queue_list[i].get()
  391. proc_list[i].join()
  392. if error:
  393. errors.append(error)
  394. for key in sorted(maps.keys()):
  395. mapset_path = os.path.join(location_nodes[i].parent.data['name'],
  396. location_nodes[i].data['name'],
  397. key)
  398. mapset_node = self._model.AppendNode(
  399. parent=location_nodes[i],
  400. data=dict(type='mapset',
  401. name=key,
  402. lock=is_mapset_locked(mapset_path),
  403. current=False,
  404. is_different_owner=is_different_mapset_owner(mapset_path),
  405. owner=get_mapset_owner(mapset_path)))
  406. self._populateMapsetItem(mapset_node, maps[key])
  407. proc_count = 0
  408. proc_list = []
  409. queue_list = []
  410. loc_list = []
  411. location_nodes = []
  412. for node in all_location_nodes:
  413. self._model.SortChildren(node)
  414. self._model.SortChildren(grassdb_node)
  415. self._orig_model = copy.deepcopy(self._model)
  416. return errors
  417. def _reloadTreeItems(self):
  418. """Updates grass databases, locations, mapsets and layers in the tree.
  419. Saves resulting data and error."""
  420. errors = []
  421. for grassdatabase in self.grassdatabases:
  422. grassdb_nodes = self._model.SearchNodes(name=grassdatabase,
  423. type='grassdb')
  424. if not grassdb_nodes:
  425. grassdb_node = self._model.AppendNode(parent=self._model.root,
  426. data=dict(type='grassdb',
  427. name=grassdatabase))
  428. else:
  429. grassdb_node = grassdb_nodes[0]
  430. error = self._reloadGrassDBNode(grassdb_node)
  431. if error:
  432. errors.append(error)
  433. if errors:
  434. wx.CallAfter(GWarning, '\n'.join(errors))
  435. Debug.msg(1, "Tree filled")
  436. self.UpdateCurrentDbLocationMapsetNode()
  437. self.RefreshItems()
  438. def _renameNode(self, node, name):
  439. """Rename node (map, mapset, location), sort and refresh.
  440. Should be called after actual renaming of a map, mapset, location."""
  441. node.data['name'] = name
  442. self._model.SortChildren(node.parent)
  443. self.RefreshNode(node.parent, recursive=True)
  444. def UpdateCurrentDbLocationMapsetNode(self):
  445. """Update variables storing current mapset/location/grassdb node.
  446. Updates associated mapset node data ('lock' and 'current').
  447. """
  448. def is_current_mapset_node_locked():
  449. mapset_path = os.path.join(self.current_grassdb_node.data['name'],
  450. self.current_location_node.data['name'],
  451. self.current_mapset_node.data["name"])
  452. return is_mapset_locked(mapset_path)
  453. if self.current_mapset_node:
  454. self.current_mapset_node.data["current"] = False
  455. self.current_mapset_node.data["lock"] = is_current_mapset_node_locked()
  456. self.current_grassdb_node, self.current_location_node, self.current_mapset_node = \
  457. self.GetCurrentDbLocationMapsetNode()
  458. if self.current_mapset_node:
  459. self.current_mapset_node.data["current"] = True
  460. self.current_mapset_node.data["lock"] = is_current_mapset_node_locked()
  461. def ReloadTreeItems(self):
  462. """Reload dbs, locations, mapsets and layers in the tree."""
  463. self._reloadTreeItems()
  464. def ReloadCurrentMapset(self):
  465. """Reload current mapset tree only."""
  466. self.UpdateCurrentDbLocationMapsetNode()
  467. if not self.current_grassdb_node or not self.current_location_node or not self.current_mapset_node:
  468. return
  469. self._reloadMapsetNode(self.current_mapset_node)
  470. self.RefreshNode(self.current_mapset_node, recursive=True)
  471. def _populateMapsetItem(self, mapset_node, data):
  472. for item in data:
  473. self._model.AppendNode(parent=mapset_node,
  474. data=dict(**item))
  475. self._model.SortChildren(mapset_node)
  476. def _initImages(self):
  477. bmpsize = (16, 16)
  478. icons = {
  479. 'grassdb': MetaIcon(img='grassdb').GetBitmap(bmpsize),
  480. 'location': MetaIcon(img='location').GetBitmap(bmpsize),
  481. 'mapset': MetaIcon(img='mapset').GetBitmap(bmpsize),
  482. 'raster': MetaIcon(img='raster').GetBitmap(bmpsize),
  483. 'vector': MetaIcon(img='vector').GetBitmap(bmpsize),
  484. 'raster_3d': MetaIcon(img='raster3d').GetBitmap(bmpsize)
  485. }
  486. il = wx.ImageList(bmpsize[0], bmpsize[1], mask=False)
  487. for each in self._iconTypes:
  488. il.Add(icons[each])
  489. self.AssignImageList(il)
  490. def GetControl(self):
  491. """Returns control itself."""
  492. return self
  493. def DefineItems(self, selected):
  494. """Set selected items."""
  495. self._resetSelectVariables()
  496. mixed = []
  497. for item in selected:
  498. type = item.data['type']
  499. if type in ('raster', 'raster_3d', 'vector'):
  500. self.selected_layer.append(item)
  501. self.selected_mapset.append(item.parent)
  502. self.selected_location.append(item.parent.parent)
  503. self.selected_grassdb.append(item.parent.parent.parent)
  504. mixed.append('layer')
  505. elif type == 'mapset':
  506. self.selected_layer.append(None)
  507. self.selected_mapset.append(item)
  508. self.selected_location.append(item.parent)
  509. self.selected_grassdb.append(item.parent.parent)
  510. mixed.append('mapset')
  511. elif type == 'location':
  512. self.selected_layer.append(None)
  513. self.selected_mapset.append(None)
  514. self.selected_location.append(item)
  515. self.selected_grassdb.append(item.parent)
  516. mixed.append('location')
  517. elif type == 'grassdb':
  518. self.selected_layer.append(None)
  519. self.selected_mapset.append(None)
  520. self.selected_location.append(None)
  521. self.selected_grassdb.append(item)
  522. mixed.append('grassdb')
  523. self.mixed = False
  524. if len(set(mixed)) > 1:
  525. self.mixed = True
  526. def OnSelChanged(self, event):
  527. self.selected_layer = None
  528. def OnRightClick(self, node):
  529. """Display popup menu."""
  530. self.DefineItems(self.GetSelected())
  531. if self.mixed:
  532. self._popupMenuEmpty()
  533. return
  534. if not self.selected_layer:
  535. self._popupMenuEmpty()
  536. elif self.selected_layer[0]:
  537. self._popupMenuLayer()
  538. elif self.selected_mapset[0] and len(self.selected_mapset) == 1:
  539. self._popupMenuMapset()
  540. elif self.selected_location[0] and not self.selected_mapset[0] and len(self.selected_location) == 1:
  541. self._popupMenuLocation()
  542. elif self.selected_grassdb[0] and not self.selected_location[0] and len(self.selected_grassdb) == 1:
  543. self._popupMenuGrassDb()
  544. elif len(self.selected_location) > 1 and not self.selected_mapset[0]:
  545. self._popupMenuMultipleLocations()
  546. elif len(self.selected_mapset) > 1:
  547. self._popupMenuMultipleMapsets()
  548. else:
  549. self._popupMenuEmpty()
  550. def OnDoubleClick(self, node):
  551. """Double click on item/node.
  552. Display selected layer if node is a map layer otherwise
  553. expand/collapse node.
  554. """
  555. if not isinstance(self._giface, StandaloneGrassInterface):
  556. self.DefineItems([node])
  557. selected_layer = self.selected_layer[0]
  558. selected_mapset = self.selected_mapset[0]
  559. selected_loc = self.selected_location[0]
  560. if selected_layer is not None:
  561. genv = gisenv()
  562. # Check if the layer is in different location
  563. if selected_loc.data['name'] != genv['LOCATION_NAME']:
  564. dlg = wx.MessageDialog(
  565. parent=self,
  566. message=_(
  567. "Map <{0}@{1}> is not in the current location"
  568. " and therefore cannot be displayed."
  569. "\n\n"
  570. "To display this map switch to mapset <{1}> first."
  571. ).format(selected_layer.data['name'],
  572. selected_mapset.data['name']),
  573. caption=_("Unable to display the map"),
  574. style=wx.OK | wx.ICON_WARNING
  575. )
  576. dlg.ShowModal()
  577. dlg.Destroy()
  578. else:
  579. self.DisplayLayer()
  580. return
  581. # expand/collapse location/mapset...
  582. if self.IsNodeExpanded(node):
  583. self.CollapseNode(node, recursive=False)
  584. else:
  585. self.ExpandNode(node, recursive=False)
  586. def ExpandCurrentLocation(self):
  587. """Expand current location"""
  588. location = gscript.gisenv()['LOCATION_NAME']
  589. item = self._model.SearchNodes(name=location, type='location')
  590. if item:
  591. self.Select(item[0], select=True)
  592. self.ExpandNode(item[0], recursive=False)
  593. else:
  594. Debug.msg(1, "Location <%s> not found" % location)
  595. def GetCurrentDbLocationMapsetNode(self):
  596. """Get current mapset node"""
  597. genv = gisenv()
  598. gisdbase = genv['GISDBASE']
  599. location = genv['LOCATION_NAME']
  600. mapset = genv['MAPSET']
  601. grassdbItem = self._model.SearchNodes(
  602. name=gisdbase, type='grassdb')
  603. if not grassdbItem:
  604. return None, None, None
  605. locationItem = self._model.SearchNodes(
  606. parent=grassdbItem[0],
  607. name=location, type='location')
  608. if not locationItem:
  609. return grassdbItem[0], None, None
  610. mapsetItem = self._model.SearchNodes(
  611. parent=locationItem[0],
  612. name=mapset,
  613. type='mapset')
  614. if not mapsetItem:
  615. return grassdbItem[0], locationItem[0], None
  616. return grassdbItem[0], locationItem[0], mapsetItem[0]
  617. def OnGetItemImage(self, index, which=wx.TreeItemIcon_Normal, column=0):
  618. """Overriden method to return image for each item."""
  619. node = self._model.GetNodeByIndex(index)
  620. try:
  621. return self._iconTypes.index(node.data['type'])
  622. except ValueError:
  623. return 0
  624. def OnGetItemTextColour(self, index):
  625. """Overriden method to return colour for each item.
  626. Used to distinquish lock and ownership on mapsets."""
  627. node = self._model.GetNodeByIndex(index)
  628. if node.data['type'] == 'mapset':
  629. if node.data['current']:
  630. return wx.SystemSettings.GetColour(wx.SYS_COLOUR_WINDOWTEXT)
  631. elif node.data['lock'] or node.data['is_different_owner']:
  632. return wx.SystemSettings.GetColour(wx.SYS_COLOUR_GRAYTEXT)
  633. return wx.SystemSettings.GetColour(wx.SYS_COLOUR_WINDOWTEXT)
  634. def OnGetItemFont(self, index):
  635. """Overriden method to return font for each item.
  636. Used to highlight current db/loc/mapset."""
  637. node = self._model.GetNodeByIndex(index)
  638. font = self.GetFont()
  639. if node.data['type'] in ('grassdb', 'location', 'mapset'):
  640. if node in (self.current_grassdb_node, self.current_location_node, self.current_mapset_node):
  641. font.SetWeight(wx.FONTWEIGHT_BOLD)
  642. else:
  643. font.SetWeight(wx.FONTWEIGHT_NORMAL)
  644. return font
  645. def ExpandCurrentMapset(self):
  646. """Expand current mapset"""
  647. if self.current_mapset_node:
  648. self.Select(self.current_mapset_node, select=True)
  649. self.ExpandNode(self.current_mapset_node, recursive=True)
  650. def SetRestriction(self, restrict):
  651. self._restricted = restrict
  652. def _runCommand(self, prog, **kwargs):
  653. cmdString = ' '.join(gscript.make_command(prog, **kwargs))
  654. ret = RunCommand(prog, parent=self, **kwargs)
  655. return ret, cmdString
  656. def OnMoveMap(self, event):
  657. """Move layer or mapset (just save it temporarily, copying is done by paste)"""
  658. self.copy_mode = False
  659. self.copy_layer = self.selected_layer[:]
  660. self.copy_mapset = self.selected_mapset[:]
  661. self.copy_location = self.selected_location[:]
  662. self.copy_grassdb = self.selected_grassdb[:]
  663. if len(self.copy_layer) > 1:
  664. label = _("{c} maps marked for moving.").format(c=len(self.selected_layer))
  665. else:
  666. label = _("Map <{layer}> marked for moving.").format(layer=self.copy_layer[0].data['name'])
  667. self.showNotification.emit(message=label)
  668. def OnCopyMap(self, event):
  669. """Copy layer or mapset (just save it temporarily, copying is done by paste)"""
  670. self.copy_mode = True
  671. self.copy_layer = self.selected_layer[:]
  672. self.copy_mapset = self.selected_mapset[:]
  673. self.copy_location = self.selected_location[:]
  674. self.copy_grassdb = self.selected_grassdb[:]
  675. if len(self.copy_layer) > 1:
  676. label = _("{c} maps marked for copying.").format(c=len(self.selected_layer))
  677. else:
  678. label = _("Map <{layer}> marked for copying.").format(layer=self.copy_layer[0].data['name'])
  679. self.showNotification.emit(message=label)
  680. def OnRenameMap(self, event):
  681. """Rename layer with dialog"""
  682. old_name = self.selected_layer[0].data['name']
  683. gisrc, env = gscript.create_environment(
  684. self.selected_grassdb[0].data['name'],
  685. self.selected_location[0].data['name'],
  686. self.selected_mapset[0].data['name'])
  687. new_name = self._getNewMapName(
  688. _('New name'),
  689. _('Rename map'),
  690. old_name,
  691. env=env,
  692. mapset=self.selected_mapset[0].data['name'],
  693. element=self.selected_layer[0].data['type'])
  694. if new_name:
  695. self.Rename(old_name, new_name)
  696. def CreateMapset(self, grassdb_node, location_node):
  697. """Creates new mapset interactively and adds it to the tree."""
  698. mapset = create_mapset_interactively(self, grassdb_node.data['name'],
  699. location_node.data['name'])
  700. if mapset:
  701. self.InsertMapset(name=mapset,
  702. location_node=location_node)
  703. def OnCreateMapset(self, event):
  704. """Create new mapset"""
  705. self.CreateMapset(self.selected_grassdb[0], self.selected_location[0])
  706. def CreateLocation(self, grassdb_node):
  707. """
  708. Creates new location interactively and adds it to the tree.
  709. """
  710. grassdatabase, location, mapset = (
  711. create_location_interactively(self, grassdb_node.data['name'])
  712. )
  713. if location:
  714. grassdb_nodes = self._model.SearchNodes(name=grassdatabase, type='grassdb')
  715. if not grassdb_nodes:
  716. grassdb_node = self.InsertGrassDb(name=grassdatabase)
  717. self.InsertLocation(location, grassdb_node)
  718. def OnCreateLocation(self, event):
  719. """Create new location"""
  720. self.CreateLocation(self.selected_grassdb[0])
  721. def OnRenameMapset(self, event):
  722. """
  723. Rename selected mapset
  724. """
  725. newmapset = rename_mapset_interactively(
  726. self,
  727. self.selected_grassdb[0].data['name'],
  728. self.selected_location[0].data['name'],
  729. self.selected_mapset[0].data['name'])
  730. if newmapset:
  731. self._renameNode(self.selected_mapset[0], newmapset)
  732. def OnRenameLocation(self, event):
  733. """
  734. Rename selected location
  735. """
  736. newlocation = rename_location_interactively(
  737. self,
  738. self.selected_grassdb[0].data['name'],
  739. self.selected_location[0].data['name'])
  740. if newlocation:
  741. self._renameNode(self.selected_location[0], newlocation)
  742. def OnStartEditLabel(self, node, event):
  743. """Start label editing"""
  744. self.DefineItems([node])
  745. # TODO: add renaming mapset/location
  746. if not self.selected_layer[0]:
  747. event.Veto()
  748. return
  749. Debug.msg(1, "Start label edit {name}".format(name=node.data['name']))
  750. label = _("Editing {name}").format(name=node.data['name'])
  751. self.showNotification.emit(message=label)
  752. def OnEditLabel(self, node, event):
  753. """End label editing"""
  754. if self.selected_layer and not event.IsEditCancelled():
  755. old_name = node.data['name']
  756. Debug.msg(1, "End label edit {name}".format(name=old_name))
  757. new_name = event.GetLabel()
  758. self.Rename(old_name, new_name)
  759. def Rename(self, old, new):
  760. """Rename layer"""
  761. string = old + ',' + new
  762. gisrc, env = gscript.create_environment(
  763. self.selected_grassdb[0].data['name'],
  764. self.selected_location[0].data['name'],
  765. self.selected_mapset[0].data['name'])
  766. label = _("Renaming map <{name}>...").format(name=string)
  767. self.showNotification.emit(message=label)
  768. if self.selected_layer[0].data['type'] == 'vector':
  769. renamed, cmd = self._runCommand('g.rename', vector=string, env=env)
  770. elif self.selected_layer[0].data['type'] == 'raster':
  771. renamed, cmd = self._runCommand('g.rename', raster=string, env=env)
  772. else:
  773. renamed, cmd = self._runCommand(
  774. 'g.rename', raster3d=string, env=env)
  775. if renamed == 0:
  776. self._renameNode(self.selected_layer[0], new)
  777. self.showNotification.emit(
  778. message=_("{cmd} -- completed").format(cmd=cmd))
  779. Debug.msg(1, "LAYER RENAMED TO: " + new)
  780. gscript.try_remove(gisrc)
  781. def OnPasteMap(self, event):
  782. # copying between mapsets of one location
  783. if not self.copy_layer:
  784. if self.copy_mode:
  785. GMessage(_("No map selected for copying."), parent=self)
  786. else:
  787. GMessage(_("No map selected for moving."), parent=self)
  788. return
  789. for i in range(len(self.copy_layer)):
  790. gisrc, env = gscript.create_environment(self.selected_grassdb[0].data['name'],
  791. self.selected_location[0].data['name'],
  792. self.selected_mapset[0].data['name'])
  793. gisrc2, env2 = gscript.create_environment(self.copy_grassdb[i].data['name'],
  794. self.copy_location[i].data['name'],
  795. self.copy_mapset[i].data['name'])
  796. new_name = self.copy_layer[i].data['name']
  797. if self.selected_location[0] == self.copy_location[i]:
  798. # within one mapset
  799. if self.selected_mapset[0] == self.copy_mapset[i]:
  800. # ignore when just moves map
  801. if self.copy_mode is False:
  802. return
  803. new_name = self._getNewMapName(_('New name for <{n}>').format(n=self.copy_layer[i].data['name']),
  804. _('Select new name'),
  805. self.copy_layer[i].data['name'], env=env,
  806. mapset=self.selected_mapset[0].data['name'],
  807. element=self.copy_layer[i].data['type'])
  808. if not new_name:
  809. return
  810. # within one location, different mapsets
  811. else:
  812. if map_exists(new_name, element=self.copy_layer[i].data['type'], env=env,
  813. mapset=self.selected_mapset[0].data['name']):
  814. new_name = self._getNewMapName(_('New name for <{n}>').format(n=self.copy_layer[i].data['name']),
  815. _('Select new name'),
  816. self.copy_layer[i].data['name'], env=env,
  817. mapset=self.selected_mapset[0].data['name'],
  818. element=self.copy_layer[i].data['type'])
  819. if not new_name:
  820. return
  821. string = self.copy_layer[i].data['name'] + '@' + self.copy_mapset[i].data['name'] + ',' + new_name
  822. pasted = 0
  823. if self.copy_mode:
  824. label = _("Copying <{name}>...").format(name=string)
  825. else:
  826. label = _("Moving <{name}>...").format(name=string)
  827. self.showNotification.emit(message=label)
  828. if self.copy_layer[i].data['type'] == 'vector':
  829. pasted, cmd = self._runCommand('g.copy', vector=string, env=env)
  830. node = 'vector'
  831. elif self.copy_layer[i].data['type'] == 'raster':
  832. pasted, cmd = self._runCommand('g.copy', raster=string, env=env)
  833. node = 'raster'
  834. else:
  835. pasted, cmd = self._runCommand('g.copy', raster_3d=string, env=env)
  836. node = 'raster_3d'
  837. if pasted == 0:
  838. self.InsertLayer(name=new_name, mapset_node=self.selected_mapset[0],
  839. element_name=node)
  840. Debug.msg(1, "COPIED TO: " + new_name)
  841. if self.copy_mode:
  842. self.showNotification.emit(message=_("g.copy completed"))
  843. else:
  844. self.showNotification.emit(message=_("g.copy completed"))
  845. # remove old
  846. if not self.copy_mode:
  847. self._removeMapAfterCopy(self.copy_layer[i], self.copy_mapset[i], env2)
  848. gscript.try_remove(gisrc)
  849. gscript.try_remove(gisrc2)
  850. # expand selected mapset
  851. else:
  852. if self.copy_layer[i].data['type'] == 'raster_3d':
  853. GError(_("Reprojection is not implemented for 3D rasters"), parent=self)
  854. return
  855. if map_exists(new_name, element=self.copy_layer[i].data['type'], env=env,
  856. mapset=self.selected_mapset[0].data['name']):
  857. new_name = self._getNewMapName(_('New name'), _('Select new name'),
  858. self.copy_layer[i].data['name'], env=env,
  859. mapset=self.selected_mapset[0].data['name'],
  860. element=self.copy_layer[i].data['type'])
  861. if not new_name:
  862. continue
  863. callback = lambda gisrc2=gisrc2, gisrc=gisrc, cLayer=self.copy_layer[i], \
  864. cMapset=self.copy_mapset[i], cMode=self.copy_mode, name=new_name: \
  865. self._onDoneReprojection(env2, gisrc2, gisrc, cLayer, cMapset, cMode, name)
  866. dlg = CatalogReprojectionDialog(self, self._giface,
  867. self.copy_grassdb[i].data['name'],
  868. self.copy_location[i].data['name'],
  869. self.copy_mapset[i].data['name'],
  870. self.copy_layer[i].data['name'],
  871. env2,
  872. self.selected_grassdb[0].data['name'],
  873. self.selected_location[0].data['name'],
  874. self.selected_mapset[0].data['name'],
  875. new_name,
  876. self.copy_layer[i].data['type'],
  877. env, callback)
  878. if dlg.ShowModal() == wx.ID_CANCEL:
  879. return
  880. self.ExpandNode(self.selected_mapset[0], recursive=True)
  881. self._resetCopyVariables()
  882. def _onDoneReprojection(self, iEnv, iGisrc, oGisrc, cLayer, cMapset, cMode, name):
  883. self.InsertLayer(name=name, mapset_node=self.selected_mapset[0],
  884. element_name=cLayer.data['type'])
  885. if not cMode:
  886. self._removeMapAfterCopy(cLayer, cMapset, iEnv)
  887. gscript.try_remove(iGisrc)
  888. gscript.try_remove(oGisrc)
  889. self.ExpandNode(self.selected_mapset[0], recursive=True)
  890. def _removeMapAfterCopy(self, cLayer, cMapset, env):
  891. removed, cmd = self._runCommand('g.remove', type=cLayer.data['type'],
  892. name=cLayer.data['name'], flags='f', env=env)
  893. if removed == 0:
  894. self._model.RemoveNode(cLayer)
  895. self.RefreshNode(cMapset, recursive=True)
  896. Debug.msg(1, "LAYER " + cLayer.data['name'] + " DELETED")
  897. self.showNotification.emit(message=_("g.remove completed"))
  898. def InsertLayer(self, name, mapset_node, element_name):
  899. """Insert layer into model and refresh tree"""
  900. self._model.AppendNode(parent=mapset_node,
  901. data=dict(type=element_name, name=name))
  902. self._model.SortChildren(mapset_node)
  903. self.RefreshNode(mapset_node, recursive=True)
  904. def InsertMapset(self, name, location_node):
  905. """Insert new mapset into model and refresh tree.
  906. Assumes mapset is empty."""
  907. mapset_path = os.path.join(location_node.parent.data['name'],
  908. location_node.data['name'],
  909. name)
  910. mapset_node = self._model.AppendNode(parent=location_node,
  911. data=dict(type='mapset',
  912. name=name,
  913. current=False,
  914. lock=is_mapset_locked(mapset_path),
  915. is_different_owner=is_different_mapset_owner(mapset_path),
  916. owner=get_mapset_owner(mapset_path)))
  917. self._model.SortChildren(location_node)
  918. self.RefreshNode(location_node, recursive=True)
  919. return mapset_node
  920. def InsertLocation(self, name, grassdb_node):
  921. """Insert new location into model and refresh tree"""
  922. location_node = self._model.AppendNode(parent=grassdb_node,
  923. data=dict(type='location', name=name))
  924. # reload new location since it has a mapset
  925. self._reloadLocationNode(location_node)
  926. self._model.SortChildren(grassdb_node)
  927. self.RefreshNode(grassdb_node, recursive=True)
  928. return location_node
  929. def InsertGrassDb(self, name):
  930. """
  931. Insert new grass db into model, update user setting and refresh tree.
  932. Check if not already added.
  933. """
  934. grassdb_node = self._model.SearchNodes(name=name,
  935. type='grassdb')
  936. if not grassdb_node:
  937. grassdb_node = self._model.AppendNode(parent=self._model.root,
  938. data=dict(type="grassdb", name=name))
  939. self._reloadGrassDBNode(grassdb_node)
  940. self.RefreshItems()
  941. # Update user's settings
  942. self.grassdatabases.append(name)
  943. self._saveGrassDBs()
  944. return grassdb_node
  945. def OnDeleteMap(self, event):
  946. """Delete layer or mapset"""
  947. names = [self.selected_layer[i].data['name'] + '@' + self.selected_mapset[i].data['name']
  948. for i in range(len(self.selected_layer))]
  949. if len(names) < 10:
  950. question = _("Do you really want to delete map(s) <{m}>?").format(m=', '.join(names))
  951. else:
  952. question = _("Do you really want to delete {n} maps?").format(n=len(names))
  953. if self._confirmDialog(question, title=_('Delete map')) == wx.ID_YES:
  954. label = _("Deleting {name}...").format(name=names)
  955. self.showNotification.emit(message=label)
  956. for i in range(len(self.selected_layer)):
  957. gisrc, env = gscript.create_environment(
  958. self.selected_grassdb[i].data['name'],
  959. self.selected_location[i].data['name'],
  960. self.selected_mapset[i].data['name'])
  961. removed, cmd = self._runCommand(
  962. 'g.remove', flags='f', type=self.selected_layer[i].data['type'],
  963. name=self.selected_layer[i].data['name'], env=env)
  964. if removed == 0:
  965. self._model.RemoveNode(self.selected_layer[i])
  966. self.RefreshNode(self.selected_mapset[i], recursive=True)
  967. Debug.msg(1, "LAYER " + self.selected_layer[i].data['name'] + " DELETED")
  968. # remove map layer from layer tree if exists
  969. if not isinstance(self._giface, StandaloneGrassInterface):
  970. name = self.selected_layer[i].data['name'] + '@' + self.selected_mapset[i].data['name']
  971. layers = self._giface.GetLayerList().GetLayersByName(name)
  972. for layer in layers:
  973. self._giface.GetLayerList().DeleteLayer(layer)
  974. gscript.try_remove(gisrc)
  975. self.UnselectAll()
  976. self.showNotification.emit(message=_("g.remove completed"))
  977. def OnDeleteMapset(self, event):
  978. """
  979. Delete selected mapset or mapsets
  980. """
  981. mapsets = []
  982. for i in range(len(self.selected_mapset)):
  983. # Append to the list of tuples
  984. mapsets.append((
  985. self.selected_grassdb[i].data['name'],
  986. self.selected_location[i].data['name'],
  987. self.selected_mapset[i].data['name']
  988. ))
  989. if delete_mapsets_interactively(self, mapsets):
  990. locations = set([each for each in self.selected_location])
  991. for loc_node in locations:
  992. self._reloadLocationNode(loc_node)
  993. self.UpdateCurrentDbLocationMapsetNode()
  994. self.RefreshNode(loc_node, recursive=True)
  995. def OnDeleteLocation(self, event):
  996. """
  997. Delete selected location or locations
  998. """
  999. locations = []
  1000. for i in range(len(self.selected_location)):
  1001. # Append to the list of tuples
  1002. locations.append((
  1003. self.selected_grassdb[i].data['name'],
  1004. self.selected_location[i].data['name']
  1005. ))
  1006. if delete_locations_interactively(self, locations):
  1007. grassdbs = set([each for each in self.selected_grassdb])
  1008. for grassdb_node in grassdbs:
  1009. self._reloadGrassDBNode(grassdb_node)
  1010. self.UpdateCurrentDbLocationMapsetNode()
  1011. self.RefreshNode(grassdb_node, recursive=True)
  1012. def DownloadLocation(self, grassdb_node):
  1013. """
  1014. Download new location interactively.
  1015. """
  1016. grassdatabase, location, mapset = (
  1017. download_location_interactively(self, grassdb_node.data['name'])
  1018. )
  1019. if location:
  1020. self._reloadGrassDBNode(grassdb_node)
  1021. self.UpdateCurrentDbLocationMapsetNode()
  1022. self.RefreshItems()
  1023. def OnDownloadLocation(self, event):
  1024. """
  1025. Download location online
  1026. """
  1027. self.DownloadLocation(self.selected_grassdb[0])
  1028. def DeleteGrassDb(self, grassdb_node):
  1029. """
  1030. Delete grassdb from disk.
  1031. """
  1032. grassdb = grassdb_node.data['name']
  1033. if (delete_grassdb_interactively(self, grassdb)):
  1034. self.RemoveGrassDB(grassdb_node)
  1035. def OnDeleteGrassDb(self, event):
  1036. """
  1037. Delete grassdb from disk.
  1038. """
  1039. self.DeleteGrassDb(self.selected_grassdb[0])
  1040. def OnRemoveGrassDb(self, event):
  1041. """
  1042. Remove grassdb node from data catalogue.
  1043. """
  1044. self.RemoveGrassDB(self.selected_grassdb[0])
  1045. def RemoveGrassDB(self, grassdb_node):
  1046. """
  1047. Remove grassdb node from tree
  1048. and updates settings. Doesn't check if it's current db.
  1049. """
  1050. self.grassdatabases.remove(grassdb_node.data['name'])
  1051. self._model.RemoveNode(grassdb_node)
  1052. self.RefreshItems()
  1053. # Update user's settings
  1054. self._saveGrassDBs()
  1055. def OnDisplayLayer(self, event):
  1056. """
  1057. Display layer in current graphics view
  1058. """
  1059. self.DisplayLayer()
  1060. def DisplayLayer(self):
  1061. """Display selected layer in current graphics view"""
  1062. all_names = []
  1063. names = {'raster': [], 'vector': [], 'raster_3d': []}
  1064. for i in range(len(self.selected_layer)):
  1065. name = self.selected_layer[i].data['name'] + '@' + self.selected_mapset[i].data['name']
  1066. names[self.selected_layer[i].data['type']].append(name)
  1067. all_names.append(name)
  1068. #if self.selected_location[0].data['name'] == gisenv()['LOCATION_NAME'] and self.selected_mapset[0]:
  1069. for ltype in names:
  1070. if names[ltype]:
  1071. self._giface.lmgr.AddMaps(list(reversed(names[ltype])), ltype, True)
  1072. if len(self._giface.GetLayerList()) == 1:
  1073. # zoom to map if there is only one map layer
  1074. self._giface.GetMapWindow().ZoomToMap()
  1075. Debug.msg(1, "Displayed layer(s): " + str(all_names))
  1076. def OnBeginDrag(self, node, event):
  1077. """Just copy necessary data"""
  1078. self.DefineItems(self.GetSelected())
  1079. if self.selected_location and None in self.selected_mapset and \
  1080. None in self.selected_layer:
  1081. GMessage(_("Move or copy location isn't allowed"))
  1082. event.Veto()
  1083. return
  1084. elif self.selected_location and self.selected_mapset and \
  1085. None in self.selected_layer:
  1086. GMessage(_("Move or copy mapset isn't allowed"))
  1087. event.Veto()
  1088. return
  1089. if self.selected_layer and not (self._restricted and gisenv()[
  1090. 'LOCATION_NAME'] != self.selected_location[0].data['name']):
  1091. event.Allow()
  1092. self.OnCopyMap(event)
  1093. Debug.msg(1, "DRAG")
  1094. else:
  1095. event.Veto()
  1096. def OnEndDrag(self, node, event):
  1097. """Copy layer into target"""
  1098. self.copy_mode = wx.GetMouseState().ControlDown()
  1099. if node:
  1100. self.DefineItems([node])
  1101. if None not in self.selected_mapset:
  1102. if self._restricted and gisenv()['MAPSET'] != self.selected_mapset[0].data['name']:
  1103. GMessage(_("To move or copy maps to other mapsets, unlock editing of other mapsets"),
  1104. parent=self)
  1105. event.Veto()
  1106. return
  1107. event.Allow()
  1108. Debug.msg(1, "DROP DONE")
  1109. self.OnPasteMap(event)
  1110. else:
  1111. GMessage(_("To move or copy maps to other location, "
  1112. "please drag them to a mapset in the "
  1113. "destination location"),
  1114. parent=self)
  1115. event.Veto()
  1116. return
  1117. def OnSwitchDbLocationMapset(self, event):
  1118. """Switch to location and mapset"""
  1119. self._SwitchDbLocationMapset()
  1120. def _SwitchDbLocationMapset(self):
  1121. """Switch to location and mapset"""
  1122. genv = gisenv()
  1123. # Distinguish when only part of db/location/mapset is changed.
  1124. if (
  1125. self.selected_grassdb[0].data['name'] == genv['GISDBASE']
  1126. and self.selected_location[0].data['name'] == genv['LOCATION_NAME']
  1127. ):
  1128. self.changeMapset.emit(mapset=self.selected_mapset[0].data['name'])
  1129. elif self.selected_grassdb[0].data['name'] == genv['GISDBASE']:
  1130. self.changeLocation.emit(mapset=self.selected_mapset[0].data['name'],
  1131. location=self.selected_location[0].data['name'],
  1132. dbase=None)
  1133. else:
  1134. self.changeLocation.emit(mapset=self.selected_mapset[0].data['name'],
  1135. location=self.selected_location[0].data['name'],
  1136. dbase=self.selected_grassdb[0].data['name'])
  1137. self.UpdateCurrentDbLocationMapsetNode()
  1138. self.ExpandCurrentMapset()
  1139. self.RefreshItems()
  1140. def OnMetadata(self, event):
  1141. """Show metadata of any raster/vector/3draster"""
  1142. def done(event):
  1143. gscript.try_remove(event.userData)
  1144. for i in range(len(self.selected_layer)):
  1145. if self.selected_layer[i].data['type'] == 'raster':
  1146. cmd = ['r.info']
  1147. elif self.selected_layer[i].data['type'] == 'vector':
  1148. cmd = ['v.info']
  1149. elif self.selected_layer[i].data['type'] == 'raster_3d':
  1150. cmd = ['r3.info']
  1151. cmd.append('map=%s@%s' % (self.selected_layer[i].data['name'], self.selected_mapset[i].data['name']))
  1152. gisrc, env = gscript.create_environment(
  1153. self.selected_grassdb[i].data['name'],
  1154. self.selected_location[i].data['name'],
  1155. self.selected_mapset[i].data['name'])
  1156. # print output to command log area
  1157. # temp gisrc file must be deleted onDone
  1158. self._giface.RunCmd(cmd, env=env, onDone=done, userData=gisrc)
  1159. def OnCopyName(self, event):
  1160. """Copy layer name to clipboard"""
  1161. if wx.TheClipboard.Open():
  1162. do = wx.TextDataObject()
  1163. text = []
  1164. for i in range(len(self.selected_layer)):
  1165. text.append('%s@%s' % (self.selected_layer[i].data['name'], self.selected_mapset[i].data['name']))
  1166. do.SetText(','.join(text))
  1167. wx.TheClipboard.SetData(do)
  1168. wx.TheClipboard.Close()
  1169. def Filter(self, text):
  1170. """Filter tree based on name and type."""
  1171. text = text.strip()
  1172. if len(text.split(':')) > 1:
  1173. name = text.split(':')[1].strip()
  1174. elem = text.split(':')[0].strip()
  1175. if 'r' == elem:
  1176. element = 'raster'
  1177. elif 'r3' == elem:
  1178. element = 'raster_3d'
  1179. elif 'v' == elem:
  1180. element = 'vector'
  1181. else:
  1182. element = None
  1183. else:
  1184. element = None
  1185. name = text.strip()
  1186. self._model = filterModel(self._orig_model, name=name, element=element)
  1187. self.UpdateCurrentDbLocationMapsetNode()
  1188. self.RefreshItems()
  1189. self.ExpandCurrentMapset()
  1190. def _getNewMapName(self, message, title, value, element, mapset, env):
  1191. """Dialog for simple text entry"""
  1192. dlg = NameEntryDialog(parent=self, message=message, caption=title,
  1193. element=element, env=env, mapset=mapset)
  1194. dlg.SetValue(value)
  1195. if dlg.ShowModal() == wx.ID_OK:
  1196. name = dlg.GetValue()
  1197. else:
  1198. name = None
  1199. dlg.Destroy()
  1200. return name
  1201. def _confirmDialog(self, question, title):
  1202. """Confirm dialog"""
  1203. dlg = wx.MessageDialog(self, question, title, wx.YES_NO)
  1204. res = dlg.ShowModal()
  1205. dlg.Destroy()
  1206. return res
  1207. def _isCurrent(self, genv):
  1208. if self._restricted:
  1209. currentMapset = currentLocation = currentGrassDb = True
  1210. for i in range(len(self.selected_grassdb)):
  1211. if self.selected_grassdb[i].data['name'] != genv['GISDBASE']:
  1212. currentGrassDb = False
  1213. currentLocation = False
  1214. currentMapset = False
  1215. break
  1216. if currentLocation and self.selected_location[0]:
  1217. for i in range(len(self.selected_location)):
  1218. if self.selected_location[i].data['name'] != genv['LOCATION_NAME']:
  1219. currentLocation = False
  1220. currentMapset = False
  1221. break
  1222. if currentMapset and self.selected_mapset[0]:
  1223. for i in range(len(self.selected_mapset)):
  1224. if self.selected_mapset[i].data['name'] != genv['MAPSET']:
  1225. currentMapset = False
  1226. break
  1227. return currentGrassDb, currentLocation, currentMapset
  1228. else:
  1229. return True, True, True
  1230. def _popupMenuLayer(self):
  1231. """Create popup menu for layers"""
  1232. menu = Menu()
  1233. genv = gisenv()
  1234. currentGrassDb, currentLocation, currentMapset = self._isCurrent(genv)
  1235. item = wx.MenuItem(menu, wx.ID_ANY, _("&Cut"))
  1236. menu.AppendItem(item)
  1237. self.Bind(wx.EVT_MENU, self.OnMoveMap, item)
  1238. if not currentMapset:
  1239. item.Enable(False)
  1240. item = wx.MenuItem(menu, wx.ID_ANY, _("&Copy"))
  1241. menu.AppendItem(item)
  1242. self.Bind(wx.EVT_MENU, self.OnCopyMap, item)
  1243. item = wx.MenuItem(menu, wx.ID_ANY, _("Copy &name"))
  1244. menu.AppendItem(item)
  1245. self.Bind(wx.EVT_MENU, self.OnCopyName, item)
  1246. item = wx.MenuItem(menu, wx.ID_ANY, _("&Paste"))
  1247. menu.AppendItem(item)
  1248. self.Bind(wx.EVT_MENU, self.OnPasteMap, item)
  1249. if not(currentMapset and self.copy_layer):
  1250. item.Enable(False)
  1251. item = wx.MenuItem(menu, wx.ID_ANY, _("&Delete"))
  1252. menu.AppendItem(item)
  1253. self.Bind(wx.EVT_MENU, self.OnDeleteMap, item)
  1254. item.Enable(currentMapset)
  1255. item = wx.MenuItem(menu, wx.ID_ANY, _("&Rename"))
  1256. menu.AppendItem(item)
  1257. self.Bind(wx.EVT_MENU, self.OnRenameMap, item)
  1258. item.Enable(currentMapset and len(self.selected_layer) == 1)
  1259. menu.AppendSeparator()
  1260. if not isinstance(self._giface, StandaloneGrassInterface):
  1261. if all([each.data['name'] == genv['LOCATION_NAME'] for each in self.selected_location]):
  1262. if len(self.selected_layer) > 1:
  1263. item = wx.MenuItem(menu, wx.ID_ANY, _("&Display layers"))
  1264. else:
  1265. item = wx.MenuItem(menu, wx.ID_ANY, _("&Display layer"))
  1266. menu.AppendItem(item)
  1267. self.Bind(wx.EVT_MENU, self.OnDisplayLayer, item)
  1268. item = wx.MenuItem(menu, wx.ID_ANY, _("Show &metadata"))
  1269. menu.AppendItem(item)
  1270. self.Bind(wx.EVT_MENU, self.OnMetadata, item)
  1271. self.PopupMenu(menu)
  1272. menu.Destroy()
  1273. def _popupMenuMapset(self):
  1274. """Create popup menu for mapsets"""
  1275. menu = Menu()
  1276. genv = gisenv()
  1277. currentGrassDb, currentLocation, currentMapset = self._isCurrent(genv)
  1278. item = wx.MenuItem(menu, wx.ID_ANY, _("&Paste"))
  1279. menu.AppendItem(item)
  1280. self.Bind(wx.EVT_MENU, self.OnPasteMap, item)
  1281. if not(currentMapset and self.copy_layer):
  1282. item.Enable(False)
  1283. item = wx.MenuItem(menu, wx.ID_ANY, _("&Switch mapset"))
  1284. menu.AppendItem(item)
  1285. self.Bind(wx.EVT_MENU, self.OnSwitchDbLocationMapset, item)
  1286. if (
  1287. self.selected_grassdb[0].data['name'] == genv['GISDBASE']
  1288. and self.selected_location[0].data['name'] == genv['LOCATION_NAME']
  1289. and self.selected_mapset[0].data['name'] == genv['MAPSET']
  1290. ):
  1291. item.Enable(False)
  1292. item = wx.MenuItem(menu, wx.ID_ANY, _("&Delete mapset"))
  1293. menu.AppendItem(item)
  1294. self.Bind(wx.EVT_MENU, self.OnDeleteMapset, item)
  1295. if self._restricted:
  1296. item.Enable(False)
  1297. item = wx.MenuItem(menu, wx.ID_ANY, _("&Rename mapset"))
  1298. menu.AppendItem(item)
  1299. self.Bind(wx.EVT_MENU, self.OnRenameMapset, item)
  1300. if self._restricted:
  1301. item.Enable(False)
  1302. self.PopupMenu(menu)
  1303. menu.Destroy()
  1304. def _popupMenuLocation(self):
  1305. """Create popup menu for locations"""
  1306. menu = Menu()
  1307. item = wx.MenuItem(menu, wx.ID_ANY, _("&Create mapset"))
  1308. menu.AppendItem(item)
  1309. self.Bind(wx.EVT_MENU, self.OnCreateMapset, item)
  1310. item = wx.MenuItem(menu, wx.ID_ANY, _("&Delete location"))
  1311. menu.AppendItem(item)
  1312. self.Bind(wx.EVT_MENU, self.OnDeleteLocation, item)
  1313. if self._restricted:
  1314. item.Enable(False)
  1315. item = wx.MenuItem(menu, wx.ID_ANY, _("&Rename location"))
  1316. menu.AppendItem(item)
  1317. self.Bind(wx.EVT_MENU, self.OnRenameLocation, item)
  1318. if self._restricted:
  1319. item.Enable(False)
  1320. self.PopupMenu(menu)
  1321. menu.Destroy()
  1322. def _popupMenuGrassDb(self):
  1323. """Create popup menu for grass db"""
  1324. menu = Menu()
  1325. genv = gisenv()
  1326. currentGrassDb, currentLocation, currentMapset = self._isCurrent(genv)
  1327. item = wx.MenuItem(menu, wx.ID_ANY, _("&Create new location"))
  1328. menu.AppendItem(item)
  1329. self.Bind(wx.EVT_MENU, self.OnCreateLocation, item)
  1330. item = wx.MenuItem(menu, wx.ID_ANY, _("&Download sample location"))
  1331. menu.AppendItem(item)
  1332. self.Bind(wx.EVT_MENU, self.OnDownloadLocation, item)
  1333. item = wx.MenuItem(menu, wx.ID_ANY, _("&Remove GRASS database from data catalog"))
  1334. menu.AppendItem(item)
  1335. self.Bind(wx.EVT_MENU, self.OnRemoveGrassDb, item)
  1336. if currentGrassDb:
  1337. item.Enable(False)
  1338. item = wx.MenuItem(menu, wx.ID_ANY, _("&Delete GRASS database from disk"))
  1339. menu.AppendItem(item)
  1340. self.Bind(wx.EVT_MENU, self.OnDeleteGrassDb, item)
  1341. if currentGrassDb:
  1342. item.Enable(False)
  1343. self.PopupMenu(menu)
  1344. menu.Destroy()
  1345. def _popupMenuElement(self):
  1346. """Create popup menu for elements"""
  1347. menu = Menu()
  1348. item = wx.MenuItem(menu, wx.ID_ANY, _("&Paste"))
  1349. menu.AppendItem(item)
  1350. self.Bind(wx.EVT_MENU, self.OnPasteMap, item)
  1351. genv = gisenv()
  1352. currentGrassDb, currentLocation, currentMapset = self._isCurrent(genv)
  1353. if not(currentMapset and self.copy_layer):
  1354. item.Enable(False)
  1355. self.PopupMenu(menu)
  1356. menu.Destroy()
  1357. def _popupMenuMultipleLocations(self):
  1358. """Create popup menu for multiple selected locations"""
  1359. menu = Menu()
  1360. item = wx.MenuItem(menu, wx.ID_ANY, _("&Delete locations"))
  1361. menu.AppendItem(item)
  1362. self.Bind(wx.EVT_MENU, self.OnDeleteLocation, item)
  1363. if self._restricted:
  1364. item.Enable(False)
  1365. self.PopupMenu(menu)
  1366. menu.Destroy()
  1367. def _popupMenuMultipleMapsets(self):
  1368. """Create popup menu for multiple selected mapsets"""
  1369. menu = Menu()
  1370. item = wx.MenuItem(menu, wx.ID_ANY, _("&Delete mapsets"))
  1371. menu.AppendItem(item)
  1372. self.Bind(wx.EVT_MENU, self.OnDeleteMapset, item)
  1373. if self._restricted:
  1374. item.Enable(False)
  1375. self.PopupMenu(menu)
  1376. menu.Destroy()
  1377. def _popupMenuEmpty(self):
  1378. """Create empty popup when multiple different types of items are selected"""
  1379. menu = Menu()
  1380. item = wx.MenuItem(menu, wx.ID_ANY, _("No available options"))
  1381. menu.AppendItem(item)
  1382. item.Enable(False)
  1383. self.PopupMenu(menu)
  1384. menu.Destroy()