CalibrationHelpers.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/CalibrationHelpers.h"
#include "MantidAPI/ResizeRectangularDetectorHelper.h"
#include "MantidCrystal/SCDCalibratePanels.h"
#include "MantidGeometry/Instrument/ComponentInfo.h"
 
using namespace Mantid::API;
using namespace Mantid::Geometry;
using namespace Mantid::Kernel;
 
namespace Mantid::Crystal::CalibrationHelpers {
void adjustUpSampleAndSourcePositions(const double L0, const V3D &newSampPos, ComponentInfo &componentInfo) {
 
  if (L0 <= 0)
    throw std::runtime_error("L0 is negative, must be positive.");
 
  const V3D &oldSourceToSampleDir = componentInfo.samplePosition() - componentInfo.sourcePosition();
  const double oldL1 = componentInfo.l1();
 
  V3D samplePos = componentInfo.samplePosition();
  if (samplePos != newSampPos) {
    componentInfo.setPosition(componentInfo.sample(), newSampPos);
  }
 
  double scalee = L0 / oldL1;
  V3D newSourcePos = newSampPos - oldSourceToSampleDir * scalee;
 
  componentInfo.setPosition(componentInfo.source(), newSourcePos);
}
 
void adjustBankPositionsAndSizes(const std::vector<std::string> &bankNames, const Instrument &newInstrument,
                                 const V3D &pos, const Quat &rot, const double detWScale, const double detHtScale,
                                 ComponentInfo &componentInfo) {
  std::shared_ptr<ParameterMap> pmap = newInstrument.getParameterMap();
 
  for (const auto &bankName : bankNames) {
    const size_t bankComponentIndex = componentInfo.indexOfAny(bankName);
    const size_t parentIndex = componentInfo.parent(bankComponentIndex);
 
    Quat relRot = componentInfo.relativeRotation(bankComponentIndex);
    Quat parentRot = componentInfo.rotation(parentIndex);
    Quat newRot = parentRot * rot * relRot;
 
    componentInfo.setRotation(bankComponentIndex, newRot);
 
    V3D rotatedPos = V3D(pos);
    parentRot.rotate(rotatedPos);
 
    componentInfo.setPosition(bankComponentIndex, rotatedPos + componentInfo.position(bankComponentIndex));
 
    const std::string &resolvedBankName = componentInfo.name(bankComponentIndex);
    std::vector<double> oldScalex = pmap->getDouble(resolvedBankName, std::string("scalex"));
    std::vector<double> oldScaley = pmap->getDouble(resolvedBankName, std::string("scaley"));
 
    double scalex, scaley;
    if (!oldScalex.empty())
      scalex = oldScalex[0] * detWScale;
    else
      scalex = detWScale;
 
    if (!oldScaley.empty())
      scaley = oldScaley[0] * detHtScale;
    else
      scaley = detHtScale;
 
    auto *bankComponentID = const_cast<IComponent *>(componentInfo.componentID(bankComponentIndex));
    pmap->addDouble(bankComponentID, std::string("scalex"), scalex);
    pmap->addDouble(bankComponentID, std::string("scaley"), scaley);
 
    if (detWScale != 1.0 || detHtScale != 1.0)
      applyRectangularDetectorScaleToComponentInfo(componentInfo, bankComponentID, detWScale, detHtScale);
  }
}
 
} // namespace Mantid::Crystal::CalibrationHelpers