ClearUB.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 "MantidCrystal/ClearUB.h"
#include "MantidAPI/ExperimentInfo.h"
#include "MantidAPI/MultipleExperimentInfos.h"
#include "MantidAPI/Sample.h"
#include "MantidAPI/Workspace.h"
 
using namespace Mantid::Kernel;
using namespace Mantid::API;
 
namespace Mantid::Crystal {
 
// Register the algorithm into the AlgorithmFactory
DECLARE_ALGORITHM(ClearUB)
 
 
const std::string ClearUB::name() const { return "ClearUB"; }
 
int ClearUB::version() const { return 1; }
 
const std::string ClearUB::category() const { return "Crystal\\UBMatrix"; }
 
void ClearUB::init() {
  declareProperty(std::make_unique<WorkspaceProperty<Workspace>>("Workspace", "", Direction::InOut),
                  "Workspace to clear the UB from.");
  declareProperty(std::make_unique<PropertyWithValue<bool>>("DoesClear", false, Direction::Output),
                  "Indicates action performed. DoesClear returns true only if one or more "
                  "OrientedLattices have been removed.");
}
 
bool ClearUB::clearSingleExperimentInfo(ExperimentInfo *const experimentInfo, const bool dryRun) const {
  bool doesClear = false;
  Sample &sampleObject = experimentInfo->mutableSample();
  if (!sampleObject.hasOrientedLattice()) {
    this->g_log.notice("Experiment Info has no oriented lattice.");
  } else {
    this->g_log.notice("Experiment Info has an oriented lattice.");
    // Only actually clear the orientedlattice if this is NOT a dry run.
    if (!dryRun) {
      sampleObject.clearOrientedLattice();
    }
    doesClear = true;
  }
  return doesClear;
}
 
bool ClearUB::doExecute(Workspace *const ws, bool dryRun) {
  bool doesClear = false;
  auto *experimentInfo = dynamic_cast<ExperimentInfo *>(ws);
  if (experimentInfo) {
    doesClear = clearSingleExperimentInfo(experimentInfo, dryRun);
  } else {
    auto *experimentInfos = dynamic_cast<MultipleExperimentInfos *>(ws);
    if (!experimentInfos) {
      if (!dryRun) {
        throw std::invalid_argument("Input workspace is neither of type "
                                    "ExperimentInfo or MultipleExperimentInfo, "
                                    "cannot process.");
      }
    } else {
      const uint16_t nInfos = experimentInfos->getNumExperimentInfo();
      for (uint16_t i = 0; i < nInfos; ++i) {
        ExperimentInfo_sptr info = experimentInfos->getExperimentInfo(i);
        doesClear = clearSingleExperimentInfo(info.get(), dryRun) || doesClear;
      }
    }
  }
  return doesClear;
}
 
//----------------------------------------------------------------------------------------------
void ClearUB::exec() {
  Workspace_sptr ws = getProperty("Workspace");
  bool doesClear = doExecute(ws.get(), false /* Not a dry run*/);
  this->setProperty("DoesClear", doesClear);
}
 
} // namespace Mantid::Crystal