etisserant@48: #!/usr/bin/env python etisserant@48: # -*- coding: utf-8 -*- etisserant@48: etisserant@48: #This file is part of Beremiz, a Integrated Development Environment for etisserant@48: #programming IEC 61131-3 automates supporting plcopen standard and CanFestival. etisserant@48: # etisserant@48: #Copyright (C) 2007: Edouard TISSERANT and Laurent BESSARD etisserant@48: # etisserant@48: #See COPYING file for copyrights details. etisserant@48: # etisserant@48: #This library is free software; you can redistribute it and/or etisserant@48: #modify it under the terms of the GNU General Public etisserant@48: #License as published by the Free Software Foundation; either etisserant@48: #version 2.1 of the License, or (at your option) any later version. etisserant@48: # etisserant@48: #This library is distributed in the hope that it will be useful, etisserant@48: #but WITHOUT ANY WARRANTY; without even the implied warranty of etisserant@48: #MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU etisserant@48: #General Public License for more details. etisserant@48: # etisserant@48: #You should have received a copy of the GNU General Public etisserant@48: #License along with this library; if not, write to the Free Software etisserant@48: #Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA etisserant@48: etisserant@48: # etisserant@48: # based on wxPopen.py from boa-constructor etisserant@48: # etisserant@48: etisserant@48: import time etisserant@48: from StringIO import StringIO etisserant@48: lbessard@70: import wx etisserant@48: etisserant@48: class ProcessRunnerMix: lbessard@70: lbessard@70: if wx.VERSION < (2, 6, 0): lbessard@70: def Bind(self, event, function, id = None): lbessard@70: if id is not None: lbessard@70: event(self, id, function) lbessard@70: else: lbessard@70: event(self, function) lbessard@70: etisserant@48: def __init__(self, input, handler=None): etisserant@48: if handler is None: etisserant@48: handler = self etisserant@48: self.handler = handler lbessard@70: handler.Bind(wx.EVT_MENU, self.OnIdle) lbessard@70: handler.Bind(wx.EVT_END_PROCESS, self.OnProcessEnded, id=-1) lbessard@70: etisserant@48: input.reverse() # so we can pop etisserant@48: self.input = input etisserant@48: etisserant@48: self.reset() etisserant@48: etisserant@48: def reset(self): etisserant@48: self.process = None etisserant@48: self.pid = -1 etisserant@48: self.output = [] etisserant@48: self.errors = [] etisserant@48: self.inputStream = None etisserant@48: self.errorStream = None etisserant@48: self.outputStream = None etisserant@48: self.outputFunc = None etisserant@48: self.errorsFunc = None etisserant@48: self.finishedFunc = None lbessard@70: self.finished = False lbessard@70: self.responded = False etisserant@48: etisserant@48: def execute(self, cmd): lbessard@70: self.process = wx.Process(self.handler) etisserant@48: self.process.Redirect() etisserant@48: lbessard@70: self.pid = wx.Execute(cmd, wx.EXEC_NOHIDE, self.process) etisserant@48: etisserant@48: self.inputStream = self.process.GetOutputStream() etisserant@48: self.errorStream = self.process.GetErrorStream() etisserant@48: self.outputStream = self.process.GetInputStream() etisserant@48: etisserant@48: #self.OnIdle() lbessard@70: wx.WakeUpIdle() etisserant@48: etisserant@48: def setCallbacks(self, output, errors, finished): etisserant@48: self.outputFunc = output etisserant@48: self.errorsFunc = errors etisserant@48: self.finishedFunc = finished etisserant@48: etisserant@48: def detach(self): etisserant@48: if self.process is not None: etisserant@48: self.process.CloseOutput() etisserant@48: self.process.Detach() etisserant@48: self.process = None etisserant@48: etisserant@48: def kill(self): etisserant@48: if self.process is not None: etisserant@48: self.process.CloseOutput() lbessard@70: if wx.Process_Kill(self.pid, wx.SIGTERM) != wx.KILL_OK: lbessard@70: wx.Process_Kill(self.pid, wx.SIGKILL) etisserant@48: self.process = None etisserant@48: etisserant@48: def updateStream(self, stream, data): etisserant@48: if stream and stream.CanRead(): etisserant@48: if not self.responded: lbessard@70: self.responded = True etisserant@48: text = stream.read() etisserant@48: data.append(text) etisserant@48: return text etisserant@48: else: etisserant@48: return None etisserant@48: etisserant@48: def updateInpStream(self, stream, input): etisserant@48: if stream and input: etisserant@48: line = input.pop() etisserant@48: stream.write(line) etisserant@48: etisserant@48: def updateErrStream(self, stream, data): etisserant@48: return self.updateStream(stream, data) etisserant@48: etisserant@48: def updateOutStream(self, stream, data): etisserant@48: return self.updateStream(stream, data) etisserant@48: etisserant@48: def OnIdle(self, event=None): etisserant@48: if self.process is not None: etisserant@48: self.updateInpStream(self.inputStream, self.input) etisserant@48: e = self.updateErrStream(self.errorStream, self.errors) etisserant@48: if e is not None and self.errorsFunc is not None: lbessard@70: wx.CallAfter(self.errorsFunc, e) etisserant@48: o = self.updateOutStream(self.outputStream, self.output) etisserant@48: if o is not None and self.outputFunc is not None: lbessard@70: wx.CallAfter(self.outputFunc, o) etisserant@48: lbessard@70: #wx.WakeUpIdle() etisserant@48: #time.sleep(0.001) etisserant@48: etisserant@48: def OnProcessEnded(self, event): etisserant@48: self.OnIdle() etisserant@48: pid,exitcode = event.GetPid(), event.GetExitCode() etisserant@48: if self.process: etisserant@48: self.process.Destroy() etisserant@48: self.process = None etisserant@48: lbessard@70: self.finished = True etisserant@48: etisserant@48: # XXX doesn't work ??? lbessard@70: #self.handler.Disconnect(-1, wx.EVT_IDLE) etisserant@48: etisserant@48: if self.finishedFunc: lbessard@70: wx.CallAfter(self.finishedFunc, pid, exitcode) etisserant@48: lbessard@70: class ProcessRunner(wx.EvtHandler, ProcessRunnerMix): etisserant@48: def __init__(self, input): lbessard@70: wx.EvtHandler.__init__(self) etisserant@48: ProcessRunnerMix.__init__(self, input) etisserant@48: etisserant@48: def wxPopen3(cmd, input, output, errors, finish, handler=None): etisserant@48: p = ProcessRunnerMix(input, handler) etisserant@48: p.setCallbacks(output, errors, finish) etisserant@48: p.execute(cmd) etisserant@48: return p etisserant@48: etisserant@48: