dialogs.py 36 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989
  1. """!
  2. @package gmodeler.dialogs
  3. @brief wxGUI Graphical Modeler - dialogs
  4. Classes:
  5. - dialogs::ModelDataDialog
  6. - dialogs::ModelSearchDialog
  7. - dialogs::ModelRelationDialog
  8. - dialogs::ModelItemDialog
  9. - dialogs::ModelLoopDialog
  10. - dialogs::ModelConditionDialog
  11. - dialogs::ModelListCtrl
  12. - dialogs::ValiableListCtrl
  13. - dialogs::ItemListCtrl
  14. - dialogs::ItemCheckListCtrl
  15. (C) 2010-2011 by the GRASS Development Team
  16. This program is free software under the GNU General Public License
  17. (>=v2). Read the file COPYING that comes with GRASS for details.
  18. @author Martin Landa <landa.martin gmail.com>
  19. """
  20. import os
  21. import sys
  22. import wx
  23. import wx.lib.mixins.listctrl as listmix
  24. from core import globalvar
  25. from core import utils
  26. from core.utils import _
  27. from gui_core.widgets import SearchModuleWidget, SimpleValidator
  28. from core.gcmd import GError, EncodeString
  29. from gui_core.dialogs import SimpleDialog, MapLayersDialogForModeler
  30. from gui_core.prompt import GPromptSTC
  31. from gui_core.forms import CmdPanel
  32. from gui_core.gselect import Select, ElementSelect
  33. from gmodeler.model import *
  34. from lmgr.menudata import LayerManagerMenuData
  35. from grass.script import task as gtask
  36. class ModelDataDialog(SimpleDialog):
  37. """!Data item properties dialog"""
  38. def __init__(self, parent, shape, title = _("Data properties")):
  39. self.parent = parent
  40. self.shape = shape
  41. label, etype = self._getLabel()
  42. self.etype = etype
  43. SimpleDialog.__init__(self, parent, title)
  44. self.element = Select(parent = self.panel,
  45. validator = SimpleValidator(callback = self.ValidatorCallback))
  46. self.element.SetValue(shape.GetValue())
  47. self.Bind(wx.EVT_BUTTON, self.OnOK, self.btnOK)
  48. self.Bind(wx.EVT_BUTTON, self.OnCancel, self.btnCancel)
  49. if self.etype:
  50. self.typeSelect = ElementSelect(parent = self.panel,
  51. size = globalvar.DIALOG_GSELECT_SIZE)
  52. self.typeSelect.Bind(wx.EVT_CHOICE, self.OnType)
  53. if shape.GetValue():
  54. self.btnOK.Enable()
  55. self._layout()
  56. self.SetMinSize(self.GetSize())
  57. def _getLabel(self):
  58. etype = False
  59. prompt = self.shape.GetPrompt()
  60. if prompt == 'raster':
  61. label = _('Name of raster map:')
  62. elif prompt == 'vector':
  63. label = _('Name of vector map:')
  64. else:
  65. etype = True
  66. label = _('Name of element:')
  67. return label, etype
  68. def _layout(self):
  69. """!Do layout"""
  70. if self.etype:
  71. self.dataSizer.Add(item = wx.StaticText(parent = self.panel, id = wx.ID_ANY,
  72. label = _("Type of element:")),
  73. proportion = 0, flag = wx.ALL, border = 1)
  74. self.dataSizer.Add(item = self.typeSelect,
  75. proportion = 0, flag = wx.ALL, border = 1)
  76. self.dataSizer.Add(item = wx.StaticText(parent = self.panel, id = wx.ID_ANY,
  77. label = _("Name of element:")),
  78. proportion = 0, flag = wx.ALL, border = 1)
  79. self.dataSizer.Add(self.element, proportion=0,
  80. flag=wx.EXPAND | wx.ALL, border=1)
  81. self.panel.SetSizer(self.sizer)
  82. self.sizer.Fit(self)
  83. def GetType(self):
  84. """!Get element type"""
  85. if not self.etype:
  86. return
  87. return self.element.tcp.GetType()
  88. def OnType(self, event):
  89. """!Select element type"""
  90. evalue = self.typeSelect.GetValue(event.GetString())
  91. self.element.SetType(evalue)
  92. def OnOK(self, event):
  93. """!Ok pressed"""
  94. self.shape.SetValue(self.element.GetValue())
  95. if self.etype:
  96. elem = self.GetType()
  97. if elem == 'rast':
  98. self.shape.SetPrompt('raster')
  99. elif elem == 'vect':
  100. self.shape.SetPrompt('raster')
  101. self.parent.canvas.Refresh()
  102. self.parent.SetStatusText('', 0)
  103. self.shape.SetPropDialog(None)
  104. if self.IsModal():
  105. event.Skip()
  106. else:
  107. self.Destroy()
  108. def OnCancel(self, event):
  109. """!Cancel pressed"""
  110. self.shape.SetPropDialog(None)
  111. if self.IsModal():
  112. event.Skip()
  113. else:
  114. self.Destroy()
  115. class ModelSearchDialog(wx.Dialog):
  116. def __init__(self, parent, title = _("Add new GRASS module to the model"),
  117. style = wx.DEFAULT_DIALOG_STYLE | wx.RESIZE_BORDER, **kwargs):
  118. """!Graphical modeler module search window
  119. @param parent parent window
  120. @param id window id
  121. @param title window title
  122. @param kwargs wx.Dialogs' arguments
  123. """
  124. self.parent = parent
  125. wx.Dialog.__init__(self, parent = parent, id = wx.ID_ANY, title = title, **kwargs)
  126. self.SetName("ModelerDialog")
  127. self.SetIcon(wx.Icon(os.path.join(globalvar.ETCICONDIR, 'grass.ico'), wx.BITMAP_TYPE_ICO))
  128. self._command = None
  129. self.panel = wx.Panel(parent = self, id = wx.ID_ANY)
  130. self.cmdBox = wx.StaticBox(parent = self.panel, id = wx.ID_ANY,
  131. label=" %s " % _("Command"))
  132. # menu data for search widget and prompt
  133. menuModel = LayerManagerMenuData()
  134. self.cmd_prompt = GPromptSTC(parent = self, menuModel = menuModel.GetModel(), updateCmdHistory = False)
  135. self.cmd_prompt.promptRunCmd.connect(self.OnCommand)
  136. self.search = SearchModuleWidget(parent = self.panel,
  137. model = menuModel.GetModel(),
  138. showTip = True)
  139. self.search.moduleSelected.connect(lambda name:
  140. self.cmd_prompt.SetTextAndFocus(name + ' '))
  141. wx.CallAfter(self.cmd_prompt.SetFocus)
  142. self.btnCancel = wx.Button(self.panel, wx.ID_CANCEL)
  143. self.btnOk = wx.Button(self.panel, wx.ID_OK)
  144. self.btnOk.SetDefault()
  145. self.Bind(wx.EVT_BUTTON, self.OnOk, self.btnOk)
  146. self.Bind(wx.EVT_BUTTON, self.OnCancel, self.btnCancel)
  147. self._layout()
  148. self.SetSize((500, 275))
  149. def _layout(self):
  150. cmdSizer = wx.StaticBoxSizer(self.cmdBox, wx.VERTICAL)
  151. cmdSizer.Add(item = self.cmd_prompt, proportion = 1,
  152. flag = wx.EXPAND)
  153. btnSizer = wx.StdDialogButtonSizer()
  154. btnSizer.AddButton(self.btnCancel)
  155. btnSizer.AddButton(self.btnOk)
  156. btnSizer.Realize()
  157. mainSizer = wx.BoxSizer(wx.VERTICAL)
  158. mainSizer.Add(item = self.search, proportion = 0,
  159. flag = wx.EXPAND | wx.ALL, border = 3)
  160. mainSizer.Add(item = cmdSizer, proportion = 1,
  161. flag = wx.EXPAND | wx.LEFT | wx.RIGHT | wx.TOP, border = 3)
  162. mainSizer.Add(item = btnSizer, proportion = 0,
  163. flag = wx.EXPAND | wx.ALL | wx.ALIGN_CENTER, border = 5)
  164. self.panel.SetSizer(mainSizer)
  165. mainSizer.Fit(self.panel)
  166. self.Layout()
  167. def GetPanel(self):
  168. """!Get dialog panel"""
  169. return self.panel
  170. def _getCmd(self):
  171. line = self.cmd_prompt.GetCurLine()[0].strip()
  172. if len(line) == 0:
  173. cmd = list()
  174. else:
  175. try:
  176. cmd = utils.split(str(line))
  177. except UnicodeError:
  178. cmd = utils.split(EncodeString((line)))
  179. return cmd
  180. def GetCmd(self):
  181. """!Get command"""
  182. return self._command
  183. def ValidateCmd(self, cmd):
  184. if len(cmd) < 1:
  185. GError(parent = self,
  186. message = _("Command not defined.\n\n"
  187. "Unable to add new action to the model."))
  188. return False
  189. if cmd[0] not in globalvar.grassCmd:
  190. GError(parent = self,
  191. message = _("'%s' is not a GRASS module.\n\n"
  192. "Unable to add new action to the model.") % cmd[0])
  193. return False
  194. return True
  195. def OnCommand(self, cmd):
  196. """!Command in prompt confirmed"""
  197. if self.ValidateCmd(cmd):
  198. self._command = cmd
  199. self.EndModal(wx.ID_OK)
  200. def OnOk(self, event):
  201. """!Button 'OK' pressed"""
  202. cmd = self._getCmd()
  203. if self.ValidateCmd(cmd):
  204. self._command = cmd
  205. self.EndModal(wx.ID_OK)
  206. def OnCancel(self, event):
  207. """Cancel pressed, close window"""
  208. self.Hide()
  209. def Reset(self):
  210. """!Reset dialog"""
  211. self.search.Reset()
  212. self.cmd_prompt.OnCmdErase(None)
  213. self.cmd_prompt.SetFocus()
  214. class ModelRelationDialog(wx.Dialog):
  215. """!Relation properties dialog"""
  216. def __init__(self, parent, shape, id = wx.ID_ANY, title = _("Relation properties"),
  217. style = wx.DEFAULT_DIALOG_STYLE | wx.RESIZE_BORDER, **kwargs):
  218. self.parent = parent
  219. self.shape = shape
  220. options = self._getOptions()
  221. if not options:
  222. self.valid = False
  223. return
  224. self.valid = True
  225. wx.Dialog.__init__(self, parent, id, title, style = style, **kwargs)
  226. self.SetIcon(wx.Icon(os.path.join(globalvar.ETCICONDIR, 'grass.ico'), wx.BITMAP_TYPE_ICO))
  227. self.panel = wx.Panel(parent = self, id = wx.ID_ANY)
  228. self.fromBox = wx.StaticBox(parent = self.panel, id = wx.ID_ANY,
  229. label = " %s " % _("From"))
  230. self.toBox = wx.StaticBox(parent = self.panel, id = wx.ID_ANY,
  231. label = " %s " % _("To"))
  232. self.option = wx.ComboBox(parent = self.panel, id = wx.ID_ANY,
  233. style = wx.CB_READONLY,
  234. choices = options)
  235. self.option.Bind(wx.EVT_COMBOBOX, self.OnOption)
  236. self.btnCancel = wx.Button(self.panel, wx.ID_CANCEL)
  237. self.btnOk = wx.Button(self.panel, wx.ID_OK)
  238. self.btnOk.Enable(False)
  239. self._layout()
  240. def _layout(self):
  241. mainSizer = wx.BoxSizer(wx.VERTICAL)
  242. fromSizer = wx.StaticBoxSizer(self.fromBox, wx.VERTICAL)
  243. self._layoutShape(shape = self.shape.GetFrom(), sizer = fromSizer)
  244. toSizer = wx.StaticBoxSizer(self.toBox, wx.VERTICAL)
  245. self._layoutShape(shape = self.shape.GetTo(), sizer = toSizer)
  246. btnSizer = wx.StdDialogButtonSizer()
  247. btnSizer.AddButton(self.btnCancel)
  248. btnSizer.AddButton(self.btnOk)
  249. btnSizer.Realize()
  250. mainSizer.Add(item = fromSizer, proportion = 0,
  251. flag = wx.EXPAND | wx.ALL, border = 5)
  252. mainSizer.Add(item = toSizer, proportion = 0,
  253. flag = wx.EXPAND | wx.LEFT | wx.RIGHT | wx.BOTTOM, border = 5)
  254. mainSizer.Add(item = btnSizer, proportion = 0,
  255. flag = wx.EXPAND | wx.ALL | wx.ALIGN_CENTER, border = 5)
  256. self.panel.SetSizer(mainSizer)
  257. mainSizer.Fit(self.panel)
  258. self.Layout()
  259. self.SetSize(self.GetBestSize())
  260. def _layoutShape(self, shape, sizer):
  261. if isinstance(shape, ModelData):
  262. sizer.Add(item = wx.StaticText(parent = self.panel, id = wx.ID_ANY,
  263. label = _("Data: %s") % shape.GetLog()),
  264. proportion = 1, flag = wx.EXPAND | wx.ALL,
  265. border = 5)
  266. elif isinstance(shape, ModelAction):
  267. gridSizer = wx.GridBagSizer (hgap = 5, vgap = 5)
  268. gridSizer.Add(item = wx.StaticText(parent = self.panel, id = wx.ID_ANY,
  269. label = _("Command:")),
  270. pos = (0, 0))
  271. gridSizer.Add(item = wx.StaticText(parent = self.panel, id = wx.ID_ANY,
  272. label = shape.GetName()),
  273. pos = (0, 1))
  274. gridSizer.Add(item = wx.StaticText(parent = self.panel, id = wx.ID_ANY,
  275. label = _("Option:")),
  276. flag = wx.ALIGN_CENTER_VERTICAL,
  277. pos = (1, 0))
  278. gridSizer.Add(item = self.option,
  279. pos = (1, 1))
  280. sizer.Add(item = gridSizer,
  281. proportion = 1, flag = wx.EXPAND | wx.ALL,
  282. border = 5)
  283. def _getOptions(self):
  284. """!Get relevant options"""
  285. items = []
  286. fromShape = self.shape.GetFrom()
  287. if not isinstance(fromShape, ModelData):
  288. GError(parent = self.parent,
  289. message = _("Relation doesn't start with data item.\n"
  290. "Unable to add relation."))
  291. return items
  292. toShape = self.shape.GetTo()
  293. if not isinstance(toShape, ModelAction):
  294. GError(parent = self.parent,
  295. message = _("Relation doesn't point to GRASS command.\n"
  296. "Unable to add relation."))
  297. return items
  298. prompt = fromShape.GetPrompt()
  299. task = toShape.GetTask()
  300. for p in task.get_options()['params']:
  301. if p.get('prompt', '') == prompt and \
  302. 'name' in p:
  303. items.append(p['name'])
  304. if not items:
  305. GError(parent = self.parent,
  306. message = _("No relevant option found.\n"
  307. "Unable to add relation."))
  308. return items
  309. def GetOption(self):
  310. """!Get selected option"""
  311. return self.option.GetStringSelection()
  312. def IsValid(self):
  313. """!Check if relation is valid"""
  314. return self.valid
  315. def OnOption(self, event):
  316. """!Set option"""
  317. if event.GetString():
  318. self.btnOk.Enable()
  319. else:
  320. self.btnOk.Enable(False)
  321. class ModelItemDialog(wx.Dialog):
  322. """!Abstract item properties dialog"""
  323. def __init__(self, parent, shape, title, id = wx.ID_ANY,
  324. style = wx.DEFAULT_DIALOG_STYLE | wx.RESIZE_BORDER, **kwargs):
  325. self.parent = parent
  326. self.shape = shape
  327. wx.Dialog.__init__(self, parent, id, title = title, style = style, **kwargs)
  328. self.panel = wx.Panel(parent = self, id = wx.ID_ANY)
  329. self.condBox = wx.StaticBox(parent = self.panel, id = wx.ID_ANY,
  330. label=" %s " % _("Condition"))
  331. self.condText = wx.TextCtrl(parent = self.panel, id = wx.ID_ANY,
  332. value = shape.GetText())
  333. self.itemList = ItemCheckListCtrl(parent = self.panel,
  334. columns = [_("Name"),
  335. _("Command")],
  336. shape = shape,
  337. frame = parent)
  338. self.itemList.Populate(self.parent.GetModel().GetItems())
  339. self.btnCancel = wx.Button(parent = self.panel, id = wx.ID_CANCEL)
  340. self.btnOk = wx.Button(parent = self.panel, id = wx.ID_OK)
  341. self.btnOk.SetDefault()
  342. def _layout(self):
  343. """!Do layout (virtual method)"""
  344. pass
  345. def GetCondition(self):
  346. """!Get loop condition"""
  347. return self.condText.GetValue()
  348. class ModelLoopDialog(ModelItemDialog):
  349. """!Loop properties dialog"""
  350. def __init__(self, parent, shape, id = wx.ID_ANY, title = _("Loop properties"),
  351. style = wx.DEFAULT_DIALOG_STYLE | wx.RESIZE_BORDER, **kwargs):
  352. ModelItemDialog.__init__(self, parent, shape, title,
  353. style = style, **kwargs)
  354. self.listBox = wx.StaticBox(parent = self.panel, id = wx.ID_ANY,
  355. label=" %s " % _("List of items in loop"))
  356. self.btnSeries = wx.Button(parent = self.panel, id = wx.ID_ANY,
  357. label = _("Series"))
  358. self.btnSeries.SetToolTipString(_("Define map series as condition for the loop"))
  359. self.btnSeries.Bind(wx.EVT_BUTTON, self.OnSeries)
  360. self._layout()
  361. self.SetMinSize(self.GetSize())
  362. self.SetSize((500, 400))
  363. def _layout(self):
  364. """!Do layout"""
  365. sizer = wx.BoxSizer(wx.VERTICAL)
  366. condSizer = wx.StaticBoxSizer(self.condBox, wx.HORIZONTAL)
  367. condSizer.Add(item = self.condText, proportion = 1,
  368. flag = wx.ALL, border = 3)
  369. condSizer.Add(item = self.btnSeries, proportion = 0,
  370. flag = wx.EXPAND)
  371. listSizer = wx.StaticBoxSizer(self.listBox, wx.VERTICAL)
  372. listSizer.Add(item = self.itemList, proportion = 1,
  373. flag = wx.EXPAND | wx.ALL, border = 3)
  374. btnSizer = wx.StdDialogButtonSizer()
  375. btnSizer.AddButton(self.btnCancel)
  376. btnSizer.AddButton(self.btnOk)
  377. btnSizer.Realize()
  378. sizer.Add(item = condSizer, proportion = 0,
  379. flag = wx.EXPAND | wx.ALL, border = 3)
  380. sizer.Add(item = listSizer, proportion = 1,
  381. flag = wx.EXPAND | wx.LEFT | wx.RIGHT, border = 3)
  382. sizer.Add(item = btnSizer, proportion=0,
  383. flag = wx.EXPAND | wx.ALL | wx.ALIGN_CENTER, border=5)
  384. self.panel.SetSizer(sizer)
  385. sizer.Fit(self.panel)
  386. self.Layout()
  387. def GetItems(self):
  388. """!Get list of selected actions"""
  389. return self.itemList.GetItems()
  390. def OnSeries(self, event):
  391. """!Define map series as condition"""
  392. dialog = MapLayersDialogForModeler(parent = self, title = _("Define series of maps"))
  393. if dialog.ShowModal() != wx.ID_OK:
  394. dialog.Destroy()
  395. return
  396. cond = dialog.GetDSeries()
  397. if not cond:
  398. cond = 'map in %s' % map(lambda x: str(x), dialog.GetMapLayers())
  399. self.condText.SetValue(cond)
  400. dialog.Destroy()
  401. class ModelConditionDialog(ModelItemDialog):
  402. """!Condition properties dialog"""
  403. def __init__(self, parent, shape, id = wx.ID_ANY, title = _("If-else properties"),
  404. style = wx.DEFAULT_DIALOG_STYLE | wx.RESIZE_BORDER, **kwargs):
  405. ModelItemDialog.__init__(self, parent, shape, title,
  406. style = style, **kwargs)
  407. self.listBoxIf = wx.StaticBox(parent = self.panel, id = wx.ID_ANY,
  408. label=" %s " % _("List of items in 'if' block"))
  409. self.itemListIf = self.itemList
  410. self.itemListIf.SetName('IfBlockList')
  411. self.listBoxElse = wx.StaticBox(parent = self.panel, id = wx.ID_ANY,
  412. label=" %s " % _("List of items in 'else' block"))
  413. self.itemListElse = ItemCheckListCtrl(parent = self.panel,
  414. window = self,
  415. columns = [_("Name"),
  416. _("Command")],
  417. shape = shape)
  418. self.itemListElse.SetName('ElseBlockList')
  419. self.itemListElse.Populate(self.parent.GetModel().GetItems())
  420. self._layout()
  421. self.SetMinSize(self.GetSize())
  422. self.SetSize((500, 400))
  423. def _layout(self):
  424. """!Do layout"""
  425. sizer = wx.BoxSizer(wx.VERTICAL)
  426. condSizer = wx.StaticBoxSizer(self.condBox, wx.VERTICAL)
  427. condSizer.Add(item = self.condText, proportion = 1,
  428. flag = wx.EXPAND)
  429. listIfSizer = wx.StaticBoxSizer(self.listBoxIf, wx.VERTICAL)
  430. listIfSizer.Add(item = self.itemListIf, proportion = 1,
  431. flag = wx.EXPAND)
  432. listElseSizer = wx.StaticBoxSizer(self.listBoxElse, wx.VERTICAL)
  433. listElseSizer.Add(item = self.itemListElse, proportion = 1,
  434. flag = wx.EXPAND)
  435. btnSizer = wx.StdDialogButtonSizer()
  436. btnSizer.AddButton(self.btnCancel)
  437. btnSizer.AddButton(self.btnOk)
  438. btnSizer.Realize()
  439. sizer.Add(item = condSizer, proportion = 0,
  440. flag = wx.EXPAND | wx.ALL, border = 3)
  441. sizer.Add(item = listIfSizer, proportion = 1,
  442. flag = wx.EXPAND | wx.LEFT | wx.RIGHT, border = 3)
  443. sizer.Add(item = listElseSizer, proportion = 1,
  444. flag = wx.EXPAND | wx.LEFT | wx.RIGHT, border = 3)
  445. sizer.Add(item = btnSizer, proportion=0,
  446. flag = wx.EXPAND | wx.ALL | wx.ALIGN_CENTER, border=5)
  447. self.panel.SetSizer(sizer)
  448. sizer.Fit(self.panel)
  449. self.Layout()
  450. def OnCheckItemIf(self, index, flag):
  451. """!Item in if-block checked/unchecked"""
  452. if flag is False:
  453. return
  454. aId = int(self.itemListIf.GetItem(index, 0).GetText())
  455. if aId in self.itemListElse.GetItems()['checked']:
  456. self.itemListElse.CheckItemById(aId, False)
  457. def OnCheckItemElse(self, index, flag):
  458. """!Item in else-block checked/unchecked"""
  459. if flag is False:
  460. return
  461. aId = int(self.itemListElse.GetItem(index, 0).GetText())
  462. if aId in self.itemListIf.GetItems()['checked']:
  463. self.itemListIf.CheckItemById(aId, False)
  464. def GetItems(self):
  465. """!Get items"""
  466. return { 'if' : self.itemListIf.GetItems(),
  467. 'else' : self.itemListElse.GetItems() }
  468. class ModelListCtrl(wx.ListCtrl,
  469. listmix.ListCtrlAutoWidthMixin,
  470. listmix.TextEditMixin,
  471. listmix.ColumnSorterMixin):
  472. def __init__(self, parent, columns, frame, id = wx.ID_ANY, columnsNotEditable = [],
  473. style = wx.LC_REPORT | wx.BORDER_NONE |
  474. wx.LC_SORT_ASCENDING |wx.LC_HRULES |
  475. wx.LC_VRULES, **kwargs):
  476. """!List of model variables"""
  477. self.parent = parent
  478. self.columns = columns
  479. self.shape = None
  480. self.frame = frame
  481. self.columnNotEditable = columnsNotEditable
  482. wx.ListCtrl.__init__(self, parent, id = id, style = style, **kwargs)
  483. listmix.ListCtrlAutoWidthMixin.__init__(self)
  484. listmix.TextEditMixin.__init__(self)
  485. listmix.ColumnSorterMixin.__init__(self, 4)
  486. i = 0
  487. for col in columns:
  488. self.InsertColumn(i, col)
  489. self.SetColumnWidth(i, wx.LIST_AUTOSIZE_USEHEADER)
  490. i += 1
  491. self.itemDataMap = {} # requested by sorter
  492. self.itemCount = 0
  493. self.Bind(wx.EVT_LIST_BEGIN_LABEL_EDIT, self.OnBeginEdit)
  494. self.Bind(wx.EVT_LIST_END_LABEL_EDIT, self.OnEndEdit)
  495. self.Bind(wx.EVT_LIST_COL_CLICK, self.OnColClick)
  496. self.Bind(wx.EVT_COMMAND_RIGHT_CLICK, self.OnRightUp) #wxMSW
  497. self.Bind(wx.EVT_RIGHT_UP, self.OnRightUp) #wxGTK
  498. def OnBeginEdit(self, event):
  499. """!Editing of item started"""
  500. if self.columnNotEditable and event.m_col in self.columnNotEditable:
  501. event.Veto()
  502. self.SetItemState(event.m_itemIndex,
  503. wx.LIST_STATE_SELECTED, wx.LIST_STATE_SELECTED | wx.LIST_STATE_FOCUSED)
  504. else:
  505. event.Allow()
  506. def OnEndEdit(self, event):
  507. """!Finish editing of item"""
  508. pass
  509. def OnColClick(self, event):
  510. """!Click on column header (order by)"""
  511. event.Skip()
  512. class VariableListCtrl(ModelListCtrl):
  513. def __init__(self, parent, columns, **kwargs):
  514. """!List of model variables"""
  515. ModelListCtrl.__init__(self, parent, columns, **kwargs)
  516. self.SetColumnWidth(2, 200) # default value
  517. def GetListCtrl(self):
  518. """!Used by ColumnSorterMixin"""
  519. return self
  520. def GetData(self):
  521. """!Get list data"""
  522. return self.itemDataMap
  523. def Populate(self, data):
  524. """!Populate the list"""
  525. self.itemDataMap = dict()
  526. i = 0
  527. for name, values in data.iteritems():
  528. self.itemDataMap[i] = [name, values['type'],
  529. values.get('value', ''),
  530. values.get('description', '')]
  531. i += 1
  532. self.itemCount = len(self.itemDataMap.keys())
  533. self.DeleteAllItems()
  534. i = 0
  535. for name, vtype, value, desc in self.itemDataMap.itervalues():
  536. index = self.InsertStringItem(sys.maxint, name)
  537. self.SetStringItem(index, 0, name)
  538. self.SetStringItem(index, 1, vtype)
  539. self.SetStringItem(index, 2, value)
  540. self.SetStringItem(index, 3, desc)
  541. self.SetItemData(index, i)
  542. i += 1
  543. def Append(self, name, vtype, value, desc):
  544. """!Append new item to the list
  545. @return None on success
  546. @return error string
  547. """
  548. for iname, ivtype, ivalue, idesc in self.itemDataMap.itervalues():
  549. if iname == name:
  550. return _("Variable <%s> already exists in the model. "
  551. "Adding variable failed.") % name
  552. index = self.InsertStringItem(sys.maxint, name)
  553. self.SetStringItem(index, 0, name)
  554. self.SetStringItem(index, 1, vtype)
  555. self.SetStringItem(index, 2, value)
  556. self.SetStringItem(index, 3, desc)
  557. self.SetItemData(index, self.itemCount)
  558. self.itemDataMap[self.itemCount] = [name, vtype, value, desc]
  559. self.itemCount += 1
  560. return None
  561. def OnRemove(self, event):
  562. """!Remove selected variable(s) from the model"""
  563. item = self.GetFirstSelected()
  564. while item != -1:
  565. self.DeleteItem(item)
  566. del self.itemDataMap[item]
  567. item = self.GetFirstSelected()
  568. self.parent.UpdateModelVariables()
  569. event.Skip()
  570. def OnRemoveAll(self, event):
  571. """!Remove all variable(s) from the model"""
  572. dlg = wx.MessageBox(parent=self,
  573. message=_("Do you want to delete all variables from "
  574. "the model?"),
  575. caption=_("Delete variables"),
  576. style=wx.YES_NO | wx.CENTRE)
  577. if dlg != wx.YES:
  578. return
  579. self.DeleteAllItems()
  580. self.itemDataMap = dict()
  581. self.parent.UpdateModelVariables()
  582. def OnEndEdit(self, event):
  583. """!Finish editing of item"""
  584. itemIndex = event.GetIndex()
  585. columnIndex = event.GetColumn()
  586. nameOld = self.GetItem(itemIndex, 0).GetText()
  587. if columnIndex == 0: # TODO
  588. event.Veto()
  589. self.itemDataMap[itemIndex][columnIndex] = event.GetText()
  590. self.parent.UpdateModelVariables()
  591. def OnReload(self, event):
  592. """!Reload list of variables"""
  593. self.Populate(self.parent.parent.GetModel().GetVariables())
  594. def OnRightUp(self, event):
  595. """!Mouse right button up"""
  596. if not hasattr(self, "popupID1"):
  597. self.popupID1 = wx.NewId()
  598. self.popupID2 = wx.NewId()
  599. self.popupID3 = wx.NewId()
  600. self.Bind(wx.EVT_MENU, self.OnRemove, id = self.popupID1)
  601. self.Bind(wx.EVT_MENU, self.OnRemoveAll, id = self.popupID2)
  602. self.Bind(wx.EVT_MENU, self.OnReload, id = self.popupID3)
  603. # generate popup-menu
  604. menu = wx.Menu()
  605. menu.Append(self.popupID1, _("Delete selected"))
  606. menu.Append(self.popupID2, _("Delete all"))
  607. if self.GetFirstSelected() == -1:
  608. menu.Enable(self.popupID1, False)
  609. menu.Enable(self.popupID2, False)
  610. menu.AppendSeparator()
  611. menu.Append(self.popupID3, _("Reload"))
  612. self.PopupMenu(menu)
  613. menu.Destroy()
  614. class ItemListCtrl(ModelListCtrl):
  615. def __init__(self, parent, columns, frame, disablePopup = False, **kwargs):
  616. """!List of model actions"""
  617. self.disablePopup = disablePopup
  618. ModelListCtrl.__init__(self, parent, columns, frame, **kwargs)
  619. self.SetColumnWidth(1, 100)
  620. self.SetColumnWidth(2, 65)
  621. def GetListCtrl(self):
  622. """!Used by ColumnSorterMixin"""
  623. return self
  624. def GetData(self):
  625. """!Get list data"""
  626. return self.itemDataMap
  627. def Populate(self, data):
  628. """!Populate the list"""
  629. self.itemDataMap = dict()
  630. if self.shape:
  631. if isinstance(self.shape, ModelCondition):
  632. if self.GetName() == 'ElseBlockList':
  633. shapeItems = map(lambda x: x.GetId(), self.shape.GetItems()['else'])
  634. else:
  635. shapeItems = map(lambda x: x.GetId(), self.shape.GetItems()['if'])
  636. else:
  637. shapeItems = map(lambda x: x.GetId(),
  638. self.shape.GetItems(self.frame.GetModel().GetItems(objType=ModelAction)))
  639. else:
  640. shapeItems = list()
  641. i = 0
  642. if len(self.columns) == 2: # ItemCheckList
  643. checked = list()
  644. for action in data:
  645. if isinstance(action, ModelData) or \
  646. action == self.shape:
  647. continue
  648. if len(self.columns) == 2:
  649. self.itemDataMap[i] = [action.GetName(),
  650. action.GetLog()]
  651. aId = action.GetBlockId()
  652. if action.GetId() in shapeItems:
  653. checked.append(aId)
  654. else:
  655. checked.append(None)
  656. else:
  657. bId = action.GetBlockId()
  658. if not bId:
  659. bId = _('No')
  660. else:
  661. bId = _("Yes")
  662. self.itemDataMap[i] = [action.GetName(),
  663. bId,
  664. action.GetLog()]
  665. i += 1
  666. self.itemCount = len(self.itemDataMap.keys())
  667. self.DeleteAllItems()
  668. i = 0
  669. if len(self.columns) == 2:
  670. for name, desc in self.itemDataMap.itervalues():
  671. index = self.InsertStringItem(sys.maxint, str(i))
  672. self.SetStringItem(index, 0, name)
  673. self.SetStringItem(index, 1, desc)
  674. self.SetItemData(index, i)
  675. if checked[i]:
  676. self.CheckItem(index, True)
  677. i += 1
  678. else:
  679. for name, inloop, desc in self.itemDataMap.itervalues():
  680. index = self.InsertStringItem(sys.maxint, str(i))
  681. self.SetStringItem(index, 0, name)
  682. self.SetStringItem(index, 1, inloop)
  683. self.SetStringItem(index, 2, desc)
  684. self.SetItemData(index, i)
  685. i += 1
  686. def OnRemove(self, event):
  687. """!Remove selected action(s) from the model"""
  688. model = self.frame.GetModel()
  689. canvas = self.frame.GetCanvas()
  690. item = self.GetFirstSelected()
  691. while item != -1:
  692. self.DeleteItem(item)
  693. del self.itemDataMap[item]
  694. action = model.GetItem(item+1) # action indeces starts at 1
  695. if not action:
  696. item = self.GetFirstSelected()
  697. continue
  698. canvas.RemoveShapes([action])
  699. self.frame.ModelChanged()
  700. item = self.GetFirstSelected()
  701. canvas.Refresh()
  702. event.Skip()
  703. def OnEndEdit(self, event):
  704. """!Finish editing of item"""
  705. itemIndex = event.GetIndex()
  706. columnIndex = event.GetColumn()
  707. self.itemDataMap[itemIndex][columnIndex] = event.GetText()
  708. action = self.frame.GetModel().GetItem(itemIndex)
  709. if not action:
  710. event.Veto()
  711. self.frame.ModelChanged()
  712. def OnReload(self, event = None):
  713. """!Reload list of actions"""
  714. self.Populate(self.frame.GetModel().GetItems(objType=ModelAction))
  715. def OnRightUp(self, event):
  716. """!Mouse right button up"""
  717. if self.disablePopup:
  718. return
  719. if not hasattr(self, "popupId"):
  720. self.popupID = dict()
  721. self.popupID['remove'] = wx.NewId()
  722. self.popupID['reload'] = wx.NewId()
  723. self.Bind(wx.EVT_MENU, self.OnRemove, id = self.popupID['remove'])
  724. self.Bind(wx.EVT_MENU, self.OnReload, id = self.popupID['reload'])
  725. # generate popup-menu
  726. menu = wx.Menu()
  727. menu.Append(self.popupID['remove'], _("Delete selected"))
  728. if self.GetFirstSelected() == -1:
  729. menu.Enable(self.popupID['remove'], False)
  730. menu.AppendSeparator()
  731. menu.Append(self.popupID['reload'], _("Reload"))
  732. self.PopupMenu(menu)
  733. menu.Destroy()
  734. def MoveItems(self, items, up):
  735. """!Move items in the list
  736. @param items list of items to move
  737. @param up True to move up otherwise down
  738. """
  739. if len(items) < 1:
  740. return
  741. if items[0] == 0 and up:
  742. del items[0]
  743. if len(items) < 1:
  744. return
  745. if items[-1] == len(self.itemDataMap.keys())-1 and not up:
  746. del items[-1]
  747. if len(items) < 1:
  748. return
  749. model = self.frame.GetModel()
  750. modelActions = model.GetItems(objType=ModelAction)
  751. idxList = dict()
  752. for i in items:
  753. if up:
  754. idx = i-1
  755. else:
  756. idx = i+1
  757. idxList[model.GetItemIndex(modelActions[i])] = model.GetItemIndex(modelActions[idx])
  758. # reorganize model items
  759. model.ReorderItems(idxList)
  760. model.Normalize()
  761. self.Populate(model.GetItems(objType=ModelAction))
  762. # re-selected originaly selected item
  763. for item in idxList.itervalues():
  764. self.SetItemState(item, wx.LIST_STATE_SELECTED, wx.LIST_STATE_SELECTED | wx.LIST_STATE_FOCUSED)
  765. class ItemCheckListCtrl(ItemListCtrl, listmix.CheckListCtrlMixin):
  766. def __init__(self, parent, shape, columns, frame, **kwargs):
  767. self.parent = parent
  768. self.frame = frame
  769. ItemListCtrl.__init__(self, parent, columns, frame,
  770. disablePopup = True, **kwargs)
  771. listmix.CheckListCtrlMixin.__init__(self)
  772. self.SetColumnWidth(0, 100)
  773. self.shape = shape
  774. def OnBeginEdit(self, event):
  775. """!Disable editing"""
  776. event.Veto()
  777. def OnCheckItem(self, index, flag):
  778. """!Item checked/unchecked"""
  779. name = self.GetName()
  780. if name == 'IfBlockList' and self.window:
  781. self.window.OnCheckItemIf(index, flag)
  782. elif name == 'ElseBlockList' and self.window:
  783. self.window.OnCheckItemElse(index, flag)
  784. def GetItems(self):
  785. """!Get list of selected actions"""
  786. ids = { 'checked' : list(),
  787. 'unchecked' : list() }
  788. # action ids start at 1
  789. for i in range(self.GetItemCount()):
  790. if self.IsChecked(i):
  791. ids['checked'].append(i+1)
  792. else:
  793. ids['unchecked'].append(i+1)
  794. return ids
  795. def CheckItemById(self, aId, flag):
  796. """!Check/uncheck given item by id"""
  797. for i in range(self.GetItemCount()):
  798. iId = int(self.GetItem(i, 0).GetText())
  799. if iId == aId:
  800. self.CheckItem(i, flag)
  801. break