mcalc_builder.py 24 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565
  1. """!
  2. @package mcalc_builder.py
  3. @brief Map calculator, wrapper for r.mapcalc
  4. Classes:
  5. - MapCalcFrame
  6. (C) 2008, 2010 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 time
  16. import globalvar
  17. if not os.getenv("GRASS_WXBUNDLED"):
  18. globalvar.CheckForWx()
  19. import wx
  20. import grass.script as grass
  21. import gcmd
  22. import gselect
  23. import menuform
  24. try:
  25. import subprocess
  26. except:
  27. sys.path.append(os.path.join(globalvar.ETCWXDIR, "compat"))
  28. import subprocess
  29. from preferences import globalSettings as UserSettings
  30. class MapCalcFrame(wx.Frame):
  31. """!Mapcalc Frame class. Calculator-style window to create and run
  32. r(3).mapcalc statements
  33. """
  34. def __init__(self, parent, cmd, id = wx.ID_ANY,
  35. style = wx.DEFAULT_FRAME_STYLE | wx.RESIZE_BORDER, **kwargs):
  36. self.parent = parent
  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. #
  53. # variables
  54. #
  55. self.heading = _('mapcalc statement')
  56. self.funct_list = [
  57. 'abs(x)',
  58. 'acos(x)',
  59. 'asin(x)',
  60. 'atan(x)',
  61. 'atan(x,y)',
  62. 'cos(x)',
  63. 'double(x)',
  64. 'eval([x,y,...,]z)',
  65. 'exp(x)',
  66. 'exp(x,y)',
  67. 'float(x)',
  68. 'graph(x,x1,y1[x2,y2..])',
  69. 'if(x)',
  70. 'if(x,a)',
  71. 'if(x,a,b)',
  72. 'if(x,a,b,c)',
  73. 'int(x)',
  74. 'isnull(x)',
  75. 'log(x)',
  76. 'log(x,b)',
  77. 'max(x,y[,z...])',
  78. 'median(x,y[,z...])',
  79. 'min(x,y[,z...])',
  80. 'mode(x,y[,z...])',
  81. 'not(x)',
  82. 'pow(x,y)',
  83. 'rand(a,b)',
  84. 'round(x)',
  85. 'sin(x)',
  86. 'sqrt(x)',
  87. 'tan(x)',
  88. 'xor(x,y)',
  89. 'row()',
  90. 'col()',
  91. 'x()',
  92. 'y()',
  93. 'ewres()',
  94. 'nsres()',
  95. 'null()'
  96. ]
  97. if self.rast3d:
  98. indx = self.funct_list.index('y()') +1
  99. self.funct_list.insert(indx, 'z()')
  100. indx = self.funct_list.index('nsres()') +1
  101. self.funct_list.insert(indx, 'tbres()')
  102. maplabel = _('3D raster map')
  103. element = 'rast3d'
  104. else:
  105. maplabel = _('raster map')
  106. element = 'cell'
  107. self.operatorBox = wx.StaticBox(parent = self.panel, id = wx.ID_ANY,
  108. label=" %s " % _('Operators'))
  109. self.operandBox = wx.StaticBox(parent = self.panel, id = wx.ID_ANY,
  110. label=" %s " % _('Operands'))
  111. self.expressBox = wx.StaticBox(parent = self.panel, id = wx.ID_ANY,
  112. label=" %s " % _('Expression'))
  113. #
  114. # Buttons
  115. #
  116. self.btn_clear = wx.Button(parent = self.panel, id = wx.ID_CLEAR)
  117. self.btn_help = wx.Button(parent = self.panel, id = wx.ID_HELP)
  118. self.btn_run = wx.Button(parent = self.panel, id = wx.ID_ANY, label = _("&Run"))
  119. self.btn_run.SetDefault()
  120. self.btn_close = wx.Button(parent = self.panel, id = wx.ID_CLOSE)
  121. self.btn_cmd = wx.Button(parent = self.panel, id = wx.ID_ANY,
  122. label = _("Command dialog"))
  123. self.btn_cmd.SetToolTipString(_('Open %s dialog') % self.cmd)
  124. self.btn_save = wx.Button(parent = self.panel, id = wx.ID_SAVE)
  125. self.btn_save.SetToolTipString(_('Save expression to file'))
  126. self.btn_load = wx.Button(parent = self.panel, id = wx.ID_ANY,
  127. label = _("&Load"))
  128. self.btn_load.SetToolTipString(_('Load expression from file'))
  129. self.btn = dict()
  130. self.btn['pow'] = wx.Button(parent = self.panel, id = wx.ID_ANY, label = "^")
  131. self.btn['pow'].SetToolTipString(_('exponent'))
  132. self.btn['div'] = wx.Button(parent = self.panel, id = wx.ID_ANY, label = "/")
  133. self.btn['div'].SetToolTipString(_('divide'))
  134. self.btn['add'] = wx.Button(parent = self.panel, id = wx.ID_ANY, label = "+")
  135. self.btn['add'].SetToolTipString(_('add'))
  136. self.btn['minus'] = wx.Button(parent = self.panel, id = wx.ID_ANY, label = "-")
  137. self.btn['minus'].SetToolTipString(_('subtract'))
  138. self.btn['mod'] = wx.Button(parent = self.panel, id = wx.ID_ANY, label = "%")
  139. self.btn['mod'].SetToolTipString(_('modulus'))
  140. self.btn['mult'] = wx.Button(parent = self.panel, id = wx.ID_ANY, label = "*")
  141. self.btn['mult'].SetToolTipString(_('multiply'))
  142. self.btn['paren'] = wx.Button(parent = self.panel, id = wx.ID_ANY, label = "( )")
  143. self.btn['lshift'] = wx.Button(parent = self.panel, id = wx.ID_ANY, label = "<<")
  144. self.btn['lshift'].SetToolTipString(_('left shift'))
  145. self.btn['rshift'] = wx.Button(parent = self.panel, id = wx.ID_ANY, label = ">>")
  146. self.btn['rshift'].SetToolTipString(_('right shift'))
  147. self.btn['rshiftu'] = wx.Button(parent = self.panel, id = wx.ID_ANY, label = ">>>")
  148. self.btn['rshiftu'].SetToolTipString(_('right shift (unsigned)'))
  149. self.btn['gt'] = wx.Button(parent = self.panel, id = wx.ID_ANY, label = ">")
  150. self.btn['gt'].SetToolTipString(_('greater than'))
  151. self.btn['gteq'] = wx.Button(parent = self.panel, id = wx.ID_ANY, label = ">=")
  152. self.btn['gteq'].SetToolTipString(_('greater than or equal to'))
  153. self.btn['lt'] = wx.Button(parent = self.panel, id = wx.ID_ANY, label = "<")
  154. self.btn['lt'].SetToolTipString(_('less than'))
  155. self.btn['lteq'] = wx.Button(parent = self.panel, id = wx.ID_ANY, label = "<=")
  156. self.btn['lteq'].SetToolTipString(_('less than or equal to'))
  157. self.btn['eq'] = wx.Button(parent = self.panel, id = wx.ID_ANY, label = "==")
  158. self.btn['eq'].SetToolTipString(_('equal to'))
  159. self.btn['noteq'] = wx.Button(parent = self.panel, id = wx.ID_ANY, label = "!=")
  160. self.btn['noteq'].SetToolTipString(_('not equal to'))
  161. self.btn['compl'] = wx.Button(parent = self.panel, id = wx.ID_ANY, label = "~")
  162. self.btn['compl'].SetToolTipString(_('one\'s complement'))
  163. self.btn['not'] = wx.Button(parent = self.panel, id = wx.ID_ANY, label = "!")
  164. self.btn['not'].SetToolTipString(_('NOT'))
  165. self.btn['andbit'] = wx.Button(parent = self.panel, id = wx.ID_ANY, label = '&&')
  166. self.btn['andbit'].SetToolTipString(_('bitwise AND'))
  167. self.btn['orbit'] = wx.Button(parent = self.panel, id = wx.ID_ANY, label = "|")
  168. self.btn['orbit'].SetToolTipString(_('bitwise OR'))
  169. self.btn['and'] = wx.Button(parent = self.panel, id = wx.ID_ANY, label = "&&&&")
  170. self.btn['and'].SetToolTipString(_('logical AND'))
  171. self.btn['andnull'] = wx.Button(parent = self.panel, id = wx.ID_ANY, label = "&&&&&&")
  172. self.btn['andnull'].SetToolTipString(_('logical AND (ignores NULLs)'))
  173. self.btn['or'] = wx.Button(parent = self.panel, id = wx.ID_ANY, label = "||")
  174. self.btn['or'].SetToolTipString(_('logical OR'))
  175. self.btn['ornull'] = wx.Button(parent = self.panel, id = wx.ID_ANY, label = "|||")
  176. self.btn['ornull'].SetToolTipString(_('logical OR (ignores NULLs)'))
  177. self.btn['cond'] = wx.Button(parent = self.panel, id = wx.ID_ANY, label = "a ? b : c")
  178. self.btn['cond'].SetToolTipString(_('conditional'))
  179. #
  180. # Text area
  181. #
  182. self.text_mcalc = wx.TextCtrl(parent = self.panel, id = wx.ID_ANY, size = (-1, 75),
  183. style = wx.TE_MULTILINE)
  184. wx.CallAfter(self.text_mcalc.SetFocus)
  185. #
  186. # Map and function insertion text and ComboBoxes
  187. self.newmaplabel = wx.StaticText(parent = self.panel, id = wx.ID_ANY,
  188. label= _('Name for new %s to create') % maplabel)
  189. self.newmaptxt = wx.TextCtrl(parent = self.panel, id = wx.ID_ANY, size=(250, -1))
  190. self.mapsellabel = wx.StaticText(parent = self.panel, id = wx.ID_ANY,
  191. label = _('Insert existing %s') % maplabel)
  192. self.mapselect = gselect.Select(parent = self.panel, id = wx.ID_ANY, size = (250, -1),
  193. type = element, multiple = False)
  194. self.functlabel = wx.StaticText(parent = self.panel, id = wx.ID_ANY,
  195. label = _('Insert mapcalc function'))
  196. self.function = wx.ComboBox(parent = self.panel, id = wx.ID_ANY,
  197. size = (250, -1), choices = self.funct_list,
  198. style = wx.CB_DROPDOWN |
  199. wx.CB_READONLY | wx.TE_PROCESS_ENTER)
  200. self.overwrite = wx.CheckBox(parent = self.panel, id = wx.ID_ANY,
  201. label=_("Allow output files to overwrite existing files"))
  202. self.overwrite.SetValue(UserSettings.Get(group='cmd', key='overwrite', subkey='enabled'))
  203. self.addbox = wx.CheckBox(parent=self.panel,
  204. label=_('Add created raster map into layer tree'), style = wx.NO_BORDER)
  205. self.addbox.SetValue(UserSettings.Get(group='cmd', key='addNewLayer', subkey='enabled'))
  206. if not self.parent or self.parent.GetName() != 'LayerManager':
  207. self.addbox.Hide()
  208. #
  209. # Bindings
  210. #
  211. for btn in self.btn.keys():
  212. self.btn[btn].Bind(wx.EVT_BUTTON, self.AddMark)
  213. self.btn_close.Bind(wx.EVT_BUTTON, self.OnClose)
  214. self.btn_clear.Bind(wx.EVT_BUTTON, self.OnClear)
  215. self.btn_run.Bind(wx.EVT_BUTTON, self.OnMCalcRun)
  216. self.btn_help.Bind(wx.EVT_BUTTON, self.OnHelp)
  217. self.btn_cmd.Bind(wx.EVT_BUTTON, self.OnCmdDialog)
  218. self.btn_save.Bind(wx.EVT_BUTTON, self.OnSaveExpression)
  219. self.btn_load.Bind(wx.EVT_BUTTON, self.OnLoadExpression)
  220. self.mapselect.Bind(wx.EVT_TEXT, self.OnSelect)
  221. self.function.Bind(wx.EVT_COMBOBOX, self.OnSelect)
  222. self.function.Bind(wx.EVT_TEXT_ENTER, self.OnSelect)
  223. self._layout()
  224. self.SetMinSize(self.GetBestSize())
  225. def _layout(self):
  226. sizer = wx.BoxSizer(wx.VERTICAL)
  227. controlSizer = wx.BoxSizer(wx.HORIZONTAL)
  228. operatorSizer = wx.StaticBoxSizer(self.operatorBox, wx.HORIZONTAL)
  229. buttonSizer1 = wx.GridBagSizer(5, 1)
  230. buttonSizer1.Add(item = self.btn['add'], pos = (0,0))
  231. buttonSizer1.Add(item = self.btn['minus'], pos = (0,1))
  232. buttonSizer1.Add(item = self.btn['mod'], pos = (5,0))
  233. buttonSizer1.Add(item = self.btn['mult'], pos = (1,0))
  234. buttonSizer1.Add(item = self.btn['div'], pos = (1,1))
  235. buttonSizer1.Add(item = self.btn['pow'], pos = (5,1))
  236. buttonSizer1.Add(item = self.btn['gt'], pos = (2,0))
  237. buttonSizer1.Add(item = self.btn['gteq'], pos = (2,1))
  238. buttonSizer1.Add(item = self.btn['eq'], pos = (4,0))
  239. buttonSizer1.Add(item = self.btn['lt'], pos = (3,0))
  240. buttonSizer1.Add(item = self.btn['lteq'], pos = (3,1))
  241. buttonSizer1.Add(item = self.btn['noteq'], pos = (4,1))
  242. buttonSizer2 = wx.GridBagSizer(5, 1)
  243. buttonSizer2.Add(item = self.btn['and'], pos = (0,0))
  244. buttonSizer2.Add(item = self.btn['andbit'], pos = (1,0))
  245. buttonSizer2.Add(item = self.btn['andnull'], pos = (2,0))
  246. buttonSizer2.Add(item = self.btn['or'], pos = (0,1))
  247. buttonSizer2.Add(item = self.btn['orbit'], pos = (1,1))
  248. buttonSizer2.Add(item = self.btn['ornull'], pos = (2,1))
  249. buttonSizer2.Add(item = self.btn['lshift'], pos = (3,0))
  250. buttonSizer2.Add(item = self.btn['rshift'], pos = (3,1))
  251. buttonSizer2.Add(item = self.btn['rshiftu'], pos = (4,0))
  252. buttonSizer2.Add(item = self.btn['cond'], pos = (5,0))
  253. buttonSizer2.Add(item = self.btn['compl'], pos = (5,1))
  254. buttonSizer2.Add(item = self.btn['not'], pos = (4,1))
  255. operandSizer = wx.StaticBoxSizer(self.operandBox, wx.HORIZONTAL)
  256. buttonSizer3 = wx.GridBagSizer(7, 1)
  257. buttonSizer3.Add(item = self.newmaplabel, pos = (0, 0),
  258. span = (1, 2), flag = wx.ALIGN_CENTER)
  259. buttonSizer3.Add(item = self.newmaptxt, pos = (1, 0),
  260. span = (1, 2))
  261. buttonSizer3.Add(item = self.mapsellabel, pos = (2, 0),
  262. span = (1, 2), flag = wx.ALIGN_CENTER)
  263. buttonSizer3.Add(item = self.mapselect, pos = (3, 0),
  264. span = (1, 2))
  265. buttonSizer3.Add(item = self.functlabel, pos = (4, 0),
  266. span = (1, 2), flag = wx.ALIGN_CENTER)
  267. buttonSizer3.Add(item = self.function, pos = (5, 0),
  268. span = (1, 2))
  269. buttonSizer3.Add(item = self.btn['paren'], pos = (6, 0),
  270. flag = wx.ALIGN_LEFT)
  271. buttonSizer3.Add(item = self.btn_clear, pos = (6, 1),
  272. flag = wx.ALIGN_RIGHT)
  273. buttonSizer4 = wx.BoxSizer(wx.HORIZONTAL)
  274. buttonSizer4.Add(item = self.btn_cmd,
  275. flag = wx.ALL, border = 5)
  276. buttonSizer4.AddSpacer(10)
  277. buttonSizer4.Add(item = self.btn_load,
  278. flag = wx.ALL, border = 5)
  279. buttonSizer4.Add(item = self.btn_save,
  280. flag = wx.ALL, border = 5)
  281. buttonSizer4.AddSpacer(10)
  282. buttonSizer4.Add(item = self.btn_help,
  283. flag = wx.ALL, border = 5)
  284. buttonSizer4.Add(item = self.btn_run,
  285. flag = wx.ALL, border = 5)
  286. buttonSizer4.Add(item = self.btn_close,
  287. flag = wx.ALL, border = 5)
  288. operatorSizer.Add(item = buttonSizer1, proportion = 0,
  289. flag = wx.ALL | wx.EXPAND, border = 5)
  290. operatorSizer.Add(item = buttonSizer2, proportion = 0,
  291. flag = wx.TOP | wx.BOTTOM | wx.RIGHT | wx.EXPAND, border = 5)
  292. operandSizer.Add(item = buttonSizer3, proportion = 0,
  293. flag = wx.TOP | wx.BOTTOM | wx.RIGHT, border = 5)
  294. controlSizer.Add(item = operatorSizer, proportion = 1,
  295. flag = wx.RIGHT, border = 5)
  296. controlSizer.Add(item = operandSizer, proportion = 0,
  297. flag = wx.EXPAND)
  298. expressSizer = wx.StaticBoxSizer(self.expressBox, wx.HORIZONTAL)
  299. expressSizer.Add(item = self.text_mcalc, proportion = 1,
  300. flag = wx.EXPAND)
  301. sizer.Add(item = controlSizer, proportion = 0,
  302. flag = wx.EXPAND | wx.ALL,
  303. border = 5)
  304. sizer.Add(item = expressSizer, proportion = 1,
  305. flag = wx.EXPAND | wx.LEFT | wx.RIGHT,
  306. border = 5)
  307. sizer.Add(item = self.overwrite, proportion = 0,
  308. flag = wx.EXPAND | wx.LEFT | wx.RIGHT,
  309. border = 5)
  310. if self.addbox.IsShown():
  311. sizer.Add(item = self.addbox, proportion = 0,
  312. flag = wx.EXPAND | wx.LEFT | wx.TOP | wx.RIGHT,
  313. border = 5)
  314. sizer.Add(item = buttonSizer4, proportion = 0,
  315. flag = wx.ALL | wx.ALIGN_RIGHT, border = 1)
  316. self.panel.SetAutoLayout(True)
  317. self.panel.SetSizer(sizer)
  318. sizer.Fit(self.panel)
  319. self.Layout()
  320. def AddMark(self,event):
  321. """!Sends operators to insertion method
  322. """
  323. if event.GetId() == self.btn['compl'].GetId(): mark = "~"
  324. elif event.GetId() == self.btn['not'].GetId(): mark = "!"
  325. elif event.GetId() == self.btn['pow'].GetId(): mark = "^"
  326. elif event.GetId() == self.btn['div'].GetId(): mark = "/"
  327. elif event.GetId() == self.btn['add'].GetId(): mark = "+"
  328. elif event.GetId() == self.btn['minus'].GetId(): mark = "-"
  329. elif event.GetId() == self.btn['mod'].GetId(): mark = "%"
  330. elif event.GetId() == self.btn['mult'].GetId(): mark = "*"
  331. elif event.GetId() == self.btn['lshift'].GetId(): mark = "<<"
  332. elif event.GetId() == self.btn['rshift'].GetId(): mark = ">>"
  333. elif event.GetId() == self.btn['rshiftu'].GetId(): mark = ">>>"
  334. elif event.GetId() == self.btn['gt'].GetId(): mark = ">"
  335. elif event.GetId() == self.btn['gteq'].GetId(): mark = ">="
  336. elif event.GetId() == self.btn['lt'].GetId(): mark = "<"
  337. elif event.GetId() == self.btn['lteq'].GetId(): mark = "<="
  338. elif event.GetId() == self.btn['eq'].GetId(): mark = "=="
  339. elif event.GetId() == self.btn['noteq'].GetId(): mark = "!="
  340. elif event.GetId() == self.btn['andbit'].GetId(): mark = "&"
  341. elif event.GetId() == self.btn['orbit'].GetId(): mark = "|"
  342. elif event.GetId() == self.btn['or'].GetId(): mark = "||"
  343. elif event.GetId() == self.btn['ornull'].GetId(): mark = "|||"
  344. elif event.GetId() == self.btn['and'].GetId(): mark = "&&"
  345. elif event.GetId() == self.btn['andnull'].GetId(): mark = "&&&"
  346. elif event.GetId() == self.btn['cond'].GetId(): mark = " ? : "
  347. elif event.GetId() == self.btn['paren'].GetId(): mark = "( )"
  348. self._addSomething(mark)
  349. def OnSelect(self, event):
  350. """!Gets raster map or function selection and send it to
  351. insertion method
  352. """
  353. item = event.GetString()
  354. self._addSomething(item)
  355. def _addSomething(self, what):
  356. """!Inserts operators, map names, and functions into text area
  357. """
  358. self.text_mcalc.SetFocus()
  359. mcalcstr = self.text_mcalc.GetValue()
  360. position = self.text_mcalc.GetInsertionPoint()
  361. newmcalcstr = mcalcstr[:position]
  362. position_offset = 0
  363. try:
  364. if newmcalcstr[-1] != ' ':
  365. newmcalcstr += ' '
  366. position_offset += 1
  367. except:
  368. pass
  369. newmcalcstr += what
  370. position_offset += len(what)
  371. newmcalcstr += ' ' + mcalcstr[position:]
  372. self.text_mcalc.SetValue(newmcalcstr)
  373. if what == '()':
  374. position_offset -= 1
  375. self.text_mcalc.SetInsertionPoint(position + position_offset)
  376. self.text_mcalc.Update()
  377. def OnMCalcRun(self,event):
  378. """!Builds and runs r.mapcalc statement
  379. """
  380. name = self.newmaptxt.GetValue().strip()
  381. if not name:
  382. gcmd.GError(parent = self,
  383. message = _("You must enter the name of a new map to create"))
  384. return
  385. if not self.text_mcalc.GetValue().strip():
  386. gcmd.GError(parent = self,
  387. message = _("You must enter a mapcalc statement to create a new map"))
  388. return
  389. mctxt = self.text_mcalc.GetValue().strip().replace("\n"," ")
  390. mctxt = mctxt.replace(" " , "")
  391. if self.log:
  392. cmd = [self.cmd, str('expression=%s = %s' % (name, mctxt))]
  393. if self.overwrite.IsChecked():
  394. cmd.append('--overwrite')
  395. self.log.RunCmd(cmd, onDone = self.OnDone)
  396. self.parent.Raise()
  397. else:
  398. if self.overwrite.IsChecked():
  399. overwrite = True
  400. else:
  401. overwrite = False
  402. gcmd.RunCommand(self.cmd,
  403. expression = "%s=%s" % (name, mctxt),
  404. overwrite = overwrite)
  405. def OnDone(self, cmd, returncode):
  406. """!Add create map to the layer tree"""
  407. if not self.addbox.IsChecked():
  408. return
  409. name = self.newmaptxt.GetValue().strip() + '@' + grass.gisenv()['MAPSET']
  410. self.parent.GetLayerTree().AddLayer(ltype = 'raster',
  411. lname = name,
  412. lcmd = ['d.rast', 'map=%s' % name],
  413. multiple = False)
  414. display = self.parent.GetLayerTree().GetMapDisplay()
  415. if display and display.IsAutoRendered():
  416. display.GetWindow().UpdateMap(render = True)
  417. def OnSaveExpression(self, event):
  418. """!Saves expression to file
  419. """
  420. mctxt = self.newmaptxt.GetValue() + ' = ' + self.text_mcalc.GetValue() + os.linesep
  421. #dialog
  422. dlg = wx.FileDialog(parent = self,
  423. message = _("Choose a file name to save the expression"),
  424. wildcard = _("Expression file (*)|*"),
  425. style = wx.SAVE | wx.FD_OVERWRITE_PROMPT)
  426. if dlg.ShowModal() == wx.ID_OK:
  427. path = dlg.GetPath()
  428. if not path:
  429. dlg.Destroy()
  430. return
  431. try:
  432. fobj = open(path, 'w')
  433. fobj.write(mctxt)
  434. finally:
  435. fobj.close()
  436. dlg.Destroy()
  437. def OnLoadExpression(self, event):
  438. """!Load expression from file
  439. """
  440. dlg = wx.FileDialog(parent = self,
  441. message = _("Choose a file name to load the expression"),
  442. wildcard = _("Expression file (*)|*"),
  443. style = wx.OPEN)
  444. if dlg.ShowModal() == wx.ID_OK:
  445. path = dlg.GetPath()
  446. if not path:
  447. dlg.Destroy()
  448. return
  449. try:
  450. fobj = open(path,'r')
  451. mctxt = fobj.read()
  452. finally:
  453. fobj.close()
  454. try:
  455. result, exp = mctxt.split('=', 1)
  456. except ValueError:
  457. result = ''
  458. exp = mctxt
  459. self.newmaptxt.SetValue(result.strip())
  460. self.text_mcalc.SetValue(exp.strip())
  461. self.text_mcalc.SetFocus()
  462. self.text_mcalc.SetInsertionPointEnd()
  463. dlg.Destroy()
  464. def OnClear(self, event):
  465. """!Clears text area
  466. """
  467. self.text_mcalc.SetValue('')
  468. def OnHelp(self, event):
  469. """!Launches r.mapcalc help
  470. """
  471. gcmd.RunCommand('g.manual', parent = self, entry = self.cmd)
  472. def OnClose(self,event):
  473. """!Close window"""
  474. self.Destroy()
  475. def OnCmdDialog(self, event):
  476. """!Shows command dialog"""
  477. name = self.newmaptxt.GetValue().strip()
  478. mctxt = self.text_mcalc.GetValue().strip().replace("\n"," ")
  479. mctxt = mctxt.replace(" " , "")
  480. expr = name
  481. if expr:
  482. expr += '='
  483. expr += mctxt
  484. menuform.GUI().ParseCommand(cmd = [self.cmd, 'expression=' + expr],
  485. parentframe = self)
  486. if __name__ == "__main__":
  487. app = wx.App(0)
  488. frame = MapCalcFrame(None, cmd = 'r.mapcalc')
  489. frame.Show()
  490. app.MainLoop()