gselect.py 96 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768176917701771177217731774177517761777177817791780178117821783178417851786178717881789179017911792179317941795179617971798179918001801180218031804180518061807180818091810181118121813181418151816181718181819182018211822182318241825182618271828182918301831183218331834183518361837183818391840184118421843184418451846184718481849185018511852185318541855185618571858185918601861186218631864186518661867186818691870187118721873187418751876187718781879188018811882188318841885188618871888188918901891189218931894189518961897189818991900190119021903190419051906190719081909191019111912191319141915191619171918191919201921192219231924192519261927192819291930193119321933193419351936193719381939194019411942194319441945194619471948194919501951195219531954195519561957195819591960196119621963196419651966196719681969197019711972197319741975197619771978197919801981198219831984198519861987198819891990199119921993199419951996199719981999200020012002200320042005200620072008200920102011201220132014201520162017201820192020202120222023202420252026202720282029203020312032203320342035203620372038203920402041204220432044204520462047204820492050205120522053205420552056205720582059206020612062206320642065206620672068206920702071207220732074207520762077207820792080208120822083208420852086208720882089209020912092209320942095209620972098209921002101210221032104210521062107210821092110211121122113211421152116211721182119212021212122212321242125212621272128212921302131213221332134213521362137213821392140214121422143214421452146214721482149215021512152215321542155215621572158215921602161216221632164216521662167216821692170217121722173217421752176217721782179218021812182218321842185218621872188218921902191219221932194219521962197219821992200220122022203220422052206220722082209221022112212221322142215221622172218221922202221222222232224222522262227222822292230223122322233223422352236223722382239224022412242224322442245224622472248224922502251225222532254225522562257225822592260226122622263226422652266226722682269227022712272227322742275227622772278227922802281228222832284228522862287228822892290229122922293229422952296229722982299230023012302230323042305230623072308230923102311231223132314231523162317231823192320232123222323232423252326232723282329233023312332233323342335233623372338233923402341234223432344234523462347234823492350235123522353235423552356235723582359236023612362236323642365236623672368236923702371237223732374237523762377237823792380238123822383238423852386238723882389239023912392239323942395239623972398239924002401240224032404240524062407240824092410241124122413241424152416241724182419242024212422242324242425242624272428242924302431243224332434243524362437243824392440244124422443244424452446244724482449245024512452245324542455245624572458245924602461246224632464246524662467246824692470247124722473247424752476247724782479248024812482248324842485248624872488248924902491249224932494249524962497249824992500250125022503250425052506250725082509251025112512251325142515251625172518251925202521252225232524252525262527252825292530253125322533253425352536253725382539254025412542254325442545254625472548254925502551255225532554255525562557255825592560256125622563256425652566256725682569257025712572257325742575257625772578257925802581258225832584258525862587258825892590259125922593259425952596259725982599260026012602260326042605260626072608260926102611261226132614261526162617261826192620262126222623262426252626262726282629263026312632263326342635263626372638263926402641264226432644264526462647264826492650265126522653265426552656265726582659266026612662266326642665266626672668266926702671267226732674267526762677267826792680268126822683268426852686268726882689269026912692269326942695269626972698269927002701270227032704270527062707270827092710271127122713271427152716271727182719272027212722272327242725272627272728272927302731273227332734273527362737273827392740274127422743274427452746274727482749275027512752275327542755275627572758275927602761
  1. """
  2. @package gui_core.gselect
  3. @brief Custom control that selects elements
  4. Classes:
  5. - :class:`Select`
  6. - :class:`VectorSelect`
  7. - :class:`ListCtrlComboPopup`
  8. - :class:`TreeCrtlComboPopup`
  9. - :class:`VectorDBInfo`
  10. - :class:`LayerSelect`
  11. - :class:`DriverSelect`
  12. - :class:`DatabaseSelect`
  13. - :class:`TableSelect`
  14. - :class:`ColumnSelect`
  15. - :class:`DbaseSelect`
  16. - :class:`LocationSelect`
  17. - :class:`MapsetSelect`
  18. - :class:`SubGroupSelect`
  19. - :class:`FormatSelect`
  20. - :class:`GdalSelect`
  21. - :class:`ProjSelect`
  22. - :class:`ElementSelect`
  23. - :class:`OgrTypeSelect`
  24. - :class:`CoordinatesSelect`
  25. - :class:`VectorCategorySelect`
  26. - :class:`SignatureSelect`
  27. - :class:`SeparatorSelect`
  28. - :class:`SqlWhereSelect`
  29. (C) 2007-2018 by the GRASS Development Team
  30. This program is free software under the GNU General Public License
  31. (>=v2). Read the file COPYING that comes with GRASS for details.
  32. @author Michael Barton
  33. @author Martin Landa <landa.martin gmail.com>
  34. @author Vaclav Petras <wenzeslaus gmail.com> (menu customization)
  35. @author Stepan Turek <stepan.turek seznam.cz> (CoordinatesSelect, ListCtrlComboPopup)
  36. @author Matej Krejci <matejkrejci gmail.com> (VectorCategorySelect)
  37. """
  38. from __future__ import print_function
  39. import os
  40. import sys
  41. import glob
  42. import six
  43. import wx
  44. from core import globalvar
  45. import wx.lib.buttons as buttons
  46. import wx.lib.filebrowsebutton as filebrowse
  47. import grass.script as grass
  48. from grass.script import task as gtask
  49. from grass.exceptions import CalledModuleError
  50. from gui_core.widgets import ManageSettingsWidget, CoordinatesValidator
  51. from core.gcmd import RunCommand, GMessage, GWarning, GException
  52. from core.utils import GetListOfLocations, GetListOfMapsets, \
  53. GetFormats, rasterFormatExtension, vectorFormatExtension
  54. from core.utils import GetSettingsPath, GetValidLayerName, ListSortLower
  55. from core.utils import GetVectorNumberOfLayers
  56. from core.settings import UserSettings
  57. from core.debug import Debug
  58. from gui_core.vselect import VectorSelectBase
  59. from gui_core.wrap import TreeCtrl, Button, StaticText, StaticBox, \
  60. TextCtrl, Panel, ComboPopup, ComboCtrl
  61. from grass.pydispatch.signal import Signal
  62. class Select(ComboCtrl):
  63. def __init__(
  64. self, parent, id=wx.ID_ANY, size=globalvar.DIALOG_GSELECT_SIZE,
  65. type=None, multiple=False, nmaps=1, mapsets=None,
  66. updateOnPopup=True, onPopup=None, fullyQualified=True,
  67. extraItems={},
  68. layerTree=None, validator=wx.DefaultValidator):
  69. """Custom control to create a ComboBox with a tree control to
  70. display and select GIS elements within acessible mapsets.
  71. Elements can be selected with mouse. Can allow multiple
  72. selections, when argument <em>multiple</em> is True. Multiple
  73. selections are separated by commas.
  74. :param type: type of GIS elements ('raster, 'vector', ...)
  75. :param multiple: True for multiple input
  76. :param nmaps: number of maps to be entered
  77. :param mapsets: force list of mapsets (otherwise search path)
  78. :param updateOnPopup: True for updating list of elements on popup
  79. :param onPopup: function to be called on Popup
  80. :param fullyQualified: True to provide fully qualified names (map@mapset)
  81. :param extraItems: extra items to add (given as dictionary) - see gmodeler for usage
  82. :param layerTree: show only elements from given layer tree if not None
  83. :param validator: validator for TextCtrl
  84. """
  85. ComboCtrl.__init__(
  86. self,
  87. parent=parent,
  88. id=id,
  89. size=size,
  90. validator=validator)
  91. if globalvar.CheckWxVersion([3]):
  92. self.SetName("Select")
  93. else:
  94. self.GetChildren()[0].SetName("Select")
  95. self.GetChildren()[0].type = type
  96. self.tcp = TreeCtrlComboPopup()
  97. self.SetPopupControl(self.tcp)
  98. self.SetPopupExtents(0, 100)
  99. if type:
  100. self.tcp.SetData(
  101. type=type,
  102. mapsets=mapsets,
  103. multiple=multiple,
  104. nmaps=nmaps,
  105. updateOnPopup=updateOnPopup,
  106. onPopup=onPopup,
  107. fullyQualified=fullyQualified,
  108. extraItems=extraItems,
  109. layerTree=layerTree)
  110. self.Bind(wx.EVT_KEY_DOWN, self.OnKeyDown)
  111. def OnKeyDown(self, event):
  112. """Open popup and send key events to the tree."""
  113. if self.IsPopupShown():
  114. # on some configurations we get key down, with some only key up
  115. # so we are trying to catch either key up or down
  116. # this mess shouldn't be necessary with wxPython 3
  117. self.tcp.OnKeyUp(event)
  118. else:
  119. if event.GetKeyCode() == wx.WXK_DOWN:
  120. self.ShowPopup()
  121. event.Skip()
  122. def SetElementList(self, type, mapsets=None):
  123. """Set element list
  124. :param type: GIS element type
  125. :param mapsets: list of acceptable mapsets (None for all in search path)
  126. """
  127. self.tcp.SetData(type=type, mapsets=mapsets)
  128. def GetElementList(self):
  129. """Load elements"""
  130. self.tcp.GetElementList()
  131. def SetType(self, etype, multiple=False, nmaps=1,
  132. mapsets=None, updateOnPopup=True, onPopup=None):
  133. """Param set element type for widget
  134. :param etype: element type, see gselect.ElementSelect
  135. """
  136. self.tcp.SetData(type=etype, mapsets=mapsets,
  137. multiple=multiple, nmaps=nmaps,
  138. updateOnPopup=updateOnPopup, onPopup=onPopup)
  139. class VectorSelect(Select):
  140. def __init__(self, parent, ftype, **kwargs):
  141. """Custom to create a ComboBox with a tree control to display and
  142. select vector maps. You can filter the vector maps. If you
  143. don't need this feature use Select class instead
  144. :param ftype: filter vector maps based on feature type
  145. """
  146. Select.__init__(self, parent=parent, id=wx.ID_ANY,
  147. type='vector', **kwargs)
  148. self.ftype = ftype
  149. # remove vector maps which do not contain given feature type
  150. self.tcp.SetFilter(self._isElement)
  151. def _isElement(self, vectorName):
  152. """Check if element should be filtered out"""
  153. try:
  154. if int(grass.vector_info_topo(vectorName)[self.ftype]) < 1:
  155. return False
  156. except KeyError:
  157. return False
  158. return True
  159. class ListCtrlComboPopup(ComboPopup):
  160. """Create a list ComboBox using TreeCtrl with hidden root.
  161. .. todo::
  162. use event.EventObject instead of hardcoding (see forms.py)
  163. https://groups.google.com/forum/#!topic/wxpython-users/pRz6bi0k0XY
  164. """
  165. # overridden ComboPopup methods
  166. def Init(self):
  167. self.value = [] # for multiple is False ->
  168. # len(self.value) in [0,1]
  169. self.curitem = None
  170. self.multiple = False
  171. self.updateOnPopup = True
  172. self.filterItems = [] # limit items based on this list,
  173. # see layerTree parameter
  174. def Create(self, parent):
  175. self.seltree = TreeCtrl(parent, style=wx.TR_HIDE_ROOT |
  176. wx.TR_HAS_BUTTONS |
  177. wx.TR_SINGLE |
  178. wx.TR_LINES_AT_ROOT |
  179. wx.SIMPLE_BORDER |
  180. wx.TR_FULL_ROW_HIGHLIGHT)
  181. self.seltree.Bind(wx.EVT_MOTION, self.OnMotion)
  182. self.seltree.Bind(wx.EVT_LEFT_DOWN, self.OnLeftDown)
  183. # the following dummy handler are needed to keep tree events
  184. # from propagating up to the parent GIS Manager layer tree
  185. self.seltree.Bind(wx.EVT_TREE_ITEM_EXPANDING, lambda x: None)
  186. self.seltree.Bind(wx.EVT_TREE_ITEM_COLLAPSED, lambda x: None)
  187. self.seltree.Bind(wx.EVT_TREE_SEL_CHANGED, lambda x: None)
  188. self.seltree.Bind(wx.EVT_TREE_DELETE_ITEM, lambda x: None)
  189. self.seltree.Bind(wx.EVT_TREE_BEGIN_DRAG, lambda x: None)
  190. self.seltree.Bind(wx.EVT_TREE_ITEM_RIGHT_CLICK, lambda x: None)
  191. # navigation in list/tree is handled automatically since wxPython 3
  192. # for older versions, we have to workaround it and write our own
  193. # navigation
  194. if globalvar.CheckWxVersion(version=[3]):
  195. self.seltree.Bind(
  196. wx.EVT_TREE_ITEM_ACTIVATED,
  197. self._onItemConfirmed)
  198. self.seltree.Bind(wx.EVT_TREE_KEY_DOWN, self._onDismissPopup)
  199. else:
  200. self.seltree.Bind(wx.EVT_TREE_ITEM_ACTIVATED, lambda x: None)
  201. self.seltree.Bind(wx.EVT_KEY_UP, self.OnKeyUp)
  202. return True
  203. def GetControl(self):
  204. return self.seltree
  205. def GetComboCtrl(self):
  206. if globalvar.wxPythonPhoenix:
  207. return super(ListCtrlComboPopup, self).GetComboCtrl()
  208. else:
  209. return self.GetCombo()
  210. def GetStringValue(self):
  211. """Get value as a string separated by commas
  212. """
  213. return ','.join(self.value)
  214. def SetStringValue(self, value):
  215. # this assumes that item strings are unique...
  216. root = self.seltree.GetRootItem()
  217. if not root:
  218. return
  219. winValue = self.GetComboCtrl().GetValue().strip(',')
  220. self.value = []
  221. if winValue:
  222. self.value = winValue.split(',')
  223. def OnPopup(self, force=False):
  224. """Limited only for first selected
  225. """
  226. if not force and not self.updateOnPopup:
  227. return
  228. # selects map starting according to written text
  229. inputText = self.GetComboCtrl().GetValue().strip()
  230. if inputText:
  231. root = self.seltree.GetRootItem()
  232. match = self.FindItem(root, inputText, startLetters=True)
  233. if match.IsOk():
  234. self.seltree.EnsureVisible(match)
  235. self.seltree.SelectItem(match)
  236. def GetAdjustedSize(self, minWidth, prefHeight, maxHeight):
  237. """Reads UserSettings to get height (which was 200 in old implementation).
  238. """
  239. height = UserSettings.Get(
  240. group='appearance',
  241. key='gSelectPopupHeight',
  242. subkey='value')
  243. return wx.Size(minWidth, min(height, maxHeight))
  244. def FindItem(self, parentItem, text, startLetters=False):
  245. """Finds item with given name or starting with given text
  246. """
  247. startletters = startLetters
  248. item, cookie = self.seltree.GetFirstChild(parentItem)
  249. while wx.TreeItemId.IsOk(item):
  250. if self.seltree.GetItemText(item) == text:
  251. return item
  252. if self.seltree.ItemHasChildren(item):
  253. item = self.FindItem(item, text, startLetters=startletters)
  254. if wx.TreeItemId.IsOk(item):
  255. return item
  256. elif startletters and self.seltree.GetItemText(item).startswith(text.split('@', 1)[0]):
  257. return item
  258. item, cookie = self.seltree.GetNextChild(parentItem, cookie)
  259. return wx.TreeItemId()
  260. def AddItem(self, value):
  261. root = self.seltree.GetRootItem()
  262. if not root:
  263. root = self.seltree.AddRoot("<hidden root>")
  264. self.seltree.AppendItem(root, text=value)
  265. def SetItems(self, items):
  266. root = self.seltree.GetRootItem()
  267. if not root:
  268. root = self.seltree.AddRoot("<hidden root>")
  269. for item in items:
  270. self.seltree.AppendItem(root, text=item)
  271. def OnKeyUp(self, event):
  272. """Enable to select items using keyboard.
  273. Unused with wxPython 3, can be removed in the future.
  274. """
  275. item = self.seltree.GetSelection()
  276. if event.GetKeyCode() == wx.WXK_DOWN:
  277. self.seltree.SelectItem(self.seltree.GetNextVisible(item))
  278. elif event.GetKeyCode() == wx.WXK_UP:
  279. itemPrev = self.seltree.GetPrevSibling(item)
  280. self.seltree.SelectItem(itemPrev)
  281. elif event.GetKeyCode() == wx.WXK_ESCAPE:
  282. self.Dismiss()
  283. elif event.GetKeyCode() == wx.WXK_RETURN:
  284. self.seltree.SelectItem(item)
  285. self.curitem = item
  286. self._selectTreeItem(item)
  287. self.Dismiss()
  288. def _onDismissPopup(self, event):
  289. """Hide popup without selecting item on Esc"""
  290. if event.GetKeyCode() == wx.WXK_ESCAPE:
  291. self.Dismiss()
  292. else:
  293. event.Skip()
  294. def _selectTreeItem(self, item):
  295. item_str = self.seltree.GetItemText(item)
  296. if self.multiple:
  297. if item_str not in self.value:
  298. self.value.append(item_str)
  299. else:
  300. self.value = [item_str]
  301. def _onItemConfirmed(self, event):
  302. item = event.GetItem()
  303. self._selectTreeItem(item)
  304. self.Dismiss()
  305. def OnMotion(self, evt):
  306. """Have the selection follow the mouse, like in a real combobox
  307. """
  308. item, flags = self.seltree.HitTest(evt.GetPosition())
  309. if item and flags & wx.TREE_HITTEST_ONITEMLABEL:
  310. self.seltree.SelectItem(item)
  311. self.curitem = item
  312. evt.Skip()
  313. def OnLeftDown(self, evt):
  314. """Do the combobox selection
  315. """
  316. if self.curitem is None:
  317. return
  318. self._selectTreeItem(self.curitem)
  319. self.Dismiss()
  320. evt.Skip()
  321. def SetData(self, **kargs):
  322. """Set object properties"""
  323. if 'multiple' in kargs:
  324. self.multiple = kargs['multiple']
  325. if 'onPopup' in kargs:
  326. self.onPopup = kargs['onPopup']
  327. if kargs.get('layerTree', None):
  328. self.filterItems = [] # reset
  329. ltype = kargs['type']
  330. for layer in kargs['layerTree'].GetVisibleLayers(
  331. skipDigitized=True):
  332. if layer.GetType() != ltype:
  333. continue
  334. self.filterItems.append(layer.GetName())
  335. def DeleteAllItems(self):
  336. """Delete all items in popup"""
  337. self.seltree.DeleteAllItems()
  338. class TreeCtrlComboPopup(ListCtrlComboPopup):
  339. """Create a tree ComboBox for selecting maps and other GIS elements
  340. in accessible mapsets within the current location
  341. """
  342. # overridden ComboPopup methods
  343. def Init(self):
  344. ListCtrlComboPopup.Init(self)
  345. self.nmaps = 1
  346. self.type = None
  347. self.mapsets = None
  348. self.onPopup = None
  349. self.fullyQualified = True
  350. self.extraItems = dict()
  351. self.SetFilter(None)
  352. self.tgis_error = False
  353. def SetFilter(self, filter):
  354. """Set filter for GIS elements, see e.g. VectorSelect"""
  355. self.filterElements = filter
  356. def OnPopup(self, force=False):
  357. """Limited only for first selected"""
  358. if not force and not self.updateOnPopup:
  359. return
  360. if self.onPopup:
  361. selected, exclude = self.onPopup(self.type)
  362. else:
  363. selected = None
  364. exclude = False
  365. self.GetElementList(selected, exclude)
  366. ListCtrlComboPopup.OnPopup(self, force)
  367. def GetElementList(self, elements=None, exclude=False):
  368. """Get filtered list of GIS elements in accessible mapsets
  369. and display as tree with all relevant elements displayed
  370. beneath each mapset branch
  371. """
  372. # update list
  373. self.seltree.DeleteAllItems()
  374. if self.type:
  375. self._getElementList(self.type, self.mapsets, elements, exclude)
  376. if len(self.value) > 0:
  377. root = self.seltree.GetRootItem()
  378. if not root:
  379. return
  380. item = self.FindItem(root, self.value[0])
  381. try:
  382. self.seltree.EnsureVisible(item)
  383. self.seltree.SelectItem(item)
  384. except:
  385. pass
  386. def _getElementList(self, element, mapsets=None,
  387. elements=None, exclude=False):
  388. """Get list of GIS elements in accessible mapsets and display as tree
  389. with all relevant elements displayed beneath each mapset branch
  390. :param element: GIS element
  391. :param mapsets: list of acceptable mapsets (None for all mapsets in search path)
  392. :param elements: list of forced GIS elements
  393. :param exclude: True to exclude, False for forcing the list (elements)
  394. """
  395. # get current mapset
  396. curr_mapset = grass.gisenv()['MAPSET']
  397. # map element types to g.list types
  398. elementdict = {'cell': 'raster',
  399. 'raster': 'raster',
  400. 'grid3': 'raster_3d',
  401. 'raster_3d': 'raster_3d',
  402. 'vector': 'vector',
  403. 'paint/labels': 'label',
  404. 'label': 'label',
  405. 'windows': 'region',
  406. 'region': 'region',
  407. 'group': 'group',
  408. 'stds': 'stds',
  409. 'strds': 'strds',
  410. 'str3ds': 'str3ds',
  411. 'stvds': 'stvds'}
  412. # to support multiple elements
  413. element_list = element.split(',')
  414. renamed_elements = []
  415. for elem in element_list:
  416. if elem not in elementdict:
  417. self.AddItem(_('Not selectable element'), node=False)
  418. return
  419. else:
  420. renamed_elements.append(elementdict[elem])
  421. if element in ('stds', 'strds', 'str3ds', 'stvds'):
  422. if not self.tgis_error:
  423. import grass.temporal as tgis
  424. filesdict = tgis.tlist_grouped(
  425. elementdict[element], element == 'stds')
  426. else:
  427. filesdict = None
  428. else:
  429. filesdict = grass.list_grouped(renamed_elements,
  430. check_search_path=False)
  431. # add extra items first
  432. if self.extraItems:
  433. for group, items in six.iteritems(self.extraItems):
  434. node = self.AddItem(group, node=True)
  435. self.seltree.SetItemTextColour(node, wx.Colour(50, 50, 200))
  436. for item in items:
  437. self.AddItem(item, node=False, parent=node)
  438. self.seltree.ExpandAllChildren(node)
  439. # list of mapsets in current location
  440. if mapsets is None:
  441. mapsets = grass.mapsets(search_path=True)
  442. # current mapset first
  443. if curr_mapset in mapsets and mapsets[0] != curr_mapset:
  444. mapsets.remove(curr_mapset)
  445. mapsets.insert(0, curr_mapset)
  446. first_mapset = None
  447. for mapset in mapsets:
  448. mapset_node = self.AddItem(
  449. _('Mapset') + ': ' + mapset, node=True, mapset=mapset)
  450. node = mapset_node
  451. if not first_mapset:
  452. first_mapset = mapset_node
  453. self.seltree.SetItemTextColour(mapset_node, wx.Colour(50, 50, 200))
  454. if mapset not in filesdict:
  455. continue
  456. try:
  457. if isinstance(filesdict[mapset], dict):
  458. for elementType in filesdict[mapset].keys():
  459. node = self.AddItem(
  460. _('Type: ') + elementType,
  461. mapset=mapset,
  462. node=True,
  463. parent=mapset_node)
  464. self.seltree.SetItemTextColour(
  465. node, wx.Colour(50, 50, 200))
  466. elem_list = filesdict[mapset][elementType]
  467. self._addItems(
  468. elist=elem_list,
  469. elements=elements,
  470. mapset=mapset,
  471. exclude=exclude,
  472. node=node)
  473. else:
  474. elem_list = filesdict[mapset]
  475. self._addItems(elist=elem_list, elements=elements,
  476. mapset=mapset, exclude=exclude, node=node)
  477. except Exception as e:
  478. sys.stderr.write(_("GSelect: invalid item: %s") % e)
  479. continue
  480. if self.seltree.ItemHasChildren(mapset_node):
  481. sel = UserSettings.Get(
  482. group='appearance',
  483. key='elementListExpand',
  484. subkey='selection')
  485. collapse = True
  486. if sel == 0: # collapse all except PERMANENT and current
  487. if mapset in ('PERMANENT', curr_mapset):
  488. collapse = False
  489. elif sel == 1: # collapse all except PERMANENT
  490. if mapset == 'PERMANENT':
  491. collapse = False
  492. elif sel == 2: # collapse all except current
  493. if mapset == curr_mapset:
  494. collapse = False
  495. elif sel == 3: # collapse all
  496. pass
  497. elif sel == 4: # expand all
  498. collapse = False
  499. if collapse:
  500. self.seltree.CollapseAllChildren(mapset_node)
  501. else:
  502. self.seltree.ExpandAllChildren(mapset_node)
  503. if first_mapset:
  504. # select first mapset (MSW hack)
  505. self.seltree.SelectItem(first_mapset)
  506. # helpers
  507. def _addItems(self, elist, elements, mapset, exclude, node):
  508. """Helper function for adding multiple items (maps, stds).
  509. :param list elist: list of map/stds names
  510. :param list elements: list of forced elements
  511. :param str mapset: mapset name
  512. :param exclude: True to exclude, False for forcing the list
  513. :param node: parent node
  514. """
  515. elist = grass.naturally_sorted(elist)
  516. for elem in elist:
  517. if elem != '':
  518. fullqElem = elem + '@' + mapset
  519. if self.filterItems and fullqElem not in self.filterItems:
  520. continue # skip items missed in self.filterItems
  521. if elements is not None:
  522. if (exclude and fullqElem in elements) or \
  523. (not exclude and fullqElem not in elements):
  524. continue
  525. if self.filterElements:
  526. if self.filterElements(fullqElem):
  527. self.AddItem(
  528. elem, mapset=mapset, node=False, parent=node)
  529. else:
  530. self.AddItem(elem, mapset=mapset, node=False, parent=node)
  531. def AddItem(self, value, mapset=None, node=True, parent=None):
  532. if not parent:
  533. root = self.seltree.GetRootItem()
  534. if not root:
  535. root = self.seltree.AddRoot("<hidden root>")
  536. parent = root
  537. data = {'node': node, 'mapset': mapset}
  538. item = self.seltree.AppendItem(
  539. parent, text=value, data=data)
  540. return item
  541. def OnKeyUp(self, event):
  542. """Enables to select items using keyboard
  543. Unused with wxPython 3, can be removed in the future.
  544. """
  545. item = self.seltree.GetSelection()
  546. if event.GetKeyCode() == wx.WXK_DOWN:
  547. self.seltree.SelectItem(self.seltree.GetNextVisible(item))
  548. # problem with GetPrevVisible
  549. elif event.GetKeyCode() == wx.WXK_UP:
  550. if self.seltree.ItemHasChildren(item) and self.seltree.IsExpanded(
  551. self.seltree.GetPrevSibling(item)):
  552. itemPrev = self.seltree.GetLastChild(
  553. self.seltree.GetPrevSibling(item))
  554. else:
  555. itemPrev = self.seltree.GetPrevSibling(item)
  556. if not wx.TreeItemId.IsOk(itemPrev):
  557. itemPrev = self.seltree.GetItemParent(item)
  558. if item == self.seltree.GetFirstChild(
  559. self.seltree.GetRootItem())[0]:
  560. itemPrev = item
  561. self.seltree.SelectItem(itemPrev)
  562. # selects first item starting with the written text in next mapset
  563. elif event.GetKeyCode() == wx.WXK_TAB:
  564. selected = self.seltree.GetSelection()
  565. if self.seltree.ItemHasChildren(selected):
  566. parent = selected
  567. else:
  568. parent = self.seltree.GetItemParent(selected)
  569. nextSibling = self.seltree.GetNextSibling(parent)
  570. if wx.TreeItemId.IsOk(nextSibling):
  571. match = self.FindItem(
  572. nextSibling, self.GetCombo().GetValue().strip(), True)
  573. else:
  574. match = self.FindItem(
  575. self.seltree.GetFirstChild(
  576. self.seltree.GetItemParent(parent))[0],
  577. self.GetCombo().GetValue().strip(),
  578. True)
  579. self.seltree.SelectItem(match)
  580. elif event.GetKeyCode() == wx.WXK_RIGHT:
  581. if self.seltree.ItemHasChildren(item):
  582. self.seltree.Expand(item)
  583. elif event.GetKeyCode() == wx.WXK_LEFT:
  584. if self.seltree.ItemHasChildren(item):
  585. self.seltree.Collapse(item)
  586. elif event.GetKeyCode() == wx.WXK_ESCAPE:
  587. self.Dismiss()
  588. elif event.GetKeyCode() == wx.WXK_RETURN:
  589. if self.seltree.GetItemData(item)['node']:
  590. self.value = []
  591. else:
  592. self._selectTreeItem(item)
  593. self.Dismiss()
  594. def OnLeftDown(self, evt):
  595. """Do the combobox selection
  596. """
  597. item, flags = self.seltree.HitTest(evt.GetPosition())
  598. if item and flags & wx.TREE_HITTEST_ONITEMLABEL:
  599. self.curitem = item
  600. if self.seltree.GetItemData(item)['node']:
  601. evt.Skip()
  602. return
  603. self._selectTreeItem(item)
  604. self.Dismiss()
  605. evt.Skip()
  606. def _selectTreeItem(self, item):
  607. fullName = self.seltree.GetItemText(item)
  608. if self.fullyQualified and self.seltree.GetItemData(item)['mapset']:
  609. fullName += '@' + self.seltree.GetItemData(item)['mapset']
  610. if self.multiple:
  611. self.value.append(fullName)
  612. else:
  613. if self.nmaps > 1: # see key_desc
  614. if len(self.value) >= self.nmaps:
  615. self.value = [fullName]
  616. else:
  617. self.value.append(fullName)
  618. else:
  619. self.value = [fullName]
  620. def _onItemConfirmed(self, event):
  621. item = event.GetItem()
  622. if self.seltree.GetItemData(item)['node']:
  623. self.value = []
  624. else:
  625. self._selectTreeItem(item)
  626. self.Dismiss()
  627. def SetData(self, **kargs):
  628. """Set object properties"""
  629. ListCtrlComboPopup.SetData(self, **kargs)
  630. if 'type' in kargs:
  631. self.type = kargs['type']
  632. if self.type in ('stds', 'strds', 'str3ds', 'stvds'):
  633. # Initiate the temporal framework. Catch database error
  634. # and set the error flag for the stds listing.
  635. try:
  636. import grass.temporal as tgis
  637. from grass.pygrass import messages
  638. try:
  639. tgis.init(True)
  640. except messages.FatalError as e:
  641. sys.stderr.write(_("Temporal GIS error:\n%s") % e)
  642. self.tgis_error = True
  643. except ImportError as e:
  644. # PyGRASS (ctypes) is the likely cause
  645. sys.stderr.write(_(
  646. "Unable to import pyGRASS: %s\n"
  647. "Some functionality will be not accessible") % e)
  648. self.tgis_error = True
  649. if 'mapsets' in kargs:
  650. self.mapsets = kargs['mapsets']
  651. if 'nmaps' in kargs:
  652. self.nmaps = kargs['nmaps']
  653. if 'updateOnPopup' in kargs:
  654. self.updateOnPopup = kargs['updateOnPopup']
  655. if 'fullyQualified' in kargs:
  656. self.fullyQualified = kargs['fullyQualified']
  657. if 'extraItems' in kargs:
  658. self.extraItems = kargs['extraItems']
  659. def GetType(self):
  660. """Get element type
  661. """
  662. return self.type
  663. class VectorDBInfo:
  664. """Class providing information about attribute tables
  665. linked to a vector map"""
  666. def __init__(self, map):
  667. self.map = map
  668. # dictionary of layer number and associated (driver, database, table)
  669. self.layers = {}
  670. # dictionary of table and associated columns (type, length, values,
  671. # ids)
  672. self.tables = {}
  673. if not self._CheckDBConnection(): # -> self.layers
  674. return
  675. self._DescribeTables() # -> self.tables
  676. def _CheckDBConnection(self):
  677. """Check DB connection"""
  678. nuldev = open(os.devnull, 'w+')
  679. # if map is not defined (happens with vnet initialization) or it
  680. # doesn't exist
  681. try:
  682. self.layers = grass.vector_db(map=self.map, stderr=nuldev)
  683. except CalledModuleError:
  684. return False
  685. finally: # always close nuldev
  686. nuldev.close()
  687. return bool(len(self.layers.keys()) > 0)
  688. def _DescribeTables(self):
  689. """Describe linked tables"""
  690. for layer in self.layers.keys():
  691. # determine column names and types
  692. table = self.layers[layer]["table"]
  693. columns = {} # {name: {type, length, [values], [ids]}}
  694. i = 0
  695. Debug.msg(
  696. 1,
  697. "gselect.VectorDBInfo._DescribeTables(): table=%s driver=%s database=%s" %
  698. (self.layers[layer]["table"],
  699. self.layers[layer]["driver"],
  700. self.layers[layer]["database"]))
  701. for item in grass.db_describe(
  702. table=self.layers[layer]["table"],
  703. driver=self.layers[layer]["driver"],
  704. database=self.layers[layer]["database"])['cols']:
  705. name, type, length = item
  706. # FIXME: support more datatypes
  707. if type.lower() == "integer":
  708. ctype = int
  709. elif type.lower() == "double precision":
  710. ctype = float
  711. else:
  712. ctype = str
  713. columns[name.strip()] = {'index': i,
  714. 'type': type.lower(),
  715. 'ctype': ctype,
  716. 'length': int(length),
  717. 'values': [],
  718. 'ids': []}
  719. i += 1
  720. # check for key column
  721. # v.db.connect -g/p returns always key column name lowercase
  722. if self.layers[layer]["key"] not in columns.keys():
  723. for col in columns.keys():
  724. if col.lower() == self.layers[layer]["key"]:
  725. self.layers[layer]["key"] = col.upper()
  726. break
  727. self.tables[table] = columns
  728. return True
  729. def Reset(self):
  730. """Reset"""
  731. for layer in self.layers:
  732. table = self.layers[layer]["table"] # get table desc
  733. for name in self.tables[table].keys():
  734. self.tables[table][name]['values'] = []
  735. self.tables[table][name]['ids'] = []
  736. def GetName(self):
  737. """Get vector name"""
  738. return self.map
  739. def GetKeyColumn(self, layer):
  740. """Get key column of given layer
  741. :param layer: vector layer number
  742. """
  743. return str(self.layers[layer]['key'])
  744. def GetTable(self, layer):
  745. """Get table name of given layer
  746. :param layer: vector layer number
  747. """
  748. if layer not in self.layers:
  749. raise GException(_("No table linked to layer <{}>.".format(layer)))
  750. return self.layers[layer]['table']
  751. def GetDbSettings(self, layer):
  752. """Get database settings
  753. :param layer: layer number
  754. :return: (driver, database)
  755. """
  756. return self.layers[layer]['driver'], self.layers[layer]['database']
  757. def GetTableDesc(self, table):
  758. """Get table columns
  759. :param table: table name
  760. """
  761. return self.tables[table]
  762. class LayerSelect(wx.ComboBox):
  763. def __init__(self, parent, id=wx.ID_ANY,
  764. size=globalvar.DIALOG_COMBOBOX_SIZE,
  765. vector=None, dsn=None, choices=[], all=False, default=None):
  766. """Creates combo box for selecting vector map layer names
  767. :param str vector: vector map name (native or connected via v.external)
  768. :param str dsn: OGR data source name
  769. """
  770. super(
  771. LayerSelect,
  772. self).__init__(
  773. parent,
  774. id,
  775. size=size,
  776. choices=choices)
  777. self.all = all
  778. self.SetName("LayerSelect")
  779. # default value
  780. self.default = default
  781. self.InsertLayers(vector=vector, dsn=dsn)
  782. def InsertLayers(self, vector=None, dsn=None):
  783. """Insert layers for a vector into the layer combobox
  784. :param str vector: vector map name (native or connected via v.external)
  785. :param str dsn: OGR data source name
  786. """
  787. layers = list()
  788. if vector:
  789. layers = GetVectorNumberOfLayers(vector)
  790. elif dsn:
  791. ret = RunCommand('v.in.ogr',
  792. read=True,
  793. quiet=True,
  794. flags='l',
  795. input=dsn)
  796. if ret:
  797. layers = ret.splitlines()
  798. if self.default:
  799. if len(layers) == 0:
  800. layers.insert(0, str(self.default))
  801. elif self.default not in layers:
  802. layers.append(self.default)
  803. if self.all:
  804. layers.insert(0, '-1')
  805. if len(layers) > 0:
  806. self.SetItems(layers)
  807. else:
  808. self.Clear()
  809. self.SetValue("")
  810. if self.default and self.default in layers:
  811. self.SetValue(self.default)
  812. class DriverSelect(wx.ComboBox):
  813. """Creates combo box for selecting database driver.
  814. """
  815. def __init__(self, parent, choices, value,
  816. id=wx.ID_ANY, pos=wx.DefaultPosition,
  817. size=globalvar.DIALOG_LAYER_SIZE, **kargs):
  818. super(DriverSelect, self).__init__(parent, id, value, pos, size,
  819. choices, style=wx.CB_READONLY)
  820. self.SetName("DriverSelect")
  821. self.SetStringSelection(value)
  822. class DatabaseSelect(TextCtrl):
  823. """Creates combo box for selecting database driver.
  824. """
  825. def __init__(self, parent, value='', id=wx.ID_ANY,
  826. size=globalvar.DIALOG_TEXTCTRL_SIZE, **kargs):
  827. super(
  828. DatabaseSelect,
  829. self).__init__(
  830. parent,
  831. id,
  832. value,
  833. size=size,
  834. **kargs)
  835. self.SetName("DatabaseSelect")
  836. class TableSelect(wx.ComboBox):
  837. """Creates combo box for selecting attribute tables from the database
  838. """
  839. def __init__(self, parent,
  840. id=wx.ID_ANY, value='',
  841. size=globalvar.DIALOG_COMBOBOX_SIZE, choices=[], **kargs):
  842. super(
  843. TableSelect,
  844. self).__init__(
  845. parent,
  846. id,
  847. value,
  848. size=size,
  849. choices=choices,
  850. style=wx.CB_READONLY,
  851. **kargs)
  852. self.SetName("TableSelect")
  853. if not choices:
  854. self.InsertTables()
  855. def InsertTables(self, driver=None, database=None):
  856. """Insert attribute tables into combobox"""
  857. items = []
  858. if not driver or not database:
  859. connect = grass.db_connection()
  860. driver = connect['driver']
  861. database = connect['database']
  862. ret = RunCommand('db.tables',
  863. flags='p',
  864. read=True,
  865. driver=driver,
  866. database=database)
  867. if ret:
  868. for table in ret.splitlines():
  869. items.append(table)
  870. self.SetItems(items)
  871. self.SetValue('')
  872. class ColumnSelect(ComboCtrl):
  873. """Creates combo box for selecting columns in the attribute table
  874. for a vector map.
  875. :param parent: window parent
  876. :param id: window id
  877. :param value: default value
  878. :param size: window size
  879. :param str vector: vector map name
  880. :param layer: layer number
  881. :param multiple: True if it is possible to add multiple columns
  882. :param param: parameters list (see menuform.py)
  883. :param kwags: wx.ComboBox parameters
  884. """
  885. def __init__(self, parent, id=wx.ID_ANY, value='',
  886. size=globalvar.DIALOG_COMBOBOX_SIZE,
  887. vector=None, layer=1, multiple=False,
  888. param=None, **kwargs):
  889. self.defaultValue = value
  890. self.param = param
  891. self.columns = []
  892. ComboCtrl.__init__(self, parent, id, size=size, **kwargs)
  893. self.GetChildren()[0].SetName("ColumnSelect")
  894. self.GetChildren()[0].type = type
  895. self.tcp = ListCtrlComboPopup()
  896. self.SetPopupControl(self.tcp)
  897. self.tcp.SetData(multiple=multiple)
  898. if vector:
  899. self.InsertColumns(vector, layer)
  900. self.GetChildren()[0].Bind(wx.EVT_KEY_UP, self.OnKeyUp)
  901. def GetColumns(self):
  902. return self.columns
  903. def OnKeyUp(self, event):
  904. """Shows popupwindow if down arrow key is released"""
  905. if event.GetKeyCode() == wx.WXK_DOWN and not self.IsPopupShown():
  906. self.ShowPopup()
  907. else:
  908. event.Skip()
  909. def Clear(self):
  910. self.tcp.DeleteAllItems()
  911. self.SetValue('')
  912. def InsertColumns(self, vector, layer, excludeKey=False,
  913. excludeCols=None, type=None, dbInfo=None):
  914. """Insert columns for a vector attribute table into the columns combobox
  915. :param str vector: vector name
  916. :param int layer: vector layer number
  917. :param excludeKey: exclude key column from the list?
  918. :param excludeCols: list of columns to be removed from the list
  919. :param type: only columns of given type (given as list)
  920. """
  921. if not dbInfo:
  922. dbInfo = VectorDBInfo(vector)
  923. try:
  924. try:
  925. layer = int(layer)
  926. except TypeError:
  927. # assuming layer 1
  928. layer = 1
  929. table = dbInfo.GetTable(layer)
  930. columnchoices = dbInfo.GetTableDesc(table)
  931. keyColumn = dbInfo.GetKeyColumn(layer)
  932. self.columns = len(columnchoices.keys()) * ['']
  933. for key, val in six.iteritems(columnchoices):
  934. self.columns[val['index']] = key
  935. if excludeKey: # exclude key column
  936. self.columns.remove(keyColumn)
  937. if excludeCols: # exclude key column
  938. for key in six.iterkeys(columnchoices):
  939. if key in excludeCols:
  940. self.columns.remove(key)
  941. if type: # only selected column types
  942. for key, value in six.iteritems(columnchoices):
  943. if value['type'] not in type:
  944. try:
  945. self.columns.remove(key)
  946. except ValueError:
  947. pass
  948. except (KeyError, ValueError, GException):
  949. self.columns[:] = []
  950. # update list
  951. self.tcp.DeleteAllItems()
  952. for col in self.columns:
  953. self.tcp.AddItem(col)
  954. self.SetValue(self.defaultValue)
  955. if self.param:
  956. value = self.param.get('value', '')
  957. if value != '' and value in self.columns:
  958. self.SetValue(value)
  959. def InsertTableColumns(self, table, driver=None, database=None):
  960. """Insert table columns
  961. :param str table: table name
  962. :param str driver: driver name
  963. :param str database: database name
  964. """
  965. self.columns[:] = []
  966. ret = RunCommand('db.columns',
  967. read=True,
  968. driver=driver,
  969. database=database,
  970. table=table)
  971. if ret:
  972. self.columns = ret.splitlines()
  973. # update list
  974. self.tcp.DeleteAllItems()
  975. self.SetValue(self.defaultValue)
  976. for col in self.columns:
  977. self.tcp.AddItem(col)
  978. if self.param:
  979. value = self.param.get('value', '')
  980. if value != '' and value in self.columns:
  981. self.SetValue(value)
  982. class DbaseSelect(wx.lib.filebrowsebutton.DirBrowseButton):
  983. """Widget for selecting GRASS Database"""
  984. def __init__(self, parent, **kwargs):
  985. super(
  986. DbaseSelect,
  987. self).__init__(
  988. parent,
  989. id=wx.ID_ANY,
  990. size=globalvar.DIALOG_GSELECT_SIZE,
  991. labelText='',
  992. dialogTitle=_('Choose GIS Data Directory'),
  993. buttonText=_('Browse'),
  994. startDirectory=grass.gisenv()['GISDBASE'],
  995. **kwargs)
  996. class LocationSelect(wx.ComboBox):
  997. """Widget for selecting GRASS location"""
  998. def __init__(
  999. self, parent, id=wx.ID_ANY, size=globalvar.DIALOG_COMBOBOX_SIZE,
  1000. gisdbase=None, **kwargs):
  1001. super(LocationSelect, self).__init__(parent, id, size=size, **kwargs)
  1002. self.SetName("LocationSelect")
  1003. if not gisdbase:
  1004. self.gisdbase = grass.gisenv()['GISDBASE']
  1005. else:
  1006. self.gisdbase = gisdbase
  1007. self.SetItems(GetListOfLocations(self.gisdbase))
  1008. def UpdateItems(self, dbase):
  1009. """Update list of locations
  1010. :param str dbase: path to GIS database
  1011. """
  1012. self.gisdbase = dbase
  1013. if dbase:
  1014. self.SetItems(GetListOfLocations(self.gisdbase))
  1015. else:
  1016. self.SetItems([])
  1017. class MapsetSelect(wx.ComboBox):
  1018. """Widget for selecting GRASS mapset"""
  1019. def __init__(self, parent, id=wx.ID_ANY,
  1020. size=globalvar.DIALOG_COMBOBOX_SIZE, gisdbase=None,
  1021. location=None, setItems=True, searchPath=False, new=False,
  1022. skipCurrent=False, multiple=False, **kwargs):
  1023. style = 0
  1024. # disabled, read-only widget has no TextCtrl children (TODO: rewrite)
  1025. # if not new and not multiple:
  1026. ### style = wx.CB_READONLY
  1027. wx.ComboBox.__init__(self, parent, id, size=size,
  1028. style=style, **kwargs)
  1029. self.searchPath = searchPath
  1030. self.skipCurrent = skipCurrent
  1031. self.SetName("MapsetSelect")
  1032. self.value = ''
  1033. self.multiple = multiple
  1034. if not gisdbase:
  1035. self.gisdbase = grass.gisenv()['GISDBASE']
  1036. else:
  1037. self.gisdbase = gisdbase
  1038. if not location:
  1039. self.location = grass.gisenv()['LOCATION_NAME']
  1040. else:
  1041. self.location = location
  1042. if setItems:
  1043. self.SetItems(self._getMapsets())
  1044. if self.multiple:
  1045. self.Bind(wx.EVT_COMBOBOX, self._onSelection)
  1046. self.Bind(wx.EVT_TEXT, self._onSelection)
  1047. def _onSelection(self, event):
  1048. value = self.GetValue()
  1049. if value:
  1050. if self.value:
  1051. self.value += ','
  1052. self.value += value
  1053. self.SetValue(self.value)
  1054. else:
  1055. self.value = value
  1056. event.Skip()
  1057. def UpdateItems(self, location, dbase=None):
  1058. """Update list of mapsets for given location
  1059. :param str dbase: path to GIS database (None to use currently
  1060. selected)
  1061. :param str location: name of location
  1062. """
  1063. if dbase:
  1064. self.gisdbase = dbase
  1065. self.location = location
  1066. if location:
  1067. self.SetItems(self._getMapsets())
  1068. else:
  1069. self.SetItems([])
  1070. def _getMapsets(self):
  1071. if self.searchPath:
  1072. mlist = RunCommand('g.mapsets',
  1073. read=True, flags='p',
  1074. sep='newline').splitlines()
  1075. else:
  1076. mlist = GetListOfMapsets(self.gisdbase, self.location,
  1077. selectable=False)
  1078. gisenv = grass.gisenv()
  1079. if self.skipCurrent and \
  1080. gisenv['LOCATION_NAME'] == self.location and \
  1081. gisenv['MAPSET'] in mlist:
  1082. mlist.remove(gisenv['MAPSET'])
  1083. return mlist
  1084. class SubGroupSelect(wx.ComboBox):
  1085. """Widget for selecting subgroups"""
  1086. def __init__(self, parent, id=wx.ID_ANY,
  1087. size=globalvar.DIALOG_GSELECT_SIZE, **kwargs):
  1088. super(SubGroupSelect, self).__init__(parent, id, size=size,
  1089. **kwargs)
  1090. self.SetName("SubGroupSelect")
  1091. def Insert(self, group):
  1092. """Insert subgroups for defined group"""
  1093. if not group:
  1094. return
  1095. gisenv = grass.gisenv()
  1096. try:
  1097. name, mapset = group.split('@', 1)
  1098. except ValueError:
  1099. name = group
  1100. mapset = gisenv['MAPSET']
  1101. mlist = RunCommand('i.group', group=group,
  1102. read=True, flags='sg').splitlines()
  1103. try:
  1104. self.SetItems(mlist)
  1105. except OSError:
  1106. self.SetItems([])
  1107. class FormatSelect(wx.Choice):
  1108. def __init__(self, parent, srcType, ogr=False,
  1109. size=globalvar.DIALOG_SPIN_SIZE,
  1110. **kwargs):
  1111. """Widget for selecting external (GDAL/OGR) format
  1112. :param parent: parent window
  1113. :param srcType: source type ('file', 'database', 'protocol')
  1114. :param ogr: True for OGR otherwise GDAL
  1115. """
  1116. super(FormatSelect, self).__init__(parent, id=wx.ID_ANY, size=size,
  1117. **kwargs)
  1118. self.SetName("FormatSelect")
  1119. if ogr:
  1120. ftype = 'ogr'
  1121. else:
  1122. ftype = 'gdal'
  1123. formats = list()
  1124. for f in GetFormats()[ftype][srcType].values():
  1125. formats += f
  1126. self.SetItems(formats)
  1127. def GetExtension(self, name):
  1128. """Get file extension by format name"""
  1129. formatToExt = dict()
  1130. formatToExt.update(rasterFormatExtension)
  1131. formatToExt.update(vectorFormatExtension)
  1132. return formatToExt.get(name, '')
  1133. class GdalSelect(wx.Panel):
  1134. def __init__(self, parent, panel, ogr=False, link=False, dest=False,
  1135. exclude=None, settings=True):
  1136. """Widget for selecting GDAL/OGR datasource, format
  1137. .. todo::
  1138. Split into GdalSelect and OgrSelect and optionally to
  1139. GdalSelectOutput, OgrSelectOutput
  1140. :param parent: parent window
  1141. :param bool ogr: use OGR selector instead of GDAL
  1142. :param bool dest: True for output (destination)
  1143. :param default: default type (ignored when dest == True)
  1144. :param exclude: list of types to be excluded
  1145. """
  1146. self.parent = parent
  1147. self.ogr = ogr
  1148. self.link = link
  1149. self.dest = dest
  1150. self._sourceType = None
  1151. wx.Panel.__init__(self, parent=panel, name='GdalSelect')
  1152. self.reloadDataRequired = Signal('GdalSelect.reloadDataRequired')
  1153. self.inputBox = StaticBox(parent=self)
  1154. if dest:
  1155. self.inputBox.SetLabel(" %s " % _("Output settings"))
  1156. else:
  1157. self.inputBox.SetLabel(" %s " % _("Source input"))
  1158. # source type
  1159. sources = list()
  1160. self.sourceMap = {'file': -1,
  1161. 'dir': -1,
  1162. 'db': -1,
  1163. 'pro': -1,
  1164. 'native': -1}
  1165. idx = 0
  1166. if dest:
  1167. sources.append(_("Native"))
  1168. self.sourceMap['native'] = idx
  1169. idx += 1
  1170. if exclude is None:
  1171. exclude = []
  1172. if 'file' not in exclude:
  1173. sources.append(_("File"))
  1174. self.sourceMap['file'] = idx
  1175. idx += 1
  1176. if 'directory' not in exclude:
  1177. sources.append(_("Directory"))
  1178. self.sourceMap['dir'] = idx
  1179. idx += 1
  1180. if 'database' not in exclude:
  1181. sources.append(_("Database"))
  1182. self.sourceMap['db'] = idx
  1183. idx += 1
  1184. if 'protocol' not in exclude:
  1185. sources.append(_("Protocol"))
  1186. self.sourceMap['pro'] = idx
  1187. idx += 1
  1188. self.sourceMapByIdx = {}
  1189. for name, idx in self.sourceMap.items():
  1190. self.sourceMapByIdx[idx] = name
  1191. self.source = wx.RadioBox(parent=self, id=wx.ID_ANY,
  1192. style=wx.RA_SPECIFY_COLS,
  1193. choices=sources)
  1194. if dest:
  1195. self.source.SetLabel(" %s " % _('Output type'))
  1196. else:
  1197. self.source.SetLabel(" %s " % _('Source type'))
  1198. self.source.SetSelection(0)
  1199. self.source.Bind(
  1200. wx.EVT_RADIOBOX,
  1201. lambda evt: self.SetSourceType(
  1202. self.sourceMapByIdx[
  1203. evt.GetInt()]))
  1204. self.nativeWidgets = {}
  1205. self.fileWidgets = {}
  1206. self.dirWidgets = {}
  1207. self.dbWidgets = {}
  1208. self.protocolWidgets = {}
  1209. self.pgWidgets = {}
  1210. if ogr:
  1211. fType = 'ogr'
  1212. else:
  1213. fType = 'gdal'
  1214. # file
  1215. fileMask = '%(all)s (*)|*|' % {'all': _('All files')}
  1216. if not ogr:
  1217. extList = rasterFormatExtension
  1218. fileMask += ('%(name)s (*.%(low1)s;*.%(low2)s;*.%(up1)s;*.%(up2)s)|'
  1219. '*.%(low1)s;*.%(low2)s;*.%(up1)s;*.%(up2)s|' %
  1220. {'name': 'GeoTIFF', 'low1': 'tif', 'low2': 'tiff', 'up1': 'TIF', 'up2': 'TIFF'})
  1221. else:
  1222. extList = vectorFormatExtension
  1223. fileMask += '%(name)s (*.%(low)s;*.%(up)s)|*.%(low)s;*.%(up)s|' % {
  1224. 'name': 'ESRI Shapefile', 'low': 'shp', 'up': 'SHP'}
  1225. for name, ext in sorted(extList.items()):
  1226. if name in ('ESRI Shapefile', 'GeoTIFF'):
  1227. continue
  1228. fileMask += '%(name)s (*.%(low)s;*.%(up)s)|*.%(low)s;*.%(up)s|' % {
  1229. 'name': name, 'low': ext.lower(), 'up': ext.upper()}
  1230. fileMask += '%s (*.zip;*.ZIP)|*.zip;*.ZIP|' % _('ZIP files')
  1231. fileMask += '%s (*.gz;*.GZ)|*.gz;*.GZ|' % _('GZIP files')
  1232. fileMask += '%s (*.tar;*.TAR)|*.tar;*.TAR|' % _('TAR files')
  1233. # don't include last '|' - windows and mac throw error
  1234. fileMask += '%s (*.tar.gz;*.TAR.GZ;*.tgz;*.TGZ)|*.tar.gz;*.TAR.GZ;*.tgz;*.TGZ' % _('TARGZ files')
  1235. # only contains formats with extensions hardcoded
  1236. self.filePanel = wx.Panel(parent=self)
  1237. browse = filebrowse.FileBrowseButton(
  1238. parent=self.filePanel,
  1239. id=wx.ID_ANY,
  1240. size=globalvar.DIALOG_GSELECT_SIZE,
  1241. labelText=_('File:'),
  1242. dialogTitle=_('Choose file to import'),
  1243. buttonText=_('Browse'),
  1244. startDirectory=os.getcwd(),
  1245. changeCallback=self.OnUpdate,
  1246. fileMask=fileMask)
  1247. browse.GetChildren()[1].SetName('GdalSelectDataSource')
  1248. self.fileWidgets['browse'] = browse
  1249. self.fileWidgets['options'] = TextCtrl(parent=self.filePanel)
  1250. # directory
  1251. self.dirPanel = wx.Panel(parent=self)
  1252. browse = filebrowse.DirBrowseButton(
  1253. parent=self.dirPanel,
  1254. id=wx.ID_ANY,
  1255. size=globalvar.DIALOG_GSELECT_SIZE,
  1256. labelText=_('Directory:'),
  1257. dialogTitle=_('Choose input directory'),
  1258. buttonText=_('Browse'),
  1259. startDirectory=os.getcwd(),
  1260. changeCallback=self.OnUpdate)
  1261. browse.GetChildren()[1].SetName('GdalSelectDataSource')
  1262. self.dirWidgets['browse'] = browse
  1263. formatSelect = wx.Choice(parent=self.dirPanel)
  1264. self.dirWidgets['format'] = formatSelect
  1265. fileFormats = GetFormats(writableOnly=dest)[fType]['file']
  1266. formatSelect.SetItems(sorted(list(fileFormats)))
  1267. formatSelect.Bind(
  1268. wx.EVT_CHOICE, lambda evt: self.SetExtension(
  1269. self.dirWidgets['format'].GetStringSelection()))
  1270. formatSelect.Bind(wx.EVT_CHOICE, self.OnUpdate)
  1271. self.dirWidgets['extensionLabel'] = StaticText(
  1272. parent=self.dirPanel, label=_("Extension:"))
  1273. self.dirWidgets['extension'] = TextCtrl(parent=self.dirPanel)
  1274. self.dirWidgets['extension'].Bind(wx.EVT_TEXT, self.ExtensionChanged)
  1275. self.dirWidgets['options'] = TextCtrl(parent=self.dirPanel)
  1276. if self.ogr:
  1277. shapefile = 'ESRI Shapefile'
  1278. if shapefile in fileFormats:
  1279. formatSelect.SetStringSelection(shapefile)
  1280. self.SetExtension(shapefile)
  1281. else:
  1282. tiff = 'GeoTIFF'
  1283. if tiff in fileFormats:
  1284. formatSelect.SetStringSelection(tiff)
  1285. self.SetExtension(tiff)
  1286. # database
  1287. self.dbPanel = wx.Panel(parent=self)
  1288. self.dbFormats = GetFormats(writableOnly=dest)[fType]['database']
  1289. dbChoice = wx.Choice(parent=self.dbPanel, choices=self.dbFormats)
  1290. dbChoice.Bind(
  1291. wx.EVT_CHOICE,
  1292. lambda evt: self.SetDatabase(
  1293. db=dbChoice.GetStringSelection()))
  1294. self.dbWidgets['format'] = dbChoice
  1295. browse = filebrowse.FileBrowseButton(
  1296. parent=self.dbPanel,
  1297. id=wx.ID_ANY,
  1298. size=globalvar.DIALOG_GSELECT_SIZE,
  1299. labelText=_("Name:"),
  1300. dialogTitle=_('Choose file'),
  1301. buttonText=_('Browse'),
  1302. startDirectory=os.getcwd(),
  1303. changeCallback=self.OnUpdate)
  1304. browse.GetChildren()[1].SetName('GdalSelectDataSource')
  1305. self.dbWidgets['browse'] = browse
  1306. self.dbWidgets['choice'] = wx.Choice(
  1307. parent=self.dbPanel, name='GdalSelectDataSource')
  1308. self.dbWidgets['choice'].Bind(wx.EVT_CHOICE, self.OnUpdate)
  1309. self.dbWidgets['text'] = TextCtrl(
  1310. parent=self.dbPanel, name='GdalSelectDataSource')
  1311. self.dbWidgets['text'].Bind(wx.EVT_TEXT, self.OnUpdate)
  1312. self.dbWidgets['textLabel1'] = StaticText(
  1313. parent=self.dbPanel, label=_("Name:"))
  1314. self.dbWidgets['textLabel2'] = StaticText(
  1315. parent=self.dbPanel, label=_("Name:"))
  1316. self.dbWidgets['featType'] = wx.RadioBox(
  1317. parent=self.dbPanel,
  1318. id=wx.ID_ANY,
  1319. label=" %s " %
  1320. _("Feature type:"),
  1321. choices=[
  1322. _("simple features"),
  1323. _("topological")],
  1324. majorDimension=2,
  1325. style=wx.RA_SPECIFY_COLS)
  1326. if dest:
  1327. self.dbWidgets['featType'].Disable()
  1328. else:
  1329. self.dbWidgets['featType'].Hide()
  1330. browse = filebrowse.DirBrowseButton(
  1331. parent=self.dbPanel,
  1332. id=wx.ID_ANY,
  1333. size=globalvar.DIALOG_GSELECT_SIZE,
  1334. labelText=_('Directory:'),
  1335. dialogTitle=_('Choose input directory'),
  1336. buttonText=_('Browse'),
  1337. startDirectory=os.getcwd(),
  1338. changeCallback=self.OnUpdate)
  1339. self.dbWidgets['dirbrowse'] = browse
  1340. self.dbWidgets['options'] = TextCtrl(parent=self.dbPanel)
  1341. # protocol
  1342. self.protocolPanel = wx.Panel(parent=self)
  1343. protocolFormats = GetFormats(writableOnly=self.dest)[fType]['protocol']
  1344. protocolChoice = wx.Choice(
  1345. parent=self.protocolPanel,
  1346. choices=protocolFormats)
  1347. self.protocolWidgets['format'] = protocolChoice
  1348. self.protocolWidgets['text'] = TextCtrl(parent=self.protocolPanel)
  1349. self.protocolWidgets['text'].Bind(wx.EVT_TEXT, self.OnUpdate)
  1350. self.protocolWidgets['options'] = TextCtrl(
  1351. parent=self.protocolPanel)
  1352. # native
  1353. self.nativePanel = wx.Panel(parent=self)
  1354. self._layout()
  1355. sourceType = 'file'
  1356. self.SetSourceType(sourceType) # needed always to fit dialog size
  1357. if self.dest:
  1358. current = RunCommand('v.external.out',
  1359. parent=self,
  1360. read=True, parse=grass.parse_key_val,
  1361. flags='g')
  1362. if current['format'] == 'native':
  1363. sourceType = 'native'
  1364. elif current['format'] in GetFormats()['ogr']['database']:
  1365. sourceType = 'db'
  1366. else:
  1367. sourceType = 'dir'
  1368. if self.dest:
  1369. wx.CallAfter(self._postInit, sourceType, current)
  1370. def _postInit(self, sourceType, data):
  1371. """Fill in default values."""
  1372. format = data.get('format', '')
  1373. pg = 'conninfo' in data.keys()
  1374. if pg:
  1375. dsn = ''
  1376. for item in data.get('conninfo').split(' '):
  1377. k, v = item.split('=')
  1378. if k == 'dbname':
  1379. dsn = v
  1380. break
  1381. optList = list()
  1382. for k, v in six.iteritems(data):
  1383. if k in ('format', 'conninfo', 'topology'):
  1384. continue
  1385. optList.append('%s=%s' % (k, v))
  1386. options = ','.join(optList)
  1387. else:
  1388. dsn = data.get('dsn')
  1389. options = data.get('options', '')
  1390. self.SetSourceType(sourceType)
  1391. self.source.SetSelection(self.sourceMap[sourceType])
  1392. # v.external.out does not return dsn for the native format
  1393. if dsn:
  1394. dsn = os.path.expandvars(dsn) # v.external.out uses $HOME
  1395. # fill in default values
  1396. if sourceType == 'dir':
  1397. self.dirWidgets['format'].SetStringSelection(format)
  1398. self.dirWidgets['browse'].SetValue(dsn)
  1399. self.dirWidgets['options'].SetValue(options)
  1400. elif sourceType == 'db':
  1401. self.dbWidgets['format'].SetStringSelection(format)
  1402. self.dbWidgets['options'].SetValue(options)
  1403. name = self._getCurrentDbWidgetName()
  1404. if name == 'choice':
  1405. if dsn in self.dbWidgets[name].GetItems():
  1406. self.dbWidgets[name].SetStringSelection(dsn)
  1407. if 'topology' in data.keys():
  1408. self.dbWidgets['featType'].SetSelection(1)
  1409. else:
  1410. self.dbWidgets[name].SetValue(dsn)
  1411. def _layout(self):
  1412. """Layout"""
  1413. self.mainSizer = wx.BoxSizer(wx.VERTICAL)
  1414. self.changingSizer = wx.StaticBoxSizer(self.inputBox, wx.VERTICAL)
  1415. # file
  1416. paddingSizer = wx.BoxSizer(wx.VERTICAL)
  1417. sizer = wx.GridBagSizer(vgap=5, hgap=10)
  1418. paddingSizer.Add(self.fileWidgets['browse'],
  1419. flag=wx.BOTTOM | wx.EXPAND,
  1420. border=35)
  1421. sizer.Add(paddingSizer, flag=wx.EXPAND, pos=(0, 0), span=(1, 2))
  1422. sizer.AddGrowableCol(0)
  1423. if self.dest:
  1424. sizer.Add(StaticText(parent=self.filePanel,
  1425. label=_("Creation options:")),
  1426. flag=wx.ALIGN_CENTER_VERTICAL,
  1427. pos=(1, 0))
  1428. sizer.Add(self.fileWidgets['options'],
  1429. flag=wx.ALIGN_CENTER_VERTICAL | wx.EXPAND,
  1430. pos=(1, 1))
  1431. else:
  1432. self.fileWidgets['options'].Hide()
  1433. self.filePanel.SetSizer(sizer)
  1434. # directory
  1435. sizer = wx.GridBagSizer(vgap=3, hgap=10)
  1436. sizer.Add(StaticText(parent=self.dirPanel,
  1437. label=_("Format:")),
  1438. flag=wx.ALIGN_CENTER_VERTICAL,
  1439. pos=(0, 0))
  1440. sizer.Add(self.dirWidgets['format'],
  1441. flag=wx.ALIGN_CENTER_VERTICAL | wx.EXPAND,
  1442. pos=(0, 1))
  1443. sizer.Add(self.dirWidgets['extensionLabel'],
  1444. flag=wx.ALIGN_CENTER_VERTICAL,
  1445. pos=(0, 2))
  1446. sizer.Add(self.dirWidgets['extension'],
  1447. flag=wx.ALIGN_CENTER_VERTICAL,
  1448. pos=(0, 3))
  1449. sizer.Add(self.dirWidgets['browse'],
  1450. flag=wx.ALIGN_CENTER_VERTICAL | wx.EXPAND,
  1451. pos=(1, 0), span=(1, 4))
  1452. if self.dest:
  1453. sizer.Add(StaticText(parent=self.dirPanel,
  1454. label=_("Creation options:")),
  1455. flag=wx.ALIGN_CENTER_VERTICAL,
  1456. pos=(2, 0))
  1457. sizer.Add(self.dirWidgets['options'],
  1458. flag=wx.ALIGN_CENTER_VERTICAL | wx.EXPAND,
  1459. pos=(2, 1))
  1460. helpBtn = Button(parent=self.dirPanel, id=wx.ID_HELP)
  1461. helpBtn.Bind(wx.EVT_BUTTON, self.OnHelp)
  1462. sizer.Add(helpBtn,
  1463. flag=wx.ALIGN_CENTER_VERTICAL | wx.EXPAND,
  1464. pos=(2, 2))
  1465. self.dirWidgets['extensionLabel'].Hide()
  1466. self.dirWidgets['extension'].Hide()
  1467. else:
  1468. self.dirWidgets['options'].Hide()
  1469. sizer.AddGrowableCol(1)
  1470. self.dirPanel.SetSizer(sizer)
  1471. # database
  1472. sizer = wx.GridBagSizer(vgap=1, hgap=5)
  1473. sizer.Add(StaticText(parent=self.dbPanel,
  1474. label=_("Format:")),
  1475. flag=wx.ALIGN_CENTER_VERTICAL,
  1476. pos=(0, 0))
  1477. sizer.Add(self.dbWidgets['format'],
  1478. flag=wx.ALIGN_CENTER_VERTICAL,
  1479. pos=(0, 1))
  1480. sizer.Add(self.dbWidgets['textLabel1'],
  1481. flag=wx.ALIGN_CENTER_VERTICAL,
  1482. pos=(1, 0))
  1483. sizer.Add(self.dbWidgets['text'],
  1484. flag=wx.ALIGN_CENTER_VERTICAL | wx.EXPAND,
  1485. pos=(1, 1), span=(1, 2))
  1486. sizer.Add(self.dbWidgets['browse'],
  1487. flag=wx.ALIGN_CENTER_VERTICAL | wx.EXPAND,
  1488. pos=(2, 0), span=(1, 3))
  1489. sizer.Add(self.dbWidgets['dirbrowse'],
  1490. flag=wx.ALIGN_CENTER_VERTICAL | wx.EXPAND,
  1491. pos=(3, 0), span=(1, 2))
  1492. sizer.Add(self.dbWidgets['textLabel2'],
  1493. flag=wx.ALIGN_CENTER_VERTICAL,
  1494. pos=(4, 0))
  1495. sizer.Add(self.dbWidgets['choice'],
  1496. flag=wx.ALIGN_CENTER_VERTICAL | wx.EXPAND,
  1497. pos=(4, 1), span=(1, 2))
  1498. if self.dest:
  1499. sizer.Add(self.dbWidgets['featType'],
  1500. pos=(0, 2), flag=wx.EXPAND)
  1501. sizer.Add(StaticText(parent=self.dbPanel,
  1502. label=_("Creation options:")),
  1503. flag=wx.ALIGN_CENTER_VERTICAL,
  1504. pos=(5, 0))
  1505. sizer.Add(self.dbWidgets['options'],
  1506. flag=wx.ALIGN_CENTER_VERTICAL | wx.EXPAND,
  1507. pos=(5, 1), span=(1, 2))
  1508. # help button
  1509. helpBtn = Button(parent=self.dbPanel, id=wx.ID_HELP)
  1510. helpBtn.Bind(wx.EVT_BUTTON, self.OnHelp)
  1511. sizer.Add(helpBtn,
  1512. pos=(5, 3))
  1513. else:
  1514. self.dbWidgets['options'].Hide()
  1515. self.dbPanel.SetSizer(sizer)
  1516. sizer.SetEmptyCellSize((0, 0))
  1517. sizer.AddGrowableCol(1)
  1518. # protocol
  1519. sizer = wx.GridBagSizer(vgap=3, hgap=3)
  1520. sizer.Add(StaticText(parent=self.protocolPanel,
  1521. label=_("Format:")),
  1522. flag=wx.ALIGN_CENTER_VERTICAL,
  1523. pos=(0, 0))
  1524. sizer.Add(self.protocolWidgets['format'],
  1525. flag=wx.ALIGN_CENTER_VERTICAL,
  1526. pos=(0, 1))
  1527. sizer.Add(StaticText(parent=self.protocolPanel,
  1528. label=_("Protocol:")),
  1529. flag=wx.ALIGN_CENTER_VERTICAL,
  1530. pos=(1, 0))
  1531. sizer.Add(self.protocolWidgets['text'],
  1532. flag=wx.ALIGN_CENTER_VERTICAL | wx.EXPAND,
  1533. pos=(1, 1))
  1534. if self.dest:
  1535. sizer.Add(StaticText(parent=self.protocolPanel,
  1536. label=_("Creation options:")),
  1537. flag=wx.ALIGN_CENTER_VERTICAL,
  1538. pos=(2, 0))
  1539. sizer.Add(self.protocolWidgets['options'],
  1540. flag=wx.ALIGN_CENTER_VERTICAL | wx.EXPAND,
  1541. pos=(2, 1))
  1542. else:
  1543. self.protocolWidgets['options'].Hide()
  1544. sizer.AddGrowableCol(1)
  1545. self.protocolPanel.SetSizer(sizer)
  1546. # native
  1547. sizer = wx.BoxSizer(wx.VERTICAL)
  1548. sizer.Add(StaticText(parent=self.nativePanel,
  1549. label=_("No settings available")),
  1550. flag=wx.ALL | wx.EXPAND, border=5)
  1551. self.nativePanel.SetSizer(sizer)
  1552. for panel in (self.nativePanel, self.filePanel,
  1553. self.dirPanel, self.dbPanel,
  1554. self.protocolPanel):
  1555. self.changingSizer.Add(panel, proportion=1,
  1556. flag=wx.EXPAND)
  1557. self.mainSizer.Add(self.source, proportion=0,
  1558. flag=wx.LEFT | wx.RIGHT | wx.EXPAND, border=5)
  1559. self.mainSizer.Add(self.changingSizer, proportion=1,
  1560. flag=wx.ALL | wx.EXPAND, border=5)
  1561. self.SetSizer(self.mainSizer)
  1562. self.mainSizer.Fit(self)
  1563. def _getExtension(self, name):
  1564. """Get file extension by format name"""
  1565. formatToExt = dict()
  1566. formatToExt.update(rasterFormatExtension)
  1567. formatToExt.update(vectorFormatExtension)
  1568. return formatToExt.get(name, '')
  1569. def SetSourceType(self, sourceType):
  1570. """Set source type (db, file, dir, ...).
  1571. Does not switch radioboxes."""
  1572. self._sourceType = sourceType
  1573. self.changingSizer.Show(
  1574. self.filePanel, show=(
  1575. sourceType == 'file'))
  1576. self.changingSizer.Show(
  1577. self.nativePanel, show=(
  1578. sourceType == 'native'))
  1579. self.changingSizer.Show(self.dirPanel, show=(sourceType == 'dir'))
  1580. self.changingSizer.Show(
  1581. self.protocolPanel, show=(
  1582. sourceType == 'pro'))
  1583. self.changingSizer.Show(self.dbPanel, show=(sourceType == 'db'))
  1584. self.changingSizer.Layout()
  1585. if sourceType == 'db':
  1586. self.dbWidgets['format'].SetItems(self.dbFormats)
  1587. if self.dbFormats:
  1588. if 'PostgreSQL' in self.dbFormats:
  1589. self.dbWidgets['format'].SetStringSelection('PostgreSQL')
  1590. else:
  1591. self.dbWidgets['format'].SetSelection(0)
  1592. self.dbWidgets['format'].Enable()
  1593. if sourceType == 'db':
  1594. db = self.dbWidgets['format'].GetStringSelection()
  1595. self.SetDatabase(db)
  1596. if not self.dest:
  1597. self.reloadDataRequired.emit(listData=None, data=None)
  1598. self._reloadLayers()
  1599. def OnSettingsChanged(self, data):
  1600. """User changed setting"""
  1601. # data list: [type, dsn, format, options]
  1602. if len(data) == 3:
  1603. data.append('')
  1604. elif len(data) < 3:
  1605. return
  1606. self.source.SetSelection(self.sourceMap[data[0]])
  1607. self.SetSourceType(data[0])
  1608. if data[0] == 'file':
  1609. self.fileWidgets['browse'].SetValue(data[1])
  1610. self.fileWidgets['options'].SetValue(data[3])
  1611. elif data[0] == 'dir':
  1612. self.dirWidgets['browse'].SetValue(data[1])
  1613. self.dirWidgets['format'].SetStringSelection(data[2])
  1614. self.dirWidgets['options'].SetValue(data[3])
  1615. self.SetExtension(data[2])
  1616. elif data[0] == 'pro':
  1617. self.protocolWidgets['text'].SetValue(data[2])
  1618. self.protocolWidgets['options'].SetValue(data[3])
  1619. elif data[0] == 'db':
  1620. name = self._getCurrentDbWidgetName()
  1621. if name == 'choice':
  1622. if len(data[1].split(':', 1)) > 1:
  1623. for item in data[1].split(':', 1)[1].split(','):
  1624. key, value = item.split('=', 1)
  1625. if key == 'dbname':
  1626. self.dbWidgets[name].SetStringSelection(value)
  1627. break
  1628. else:
  1629. self.dbWidgets[name].SetStringSelection(data[1])
  1630. else:
  1631. self.dbWidgets[name].SetValue(data[1])
  1632. self.dbWidgets['options'].SetValue(data[3])
  1633. if not self.dest:
  1634. self.reloadDataRequired.emit(listData=None, data=None)
  1635. self._reloadLayers()
  1636. def AttachSettings(self):
  1637. if self.ogr:
  1638. settingsFile = os.path.join(GetSettingsPath(), 'wxOGR')
  1639. else:
  1640. settingsFile = os.path.join(GetSettingsPath(), 'wxGDAL')
  1641. self.settsManager = ManageSettingsWidget(parent=self,
  1642. settingsFile=settingsFile)
  1643. self.settsManager.settingsChanged.connect(self.OnSettingsChanged)
  1644. self.settsManager.settingsSaving.connect(self.OnSettingsSaving)
  1645. # do layout
  1646. self.mainSizer.Insert(0, self.settsManager,
  1647. flag=wx.ALL | wx.EXPAND, border=5)
  1648. def OnSettingsSaving(self, name):
  1649. """Saving data"""
  1650. if not self.GetDsn():
  1651. GMessage(parent=self, message=_(
  1652. "No data source defined, settings are not saved."))
  1653. return
  1654. self.settsManager.SetDataToSave((self._sourceType, self.GetDsn(),
  1655. self.GetFormat(), self.GetOptions()))
  1656. self.settsManager.SaveSettings(name)
  1657. def _getExtPatternGlob(self, ext):
  1658. """Get pattern for case-insensitive globing"""
  1659. pattern = '*.'
  1660. for c in ext:
  1661. pattern += '[%s%s]' % (c.lower(), c.upper())
  1662. return pattern
  1663. def _getCurrentDbWidgetName(self):
  1664. """Returns active dns database widget name."""
  1665. for widget in ('browse', 'dirbrowse', 'text', 'choice'):
  1666. if self.dbWidgets[widget].IsShown():
  1667. return widget
  1668. def GetDsn(self):
  1669. """Get datasource name
  1670. """
  1671. if self._sourceType == 'db':
  1672. if self.dbWidgets['format'].GetStringSelection() in(
  1673. 'PostgreSQL', 'PostGIS Raster driver'):
  1674. dsn = 'PG:dbname=%s' % self.dbWidgets[
  1675. 'choice'].GetStringSelection()
  1676. else:
  1677. name = self._getCurrentDbWidgetName()
  1678. if name == 'choice':
  1679. dsn = self.dbWidgets[name].GetStringSelection()
  1680. else:
  1681. dsn = self.dbWidgets[name].GetValue()
  1682. else:
  1683. if self._sourceType == 'file':
  1684. dsn = self.fileWidgets['browse'].GetValue()
  1685. elif self._sourceType == 'dir':
  1686. dsn = self.dirWidgets['browse'].GetValue()
  1687. elif self._sourceType == 'pro':
  1688. dsn = self.protocolWidgets['text'].GetValue()
  1689. else:
  1690. dsn = ''
  1691. # check compressed files
  1692. try:
  1693. ext = os.path.splitext(dsn)[1].lower()
  1694. except KeyError:
  1695. ext = None
  1696. if ext == '.zip':
  1697. dsn = '/vsizip/' + dsn
  1698. elif ext == '.gzip':
  1699. dsn = '/vsigzip/' + dsn
  1700. elif ext in ('.tar', '.tar.gz', '.tgz'):
  1701. dsn = '/vsitar/' + dsn
  1702. return dsn
  1703. def SetDatabase(self, db):
  1704. """Update database panel."""
  1705. sizer = self.dbPanel.GetSizer()
  1706. showBrowse = db in ('SQLite', 'Rasterlite')
  1707. showDirbrowse = db in ('FileGDB')
  1708. showChoice = db in ('PostgreSQL', 'PostGIS WKT Raster driver',
  1709. 'PostGIS Raster driver')
  1710. enableFeatType = self.dest and self.ogr and db in ('PostgreSQL')
  1711. showText = not(showBrowse or showChoice or showDirbrowse)
  1712. sizer.Show(self.dbWidgets['browse'], show=showBrowse)
  1713. sizer.Show(self.dbWidgets['dirbrowse'], show=showDirbrowse)
  1714. sizer.Show(self.dbWidgets['choice'], show=showChoice)
  1715. sizer.Show(self.dbWidgets['textLabel2'], show=showChoice)
  1716. sizer.Show(self.dbWidgets['text'], show=showText)
  1717. sizer.Show(self.dbWidgets['textLabel1'], show=showText)
  1718. self.dbWidgets['featType'].Enable(enableFeatType)
  1719. if showChoice:
  1720. # try to get list of PG databases
  1721. dbNames = RunCommand(
  1722. 'db.databases',
  1723. parent=self,
  1724. quiet=True,
  1725. read=True,
  1726. driver='pg').splitlines()
  1727. if dbNames is not None:
  1728. self.dbWidgets['choice'].SetItems(sorted(dbNames))
  1729. self.dbWidgets['choice'].SetSelection(0)
  1730. elif grass.find_program('psql', '--help'):
  1731. if not self.dbWidgets['choice'].GetItems():
  1732. p = grass.Popen(['psql', '-ltA'], stdout=grass.PIPE)
  1733. ret = p.communicate()[0]
  1734. if ret:
  1735. dbNames = list()
  1736. for line in ret.splitlines():
  1737. sline = line.split('|')
  1738. if len(sline) < 2:
  1739. continue
  1740. dbname = sline[0]
  1741. if dbname:
  1742. dbNames.append(dbname)
  1743. self.dbWidgets['choice'].SetItems(db)
  1744. self.dbWidgets['choice'].SetSelection(0)
  1745. else:
  1746. sizer.Show(self.dbWidgets['text'])
  1747. sizer.Show(self.dbWidgets['choice'], False)
  1748. sizer.Layout()
  1749. def OnUpdate(self, event):
  1750. """Update required - load layers."""
  1751. if not self.dest:
  1752. self._reloadLayers()
  1753. event.Skip()
  1754. def _reloadLayers(self):
  1755. """Reload list of layers"""
  1756. def hasRastSameProjAsLocation(dsn):
  1757. ret = RunCommand('r.external',
  1758. quiet=True,
  1759. read=True,
  1760. flags='t',
  1761. input=dsn)
  1762. # v.external returns info for individual bands, however projection is shared by all bands ->
  1763. # (it is possible to take first line)
  1764. lines = ret.splitlines()
  1765. projectionMatch = '0'
  1766. if lines:
  1767. bandNumber, bandType, projectionMatch = map(
  1768. lambda x: x.strip(), lines[0].split(','))
  1769. return projectionMatch
  1770. def getProjMatchCaption(projectionMatch):
  1771. if projectionMatch == '0':
  1772. projectionMatchCaption = _("No")
  1773. else:
  1774. projectionMatchCaption = _("Yes")
  1775. return projectionMatchCaption
  1776. dsn = self.GetDsn()
  1777. if not dsn:
  1778. return
  1779. data = list()
  1780. listData = list()
  1781. layerId = 1
  1782. if self.ogr:
  1783. ret = RunCommand('v.external',
  1784. quiet=True,
  1785. read=True,
  1786. flags='t',
  1787. input=dsn)
  1788. if not ret:
  1789. self.reloadDataRequired.emit(listData=None, data=None)
  1790. return
  1791. layerId = 1
  1792. for line in ret.splitlines():
  1793. layerName, featureType, projectionMatch, geometryColumn = map(
  1794. lambda x: x.strip(), line.split(','))
  1795. projectionMatchCaption = getProjMatchCaption(projectionMatch)
  1796. grassName = GetValidLayerName(layerName)
  1797. if geometryColumn:
  1798. featureType = geometryColumn + '/' + featureType
  1799. listData.append(
  1800. (layerId,
  1801. layerName,
  1802. featureType,
  1803. projectionMatchCaption,
  1804. grassName))
  1805. data.append(
  1806. (layerId,
  1807. layerName,
  1808. featureType,
  1809. int(projectionMatch),
  1810. grassName))
  1811. layerId += 1
  1812. else:
  1813. if self._sourceType == 'file':
  1814. baseName = os.path.basename(dsn)
  1815. grassName = GetValidLayerName(baseName.split('.', -1)[0])
  1816. projectionMatch = hasRastSameProjAsLocation(dsn)
  1817. projectionMatchCaption = getProjMatchCaption(projectionMatch)
  1818. listData.append(
  1819. (layerId, baseName, projectionMatchCaption, grassName))
  1820. data.append(
  1821. (layerId, baseName, int(projectionMatch), grassName))
  1822. elif self._sourceType == 'dir':
  1823. ext = self.dirWidgets['extension'].GetValue()
  1824. for filename in glob.glob(os.path.join(
  1825. dsn, "%s") % self._getExtPatternGlob(ext)):
  1826. baseName = os.path.basename(filename)
  1827. grassName = GetValidLayerName(baseName.split('.', -1)[0])
  1828. projectionMatch = hasRastSameProjAsLocation(filename)
  1829. projectionMatchCaption = getProjMatchCaption(
  1830. projectionMatch)
  1831. listData.append(
  1832. (layerId, baseName, projectionMatchCaption, grassName))
  1833. data.append(
  1834. (layerId, baseName, int(projectionMatch), grassName))
  1835. layerId += 1
  1836. # emit signal
  1837. self.reloadDataRequired.emit(listData=listData, data=data)
  1838. def ExtensionChanged(self, event):
  1839. if not self.dest:
  1840. # reload layers
  1841. self._reloadLayers()
  1842. def SetExtension(self, name):
  1843. """Extension changed"""
  1844. ext = self._getExtension(name)
  1845. self.dirWidgets['extension'].SetValue(ext)
  1846. def GetType(self):
  1847. """Get source type"""
  1848. return self._sourceType
  1849. def GetFormat(self):
  1850. """Get format as string"""
  1851. if self._sourceType == 'dir':
  1852. format = self.dirWidgets['format'].GetStringSelection()
  1853. elif self._sourceType == 'pro':
  1854. format = self.protocolWidgets['format'].GetStringSelection()
  1855. elif self._sourceType == 'db':
  1856. format = self.dbWidgets['format'].GetStringSelection()
  1857. else:
  1858. format = ''
  1859. return format.replace(' ', '_')
  1860. def GetFormatExt(self):
  1861. """Get format extension"""
  1862. return self._getExtension(self.GetFormat())
  1863. def GetOptions(self):
  1864. """Get creation options"""
  1865. if self._sourceType == 'file':
  1866. options = self.fileWidgets['options'].GetValue()
  1867. elif self._sourceType == 'dir':
  1868. options = self.dirWidgets['options'].GetValue()
  1869. elif self._sourceType == 'pro':
  1870. options = self.protocolWidgets['options'].GetValue()
  1871. elif self._sourceType == 'db':
  1872. if self.dbWidgets['featType'].GetSelection() == 1:
  1873. options = 'topology=yes '
  1874. else:
  1875. options = ''
  1876. options += self.dbWidgets['options'].GetValue()
  1877. return options.strip()
  1878. def OnHelp(self, event):
  1879. """Show related manual page"""
  1880. cmd = ''
  1881. if self.dest:
  1882. if self.ogr:
  1883. cmd = 'v.external.out'
  1884. else:
  1885. cmd = 'r.external.out'
  1886. else:
  1887. if self.link:
  1888. if self.ogr:
  1889. cmd = 'v.external'
  1890. else:
  1891. cmd = 'r.external'
  1892. else:
  1893. if self.ogr:
  1894. cmd = 'v.in.ogr'
  1895. else:
  1896. cmd = 'r.in.gdal'
  1897. RunCommand('g.manual', entry=cmd)
  1898. class ProjSelect(wx.ComboBox):
  1899. """Widget for selecting input raster/vector map used by
  1900. r.proj/v.proj modules."""
  1901. def __init__(self, parent, isRaster, id=wx.ID_ANY,
  1902. size=globalvar.DIALOG_COMBOBOX_SIZE, **kwargs):
  1903. super(ProjSelect, self).__init__(parent, id, size=size, **kwargs)
  1904. self.SetName("ProjSelect")
  1905. self.isRaster = isRaster
  1906. def UpdateItems(self, dbase, location, mapset):
  1907. """Update list of maps
  1908. """
  1909. if not dbase:
  1910. dbase = grass.gisenv()['GISDBASE']
  1911. if not mapset:
  1912. mapset = grass.gisenv()['MAPSET']
  1913. if self.isRaster:
  1914. ret = RunCommand('r.proj',
  1915. quiet=True,
  1916. read=True,
  1917. flags='l',
  1918. dbase=dbase,
  1919. location=location,
  1920. mapset=mapset)
  1921. else:
  1922. ret = RunCommand('v.proj',
  1923. quiet=True,
  1924. read=True,
  1925. flags='l',
  1926. dbase=dbase,
  1927. location=location,
  1928. mapset=mapset)
  1929. listMaps = list()
  1930. if ret:
  1931. for line in ret.splitlines():
  1932. listMaps.append(line.strip())
  1933. ListSortLower(listMaps)
  1934. self.SetItems(listMaps)
  1935. self.SetValue('')
  1936. class ElementSelect(wx.Choice):
  1937. def __init__(self, parent, id=wx.ID_ANY, elements=None,
  1938. size=globalvar.DIALOG_COMBOBOX_SIZE,
  1939. **kwargs):
  1940. """Widget for selecting GIS element
  1941. :param parent: parent window
  1942. :param elements: filter elements
  1943. """
  1944. super(ElementSelect, self).__init__(parent, id, size=size,
  1945. **kwargs)
  1946. self.SetName("ElementSelect")
  1947. task = gtask.parse_interface('g.list')
  1948. p = task.get_param(value='type')
  1949. self.values = p.get('values', [])
  1950. self.valuesDesc = p.get('values_desc', [])
  1951. if elements:
  1952. values = []
  1953. valuesDesc = []
  1954. for idx in range(0, len(self.values)):
  1955. value = self.values[idx]
  1956. if value in elements:
  1957. values.append(value)
  1958. valuesDesc.append(self.valuesDesc[idx])
  1959. self.values = values
  1960. self.valuesDesc = valuesDesc
  1961. self.SetItems(self.valuesDesc)
  1962. def GetValue(self, name):
  1963. """Translate value
  1964. :param name: element name
  1965. """
  1966. idx = self.valuesDesc.index(name)
  1967. if idx > -1:
  1968. return self.values[idx]
  1969. return ''
  1970. class OgrTypeSelect(wx.Panel):
  1971. def __init__(self, parent, panel, **kwargs):
  1972. """Widget to choose OGR feature type
  1973. :param parent: parent window
  1974. :param panel: wx.Panel instance used as parent window
  1975. """
  1976. wx.Panel.__init__(self, parent=panel, id=wx.ID_ANY)
  1977. self.ftype = wx.Choice(parent=self, id=wx.ID_ANY, size=(
  1978. 200, -1), choices=(_("Point"), _("LineString"), _("Polygon")))
  1979. self._layout()
  1980. def _layout(self):
  1981. """Do layout"""
  1982. sizer = wx.BoxSizer(wx.HORIZONTAL)
  1983. sizer.Add(StaticText(parent=self,
  1984. id=wx.ID_ANY,
  1985. label=_("Feature type:")),
  1986. proportion=1,
  1987. flag=wx.ALIGN_CENTER_VERTICAL,
  1988. border=5)
  1989. sizer.Add(self.ftype,
  1990. proportion=0,
  1991. flag=wx.EXPAND | wx.ALIGN_RIGHT)
  1992. self.SetSizer(sizer)
  1993. sizer.Fit(self)
  1994. def GetType(self):
  1995. """Get selected type as string
  1996. :return: feature type as string
  1997. """
  1998. sel = self.ftype.GetSelection()
  1999. if sel == 0:
  2000. return 'point'
  2001. elif sel == 1:
  2002. return 'line'
  2003. elif sel == 2:
  2004. return 'boundary'
  2005. class CoordinatesSelect(Panel):
  2006. def __init__(self, parent, giface, multiple=False, **kwargs):
  2007. """Widget to get coordinates from map window by mouse click
  2008. :param parent: parent window
  2009. :param giface: GRASS interface
  2010. :param multiple: - True if it is possible to insert more coordinates
  2011. """
  2012. self._giface = giface
  2013. self.multiple = multiple
  2014. self.mapWin = None
  2015. self.drawMapWin = None
  2016. super(CoordinatesSelect, self).__init__(parent=parent, id=wx.ID_ANY)
  2017. self.coordsField = TextCtrl(parent=self, id=wx.ID_ANY,
  2018. size=globalvar.DIALOG_TEXTCTRL_SIZE,
  2019. validator=CoordinatesValidator())
  2020. icon = wx.Bitmap(
  2021. os.path.join(
  2022. globalvar.ICONDIR,
  2023. "grass",
  2024. "pointer.png"))
  2025. self.buttonInsCoords = buttons.ThemedGenBitmapToggleButton(
  2026. parent=self, id=wx.ID_ANY, bitmap=icon, size=globalvar.DIALOG_COLOR_SIZE)
  2027. self.registered = False
  2028. self.buttonInsCoords.Bind(wx.EVT_BUTTON, self._onClick)
  2029. mapdisp = self._giface.GetMapDisplay()
  2030. if mapdisp:
  2031. switcher = mapdisp.GetToolSwitcher()
  2032. switcher.AddCustomToolToGroup(
  2033. group='mouseUse',
  2034. btnId=self.buttonInsCoords.GetId(),
  2035. toggleHandler=self.buttonInsCoords.SetValue)
  2036. self._doLayout()
  2037. self.coordsField.Bind(wx.EVT_TEXT, lambda event: self._draw(delay=1))
  2038. def _doLayout(self):
  2039. self.dialogSizer = wx.BoxSizer(wx.HORIZONTAL)
  2040. self.dialogSizer.Add(self.coordsField,
  2041. proportion=1,
  2042. flag=wx.EXPAND)
  2043. self.dialogSizer.Add(self.buttonInsCoords)
  2044. self.SetSizer(self.dialogSizer)
  2045. def _onClick(self, event):
  2046. """Button for interacitve inserting of coordinates clicked"""
  2047. self.mapWin = self._giface.GetMapWindow()
  2048. if self.buttonInsCoords.GetToggle() and self.mapWin:
  2049. switcher = self._giface.GetMapDisplay().GetToolSwitcher()
  2050. switcher.ToolChanged(self.buttonInsCoords.GetId())
  2051. if not self.mapWin.RegisterMouseEventHandler(wx.EVT_LEFT_DOWN,
  2052. self._onMapClickHandler,
  2053. 'cross'):
  2054. return
  2055. self.registered = True
  2056. self._giface.GetMapDisplay().Raise()
  2057. else:
  2058. if self.mapWin and self.mapWin.UnregisterMouseEventHandler(
  2059. wx.EVT_LEFT_DOWN, self._onMapClickHandler):
  2060. self.registered = False
  2061. return
  2062. def drawCleanUp(self):
  2063. if self.drawMapWin:
  2064. self.drawMapWin.UnregisterGraphicsToDraw(self.pointsToDraw)
  2065. def _draw(self, delay):
  2066. """Draws points representing inserted coordinates in mapwindow."""
  2067. if self.drawMapWin != self.mapWin:
  2068. self.drawCleanUp()
  2069. if self.mapWin:
  2070. self.drawMapWin = self.mapWin
  2071. self.pointsToDraw = self.drawMapWin.RegisterGraphicsToDraw(
  2072. graphicsType="point")
  2073. if self.drawMapWin:
  2074. items = self.pointsToDraw.GetAllItems()
  2075. for i in items:
  2076. self.pointsToDraw.DeleteItem(i)
  2077. coords = self._getCoords()
  2078. if coords is not None:
  2079. for i in range(len(coords) // 2):
  2080. i = i * 2
  2081. self.pointsToDraw.AddItem(
  2082. coords=(coords[i], coords[i + 1]))
  2083. self._giface.updateMap.emit(
  2084. render=False, renderVector=False, delay=delay)
  2085. def _getCoords(self):
  2086. """Get list of coordinates.
  2087. :return: None if values are not valid
  2088. """
  2089. if self.coordsField.GetValidator().Validate(self):
  2090. return self.coordsField.GetValue().split(',')
  2091. return None
  2092. def _onMapClickHandler(self, event):
  2093. """Gets coordinates from mapwindow"""
  2094. if event == "unregistered":
  2095. return
  2096. e, n = self.mapWin.GetLastEN()
  2097. prevCoords = ""
  2098. if self.multiple:
  2099. prevCoords = self.coordsField.GetValue().strip()
  2100. if prevCoords != "":
  2101. prevCoords += ","
  2102. value = prevCoords + str(e) + "," + str(n)
  2103. self.coordsField.SetValue(value)
  2104. self._draw(delay=0)
  2105. def OnClose(self):
  2106. """Unregistrates _onMapClickHandler from mapWin"""
  2107. self.drawCleanUp()
  2108. self._giface.updateMap.emit(render=False, renderVector=False)
  2109. mapdisp = self._giface.GetMapDisplay()
  2110. if mapdisp:
  2111. switcher = mapdisp.GetToolSwitcher()
  2112. switcher.RemoveCustomToolFromGroup(self.buttonInsCoords.GetId())
  2113. if self.mapWin and self.registered:
  2114. self.mapWin.UnregisterMouseEventHandler(wx.EVT_LEFT_DOWN,
  2115. self._onMapClickHandler)
  2116. def GetTextWin(self):
  2117. """Get TextCtrl widget"""
  2118. return self.coordsField
  2119. class VectorCategorySelect(wx.Panel):
  2120. """Widget that allows interactive selection of vector features"""
  2121. def __init__(self, parent, giface, task=None):
  2122. super(VectorCategorySelect, self).__init__(parent=parent, id=wx.ID_ANY)
  2123. self.task = task
  2124. self.parent = parent
  2125. self.giface = giface
  2126. self.selectedFeatures = None
  2127. self.registered = False
  2128. self._vectorSelect = None
  2129. self.mapdisp = self.giface.GetMapDisplay()
  2130. self.catsField = TextCtrl(parent=self, id=wx.ID_ANY,
  2131. size=globalvar.DIALOG_TEXTCTRL_SIZE)
  2132. icon = wx.Bitmap(
  2133. os.path.join(
  2134. globalvar.ICONDIR,
  2135. "grass",
  2136. "select.png"))
  2137. self.buttonVecSelect = buttons.ThemedGenBitmapToggleButton(
  2138. parent=self, id=wx.ID_ANY, bitmap=icon, size=globalvar.DIALOG_COLOR_SIZE)
  2139. self.buttonVecSelect.Bind(wx.EVT_BUTTON, self._onClick)
  2140. if self.mapdisp:
  2141. switcher = self.mapdisp.GetToolSwitcher()
  2142. switcher.AddCustomToolToGroup(
  2143. group='mouseUse',
  2144. btnId=self.buttonVecSelect.GetId(),
  2145. toggleHandler=self.buttonVecSelect.SetValue)
  2146. self._layout()
  2147. def _isMapSelected(self):
  2148. """Check if layer list contains at least one selected map
  2149. """
  2150. layerList = self.giface.GetLayerList()
  2151. layerSelected = layerList.GetSelectedLayer()
  2152. if layerSelected is None:
  2153. GWarning(
  2154. _("No vector map selected in layer manager. Operation canceled."))
  2155. return False
  2156. return True
  2157. def _chckMap(self):
  2158. """Check if selected map in 'input' widget is the same as selected map in lmgr """
  2159. if self._isMapSelected():
  2160. layerList = self.giface.GetLayerList()
  2161. layerSelected = layerList.GetSelectedLayer()
  2162. # d.vect module
  2163. inputName = self.task.get_param(value='map', raiseError=False)
  2164. if not inputName:
  2165. inputName = self.task.get_param('input')
  2166. if inputName['value'] != str(layerSelected):
  2167. if inputName['value'] == '' or inputName['value'] is None:
  2168. GWarning(_("Input vector map is not selected"))
  2169. return False
  2170. GWarning(
  2171. _(
  2172. "Input vector map <%s> and selected map <%s> in layer manager are different. "
  2173. "Operation canceled.") %
  2174. (inputName['value'], str(layerSelected)))
  2175. return False
  2176. return True
  2177. return False
  2178. def _onClick(self, evt=None):
  2179. if self.task is not None:
  2180. if not self._chckMap():
  2181. self.buttonVecSelect.SetValue(False)
  2182. return
  2183. else:
  2184. if not self._isMapSelected():
  2185. self.buttonVecSelect.SetValue(False)
  2186. return
  2187. if self._vectorSelect is None:
  2188. if self.mapdisp:
  2189. if self.buttonVecSelect.IsEnabled():
  2190. switcher = self.mapdisp.GetToolSwitcher()
  2191. switcher.ToolChanged(self.buttonVecSelect.GetId())
  2192. self._vectorSelect = VectorSelectBase(
  2193. self.mapdisp, self.giface)
  2194. if not self.mapdisp.GetWindow().RegisterMouseEventHandler(
  2195. wx.EVT_LEFT_DOWN, self._onMapClickHandler, 'cross'):
  2196. return
  2197. self.registered = True
  2198. self.mapdisp.Raise()
  2199. else:
  2200. self.OnClose()
  2201. def OnClose(self, event=None):
  2202. if not self.mapdisp:
  2203. return
  2204. switcher = self.mapdisp.GetToolSwitcher()
  2205. switcher.RemoveCustomToolFromGroup(self.buttonVecSelect.GetId())
  2206. if self._vectorSelect is not None:
  2207. tmp = self._vectorSelect.GetLineStringSelectedCats()
  2208. self._vectorSelect.OnClose()
  2209. self.catsField.SetValue(tmp)
  2210. self._vectorSelect = None
  2211. def _onMapClickHandler(self, event):
  2212. """Update category text input widget"""
  2213. if event == "unregistered":
  2214. return
  2215. if self.task is None:
  2216. if not self._isMapSelected():
  2217. self.OnClose()
  2218. else:
  2219. self.catsField.SetValue(
  2220. self._vectorSelect.GetLineStringSelectedCats())
  2221. else:
  2222. if not self._chckMap():
  2223. self.OnClose()
  2224. else:
  2225. self.catsField.SetValue(
  2226. self._vectorSelect.GetLineStringSelectedCats())
  2227. def GetTextWin(self):
  2228. return self.catsField
  2229. def GetValue(self):
  2230. return self.catsField.GetValue()
  2231. def SetValue(self, value):
  2232. self.catsField.SetValue(value)
  2233. def _layout(self):
  2234. self.dialogSizer = wx.BoxSizer(wx.HORIZONTAL)
  2235. self.dialogSizer.Add(self.catsField,
  2236. proportion=1,
  2237. flag=wx.EXPAND)
  2238. self.dialogSizer.Add(self.buttonVecSelect)
  2239. self.SetSizer(self.dialogSizer)
  2240. class SignatureSelect(wx.ComboBox):
  2241. """Widget for selecting signatures"""
  2242. def __init__(self, parent, element, id=wx.ID_ANY,
  2243. size=globalvar.DIALOG_GSELECT_SIZE, **kwargs):
  2244. super(SignatureSelect, self).__init__(parent, id, size=size,
  2245. **kwargs)
  2246. self.element = element
  2247. self.SetName("SignatureSelect")
  2248. def Insert(self, group, subgroup=None):
  2249. """Insert signatures for defined group/subgroup
  2250. :param group: group name (can be fully-qualified)
  2251. :param subgroup: non fully-qualified name of subgroup
  2252. """
  2253. if not group:
  2254. return
  2255. gisenv = grass.gisenv()
  2256. try:
  2257. name, mapset = group.split('@', 1)
  2258. except ValueError:
  2259. name = group
  2260. mapset = gisenv['MAPSET']
  2261. path = os.path.join(
  2262. gisenv['GISDBASE'],
  2263. gisenv['LOCATION_NAME'],
  2264. mapset, 'group', name)
  2265. if subgroup:
  2266. path = os.path.join(path, 'subgroup', subgroup)
  2267. try:
  2268. items = list()
  2269. for element in os.listdir(os.path.join(path, self.element)):
  2270. items.append(element)
  2271. self.SetItems(items)
  2272. except OSError:
  2273. self.SetItems([])
  2274. self.SetValue('')
  2275. class SeparatorSelect(wx.ComboBox):
  2276. """Widget for selecting seperator"""
  2277. def __init__(self, parent, id=wx.ID_ANY,
  2278. size=globalvar.DIALOG_GSELECT_SIZE, **kwargs):
  2279. super(SeparatorSelect, self).__init__(parent, id, size=size,
  2280. **kwargs)
  2281. self.SetName("SeparatorSelect")
  2282. self.SetItems(['pipe', 'comma', 'space', 'tab', 'newline'])
  2283. class SqlWhereSelect(wx.Panel):
  2284. def __init__(self, parent, **kwargs):
  2285. """Widget to define SQL WHERE condition.
  2286. :param parent: parent window
  2287. """
  2288. super(SqlWhereSelect, self).__init__(parent=parent, id=wx.ID_ANY)
  2289. self.parent = parent
  2290. self.vector_map = None
  2291. self.sqlField = TextCtrl(parent=self, id=wx.ID_ANY,
  2292. size=globalvar.DIALOG_TEXTCTRL_SIZE)
  2293. self.GetChildren()[0].SetName("SqlWhereSelect")
  2294. icon = wx.Bitmap(
  2295. os.path.join(
  2296. globalvar.ICONDIR,
  2297. "grass",
  2298. "table.png"))
  2299. self.buttonInsSql = buttons.ThemedGenBitmapButton(
  2300. parent=self, id=wx.ID_ANY, bitmap=icon, size=globalvar.DIALOG_COLOR_SIZE)
  2301. self.buttonInsSql.Bind(wx.EVT_BUTTON, self._onClick)
  2302. self._doLayout()
  2303. def _doLayout(self):
  2304. self.dialogSizer = wx.BoxSizer(wx.HORIZONTAL)
  2305. self.dialogSizer.Add(self.sqlField,
  2306. proportion=1,
  2307. flag=wx.EXPAND)
  2308. self.dialogSizer.Add(self.buttonInsSql)
  2309. self.SetSizer(self.dialogSizer)
  2310. def GetTextWin(self):
  2311. return self.sqlField
  2312. def _onClick(self, event):
  2313. from dbmgr.sqlbuilder import SQLBuilderWhere
  2314. try:
  2315. if not self.vector_map:
  2316. raise GException(_('No vector map selected'))
  2317. win = SQLBuilderWhere(parent=self,
  2318. vectmap=self.vector_map,
  2319. layer=self.vector_layer)
  2320. win.Show()
  2321. except GException as e:
  2322. GMessage(parent=self.parent, message='{}'.format(e))
  2323. def SetData(self, vector, layer):
  2324. self.vector_map = vector
  2325. self.vector_layer = int(layer) # TODO: support layer names
  2326. def SetValue(self, value):
  2327. self.sqlField.SetValue(value)