gselect.py 94 KB

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