BatchAlgorithmRunner.cpp

Go to the documentation of this file.

// Mantid Repository : https://github.com/mantidproject/mantid
//
// Copyright © 2018 ISIS Rutherford Appleton Laboratory UKRI,
//   NScD Oak Ridge National Laboratory, European Spallation Source,
//   Institut Laue - Langevin & CSNS, Institute of High Energy Physics, CAS
// SPDX - License - Identifier: GPL - 3.0 +
 
#include "MantidQtWidgets/Common/BatchAlgorithmRunner.h"
#include "MantidQtWidgets/Common/AlgorithmRuntimeProps.h"
#include "MantidQtWidgets/Common/ConfiguredAlgorithm.h"
#include "MantidQtWidgets/Common/IAlgorithmRuntimeProps.h"
 
#include "MantidAPI/AlgorithmManager.h"
#include "MantidKernel/Exception.h"
#include "MantidKernel/Logger.h"
 
#include <memory>
#include <utility>
 
using namespace Mantid::API;
 
namespace {
Mantid::Kernel::Logger g_log("BatchAlgorithmRunner");
 
// Throw if any of the given properties do not exist in the algorithm's declared property names
void throwIfAnyPropertiesInvalid(IAlgorithm_sptr alg, MantidQt::API::IAlgorithmRuntimeProps const &props) {
  auto allowedPropNames = alg->getDeclaredPropertyNames();
  auto propNamesToUpdate = props.getDeclaredPropertyNames();
 
  // Note that for std::set_difference the lists need to be sorted
  std::sort(allowedPropNames.begin(), allowedPropNames.end());
  std::sort(propNamesToUpdate.begin(), propNamesToUpdate.end());
 
  std::vector<std::string> invalidProps;
  std::set_difference(propNamesToUpdate.cbegin(), propNamesToUpdate.cend(), allowedPropNames.cbegin(),
                      allowedPropNames.cend(), std::back_inserter(invalidProps));
 
  if (invalidProps.size() > 0) {
    const auto invalidPropsStr = std::accumulate(std::next(invalidProps.cbegin()), invalidProps.cend(), invalidProps[0],
                                                 [](const auto &a, const auto &b) { return a + "," + b; });
    throw Mantid::Kernel::Exception::NotFoundError("Invalid Properties given: ", invalidPropsStr);
  }
}
} // namespace
 
namespace MantidQt::API {
 
BatchAlgorithmRunner::BatchAlgorithmRunner(QObject *parent)
    : QObject(parent), m_stopOnFailure(true), m_cancelRequested(false), m_notificationCenter(),
      m_batchCompleteObserver(*this, &BatchAlgorithmRunner::handleBatchComplete),
      m_batchCancelledObserver(*this, &BatchAlgorithmRunner::handleBatchCancelled),
      m_algorithmStartedObserver(*this, &BatchAlgorithmRunner::handleAlgorithmStarted),
      m_algorithmCompleteObserver(*this, &BatchAlgorithmRunner::handleAlgorithmComplete),
      m_algorithmErrorObserver(*this, &BatchAlgorithmRunner::handleAlgorithmError),
      m_executeAsync(this, &BatchAlgorithmRunner::executeBatchAsyncImpl) {}
 
BatchAlgorithmRunner::~BatchAlgorithmRunner() { removeAllObservers(); }
 
void BatchAlgorithmRunner::addAllObservers() {
  m_notificationCenter.addObserver(m_batchCompleteObserver);
  m_notificationCenter.addObserver(m_batchCancelledObserver);
  m_notificationCenter.addObserver(m_algorithmStartedObserver);
  m_notificationCenter.addObserver(m_algorithmCompleteObserver);
  m_notificationCenter.addObserver(m_algorithmErrorObserver);
}
 
void BatchAlgorithmRunner::removeAllObservers() {
  m_notificationCenter.removeObserver(m_batchCompleteObserver);
  m_notificationCenter.removeObserver(m_batchCancelledObserver);
  m_notificationCenter.removeObserver(m_algorithmStartedObserver);
  m_notificationCenter.removeObserver(m_algorithmCompleteObserver);
  m_notificationCenter.removeObserver(m_algorithmErrorObserver);
}
 
void BatchAlgorithmRunner::stopOnFailure(bool stopOnFailure) { m_stopOnFailure = stopOnFailure; }
 
void BatchAlgorithmRunner::addAlgorithm(const IAlgorithm_sptr &algo) {
  this->addAlgorithm(algo, std::make_unique<AlgorithmRuntimeProps>());
}
 
void BatchAlgorithmRunner::addAlgorithm(const IAlgorithm_sptr &algo, std::unique_ptr<IAlgorithmRuntimeProps> props) {
  m_algorithms.emplace_back(std::make_unique<ConfiguredAlgorithm>(algo, std::move(props)));
 
  g_log.debug() << "Added algorithm \"" << m_algorithms.back()->algorithm()->name() << "\" to batch queue\n";
}
 
void BatchAlgorithmRunner::setQueue(std::deque<IConfiguredAlgorithm_sptr> algorithms) {
  g_log.debug() << "Set batch queue to algorithm list:\n";
  for (auto &algorithm : algorithms)
    g_log.debug() << algorithm->algorithm()->name() << "\n";
 
  m_algorithms = std::move(algorithms);
}
 
void BatchAlgorithmRunner::clearQueue() { m_algorithms.clear(); }
 
size_t BatchAlgorithmRunner::queueLength() { return m_algorithms.size(); }
 
bool BatchAlgorithmRunner::executeBatch() {
  addAllObservers();
  Poco::ActiveResult<bool> result = m_executeAsync(Poco::Void());
  result.wait();
  removeAllObservers();
  return result.data();
}
 
void BatchAlgorithmRunner::executeBatchAsync() {
  addAllObservers();
  m_executeAsync(Poco::Void());
}
 
void BatchAlgorithmRunner::cancelBatch() {
  std::lock_guard<std::mutex> lock(m_mutex);
  // If the queue is empty, notify straight away that the batch has been
  // cancelled. Otherwise, set a flag so that it will be cancelled after the
  // current algorithm finishes processing
  if (queueLength() < 1) {
    addAllObservers();
    m_notificationCenter.postNotification(new BatchCancelledNotification());
    removeAllObservers();
  } else {
    m_cancelRequested = true;
  }
}
 
void BatchAlgorithmRunner::resetState() {
  std::lock_guard<std::mutex> lock(m_mutex);
  removeAllObservers();
  clearQueue();
  m_cancelRequested = false;
}
 
bool BatchAlgorithmRunner::cancelRequested() {
  std::lock_guard<std::mutex> lock(m_mutex);
  return m_cancelRequested;
}
 
bool BatchAlgorithmRunner::executeBatchAsyncImpl(const Poco::Void & /*unused*/) {
  bool errorFlag = false;
 
  for (auto &it : m_algorithms) {
    if (cancelRequested()) {
      g_log.information("Stopping batch algorithm execution: cancelled");
      break;
    }
 
    // Try to execute the algorithm
    if (!executeAlgo(it)) {
      g_log.warning() << "Got error from algorithm \"" << m_currentAlgorithm->name() << "\"\n";
 
      // Stop executing the entire batch if appropriate
      if (m_stopOnFailure) {
        g_log.warning("Stopping batch algorithm because of execution error");
        errorFlag = true;
        break;
      }
    } else {
      g_log.information() << "Algorithm \"" << m_currentAlgorithm->name() << "\" finished\n";
    }
  }
 
  // Notify observers
  if (cancelRequested())
    m_notificationCenter.postNotification(new BatchCancelledNotification());
  else
    m_notificationCenter.postNotification(new BatchCompleteNotification(false, errorFlag));
 
  resetState();
 
  return !errorFlag;
}
 
bool BatchAlgorithmRunner::executeAlgo(const IConfiguredAlgorithm_sptr &algorithm) {
  try {
    m_currentAlgorithm = algorithm->algorithm();
    auto const &props = algorithm->getAlgorithmRuntimeProps();
    throwIfAnyPropertiesInvalid(m_currentAlgorithm, props);
 
    // Assign the properties to be set at runtime
    m_currentAlgorithm->updatePropertyValues(props);
 
    g_log.information() << "Starting next algorithm in queue: " << m_currentAlgorithm->name() << "\n";
 
    // Start algorithm running
    m_notificationCenter.postNotification(new AlgorithmStartedNotification(algorithm));
    auto result = m_currentAlgorithm->execute();
 
    if (!result) {
      auto message = std::string("Algorithm") + algorithm->algorithm()->name() + std::string(" execution failed");
      m_notificationCenter.postNotification(new AlgorithmErrorNotification(algorithm, message));
    } else {
      m_notificationCenter.postNotification(new AlgorithmCompleteNotification(algorithm));
    }
 
    return result;
  }
  // If a property name was given that does not match a property
  catch (Mantid::Kernel::Exception::NotFoundError &notFoundEx) {
    UNUSED_ARG(notFoundEx);
    g_log.warning("Algorithm property does not exist.\nStopping queue execution.");
    m_notificationCenter.postNotification(new AlgorithmErrorNotification(algorithm, notFoundEx.what()));
    return false;
  }
  // If a property was assigned a value of the wrong type
  catch (std::invalid_argument &invalidArgEx) {
    UNUSED_ARG(invalidArgEx);
    g_log.warning("Algorithm property given value of incorrect type.\nStopping "
                  "queue execution.");
    m_notificationCenter.postNotification(new AlgorithmErrorNotification(algorithm, invalidArgEx.what()));
    return false;
  }
  // For anything else that could go wrong
  catch (std::exception &ex) {
    g_log.warning("Error starting batch algorithm");
    m_notificationCenter.postNotification(new AlgorithmErrorNotification(algorithm, ex.what()));
    return false;
  } catch (...) {
    g_log.warning("Unknown error starting next batch algorithm");
    m_notificationCenter.postNotification(
        new AlgorithmErrorNotification(algorithm, "Unknown error starting algorithm"));
    return false;
  }
}
 
void BatchAlgorithmRunner::handleBatchComplete(const Poco::AutoPtr<BatchCompleteNotification> &pNf) {
  bool inProgress = pNf->isInProgress();
  if (!inProgress) {
    // Notify UI elements
    emit batchComplete(pNf->hasError());
  }
}
 
void BatchAlgorithmRunner::handleBatchCancelled(const Poco::AutoPtr<BatchCancelledNotification> &pNf) {
  UNUSED_ARG(pNf);
  // Notify UI elements
  emit batchCancelled();
}
 
void BatchAlgorithmRunner::handleAlgorithmStarted(const Poco::AutoPtr<AlgorithmStartedNotification> &pNf) {
  // Notify UI elements
  emit algorithmStarted(pNf->algorithm());
}
 
void BatchAlgorithmRunner::handleAlgorithmComplete(const Poco::AutoPtr<AlgorithmCompleteNotification> &pNf) {
  // Notify UI elements
  emit algorithmComplete(pNf->algorithm());
}
 
void BatchAlgorithmRunner::handleAlgorithmError(const Poco::AutoPtr<AlgorithmErrorNotification> &pNf) {
  auto errorMessage = pNf->errorMessage();
  // Notify UI elements
  emit algorithmError(pNf->algorithm(), errorMessage);
}
} // namespace MantidQt::API