tree.py 56 KB

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