widgets.py 29 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859
  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. (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. """
  24. import os
  25. import sys
  26. import string
  27. import wx
  28. import wx.lib.mixins.listctrl as listmix
  29. import wx.lib.scrolledpanel as SP
  30. try:
  31. import wx.lib.agw.flatnotebook as FN
  32. except ImportError:
  33. import wx.lib.flatnotebook as FN
  34. try:
  35. from wx.lib.buttons import ThemedGenBitmapTextButton as BitmapTextButton
  36. except ImportError: # not sure about TGBTButton version
  37. from wx.lib.buttons import GenBitmapTextButton as BitmapTextButton
  38. try:
  39. import wx.lib.agw.customtreectrl as CT
  40. except ImportError:
  41. import wx.lib.customtreectrl as CT
  42. from core import globalvar
  43. from core.debug import Debug
  44. from core.events import gShowNotification
  45. from wx.lib.newevent import NewEvent
  46. wxSymbolSelectionChanged, EVT_SYMBOL_SELECTION_CHANGED = NewEvent()
  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. # TODO: (...) should be only once in wxGUI
  64. self.highlightedTextEnd = _(" (...)")
  65. def AddPage(self, **kwargs):
  66. """!Add a new page
  67. """
  68. if 'name' in kwargs:
  69. self.notebookPages[kwargs['name']] = kwargs['page']
  70. del kwargs['name']
  71. self.classObject.AddPage(self.widget, **kwargs)
  72. def InsertPage(self, **kwargs):
  73. """!Insert a new page
  74. """
  75. if 'name' in kwargs:
  76. self.notebookPages[kwargs['name']] = kwargs['page']
  77. del kwargs['name']
  78. self.classObject.InsertPage(self.widget, **kwargs)
  79. def DeletePage(self, page):
  80. """!Delete page
  81. @param page name
  82. @return True if page was deleted, False if not exists
  83. """
  84. delPageIndex = self.GetPageIndexByName(page)
  85. if delPageIndex != -1:
  86. ret = self.classObject.DeletePage(self.widget, delPageIndex)
  87. if ret:
  88. del self.notebookPages[page]
  89. return ret
  90. else:
  91. return False
  92. def RemovePage(self, page):
  93. """!Delete page without deleting the associated window.
  94. @param page name
  95. @return True if page was deleted, False if not exists
  96. """
  97. delPageIndex = self.GetPageIndexByName(page)
  98. if delPageIndex != -1:
  99. ret = self.classObject.RemovePage(self.widget, delPageIndex)
  100. if ret:
  101. del self.notebookPages[page]
  102. return ret
  103. else:
  104. return False
  105. def SetSelectionByName(self, page):
  106. """!Set active notebook page.
  107. @param page name, eg. 'layers', 'output', 'search', 'pyshell', 'nviz'
  108. (depends on concrete notebook instance)
  109. """
  110. idx = self.GetPageIndexByName(page)
  111. if self.classObject.GetSelection(self.widget) != idx:
  112. self.classObject.SetSelection(self.widget, idx)
  113. # FIXME: this code was not explicitly tested
  114. # FIXME: remove this from function lmgr.frame.Frame.OnPageChanged
  115. text = self.classObject.GetPageText(self.widget, idx)
  116. if text.endswith(self.highlightedTextEnd):
  117. text.replace(self.highlightedTextEnd, '')
  118. self.classObject.SetPageText(self.widget, idx, text)
  119. def GetPageIndexByName(self, page):
  120. """!Get notebook page index
  121. @param page name
  122. """
  123. if page not in self.notebookPages:
  124. return -1
  125. for pageIndex in range(self.classObject.GetPageCount(self.widget)):
  126. if self.notebookPages[page] == self.classObject.GetPage(self.widget, pageIndex):
  127. break
  128. return pageIndex
  129. def HighlightPageByName(self, page):
  130. pageIndex = self.GetPageIndexByName(page)
  131. self.HighlightPage(pageIndex)
  132. def HighlightPage(self, index):
  133. if self.classObject.GetSelection(self.widget) != index:
  134. text = self.classObject.GetPageText(self.widget, index)
  135. if not text.endswith(self.highlightedTextEnd):
  136. text += self.highlightedTextEnd
  137. self.classObject.SetPageText(self.widget, index, text)
  138. def SetPageImage(self, page, index):
  139. """!Sets image index for page
  140. @param page page name
  141. @param index image index (in wx.ImageList)
  142. """
  143. pageIndex = self.GetPageIndexByName(page)
  144. self.classObject.SetPageImage(self.widget, pageIndex, index)
  145. class FlatNotebookController(NotebookController):
  146. """!Controller specialized for FN.FlatNotebook subclasses"""
  147. def __init__(self, classObject, widget):
  148. NotebookController.__init__(self, classObject, widget)
  149. def GetPageIndexByName(self, page):
  150. """!Get notebook page index
  151. @param page name
  152. """
  153. if page not in self.notebookPages:
  154. return -1
  155. return self.classObject.GetPageIndex(self.widget, self.notebookPages[page])
  156. class GNotebook(FN.FlatNotebook):
  157. """!Generic notebook widget.
  158. Enables advanced style settings.
  159. Problems with hidden tabs and does not respect system colors (native look).
  160. """
  161. def __init__(self, parent, style, **kwargs):
  162. if globalvar.hasAgw:
  163. FN.FlatNotebook.__init__(self, parent, id = wx.ID_ANY, agwStyle = style, **kwargs)
  164. else:
  165. FN.FlatNotebook.__init__(self, parent, id = wx.ID_ANY, style = style, **kwargs)
  166. self.controller = FlatNotebookController(classObject = FN.FlatNotebook, widget = self)
  167. def AddPage(self, **kwargs):
  168. """! @copydoc NotebookController::AddPage()"""
  169. self.controller.AddPage(**kwargs)
  170. def InsertPage(self, **kwargs):
  171. """! @copydoc NotebookController::InsertPage()"""
  172. self.controller.InsertPage(**kwargs)
  173. def DeletePage(self, page):
  174. """! @copydoc NotebookController::DeletePage()"""
  175. return self.controller.DeletePage(page)
  176. def RemovePage(self, page):
  177. """! @copydoc NotebookController::RemovePage()"""
  178. return self.controller.RemovePage(page)
  179. def SetPageImage(self, page, index):
  180. """!Does nothing because we don't want images for this style"""
  181. pass
  182. def __getattr__(self, name):
  183. return getattr(self.controller, name)
  184. class FormNotebook(wx.Notebook):
  185. """!Notebook widget.
  186. Respects native look.
  187. """
  188. def __init__(self, parent, style):
  189. wx.Notebook.__init__(self, parent, id = wx.ID_ANY, style = style)
  190. self.controller = NotebookController(classObject = wx.Notebook, widget = self)
  191. def AddPage(self, **kwargs):
  192. """!@copydoc NotebookController::AddPage()"""
  193. self.controller.AddPage(**kwargs)
  194. def InsertPage(self, **kwargs):
  195. """! @copydoc NotebookController::InsertPage()"""
  196. self.controller.InsertPage(**kwargs)
  197. def DeletePage(self, page):
  198. """ @copydoc NotebookController::DeletePage()"""
  199. return self.controller.DeletePage(page)
  200. def RemovePage(self, page):
  201. """ @copydoc NotebookController::RemovePage()"""
  202. return self.controller.RemovePage(page)
  203. def SetPageImage(self, page, index):
  204. """! @copydoc NotebookController::SetPageImage()"""
  205. return self.controller.SetPageImage(page, index)
  206. def __getattr__(self, name):
  207. return getattr(self.controller, name)
  208. class FormListbook(wx.Listbook):
  209. """!Notebook widget.
  210. Respects native look.
  211. """
  212. def __init__(self, parent, style):
  213. wx.Listbook.__init__(self, parent, id = wx.ID_ANY, style = style)
  214. self.controller = NotebookController(classObject = wx.Listbook, widget = self)
  215. def AddPage(self, **kwargs):
  216. """!@copydoc NotebookController::AddPage()"""
  217. self.controller.AddPage(**kwargs)
  218. def InsertPage(self, **kwargs):
  219. """! @copydoc NotebookController::InsertPage()"""
  220. self.controller.InsertPage(**kwargs)
  221. def DeletePage(self, page):
  222. """ @copydoc NotebookController::DeletePage()"""
  223. return self.controller.DeletePage(page)
  224. def RemovePage(self, page):
  225. """ @copydoc NotebookController::RemovePage()"""
  226. return self.controller.RemovePage(page)
  227. def SetPageImage(self, page, index):
  228. """! @copydoc NotebookController::SetPageImage()"""
  229. return self.controller.SetPageImage(page, index)
  230. def __getattr__(self, name):
  231. return getattr(self.controller, name)
  232. class ScrolledPanel(SP.ScrolledPanel):
  233. """!Custom ScrolledPanel to avoid strange behaviour concerning focus"""
  234. def __init__(self, parent, style = wx.TAB_TRAVERSAL):
  235. SP.ScrolledPanel.__init__(self, parent = parent, id = wx.ID_ANY, style = style)
  236. def OnChildFocus(self, event):
  237. pass
  238. class NumTextCtrl(wx.TextCtrl):
  239. """!Class derived from wx.TextCtrl for numerical values only"""
  240. def __init__(self, parent, **kwargs):
  241. ## self.precision = kwargs.pop('prec')
  242. wx.TextCtrl.__init__(self, parent = parent,
  243. validator = NTCValidator(flag = 'DIGIT_ONLY'), **kwargs)
  244. def SetValue(self, value):
  245. super(NumTextCtrl, self).SetValue( str(value))
  246. def GetValue(self):
  247. val = super(NumTextCtrl, self).GetValue()
  248. if val == '':
  249. val = '0'
  250. try:
  251. return float(val)
  252. except ValueError:
  253. val = ''.join(''.join(val.split('-')).split('.'))
  254. return float(val)
  255. def SetRange(self, min, max):
  256. pass
  257. class FloatSlider(wx.Slider):
  258. """!Class derived from wx.Slider for floats"""
  259. def __init__(self, **kwargs):
  260. Debug.msg(1, "FloatSlider.__init__()")
  261. wx.Slider.__init__(self, **kwargs)
  262. self.coef = 1.
  263. #init range
  264. self.minValueOrig = 0
  265. self.maxValueOrig = 1
  266. def SetValue(self, value):
  267. value *= self.coef
  268. if abs(value) < 1 and value != 0:
  269. while abs(value) < 1:
  270. value *= 100
  271. self.coef *= 100
  272. super(FloatSlider, self).SetRange(self.minValueOrig * self.coef, self.maxValueOrig * self.coef)
  273. super(FloatSlider, self).SetValue(value)
  274. Debug.msg(4, "FloatSlider.SetValue(): value = %f" % value)
  275. def SetRange(self, minValue, maxValue):
  276. self.coef = 1.
  277. self.minValueOrig = minValue
  278. self.maxValueOrig = maxValue
  279. if abs(minValue) < 1 or abs(maxValue) < 1:
  280. while (abs(minValue) < 1 and minValue != 0) or (abs(maxValue) < 1 and maxValue != 0):
  281. minValue *= 100
  282. maxValue *= 100
  283. self.coef *= 100
  284. super(FloatSlider, self).SetValue(super(FloatSlider, self).GetValue() * self.coef)
  285. super(FloatSlider, self).SetRange(minValue, maxValue)
  286. Debug.msg(4, "FloatSlider.SetRange(): minValue = %f, maxValue = %f" % (minValue, maxValue))
  287. def GetValue(self):
  288. val = super(FloatSlider, self).GetValue()
  289. Debug.msg(4, "FloatSlider.GetValue(): value = %f" % (val/self.coef))
  290. return val/self.coef
  291. class SymbolButton(BitmapTextButton):
  292. """!Button with symbol and label."""
  293. def __init__(self, parent, usage, label, **kwargs):
  294. """!Constructor
  295. @param parent parent (usually wx.Panel)
  296. @param usage determines usage and picture
  297. @param label displayed label
  298. """
  299. size = (15, 15)
  300. buffer = wx.EmptyBitmap(*size)
  301. BitmapTextButton.__init__(self, parent = parent, label = " " + label, bitmap = buffer, **kwargs)
  302. dc = wx.MemoryDC()
  303. dc.SelectObject(buffer)
  304. maskColor = wx.Color(255, 255, 255)
  305. dc.SetBrush(wx.Brush(maskColor))
  306. dc.Clear()
  307. if usage == 'record':
  308. self.DrawRecord(dc, size)
  309. elif usage == 'stop':
  310. self.DrawStop(dc, size)
  311. elif usage == 'play':
  312. self.DrawPlay(dc, size)
  313. elif usage == 'pause':
  314. self.DrawPause(dc, size)
  315. if sys.platform != "win32":
  316. buffer.SetMaskColour(maskColor)
  317. self.SetBitmapLabel(buffer)
  318. dc.SelectObject(wx.NullBitmap)
  319. def DrawRecord(self, dc, size):
  320. """!Draw record symbol"""
  321. dc.SetBrush(wx.Brush(wx.Color(255, 0, 0)))
  322. dc.DrawCircle(size[0]/2, size[1] / 2, size[0] / 2)
  323. def DrawStop(self, dc, size):
  324. """!Draw stop symbol"""
  325. dc.SetBrush(wx.Brush(wx.Color(50, 50, 50)))
  326. dc.DrawRectangle(0, 0, size[0], size[1])
  327. def DrawPlay(self, dc, size):
  328. """!Draw play symbol"""
  329. dc.SetBrush(wx.Brush(wx.Color(0, 255, 0)))
  330. points = (wx.Point(0, 0), wx.Point(0, size[1]), wx.Point(size[0], size[1] / 2))
  331. dc.DrawPolygon(points)
  332. def DrawPause(self, dc, size):
  333. """!Draw pause symbol"""
  334. dc.SetBrush(wx.Brush(wx.Color(50, 50, 50)))
  335. dc.DrawRectangle(0, 0, 2 * size[0] / 5, size[1])
  336. dc.DrawRectangle(3 * size[0] / 5, 0, 2 * size[0] / 5, size[1])
  337. class StaticWrapText(wx.StaticText):
  338. """!A Static Text field that wraps its text to fit its width,
  339. enlarging its height if necessary.
  340. """
  341. def __init__(self, parent, id = wx.ID_ANY, label = '', *args, **kwds):
  342. self.parent = parent
  343. self.originalLabel = label
  344. wx.StaticText.__init__(self, parent, id, label = '', *args, **kwds)
  345. self.SetLabel(label)
  346. self.Bind(wx.EVT_SIZE, self.OnResize)
  347. def SetLabel(self, label):
  348. self.originalLabel = label
  349. self.wrappedSize = None
  350. self.OnResize(None)
  351. def OnResize(self, event):
  352. if not getattr(self, "resizing", False):
  353. self.resizing = True
  354. newSize = wx.Size(self.parent.GetSize().width - 50,
  355. self.GetSize().height)
  356. if self.wrappedSize != newSize:
  357. wx.StaticText.SetLabel(self, self.originalLabel)
  358. self.Wrap(newSize.width)
  359. self.wrappedSize = newSize
  360. self.SetSize(self.wrappedSize)
  361. del self.resizing
  362. class BaseValidator(wx.PyValidator):
  363. def __init__(self):
  364. wx.PyValidator.__init__(self)
  365. self.Bind(wx.EVT_TEXT, self.OnText)
  366. def OnText(self, event):
  367. """!Do validation"""
  368. self.Validate()
  369. event.Skip()
  370. def Validate(self):
  371. """Validate input"""
  372. textCtrl = self.GetWindow()
  373. text = textCtrl.GetValue()
  374. if text:
  375. try:
  376. self.type(text)
  377. except ValueError:
  378. textCtrl.SetBackgroundColour("grey")
  379. textCtrl.SetFocus()
  380. textCtrl.Refresh()
  381. return False
  382. sysColor = wx.SystemSettings_GetColour(wx.SYS_COLOUR_WINDOW)
  383. textCtrl.SetBackgroundColour(sysColor)
  384. textCtrl.Refresh()
  385. return True
  386. def TransferToWindow(self):
  387. return True # Prevent wxDialog from complaining.
  388. def TransferFromWindow(self):
  389. return True # Prevent wxDialog from complaining.
  390. class IntegerValidator(BaseValidator):
  391. """!Validator for floating-point input"""
  392. def __init__(self):
  393. BaseValidator.__init__(self)
  394. self.type = int
  395. def Clone(self):
  396. """!Clone validator"""
  397. return IntegerValidator()
  398. class FloatValidator(BaseValidator):
  399. """!Validator for floating-point input"""
  400. def __init__(self):
  401. BaseValidator.__init__(self)
  402. self.type = float
  403. def Clone(self):
  404. """!Clone validator"""
  405. return FloatValidator()
  406. class NTCValidator(wx.PyValidator):
  407. """!validates input in textctrls, taken from wxpython demo"""
  408. def __init__(self, flag = None):
  409. wx.PyValidator.__init__(self)
  410. self.flag = flag
  411. self.Bind(wx.EVT_CHAR, self.OnChar)
  412. def Clone(self):
  413. return NTCValidator(self.flag)
  414. def OnChar(self, event):
  415. key = event.GetKeyCode()
  416. if key < wx.WXK_SPACE or key == wx.WXK_DELETE or key > 255:
  417. event.Skip()
  418. return
  419. if self.flag == 'DIGIT_ONLY' and chr(key) in string.digits + '.-':
  420. event.Skip()
  421. return
  422. if not wx.Validator_IsSilent():
  423. wx.Bell()
  424. # Returning without calling even.Skip eats the event before it
  425. # gets to the text control
  426. return
  427. class ItemTree(CT.CustomTreeCtrl):
  428. def __init__(self, parent, id = wx.ID_ANY,
  429. ctstyle = CT.TR_HIDE_ROOT | CT.TR_FULL_ROW_HIGHLIGHT | CT.TR_HAS_BUTTONS |
  430. CT.TR_LINES_AT_ROOT | CT.TR_SINGLE, **kwargs):
  431. if globalvar.hasAgw:
  432. super(ItemTree, self).__init__(parent, id, agwStyle = ctstyle, **kwargs)
  433. else:
  434. super(ItemTree, self).__init__(parent, id, style = ctstyle, **kwargs)
  435. self.root = self.AddRoot(_("Menu tree"))
  436. self.itemsMarked = [] # list of marked items
  437. self.itemSelected = None
  438. def SearchItems(self, element, value):
  439. """!Search item
  440. @param element element index (see self.searchBy)
  441. @param value
  442. @return list of found tree items
  443. """
  444. items = list()
  445. if not value:
  446. return items
  447. item = self.GetFirstChild(self.root)[0]
  448. self._processItem(item, element, value, items)
  449. self.itemsMarked = items
  450. self.itemSelected = None
  451. return items
  452. def _processItem(self, item, element, value, listOfItems):
  453. """!Search items (used by SearchItems)
  454. @param item reference item
  455. @param listOfItems list of found items
  456. """
  457. while item and item.IsOk():
  458. subItem = self.GetFirstChild(item)[0]
  459. if subItem:
  460. self._processItem(subItem, element, value, listOfItems)
  461. data = self.GetPyData(item)
  462. if data and element in data and \
  463. value.lower() in data[element].lower():
  464. listOfItems.append(item)
  465. item = self.GetNextSibling(item)
  466. def GetSelected(self):
  467. """!Get selected item"""
  468. return self.itemSelected
  469. def OnShowItem(self, event):
  470. """!Highlight first found item in menu tree"""
  471. if len(self.itemsMarked) > 0:
  472. if self.GetSelected():
  473. self.ToggleItemSelection(self.GetSelected())
  474. idx = self.itemsMarked.index(self.GetSelected()) + 1
  475. else:
  476. idx = 0
  477. try:
  478. self.ToggleItemSelection(self.itemsMarked[idx])
  479. self.itemSelected = self.itemsMarked[idx]
  480. self.EnsureVisible(self.itemsMarked[idx])
  481. except IndexError:
  482. self.ToggleItemSelection(self.itemsMarked[0]) # reselect first item
  483. self.EnsureVisible(self.itemsMarked[0])
  484. self.itemSelected = self.itemsMarked[0]
  485. else:
  486. for item in self.root.GetChildren():
  487. self.Collapse(item)
  488. itemSelected = self.GetSelection()
  489. if itemSelected:
  490. self.ToggleItemSelection(itemSelected)
  491. self.itemSelected = None
  492. class SingleSymbolPanel(wx.Panel):
  493. """!Panel for displaying one symbol.
  494. Changes background when selected. Assumes that parent will catch
  495. events emitted on mouse click. Used in gui_core::dialog::SymbolDialog.
  496. """
  497. def __init__(self, parent, symbolPath):
  498. """!Panel constructor
  499. @param parent parent (gui_core::dialog::SymbolDialog)
  500. @param symbolPath absolute path to symbol
  501. """
  502. wx.Panel.__init__(self, parent, id = wx.ID_ANY, style = wx.BORDER_RAISED)
  503. self.SetName(os.path.splitext(os.path.basename(symbolPath))[0])
  504. self.sBmp = wx.StaticBitmap(self, wx.ID_ANY, wx.Bitmap(symbolPath))
  505. self.selected = False
  506. self.selectColor = wx.SystemSettings.GetColour(wx.SYS_COLOUR_HIGHLIGHT)
  507. self.deselectColor = wx.SystemSettings.GetColour(wx.SYS_COLOUR_WINDOW)
  508. sizer = wx.BoxSizer()
  509. sizer.Add(item = self.sBmp, proportion = 0, flag = wx.ALL | wx.ALIGN_CENTER, border = 5)
  510. self.SetBackgroundColour(self.deselectColor)
  511. self.SetMinSize(self.GetBestSize())
  512. self.SetSizerAndFit(sizer)
  513. # binding to both (staticBitmap, Panel) necessary
  514. self.sBmp.Bind(wx.EVT_LEFT_DOWN, self.OnLeftDown)
  515. self.Bind(wx.EVT_LEFT_DOWN, self.OnLeftDown)
  516. self.Bind(wx.EVT_LEFT_DCLICK, self.OnDoubleClick)
  517. self.sBmp.Bind(wx.EVT_LEFT_DCLICK, self.OnDoubleClick)
  518. def OnLeftDown(self, event):
  519. """!Panel selected, background changes"""
  520. self.selected = True
  521. self.SetBackgroundColour(self.selectColor)
  522. event.Skip()
  523. event = wxSymbolSelectionChanged(name = self.GetName(), doubleClick = False)
  524. wx.PostEvent(self.GetParent(), event)
  525. def OnDoubleClick(self, event):
  526. event = wxSymbolSelectionChanged(name = self.GetName(), doubleClick = True)
  527. wx.PostEvent(self.GetParent(), event)
  528. def Deselect(self):
  529. """!Panel deselected, background changes back to default"""
  530. self.selected = False
  531. self.SetBackgroundColour(self.deselectColor)
  532. def Select(self):
  533. """!Select panel, no event emitted"""
  534. self.selected = True
  535. self.SetBackgroundColour(self.selectColor)
  536. class GListCtrl(wx.ListCtrl, listmix.ListCtrlAutoWidthMixin, listmix.CheckListCtrlMixin):
  537. """!Generic ListCtrl with popup menu to select/deselect all
  538. items"""
  539. def __init__(self, parent):
  540. self.parent = parent
  541. wx.ListCtrl.__init__(self, parent, id = wx.ID_ANY,
  542. style = wx.LC_REPORT)
  543. listmix.CheckListCtrlMixin.__init__(self)
  544. # setup mixins
  545. listmix.ListCtrlAutoWidthMixin.__init__(self)
  546. self.Bind(wx.EVT_COMMAND_RIGHT_CLICK, self.OnPopupMenu) #wxMSW
  547. self.Bind(wx.EVT_RIGHT_UP, self.OnPopupMenu) #wxGTK
  548. def LoadData(self):
  549. """!Load data into list"""
  550. pass
  551. def OnPopupMenu(self, event):
  552. """!Show popup menu"""
  553. if self.GetItemCount() < 1:
  554. return
  555. if not hasattr(self, "popupDataID1"):
  556. self.popupDataID1 = wx.NewId()
  557. self.popupDataID2 = wx.NewId()
  558. self.Bind(wx.EVT_MENU, self.OnSelectAll, id = self.popupDataID1)
  559. self.Bind(wx.EVT_MENU, self.OnSelectNone, id = self.popupDataID2)
  560. # generate popup-menu
  561. menu = wx.Menu()
  562. menu.Append(self.popupDataID1, _("Select all"))
  563. menu.Append(self.popupDataID2, _("Deselect all"))
  564. self.PopupMenu(menu)
  565. menu.Destroy()
  566. def OnSelectAll(self, event):
  567. """!Select all items"""
  568. item = -1
  569. while True:
  570. item = self.GetNextItem(item)
  571. if item == -1:
  572. break
  573. self.CheckItem(item, True)
  574. event.Skip()
  575. def OnSelectNone(self, event):
  576. """!Deselect items"""
  577. item = -1
  578. while True:
  579. item = self.GetNextItem(item, wx.LIST_STATE_SELECTED)
  580. if item == -1:
  581. break
  582. self.CheckItem(item, False)
  583. event.Skip()
  584. gModuleSelected, EVT_MODULE_SELECTED = NewEvent()
  585. class SearchModuleWidget(wx.Panel):
  586. """!Search module widget (used in SearchModuleWindow)"""
  587. def __init__(self, parent, modulesData, id = wx.ID_ANY,
  588. showChoice = True, showTip = False, **kwargs):
  589. self.showTip = showTip
  590. self.showChoice = showChoice
  591. self.modulesData = modulesData
  592. wx.Panel.__init__(self, parent = parent, id = id, **kwargs)
  593. self._searchDict = { _('description') : 'description',
  594. _('command') : 'command',
  595. _('keywords') : 'keywords' }
  596. self.box = wx.StaticBox(parent = self, id = wx.ID_ANY,
  597. label = " %s " % _("Find module - (press Enter for next match)"))
  598. self.searchBy = wx.Choice(parent = self, id = wx.ID_ANY)
  599. items = [_('description'), _('keywords'), _('command')]
  600. datas = ['description', 'keywords', 'command']
  601. for item, data in zip(items, datas):
  602. self.searchBy.Append(item = item, clientData = data)
  603. self.searchBy.SetSelection(0)
  604. self.search = wx.SearchCtrl(parent = self, id = wx.ID_ANY,
  605. size = (-1, 25), style = wx.TE_PROCESS_ENTER)
  606. self.search.Bind(wx.EVT_TEXT, self.OnSearchModule)
  607. if self.showTip:
  608. self.searchTip = StaticWrapText(parent = self, id = wx.ID_ANY,
  609. size = (-1, 35))
  610. if self.showChoice:
  611. self.searchChoice = wx.Choice(parent = self, id = wx.ID_ANY)
  612. self.searchChoice.SetItems(self.modulesData.GetCommandItems())
  613. self.searchChoice.Bind(wx.EVT_CHOICE, self.OnSelectModule)
  614. self._layout()
  615. def _layout(self):
  616. """!Do layout"""
  617. sizer = wx.StaticBoxSizer(self.box, wx.HORIZONTAL)
  618. gridSizer = wx.GridBagSizer(hgap = 3, vgap = 3)
  619. gridSizer.Add(item = self.searchBy,
  620. flag = wx.ALIGN_CENTER_VERTICAL, pos = (0, 0))
  621. gridSizer.Add(item = self.search,
  622. flag = wx.ALIGN_CENTER_VERTICAL | wx.EXPAND, pos = (0, 1))
  623. row = 1
  624. if self.showChoice:
  625. gridSizer.Add(item = self.searchChoice,
  626. flag = wx.ALIGN_CENTER_VERTICAL | wx.EXPAND, pos = (row, 0), span = (1, 2))
  627. row += 1
  628. if self.showTip:
  629. gridSizer.Add(item = self.searchTip,
  630. flag = wx.ALIGN_CENTER_VERTICAL | wx.EXPAND, pos = (row, 0), span = (1, 2))
  631. row += 1
  632. gridSizer.AddGrowableCol(1)
  633. sizer.Add(item = gridSizer, proportion = 1)
  634. self.SetSizer(sizer)
  635. sizer.Fit(self)
  636. def GetCtrl(self):
  637. """!Get SearchCtrl widget"""
  638. return self.search
  639. def GetSelection(self):
  640. """!Get selected element"""
  641. selection = self.searchBy.GetStringSelection()
  642. return self._searchDict[selection]
  643. def SetSelection(self, i):
  644. """!Set selection element"""
  645. self.searchBy.SetSelection(i)
  646. def OnSearchModule(self, event):
  647. """!Search module by keywords or description"""
  648. text = event.GetEventObject().GetValue()
  649. if not text:
  650. self.modulesData.SetFilter()
  651. mList = self.modulesData.GetCommandItems()
  652. if self.showChoice:
  653. self.searchChoice.SetItems(mList)
  654. label = _("%d modules found") % len(mList)
  655. else:
  656. findIn = self.searchBy.GetClientData(self.searchBy.GetSelection())
  657. modules, nFound = self.modulesData.FindModules(text = text, findIn = findIn)
  658. self.modulesData.SetFilter(modules)
  659. if self.showChoice:
  660. self.searchChoice.SetItems(self.modulesData.GetCommandItems())
  661. self.searchChoice.SetSelection(0)
  662. label = _("%d modules match") % nFound
  663. if self.showTip:
  664. self.searchTip.SetLabel(label)
  665. newEvent = gShowNotification(self.GetId(), message = label)
  666. wx.PostEvent(self, newEvent)
  667. event.Skip()
  668. def OnSelectModule(self, event):
  669. """!Module selected from choice, update command prompt"""
  670. cmd = event.GetString().split(' ', 1)[0]
  671. moduleEvent = gModuleSelected(name = cmd)
  672. wx.PostEvent(self, moduleEvent)
  673. desc = self.modulesData.GetCommandDesc(cmd)
  674. if self.showTip:
  675. self.searchTip.SetLabel(desc)
  676. def Reset(self):
  677. """!Reset widget"""
  678. self.searchBy.SetSelection(0)
  679. self.search.SetValue('')
  680. if self.showTip:
  681. self.searchTip.SetLabel('')