widgets.py 51 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676
  1. """
  2. @package gui_core.widgets
  3. @brief Core GUI widgets
  4. Classes:
  5. - widgets::GNotebook
  6. - widgets::ScrolledPanel
  7. - widgets::NumTextCtrl
  8. - widgets::FloatSlider
  9. - widgets::SymbolButton
  10. - widgets::StaticWrapText
  11. - widgets::BaseValidator
  12. - widgets::CoordinatesValidator
  13. - widgets::IntegerValidator
  14. - widgets::FloatValidator
  15. - widgets::EmailValidator
  16. - widgets::TimeISOValidator
  17. - widgets::MapValidator
  18. - widgets::NTCValidator
  19. - widgets::SimpleValidator
  20. - widgets::GenericValidator
  21. - widgets::GListCtrl
  22. - widgets::SearchModuleWidget
  23. - widgets::ManageSettingsWidget
  24. - widgets::PictureComboBox
  25. - widgets::ColorTablesComboBox
  26. - widgets::BarscalesComboBox
  27. - widgets::NArrowsComboBox
  28. - widgets::LayersList
  29. @todo:
  30. - move validators to a separate file gui_core/validators.py
  31. (C) 2008-2014 by the GRASS Development Team
  32. This program is free software under the GNU General Public License
  33. (>=v2). Read the file COPYING that comes with GRASS for details.
  34. @author Martin Landa <landa.martin gmail.com> (Google SoC 2008/2010)
  35. @author Enhancements by Michael Barton <michael.barton asu.edu>
  36. @author Anna Kratochvilova <kratochanna gmail.com> (Google SoC 2011)
  37. @author Stepan Turek <stepan.turek seznam.cz> (ManageSettingsWidget - created from GdalSelect)
  38. @author Matej Krejci <matejkrejci gmail.com> (Google GSoC 2014; EmailValidator, TimeISOValidator)
  39. """
  40. import os
  41. import sys
  42. import string
  43. import re
  44. from bisect import bisect
  45. from datetime import datetime
  46. from core.globalvar import wxPythonPhoenix
  47. import wx
  48. import wx.lib.mixins.listctrl as listmix
  49. import wx.lib.scrolledpanel as SP
  50. from wx.lib.stattext import GenStaticText
  51. from wx.lib.wordwrap import wordwrap
  52. if wxPythonPhoenix:
  53. import wx.adv
  54. from wx.adv import OwnerDrawnComboBox
  55. else:
  56. import wx.combo
  57. from wx.combo import OwnerDrawnComboBox
  58. try:
  59. import wx.lib.agw.flatnotebook as FN
  60. except ImportError:
  61. import wx.lib.flatnotebook as FN
  62. try:
  63. from wx.lib.buttons import ThemedGenBitmapTextButton as BitmapTextButton
  64. except ImportError: # not sure about TGBTButton version
  65. from wx.lib.buttons import GenBitmapTextButton as BitmapTextButton
  66. try:
  67. import wx.lib.agw.customtreectrl as CT
  68. except ImportError:
  69. import wx.lib.customtreectrl as CT
  70. if wxPythonPhoenix:
  71. from wx import Validator as Validator
  72. else:
  73. from wx import PyValidator as Validator
  74. from grass.script import core as grass
  75. from grass.pydispatch.signal import Signal
  76. from core import globalvar
  77. from core.utils import _
  78. from core.gcmd import GMessage, GError
  79. from core.debug import Debug
  80. from gui_core.wrap import Button, SearchCtrl
  81. class NotebookController:
  82. """Provides handling of notebook page names.
  83. Translates page names to page indices.
  84. Class is aggregated in notebook subclasses.
  85. Notebook subclasses must delegate methods to controller.
  86. Methods inherited from notebook class must be delegated explicitly
  87. and other methods can be delegated by @c __getattr__.
  88. """
  89. def __init__(self, classObject, widget):
  90. """
  91. :param classObject: notebook class name (object, i.e. FlatNotebook)
  92. :param widget: notebook instance
  93. """
  94. self.notebookPages = {}
  95. self.classObject = classObject
  96. self.widget = widget
  97. self.highlightedTextEnd = _(" (...)")
  98. self.BindPageChanged()
  99. def BindPageChanged(self):
  100. """Binds page changed event."""
  101. self.widget.Bind(wx.EVT_NOTEBOOK_PAGE_CHANGED, self.OnRemoveHighlight)
  102. def AddPage(self, **kwargs):
  103. """Add a new page
  104. """
  105. if 'name' in kwargs:
  106. self.notebookPages[kwargs['name']] = kwargs['page']
  107. del kwargs['name']
  108. self.classObject.AddPage(self.widget, **kwargs)
  109. def InsertPage(self, **kwargs):
  110. """Insert a new page
  111. """
  112. if 'name' in kwargs:
  113. self.notebookPages[kwargs['name']] = kwargs['page']
  114. del kwargs['name']
  115. try:
  116. self.classObject.InsertPage(self.widget, **kwargs)
  117. except TypeError as e: # documentation says 'index', but certain versions of wx require 'n'
  118. kwargs['n'] = kwargs['index']
  119. del kwargs['index']
  120. self.classObject.InsertPage(self.widget, **kwargs)
  121. def DeletePage(self, page):
  122. """Delete page
  123. :param page: name
  124. :return: True if page was deleted, False if not exists
  125. """
  126. delPageIndex = self.GetPageIndexByName(page)
  127. if delPageIndex != -1:
  128. ret = self.classObject.DeletePage(self.widget, delPageIndex)
  129. if ret:
  130. del self.notebookPages[page]
  131. return ret
  132. else:
  133. return False
  134. def RemovePage(self, page):
  135. """Delete page without deleting the associated window.
  136. :param page: name
  137. :return: True if page was deleted, False if not exists
  138. """
  139. delPageIndex = self.GetPageIndexByName(page)
  140. if delPageIndex != -1:
  141. ret = self.classObject.RemovePage(self.widget, delPageIndex)
  142. if ret:
  143. del self.notebookPages[page]
  144. return ret
  145. else:
  146. return False
  147. def SetSelectionByName(self, page):
  148. """Set active notebook page.
  149. :param page: name, eg. 'layers', 'output', 'search', 'pyshell', 'nviz'
  150. (depends on concrete notebook instance)
  151. """
  152. idx = self.GetPageIndexByName(page)
  153. if self.classObject.GetSelection(self.widget) != idx:
  154. self.classObject.SetSelection(self.widget, idx)
  155. self.RemoveHighlight(idx)
  156. def OnRemoveHighlight(self, event):
  157. """Highlighted tab name should be removed."""
  158. page = event.GetSelection()
  159. self.RemoveHighlight(page)
  160. event.Skip()
  161. def RemoveHighlight(self, page):
  162. """Removes highlight string from notebook tab name if necessary.
  163. :param page: index
  164. """
  165. text = self.classObject.GetPageText(self.widget, page)
  166. if text.endswith(self.highlightedTextEnd):
  167. text = text.replace(self.highlightedTextEnd, '')
  168. self.classObject.SetPageText(self.widget, page, text)
  169. def GetPageIndexByName(self, page):
  170. """Get notebook page index
  171. :param page: name
  172. """
  173. if page not in self.notebookPages:
  174. return -1
  175. for pageIndex in range(self.classObject.GetPageCount(self.widget)):
  176. if self.notebookPages[page] == self.classObject.GetPage(
  177. self.widget, pageIndex):
  178. break
  179. return pageIndex
  180. def HighlightPageByName(self, page):
  181. pageIndex = self.GetPageIndexByName(page)
  182. self.HighlightPage(pageIndex)
  183. def HighlightPage(self, index):
  184. if self.classObject.GetSelection(self.widget) != index:
  185. text = self.classObject.GetPageText(self.widget, index)
  186. if not text.endswith(self.highlightedTextEnd):
  187. text += self.highlightedTextEnd
  188. self.classObject.SetPageText(self.widget, index, text)
  189. def SetPageImage(self, page, index):
  190. """Sets image index for page
  191. :param page: page name
  192. :param index: image index (in wx.ImageList)
  193. """
  194. pageIndex = self.GetPageIndexByName(page)
  195. self.classObject.SetPageImage(self.widget, pageIndex, index)
  196. class FlatNotebookController(NotebookController):
  197. """Controller specialized for FN.FlatNotebook subclasses"""
  198. def __init__(self, classObject, widget):
  199. NotebookController.__init__(self, classObject, widget)
  200. def BindPageChanged(self):
  201. self.widget.Bind(
  202. FN.EVT_FLATNOTEBOOK_PAGE_CHANGED,
  203. self.OnRemoveHighlight)
  204. def GetPageIndexByName(self, page):
  205. """Get notebook page index
  206. :param page: name
  207. """
  208. if page not in self.notebookPages:
  209. return -1
  210. return self.classObject.GetPageIndex(
  211. self.widget, self.notebookPages[page])
  212. def InsertPage(self, **kwargs):
  213. """Insert a new page
  214. """
  215. if 'name' in kwargs:
  216. self.notebookPages[kwargs['name']] = kwargs['page']
  217. del kwargs['name']
  218. kwargs['indx'] = kwargs['index']
  219. del kwargs['index']
  220. self.classObject.InsertPage(self.widget, **kwargs)
  221. class GNotebook(FN.FlatNotebook):
  222. """Generic notebook widget.
  223. Enables advanced style settings.
  224. Problems with hidden tabs and does not respect system colors (native look).
  225. """
  226. def __init__(self, parent, style, **kwargs):
  227. if globalvar.hasAgw:
  228. FN.FlatNotebook.__init__(self, parent, id=wx.ID_ANY,
  229. agwStyle=style, **kwargs)
  230. else:
  231. FN.FlatNotebook.__init__(self, parent, id=wx.ID_ANY,
  232. style=style, **kwargs)
  233. self.controller = FlatNotebookController(classObject=FN.FlatNotebook,
  234. widget=self)
  235. def AddPage(self, **kwargs):
  236. """@copydoc NotebookController::AddPage()"""
  237. self.controller.AddPage(**kwargs)
  238. def InsertNBPage(self, **kwargs):
  239. """@copydoc NotebookController::InsertPage()"""
  240. self.controller.InsertPage(**kwargs)
  241. def DeleteNBPage(self, page):
  242. """@copydoc NotebookController::DeletePage()"""
  243. return self.controller.DeletePage(page)
  244. def RemoveNBPage(self, page):
  245. """@copydoc NotebookController::RemovePage()"""
  246. return self.controller.RemovePage(page)
  247. def SetPageImage(self, page, index):
  248. """Does nothing because we don't want images for this style"""
  249. pass
  250. def __getattr__(self, name):
  251. return getattr(self.controller, name)
  252. class FormNotebook(wx.Notebook):
  253. """Notebook widget.
  254. Respects native look.
  255. """
  256. def __init__(self, parent, style):
  257. wx.Notebook.__init__(self, parent, id=wx.ID_ANY, style=style)
  258. self.controller = NotebookController(classObject=wx.Notebook,
  259. widget=self)
  260. def AddPage(self, **kwargs):
  261. """@copydoc NotebookController::AddPage()"""
  262. self.controller.AddPage(**kwargs)
  263. def InsertNBPage(self, **kwargs):
  264. """@copydoc NotebookController::InsertPage()"""
  265. self.controller.InsertPage(**kwargs)
  266. def DeleteNBPage(self, page):
  267. """@copydoc NotebookController::DeletePage()"""
  268. return self.controller.DeletePage(page)
  269. def RemoveNBPage(self, page):
  270. """@copydoc NotebookController::RemovePage()"""
  271. return self.controller.RemovePage(page)
  272. def SetPageImage(self, page, index):
  273. """@copydoc NotebookController::SetPageImage()"""
  274. return self.controller.SetPageImage(page, index)
  275. def __getattr__(self, name):
  276. return getattr(self.controller, name)
  277. class FormListbook(wx.Listbook):
  278. """Notebook widget.
  279. Respects native look.
  280. """
  281. def __init__(self, parent, style):
  282. wx.Listbook.__init__(self, parent, id=wx.ID_ANY, style=style)
  283. self.controller = NotebookController(classObject=wx.Listbook,
  284. widget=self)
  285. def AddPage(self, **kwargs):
  286. """@copydoc NotebookController::AddPage()"""
  287. self.controller.AddPage(**kwargs)
  288. def InsertPage_(self, **kwargs):
  289. """@copydoc NotebookController::InsertPage()"""
  290. self.controller.InsertPage(**kwargs)
  291. def DeletePage(self, page):
  292. """@copydoc NotebookController::DeletePage()"""
  293. return self.controller.DeletePage(page)
  294. def RemovePage(self, page):
  295. """@copydoc NotebookController::RemovePage()"""
  296. return self.controller.RemovePage(page)
  297. def SetPageImage(self, page, index):
  298. """@copydoc NotebookController::SetPageImage()"""
  299. return self.controller.SetPageImage(page, index)
  300. def __getattr__(self, name):
  301. return getattr(self.controller, name)
  302. class ScrolledPanel(SP.ScrolledPanel):
  303. """Custom ScrolledPanel to avoid strange behaviour concerning focus"""
  304. def __init__(self, parent, style=wx.TAB_TRAVERSAL):
  305. SP.ScrolledPanel.__init__(self, parent=parent, id=wx.ID_ANY,
  306. style=style)
  307. def OnChildFocus(self, event):
  308. pass
  309. class NumTextCtrl(wx.TextCtrl):
  310. """Class derived from wx.TextCtrl for numerical values only"""
  311. def __init__(self, parent, **kwargs):
  312. ## self.precision = kwargs.pop('prec')
  313. wx.TextCtrl.__init__(self, parent=parent,
  314. validator=NTCValidator(flag='DIGIT_ONLY'),
  315. **kwargs)
  316. def SetValue(self, value):
  317. super(NumTextCtrl, self).SetValue(str(value))
  318. def GetValue(self):
  319. val = super(NumTextCtrl, self).GetValue()
  320. if val == '':
  321. val = '0'
  322. try:
  323. return float(val)
  324. except ValueError:
  325. val = ''.join(''.join(val.split('-')).split('.'))
  326. return float(val)
  327. def SetRange(self, min, max):
  328. pass
  329. class FloatSlider(wx.Slider):
  330. """Class derived from wx.Slider for floats"""
  331. def __init__(self, **kwargs):
  332. Debug.msg(1, "FloatSlider.__init__()")
  333. wx.Slider.__init__(self, **kwargs)
  334. self.coef = 1.
  335. # init range
  336. self.minValueOrig = 0
  337. self.maxValueOrig = 1
  338. def SetValue(self, value):
  339. value *= self.coef
  340. if abs(value) < 1 and value != 0:
  341. while abs(value) < 1:
  342. value *= 100
  343. self.coef *= 100
  344. super(FloatSlider, self).SetRange(self.minValueOrig * self.coef,
  345. self.maxValueOrig * self.coef)
  346. super(FloatSlider, self).SetValue(value)
  347. Debug.msg(4, "FloatSlider.SetValue(): value = %f" % value)
  348. def SetRange(self, minValue, maxValue):
  349. self.coef = 1.
  350. self.minValueOrig = minValue
  351. self.maxValueOrig = maxValue
  352. if abs(minValue) < 1 or abs(maxValue) < 1:
  353. while (abs(minValue) < 1 and minValue != 0) or (
  354. abs(maxValue) < 1 and maxValue != 0):
  355. minValue *= 100
  356. maxValue *= 100
  357. self.coef *= 100
  358. super(
  359. FloatSlider,
  360. self).SetValue(
  361. super(
  362. FloatSlider,
  363. self).GetValue() *
  364. self.coef)
  365. super(FloatSlider, self).SetRange(minValue, maxValue)
  366. Debug.msg(
  367. 4, "FloatSlider.SetRange(): minValue = %f, maxValue = %f" %
  368. (minValue, maxValue))
  369. def GetValue(self):
  370. val = super(FloatSlider, self).GetValue()
  371. Debug.msg(4, "FloatSlider.GetValue(): value = %f" % (val / self.coef))
  372. return val / self.coef
  373. class SymbolButton(BitmapTextButton):
  374. """Button with symbol and label."""
  375. def __init__(self, parent, usage, label, **kwargs):
  376. """Constructor
  377. :param parent: parent (usually wx.Panel)
  378. :param usage: determines usage and picture
  379. :param label: displayed label
  380. """
  381. size = (15, 15)
  382. buffer = wx.EmptyBitmap(*size)
  383. BitmapTextButton.__init__(self, parent=parent, label=" " + label,
  384. bitmap=buffer, **kwargs)
  385. dc = wx.MemoryDC()
  386. dc.SelectObject(buffer)
  387. maskColor = wx.Colour(255, 255, 255)
  388. dc.SetBrush(wx.Brush(maskColor))
  389. dc.Clear()
  390. if usage == 'record':
  391. self.DrawRecord(dc, size)
  392. elif usage == 'stop':
  393. self.DrawStop(dc, size)
  394. elif usage == 'play':
  395. self.DrawPlay(dc, size)
  396. elif usage == 'pause':
  397. self.DrawPause(dc, size)
  398. if sys.platform != "win32":
  399. buffer.SetMaskColour(maskColor)
  400. self.SetBitmapLabel(buffer)
  401. dc.SelectObject(wx.NullBitmap)
  402. def DrawRecord(self, dc, size):
  403. """Draw record symbol"""
  404. dc.SetBrush(wx.Brush(wx.Colour(255, 0, 0)))
  405. dc.DrawCircle(size[0] / 2, size[1] / 2, size[0] / 2)
  406. def DrawStop(self, dc, size):
  407. """Draw stop symbol"""
  408. dc.SetBrush(wx.Brush(wx.Colour(50, 50, 50)))
  409. dc.DrawRectangle(0, 0, size[0], size[1])
  410. def DrawPlay(self, dc, size):
  411. """Draw play symbol"""
  412. dc.SetBrush(wx.Brush(wx.Colour(0, 255, 0)))
  413. points = (wx.Point(0, 0), wx.Point(0, size[1]), wx.Point(size[0],
  414. size[1] / 2))
  415. dc.DrawPolygon(points)
  416. def DrawPause(self, dc, size):
  417. """Draw pause symbol"""
  418. dc.SetBrush(wx.Brush(wx.Colour(50, 50, 50)))
  419. dc.DrawRectangle(0, 0, 2 * size[0] / 5, size[1])
  420. dc.DrawRectangle(3 * size[0] / 5, 0, 2 * size[0] / 5, size[1])
  421. class StaticWrapText(GenStaticText):
  422. """A Static Text widget that wraps its text to fit parents width,
  423. enlarging its height if necessary."""
  424. def __init__(self, parent, id=wx.ID_ANY,
  425. label='', margin=0, *args, **kwds):
  426. self._margin = margin
  427. self._initialLabel = label
  428. self.init = False
  429. GenStaticText.__init__(self, parent, id, label, *args, **kwds)
  430. self.Bind(wx.EVT_SIZE, self.OnSize)
  431. def DoGetBestSize(self):
  432. """Overriden method which reports widget's best size."""
  433. if not self.init:
  434. self.init = True
  435. self._updateLabel()
  436. parent = self.GetParent()
  437. newExtent = wx.ClientDC(parent).GetMultiLineTextExtent(self.GetLabel())
  438. # when starting, width is very small and height is big which creates
  439. # very high windows
  440. if newExtent[0] < newExtent[1]:
  441. return (0, 0)
  442. return newExtent[:2]
  443. def OnSize(self, event):
  444. self._updateLabel()
  445. event.Skip()
  446. def _updateLabel(self):
  447. """Calculates size of wrapped label"""
  448. parent = self.GetParent()
  449. newLabel = wordwrap(text=self._initialLabel, width=parent.GetSize()[0],
  450. dc=wx.ClientDC(parent), breakLongWords=True,
  451. margin=self._margin)
  452. GenStaticText.SetLabel(self, newLabel)
  453. def SetLabel(self, label):
  454. self._initialLabel = label
  455. self._updateLabel()
  456. class BaseValidator(Validator):
  457. def __init__(self):
  458. Validator.__init__(self)
  459. self.Bind(wx.EVT_TEXT, self.OnText)
  460. def OnText(self, event):
  461. """Do validation"""
  462. self.Validate()
  463. event.Skip()
  464. def Validate(self):
  465. """Validate input"""
  466. textCtrl = self.GetWindow()
  467. text = textCtrl.GetValue()
  468. if text:
  469. try:
  470. self.type(text)
  471. except ValueError:
  472. self._notvalid()
  473. return False
  474. self._valid()
  475. return True
  476. def _notvalid(self):
  477. textCtrl = self.GetWindow()
  478. textCtrl.SetBackgroundColour("grey")
  479. textCtrl.SetFocus()
  480. textCtrl.Refresh()
  481. def _valid(self):
  482. textCtrl = self.GetWindow()
  483. sysColor = wx.SystemSettings.GetColour(wx.SYS_COLOUR_WINDOW)
  484. textCtrl.SetBackgroundColour(sysColor)
  485. textCtrl.Refresh()
  486. return True
  487. def TransferToWindow(self):
  488. return True # Prevent wxDialog from complaining.
  489. def TransferFromWindow(self):
  490. return True # Prevent wxDialog from complaining.
  491. class CoordinatesValidator(BaseValidator):
  492. """Validator for coordinates input (list of floats separated by comma)"""
  493. def __init__(self):
  494. BaseValidator.__init__(self)
  495. def Validate(self):
  496. """Validate input"""
  497. textCtrl = self.GetWindow()
  498. text = textCtrl.GetValue()
  499. if text:
  500. try:
  501. text = text.split(',')
  502. for t in text:
  503. float(t)
  504. if len(text) % 2 != 0:
  505. return False
  506. except ValueError:
  507. self._notvalid()
  508. return False
  509. self._valid()
  510. return True
  511. def Clone(self):
  512. """Clone validator"""
  513. return CoordinatesValidator()
  514. class IntegerValidator(BaseValidator):
  515. """Validator for floating-point input"""
  516. def __init__(self):
  517. BaseValidator.__init__(self)
  518. self.type = int
  519. def Clone(self):
  520. """Clone validator"""
  521. return IntegerValidator()
  522. class FloatValidator(BaseValidator):
  523. """Validator for floating-point input"""
  524. def __init__(self):
  525. BaseValidator.__init__(self)
  526. self.type = float
  527. def Clone(self):
  528. """Clone validator"""
  529. return FloatValidator()
  530. class EmailValidator(BaseValidator):
  531. """Validator for email input"""
  532. def __init__(self):
  533. BaseValidator.__init__(self)
  534. def Validate(self):
  535. """Validate input"""
  536. textCtrl = self.GetWindow()
  537. text = textCtrl.GetValue()
  538. if text:
  539. if re.match(r'\b[\w.-]+@[\w.-]+.\w{2,4}\b', text) is None:
  540. self._notvalid()
  541. return False
  542. self._valid()
  543. return True
  544. def Clone(self):
  545. """Clone validator"""
  546. return EmailValidator()
  547. class TimeISOValidator(BaseValidator):
  548. """Validator for time ISO format (YYYY-MM-DD) input"""
  549. def __init__(self):
  550. BaseValidator.__init__(self)
  551. def Validate(self):
  552. """Validate input"""
  553. textCtrl = self.GetWindow()
  554. text = textCtrl.GetValue()
  555. if text:
  556. try:
  557. datetime.strptime(text, '%Y-%m-%d')
  558. except:
  559. self._notvalid()
  560. return False
  561. self._valid()
  562. return True
  563. def Clone(self):
  564. """Clone validator"""
  565. return TimeISOValidator()
  566. class NTCValidator(Validator):
  567. """validates input in textctrls, taken from wxpython demo"""
  568. def __init__(self, flag=None):
  569. Validator.__init__(self)
  570. self.flag = flag
  571. self.Bind(wx.EVT_CHAR, self.OnChar)
  572. def Clone(self):
  573. return NTCValidator(self.flag)
  574. def OnChar(self, event):
  575. key = event.GetKeyCode()
  576. if key < wx.WXK_SPACE or key == wx.WXK_DELETE or key > 255:
  577. event.Skip()
  578. return
  579. if self.flag == 'DIGIT_ONLY' and chr(key) in string.digits + '.-':
  580. event.Skip()
  581. return
  582. if not wx.Validator_IsSilent():
  583. wx.Bell()
  584. # Returning without calling even.Skip eats the event before it
  585. # gets to the text control
  586. return
  587. class SimpleValidator(Validator):
  588. """This validator is used to ensure that the user has entered something
  589. into the text object editor dialog's text field.
  590. """
  591. def __init__(self, callback):
  592. """Standard constructor.
  593. """
  594. Validator.__init__(self)
  595. self.callback = callback
  596. def Clone(self):
  597. """Standard cloner.
  598. Note that every validator must implement the Clone() method.
  599. """
  600. return SimpleValidator(self.callback)
  601. def Validate(self, win):
  602. """Validate the contents of the given text control.
  603. """
  604. ctrl = self.GetWindow()
  605. text = ctrl.GetValue()
  606. if len(text) == 0:
  607. self.callback(ctrl)
  608. return False
  609. else:
  610. return True
  611. def TransferToWindow(self):
  612. """Transfer data from validator to window.
  613. The default implementation returns False, indicating that an
  614. error occurred. We simply return True, as we don't do any data
  615. transfer.
  616. """
  617. return True # Prevent wxDialog from complaining.
  618. def TransferFromWindow(self):
  619. """Transfer data from window to validator.
  620. The default implementation returns False, indicating that an
  621. error occurred. We simply return True, as we don't do any data
  622. transfer.
  623. """
  624. return True # Prevent wxDialog from complaining.
  625. class GenericValidator(Validator):
  626. """This validator checks condition and calls callback
  627. in case the condition is not fulfilled.
  628. """
  629. def __init__(self, condition, callback):
  630. """Standard constructor.
  631. :param condition: function which accepts string value and returns T/F
  632. :param callback: function which is called when condition is not fulfilled
  633. """
  634. Validator.__init__(self)
  635. self._condition = condition
  636. self._callback = callback
  637. def Clone(self):
  638. """Standard cloner.
  639. Note that every validator must implement the Clone() method.
  640. """
  641. return GenericValidator(self._condition, self._callback)
  642. def Validate(self, win):
  643. """Validate the contents of the given text control.
  644. """
  645. ctrl = self.GetWindow()
  646. text = ctrl.GetValue()
  647. if not self._condition(text):
  648. self._callback(ctrl)
  649. return False
  650. else:
  651. return True
  652. def TransferToWindow(self):
  653. """Transfer data from validator to window.
  654. """
  655. return True # Prevent wxDialog from complaining.
  656. def TransferFromWindow(self):
  657. """Transfer data from window to validator.
  658. """
  659. return True # Prevent wxDialog from complaining.
  660. class MapValidator(GenericValidator):
  661. """Validator for map name input
  662. See G_legal_filename()
  663. """
  664. def __init__(self):
  665. def _mapNameValidationFailed(ctrl):
  666. message = _(
  667. "Name <%(name)s> is not a valid name for GRASS map. "
  668. "Please use only ASCII characters excluding %(chars)s "
  669. "and space.") % {
  670. 'name': ctrl.GetValue(),
  671. 'chars': '/"\'@,=*~'}
  672. GError(message, caption=_("Invalid name"))
  673. GenericValidator.__init__(self,
  674. grass.legal_name,
  675. _mapNameValidationFailed)
  676. class SingleSymbolPanel(wx.Panel):
  677. """Panel for displaying one symbol.
  678. Changes background when selected. Assumes that parent will catch
  679. events emitted on mouse click. Used in gui_core::dialog::SymbolDialog.
  680. """
  681. def __init__(self, parent, symbolPath):
  682. """Panel constructor
  683. Signal symbolSelectionChanged - symbol selected
  684. - attribute 'name' (symbol name)
  685. - attribute 'doubleClick' (underlying cause)
  686. :param parent: parent (gui_core::dialog::SymbolDialog)
  687. :param symbolPath: absolute path to symbol
  688. """
  689. self.symbolSelectionChanged = Signal(
  690. 'SingleSymbolPanel.symbolSelectionChanged')
  691. wx.Panel.__init__(self, parent, id=wx.ID_ANY, style=wx.BORDER_RAISED)
  692. self.SetName(os.path.splitext(os.path.basename(symbolPath))[0])
  693. self.sBmp = wx.StaticBitmap(self, wx.ID_ANY, wx.Bitmap(symbolPath))
  694. self.selected = False
  695. self.selectColor = wx.SystemSettings.GetColour(wx.SYS_COLOUR_HIGHLIGHT)
  696. self.deselectColor = wx.SystemSettings.GetColour(wx.SYS_COLOUR_WINDOW)
  697. sizer = wx.BoxSizer()
  698. sizer.Add(
  699. self.sBmp,
  700. proportion=0,
  701. flag=wx.ALL | wx.ALIGN_CENTER,
  702. border=5)
  703. self.SetBackgroundColour(self.deselectColor)
  704. self.SetMinSize(self.GetBestSize())
  705. self.SetSizerAndFit(sizer)
  706. # binding to both (staticBitmap, Panel) necessary
  707. self.sBmp.Bind(wx.EVT_LEFT_DOWN, self.OnLeftDown)
  708. self.Bind(wx.EVT_LEFT_DOWN, self.OnLeftDown)
  709. self.Bind(wx.EVT_LEFT_DCLICK, self.OnDoubleClick)
  710. self.sBmp.Bind(wx.EVT_LEFT_DCLICK, self.OnDoubleClick)
  711. def OnLeftDown(self, event):
  712. """Panel selected, background changes"""
  713. self.selected = True
  714. self.SetBackgroundColour(self.selectColor)
  715. self.Refresh()
  716. event.Skip()
  717. self.symbolSelectionChanged.emit(
  718. name=self.GetName(), doubleClick=False)
  719. def OnDoubleClick(self, event):
  720. self.symbolSelectionChanged.emit(name=self.GetName(), doubleClick=True)
  721. def Deselect(self):
  722. """Panel deselected, background changes back to default"""
  723. self.selected = False
  724. self.SetBackgroundColour(self.deselectColor)
  725. self.Refresh()
  726. def Select(self):
  727. """Select panel, no event emitted"""
  728. self.selected = True
  729. self.SetBackgroundColour(self.selectColor)
  730. self.Refresh()
  731. class GListCtrl(wx.ListCtrl, listmix.ListCtrlAutoWidthMixin,
  732. listmix.CheckListCtrlMixin):
  733. """Generic ListCtrl with popup menu to select/deselect all
  734. items"""
  735. def __init__(self, parent):
  736. self.parent = parent
  737. wx.ListCtrl.__init__(self, parent, id=wx.ID_ANY,
  738. style=wx.LC_REPORT)
  739. listmix.CheckListCtrlMixin.__init__(self)
  740. # setup mixins
  741. listmix.ListCtrlAutoWidthMixin.__init__(self)
  742. self.Bind(wx.EVT_COMMAND_RIGHT_CLICK, self.OnPopupMenu) # wxMSW
  743. self.Bind(wx.EVT_RIGHT_UP, self.OnPopupMenu) # wxGTK
  744. def OnPopupMenu(self, event):
  745. """Show popup menu"""
  746. if self.GetItemCount() < 1:
  747. return
  748. if not hasattr(self, "popupDataID1"):
  749. self.popupDataID1 = wx.NewId()
  750. self.popupDataID2 = wx.NewId()
  751. self.Bind(wx.EVT_MENU, self.OnSelectAll, id=self.popupDataID1)
  752. self.Bind(wx.EVT_MENU, self.OnSelectNone, id=self.popupDataID2)
  753. # generate popup-menu
  754. menu = wx.Menu()
  755. menu.Append(self.popupDataID1, _("Select all"))
  756. menu.Append(self.popupDataID2, _("Deselect all"))
  757. self.PopupMenu(menu)
  758. menu.Destroy()
  759. def OnSelectAll(self, event):
  760. """Select all items"""
  761. item = -1
  762. while True:
  763. item = self.GetNextItem(item)
  764. if item == -1:
  765. break
  766. self.CheckItem(item, True)
  767. event.Skip()
  768. def OnSelectNone(self, event):
  769. """Deselect items"""
  770. item = -1
  771. while True:
  772. item = self.GetNextItem(item, wx.LIST_STATE_SELECTED)
  773. if item == -1:
  774. break
  775. self.CheckItem(item, False)
  776. event.Skip()
  777. def GetData(self, checked=None):
  778. """Get list data"""
  779. data = []
  780. checkedList = []
  781. item = -1
  782. while True:
  783. row = []
  784. item = self.GetNextItem(item)
  785. if item == -1:
  786. break
  787. isChecked = self.IsChecked(item)
  788. if checked is not None and checked != isChecked:
  789. continue
  790. checkedList.append(isChecked)
  791. for i in range(self.GetColumnCount()):
  792. row.append(self.GetItem(item, i).GetText())
  793. row.append(item)
  794. data.append(tuple(row))
  795. if checked is not None:
  796. return tuple(data)
  797. else:
  798. return (tuple(data), tuple(checkedList))
  799. def LoadData(self, data=None, selectOne=True):
  800. """Load data into list"""
  801. self.DeleteAllItems()
  802. if data is None:
  803. return
  804. idx = 0
  805. for item in data:
  806. index = self.InsertStringItem(idx, str(item[0]))
  807. for i in range(1, self.GetColumnCount()):
  808. self.SetStringItem(index, i, item[i])
  809. idx += 1
  810. # check by default only on one item
  811. if len(data) == 1 and selectOne:
  812. self.CheckItem(index, True)
  813. class SearchModuleWidget(wx.Panel):
  814. """Search module widget (used e.g. in SearchModuleWindow)
  815. Signals:
  816. moduleSelected - attribute 'name' is module name
  817. showSearchResult - attribute 'result' is a node (representing module)
  818. showNotification - attribute 'message'
  819. """
  820. def __init__(self, parent, model,
  821. showChoice=True, showTip=False, **kwargs):
  822. self._showTip = showTip
  823. self._showChoice = showChoice
  824. self._model = model
  825. self._results = [] # list of found nodes
  826. self._resultIndex = -1
  827. self._searchKeys = ['description', 'keywords', 'command']
  828. self._oldValue = ''
  829. self.moduleSelected = Signal('SearchModuleWidget.moduleSelected')
  830. self.showSearchResult = Signal('SearchModuleWidget.showSearchResult')
  831. self.showNotification = Signal('SearchModuleWidget.showNotification')
  832. wx.Panel.__init__(self, parent=parent, id=wx.ID_ANY, **kwargs)
  833. # self._box = wx.StaticBox(parent = self, id = wx.ID_ANY,
  834. # label = " %s " % _("Find module - (press Enter for next match)"))
  835. if sys.platform == 'win32':
  836. self._search = wx.TextCtrl(
  837. parent=self, id=wx.ID_ANY, size=(-1, 25),
  838. style=wx.TE_PROCESS_ENTER)
  839. else:
  840. self._search = SearchCtrl(
  841. parent=self, id=wx.ID_ANY, size=(-1, 25),
  842. style=wx.TE_PROCESS_ENTER)
  843. self._search.SetDescriptiveText(_('Fulltext search'))
  844. self._search.SetToolTip(
  845. _("Type to search in all modules. Press Enter for next match."))
  846. self._search.Bind(wx.EVT_TEXT, self.OnSearchModule)
  847. self._search.Bind(wx.EVT_TEXT_ENTER, self.OnEnter)
  848. if self._showTip:
  849. self._searchTip = StaticWrapText(parent=self, id=wx.ID_ANY,
  850. label="Choose a module", size=(-1, 35))
  851. if self._showChoice:
  852. self._searchChoice = wx.Choice(parent=self, id=wx.ID_ANY)
  853. self._searchChoice.SetItems(
  854. self._searchModule(
  855. keys=['command'], value=''))
  856. self._searchChoice.Bind(wx.EVT_CHOICE, self.OnSelectModule)
  857. self._layout()
  858. def _layout(self):
  859. """Do layout"""
  860. sizer = wx.BoxSizer(wx.HORIZONTAL)
  861. boxSizer = wx.BoxSizer(wx.VERTICAL)
  862. boxSizer.Add(self._search,
  863. flag=wx.ALIGN_CENTER_VERTICAL | wx.EXPAND | wx.BOTTOM,
  864. border=5)
  865. if self._showChoice:
  866. hSizer = wx.BoxSizer(wx.HORIZONTAL)
  867. hSizer.Add(self._searchChoice,
  868. flag=wx.ALIGN_CENTER_VERTICAL | wx.EXPAND | wx.BOTTOM,
  869. border=5)
  870. hSizer.AddStretchSpacer()
  871. boxSizer.Add(hSizer, flag=wx.EXPAND)
  872. if self._showTip:
  873. boxSizer.Add(self._searchTip,
  874. flag=wx.ALIGN_CENTER_VERTICAL | wx.EXPAND)
  875. sizer.Add(boxSizer, proportion=1)
  876. self.SetSizer(sizer)
  877. sizer.Fit(self)
  878. def OnEnter(self, event):
  879. """Process EVT_TEXT_ENTER to show search results"""
  880. self._showSearchResult()
  881. event.Skip()
  882. def _showSearchResult(self):
  883. if self._results:
  884. self._resultIndex += 1
  885. if self._resultIndex == len(self._results):
  886. self._resultIndex = 0
  887. self.showSearchResult.emit(result=self._results[self._resultIndex])
  888. def OnSearchModule(self, event):
  889. """Search module by keywords or description"""
  890. value = self._search.GetValue()
  891. if value == self._oldValue:
  892. event.Skip()
  893. return
  894. self._oldValue = value
  895. if len(value) <= 2:
  896. if len(value) == 0: # reset
  897. commands = self._searchModule(keys=['command'], value='')
  898. else:
  899. self.showNotification.emit(
  900. message=_("Searching, please type more characters."))
  901. return
  902. else:
  903. commands = self._searchModule(keys=self._searchKeys, value=value)
  904. if self._showChoice:
  905. self._searchChoice.SetItems(commands)
  906. if commands:
  907. self._searchChoice.SetSelection(0)
  908. label = _("%d modules match") % len(commands)
  909. if self._showTip:
  910. self._searchTip.SetLabel(label)
  911. self.showNotification.emit(message=label)
  912. event.Skip()
  913. def _searchModule(self, keys, value):
  914. """Search modules by keys
  915. :param keys: list of keys
  916. :param value: patter to match
  917. """
  918. nodes = set()
  919. for key in keys:
  920. nodes.update(self._model.SearchNodes(key=key, value=value))
  921. nodes = list(nodes)
  922. nodes.sort(key=lambda node: self._model.GetIndexOfNode(node))
  923. self._results = nodes
  924. self._resultIndex = -1
  925. commands = sorted([node.data['command']
  926. for node in nodes if node.data['command']])
  927. return commands
  928. def OnSelectModule(self, event):
  929. """Module selected from choice, update command prompt"""
  930. cmd = self._searchChoice.GetStringSelection()
  931. self.moduleSelected.emit(name=cmd)
  932. if self._showTip:
  933. for module in self._results:
  934. if cmd == module.data['command']:
  935. self._searchTip.SetLabel(module.data['description'])
  936. break
  937. def Reset(self):
  938. """Reset widget"""
  939. self._search.SetValue('')
  940. if self._showTip:
  941. self._searchTip.SetLabel('Choose a module')
  942. class ManageSettingsWidget(wx.Panel):
  943. """Widget which allows loading and saving settings into file."""
  944. def __init__(self, parent, settingsFile):
  945. """
  946. Signals:
  947. settingsChanged - called when users changes setting
  948. - attribute 'data' with chosen setting data
  949. settingsSaving - called when settings are saving
  950. - attribute 'name' with chosen settings name
  951. settingsLoaded - called when settings are loaded
  952. - attribute 'settings' is dict with loaded settings
  953. {nameofsetting : settingdata, ....}
  954. :param settingsFile: path to file, where settings will be saved and loaded from
  955. """
  956. self.settingsFile = settingsFile
  957. self.settingsChanged = Signal('ManageSettingsWidget.settingsChanged')
  958. self.settingsSaving = Signal('ManageSettingsWidget.settingsSaving')
  959. self.settingsLoaded = Signal('ManageSettingsWidget.settingsLoaded')
  960. wx.Panel.__init__(self, parent=parent, id=wx.ID_ANY)
  961. self.settingsBox = wx.StaticBox(parent=self, id=wx.ID_ANY,
  962. label=" %s " % _("Profiles"))
  963. self.settingsChoice = wx.Choice(parent=self, id=wx.ID_ANY)
  964. self.settingsChoice.Bind(wx.EVT_CHOICE, self.OnSettingsChanged)
  965. self.btnSettingsSave = Button(parent=self, id=wx.ID_SAVE)
  966. self.btnSettingsSave.Bind(wx.EVT_BUTTON, self.OnSettingsSave)
  967. self.btnSettingsSave.SetToolTip(_("Save current settings"))
  968. self.btnSettingsDel = Button(parent=self, id=wx.ID_REMOVE)
  969. self.btnSettingsDel.Bind(wx.EVT_BUTTON, self.OnSettingsDelete)
  970. self.btnSettingsSave.SetToolTip(
  971. _("Delete currently selected settings"))
  972. # escaping with '$' character - index in self.esc_chars
  973. self.e_char_i = 0
  974. self.esc_chars = ['$', ';']
  975. self._settings = self._loadSettings() # -> self.settingsChoice.SetItems()
  976. self.settingsLoaded.emit(settings=self._settings)
  977. self.data_to_save = []
  978. self._layout()
  979. self.SetSizer(self.settingsSizer)
  980. self.settingsSizer.Fit(self)
  981. def _layout(self):
  982. self.settingsSizer = wx.StaticBoxSizer(self.settingsBox, wx.HORIZONTAL)
  983. self.settingsSizer.Add(
  984. wx.StaticText(
  985. parent=self,
  986. id=wx.ID_ANY,
  987. label=_("Load:")),
  988. flag=wx.ALIGN_CENTER_VERTICAL | wx.RIGHT | wx.LEFT,
  989. border=5)
  990. self.settingsSizer.Add(
  991. self.settingsChoice,
  992. proportion=1,
  993. flag=wx.EXPAND | wx.BOTTOM,
  994. border=3)
  995. self.settingsSizer.Add(self.btnSettingsSave,
  996. flag=wx.LEFT | wx.RIGHT | wx.BOTTOM, border=3)
  997. self.settingsSizer.Add(self.btnSettingsDel,
  998. flag=wx.RIGHT | wx.BOTTOM, border=3)
  999. def OnSettingsChanged(self, event):
  1000. """Load named settings"""
  1001. name = event.GetString()
  1002. if name not in self._settings:
  1003. GError(parent=self,
  1004. message=_("Settings <%s> not found") % name)
  1005. return
  1006. data = self._settings[name]
  1007. self.settingsChanged.emit(data=data)
  1008. def GetSettings(self):
  1009. """Load named settings"""
  1010. return self._settings.copy()
  1011. def OnSettingsSave(self, event):
  1012. """Save settings"""
  1013. dlg = wx.TextEntryDialog(parent=self,
  1014. message=_("Name:"),
  1015. caption=_("Save settings"))
  1016. if dlg.ShowModal() == wx.ID_OK:
  1017. name = dlg.GetValue()
  1018. if not name:
  1019. GMessage(parent=self,
  1020. message=_("Name not given, settings is not saved."))
  1021. else:
  1022. self.settingsSaving.emit(name=name)
  1023. dlg.Destroy()
  1024. def SaveSettings(self, name):
  1025. # check if settings item already exists
  1026. if name in self._settings:
  1027. dlgOwt = wx.MessageDialog(
  1028. self,
  1029. message=_(
  1030. "Settings <%s> already exists. "
  1031. "Do you want to overwrite the settings?") %
  1032. name,
  1033. caption=_("Save settings"),
  1034. style=wx.YES_NO | wx.YES_DEFAULT | wx.ICON_QUESTION)
  1035. if dlgOwt.ShowModal() != wx.ID_YES:
  1036. dlgOwt.Destroy()
  1037. return
  1038. if self.data_to_save:
  1039. self._settings[name] = self.data_to_save
  1040. self._saveSettings()
  1041. self.settingsChoice.SetStringSelection(name)
  1042. self.data_to_save = []
  1043. def _saveSettings(self):
  1044. """Save settings and reload if successful"""
  1045. if self._writeSettings() == 0:
  1046. self._settings = self._loadSettings()
  1047. def SetDataToSave(self, data):
  1048. """Set data for setting, which will be saved.
  1049. :param data: - list of strings, which will be saved
  1050. """
  1051. self.data_to_save = data
  1052. def SetSettings(self, settings):
  1053. """Set settings
  1054. :param settings: - dict with all settigs {nameofsetting : settingdata, ....}
  1055. """
  1056. self._settings = settings
  1057. self._saveSettings()
  1058. def AddSettings(self, settings):
  1059. """Add settings
  1060. :param settings: - dict with all settigs {nameofsetting : settingdata, ....}
  1061. """
  1062. self._settings = dict(self._settings.items() + settings.items())
  1063. self._saveSettings()
  1064. def OnSettingsDelete(self, event):
  1065. """Save settings
  1066. """
  1067. name = self.settingsChoice.GetStringSelection()
  1068. if not name:
  1069. GMessage(parent=self,
  1070. message=_("No settings is defined. Operation canceled."))
  1071. return
  1072. self._settings.pop(name)
  1073. if self._writeSettings() == 0:
  1074. self._settings = self._loadSettings()
  1075. def _writeSettings(self):
  1076. """Save settings into the file
  1077. :return: 0 on success
  1078. :return: -1 on failure
  1079. """
  1080. try:
  1081. fd = open(self.settingsFile, 'w')
  1082. fd.write('format_version=2.0\n')
  1083. for key, values in self._settings.iteritems():
  1084. first = True
  1085. for v in values:
  1086. # escaping characters
  1087. for e_ch in self.esc_chars:
  1088. v = v.replace(
  1089. e_ch, self.esc_chars[
  1090. self.e_char_i] + e_ch)
  1091. if first:
  1092. # escaping characters
  1093. for e_ch in self.esc_chars:
  1094. key = key.replace(
  1095. e_ch, self.esc_chars[
  1096. self.e_char_i] + e_ch)
  1097. fd.write('%s;%s;' % (key, v))
  1098. first = False
  1099. else:
  1100. fd.write('%s;' % (v))
  1101. fd.write('\n')
  1102. except IOError:
  1103. GError(parent=self,
  1104. message=_("Unable to save settings"))
  1105. return -1
  1106. fd.close()
  1107. return 0
  1108. def _loadSettings(self):
  1109. """Load settings from the file
  1110. The file is defined by self.SettingsFile.
  1111. :return: parsed dict
  1112. :return: empty dict on error
  1113. """
  1114. data = dict()
  1115. if not os.path.exists(self.settingsFile):
  1116. return data
  1117. try:
  1118. fd = open(self.settingsFile, 'r')
  1119. except IOError:
  1120. return data
  1121. fd_lines = fd.readlines()
  1122. if not fd_lines:
  1123. fd.close()
  1124. return data
  1125. if fd_lines[0].strip() == 'format_version=2.0':
  1126. data = self._loadSettings_v2(fd_lines)
  1127. else:
  1128. data = self._loadSettings_v1(fd_lines)
  1129. self.settingsChoice.SetItems(sorted(data.keys()))
  1130. fd.close()
  1131. self.settingsLoaded.emit(settings=data)
  1132. return data
  1133. def _loadSettings_v2(self, fd_lines):
  1134. """Load settings from the file in format version 2.0
  1135. The file is defined by self.SettingsFile.
  1136. :return: parsed dict
  1137. :return: empty dict on error
  1138. """
  1139. data = dict()
  1140. for line in fd_lines[1:]:
  1141. try:
  1142. lineData = []
  1143. line = line.rstrip('\n')
  1144. i_last_found = i_last = 0
  1145. key = ''
  1146. while True:
  1147. idx = line.find(';', i_last)
  1148. if idx < 0:
  1149. break
  1150. elif idx != 0:
  1151. # find out whether it is separator
  1152. # $$$$; - it is separator
  1153. # $$$$$; - it is not separator
  1154. i_esc_chars = 0
  1155. while True:
  1156. if line[idx - (i_esc_chars + 1)
  1157. ] == self.esc_chars[self.e_char_i]:
  1158. i_esc_chars += 1
  1159. else:
  1160. break
  1161. if i_esc_chars % 2 != 0:
  1162. i_last = idx + 1
  1163. continue
  1164. lineItem = line[i_last_found: idx]
  1165. # unescape characters
  1166. for e_ch in self.esc_chars:
  1167. lineItem = lineItem.replace(
  1168. self.esc_chars[self.e_char_i] + e_ch, e_ch)
  1169. if i_last_found == 0:
  1170. key = lineItem
  1171. else:
  1172. lineData.append(lineItem)
  1173. i_last_found = i_last = idx + 1
  1174. if key and lineData:
  1175. data[key] = lineData
  1176. except ValueError:
  1177. pass
  1178. return data
  1179. def _loadSettings_v1(self, fd_lines):
  1180. """Load settings from the file in format version 1.0 (backward compatibility)
  1181. The file is defined by self.SettingsFile.
  1182. :return: parsed dict
  1183. :return: empty dict on error
  1184. """
  1185. data = dict()
  1186. for line in fd_lines:
  1187. try:
  1188. lineData = line.rstrip('\n').split(';')
  1189. if len(lineData) > 4:
  1190. # type, dsn, format, options
  1191. data[
  1192. lineData[0]] = (
  1193. lineData[1],
  1194. lineData[2],
  1195. lineData[3],
  1196. lineData[4])
  1197. else:
  1198. data[
  1199. lineData[0]] = (
  1200. lineData[1],
  1201. lineData[2],
  1202. lineData[3],
  1203. '')
  1204. except ValueError:
  1205. pass
  1206. return data
  1207. class PictureComboBox(OwnerDrawnComboBox):
  1208. """Abstract class of ComboBox with pictures.
  1209. Derived class has to specify has to specify _getPath method.
  1210. """
  1211. def OnDrawItem(self, dc, rect, item, flags):
  1212. """Overridden from OwnerDrawnComboBox.
  1213. Called to draw each item in the list.
  1214. """
  1215. if item == wx.NOT_FOUND:
  1216. # painting the control, but there is no valid item selected yet
  1217. return
  1218. r = wx.Rect(*rect) # make a copy
  1219. r.Deflate(3, 5)
  1220. # for painting the items in the popup
  1221. bitmap = self.GetPictureBitmap(self.GetString(item))
  1222. if bitmap:
  1223. dc.DrawBitmap(
  1224. bitmap, r.x, r.y + (r.height - bitmap.GetHeight()) / 2)
  1225. width = bitmap.GetWidth() + 10
  1226. else:
  1227. width = 0
  1228. dc.DrawText(self.GetString(item),
  1229. r.x + width,
  1230. (r.y + 0) + (r.height - dc.GetCharHeight()) / 2)
  1231. def OnMeasureItem(self, item):
  1232. """Overridden from OwnerDrawnComboBox, should return the height.
  1233. Needed to display an item in the popup, or -1 for default.
  1234. """
  1235. return 24
  1236. def GetPictureBitmap(self, name):
  1237. """Returns bitmap for given picture name.
  1238. :param str colorTable: name of color table
  1239. """
  1240. if not hasattr(self, 'bitmaps'):
  1241. self.bitmaps = {}
  1242. if name in self.bitmaps:
  1243. return self.bitmaps[name]
  1244. path = self._getPath(name)
  1245. if os.path.exists(path):
  1246. bitmap = wx.Bitmap(path)
  1247. self.bitmaps[name] = bitmap
  1248. return bitmap
  1249. return None
  1250. class ColorTablesComboBox(PictureComboBox):
  1251. """ComboBox with drawn color tables (created by thumbnails.py).
  1252. Used in r(3).colors dialog."""
  1253. def _getPath(self, name):
  1254. return os.path.join(
  1255. os.getenv("GISBASE"),
  1256. "docs", "html", "colortables", "%s.png" % name)
  1257. class BarscalesComboBox(PictureComboBox):
  1258. """ComboBox with barscales for d.barscale."""
  1259. def _getPath(self, name):
  1260. return os.path.join(
  1261. os.getenv("GISBASE"),
  1262. "docs", "html", "barscales", name + '.png')
  1263. class NArrowsComboBox(PictureComboBox):
  1264. """ComboBox with north arrows for d.barscale."""
  1265. def _getPath(self, name):
  1266. return os.path.join(
  1267. os.getenv("GISBASE"),
  1268. "docs", "html", "northarrows", "%s.png" % name)
  1269. class LayersList(GListCtrl, listmix.TextEditMixin):
  1270. """List of layers to be imported (dxf, shp...)"""
  1271. def __init__(self, parent, columns, log=None):
  1272. GListCtrl.__init__(self, parent)
  1273. self.log = log
  1274. # setup mixins
  1275. listmix.TextEditMixin.__init__(self)
  1276. for i in range(len(columns)):
  1277. self.InsertColumn(i, columns[i])
  1278. width = []
  1279. if len(columns) == 3:
  1280. width = (65, 200)
  1281. elif len(columns) == 4:
  1282. width = (65, 200, 90)
  1283. elif len(columns) == 5:
  1284. width = (65, 180, 90, 70)
  1285. for i in range(len(width)):
  1286. self.SetColumnWidth(col=i, width=width[i])
  1287. def OnLeftDown(self, event):
  1288. """Allow editing only output name
  1289. Code taken from TextEditMixin class.
  1290. """
  1291. x, y = event.GetPosition()
  1292. colLocs = [0]
  1293. loc = 0
  1294. for n in range(self.GetColumnCount()):
  1295. loc = loc + self.GetColumnWidth(n)
  1296. colLocs.append(loc)
  1297. col = bisect(colLocs, x + self.GetScrollPos(wx.HORIZONTAL)) - 1
  1298. if col == self.GetColumnCount() - 1:
  1299. listmix.TextEditMixin.OnLeftDown(self, event)
  1300. else:
  1301. event.Skip()
  1302. def GetLayers(self):
  1303. """Get list of layers (layer name, output name, list id)"""
  1304. layers = []
  1305. data = self.GetData(checked=True)
  1306. for itm in data:
  1307. layer = itm[1]
  1308. ftype = itm[2]
  1309. if '/' in ftype:
  1310. layer += '|%s' % ftype.split('/', 1)[0]
  1311. output = itm[self.GetColumnCount() - 1]
  1312. layers.append((layer, output, itm[-1]))
  1313. return layers