widgets.py 40 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195
  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::IntegerValidator
  13. - widgets::FloatValidator
  14. - widgets::GListCtrl
  15. - widgets::SearchModuleWidget
  16. - widgets::ManageSettingsWidget
  17. (C) 2008-2012 by the GRASS Development Team
  18. This program is free software under the GNU General Public License
  19. (>=v2). Read the file COPYING that comes with GRASS for details.
  20. @author Martin Landa <landa.martin gmail.com> (Google SoC 2008/2010)
  21. @author Enhancements by Michael Barton <michael.barton asu.edu>
  22. @author Anna Kratochvilova <kratochanna gmail.com> (Google SoC 2011)
  23. @author Stepan Turek <stepan.turek seznam.cz> (ManageSettingsWidget - created from GdalSelect)
  24. """
  25. import os
  26. import sys
  27. import string
  28. import wx
  29. import wx.lib.mixins.listctrl as listmix
  30. import wx.lib.scrolledpanel as SP
  31. try:
  32. import wx.lib.agw.flatnotebook as FN
  33. except ImportError:
  34. import wx.lib.flatnotebook as FN
  35. try:
  36. from wx.lib.buttons import ThemedGenBitmapTextButton as BitmapTextButton
  37. except ImportError: # not sure about TGBTButton version
  38. from wx.lib.buttons import GenBitmapTextButton as BitmapTextButton
  39. try:
  40. import wx.lib.agw.customtreectrl as CT
  41. except ImportError:
  42. import wx.lib.customtreectrl as CT
  43. from grass.pydispatch.signal import Signal
  44. from core import globalvar
  45. from core.gcmd import GMessage, GError
  46. from core.debug import Debug
  47. class NotebookController:
  48. """!Provides handling of notebook page names.
  49. Translates page names to page indices.
  50. Class is aggregated in notebook subclasses.
  51. Notebook subclasses must delegate methods to controller.
  52. Methods inherited from notebook class must be delegated explicitly
  53. and other methods can be delegated by @c __getattr__.
  54. """
  55. def __init__(self, classObject, widget):
  56. """!
  57. @param classObject notebook class name (object, i.e. FlatNotebook)
  58. @param widget notebook instance
  59. """
  60. self.notebookPages = {}
  61. self.classObject = classObject
  62. self.widget = widget
  63. self.highlightedTextEnd = _(" (...)")
  64. self.BindPageChanged()
  65. def BindPageChanged(self):
  66. """!Binds page changed event."""
  67. self.widget.Bind(wx.EVT_NOTEBOOK_PAGE_CHANGED, self.OnRemoveHighlight)
  68. def AddPage(self, **kwargs):
  69. """!Add a new page
  70. """
  71. if 'name' in kwargs:
  72. self.notebookPages[kwargs['name']] = kwargs['page']
  73. del kwargs['name']
  74. self.classObject.AddPage(self.widget, **kwargs)
  75. def InsertPage(self, **kwargs):
  76. """!Insert a new page
  77. """
  78. if 'name' in kwargs:
  79. self.notebookPages[kwargs['name']] = kwargs['page']
  80. del kwargs['name']
  81. self.classObject.InsertPage(self.widget, **kwargs)
  82. def DeletePage(self, page):
  83. """!Delete page
  84. @param page name
  85. @return True if page was deleted, False if not exists
  86. """
  87. delPageIndex = self.GetPageIndexByName(page)
  88. if delPageIndex != -1:
  89. ret = self.classObject.DeletePage(self.widget, delPageIndex)
  90. if ret:
  91. del self.notebookPages[page]
  92. return ret
  93. else:
  94. return False
  95. def RemovePage(self, page):
  96. """!Delete page without deleting the associated window.
  97. @param page name
  98. @return True if page was deleted, False if not exists
  99. """
  100. delPageIndex = self.GetPageIndexByName(page)
  101. if delPageIndex != -1:
  102. ret = self.classObject.RemovePage(self.widget, delPageIndex)
  103. if ret:
  104. del self.notebookPages[page]
  105. return ret
  106. else:
  107. return False
  108. def SetSelectionByName(self, page):
  109. """!Set active notebook page.
  110. @param page name, eg. 'layers', 'output', 'search', 'pyshell', 'nviz'
  111. (depends on concrete notebook instance)
  112. """
  113. idx = self.GetPageIndexByName(page)
  114. if self.classObject.GetSelection(self.widget) != idx:
  115. self.classObject.SetSelection(self.widget, idx)
  116. self.RemoveHighlight(idx)
  117. def OnRemoveHighlight(self, event):
  118. """!Highlighted tab name should be removed."""
  119. page = event.GetSelection()
  120. self.RemoveHighlight(page)
  121. def RemoveHighlight(self, page):
  122. """!Removes highlight string from notebook tab name if necessary.
  123. @param page index
  124. """
  125. text = self.classObject.GetPageText(self.widget, page)
  126. if text.endswith(self.highlightedTextEnd):
  127. text = text.replace(self.highlightedTextEnd, '')
  128. self.classObject.SetPageText(self.widget, page, text)
  129. def GetPageIndexByName(self, page):
  130. """!Get notebook page index
  131. @param page name
  132. """
  133. if page not in self.notebookPages:
  134. return -1
  135. for pageIndex in range(self.classObject.GetPageCount(self.widget)):
  136. if self.notebookPages[page] == self.classObject.GetPage(self.widget, pageIndex):
  137. break
  138. return pageIndex
  139. def HighlightPageByName(self, page):
  140. pageIndex = self.GetPageIndexByName(page)
  141. self.HighlightPage(pageIndex)
  142. def HighlightPage(self, index):
  143. if self.classObject.GetSelection(self.widget) != index:
  144. text = self.classObject.GetPageText(self.widget, index)
  145. if not text.endswith(self.highlightedTextEnd):
  146. text += self.highlightedTextEnd
  147. self.classObject.SetPageText(self.widget, index, text)
  148. def SetPageImage(self, page, index):
  149. """!Sets image index for page
  150. @param page page name
  151. @param index image index (in wx.ImageList)
  152. """
  153. pageIndex = self.GetPageIndexByName(page)
  154. self.classObject.SetPageImage(self.widget, pageIndex, index)
  155. class FlatNotebookController(NotebookController):
  156. """!Controller specialized for FN.FlatNotebook subclasses"""
  157. def __init__(self, classObject, widget):
  158. NotebookController.__init__(self, classObject, widget)
  159. def BindPageChanged(self):
  160. self.widget.Bind(FN.EVT_FLATNOTEBOOK_PAGE_CHANGED, self.OnRemoveHighlight)
  161. def GetPageIndexByName(self, page):
  162. """!Get notebook page index
  163. @param page name
  164. """
  165. if page not in self.notebookPages:
  166. return -1
  167. return self.classObject.GetPageIndex(self.widget, self.notebookPages[page])
  168. class GNotebook(FN.FlatNotebook):
  169. """!Generic notebook widget.
  170. Enables advanced style settings.
  171. Problems with hidden tabs and does not respect system colors (native look).
  172. """
  173. def __init__(self, parent, style, **kwargs):
  174. if globalvar.hasAgw:
  175. FN.FlatNotebook.__init__(self, parent, id = wx.ID_ANY, agwStyle = style, **kwargs)
  176. else:
  177. FN.FlatNotebook.__init__(self, parent, id = wx.ID_ANY, style = style, **kwargs)
  178. self.controller = FlatNotebookController(classObject = FN.FlatNotebook, widget = self)
  179. def AddPage(self, **kwargs):
  180. """! @copydoc NotebookController::AddPage()"""
  181. self.controller.AddPage(**kwargs)
  182. def InsertPage(self, **kwargs):
  183. """! @copydoc NotebookController::InsertPage()"""
  184. self.controller.InsertPage(**kwargs)
  185. def DeletePage(self, page):
  186. """! @copydoc NotebookController::DeletePage()"""
  187. return self.controller.DeletePage(page)
  188. def RemovePage(self, page):
  189. """! @copydoc NotebookController::RemovePage()"""
  190. return self.controller.RemovePage(page)
  191. def SetPageImage(self, page, index):
  192. """!Does nothing because we don't want images for this style"""
  193. pass
  194. def __getattr__(self, name):
  195. return getattr(self.controller, name)
  196. class FormNotebook(wx.Notebook):
  197. """!Notebook widget.
  198. Respects native look.
  199. """
  200. def __init__(self, parent, style):
  201. wx.Notebook.__init__(self, parent, id = wx.ID_ANY, style = style)
  202. self.controller = NotebookController(classObject = wx.Notebook, widget = self)
  203. def AddPage(self, **kwargs):
  204. """!@copydoc NotebookController::AddPage()"""
  205. self.controller.AddPage(**kwargs)
  206. def InsertPage(self, **kwargs):
  207. """! @copydoc NotebookController::InsertPage()"""
  208. self.controller.InsertPage(**kwargs)
  209. def DeletePage(self, page):
  210. """ @copydoc NotebookController::DeletePage()"""
  211. return self.controller.DeletePage(page)
  212. def RemovePage(self, page):
  213. """ @copydoc NotebookController::RemovePage()"""
  214. return self.controller.RemovePage(page)
  215. def SetPageImage(self, page, index):
  216. """! @copydoc NotebookController::SetPageImage()"""
  217. return self.controller.SetPageImage(page, index)
  218. def __getattr__(self, name):
  219. return getattr(self.controller, name)
  220. class FormListbook(wx.Listbook):
  221. """!Notebook widget.
  222. Respects native look.
  223. """
  224. def __init__(self, parent, style):
  225. wx.Listbook.__init__(self, parent, id = wx.ID_ANY, style = style)
  226. self.controller = NotebookController(classObject = wx.Listbook, widget = self)
  227. def AddPage(self, **kwargs):
  228. """!@copydoc NotebookController::AddPage()"""
  229. self.controller.AddPage(**kwargs)
  230. def InsertPage(self, **kwargs):
  231. """! @copydoc NotebookController::InsertPage()"""
  232. self.controller.InsertPage(**kwargs)
  233. def DeletePage(self, page):
  234. """ @copydoc NotebookController::DeletePage()"""
  235. return self.controller.DeletePage(page)
  236. def RemovePage(self, page):
  237. """ @copydoc NotebookController::RemovePage()"""
  238. return self.controller.RemovePage(page)
  239. def SetPageImage(self, page, index):
  240. """! @copydoc NotebookController::SetPageImage()"""
  241. return self.controller.SetPageImage(page, index)
  242. def __getattr__(self, name):
  243. return getattr(self.controller, name)
  244. class ScrolledPanel(SP.ScrolledPanel):
  245. """!Custom ScrolledPanel to avoid strange behaviour concerning focus"""
  246. def __init__(self, parent, style = wx.TAB_TRAVERSAL):
  247. SP.ScrolledPanel.__init__(self, parent = parent, id = wx.ID_ANY, style = style)
  248. def OnChildFocus(self, event):
  249. pass
  250. class NumTextCtrl(wx.TextCtrl):
  251. """!Class derived from wx.TextCtrl for numerical values only"""
  252. def __init__(self, parent, **kwargs):
  253. ## self.precision = kwargs.pop('prec')
  254. wx.TextCtrl.__init__(self, parent = parent,
  255. validator = NTCValidator(flag = 'DIGIT_ONLY'), **kwargs)
  256. def SetValue(self, value):
  257. super(NumTextCtrl, self).SetValue( str(value))
  258. def GetValue(self):
  259. val = super(NumTextCtrl, self).GetValue()
  260. if val == '':
  261. val = '0'
  262. try:
  263. return float(val)
  264. except ValueError:
  265. val = ''.join(''.join(val.split('-')).split('.'))
  266. return float(val)
  267. def SetRange(self, min, max):
  268. pass
  269. class FloatSlider(wx.Slider):
  270. """!Class derived from wx.Slider for floats"""
  271. def __init__(self, **kwargs):
  272. Debug.msg(1, "FloatSlider.__init__()")
  273. wx.Slider.__init__(self, **kwargs)
  274. self.coef = 1.
  275. #init range
  276. self.minValueOrig = 0
  277. self.maxValueOrig = 1
  278. def SetValue(self, value):
  279. value *= self.coef
  280. if abs(value) < 1 and value != 0:
  281. while abs(value) < 1:
  282. value *= 100
  283. self.coef *= 100
  284. super(FloatSlider, self).SetRange(self.minValueOrig * self.coef, self.maxValueOrig * self.coef)
  285. super(FloatSlider, self).SetValue(value)
  286. Debug.msg(4, "FloatSlider.SetValue(): value = %f" % value)
  287. def SetRange(self, minValue, maxValue):
  288. self.coef = 1.
  289. self.minValueOrig = minValue
  290. self.maxValueOrig = maxValue
  291. if abs(minValue) < 1 or abs(maxValue) < 1:
  292. while (abs(minValue) < 1 and minValue != 0) or (abs(maxValue) < 1 and maxValue != 0):
  293. minValue *= 100
  294. maxValue *= 100
  295. self.coef *= 100
  296. super(FloatSlider, self).SetValue(super(FloatSlider, self).GetValue() * self.coef)
  297. super(FloatSlider, self).SetRange(minValue, maxValue)
  298. Debug.msg(4, "FloatSlider.SetRange(): minValue = %f, maxValue = %f" % (minValue, maxValue))
  299. def GetValue(self):
  300. val = super(FloatSlider, self).GetValue()
  301. Debug.msg(4, "FloatSlider.GetValue(): value = %f" % (val/self.coef))
  302. return val/self.coef
  303. class SymbolButton(BitmapTextButton):
  304. """!Button with symbol and label."""
  305. def __init__(self, parent, usage, label, **kwargs):
  306. """!Constructor
  307. @param parent parent (usually wx.Panel)
  308. @param usage determines usage and picture
  309. @param label displayed label
  310. """
  311. size = (15, 15)
  312. buffer = wx.EmptyBitmap(*size)
  313. BitmapTextButton.__init__(self, parent = parent, label = " " + label, bitmap = buffer, **kwargs)
  314. dc = wx.MemoryDC()
  315. dc.SelectObject(buffer)
  316. maskColor = wx.Colour(255, 255, 255)
  317. dc.SetBrush(wx.Brush(maskColor))
  318. dc.Clear()
  319. if usage == 'record':
  320. self.DrawRecord(dc, size)
  321. elif usage == 'stop':
  322. self.DrawStop(dc, size)
  323. elif usage == 'play':
  324. self.DrawPlay(dc, size)
  325. elif usage == 'pause':
  326. self.DrawPause(dc, size)
  327. if sys.platform != "win32":
  328. buffer.SetMaskColour(maskColor)
  329. self.SetBitmapLabel(buffer)
  330. dc.SelectObject(wx.NullBitmap)
  331. def DrawRecord(self, dc, size):
  332. """!Draw record symbol"""
  333. dc.SetBrush(wx.Brush(wx.Colour(255, 0, 0)))
  334. dc.DrawCircle(size[0]/2, size[1] / 2, size[0] / 2)
  335. def DrawStop(self, dc, size):
  336. """!Draw stop symbol"""
  337. dc.SetBrush(wx.Brush(wx.Colour(50, 50, 50)))
  338. dc.DrawRectangle(0, 0, size[0], size[1])
  339. def DrawPlay(self, dc, size):
  340. """!Draw play symbol"""
  341. dc.SetBrush(wx.Brush(wx.Colour(0, 255, 0)))
  342. points = (wx.Point(0, 0), wx.Point(0, size[1]), wx.Point(size[0], size[1] / 2))
  343. dc.DrawPolygon(points)
  344. def DrawPause(self, dc, size):
  345. """!Draw pause symbol"""
  346. dc.SetBrush(wx.Brush(wx.Colour(50, 50, 50)))
  347. dc.DrawRectangle(0, 0, 2 * size[0] / 5, size[1])
  348. dc.DrawRectangle(3 * size[0] / 5, 0, 2 * size[0] / 5, size[1])
  349. class StaticWrapText(wx.StaticText):
  350. """!A Static Text field that wraps its text to fit its width,
  351. enlarging its height if necessary.
  352. """
  353. def __init__(self, parent, id = wx.ID_ANY, label = '', *args, **kwds):
  354. self.parent = parent
  355. self.originalLabel = label
  356. wx.StaticText.__init__(self, parent, id, label = '', *args, **kwds)
  357. self.SetLabel(label)
  358. self.Bind(wx.EVT_SIZE, self.OnResize)
  359. def SetLabel(self, label):
  360. self.originalLabel = label
  361. self.wrappedSize = None
  362. self.OnResize(None)
  363. def OnResize(self, event):
  364. if not getattr(self, "resizing", False):
  365. self.resizing = True
  366. newSize = wx.Size(self.parent.GetSize().width - 50,
  367. self.GetSize().height)
  368. if self.wrappedSize != newSize:
  369. wx.StaticText.SetLabel(self, self.originalLabel)
  370. self.Wrap(newSize.width)
  371. self.wrappedSize = newSize
  372. self.SetSize(self.wrappedSize)
  373. del self.resizing
  374. class BaseValidator(wx.PyValidator):
  375. def __init__(self):
  376. wx.PyValidator.__init__(self)
  377. self.Bind(wx.EVT_TEXT, self.OnText)
  378. def OnText(self, event):
  379. """!Do validation"""
  380. self.Validate()
  381. event.Skip()
  382. def Validate(self):
  383. """Validate input"""
  384. textCtrl = self.GetWindow()
  385. text = textCtrl.GetValue()
  386. if text:
  387. try:
  388. self.type(text)
  389. except ValueError:
  390. textCtrl.SetBackgroundColour("grey")
  391. textCtrl.SetFocus()
  392. textCtrl.Refresh()
  393. return False
  394. sysColor = wx.SystemSettings_GetColour(wx.SYS_COLOUR_WINDOW)
  395. textCtrl.SetBackgroundColour(sysColor)
  396. textCtrl.Refresh()
  397. return True
  398. def TransferToWindow(self):
  399. return True # Prevent wxDialog from complaining.
  400. def TransferFromWindow(self):
  401. return True # Prevent wxDialog from complaining.
  402. class IntegerValidator(BaseValidator):
  403. """!Validator for floating-point input"""
  404. def __init__(self):
  405. BaseValidator.__init__(self)
  406. self.type = int
  407. def Clone(self):
  408. """!Clone validator"""
  409. return IntegerValidator()
  410. class FloatValidator(BaseValidator):
  411. """!Validator for floating-point input"""
  412. def __init__(self):
  413. BaseValidator.__init__(self)
  414. self.type = float
  415. def Clone(self):
  416. """!Clone validator"""
  417. return FloatValidator()
  418. class NTCValidator(wx.PyValidator):
  419. """!validates input in textctrls, taken from wxpython demo"""
  420. def __init__(self, flag = None):
  421. wx.PyValidator.__init__(self)
  422. self.flag = flag
  423. self.Bind(wx.EVT_CHAR, self.OnChar)
  424. def Clone(self):
  425. return NTCValidator(self.flag)
  426. def OnChar(self, event):
  427. key = event.GetKeyCode()
  428. if key < wx.WXK_SPACE or key == wx.WXK_DELETE or key > 255:
  429. event.Skip()
  430. return
  431. if self.flag == 'DIGIT_ONLY' and chr(key) in string.digits + '.-':
  432. event.Skip()
  433. return
  434. if not wx.Validator_IsSilent():
  435. wx.Bell()
  436. # Returning without calling even.Skip eats the event before it
  437. # gets to the text control
  438. return
  439. class SimpleValidator(wx.PyValidator):
  440. """ This validator is used to ensure that the user has entered something
  441. into the text object editor dialog's text field.
  442. """
  443. def __init__(self, callback):
  444. """ Standard constructor.
  445. """
  446. wx.PyValidator.__init__(self)
  447. self.callback = callback
  448. def Clone(self):
  449. """ Standard cloner.
  450. Note that every validator must implement the Clone() method.
  451. """
  452. return SimpleValidator(self.callback)
  453. def Validate(self, win):
  454. """ Validate the contents of the given text control.
  455. """
  456. ctrl = self.GetWindow()
  457. text = ctrl.GetValue()
  458. if len(text) == 0:
  459. self.callback(ctrl)
  460. return False
  461. else:
  462. return True
  463. def TransferToWindow(self):
  464. """ Transfer data from validator to window.
  465. The default implementation returns False, indicating that an error
  466. occurred. We simply return True, as we don't do any data transfer.
  467. """
  468. return True # Prevent wxDialog from complaining.
  469. def TransferFromWindow(self):
  470. """ Transfer data from window to validator.
  471. The default implementation returns False, indicating that an error
  472. occurred. We simply return True, as we don't do any data transfer.
  473. """
  474. return True # Prevent wxDialog from complaining.
  475. class GenericValidator(wx.PyValidator):
  476. """ This validator checks condition and calls callback
  477. in case the condition is not fulfilled.
  478. """
  479. def __init__(self, condition, callback):
  480. """ Standard constructor.
  481. @param condition function which accepts string value and returns T/F
  482. @param callback function which is called when condition is not fulfilled
  483. """
  484. wx.PyValidator.__init__(self)
  485. self._condition = condition
  486. self._callback = callback
  487. def Clone(self):
  488. """ Standard cloner.
  489. Note that every validator must implement the Clone() method.
  490. """
  491. return GenericValidator(self._condition, self._callback)
  492. def Validate(self, win):
  493. """ Validate the contents of the given text control.
  494. """
  495. ctrl = self.GetWindow()
  496. text = ctrl.GetValue()
  497. if not self._condition(text):
  498. self._callback(ctrl)
  499. return False
  500. else:
  501. return True
  502. def TransferToWindow(self):
  503. """ Transfer data from validator to window.
  504. """
  505. return True # Prevent wxDialog from complaining.
  506. def TransferFromWindow(self):
  507. """ Transfer data from window to validator.
  508. """
  509. return True # Prevent wxDialog from complaining.
  510. class SingleSymbolPanel(wx.Panel):
  511. """!Panel for displaying one symbol.
  512. Changes background when selected. Assumes that parent will catch
  513. events emitted on mouse click. Used in gui_core::dialog::SymbolDialog.
  514. """
  515. def __init__(self, parent, symbolPath):
  516. """!Panel constructor
  517. Signal symbolSelectionChanged - symbol selected
  518. - attribute 'name' (symbol name)
  519. - attribute 'doubleClick' (underlying cause)
  520. @param parent parent (gui_core::dialog::SymbolDialog)
  521. @param symbolPath absolute path to symbol
  522. """
  523. self.symbolSelectionChanged = Signal('SingleSymbolPanel.symbolSelectionChanged')
  524. wx.Panel.__init__(self, parent, id = wx.ID_ANY, style = wx.BORDER_RAISED)
  525. self.SetName(os.path.splitext(os.path.basename(symbolPath))[0])
  526. self.sBmp = wx.StaticBitmap(self, wx.ID_ANY, wx.Bitmap(symbolPath))
  527. self.selected = False
  528. self.selectColor = wx.SystemSettings.GetColour(wx.SYS_COLOUR_HIGHLIGHT)
  529. self.deselectColor = wx.SystemSettings.GetColour(wx.SYS_COLOUR_WINDOW)
  530. sizer = wx.BoxSizer()
  531. sizer.Add(item = self.sBmp, proportion = 0, flag = wx.ALL | wx.ALIGN_CENTER, border = 5)
  532. self.SetBackgroundColour(self.deselectColor)
  533. self.SetMinSize(self.GetBestSize())
  534. self.SetSizerAndFit(sizer)
  535. # binding to both (staticBitmap, Panel) necessary
  536. self.sBmp.Bind(wx.EVT_LEFT_DOWN, self.OnLeftDown)
  537. self.Bind(wx.EVT_LEFT_DOWN, self.OnLeftDown)
  538. self.Bind(wx.EVT_LEFT_DCLICK, self.OnDoubleClick)
  539. self.sBmp.Bind(wx.EVT_LEFT_DCLICK, self.OnDoubleClick)
  540. def OnLeftDown(self, event):
  541. """!Panel selected, background changes"""
  542. self.selected = True
  543. self.SetBackgroundColour(self.selectColor)
  544. self.Refresh()
  545. event.Skip()
  546. self.symbolSelectionChanged.emit(name=self.GetName(), doubleClick=False)
  547. def OnDoubleClick(self, event):
  548. self.symbolSelectionChanged.emit(name=self.GetName(), doubleClick=True)
  549. def Deselect(self):
  550. """!Panel deselected, background changes back to default"""
  551. self.selected = False
  552. self.SetBackgroundColour(self.deselectColor)
  553. self.Refresh()
  554. def Select(self):
  555. """!Select panel, no event emitted"""
  556. self.selected = True
  557. self.SetBackgroundColour(self.selectColor)
  558. self.Refresh()
  559. class GListCtrl(wx.ListCtrl, listmix.ListCtrlAutoWidthMixin, listmix.CheckListCtrlMixin):
  560. """!Generic ListCtrl with popup menu to select/deselect all
  561. items"""
  562. def __init__(self, parent):
  563. self.parent = parent
  564. wx.ListCtrl.__init__(self, parent, id = wx.ID_ANY,
  565. style = wx.LC_REPORT)
  566. listmix.CheckListCtrlMixin.__init__(self)
  567. # setup mixins
  568. listmix.ListCtrlAutoWidthMixin.__init__(self)
  569. self.Bind(wx.EVT_COMMAND_RIGHT_CLICK, self.OnPopupMenu) #wxMSW
  570. self.Bind(wx.EVT_RIGHT_UP, self.OnPopupMenu) #wxGTK
  571. def LoadData(self):
  572. """!Load data into list"""
  573. pass
  574. def OnPopupMenu(self, event):
  575. """!Show popup menu"""
  576. if self.GetItemCount() < 1:
  577. return
  578. if not hasattr(self, "popupDataID1"):
  579. self.popupDataID1 = wx.NewId()
  580. self.popupDataID2 = wx.NewId()
  581. self.Bind(wx.EVT_MENU, self.OnSelectAll, id = self.popupDataID1)
  582. self.Bind(wx.EVT_MENU, self.OnSelectNone, id = self.popupDataID2)
  583. # generate popup-menu
  584. menu = wx.Menu()
  585. menu.Append(self.popupDataID1, _("Select all"))
  586. menu.Append(self.popupDataID2, _("Deselect all"))
  587. self.PopupMenu(menu)
  588. menu.Destroy()
  589. def OnSelectAll(self, event):
  590. """!Select all items"""
  591. item = -1
  592. while True:
  593. item = self.GetNextItem(item)
  594. if item == -1:
  595. break
  596. self.CheckItem(item, True)
  597. event.Skip()
  598. def OnSelectNone(self, event):
  599. """!Deselect items"""
  600. item = -1
  601. while True:
  602. item = self.GetNextItem(item, wx.LIST_STATE_SELECTED)
  603. if item == -1:
  604. break
  605. self.CheckItem(item, False)
  606. event.Skip()
  607. class SearchModuleWidget(wx.Panel):
  608. """!Search module widget (used e.g. in SearchModuleWindow)
  609. Signals:
  610. moduleSelected - attribute 'name' is module name
  611. showSearchResult - attribute 'result' is a node (representing module)
  612. showNotification - attribute 'message'
  613. """
  614. def __init__(self, parent, model,
  615. showChoice = True, showTip = False, **kwargs):
  616. self._showTip = showTip
  617. self._showChoice = showChoice
  618. self._model = model
  619. self._results = [] # list of found nodes
  620. self._resultIndex = -1
  621. self.moduleSelected = Signal('SearchModuleWidget.moduleSelected')
  622. self.showSearchResult = Signal('SearchModuleWidget.showSearchResult')
  623. self.showNotification = Signal('SearchModuleWidget.showNotification')
  624. wx.Panel.__init__(self, parent = parent, id = wx.ID_ANY, **kwargs)
  625. self._searchDict = { _('description') : 'description',
  626. _('command') : 'command',
  627. _('keywords') : 'keywords' }
  628. self._box = wx.StaticBox(parent = self, id = wx.ID_ANY,
  629. label = " %s " % _("Find module - (press Enter for next match)"))
  630. self._searchBy = wx.Choice(parent = self, id = wx.ID_ANY)
  631. items = [_('description'), _('keywords'), _('command')]
  632. datas = ['description', 'keywords', 'command']
  633. for item, data in zip(items, datas):
  634. self._searchBy.Append(item = item, clientData = data)
  635. self._searchBy.SetSelection(0)
  636. self._searchBy.Bind(wx.EVT_CHOICE, self.OnSearchModule)
  637. self._search = wx.SearchCtrl(parent = self, id = wx.ID_ANY,
  638. size = (-1, 25), style = wx.TE_PROCESS_ENTER)
  639. self._search.Bind(wx.EVT_TEXT, self.OnSearchModule)
  640. self._search.Bind(wx.EVT_KEY_UP, self.OnKeyUp)
  641. if self._showTip:
  642. self._searchTip = StaticWrapText(parent = self, id = wx.ID_ANY,
  643. size = (-1, 35))
  644. if self._showChoice:
  645. self._searchChoice = wx.Choice(parent = self, id = wx.ID_ANY)
  646. self._searchChoice.SetItems(self._searchModule(key='command', value=''))
  647. self._searchChoice.Bind(wx.EVT_CHOICE, self.OnSelectModule)
  648. self._layout()
  649. def _layout(self):
  650. """!Do layout"""
  651. sizer = wx.StaticBoxSizer(self._box, wx.HORIZONTAL)
  652. gridSizer = wx.GridBagSizer(hgap = 3, vgap = 3)
  653. gridSizer.Add(item = self._searchBy,
  654. flag = wx.ALIGN_CENTER_VERTICAL, pos = (0, 0))
  655. gridSizer.Add(item = self._search,
  656. flag = wx.ALIGN_CENTER_VERTICAL | wx.EXPAND, pos = (0, 1))
  657. row = 1
  658. if self._showChoice:
  659. gridSizer.Add(item = self._searchChoice,
  660. flag = wx.ALIGN_CENTER_VERTICAL | wx.EXPAND, pos = (row, 0), span = (1, 2))
  661. row += 1
  662. if self._showTip:
  663. gridSizer.Add(item = self._searchTip,
  664. flag = wx.ALIGN_CENTER_VERTICAL | wx.EXPAND, pos = (row, 0), span = (1, 2))
  665. row += 1
  666. gridSizer.AddGrowableCol(1)
  667. sizer.Add(item = gridSizer, proportion = 1)
  668. self.SetSizer(sizer)
  669. sizer.Fit(self)
  670. def OnKeyUp(self, event):
  671. """!Key or key combination pressed"""
  672. if event.GetKeyCode() == wx.WXK_RETURN and not event.ControlDown():
  673. if self._results:
  674. self._resultIndex += 1
  675. if self._resultIndex == len(self._results):
  676. self._resultIndex = 0
  677. self.showSearchResult.emit(result=self._results[self._resultIndex])
  678. event.Skip()
  679. def GetSelection(self):
  680. """!Get selected element"""
  681. selection = self._searchBy.GetStringSelection()
  682. return self._searchDict[selection]
  683. def SetSelection(self, i):
  684. """!Set selection element"""
  685. self._searchBy.SetSelection(i)
  686. def OnSearchModule(self, event):
  687. """!Search module by keywords or description"""
  688. commands = self._searchModule(key=self.GetSelection(), value=self._search.GetValue())
  689. if self._showChoice:
  690. self._searchChoice.SetItems(commands)
  691. if commands:
  692. self._searchChoice.SetSelection(0)
  693. label = _("%d modules match") % len(commands)
  694. if self._showTip:
  695. self._searchTip.SetLabel(label)
  696. self.showNotification.emit(message=label)
  697. event.Skip()
  698. def _searchModule(self, key, value):
  699. nodes = self._model.SearchNodes(key=key, value=value)
  700. self._results = nodes
  701. self._resultIndex = -1
  702. return [node.data['command'] for node in nodes if node.data['command']]
  703. def OnSelectModule(self, event):
  704. """!Module selected from choice, update command prompt"""
  705. cmd = self._searchChoice.GetStringSelection()
  706. self.moduleSelected.emit(name = cmd)
  707. if self._showTip:
  708. for module in self._results:
  709. if cmd == module.data['command']:
  710. self._searchTip.SetLabel(module.data['description'])
  711. break
  712. def Reset(self):
  713. """!Reset widget"""
  714. self._searchBy.SetSelection(0)
  715. self._search.SetValue('')
  716. if self._showTip:
  717. self._searchTip.SetLabel('')
  718. class ManageSettingsWidget(wx.Panel):
  719. """!Widget which allows loading and saving settings into file."""
  720. def __init__(self, parent, settingsFile, id = wx.ID_ANY):
  721. """
  722. Signals:
  723. settingsChanged - called when users changes setting
  724. - attribute 'data' with chosen setting data
  725. settingsSaving - called when settings are saving
  726. - attribute 'name' with chosen settings name
  727. settingsLoaded - called when settings are loaded
  728. - attribute 'settings' is dict with loaded settings
  729. {nameofsetting : settingdata, ....}
  730. @param settingsFile - path to file, where settings will be saved and loaded from
  731. """
  732. self.settingsFile = settingsFile
  733. self.settingsChanged = Signal('ManageSettingsWidget.settingsChanged')
  734. self.settingsSaving = Signal('ManageSettingsWidget.settingsSaving')
  735. self.settingsLoaded = Signal('ManageSettingsWidget.settingsLoaded')
  736. wx.Panel.__init__(self, parent = parent, id = wx.ID_ANY)
  737. self.settingsBox = wx.StaticBox(parent = self, id = wx.ID_ANY,
  738. label = " %s " % _("Settings"))
  739. self.settingsChoice = wx.Choice(parent = self, id = wx.ID_ANY)
  740. self.settingsChoice.Bind(wx.EVT_CHOICE, self.OnSettingsChanged)
  741. self.btnSettingsSave = wx.Button(parent = self, id = wx.ID_SAVE)
  742. self.btnSettingsSave.Bind(wx.EVT_BUTTON, self.OnSettingsSave)
  743. self.btnSettingsSave.SetToolTipString(_("Save current settings"))
  744. self.btnSettingsDel = wx.Button(parent = self, id = wx.ID_REMOVE)
  745. self.btnSettingsDel.Bind(wx.EVT_BUTTON, self.OnSettingsDelete)
  746. self.btnSettingsSave.SetToolTipString(_("Delete currently selected settings"))
  747. # escaping with '$' character - index in self.esc_chars
  748. self.e_char_i = 0
  749. self.esc_chars = ['$', ';']
  750. self._settings = self._loadSettings() # -> self.settingsChoice.SetItems()
  751. self.settingsLoaded.emit(settings=self._settings)
  752. self.data_to_save = []
  753. self._layout()
  754. def _layout(self):
  755. settingsSizer = wx.StaticBoxSizer(self.settingsBox, wx.HORIZONTAL)
  756. settingsSizer.Add(item = wx.StaticText(parent = self,
  757. id = wx.ID_ANY,
  758. label = _("Load settings:")),
  759. flag = wx.ALIGN_CENTER_VERTICAL | wx.RIGHT,
  760. border = 5)
  761. settingsSizer.Add(item = self.settingsChoice,
  762. proportion = 1,
  763. flag = wx.EXPAND)
  764. settingsSizer.Add(item = self.btnSettingsSave,
  765. flag = wx.LEFT | wx.RIGHT,
  766. border = 5)
  767. settingsSizer.Add(item = self.btnSettingsDel,
  768. flag = wx.RIGHT,
  769. border = 5)
  770. self.SetSizer(settingsSizer)
  771. settingsSizer.Fit(self)
  772. def OnSettingsChanged(self, event):
  773. """!Load named settings"""
  774. name = event.GetString()
  775. if name not in self._settings:
  776. GError(parent = self,
  777. message = _("Settings <%s> not found") % name)
  778. return
  779. data = self._settings[name]
  780. self.settingsChanged.emit(data=data)
  781. def OnSettingsSave(self, event):
  782. """!Save settings"""
  783. dlg = wx.TextEntryDialog(parent = self,
  784. message = _("Name:"),
  785. caption = _("Save settings"))
  786. if dlg.ShowModal() == wx.ID_OK:
  787. name = dlg.GetValue()
  788. if not name:
  789. GMessage(parent = self,
  790. message = _("Name not given, settings is not saved."))
  791. else:
  792. self.settingsSaving.emit(name=name)
  793. dlg.Destroy()
  794. def SaveSettings(self, name):
  795. # check if settings item already exists
  796. if name in self._settings:
  797. dlgOwt = wx.MessageDialog(self, message = _("Settings <%s> already exists. "
  798. "Do you want to overwrite the settings?") % name,
  799. caption = _("Save settings"), style = wx.YES_NO | wx.YES_DEFAULT | wx.ICON_QUESTION)
  800. if dlgOwt.ShowModal() != wx.ID_YES:
  801. dlgOwt.Destroy()
  802. return
  803. if self.data_to_save:
  804. self._settings[name] = self.data_to_save
  805. self._saveSettings()
  806. self.settingsChoice.SetStringSelection(name)
  807. self.data_to_save = []
  808. def _saveSettings(self):
  809. """!Save settings and reload if successful"""
  810. if self._writeSettings() == 0:
  811. self._settings = self._loadSettings()
  812. def SetDataToSave(self, data):
  813. """!Set data for setting, which will be saved.
  814. @param data - list of strings, which will be saved
  815. """
  816. self.data_to_save = data
  817. def SetSettings(self, settings):
  818. """!Set settings
  819. @param settings - dict with all settigs {nameofsetting : settingdata, ....}
  820. """
  821. self._settings = settings
  822. self._saveSettings()
  823. def OnSettingsDelete(self, event):
  824. """!Save settings
  825. """
  826. name = self.settingsChoice.GetStringSelection()
  827. if not name:
  828. GMessage(parent = self,
  829. message = _("No settings is defined. Operation canceled."))
  830. return
  831. self._settings.pop(name)
  832. if self._writeSettings() == 0:
  833. self._settings = self._loadSettings()
  834. def _writeSettings(self):
  835. """!Save settings into the file
  836. @return 0 on success
  837. @return -1 on failure
  838. """
  839. try:
  840. fd = open(self.settingsFile, 'w')
  841. fd.write('format_version=2.0\n')
  842. for key, values in self._settings.iteritems():
  843. first = True
  844. for v in values:
  845. # escaping characters
  846. for e_ch in self.esc_chars:
  847. v = v.replace(e_ch, self.esc_chars[self.e_char_i] + e_ch)
  848. if first:
  849. # escaping characters
  850. for e_ch in self.esc_chars:
  851. key = key.replace(e_ch, self.esc_chars[self.e_char_i] + e_ch)
  852. fd.write('%s;%s;' % (key, v))
  853. first = False
  854. else:
  855. fd.write('%s;' % (v))
  856. fd.write('\n')
  857. except IOError:
  858. GError(parent = self,
  859. message = _("Unable to save settings"))
  860. return -1
  861. fd.close()
  862. return 0
  863. def _loadSettings(self):
  864. """!Load settings from the file
  865. The file is defined by self.SettingsFile.
  866. @return parsed dict
  867. @return empty dict on error
  868. """
  869. data = dict()
  870. if not os.path.exists(self.settingsFile):
  871. return data
  872. try:
  873. fd = open(self.settingsFile, 'r')
  874. except IOError:
  875. return data
  876. fd_lines = fd.readlines()
  877. if not fd_lines:
  878. fd.close()
  879. return data
  880. if fd_lines[0].strip() == 'format_version=2.0':
  881. data = self._loadSettings_v2(fd_lines)
  882. else:
  883. data = self._loadSettings_v1(fd_lines)
  884. self.settingsChoice.SetItems(sorted(data.keys()))
  885. fd.close()
  886. self.settingsLoaded.emit(settings=data)
  887. return data
  888. def _loadSettings_v2(self, fd_lines):
  889. """Load settings from the file in format version 2.0
  890. The file is defined by self.SettingsFile.
  891. @return parsed dict
  892. @return empty dict on error
  893. """
  894. data = dict()
  895. for line in fd_lines[1:]:
  896. try:
  897. lineData = []
  898. line = line.rstrip('\n')
  899. i_last_found = i_last = 0
  900. key = ''
  901. while True:
  902. idx = line.find(';', i_last)
  903. if idx < 0:
  904. break
  905. elif idx != 0:
  906. # find out whether it is separator
  907. # $$$$; - it is separator
  908. # $$$$$; - it is not separator
  909. i_esc_chars = 0
  910. while True:
  911. if line[idx - (i_esc_chars + 1)] == self.esc_chars[self.e_char_i]:
  912. i_esc_chars += 1
  913. else:
  914. break
  915. if i_esc_chars%2 != 0:
  916. i_last = idx + 1
  917. continue
  918. lineItem = line[i_last_found : idx]
  919. # unescape characters
  920. for e_ch in self.esc_chars:
  921. lineItem = lineItem.replace(self.esc_chars[self.e_char_i] + e_ch, e_ch)
  922. if i_last_found == 0:
  923. key = lineItem
  924. else:
  925. lineData.append(lineItem)
  926. i_last_found = i_last = idx + 1
  927. if key and lineData:
  928. data[key] = lineData
  929. except ValueError:
  930. pass
  931. return data
  932. def _loadSettings_v1(self, fd_lines):
  933. """!Load settings from the file in format version 1.0 (backward compatibility)
  934. The file is defined by self.SettingsFile.
  935. @return parsed dict
  936. @return empty dict on error
  937. """
  938. data = dict()
  939. for line in fd_lines:
  940. try:
  941. lineData = line.rstrip('\n').split(';')
  942. if len(lineData) > 4:
  943. # type, dsn, format, options
  944. data[lineData[0]] = (lineData[1], lineData[2], lineData[3], lineData[4])
  945. else:
  946. data[lineData[0]] = (lineData[1], lineData[2], lineData[3], '')
  947. except ValueError:
  948. pass
  949. return data