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) {
    std::shared_ptr<const IComponent> bank1 = newInstrument.getComponentByName(bankName);
    std::shared_ptr<const Geometry::RectangularDetector> bank =
        std::dynamic_pointer_cast<const RectangularDetector>(bank1);
 
    Quat relRot = bank->getRelativeRot();
    Quat parentRot = bank->getParent()->getRotation();
    Quat newRot = parentRot * rot * relRot;
 
    const auto bankComponentIndex = componentInfo.indexOf(bank->getComponentID());
    componentInfo.setRotation(bankComponentIndex, newRot);
 
    V3D rotatedPos = V3D(pos);
    bank->getParent()->getRotation().rotate(rotatedPos);
 
    componentInfo.setPosition(bankComponentIndex, rotatedPos + bank->getPos());
 
    std::vector<double> oldScalex = pmap->getDouble(bank->getName(), std::string("scalex"));
    std::vector<double> oldScaley = pmap->getDouble(bank->getName(), 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;
 
    pmap->addDouble(bank.get(), std::string("scalex"), scalex);
    pmap->addDouble(bank.get(), std::string("scaley"), scaley);
 
    if (detWScale != 1.0 || detHtScale != 1.0)
      applyRectangularDetectorScaleToComponentInfo(componentInfo, bank->getComponentID(), detWScale, detHtScale);
  }
}
 
} // namespace Mantid::Crystal::CalibrationHelpers