gselect.py 95 KB

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