1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207 |
- """!
- @package gui_core.goutput
- @brief Command output widgets
- Classes:
- - goutput::CmdThread
- - goutput::GConsole
- - goutput::GStdout
- - goutput::GStderr
- - goutput::GStc
- (C) 2007-2012 by the GRASS Development Team
- This program is free software under the GNU General Public License
- (>=v2). Read the file COPYING that comes with GRASS for details.
- @author Michael Barton (Arizona State University)
- @author Martin Landa <landa.martin gmail.com>
- @author Vaclav Petras <wenzeslaus gmail.com> (copy&paste customization)
- """
- import os
- import sys
- import re
- import textwrap
- import time
- import threading
- import Queue
- import codecs
- import locale
- import wx
- from wx import stc
- from wx.lib.newevent import NewEvent
- import grass.script as grass
- from grass.script import task as gtask
- from core import globalvar
- from core import utils
- from core.gcmd import CommandThread, GMessage, GError, GException, EncodeString
- from core.events import gShowNotification, gMapCreated
- from gui_core.forms import GUI
- from gui_core.prompt import GPromptSTC
- from core.debug import Debug
- from core.settings import UserSettings
- from gui_core.widgets import SearchModuleWidget, EVT_MODULE_SELECTED
- from core.modulesdata import ModulesData
- wxCmdOutput, EVT_CMD_OUTPUT = NewEvent()
- wxCmdProgress, EVT_CMD_PROGRESS = NewEvent()
- wxCmdRun, EVT_CMD_RUN = NewEvent()
- wxCmdDone, EVT_CMD_DONE = NewEvent()
- wxCmdAbort, EVT_CMD_ABORT = NewEvent()
- wxCmdPrepare, EVT_CMD_PREPARE = NewEvent()
- GC_EMPTY = 0
- GC_SEARCH = 1
- GC_PROMPT = 2
- def GrassCmd(cmd, env = None, stdout = None, stderr = None):
- """!Return GRASS command thread"""
- return CommandThread(cmd, env = env,
- stdout = stdout, stderr = stderr)
- class CmdThread(threading.Thread):
- """!Thread for GRASS commands"""
- requestId = 0
- def __init__(self, receiver, requestQ = None, resultQ = None, **kwds):
- """!
- @param receiver event receiver (used in PostEvent)
- """
- threading.Thread.__init__(self, **kwds)
-
- if requestQ is None:
- self.requestQ = Queue.Queue()
- else:
- self.requestQ = requestQ
-
- if resultQ is None:
- self.resultQ = Queue.Queue()
- else:
- self.resultQ = resultQ
-
- self.setDaemon(True)
-
- self.receiver = receiver
- self._want_abort_all = False
-
- self.start()
- def RunCmd(self, *args, **kwds):
- """!Run command in queue
- @param args unnamed command arguments
- @param kwds named command arguments
- @return request id in queue
- """
- CmdThread.requestId += 1
-
- self.requestCmd = None
- self.requestQ.put((CmdThread.requestId, args, kwds))
-
- return CmdThread.requestId
- def SetId(self, id):
- """!Set starting id"""
- CmdThread.requestId = id
-
- def run(self):
- os.environ['GRASS_MESSAGE_FORMAT'] = 'gui'
- while True:
- requestId, args, kwds = self.requestQ.get()
- for key in ('callable', 'onDone', 'onPrepare', 'userData'):
- if key in kwds:
- vars()[key] = kwds[key]
- del kwds[key]
- else:
- vars()[key] = None
-
- if not vars()['callable']:
- vars()['callable'] = GrassCmd
-
- requestTime = time.time()
-
- # prepare
- event = wxCmdPrepare(cmd = args[0],
- time = requestTime,
- pid = requestId,
- onPrepare = vars()['onPrepare'],
- userData = vars()['userData'])
- wx.PostEvent(self.receiver, event)
-
- # run command
- event = wxCmdRun(cmd = args[0],
- pid = requestId)
-
- wx.PostEvent(self.receiver, event)
-
- time.sleep(.1)
- self.requestCmd = vars()['callable'](*args, **kwds)
- if self._want_abort_all:
- self.requestCmd.abort()
- if self.requestQ.empty():
- self._want_abort_all = False
-
- self.resultQ.put((requestId, self.requestCmd.run()))
-
- try:
- returncode = self.requestCmd.module.returncode
- except AttributeError:
- returncode = 0 # being optimistic
-
- try:
- aborted = self.requestCmd.aborted
- except AttributeError:
- aborted = False
-
- time.sleep(.1)
- # set default color table for raster data
- if UserSettings.Get(group = 'rasterLayer', key = 'colorTable', subkey = 'enabled') and \
- args[0][0][:2] == 'r.':
- colorTable = UserSettings.Get(group = 'rasterLayer', key = 'colorTable', subkey = 'selection')
- mapName = None
- if args[0][0] == 'r.mapcalc':
- try:
- mapName = args[0][1].split('=', 1)[0].strip()
- except KeyError:
- pass
- else:
- moduleInterface = GUI(show = None).ParseCommand(args[0])
- outputParam = moduleInterface.get_param(value = 'output', raiseError = False)
- if outputParam and outputParam['prompt'] == 'raster':
- mapName = outputParam['value']
-
- if mapName:
- argsColor = list(args)
- argsColor[0] = [ 'r.colors',
- 'map=%s' % mapName,
- 'color=%s' % colorTable ]
- self.requestCmdColor = vars()['callable'](*argsColor, **kwds)
- self.resultQ.put((requestId, self.requestCmdColor.run()))
-
- event = wxCmdDone(cmd = args[0],
- aborted = aborted,
- returncode = returncode,
- time = requestTime,
- pid = requestId,
- onDone = vars()['onDone'],
- userData = vars()['userData'])
-
- # send event
- wx.PostEvent(self.receiver, event)
-
- def abort(self, abortall = True):
- """!Abort command(s)"""
- if abortall:
- self._want_abort_all = True
- self.requestCmd.abort()
- if self.requestQ.empty():
- self._want_abort_all = False
- # Occurs when some new text appears.
- # Text priority is specified by priority attribute.
- # Priority is 1 (lowest), 2, 3 (highest);
- # value 0 is currently not used and probably will not be used.
- # In theory, it can be used when text is completely uninteresting.
- # It is similar to wx.EVT_TEXT.
- # However, the new text or the whole text are not event attributes.
- gOutputText, EVT_OUTPUT_TEXT = NewEvent()
- # Occurs when ignored command is called.
- # Attribute cmd contains command (as a list).
- gIgnoredCmdRun, EVT_IGNORED_CMD_RUN = NewEvent()
- class GConsole(wx.SplitterWindow):
- """!Create and manage output console for commands run by GUI.
- """
- def __init__(self, parent, margin = False,
- style = wx.TAB_TRAVERSAL | wx.FULL_REPAINT_ON_RESIZE,
- gcstyle = GC_EMPTY,
- ignoredCmdPattern = None,
- **kwargs):
- """!
- @param parent gui parent
- @param margin use margin in output pane (GStc)
- @param style wx.SplitterWindow style
- @param gcstyle GConsole style
- (GC_EMPTY, GC_PROMPT to show command prompt,
- GC_SEARCH to show search widget)
- @param ignoredCmdPattern regular expression specifying commads
- to be ignored (e.g. @c '^d\..*' for display commands)
- """
- wx.SplitterWindow.__init__(self, parent, id = wx.ID_ANY, style = style, **kwargs)
- self.SetName("GConsole")
-
- self.panelOutput = wx.Panel(parent = self, id = wx.ID_ANY)
- self.panelPrompt = wx.Panel(parent = self, id = wx.ID_ANY)
- # initialize variables
- self.parent = parent # GMFrame | CmdPanel | ?
- self._gcstyle = gcstyle
- self.lineWidth = 80
- self._ignoredCmdPattern = ignoredCmdPattern
- # create queues
- self.requestQ = Queue.Queue()
- self.resultQ = Queue.Queue()
-
- # progress bar
- self.progressbar = wx.Gauge(parent = self.panelOutput, id = wx.ID_ANY,
- range = 100, pos = (110, 50), size = (-1, 25),
- style = wx.GA_HORIZONTAL)
- self.Bind(EVT_CMD_PROGRESS, self.OnCmdProgress)
-
- # text control for command output
- self.cmdOutput = GStc(parent = self.panelOutput, id = wx.ID_ANY, margin = margin,
- wrap = None)
- self.cmdOutputTimer = wx.Timer(self.cmdOutput, id = wx.ID_ANY)
- self.Bind(EVT_CMD_OUTPUT, self.OnCmdOutput)
- self.Bind(wx.EVT_TIMER, self.OnProcessPendingOutputWindowEvents)
- self.Bind(EVT_CMD_RUN, self.OnCmdRun)
- self.Bind(EVT_CMD_DONE, self.OnCmdDone)
- # information about available modules
- modulesData = ModulesData()
- # search & command prompt
- # move to the if below
- # search depends on cmd prompt
- self.cmdPrompt = GPromptSTC(parent = self, modulesData = modulesData)
- if not self._gcstyle & GC_PROMPT:
- self.cmdPrompt.Hide()
- if self._gcstyle & GC_SEARCH:
- self.infoCollapseLabelExp = _("Click here to show search module engine")
- self.infoCollapseLabelCol = _("Click here to hide search module engine")
- self.searchPane = wx.CollapsiblePane(parent = self.panelOutput,
- label = self.infoCollapseLabelExp,
- style = wx.CP_DEFAULT_STYLE |
- wx.CP_NO_TLW_RESIZE | wx.EXPAND)
- self.MakeSearchPaneContent(self.searchPane.GetPane(), modulesData)
- self.searchPane.Collapse(True)
- self.Bind(wx.EVT_COLLAPSIBLEPANE_CHANGED, self.OnSearchPaneChanged, self.searchPane)
- self.search.Bind(EVT_MODULE_SELECTED,
- lambda event:
- self.cmdPrompt.SetTextAndFocus(event.name + ' '))
- else:
- self.search = None
- # stream redirection
- self.cmdStdOut = GStdout(self)
- self.cmdStdErr = GStderr(self)
-
- # thread
- self.cmdThread = CmdThread(self, self.requestQ, self.resultQ)
-
- self.outputBox = wx.StaticBox(parent = self.panelOutput, id = wx.ID_ANY,
- label = " %s " % _("Output window"))
- if self._gcstyle & GC_PROMPT:
- cmdLabel = _("Command prompt")
- else:
- cmdLabel = _("Command")
- self.cmdBox = wx.StaticBox(parent = self.panelOutput, id = wx.ID_ANY,
- label = " %s " % cmdLabel)
- # buttons
- self.btnOutputClear = wx.Button(parent = self.panelOutput, id = wx.ID_CLEAR)
- self.btnOutputClear.SetToolTipString(_("Clear output window content"))
- self.btnCmdClear = wx.Button(parent = self.panelOutput, id = wx.ID_CLEAR)
- self.btnCmdClear.SetToolTipString(_("Clear command prompt content"))
- self.btnOutputSave = wx.Button(parent = self.panelOutput, id = wx.ID_SAVE)
- self.btnOutputSave.SetToolTipString(_("Save output window content to the file"))
- self.btnCmdAbort = wx.Button(parent = self.panelOutput, id = wx.ID_STOP)
- self.btnCmdAbort.SetToolTipString(_("Abort running command"))
- self.btnCmdAbort.Enable(False)
- self.btnCmdProtocol = wx.ToggleButton(parent = self.panelOutput, id = wx.ID_ANY,
- label = _("&Protocol"),
- size = self.btnCmdClear.GetSize())
- self.btnCmdProtocol.SetToolTipString(_("Toggle to save list of executed commands into file; "
- "content saved when switching off."))
-
- if not self._gcstyle & GC_PROMPT:
- self.btnCmdClear.Hide()
- self.btnCmdProtocol.Hide()
-
- self.btnCmdClear.Bind(wx.EVT_BUTTON, self.cmdPrompt.OnCmdErase)
- self.btnOutputClear.Bind(wx.EVT_BUTTON, self.OnOutputClear)
- self.btnOutputSave.Bind(wx.EVT_BUTTON, self.OnOutputSave)
- self.btnCmdAbort.Bind(wx.EVT_BUTTON, self.OnCmdAbort)
- self.Bind(EVT_CMD_ABORT, self.OnCmdAbort)
- self.btnCmdProtocol.Bind(wx.EVT_TOGGLEBUTTON, self.OnCmdProtocol)
-
- self._layout()
-
- def _layout(self):
- """!Do layout"""
- outputSizer = wx.BoxSizer(wx.VERTICAL)
- btnSizer = wx.BoxSizer(wx.HORIZONTAL)
- outBtnSizer = wx.StaticBoxSizer(self.outputBox, wx.HORIZONTAL)
- cmdBtnSizer = wx.StaticBoxSizer(self.cmdBox, wx.HORIZONTAL)
-
- if self._gcstyle & GC_PROMPT:
- promptSizer = wx.BoxSizer(wx.VERTICAL)
- promptSizer.Add(item = self.cmdPrompt, proportion = 1,
- flag = wx.EXPAND | wx.LEFT | wx.RIGHT | wx.TOP, border = 3)
-
- if self._gcstyle & GC_SEARCH:
- outputSizer.Add(item = self.searchPane, proportion = 0,
- flag = wx.EXPAND | wx.ALL, border = 3)
- outputSizer.Add(item = self.cmdOutput, proportion = 1,
- flag = wx.EXPAND | wx.ALL, border = 3)
- outputSizer.Add(item = self.progressbar, proportion = 0,
- flag = wx.EXPAND | wx.LEFT | wx.RIGHT, border = 3)
- outBtnSizer.Add(item = self.btnOutputClear, proportion = 1,
- flag = wx.ALIGN_LEFT | wx.LEFT | wx.RIGHT, border = 5)
- outBtnSizer.Add(item = self.btnOutputSave, proportion = 1,
- flag = wx.ALIGN_RIGHT | wx.RIGHT, border = 5)
-
- cmdBtnSizer.Add(item = self.btnCmdProtocol, proportion = 1,
- flag = wx.ALIGN_CENTER | wx.ALIGN_CENTER_VERTICAL | wx.LEFT | wx.RIGHT, border = 5)
- cmdBtnSizer.Add(item = self.btnCmdClear, proportion = 1,
- flag = wx.ALIGN_CENTER | wx.RIGHT, border = 5)
- cmdBtnSizer.Add(item = self.btnCmdAbort, proportion = 1,
- flag = wx.ALIGN_CENTER | wx.RIGHT, border = 5)
-
- if self._gcstyle & GC_PROMPT:
- proportion = (2, 3)
- else:
- proportion = (1, 1)
-
- btnSizer.Add(item = outBtnSizer, proportion = proportion[0],
- flag = wx.ALL | wx.ALIGN_CENTER, border = 5)
- btnSizer.Add(item = cmdBtnSizer, proportion = proportion[1],
- flag = wx.ALIGN_CENTER | wx.TOP | wx.BOTTOM | wx.RIGHT, border = 5)
- outputSizer.Add(item = btnSizer, proportion = 0,
- flag = wx.EXPAND)
-
- outputSizer.Fit(self)
- outputSizer.SetSizeHints(self)
- self.panelOutput.SetSizer(outputSizer)
- # eliminate gtk_widget_size_allocate() warnings
- outputSizer.SetVirtualSizeHints(self.panelOutput)
-
- if self._gcstyle & GC_PROMPT:
- promptSizer.Fit(self)
- promptSizer.SetSizeHints(self)
- self.panelPrompt.SetSizer(promptSizer)
-
- # split window
- if self._gcstyle & GC_PROMPT:
- self.SplitHorizontally(self.panelOutput, self.panelPrompt, -50)
- else:
- self.SplitHorizontally(self.panelOutput, self.panelPrompt, -45)
- self.Unsplit()
- self.SetMinimumPaneSize(self.btnCmdClear.GetSize()[1] + 25)
-
- self.SetSashGravity(1.0)
-
- # layout
- self.SetAutoLayout(True)
- self.Layout()
- def MakeSearchPaneContent(self, pane, modulesData):
- """!Create search pane"""
- border = wx.BoxSizer(wx.VERTICAL)
-
- self.search = SearchModuleWidget(parent = pane,
- modulesData = modulesData)
-
- border.Add(item = self.search, proportion = 0,
- flag = wx.EXPAND | wx.ALL, border = 1)
-
- pane.SetSizer(border)
- border.Fit(pane)
-
- def OnSearchPaneChanged(self, event):
- """!Collapse search module box"""
- if self.searchPane.IsExpanded():
- self.searchPane.SetLabel(self.infoCollapseLabelCol)
- else:
- self.searchPane.SetLabel(self.infoCollapseLabelExp)
-
- self.panelOutput.Layout()
- self.panelOutput.SendSizeEvent()
-
- def GetPanel(self, prompt = True):
- """!Get panel
- @param prompt get prompt / output panel
- @return wx.Panel reference
- """
- if prompt:
- return self.panelPrompt
- return self.panelOutput
-
- def Redirect(self):
- """!Redirect stdout/stderr
- """
- if Debug.GetLevel() == 0 and int(grass.gisenv().get('DEBUG', 0)) == 0:
- # don't redirect when debugging is enabled
- sys.stdout = self.cmdStdOut
- sys.stderr = self.cmdStdErr
- else:
- enc = locale.getdefaultlocale()[1]
- if enc:
- sys.stdout = codecs.getwriter(enc)(sys.__stdout__)
- sys.stderr = codecs.getwriter(enc)(sys.__stderr__)
- else:
- sys.stdout = sys.__stdout__
- sys.stderr = sys.__stderr__
-
- def WriteLog(self, text, style = None, wrap = None,
- switchPage = False, priority = 1):
- """!Generic method for writing log message in
- given style
- @param line text line
- @param style text style (see GStc)
- @param stdout write to stdout or stderr
- @param switchPage for backward compatibility
- (replace by priority: False=1, True=2)
- @param priority priority of this message
- (0=no priority, 1=normal, 2=medium, 3=high)
- """
- self.cmdOutput.SetStyle()
- # documenting old behavior/implementation:
- # switch notebook if required
- if priority == 1:
- if switchPage:
- priority = 2
- event = gOutputText(priority = priority)
- wx.PostEvent(self, event)
-
- if not style:
- style = self.cmdOutput.StyleDefault
-
- # p1 = self.cmdOutput.GetCurrentPos()
- p1 = self.cmdOutput.GetEndStyled()
- # self.cmdOutput.GotoPos(p1)
- self.cmdOutput.DocumentEnd()
-
- for line in text.splitlines():
- # fill space
- if len(line) < self.lineWidth:
- diff = self.lineWidth - len(line)
- line += diff * ' '
-
- self.cmdOutput.AddTextWrapped(line, wrap = wrap) # adds '\n'
-
- p2 = self.cmdOutput.GetCurrentPos()
-
- self.cmdOutput.StartStyling(p1, 0xff)
- self.cmdOutput.SetStyling(p2 - p1, style)
-
- self.cmdOutput.EnsureCaretVisible()
-
- def WriteCmdLog(self, line, pid = None, switchPage = True):
- """!Write message in selected style
-
- @param line message to be printed
- @param pid process pid or None
- @param switchPage True to switch page
- """
- if pid:
- line = '(' + str(pid) + ') ' + line
- self.WriteLog(line, style = self.cmdOutput.StyleCommand, switchPage = switchPage)
- def WriteWarning(self, line):
- """!Write message in warning style"""
- self.WriteLog(line, style = self.cmdOutput.StyleWarning, switchPage = True)
- def WriteError(self, line):
- """!Write message in error style"""
- self.WriteLog(line, style = self.cmdOutput.StyleError, switchPage = True)
- def RunCmd(self, command, compReg = True, switchPage = False, skipInterface = False,
- onDone = None, onPrepare = None, userData = None):
- """!Run command typed into console command prompt (GPrompt).
- @todo Document the other event.
- @todo Solve problem with the other event
- (now uses gOutputText event but there is no text,
- use onPrepare handler instead?)
- Posts event EVT_IGNORED_CMD_RUN when command which should be ignored
- (according to ignoredCmdPattern) is run.
- For example, see layer manager which handles d.* on its own.
- @param command command given as a list (produced e.g. by utils.split())
- @param compReg True use computation region
- @param switchPage switch to output page
- @param skipInterface True to do not launch GRASS interface
- parser when command has no arguments given
- @param onDone function to be called when command is finished
- @param onPrepare function to be called before command is launched
- @param userData data defined for the command
- """
- if len(command) == 0:
- Debug.msg(2, "GPrompt:RunCmd(): empty command")
- return
-
- # update history file
- env = grass.gisenv()
- try:
- filePath = os.path.join(env['GISDBASE'],
- env['LOCATION_NAME'],
- env['MAPSET'],
- '.bash_history')
- fileHistory = codecs.open(filePath, encoding = 'utf-8', mode = 'a')
- except IOError, e:
- GError(_("Unable to write file '%(filePath)s'.\n\nDetails: %(error)s") %
- {'filePath': filePath, 'error' : e },
- parent = self)
- fileHistory = None
-
- if fileHistory:
- try:
- fileHistory.write(' '.join(command) + os.linesep)
- finally:
- fileHistory.close()
-
- if command[0] in globalvar.grassCmd:
- # send GRASS command without arguments to GUI command interface
- # except ignored commands (event is emitted)
- if self._ignoredCmdPattern and \
- re.compile(self._ignoredCmdPattern).search(command[0]) and \
- '--help' not in command:
- event = gIgnoredCmdRun(cmd = command)
- wx.PostEvent(self, event)
- return
- else:
- # other GRASS commands (r|v|g|...)
- try:
- task = GUI(show = None).ParseCommand(command)
- except GException, e:
- GError(parent = self,
- message = unicode(e),
- showTraceback = False)
- return
-
- hasParams = False
- if task:
- options = task.get_options()
- hasParams = options['params'] and options['flags']
- # check for <input>=-
- for p in options['params']:
- if p.get('prompt', '') == 'input' and \
- p.get('element', '') == 'file' and \
- p.get('age', 'new') == 'old' and \
- p.get('value', '') == '-':
- GError(parent = self,
- message = _("Unable to run command:\n%(cmd)s\n\n"
- "Option <%(opt)s>: read from standard input is not "
- "supported by wxGUI") % { 'cmd': ' '.join(command),
- 'opt': p.get('name', '') })
- return
-
- if len(command) == 1 and hasParams and \
- command[0] != 'v.krige':
- # no arguments given
- try:
- GUI(parent = self, lmgr = self.parent).ParseCommand(command)
- except GException, e:
- print >> sys.stderr, e
- return
- # documenting old behavior/implementation:
- # switch and focus if required
- # TODO: this probably should be run command event
- if switchPage:
- priority = 3
- event = gOutputText(priority = priority)
- wx.PostEvent(self, event)
- # activate computational region (set with g.region)
- # for all non-display commands.
- if compReg:
- tmpreg = os.getenv("GRASS_REGION")
- if "GRASS_REGION" in os.environ:
- del os.environ["GRASS_REGION"]
-
- # process GRASS command with argument
- self.cmdThread.RunCmd(command, stdout = self.cmdStdOut, stderr = self.cmdStdErr,
- onDone = onDone, onPrepare = onPrepare, userData = userData,
- env = os.environ.copy())
- self.cmdOutputTimer.Start(50)
-
- # deactivate computational region and return to display settings
- if compReg and tmpreg:
- os.environ["GRASS_REGION"] = tmpreg
- else:
- # Send any other command to the shell. Send output to
- # console output window
- if len(command) == 1 and not skipInterface:
- try:
- task = gtask.parse_interface(command[0])
- except:
- task = None
- else:
- task = None
-
- if task:
- # process GRASS command without argument
- GUI(parent = self, lmgr = self.parent).ParseCommand(command)
- else:
- self.cmdThread.RunCmd(command, stdout = self.cmdStdOut, stderr = self.cmdStdErr,
- onDone = onDone, onPrepare = onPrepare, userData = userData)
- self.cmdOutputTimer.Start(50)
-
- def OnOutputClear(self, event):
- """!Clear content of output window"""
- self.cmdOutput.SetReadOnly(False)
- self.cmdOutput.ClearAll()
- self.cmdOutput.SetReadOnly(True)
- self.progressbar.SetValue(0)
- def GetProgressBar(self):
- """!Return progress bar widget"""
- return self.progressbar
-
- def GetLog(self, err = False):
- """!Get widget used for logging
- @param err True to get stderr widget
- """
- if err:
- return self.cmdStdErr
-
- return self.cmdStdOut
-
- def OnOutputSave(self, event):
- """!Save (selected) text from output window to the file"""
- text = self.cmdOutput.GetSelectedText()
- if not text:
- text = self.cmdOutput.GetText()
-
- # add newline if needed
- if len(text) > 0 and text[-1] != '\n':
- text += '\n'
-
- dlg = wx.FileDialog(self, message = _("Save file as..."),
- defaultFile = "grass_cmd_output.txt",
- wildcard = _("%(txt)s (*.txt)|*.txt|%(files)s (*)|*") %
- {'txt': _("Text files"), 'files': _("Files")},
- style = wx.SAVE | wx.FD_OVERWRITE_PROMPT)
-
- # Show the dialog and retrieve the user response. If it is the OK response,
- # process the data.
- if dlg.ShowModal() == wx.ID_OK:
- path = dlg.GetPath()
-
- try:
- output = open(path, "w")
- output.write(text)
- except IOError, e:
- GError(_("Unable to write file '%(path)s'.\n\nDetails: %(error)s") % {'path': path, 'error': e})
- finally:
- output.close()
- message = _("Commands output saved into '%s'") % path
- wx.PostEvent(self, gShowNotification(self.GetId(), message = message))
-
- dlg.Destroy()
- def GetCmd(self):
- """!Get running command or None"""
- return self.requestQ.get()
-
- def SetCopyingOfSelectedText(self, copy):
- """!Enable or disable copying of selected text in to clipboard.
- Effects prompt and output.
-
- @param copy True for enable, False for disable
- """
- if copy:
- self.cmdPrompt.Bind(stc.EVT_STC_PAINTED, self.cmdPrompt.OnTextSelectionChanged)
- self.cmdOutput.Bind(stc.EVT_STC_PAINTED, self.cmdOutput.OnTextSelectionChanged)
- else:
- self.cmdPrompt.Unbind(stc.EVT_STC_PAINTED)
- self.cmdOutput.Unbind(stc.EVT_STC_PAINTED)
- def OnCmdOutput(self, event):
- """!Print command output
- Posts event EVT_OUTPUT_TEXT with priority attribute set to 1.
- """
- message = event.text
- type = event.type
- self.cmdOutput.AddStyledMessage(message, type)
- # documenting old behavior/implementation:
- # add elipses if not active
- event = gOutputText(priority = 1)
- wx.PostEvent(self, event)
- def OnCmdProgress(self, event):
- """!Update progress message info"""
- self.progressbar.SetValue(event.value)
- event.Skip()
- def CmdProtocolSave(self):
- """Save commands protocol into the file"""
- if not hasattr(self, 'cmdFileProtocol'):
- return # it should not happen
-
- try:
- output = open(self.cmdFileProtocol, "w")
- cmds = self.cmdPrompt.GetCommands()
- output.write('\n'.join(cmds))
- if len(cmds) > 0:
- output.write('\n')
- except IOError, e:
- GError(_("Unable to write file '%(filePath)s'.\n\nDetails: %(error)s") %
- {'filePath': self.cmdFileProtocol, 'error': e})
- finally:
- output.close()
-
- message = _("Commands protocol saved into '%s'") % self.cmdFileProtocol
- wx.PostEvent(self, gShowNotification(self.GetId(), message = message))
- del self.cmdFileProtocol
-
- def OnCmdProtocol(self, event = None):
- """!Save commands into file"""
- if not event.IsChecked():
- # stop capturing commands, save list of commands to the
- # protocol file
- self.CmdProtocolSave()
- else:
- # start capturing commands
- self.cmdPrompt.ClearCommands()
- # ask for the file
- dlg = wx.FileDialog(self, message = _("Save file as..."),
- defaultFile = "grass_cmd_protocol.txt",
- wildcard = _("%(txt)s (*.txt)|*.txt|%(files)s (*)|*") %
- {'txt': _("Text files"), 'files': _("Files")},
- style = wx.SAVE | wx.FD_OVERWRITE_PROMPT)
- if dlg.ShowModal() == wx.ID_OK:
- self.cmdFileProtocol = dlg.GetPath()
- else:
- wx.CallAfter(self.btnCmdProtocol.SetValue, False)
-
- dlg.Destroy()
-
- event.Skip()
-
- def OnCmdAbort(self, event):
- """!Abort running command"""
- self.cmdThread.abort()
- event.Skip()
-
- def OnCmdRun(self, event):
- """!Run command"""
- self.WriteCmdLog('(%s)\n%s' % (str(time.ctime()), ' '.join(event.cmd)))
- self.btnCmdAbort.Enable()
- event.Skip()
- def OnCmdDone(self, event):
- """!Command done (or aborted)
- Posts event EVT_MAP_CREATED.
- """
- # Process results here
- try:
- ctime = time.time() - event.time
- if ctime < 60:
- stime = _("%d sec") % int(ctime)
- else:
- mtime = int(ctime / 60)
- stime = _("%(min)d min %(sec)d sec") % { 'min' : mtime,
- 'sec' : int(ctime - (mtime * 60)) }
- except KeyError:
- # stopped deamon
- stime = _("unknown")
-
- if event.aborted:
- # Thread aborted (using our convention of None return)
- self.WriteLog(_('Please note that the data are left in inconsistent state '
- 'and may be corrupted'), self.cmdOutput.StyleWarning)
- msg = _('Command aborted')
- else:
- msg = _('Command finished')
-
- self.WriteCmdLog('(%s) %s (%s)' % (str(time.ctime()), msg, stime))
- self.btnCmdAbort.Enable(False)
-
- if event.onDone:
- event.onDone(cmd = event.cmd, returncode = event.returncode)
-
- self.progressbar.SetValue(0) # reset progress bar on '0%'
- self.cmdOutputTimer.Stop()
- if event.cmd[0] == 'g.gisenv':
- Debug.SetLevel()
- self.Redirect()
- # do nothing when no map added
- if event.returncode != 0 or event.aborted:
- event.Skip()
- return
- # find which maps were created
- try:
- task = GUI(show = None).ParseCommand(event.cmd)
- except GException, e:
- print >> sys.stderr, e
- task = None
- return
-
- for p in task.get_options()['params']:
- prompt = p.get('prompt', '')
- if prompt in ('raster', 'vector', '3d-raster') and \
- p.get('age', 'old') == 'new' and \
- p.get('value', None):
- name = p.get('value')
- if '@' not in name:
- name = name + '@' + grass.gisenv()['MAPSET']
- mapEvent = gMapCreated(self.GetId(),
- name = name, ltype = prompt, add = None)
- wx.PostEvent(self, mapEvent)
- event.Skip()
- def OnProcessPendingOutputWindowEvents(self, event):
- wx.GetApp().ProcessPendingEvents()
- def ResetFocus(self):
- """!Reset focus"""
- self.cmdPrompt.SetFocus()
-
- def GetPrompt(self):
- """!Get prompt"""
- return self.cmdPrompt
-
- class GStdout:
- """!GConsole standard output
- Based on FrameOutErr.py
- Name: FrameOutErr.py
- Purpose: Redirecting stdout / stderr
- Author: Jean-Michel Fauth, Switzerland
- Copyright: (c) 2005-2007 Jean-Michel Fauth
- Licence: GPL
- """
- def __init__(self, receiver):
- """!
- @param receiver event receiver (used in PostEvent)
- """
- self.receiver = receiver
- def write(self, s):
- if len(s) == 0 or s == '\n':
- return
- for line in s.splitlines():
- if len(line) == 0:
- continue
-
- evt = wxCmdOutput(text = line + '\n',
- type = '')
- wx.PostEvent(self.receiver, evt)
-
- class GStderr:
- """!GConsole standard error output
- Based on FrameOutErr.py
- Name: FrameOutErr.py
- Purpose: Redirecting stdout / stderr
- Author: Jean-Michel Fauth, Switzerland
- Copyright: (c) 2005-2007 Jean-Michel Fauth
- Licence: GPL
- """
- def __init__(self, receiver):
- """!
- @param receiver event receiver (used in PostEvent)
- """
- self.receiver = receiver
-
- self.type = ''
- self.message = ''
- self.printMessage = False
-
- def flush(self):
- pass
-
- def write(self, s):
- if "GtkPizza" in s:
- return
-
- # remove/replace escape sequences '\b' or '\r' from stream
- progressValue = -1
-
- for line in s.splitlines():
- if len(line) == 0:
- continue
-
- if 'GRASS_INFO_PERCENT' in line:
- value = int(line.rsplit(':', 1)[1].strip())
- if value >= 0 and value < 100:
- progressValue = value
- else:
- progressValue = 0
- elif 'GRASS_INFO_MESSAGE' in line:
- self.type = 'message'
- self.message += line.split(':', 1)[1].strip() + '\n'
- elif 'GRASS_INFO_WARNING' in line:
- self.type = 'warning'
- self.message += line.split(':', 1)[1].strip() + '\n'
- elif 'GRASS_INFO_ERROR' in line:
- self.type = 'error'
- self.message += line.split(':', 1)[1].strip() + '\n'
- elif 'GRASS_INFO_END' in line:
- self.printMessage = True
- elif self.type == '':
- if len(line) == 0:
- continue
- evt = wxCmdOutput(text = line,
- type = '')
- wx.PostEvent(self.receiver, evt)
- elif len(line) > 0:
- self.message += line.strip() + '\n'
- if self.printMessage and len(self.message) > 0:
- evt = wxCmdOutput(text = self.message,
- type = self.type)
- wx.PostEvent(self.receiver, evt)
- self.type = ''
- self.message = ''
- self.printMessage = False
- # update progress message
- if progressValue > -1:
- # self.gmgauge.SetValue(progressValue)
- evt = wxCmdProgress(value = progressValue)
- wx.PostEvent(self.receiver, evt)
-
- class GStc(stc.StyledTextCtrl):
- """!Styled text control for GRASS stdout and stderr.
- Based on FrameOutErr.py
- Name: FrameOutErr.py
- Purpose: Redirecting stdout / stderr
- Author: Jean-Michel Fauth, Switzerland
- Copyright: (c) 2005-2007 Jean-Michel Fauth
- Licence: GPL
- """
- def __init__(self, parent, id, margin = False, wrap = None):
- stc.StyledTextCtrl.__init__(self, parent, id)
- self.parent = parent
- self.SetUndoCollection(True)
- self.SetReadOnly(True)
- # remember position of line begining (used for '\r')
- self.linePos = -1
- #
- # styles
- #
- self.SetStyle()
-
- #
- # line margins
- #
- # TODO print number only from cmdlog
- self.SetMarginWidth(1, 0)
- self.SetMarginWidth(2, 0)
- if margin:
- self.SetMarginType(0, stc.STC_MARGIN_NUMBER)
- self.SetMarginWidth(0, 30)
- else:
- self.SetMarginWidth(0, 0)
- #
- # miscellaneous
- #
- self.SetViewWhiteSpace(False)
- self.SetTabWidth(4)
- self.SetUseTabs(False)
- self.UsePopUp(True)
- self.SetSelBackground(True, "#FFFF00")
- self.SetUseHorizontalScrollBar(True)
- #
- # bindings
- #
- self.Bind(wx.EVT_WINDOW_DESTROY, self.OnDestroy)
-
- def OnTextSelectionChanged(self, event):
- """!Copy selected text to clipboard and skip event.
- The same function is in TextCtrlAutoComplete class (prompt.py).
- """
- wx.CallAfter(self.Copy)
- event.Skip()
-
- def SetStyle(self):
- """!Set styles for styled text output windows with type face
- and point size selected by user (Courier New 10 is default)"""
-
- typeface = UserSettings.Get(group = 'appearance', key = 'outputfont', subkey = 'type')
- if typeface == "":
- typeface = "Courier New"
-
- typesize = UserSettings.Get(group = 'appearance', key = 'outputfont', subkey = 'size')
- if typesize == None or typesize <= 0:
- typesize = 10
- typesize = float(typesize)
-
- self.StyleDefault = 0
- self.StyleDefaultSpec = "face:%s,size:%d,fore:#000000,back:#FFFFFF" % (typeface, typesize)
- self.StyleCommand = 1
- self.StyleCommandSpec = "face:%s,size:%d,,fore:#000000,back:#bcbcbc" % (typeface, typesize)
- self.StyleOutput = 2
- self.StyleOutputSpec = "face:%s,size:%d,,fore:#000000,back:#FFFFFF" % (typeface, typesize)
- # fatal error
- self.StyleError = 3
- self.StyleErrorSpec = "face:%s,size:%d,,fore:#7F0000,back:#FFFFFF" % (typeface, typesize)
- # warning
- self.StyleWarning = 4
- self.StyleWarningSpec = "face:%s,size:%d,,fore:#0000FF,back:#FFFFFF" % (typeface, typesize)
- # message
- self.StyleMessage = 5
- self.StyleMessageSpec = "face:%s,size:%d,,fore:#000000,back:#FFFFFF" % (typeface, typesize)
- # unknown
- self.StyleUnknown = 6
- self.StyleUnknownSpec = "face:%s,size:%d,,fore:#000000,back:#FFFFFF" % (typeface, typesize)
-
- # default and clear => init
- self.StyleSetSpec(stc.STC_STYLE_DEFAULT, self.StyleDefaultSpec)
- self.StyleClearAll()
- self.StyleSetSpec(self.StyleCommand, self.StyleCommandSpec)
- self.StyleSetSpec(self.StyleOutput, self.StyleOutputSpec)
- self.StyleSetSpec(self.StyleError, self.StyleErrorSpec)
- self.StyleSetSpec(self.StyleWarning, self.StyleWarningSpec)
- self.StyleSetSpec(self.StyleMessage, self.StyleMessageSpec)
- self.StyleSetSpec(self.StyleUnknown, self.StyleUnknownSpec)
- def OnDestroy(self, evt):
- """!The clipboard contents can be preserved after
- the app has exited"""
-
- wx.TheClipboard.Flush()
- evt.Skip()
- def AddTextWrapped(self, txt, wrap = None):
- """!Add string to text area.
- String is wrapped and linesep is also added to the end
- of the string"""
- # allow writing to output window
- self.SetReadOnly(False)
- if wrap:
- txt = textwrap.fill(txt, wrap) + '\n'
- else:
- if txt[-1] != '\n':
- txt += '\n'
- if '\r' in txt:
- self.linePos = -1
- for seg in txt.split('\r'):
- if self.linePos > -1:
- self.SetCurrentPos(self.linePos)
- self.ReplaceSelection(seg)
- else:
- self.linePos = self.GetCurrentPos()
- self.AddText(seg)
- else:
- self.linePos = self.GetCurrentPos()
-
- try:
- self.AddText(txt)
- except UnicodeDecodeError:
- enc = UserSettings.Get(group = 'atm', key = 'encoding', subkey = 'value')
- if enc:
- txt = unicode(txt, enc)
- elif 'GRASS_DB_ENCODING' in os.environ:
- txt = unicode(txt, os.environ['GRASS_DB_ENCODING'])
- else:
- txt = EncodeString(txt)
-
- self.AddText(txt)
- # reset output window to read only
- self.SetReadOnly(True)
- def AddStyledMessage(self, message, style = None):
- """!Add message to text area.
- Handles messages with progress percentages.
- @param message message to be added
- @param style style of message, allowed values: 'message', 'warning', 'error' or None
- """
- # message prefix
- if style == 'warning':
- message = 'WARNING: ' + message
- elif style == 'error':
- message = 'ERROR: ' + message
-
- p1 = self.GetEndStyled()
- self.GotoPos(p1)
-
- # is this still needed?
- if '\b' in message:
- if self.linePos < 0:
- self.linePos = p1
- last_c = ''
- for c in message:
- if c == '\b':
- self.linePos -= 1
- else:
- if c == '\r':
- pos = self.GetCurLine()[1]
- # self.SetCurrentPos(pos)
- else:
- self.SetCurrentPos(self.linePos)
- self.ReplaceSelection(c)
- self.linePos = self.GetCurrentPos()
- if c != ' ':
- last_c = c
- if last_c not in ('0123456789'):
- self.AddTextWrapped('\n', wrap = None)
- self.linePos = -1
- else:
- self.linePos = -1 # don't force position
- if '\n' not in message:
- self.AddTextWrapped(message, wrap = 60)
- else:
- self.AddTextWrapped(message, wrap = None)
- p2 = self.GetCurrentPos()
-
- if p2 >= p1:
- self.StartStyling(p1, 0xff)
-
- if style == 'error':
- self.SetStyling(p2 - p1, self.StyleError)
- elif style == 'warning':
- self.SetStyling(p2 - p1, self.StyleWarning)
- elif style == 'message':
- self.SetStyling(p2 - p1, self.StyleMessage)
- else: # unknown
- self.SetStyling(p2 - p1, self.StyleUnknown)
-
- self.EnsureCaretVisible()
|