gselect.py 96 KB

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