ReflectometryTransform.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 "MantidDataObjects/ReflectometryTransform.h"
 
#include "MantidAPI/BinEdgeAxis.h"
#include "MantidAPI/MatrixWorkspace.h"
#include "MantidAPI/SpectrumDetectorMapping.h"
#include "MantidAPI/SpectrumInfo.h"
#include "MantidAPI/TableRow.h"
#include "MantidAPI/WorkspaceFactory.h"
#include "MantidDataObjects/CalculateReflectometry.h"
#include "MantidDataObjects/FractionalRebinning.h"
#include "MantidDataObjects/RebinnedOutput.h"
#include "MantidDataObjects/TableWorkspace.h"
#include "MantidDataObjects/Workspace2D.h"
#include "MantidGeometry/Crystal/AngleUnits.h"
#include "MantidGeometry/Instrument.h"
#include "MantidGeometry/Instrument/ComponentInfo.h"
#include "MantidGeometry/Instrument/DetectorGroup.h"
#include "MantidGeometry/Instrument/DetectorInfo.h"
#include "MantidGeometry/Instrument/ReferenceFrame.h"
#include "MantidGeometry/MDGeometry/MDHistoDimension.h"
#include "MantidGeometry/Objects/BoundingBox.h"
#include "MantidGeometry/Objects/IObject.h"
#include "MantidKernel/Unit.h"
#include "MantidKernel/UnitFactory.h"
#include "MantidKernel/V2D.h"
#include "MantidKernel/VectorHelper.h"
 
#include <memory>
#include <utility>
 
using namespace Mantid::API;
using namespace Mantid::Geometry;
using namespace Mantid::Kernel;
 
namespace {
void writeRow(std::shared_ptr<Mantid::DataObjects::TableWorkspace> &vertexes, const V2D &vertex, size_t nHisto,
              size_t nBins, double signal, double error) {
  TableRow row = vertexes->appendRow();
  row << vertex.X() << vertex.Y() << int(nHisto) << int(nBins) << signal << error;
}
void addColumnHeadings(Mantid::DataObjects::TableWorkspace &vertexes, const std::string &outputDimensions) {
 
  if (outputDimensions == "Q (lab frame)") {
    vertexes.addColumn("double", "Qx");
    vertexes.addColumn("double", "Qy");
    vertexes.addColumn("int", "OriginIndex");
    vertexes.addColumn("int", "OriginBin");
    vertexes.addColumn("double", "CellSignal");
    vertexes.addColumn("double", "CellError");
  }
  if (outputDimensions == "P (lab frame)") {
    vertexes.addColumn("double", "Pi+Pf");
    vertexes.addColumn("double", "Pi-Pf");
    vertexes.addColumn("int", "OriginIndex");
    vertexes.addColumn("int", "OriginBin");
    vertexes.addColumn("double", "CellSignal");
    vertexes.addColumn("double", "CellError");
  }
  if (outputDimensions == "K (incident, final)") {
    vertexes.addColumn("double", "Ki");
    vertexes.addColumn("double", "Kf");
    vertexes.addColumn("int", "OriginIndex");
    vertexes.addColumn("int", "OriginBin");
    vertexes.addColumn("double", "CellSignal");
    vertexes.addColumn("double", "CellError");
  }
}
} // namespace
namespace Mantid::DataObjects {
 
ReflectometryTransform::ReflectometryTransform(std::string d0Label, std::string d0ID, double d0Min, double d0Max,
                                               std::string d1Label, std::string d1ID, double d1Min, double d1Max,
                                               size_t d0NumBins, size_t d1NumBins, CalculateReflectometry *calc)
    : m_d0NumBins(d0NumBins), m_d1NumBins(d1NumBins), m_d0Min(d0Min), m_d1Min(d1Min), m_d0Max(d0Max), m_d1Max(d1Max),
      m_d0Label(std::move(d0Label)), m_d1Label(std::move(d1Label)), m_d0ID(std::move(d0ID)), m_d1ID(std::move(d1ID)),
      m_calculator(calc) {
  if (d0Min >= d0Max || d1Min >= d1Max)
    throw std::invalid_argument("The supplied minimum values must be less than the maximum values.");
}
 
std::shared_ptr<MDEventWorkspace2Lean>
ReflectometryTransform::createMDWorkspace(const Mantid::Geometry::IMDDimension_sptr &a,
                                          const Mantid::Geometry::IMDDimension_sptr &b,
                                          const BoxController_sptr &boxController) const {
  auto ws = std::make_shared<MDEventWorkspace2Lean>();
 
  ws->addDimension(a);
  ws->addDimension(b);
 
  BoxController_sptr wsbc = ws->getBoxController(); // Get the box controller
  wsbc->setSplitInto(boxController->getSplitInto(0));
  wsbc->setMaxDepth(boxController->getMaxDepth());
  wsbc->setSplitThreshold(boxController->getSplitThreshold());
 
  // Initialize the workspace.
  ws->initialize();
 
  // Start with a MDGridBox.
  ws->splitBox();
  return ws;
}
 
MantidVec createXAxis(MatrixWorkspace *const ws, const double gradX, const double cxToUnit, const size_t nBins,
                      const std::string &caption, const std::string &units) {
  // Create an X - Axis.
  auto xAxis = std::make_unique<BinEdgeAxis>(nBins);
  auto xAxisRaw = xAxis.get();
  ws->replaceAxis(0, std::move(xAxis));
  auto unitXBasePtr = UnitFactory::Instance().create("Label");
  std::shared_ptr<Mantid::Kernel::Units::Label> xUnit =
      std::dynamic_pointer_cast<Mantid::Kernel::Units::Label>(unitXBasePtr);
  xUnit->setLabel(caption, units);
  xAxisRaw->unit() = xUnit;
  xAxisRaw->title() = caption;
  MantidVec xAxisVec(nBins);
  for (size_t i = 0; i < nBins; ++i) {
    double qxIncrement = ((1 / gradX) * (static_cast<double>(i) + 1) + cxToUnit);
    xAxisRaw->setValue(i, qxIncrement);
    xAxisVec[i] = qxIncrement;
  }
  return xAxisVec;
}
 
void createVerticalAxis(MatrixWorkspace *const ws, const MantidVec &xAxisVec, const double gradY, const double cyToUnit,
                        const size_t nBins, const std::string &caption, const std::string &units) {
  // Create a Y (vertical) Axis
  auto verticalAxis = std::make_unique<BinEdgeAxis>(nBins);
  auto verticalAxisRaw = verticalAxis.get();
  ws->replaceAxis(1, std::move(verticalAxis));
  auto unitZBasePtr = UnitFactory::Instance().create("Label");
  std::shared_ptr<Mantid::Kernel::Units::Label> verticalUnit =
      std::dynamic_pointer_cast<Mantid::Kernel::Units::Label>(unitZBasePtr);
  verticalAxisRaw->unit() = verticalUnit;
  verticalUnit->setLabel(caption, units);
  verticalAxisRaw->title() = caption;
  auto xAxis = Kernel::make_cow<HistogramData::HistogramX>(xAxisVec);
  for (size_t i = 0; i < nBins; ++i) {
    ws->setSharedX(i, xAxis);
    double qzIncrement = ((1 / gradY) * (static_cast<double>(i) + 1) + cyToUnit);
    verticalAxisRaw->setValue(i, qzIncrement);
  }
}
 
DetectorAngularCache initAngularCaches(const MatrixWorkspace *const workspace) {
  const size_t nhist = workspace->getNumberHistograms();
  std::vector<double> twoThetas(nhist);
  std::vector<double> twoThetaWidths(nhist);
  std::vector<double> detectorHeights(nhist);
 
  auto inst = workspace->getInstrument();
  const V3D upDirVec = inst->getReferenceFrame()->vecPointingUp();
 
  const auto &spectrumInfo = workspace->spectrumInfo();
  const auto &detectorInfo = workspace->detectorInfo();
  const auto &componentInfo = workspace->componentInfo();
  for (size_t i = 0; i < nhist; ++i) {
    if (!spectrumInfo.hasDetectors(i) || spectrumInfo.isMonitor(i)) {
      // If no detector found, skip onto the next spectrum
      twoThetas[i] = -1.0; // Indicates a detector to skip
      twoThetaWidths[i] = -1.0;
      continue;
    }
 
    // We have to convert twoTheta from radians to degrees
    const double twoTheta = spectrumInfo.signedTwoTheta(i) * rad2deg;
    twoThetas[i] = twoTheta;
    // If the spectrum is based on a group of detectors assume they all have
    // same shape and same r,twoTheta
    // Use the first detector of the spectrum.
    size_t detIndex = spectrumInfo.firstDetectorIndex(i);
    double l2 = detectorInfo.l2(detIndex);
    // Get the shape
    // Get the shape, defined in its own reference frame with centre at 0,0,0
    BoundingBox bbox = componentInfo.shape(detIndex).getBoundingBox();
    auto maxPoint(bbox.maxPoint());
    auto minPoint(bbox.minPoint());
    auto span = maxPoint - minPoint;
    detectorHeights[i] = span.scalar_prod(upDirVec);
    twoThetaWidths[i] =
        (std::atan(maxPoint.scalar_prod(upDirVec) / l2) - std::atan(minPoint.scalar_prod(upDirVec) / l2)) * rad2deg;
  }
  DetectorAngularCache cache;
  cache.twoThetas = twoThetas;
  cache.twoThetaWidths = twoThetaWidths;
  cache.detectorHeights = detectorHeights;
  return cache;
}
 
Mantid::API::IMDEventWorkspace_sptr
ReflectometryTransform::executeMD(const Mantid::API::MatrixWorkspace_const_sptr &inputWs,
                                  const BoxController_sptr &boxController, Mantid::Geometry::MDFrame_uptr frame) const {
  auto dim0 = std::make_shared<MDHistoDimension>(m_d0Label, m_d0ID, *frame, static_cast<Mantid::coord_t>(m_d0Min),
                                                 static_cast<Mantid::coord_t>(m_d0Max), m_d0NumBins);
  auto dim1 = std::make_shared<MDHistoDimension>(m_d1Label, m_d1ID, *frame, static_cast<Mantid::coord_t>(m_d1Min),
                                                 static_cast<Mantid::coord_t>(m_d1Max), m_d1NumBins);
 
  auto ws = createMDWorkspace(dim0, dim1, boxController);
 
  auto spectraAxis = inputWs->getAxis(1);
  for (size_t index = 0; index < inputWs->getNumberHistograms(); ++index) {
    auto counts = inputWs->y(index);
    auto wavelengths = inputWs->x(index);
    auto errors = inputWs->e(index);
    const size_t nInputBins = wavelengths.size() - 1;
    const double twoTheta = spectraAxis->getValue(index);
    m_calculator->setTwoTheta(twoTheta);
    // Loop over all bins in spectra
    for (size_t binIndex = 0; binIndex < nInputBins; ++binIndex) {
      const double &wavelength = 0.5 * (wavelengths[binIndex] + wavelengths[binIndex + 1]);
      double _d0 = m_calculator->calculateDim0(wavelength);
      double _d1 = m_calculator->calculateDim1(wavelength);
      double centers[2] = {_d0, _d1};
 
      ws->addEvent(MDLeanEvent<2>(float(counts[binIndex]), float(errors[binIndex] * errors[binIndex]), centers));
    }
  }
  ws->splitAllIfNeeded(nullptr);
  ws->refreshCache();
  return ws;
}
 
Mantid::API::MatrixWorkspace_sptr
ReflectometryTransform::execute(const Mantid::API::MatrixWorkspace_const_sptr &inputWs) const {
  auto ws = std::make_shared<Mantid::DataObjects::Workspace2D>();
 
  ws->initialize(m_d1NumBins, m_d0NumBins,
                 m_d0NumBins); // Create the output workspace as a distribution
 
  // Mapping so that d0 and d1 values calculated can be added to the matrix
  // workspace at the correct index.
  const double gradD0 = double(m_d0NumBins) / (m_d0Max - m_d0Min); // The x - axis
  const double gradD1 = double(m_d1NumBins) / (m_d1Max - m_d1Min); // Actually the y-axis
  const double cxToIndex = -gradD0 * m_d0Min;
  const double cyToIndex = -gradD1 * m_d1Min;
  const double cxToD0 = m_d0Min - (1 / gradD0);
  const double cyToD1 = m_d1Min - (1 / gradD1);
 
  // Create an X - Axis.
  MantidVec xAxisVec = createXAxis(ws.get(), gradD0, cxToD0, m_d0NumBins, m_d0Label, "1/Angstroms");
  // Create a Y (vertical) Axis
  createVerticalAxis(ws.get(), xAxisVec, gradD1, cyToD1, m_d1NumBins, m_d1Label, "1/Angstroms");
 
  // Loop over all entries in the input workspace and calculate d0 and d1
  // for each.
  auto spectraAxis = inputWs->getAxis(1);
  for (size_t index = 0; index < inputWs->getNumberHistograms(); ++index) {
    auto counts = inputWs->y(index);
    auto wavelengths = inputWs->x(index);
    auto errors = inputWs->e(index);
    const size_t nInputBins = wavelengths.size() - 1;
    const double twoTheta = spectraAxis->getValue(index);
    m_calculator->setTwoTheta(twoTheta);
    // Loop over all bins in spectra
    for (size_t binIndex = 0; binIndex < nInputBins; ++binIndex) {
      const double wavelength = 0.5 * (wavelengths[binIndex] + wavelengths[binIndex + 1]);
      double _d0 = m_calculator->calculateDim0(wavelength);
      double _d1 = m_calculator->calculateDim1(wavelength);
 
      if (_d0 >= m_d0Min && _d0 <= m_d0Max && _d1 >= m_d1Min &&
          _d1 <= m_d1Max) // Check that the calculated ki and kf are in range
      {
        const auto outIndexX = static_cast<int>((gradD0 * _d0) + cxToIndex);
        const auto outIndexZ = static_cast<int>((gradD1 * _d1) + cyToIndex);
 
        ws->mutableY(outIndexZ)[outIndexX] += counts[binIndex];
        ws->mutableE(outIndexZ)[outIndexX] += errors[binIndex];
      }
    }
  }
  return ws;
}
 
IMDHistoWorkspace_sptr ReflectometryTransform::executeMDNormPoly(const MatrixWorkspace_const_sptr &inputWs) const {
 
  auto input_x_dim = inputWs->getXDimension();
 
  MDHistoDimension_sptr dim0 = MDHistoDimension_sptr(
      new MDHistoDimension(input_x_dim->getName(), input_x_dim->getDimensionId(), input_x_dim->getMDFrame(),
                           static_cast<Mantid::coord_t>(input_x_dim->getMinimum()),
                           static_cast<Mantid::coord_t>(input_x_dim->getMaximum()), input_x_dim->getNBins()));
 
  auto input_y_dim = inputWs->getYDimension();
 
  MDHistoDimension_sptr dim1 = MDHistoDimension_sptr(
      new MDHistoDimension(input_y_dim->getName(), input_y_dim->getDimensionId(), input_y_dim->getMDFrame(),
                           static_cast<Mantid::coord_t>(input_y_dim->getMinimum()),
                           static_cast<Mantid::coord_t>(input_y_dim->getMaximum()), input_y_dim->getNBins()));
 
  auto outWs = std::make_shared<MDHistoWorkspace>(dim0, dim1);
 
  for (size_t nHistoIndex = 0; nHistoIndex < inputWs->getNumberHistograms(); ++nHistoIndex) {
    const auto &Y = inputWs->y(nHistoIndex);
    const auto &E = inputWs->e(nHistoIndex);
 
    const size_t numBins = Y.size();
    for (size_t nBinIndex = 0; nBinIndex < numBins; ++nBinIndex) {
      const auto value_index = outWs->getLinearIndex(nBinIndex, nHistoIndex);
      outWs->setSignalAt(value_index, Y[nBinIndex]);
      outWs->setErrorSquaredAt(value_index, E[nBinIndex] * E[nBinIndex]);
    }
  }
  return outWs;
}
 
MatrixWorkspace_sptr
ReflectometryTransform::executeNormPoly(const MatrixWorkspace_const_sptr &inputWS,
                                        std::shared_ptr<Mantid::DataObjects::TableWorkspace> &vertexes,
                                        bool dumpVertexes, const std::string &outputDimensions) const {
  MatrixWorkspace_sptr temp =
      WorkspaceFactory::Instance().create("RebinnedOutput", m_d1NumBins, m_d0NumBins + 1, m_d0NumBins);
  RebinnedOutput_sptr outWS = std::static_pointer_cast<RebinnedOutput>(temp);
 
  const double widthD0 = (m_d0Max - m_d0Min) / double(m_d0NumBins);
  const double widthD1 = (m_d1Max - m_d1Min) / double(m_d1NumBins);
 
  std::vector<double> xBinsVec;
  std::vector<double> zBinsVec;
  VectorHelper::createAxisFromRebinParams({m_d1Min, widthD1, m_d1Max}, zBinsVec);
  VectorHelper::createAxisFromRebinParams({m_d0Min, widthD0, m_d0Max}, xBinsVec);
 
  // Put the correct bin boundaries into the workspace
  auto verticalAxis = std::make_unique<BinEdgeAxis>(zBinsVec);
  auto verticalAxisRaw = verticalAxis.get();
  outWS->replaceAxis(1, std::move(verticalAxis));
  HistogramData::BinEdges binEdges(xBinsVec);
  for (size_t i = 0; i < zBinsVec.size() - 1; ++i)
    outWS->setBinEdges(i, binEdges);
 
  verticalAxisRaw->title() = m_d1Label;
 
  // Prepare the required theta values
  DetectorAngularCache cache = initAngularCaches(inputWS.get());
  auto twoThetas = cache.twoThetas;
  auto twoThetaWidths = cache.twoThetaWidths;
 
  const size_t nHistos = inputWS->getNumberHistograms();
  const size_t nBins = inputWS->blocksize();
 
  // Holds the spectrum-detector mapping
  std::vector<specnum_t> specNumberMapping;
  std::vector<detid_t> detIDMapping;
  // Create a table for the output if we want to debug vertex positioning
  addColumnHeadings(*vertexes, outputDimensions);
  const auto &spectrumInfo = inputWS->spectrumInfo();
  for (size_t i = 0; i < nHistos; ++i) {
    if (!spectrumInfo.hasDetectors(i) || spectrumInfo.isMasked(i) || spectrumInfo.isMonitor(i)) {
      continue;
    }
    // ID of the (first) detector of the spectrum
    const auto detID = spectrumInfo.firstDetectorID(i);
 
    // Compute polygon points
    const double twoTheta = twoThetas[i];
    const double twoThetaWidth = twoThetaWidths[i];
    const double twoThetaHalfWidth = 0.5 * twoThetaWidth;
    const double twoThetaLower = twoTheta - twoThetaHalfWidth;
    const double twoThetaUpper = twoTheta + twoThetaHalfWidth;
 
    Mantid::HistogramData::HistogramX const &X = inputWS->x(i);
    Mantid::HistogramData::HistogramY const &Y = inputWS->y(i);
    Mantid::HistogramData::HistogramE const &E = inputWS->e(i);
    for (size_t j = 0; j < nBins; ++j) {
      const double lamLower = X[j];
      const double lamUpper = X[j + 1];
      const double signal = Y[j];
      const double error = E[j];
 
      auto inputQ = m_calculator->createQuad(lamUpper, lamLower, twoThetaUpper, twoThetaLower);
      FractionalRebinning::rebinToFractionalOutput(inputQ, inputWS, i, j, *outWS, zBinsVec);
      // Find which qy bin this point lies in
      const auto qIndex = std::upper_bound(zBinsVec.begin(), zBinsVec.end(), inputQ[0].Y()) - zBinsVec.begin();
      if (qIndex != 0 && qIndex < static_cast<int>(zBinsVec.size())) {
        // Add this spectra-detector pair to the mapping
        specNumberMapping.emplace_back(outWS->getSpectrum(qIndex - 1).getSpectrumNo());
        detIDMapping.emplace_back(detID);
      }
      // Debugging
      if (dumpVertexes) {
        writeRow(vertexes, inputQ[0], i, j, signal, error);
        writeRow(vertexes, inputQ[1], i, j, signal, error);
        writeRow(vertexes, inputQ[2], i, j, signal, error);
        writeRow(vertexes, inputQ[3], i, j, signal, error);
      }
    }
  }
  FractionalRebinning::finalizeFractionalRebin(*outWS);
  outWS->finalize();
  FractionalRebinning::normaliseOutput(outWS, inputWS);
  // Set the output spectrum-detector mapping
  SpectrumDetectorMapping outputDetectorMap(specNumberMapping, detIDMapping);
  outWS->updateSpectraUsing(outputDetectorMap);
  outWS->getAxis(0)->title() = m_d0Label;
  outWS->setYUnit("");
  outWS->setYUnitLabel("Intensity");
 
  return outWS;
}
} // namespace Mantid::DataObjects