gselect.py 89 KB

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