tree.py 62 KB

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