widgets.py 48 KB

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