pipedprocess.cpp

   1 #include <wx/timer.h>
   2 #include <wx/txtstrm.h>
   3 #include "pipedprocess.h"
   4
   5 DEFINE_EVENT_TYPE(EVT_PROCESS_INPUT)
   6 DEFINE_EVENT_TYPE(EVT_PROCESS_OUTPUT)
   7 DEFINE_EVENT_TYPE(EVT_PROCESS_CREATED)
   8 DEFINE_EVENT_TYPE(EVT_PROCESS_TERMINATED)
   9
  10 BEGIN_EVENT_TABLE(PipedProcess, wxProcess)
  11     EVT_TIMER(ID_QEMU_TIMER_POLL_PROCESS, PipedProcess::OnTimer)
  12     EVT_IDLE(PipedProcess::OnIdle)
  13 END_EVENT_TABLE()
  14
  15 PipedProcess::PipedProcess() : wxProcess(){
  16         Redirect();
  17 }
  18
  19 PipedProcess::~PipedProcess()
  20 {
  21 }
  22
  23 long PipedProcess::Launch(const wxString& cmd,unsigned int interval){
  24         wxCommandEvent event(EVT_PROCESS_CREATED,wxID_ANY);
  25         event.SetEventObject(this);
  26         event.SetString(cmd + wxT("\n"));
  27         SendEvent(event);
  28
  29         pid = wxExecute(cmd, wxEXEC_ASYNC | wxEXEC_NODISABLE, this);
  30         running = true;
  31         if (pid)
  32         {
  33                 /* close stdin of child process */
  34                 CloseOutput();
  35                 timer.SetOwner(this, ID_QEMU_TIMER_POLL_PROCESS);
  36                 timer.Start(interval);
  37         }
  38         return pid;
  39 }
  40
  41 void PipedProcess::SendString(const wxString& text)
  42 {
  43 #if 0
  44         if(!running)
  45                 return;
  46
  47         wxOutputStream *out = GetOutputStream();
  48         if(out)
  49         {
  50                 wxTextOutputStream sin(*out);
  51                 wxString msg = text + wxT("\n");
  52                 sin.WriteString(msg);
  53                 wxCommandEvent event(EVT_PROCESS_OUTPUT,wxID_ANY);
  54                 event.SetEventObject(this);
  55                 event.SetString(msg);
  56                 SendEvent(event);
  57         }
  58 #endif
  59 }
  60
  61 bool PipedProcess::HasInput()
  62 {
  63         bool hasInput = false;
  64
  65         if (IsInputAvailable())
  66         {
  67                 wxInputStream *in = GetInputStream();
  68                 if(in->CanRead()){
  69                         char buffer[4096];
  70                         in->Read(buffer,WXSIZEOF(buffer) - 1);
  71                         buffer[in->LastRead()] = '\0';
  72                         // remove the first line, because it contains some non printable characters
  73                         char *p = strchr(buffer,'\n');
  74                         if(p){
  75                                 wxString msg(++p,wxConvUTF8,in->LastRead() - (p - buffer));
  76                                 wxCommandEvent event(EVT_PROCESS_INPUT,wxID_ANY);
  77                                 event.SetEventObject(this);
  78                                 event.SetString(msg);
  79                                 SendEvent(event);
  80                         }
  81                 }
  82                 hasInput = true;
  83         }
  84
  85         if(IsErrorAvailable())
  86         {
  87                 wxTextInputStream serr(*GetErrorStream());
  88
  89                 wxString msg;
  90                 msg << serr.ReadLine();
  91
  92                 wxCommandEvent event(EVT_PROCESS_INPUT,wxID_ANY);
  93                 event.SetEventObject(this);
  94                 event.SetString(msg);
  95                 SendEvent(event);
  96
  97                 hasInput = true;
  98         }
  99
 100         return hasInput;
 101 }
 102
 103 void PipedProcess::OnTerminate(int pid, int status)
 104 {
 105         running = false;
 106         timer.Stop();
 107         // show the rest of the output
 108         while (HasInput())
 109                 ;
 110
 111         wxCommandEvent event(EVT_PROCESS_TERMINATED,wxID_ANY);
 112         event.SetEventObject(this);
 113         event.SetInt(status);
 114         SendEvent(event);
 115
 116         //delete this;
 117 }
 118
 119 void PipedProcess::OnTimer(wxTimerEvent& event){
 120         while (HasInput())
 121                 ;
 122 }
 123
 124 void PipedProcess::OnIdle(wxIdleEvent& event){
 125         while (HasInput())
 126                 ;
 127 }
 128
 129 void PipedProcess::AddEventHandler(wxEvtHandler *handler){
 130         evtHandlers.Add(handler);
 131 }
 132
 133 void PipedProcess::SendEvent(wxCommandEvent& event){
 134         for(int i = 0; i < evtHandlers.GetCount(); i++){
 135                 evtHandlers.Item(i)->ProcessEvent(event);
 136         }
 137 }