ConvertToYSpace.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 "MantidCurveFitting/Algorithms/ConvertToYSpace.h"
 
#include "MantidAPI/Axis.h"
#include "MantidAPI/HistogramValidator.h"
#include "MantidAPI/InstrumentValidator.h"
#include "MantidAPI/MatrixWorkspace.h"
#include "MantidAPI/SpectrumInfo.h"
#include "MantidAPI/WorkspaceFactory.h"
#include "MantidAPI/WorkspaceUnitValidator.h"
#include "MantidGeometry/Instrument.h"
#include "MantidGeometry/Instrument/ComponentInfo.h"
#include "MantidGeometry/Instrument/DetectorGroup.h"
#include "MantidGeometry/Instrument/ParameterMap.h"
#include "MantidKernel/BoundedValidator.h"
#include "MantidKernel/CompositeValidator.h"
#include "MantidKernel/PhysicalConstants.h"
#include "MantidKernel/Unit.h"
#include "MantidTypes/SpectrumDefinition.h"
 
namespace Mantid::CurveFitting::Algorithms {
 
// Register the algorithm into the AlgorithmFactory
DECLARE_ALGORITHM(ConvertToYSpace)
 
using namespace API;
using namespace Kernel;
 
namespace {
const double MASS_TO_MEV = 0.5 * PhysicalConstants::NeutronMass / PhysicalConstants::meV;
} // namespace
 
ConvertToYSpace::ConvertToYSpace()
    : Algorithm(), m_inputWS(), m_mass(0.0), m_l1(0.0), m_samplePos(), m_outputWS(), m_qOutputWS() {}
 
const std::string ConvertToYSpace::name() const { return "ConvertToYSpace"; }
 
int ConvertToYSpace::version() const { return 1; }
 
const std::string ConvertToYSpace::category() const { return "Transforms\\Units"; }
 
DetectorParams ConvertToYSpace::getDetectorParameters(const API::MatrixWorkspace_const_sptr &ws, const size_t index) {
  auto inst = ws->getInstrument();
  auto sample = inst->getSample();
  auto source = inst->getSource();
  if (!sample || !source) {
    throw std::invalid_argument("ConvertToYSpace - Workspace has no source/sample.");
  }
 
  const auto &spectrumInfo = ws->spectrumInfo();
  if (!spectrumInfo.hasDetectors(index))
    throw std::invalid_argument("ConvertToYSpace - Workspace has no detector "
                                "attached to histogram at index " +
                                std::to_string(index));
 
  DetectorParams detpar;
  const auto &componentInfo = ws->componentInfo();
  const auto &spectrumDefinition = spectrumInfo.spectrumDefinition(index);
  detpar.l1 = spectrumInfo.l1();
  detpar.l2 = spectrumInfo.l2(index);
  detpar.pos = spectrumInfo.position(index);
  detpar.theta = spectrumInfo.twoTheta(index);
  detpar.t0 = getComponentParameter(componentInfo, spectrumDefinition, "t0") * 1e-6; // Convert to seconds
  detpar.efixed = getComponentParameter(componentInfo, spectrumDefinition, "efixed");
  return detpar;
}
 
double ConvertToYSpace::getComponentParameter(const Geometry::IComponent &comp, const Geometry::ParameterMap &pmap,
                                              const std::string &name) {
  double result(0.0);
  if (const auto &group = dynamic_cast<const Geometry::DetectorGroup *>(&comp)) {
    double avg(0.0);
    for (const auto &det : group->getDetectors()) {
      auto param = pmap.getRecursive(det->getComponentID(), name);
      if (param)
        avg += param->value<double>();
      else
        throw std::invalid_argument("ComptonProfile - Unable to find "
                                    "DetectorGroup component parameter \"" +
                                    name + "\".");
    }
    result = avg / static_cast<double>(group->nDets());
  } else {
    auto param = pmap.getRecursive(comp.getComponentID(), name);
    if (param) {
      result = param->value<double>();
    } else {
      throw std::invalid_argument("ComptonProfile - Unable to find component parameter \"" + name + "\".");
    }
  }
  return result;
}
 
double ConvertToYSpace::getComponentParameter(const Geometry::ComponentInfo &componentInfo,
                                              const SpectrumDefinition &spectrumDefinition, const std::string &name) {
  const bool isGroup = spectrumDefinition.size() > 1;
  double sum(0.0);
  for (const auto &specDefItem : spectrumDefinition) {
    const auto values = componentInfo.getNumberParameter(specDefItem.first, name);
    if (!values.empty())
      sum += values.front();
    else if (isGroup)
      throw std::invalid_argument("ComptonProfile - Unable to find "
                                  "DetectorGroup component parameter \"" +
                                  name + "\".");
    else
      throw std::invalid_argument("ComptonProfile - Unable to find component parameter \"" + name + "\".");
  }
  return sum / static_cast<double>(spectrumDefinition.size());
}
 
//----------------------------------------------------------------------------------------------
 
void ConvertToYSpace::calculateY(double &yspace, double &qspace, double &ei, const double mass, const double tsec,
                                 const double k1, const double v1, const DetectorParams &detpar) {
  const double v0 = detpar.l1 / (tsec - detpar.t0 - (detpar.l2 / v1));
  ei = MASS_TO_MEV * v0 * v0;
  const double w = ei - detpar.efixed;
  const double k0 = std::sqrt(ei / PhysicalConstants::E_mev_toNeutronWavenumberSq);
  qspace = std::sqrt(k0 * k0 + k1 * k1 - 2.0 * k0 * k1 * std::cos(detpar.theta));
  const double wreduced = PhysicalConstants::E_mev_toNeutronWavenumberSq * qspace * qspace / mass;
  yspace = 0.2393 * (mass / qspace) * (w - wreduced);
}
 
//----------------------------------------------------------------------------------------------
 
void ConvertToYSpace::init() {
  auto wsValidator = std::make_shared<CompositeValidator>();
  wsValidator->add<HistogramValidator>(false); // point data
  wsValidator->add<InstrumentValidator>();
  wsValidator->add<WorkspaceUnitValidator>("TOF");
  declareProperty(std::make_unique<WorkspaceProperty<>>("InputWorkspace", "", Direction::Input, wsValidator),
                  "The input workspace in Time of Flight");
 
  auto mustBePositive = std::make_shared<BoundedValidator<double>>();
  mustBePositive->setLower(0.0);
  mustBePositive->setLowerExclusive(true); // strictly greater than 0.0
  declareProperty("Mass", -1.0, mustBePositive, "The mass defining the recoil peak in AMU");
 
  declareProperty(std::make_unique<WorkspaceProperty<>>("OutputWorkspace", "", Direction::Output),
                  "The output workspace in y-Space");
 
  declareProperty(std::make_unique<WorkspaceProperty<>>("QWorkspace", "", Direction::Output, PropertyMode::Optional),
                  "The output workspace in q-Space");
}
 
//----------------------------------------------------------------------------------------------
 
void ConvertToYSpace::exec() {
  retrieveInputs();
  createOutputWorkspace();
 
  const auto nhist = static_cast<int64_t>(m_inputWS->getNumberHistograms());
  const int64_t nreports = nhist;
  auto progress = std::make_shared<Progress>(this, 0.0, 1.0, nreports);
 
  auto &spectrumInfo = m_outputWS->mutableSpectrumInfo();
  SpectrumInfo *qSpectrumInfo{nullptr};
  if (m_qOutputWS)
    qSpectrumInfo = &m_qOutputWS->mutableSpectrumInfo();
 
  PARALLEL_FOR_IF(Kernel::threadSafe(*m_inputWS, *m_outputWS))
  for (int64_t i = 0; i < nhist; ++i) {
    PARALLEL_START_INTERRUPT_REGION
 
    if (!convert(i)) {
      g_log.warning("No detector defined for index=" + std::to_string(i) + ". Zeroing spectrum.");
      m_outputWS->getSpectrum(i).clearData();
      PARALLEL_CRITICAL(setMasked) {
        spectrumInfo.setMasked(i, true);
        if (m_qOutputWS) {
          m_qOutputWS->getSpectrum(i).clearData();
          qSpectrumInfo->setMasked(i, true);
        }
      }
    }
    progress->report();
    PARALLEL_END_INTERRUPT_REGION
  }
  PARALLEL_CHECK_INTERRUPT_REGION
 
  setProperty("OutputWorkspace", m_outputWS);
 
  if (m_qOutputWS)
    setProperty("QWorkspace", m_qOutputWS);
}
 
bool ConvertToYSpace::convert(const size_t index) {
  try {
    DetectorParams detPar = getDetectorParameters(m_inputWS, index);
    const double v1 = std::sqrt(detPar.efixed / MASS_TO_MEV);
    const double k1 = std::sqrt(detPar.efixed / PhysicalConstants::E_mev_toNeutronWavenumberSq);
 
    auto &outX = m_outputWS->mutableX(index);
    auto &outY = m_outputWS->mutableY(index);
    auto &outE = m_outputWS->mutableE(index);
    const auto &inX = m_inputWS->x(index);
    const auto &inY = m_inputWS->y(index);
    const auto &inE = m_inputWS->e(index);
 
    // The t->y mapping flips the order of the axis so we need to reverse it to
    // have a monotonically increasing axis
    const size_t npts = inY.size();
    for (size_t j = 0; j < npts; ++j) {
      double ys(0.0), qs(0.0), ei(0.0);
      calculateY(ys, qs, ei, m_mass, inX[j] * 1e-06, k1, v1, detPar);
      const size_t outIndex = (npts - j - 1);
      outX[outIndex] = ys;
      const double prefactor = qs / pow(ei, 0.1);
      outY[outIndex] = prefactor * inY[j];
      outE[outIndex] = prefactor * inE[j];
 
      if (m_qOutputWS) {
        m_qOutputWS->mutableX(index)[outIndex] = ys;
        m_qOutputWS->mutableY(index)[outIndex] = qs;
      }
    }
    return true;
  } catch (Exception::NotFoundError &) {
    return false;
  }
}
 
void ConvertToYSpace::retrieveInputs() {
  m_inputWS = getProperty("InputWorkspace");
  m_mass = getProperty("Mass");
  cacheInstrumentGeometry();
}
 
void ConvertToYSpace::createOutputWorkspace() {
  // y-Space output workspace
  m_outputWS = WorkspaceFactory::Instance().create(m_inputWS);
 
  auto xLabel = std::make_shared<Units::Label>("Momentum", "A^-1");
  m_outputWS->getAxis(0)->unit() = xLabel;
  m_outputWS->setYUnit("");
  m_outputWS->setYUnitLabel("");
 
  // q-Space output workspace
  if (!getPropertyValue("QWorkspace").empty()) {
    m_qOutputWS = WorkspaceFactory::Instance().create(m_inputWS);
 
    m_qOutputWS->getAxis(0)->unit() = xLabel;
    m_qOutputWS->setYUnit("");
    m_qOutputWS->setYUnitLabel("");
  }
}
 
void ConvertToYSpace::cacheInstrumentGeometry() {
  auto inst = m_inputWS->getInstrument();
  auto source = inst->getSource();
  auto sample = inst->getSample();
  m_l1 = sample->getDistance(*source);
  m_samplePos = sample->getPos();
}
 
} // namespace Mantid::CurveFitting::Algorithms