ParallelProcessRunner.h

Go to the documentation of this file.

/*$!{
* Aurora Clang AST Viewer (ACAV)
* 
* Copyright (c) 2026 Min Liu
* Copyright (c) 2026 Michael David Adams
* 
* SPDX-License-Identifier: GPL-2.0-or-later
* 
* 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, see <https://www.gnu.org/licenses/>.
}$!*/

#pragma once
 
#include <QElapsedTimer>
#include <QHash>
#include <QObject>
#include <QProcess>
#include <QString>
#include <QStringList>
#include <QTemporaryDir>
#include <vector>
 
#include "core/LogEntry.h"
 
namespace acav {

class ParallelProcessRunner : public QObject {
  Q_OBJECT
 
public:
  explicit ParallelProcessRunner(QObject *parent = nullptr);
  ~ParallelProcessRunner() override;

  bool isRunning() const;

  void setParallelCount(int count);

  int getParallelCount() const { return parallelCount_; }

  void cancel();
 
signals:
  void error(const QString &errorMessage);

  void progress(const QString &message);

  void chunkCompleted(int chunkIndex, int totalChunks);

  void chunkFailed(int chunkIndex, const QString &errorMessage);

  void logMessage(const LogEntry &entry);
 
protected:
  virtual std::vector<QStringList> chunkInputData(
      const QStringList &inputData,
      int chunkCount) const;

  virtual QStringList prepareProcessArguments(
      int chunkIndex,
      const QStringList &chunkData,
      const QString &tempOutputPath) = 0;

  virtual bool mergeResults(
      const QStringList &tempOutputPaths,
      const QString &finalOutputPath,
      QString &errorMessage) = 0;

  virtual void onAllCompleted(int successCount, int failureCount, int totalCount);

  QString getTempOutputPath(int chunkIndex) const;

  void runParallel(
      const QString &programPath,
      const QStringList &inputData,
      const QString &finalOutputPath);
 
  // Protected members accessible to subclasses
  QString finalOutputPath_;       
  QStringList errorMessages_;     
  QTemporaryDir tempDir_;         
  QElapsedTimer elapsed_;
 
private slots:
  void onProcessFinished(int exitCode, QProcess::ExitStatus exitStatus);
  void onProcessError(QProcess::ProcessError error);
  void onProcessStdOut();
  void onProcessStdErr();
 
private:
  QString processSource(QProcess *process) const;
  void checkAllProcessesCompleted();
 
  // Configuration
  QString programPath_;
  int parallelCount_;  // Number of parallel processes (0 = auto-detect)
 
  // Process management
  std::vector<QProcess*> processes_;
  QStringList tempOutputPaths_;
  int completedProcessCount_;
  int failedProcessCount_;
  QHash<QProcess*, QString> pendingStdout_;
  QHash<QProcess*, QString> pendingStderr_;
};
 
} // namespace acav