mcalc_builder.py 27 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631
  1. """!
  2. @package modules::mcalc_builder
  3. @brief Map calculator, GUI wrapper for r.mapcalc
  4. Classes:
  5. - mcalc_builder::MapCalcFrame
  6. (C) 2008, 2011-2013 by the GRASS Development Team
  7. This program is free software under the GNU General Public License
  8. (>=v2). Read the file COPYING that comes with GRASS for details.
  9. @author Michael Barton, Arizona State University
  10. @author Martin Landa <landa.martin gmail.com>
  11. @author Tim Michelsen (load/save expression)
  12. """
  13. import os
  14. import sys
  15. import re
  16. import wx
  17. import grass.script as grass
  18. if __name__ == "__main__":
  19. gui_wx_path = os.path.join(os.getenv('GISBASE'), 'etc', 'gui', 'wxpython')
  20. if gui_wx_path not in sys.path:
  21. sys.path.append(gui_wx_path)
  22. from core import globalvar
  23. from core.gcmd import GError, RunCommand
  24. from core.giface import StandaloneGrassInterface
  25. from core.utils import _
  26. from gui_core.gselect import Select
  27. from gui_core.forms import GUI
  28. from core.settings import UserSettings
  29. class MapCalcFrame(wx.Frame):
  30. """!Mapcalc Frame class. Calculator-style window to create and run
  31. r(3).mapcalc statements.
  32. """
  33. def __init__(self, parent, giface, cmd, id = wx.ID_ANY,
  34. style = wx.DEFAULT_FRAME_STYLE | wx.RESIZE_BORDER, **kwargs):
  35. self.parent = parent
  36. self._giface = giface
  37. if self.parent:
  38. self.log = self.parent.GetLogWindow()
  39. else:
  40. self.log = None
  41. # grass command
  42. self.cmd = cmd
  43. if self.cmd == 'r.mapcalc':
  44. self.rast3d = False
  45. title = _('GRASS GIS Raster Map Calculator')
  46. if self.cmd == 'r3.mapcalc':
  47. self.rast3d = True
  48. title = _('GRASS GIS 3D Raster Map Calculator')
  49. wx.Frame.__init__(self, parent, id = id, title = title, **kwargs)
  50. self.SetIcon(wx.Icon(os.path.join(globalvar.ETCICONDIR, 'grass.ico'), wx.BITMAP_TYPE_ICO))
  51. self.panel = wx.Panel(parent = self, id = wx.ID_ANY)
  52. self.CreateStatusBar()
  53. #
  54. # variables
  55. #
  56. self.heading = _('mapcalc statement')
  57. self.funct_dict = {
  58. 'abs(x)':'abs()',
  59. 'acos(x)':'acos()',
  60. 'asin(x)':'asin()',
  61. 'atan(x)':'atan()',
  62. 'atan(x,y)':'atan( , )',
  63. 'cos(x)':'cos()',
  64. 'double(x)':'double()',
  65. 'eval([x,y,...,]z)':'eval()',
  66. 'exp(x)':'exp()',
  67. 'exp(x,y)':'exp( , )',
  68. 'float(x)':'float()',
  69. 'graph(x,x1,y1[x2,y2..])':'graph( , , )',
  70. 'if(x)':'if()',
  71. 'if(x,a)':'if( , )',
  72. 'if(x,a,b)':'if( , , )',
  73. 'if(x,a,b,c)':'if( , , , )',
  74. 'int(x)':'int()',
  75. 'isnull(x)':'isnull()',
  76. 'log(x)':'log(',
  77. 'log(x,b)':'log( , )',
  78. 'max(x,y[,z...])':'max( , )',
  79. 'median(x,y[,z...])':'median( , )',
  80. 'min(x,y[,z...])':'min( , )',
  81. 'mode(x,y[,z...])':'mode( , )',
  82. 'not(x)':'not()',
  83. 'pow(x,y)':'pow( , )',
  84. 'rand(a,b)':'rand( , )',
  85. 'round(x)':'round()',
  86. 'sin(x)':'sin()',
  87. 'sqrt(x)':'sqrt()',
  88. 'tan(x)':'tan()',
  89. 'xor(x,y)':'xor( , )',
  90. 'row()':'row()',
  91. 'col()':'col()',
  92. 'x()':'x()',
  93. 'y()':'y()',
  94. 'ewres()':'ewres()',
  95. 'nsres()':'nsres()',
  96. 'null()':'null()'
  97. }
  98. if self.rast3d:
  99. self.funct_dict['z()'] = 'z()'
  100. self.funct_dict['tbres()'] = 'tbres()'
  101. element = 'rast3d'
  102. else:
  103. element = 'cell'
  104. # characters which can be in raster map name but the map name must be then quoted
  105. self.charactersToQuote = '+-&!<>%~?^|'
  106. # stores last typed map name in Select widget to distinguish typing from selection
  107. self.lastMapName = ''
  108. self.operatorBox = wx.StaticBox(parent = self.panel, id = wx.ID_ANY,
  109. label=" %s " % _('Operators'))
  110. self.outputBox = wx.StaticBox(parent = self.panel, id = wx.ID_ANY,
  111. label=" %s " % _('Output'))
  112. self.operandBox = wx.StaticBox(parent = self.panel, id = wx.ID_ANY,
  113. label=" %s " % _('Operands'))
  114. self.expressBox = wx.StaticBox(parent = self.panel, id = wx.ID_ANY,
  115. label=" %s " % _('Expression'))
  116. #
  117. # Buttons
  118. #
  119. self.btn_clear = wx.Button(parent = self.panel, id = wx.ID_CLEAR)
  120. self.btn_help = wx.Button(parent = self.panel, id = wx.ID_HELP)
  121. self.btn_run = wx.Button(parent = self.panel, id = wx.ID_ANY, label = _("&Run"))
  122. self.btn_run.SetForegroundColour(wx.Colour(35, 142, 35))
  123. self.btn_run.SetDefault()
  124. self.btn_close = wx.Button(parent = self.panel, id = wx.ID_CLOSE)
  125. self.btn_save = wx.Button(parent = self.panel, id = wx.ID_SAVE)
  126. self.btn_save.SetToolTipString(_('Save expression to file'))
  127. self.btn_load = wx.Button(parent = self.panel, id = wx.ID_ANY,
  128. label = _("&Load"))
  129. self.btn_load.SetToolTipString(_('Load expression from file'))
  130. self.btn_copy = wx.Button(parent=self.panel, id=wx.ID_COPY)
  131. self.btn_copy.SetToolTipString(_("Copy the current command string to the clipboard"))
  132. self.btn = dict()
  133. self.btn['pow'] = wx.Button(parent = self.panel, id = wx.ID_ANY, label = "^")
  134. self.btn['pow'].SetToolTipString(_('exponent'))
  135. self.btn['div'] = wx.Button(parent = self.panel, id = wx.ID_ANY, label = "/")
  136. self.btn['div'].SetToolTipString(_('divide'))
  137. self.btn['add'] = wx.Button(parent = self.panel, id = wx.ID_ANY, label = "+")
  138. self.btn['add'].SetToolTipString(_('add'))
  139. self.btn['minus'] = wx.Button(parent = self.panel, id = wx.ID_ANY, label = "-")
  140. self.btn['minus'].SetToolTipString(_('subtract'))
  141. self.btn['mod'] = wx.Button(parent = self.panel, id = wx.ID_ANY, label = "%")
  142. self.btn['mod'].SetToolTipString(_('modulus'))
  143. self.btn['mult'] = wx.Button(parent = self.panel, id = wx.ID_ANY, label = "*")
  144. self.btn['mult'].SetToolTipString(_('multiply'))
  145. self.btn['parenl'] = wx.Button(parent = self.panel, id = wx.ID_ANY, label = "(")
  146. self.btn['parenr'] = wx.Button(parent = self.panel, id = wx.ID_ANY, label = ")")
  147. self.btn['lshift'] = wx.Button(parent = self.panel, id = wx.ID_ANY, label = "<<")
  148. self.btn['lshift'].SetToolTipString(_('left shift'))
  149. self.btn['rshift'] = wx.Button(parent = self.panel, id = wx.ID_ANY, label = ">>")
  150. self.btn['rshift'].SetToolTipString(_('right shift'))
  151. self.btn['rshiftu'] = wx.Button(parent = self.panel, id = wx.ID_ANY, label = ">>>")
  152. self.btn['rshiftu'].SetToolTipString(_('right shift (unsigned)'))
  153. self.btn['gt'] = wx.Button(parent = self.panel, id = wx.ID_ANY, label = ">")
  154. self.btn['gt'].SetToolTipString(_('greater than'))
  155. self.btn['gteq'] = wx.Button(parent = self.panel, id = wx.ID_ANY, label = ">=")
  156. self.btn['gteq'].SetToolTipString(_('greater than or equal to'))
  157. self.btn['lt'] = wx.Button(parent = self.panel, id = wx.ID_ANY, label = "<")
  158. self.btn['lt'].SetToolTipString(_('less than'))
  159. self.btn['lteq'] = wx.Button(parent = self.panel, id = wx.ID_ANY, label = "<=")
  160. self.btn['lteq'].SetToolTipString(_('less than or equal to'))
  161. self.btn['eq'] = wx.Button(parent = self.panel, id = wx.ID_ANY, label = "==")
  162. self.btn['eq'].SetToolTipString(_('equal to'))
  163. self.btn['noteq'] = wx.Button(parent = self.panel, id = wx.ID_ANY, label = "!=")
  164. self.btn['noteq'].SetToolTipString(_('not equal to'))
  165. self.btn['compl'] = wx.Button(parent = self.panel, id = wx.ID_ANY, label = "~")
  166. self.btn['compl'].SetToolTipString(_('one\'s complement'))
  167. self.btn['not'] = wx.Button(parent = self.panel, id = wx.ID_ANY, label = "!")
  168. self.btn['not'].SetToolTipString(_('NOT'))
  169. self.btn['andbit'] = wx.Button(parent = self.panel, id = wx.ID_ANY, label = '&&')
  170. self.btn['andbit'].SetToolTipString(_('bitwise AND'))
  171. self.btn['orbit'] = wx.Button(parent = self.panel, id = wx.ID_ANY, label = "|")
  172. self.btn['orbit'].SetToolTipString(_('bitwise OR'))
  173. self.btn['and'] = wx.Button(parent = self.panel, id = wx.ID_ANY, label = "&&&&")
  174. self.btn['and'].SetToolTipString(_('logical AND'))
  175. self.btn['andnull'] = wx.Button(parent = self.panel, id = wx.ID_ANY, label = "&&&&&&")
  176. self.btn['andnull'].SetToolTipString(_('logical AND (ignores NULLs)'))
  177. self.btn['or'] = wx.Button(parent = self.panel, id = wx.ID_ANY, label = "||")
  178. self.btn['or'].SetToolTipString(_('logical OR'))
  179. self.btn['ornull'] = wx.Button(parent = self.panel, id = wx.ID_ANY, label = "|||")
  180. self.btn['ornull'].SetToolTipString(_('logical OR (ignores NULLs)'))
  181. self.btn['cond'] = wx.Button(parent = self.panel, id = wx.ID_ANY, label = "a ? b : c")
  182. self.btn['cond'].SetToolTipString(_('conditional'))
  183. #
  184. # Text area
  185. #
  186. self.text_mcalc = wx.TextCtrl(parent = self.panel, id = wx.ID_ANY, size = (-1, 75),
  187. style = wx.TE_MULTILINE)
  188. wx.CallAfter(self.text_mcalc.SetFocus)
  189. #
  190. # Map and function insertion text and ComboBoxes
  191. self.newmaplabel = wx.StaticText(parent = self.panel, id = wx.ID_ANY)
  192. if self.rast3d:
  193. self.newmaplabel.SetLabel(_('Name for new 3D raster map to create'))
  194. else:
  195. self.newmaplabel.SetLabel(_('Name for new raster map to create'))
  196. self.newmaptxt = wx.TextCtrl(parent = self.panel, id = wx.ID_ANY, size=(250, -1))
  197. self.mapsellabel = wx.StaticText(parent = self.panel, id = wx.ID_ANY)
  198. if self.rast3d:
  199. self.mapsellabel.SetLabel(_('Insert existing 3D raster map'))
  200. else:
  201. self.mapsellabel.SetLabel(_('Insert existing raster map'))
  202. self.mapselect = Select(parent = self.panel, id = wx.ID_ANY, size = (250, -1),
  203. type = element, multiple = False)
  204. self.functlabel = wx.StaticText(parent = self.panel, id = wx.ID_ANY,
  205. label = _('Insert mapcalc function'))
  206. self.function = wx.ComboBox(parent = self.panel, id = wx.ID_ANY,
  207. size = (250, -1), choices = sorted(self.funct_dict.keys()),
  208. style = wx.CB_DROPDOWN |
  209. wx.CB_READONLY | wx.TE_PROCESS_ENTER)
  210. self.overwrite = wx.CheckBox(parent = self.panel, id = wx.ID_ANY,
  211. label=_("Allow output files to overwrite existing files"))
  212. self.overwrite.SetValue(UserSettings.Get(group='cmd', key='overwrite', subkey='enabled'))
  213. self.addbox = wx.CheckBox(parent=self.panel,
  214. label=_('Add created raster map into layer tree'), style = wx.NO_BORDER)
  215. self.addbox.SetValue(UserSettings.Get(group='cmd', key='addNewLayer', subkey='enabled'))
  216. if not self.parent or self.parent.GetName() != 'LayerManager':
  217. self.addbox.Hide()
  218. #
  219. # Bindings
  220. #
  221. for btn in self.btn.keys():
  222. self.btn[btn].Bind(wx.EVT_BUTTON, self.AddMark)
  223. self.btn_close.Bind(wx.EVT_BUTTON, self.OnClose)
  224. self.btn_clear.Bind(wx.EVT_BUTTON, self.OnClear)
  225. self.btn_run.Bind(wx.EVT_BUTTON, self.OnMCalcRun)
  226. self.btn_help.Bind(wx.EVT_BUTTON, self.OnHelp)
  227. self.btn_save.Bind(wx.EVT_BUTTON, self.OnSaveExpression)
  228. self.btn_load.Bind(wx.EVT_BUTTON, self.OnLoadExpression)
  229. self.btn_copy.Bind(wx.EVT_BUTTON, self.OnCopy)
  230. self.mapselect.Bind(wx.EVT_TEXT, self.OnSelectTextEvt)
  231. self.function.Bind(wx.EVT_COMBOBOX, self._return_funct)
  232. self.function.Bind(wx.EVT_TEXT_ENTER, self.OnSelect)
  233. self.newmaptxt.Bind(wx.EVT_TEXT, self.OnUpdateStatusBar)
  234. self.text_mcalc.Bind(wx.EVT_TEXT, self.OnUpdateStatusBar)
  235. self.overwrite.Bind(wx.EVT_CHECKBOX, self.OnUpdateStatusBar)
  236. self._layout()
  237. self.SetMinSize(self.GetBestSize())
  238. def _return_funct(self,event):
  239. i = event.GetString()
  240. self._addSomething(self.funct_dict[i])
  241. def _layout(self):
  242. sizer = wx.BoxSizer(wx.VERTICAL)
  243. controlSizer = wx.BoxSizer(wx.HORIZONTAL)
  244. operatorSizer = wx.StaticBoxSizer(self.operatorBox, wx.HORIZONTAL)
  245. outOpeSizer = wx.BoxSizer(wx.VERTICAL)
  246. buttonSizer1 = wx.GridBagSizer(5, 1)
  247. buttonSizer1.Add(item = self.btn['add'], pos = (0,0))
  248. buttonSizer1.Add(item = self.btn['minus'], pos = (0,1))
  249. buttonSizer1.Add(item = self.btn['mod'], pos = (5,0))
  250. buttonSizer1.Add(item = self.btn['mult'], pos = (1,0))
  251. buttonSizer1.Add(item = self.btn['div'], pos = (1,1))
  252. buttonSizer1.Add(item = self.btn['pow'], pos = (5,1))
  253. buttonSizer1.Add(item = self.btn['gt'], pos = (2,0))
  254. buttonSizer1.Add(item = self.btn['gteq'], pos = (2,1))
  255. buttonSizer1.Add(item = self.btn['eq'], pos = (4,0))
  256. buttonSizer1.Add(item = self.btn['lt'], pos = (3,0))
  257. buttonSizer1.Add(item = self.btn['lteq'], pos = (3,1))
  258. buttonSizer1.Add(item = self.btn['noteq'], pos = (4,1))
  259. buttonSizer2 = wx.GridBagSizer(5, 1)
  260. buttonSizer2.Add(item = self.btn['and'], pos = (0,0))
  261. buttonSizer2.Add(item = self.btn['andbit'], pos = (1,0))
  262. buttonSizer2.Add(item = self.btn['andnull'], pos = (2,0))
  263. buttonSizer2.Add(item = self.btn['or'], pos = (0,1))
  264. buttonSizer2.Add(item = self.btn['orbit'], pos = (1,1))
  265. buttonSizer2.Add(item = self.btn['ornull'], pos = (2,1))
  266. buttonSizer2.Add(item = self.btn['lshift'], pos = (3,0))
  267. buttonSizer2.Add(item = self.btn['rshift'], pos = (3,1))
  268. buttonSizer2.Add(item = self.btn['rshiftu'], pos = (4,0))
  269. buttonSizer2.Add(item = self.btn['cond'], pos = (5,0))
  270. buttonSizer2.Add(item = self.btn['compl'], pos = (5,1))
  271. buttonSizer2.Add(item = self.btn['not'], pos = (4,1))
  272. outputSizer = wx.StaticBoxSizer(self.outputBox, wx.VERTICAL)
  273. outputSizer.Add(item = self.newmaplabel,
  274. flag = wx.ALIGN_CENTER | wx.BOTTOM | wx.TOP, border = 5)
  275. outputSizer.Add(item = self.newmaptxt,
  276. flag = wx.EXPAND)
  277. operandSizer = wx.StaticBoxSizer(self.operandBox, wx.HORIZONTAL)
  278. buttonSizer3 = wx.GridBagSizer(7, 1)
  279. buttonSizer3.Add(item = self.functlabel, pos = (0,0),
  280. span = (1,2), flag = wx.ALIGN_CENTER)
  281. buttonSizer3.Add(item = self.function, pos = (1,0),
  282. span = (1,2))
  283. buttonSizer3.Add(item = self.mapsellabel, pos = (2,0),
  284. span = (1,2), flag = wx.ALIGN_CENTER)
  285. buttonSizer3.Add(item = self.mapselect, pos = (3,0),
  286. span = (1,2))
  287. threebutton = wx.GridBagSizer(1, 2)
  288. threebutton.Add(item = self.btn['parenl'], pos = (0,0),
  289. span = (1,1), flag = wx.ALIGN_LEFT)
  290. threebutton.Add(item = self.btn['parenr'], pos = (0,1),
  291. span = (1,1), flag = wx.ALIGN_CENTER)
  292. threebutton.Add(item = self.btn_clear, pos = (0,2),
  293. span = (1,1), flag = wx.ALIGN_RIGHT)
  294. buttonSizer3.Add(item = threebutton, pos = (4,0),
  295. span = (1,1), flag = wx.ALIGN_CENTER)
  296. buttonSizer4 = wx.BoxSizer(wx.HORIZONTAL)
  297. buttonSizer4.Add(item = self.btn_load,
  298. flag = wx.ALL, border = 5)
  299. buttonSizer4.Add(item = self.btn_save,
  300. flag = wx.ALL, border = 5)
  301. buttonSizer4.Add(item = self.btn_copy,
  302. flag = wx.ALL, border = 5)
  303. buttonSizer4.AddSpacer(30)
  304. buttonSizer4.Add(item = self.btn_help,
  305. flag = wx.ALL, border = 5)
  306. buttonSizer4.Add(item = self.btn_run,
  307. flag = wx.ALL, border = 5)
  308. buttonSizer4.Add(item = self.btn_close,
  309. flag = wx.ALL, border = 5)
  310. operatorSizer.Add(item = buttonSizer1, proportion = 0,
  311. flag = wx.ALL | wx.EXPAND, border = 5)
  312. operatorSizer.Add(item = buttonSizer2, proportion = 0,
  313. flag = wx.TOP | wx.BOTTOM | wx.RIGHT | wx.EXPAND, border = 5)
  314. operandSizer.Add(item = buttonSizer3, proportion = 0,
  315. flag = wx.TOP | wx.BOTTOM | wx.RIGHT, border = 5)
  316. controlSizer.Add(item = operatorSizer, proportion = 1,
  317. flag = wx.RIGHT | wx.EXPAND, border = 5)
  318. outOpeSizer.Add(item = outputSizer, proportion = 0,
  319. flag = wx.EXPAND)
  320. outOpeSizer.Add(item = operandSizer, proportion = 1,
  321. flag = wx.EXPAND | wx.TOP, border = 5)
  322. controlSizer.Add(item = outOpeSizer, proportion = 0,
  323. flag = wx.EXPAND)
  324. expressSizer = wx.StaticBoxSizer(self.expressBox, wx.HORIZONTAL)
  325. expressSizer.Add(item = self.text_mcalc, proportion = 1,
  326. flag = wx.EXPAND)
  327. sizer.Add(item = controlSizer, proportion = 0,
  328. flag = wx.EXPAND | wx.ALL,
  329. border = 5)
  330. sizer.Add(item = expressSizer, proportion = 1,
  331. flag = wx.EXPAND | wx.LEFT | wx.RIGHT,
  332. border = 5)
  333. sizer.Add(item = buttonSizer4, proportion = 0,
  334. flag = wx.ALIGN_RIGHT | wx.ALL, border = 3)
  335. sizer.Add(item = self.overwrite, proportion = 0,
  336. flag = wx.LEFT | wx.RIGHT,
  337. border = 5)
  338. if self.addbox.IsShown():
  339. sizer.Add(item = self.addbox, proportion = 0,
  340. flag = wx.LEFT | wx.RIGHT,
  341. border = 5)
  342. self.panel.SetAutoLayout(True)
  343. self.panel.SetSizer(sizer)
  344. sizer.Fit(self.panel)
  345. self.Layout()
  346. def AddMark(self,event):
  347. """!Sends operators to insertion method
  348. """
  349. if event.GetId() == self.btn['compl'].GetId(): mark = "~"
  350. elif event.GetId() == self.btn['not'].GetId(): mark = "!"
  351. elif event.GetId() == self.btn['pow'].GetId(): mark = "^"
  352. elif event.GetId() == self.btn['div'].GetId(): mark = "/"
  353. elif event.GetId() == self.btn['add'].GetId(): mark = "+"
  354. elif event.GetId() == self.btn['minus'].GetId(): mark = "-"
  355. elif event.GetId() == self.btn['mod'].GetId(): mark = "%"
  356. elif event.GetId() == self.btn['mult'].GetId(): mark = "*"
  357. elif event.GetId() == self.btn['lshift'].GetId(): mark = "<<"
  358. elif event.GetId() == self.btn['rshift'].GetId(): mark = ">>"
  359. elif event.GetId() == self.btn['rshiftu'].GetId(): mark = ">>>"
  360. elif event.GetId() == self.btn['gt'].GetId(): mark = ">"
  361. elif event.GetId() == self.btn['gteq'].GetId(): mark = ">="
  362. elif event.GetId() == self.btn['lt'].GetId(): mark = "<"
  363. elif event.GetId() == self.btn['lteq'].GetId(): mark = "<="
  364. elif event.GetId() == self.btn['eq'].GetId(): mark = "=="
  365. elif event.GetId() == self.btn['noteq'].GetId(): mark = "!="
  366. elif event.GetId() == self.btn['andbit'].GetId(): mark = "&"
  367. elif event.GetId() == self.btn['orbit'].GetId(): mark = "|"
  368. elif event.GetId() == self.btn['or'].GetId(): mark = "||"
  369. elif event.GetId() == self.btn['ornull'].GetId(): mark = "|||"
  370. elif event.GetId() == self.btn['and'].GetId(): mark = "&&"
  371. elif event.GetId() == self.btn['andnull'].GetId(): mark = "&&&"
  372. elif event.GetId() == self.btn['cond'].GetId(): mark = " ? : "
  373. elif event.GetId() == self.btn['parenl'].GetId(): mark = "("
  374. elif event.GetId() == self.btn['parenr'].GetId(): mark = ")"
  375. self._addSomething(mark)
  376. def OnSelectTextEvt(self, event):
  377. """!Checks if user is typing or the event was emited by map selection.
  378. Prevents from changing focus.
  379. """
  380. item = self.mapselect.GetValue().strip()
  381. if not (abs(len(item) - len(self.lastMapName)) == 1 and \
  382. self.lastMapName in item or item in self.lastMapName):
  383. self.OnSelect(event)
  384. self.lastMapName = item
  385. def OnSelect(self, event):
  386. """!Gets raster map or function selection and send it to
  387. insertion method.
  388. Checks for characters which can be in raster map name but
  389. the raster map name must be then quoted.
  390. """
  391. item = self.FindWindowById(event.GetId()).GetValue().strip()
  392. if any((char in item) for char in self.charactersToQuote):
  393. item = '"' + item + '"'
  394. self._addSomething(item)
  395. def OnUpdateStatusBar(self, event):
  396. """!Update statusbar text"""
  397. command = self._getCommand()
  398. self.SetStatusText(command)
  399. event.Skip()
  400. def _getCommand(self):
  401. """!Returns entire command as string."""
  402. expr = self.text_mcalc.GetValue().strip().replace("\n", " ")
  403. cmd = 'r.mapcalc'
  404. if self.rast3d:
  405. cmd = 'r3.mapcalc'
  406. overwrite = ''
  407. if self.overwrite.IsChecked():
  408. overwrite = ' --overwrite'
  409. return '{cmd} "{new} = {expr}"{overwrite}'.format(cmd=cmd, expr=expr,
  410. new=self.newmaptxt.GetValue(),
  411. overwrite=overwrite)
  412. def _addSomething(self, what):
  413. """!Inserts operators, map names, and functions into text area
  414. """
  415. mcalcstr = self.text_mcalc.GetValue()
  416. position = self.text_mcalc.GetInsertionPoint()
  417. newmcalcstr = mcalcstr[:position]
  418. position_offset = 0
  419. try:
  420. if newmcalcstr[-1] != ' ':
  421. newmcalcstr += ' '
  422. position_offset += 1
  423. except:
  424. pass
  425. newmcalcstr += what + ' ' + mcalcstr[position:]
  426. self.text_mcalc.SetValue(newmcalcstr)
  427. if len(what) > 0:
  428. match = re.search(pattern="\(.*\)", string=what)
  429. if match:
  430. position_offset += match.start() + 1
  431. else:
  432. position_offset += len(what)
  433. self.text_mcalc.SetInsertionPoint(position + position_offset)
  434. self.text_mcalc.Update()
  435. self.text_mcalc.SetFocus()
  436. def OnMCalcRun(self,event):
  437. """!Builds and runs r.mapcalc statement
  438. """
  439. name = self.newmaptxt.GetValue().strip()
  440. if not name:
  441. GError(parent = self,
  442. message = _("You must enter the name of "
  443. "a new raster map to create."))
  444. return
  445. if not (name[0] == '"' and name[-1] == '"') and any((char in name) for char in self.charactersToQuote):
  446. name = '"' + name + '"'
  447. expr = self.text_mcalc.GetValue().strip().replace("\n", " ")
  448. if not expr:
  449. GError(parent = self,
  450. message = _("You must enter an expression "
  451. "to create a new raster map."))
  452. return
  453. if self.log:
  454. cmd = [self.cmd, str('expression=%s = %s' % (name, expr))]
  455. if self.overwrite.IsChecked():
  456. cmd.append('--overwrite')
  457. self.log.RunCmd(cmd, onDone = self.OnDone)
  458. self.parent.Raise()
  459. else:
  460. if self.overwrite.IsChecked():
  461. overwrite = True
  462. else:
  463. overwrite = False
  464. RunCommand(self.cmd,
  465. expression = "%s=%s" % (name, expr),
  466. overwrite = overwrite)
  467. def OnDone(self, cmd, returncode):
  468. """!Add create map to the layer tree
  469. Sends the mapCreated signal from the grass interface.
  470. """
  471. if returncode != 0:
  472. return
  473. name = self.newmaptxt.GetValue().strip(' "') + '@' + grass.gisenv()['MAPSET']
  474. ltype = 'rast'
  475. if self.rast3d:
  476. ltype = 'rast3d'
  477. self._giface.mapCreated.emit(name=name, ltype=ltype, add=self.addbox.IsChecked())
  478. def OnSaveExpression(self, event):
  479. """!Saves expression to file
  480. """
  481. mctxt = self.newmaptxt.GetValue() + ' = ' + self.text_mcalc.GetValue() + os.linesep
  482. #dialog
  483. dlg = wx.FileDialog(parent = self,
  484. message = _("Choose a file name to save the expression"),
  485. wildcard = _("Expression file (*)|*"),
  486. style = wx.SAVE | wx.FD_OVERWRITE_PROMPT)
  487. if dlg.ShowModal() == wx.ID_OK:
  488. path = dlg.GetPath()
  489. if not path:
  490. dlg.Destroy()
  491. return
  492. try:
  493. fobj = open(path, 'w')
  494. fobj.write(mctxt)
  495. finally:
  496. fobj.close()
  497. dlg.Destroy()
  498. def OnLoadExpression(self, event):
  499. """!Load expression from file
  500. """
  501. dlg = wx.FileDialog(parent = self,
  502. message = _("Choose a file name to load the expression"),
  503. wildcard = _("Expression file (*)|*"),
  504. style = wx.OPEN)
  505. if dlg.ShowModal() == wx.ID_OK:
  506. path = dlg.GetPath()
  507. if not path:
  508. dlg.Destroy()
  509. return
  510. try:
  511. fobj = open(path,'r')
  512. mctxt = fobj.read()
  513. finally:
  514. fobj.close()
  515. try:
  516. result, exp = mctxt.split('=', 1)
  517. except ValueError:
  518. result = ''
  519. exp = mctxt
  520. self.newmaptxt.SetValue(result.strip())
  521. self.text_mcalc.SetValue(exp.strip())
  522. self.text_mcalc.SetFocus()
  523. self.text_mcalc.SetInsertionPointEnd()
  524. dlg.Destroy()
  525. def OnCopy(self, event):
  526. command = self._getCommand()
  527. cmddata = wx.TextDataObject()
  528. cmddata.SetText(command)
  529. if wx.TheClipboard.Open():
  530. wx.TheClipboard.SetData(cmddata)
  531. wx.TheClipboard.Close()
  532. self.SetStatusText(_("'{cmd}' copied to clipboard").format(cmd=command))
  533. def OnClear(self, event):
  534. """!Clears text area
  535. """
  536. self.text_mcalc.SetValue('')
  537. def OnHelp(self, event):
  538. """!Launches r.mapcalc help
  539. """
  540. RunCommand('g.manual', parent = self, entry = self.cmd)
  541. def OnClose(self,event):
  542. """!Close window"""
  543. self.Destroy()
  544. if __name__ == "__main__":
  545. app = wx.App(0)
  546. frame = MapCalcFrame(parent = None, cmd = 'r.mapcalc', giface = StandaloneGrassInterface())
  547. frame.Show()
  548. app.MainLoop()