gselect.py 96 KB

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