CoordTransformAligned.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/CoordTransformAligned.h"
#include "MantidKernel/Matrix.h"
 
using namespace Mantid::Kernel;
using namespace Mantid::API;
 
namespace Mantid::DataObjects {
 
//----------------------------------------------------------------------------------------------
CoordTransformAligned::CoordTransformAligned(const size_t inD, const size_t outD, const size_t *dimensionToBinFrom,
                                             const coord_t *origin, const coord_t *scaling)
    : CoordTransform(inD, outD), m_dimensionToBinFrom(outD), m_origin(outD), m_scaling(outD) {
  if (!origin || !scaling || !dimensionToBinFrom)
    throw std::runtime_error("CoordTransformAligned::ctor(): at least one of "
                             "the input arrays is a NULL pointer.");
  for (size_t d = 0; d < outD; d++) {
    m_dimensionToBinFrom[d] = dimensionToBinFrom[d];
    if (m_dimensionToBinFrom[d] >= inD) {
      throw std::runtime_error("CoordTransformAligned::ctor(): invalid entry in "
                               "dimensionToBinFrom[" +
                               std::to_string(d) + "]. Cannot build the coordinate transformation.");
    }
    m_origin[d] = origin[d];
    m_scaling[d] = scaling[d];
  }
}
 
//----------------------------------------------------------------------------------------------
CoordTransformAligned::CoordTransformAligned(const size_t inD, const size_t outD,
                                             std::vector<size_t> dimensionToBinFrom, std::vector<coord_t> origin,
                                             std::vector<coord_t> scaling)
    : CoordTransform(inD, outD), m_dimensionToBinFrom(std::move(dimensionToBinFrom)), m_origin(std::move(origin)),
      m_scaling(std::move(scaling)) {
  if (m_dimensionToBinFrom.size() != outD || m_origin.size() != outD || m_scaling.size() != outD)
    throw std::runtime_error("CoordTransformAligned::ctor(): at least one of "
                             "the input vectors is the wrong size.");
  for (size_t d = 0; d < outD; d++) {
    if (m_dimensionToBinFrom[d] >= inD) {
      throw std::runtime_error("CoordTransformAligned::ctor(): invalid entry in "
                               "dimensionToBinFrom[" +
                               std::to_string(d) + "]. Cannot build the coordinate transformation.");
    }
  }
}
 
//----------------------------------------------------------------------------------------------
CoordTransform *CoordTransformAligned::clone() const {
  return new CoordTransformAligned(inD, outD, m_dimensionToBinFrom, m_origin, m_scaling);
}
 
//----------------------------------------------------------------------------------------------
void CoordTransformAligned::apply(const coord_t *inputVector, coord_t *outVector) const {
  // For each output dimension
  for (size_t out = 0; out < outD; ++out) {
    // Get the coordinate at the dimension in the INPUT workspace corresponding
    // to this OUTPUT
    coord_t x = inputVector[m_dimensionToBinFrom[out]];
    // Convert by taking the origin out then scaling.
    outVector[out] = (x - m_origin[out]) * m_scaling[out];
  }
}
 
//----------------------------------------------------------------------------------------------
Mantid::Kernel::Matrix<coord_t> CoordTransformAligned::makeAffineMatrix() const {
  // Zeroed-out affine matrix
  Matrix<coord_t> mat(outD + 1, inD + 1);
  // Bottom-right corner of the matrix is always 1.
  mat[outD][inD] = 1.0;
  // For each output dimension
  for (size_t out = 0; out < outD; ++out) {
    // The ROW is the out dimension.
    size_t row = out;
    // The COLUMN is the input dimension
    size_t col = m_dimensionToBinFrom[out];
    // So place the scaling factor at that spot
    mat[row][col] = m_scaling[out];
    // And place the origin (translation amount) at the last column
    mat[row][inD] = -m_origin[out] * m_scaling[out];
  }
  return mat;
}
 
//----------------------------------------------------------------------------------------------
std::string CoordTransformAligned::toXMLString() const {
  using namespace Poco::XML;
 
  AutoPtr<Document> pDoc = new Document;
  AutoPtr<Element> coordTransformElement = pDoc->createElement("CoordTransform");
  pDoc->appendChild(coordTransformElement);
 
  AutoPtr<Element> coordTransformTypeElement = pDoc->createElement("Type");
  coordTransformTypeElement->appendChild(pDoc->createTextNode("CoordTransformAffine"));
  coordTransformElement->appendChild(coordTransformTypeElement);
 
  AutoPtr<Element> paramListElement = pDoc->createElement("ParameterList");
 
  AutoPtr<Text> formatText = pDoc->createTextNode("%s%s%s");
  paramListElement->appendChild(formatText);
 
  coordTransformElement->appendChild(paramListElement);
 
  std::stringstream xmlstream;
 
  DOMWriter writer;
  writer.writeNode(xmlstream, pDoc);
 
  // Convert the members to parameters
  InDimParameter inD_param(inD);
  OutDimParameter outD_param(outD);
 
  std::string formattedXMLString = boost::str(boost::format(xmlstream.str().c_str()) % inD_param.toXMLString().c_str() %
                                              outD_param.toXMLString().c_str());
  return formattedXMLString;
}
 
std::string CoordTransformAligned::id() const { return "CoordTransformAligned"; }
 
} // namespace Mantid::DataObjects