gselect.py 96 KB

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