PFitOutputHandler.cpp

Go to the documentation of this file.

/****************************************************************************
 
  PFitOutputHandler.cpp
 
  Author: Andreas Suter
  e-mail: andreas.suter@psi.ch
 
*****************************************************************************/
 
/***************************************************************************
 *   Copyright (C) 2009-2026 by Andreas Suter                              *
 *   andreas.suter@psi.ch                                                  *
 *                                                                         *
 *   This program is free software; you can redistribute it and/or modify  *
 *   it under the terms of the GNU General Public License as published by  *
 *   the Free Software Foundation; either version 2 of the License, or     *
 *   (at your option) any later version.                                   *
 *                                                                         *
 *   This program is distributed in the hope that it will be useful,       *
 *   but WITHOUT ANY WARRANTY; without even the implied warranty of        *
 *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the         *
 *   GNU General Public License for more details.                          *
 *                                                                         *
 *   You should have received a copy of the GNU General Public License     *
 *   along with this program; if not, write to the                         *
 *   Free Software Foundation, Inc.,                                       *
 *   59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.             *
 ***************************************************************************/

 
#include <QTimer>
 
#include <QtDebug>
 
#include "PFitOutputHandler.h"
 
//----------------------------------------------------------------------------------------------------
PFitOutputHandler::PFitOutputHandler(QString workingDirectory, QVector<QString> &cmd)
{
  if (cmd.empty())
    return;
 
  // Layout
  fVbox = std::make_unique<QVBoxLayout>( this );
//Qt.3x  fVbox->resize(800, 500);
  fOutput = std::make_unique<QPlainTextEdit>();
  fOutput->setMaximumBlockCount(1000);
  fVbox->addWidget(fOutput.get());
  fOutput->setMinimumSize(800, 455);
  fOutput->setReadOnly(true);
  connect( fOutput.get(), SIGNAL( destroyed() ), this, SLOT( quitButtonPressed() ) );
  fQuitButton = std::make_unique<QPushButton>( tr("Fitting...") );
  fVbox->addWidget(fQuitButton.get());
  connect( fQuitButton.get(), SIGNAL( clicked() ), this, SLOT( quitButtonPressed() ) );
  resize( 800, 500 );
  fQuitButton->setFocus();
 
  // QProcess related code
  fProc = std::make_unique<QProcess>( this );
 
  // make sure that the system environment variables are properly set
  QProcessEnvironment env = QProcessEnvironment::systemEnvironment();
#if defined(Q_OS_DARWIN)
  env.insert("DYLD_LIBRARY_PATH", env.value("ROOTSYS") + "/lib:/usr/local/lib:" + env.value("DYLD_LIBRARY_PATH"));
#else
  env.insert("LD_LIBRARY_PATH", env.value("ROOTSYS") + "/lib:" + env.value("LD_LIBRARY_PATH"));
#endif
  fProc->setProcessEnvironment(env);
  fProc->setWorkingDirectory(workingDirectory);
 
  // Set up the command and arguments.
  QString program = cmd[0];
  QStringList arguments;
  for (int i=1; i<cmd.size(); i++)
    arguments << cmd[i];
 
  connect( fProc.get(), SIGNAL( readyReadStandardOutput() ), this, SLOT( readFromStdOut() ) );
  connect( fProc.get(), SIGNAL( readyReadStandardError() ), this, SLOT( readFromStdErr() ) );
  connect( fProc.get(), SIGNAL( finished(int, QProcess::ExitStatus) ),   this, SLOT( processDone(int, QProcess::ExitStatus) ) );
 
  fProc->start(program, arguments);
  if ( !fProc->waitForStarted() ) {
    // error handling
    QString msg(tr("Could not execute the output command: ")+cmd[0]);
    QMessageBox::critical( nullptr, tr("FATAL ERROR"), msg, QMessageBox::Close );
    done(0);
  }
  fProcPID = fProc->processId();
}
 
//----------------------------------------------------------------------------------------------------
PFitOutputHandler::~PFitOutputHandler()
{
  if (fProc->state() == QProcess::Running) {
    fProc->terminate();
    if (!fProc->waitForFinished()) {
      qDebug() << "fProc still running, will call kill." << Qt::endl;
      fProc->kill();
    }
    fProc->waitForFinished();
  }
  if (fProc->state() == QProcess::Running) {
    QString cmd = "kill -9 "+ QString("%1").arg(fProcPID);
    QString msg = "fProc still running even after Qt kill, will try system kill cmd: "+cmd;
    qDebug() << msg << Qt::endl;
    system(cmd.toLatin1());
  }
}
 
//----------------------------------------------------------------------------------------------------
void PFitOutputHandler::readFromStdOut()
{
  // Read and process the data.
  // Bear in mind that the data might be output in chunks.
  fOutput->appendPlainText( fProc->readAllStandardOutput() );
}
 
//----------------------------------------------------------------------------------------------------
void PFitOutputHandler::readFromStdErr()
{
  // Read and process the data.
  // Bear in mind that the data might be output in chunks.
  fOutput->appendPlainText( fProc->readAllStandardError() );
}
 
//----------------------------------------------------------------------------------------------------
void PFitOutputHandler::processDone(int exitCode, QProcess::ExitStatus exitStatus)
{
  if ((exitStatus == QProcess::CrashExit) && (exitCode != 0)) {
    qDebug() << "**ERROR** PFitOutputHandler::processDone: exitCode = " << exitCode << Qt::endl;
  }
  fQuitButton->setText("Done");
}
 
//----------------------------------------------------------------------------------------------------
void PFitOutputHandler::quitButtonPressed()
{
  // if the fitting is still taking place, kill it
  if (fProc->state() == QProcess::Running) {
    fProc->terminate();
    if (!fProc->waitForFinished()) {
      fProc->kill();
    }
  }
 
  accept();
}
 
//----------------------------------------------------------------------------------------------------
// END
//----------------------------------------------------------------------------------------------------