tree.py 63 KB

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