widgets.py 43 KB

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