PeakBackground.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 <utility>
 
#include "MantidAPI/IPeaksWorkspace.h"
#include "MantidCrystal/PeakBackground.h"
#include "MantidGeometry/Crystal/IPeak.h"
 
using namespace Mantid::API;
using namespace Mantid::Kernel;
using Mantid::Geometry::IPeak;
 
namespace Mantid::Crystal {
 
//----------------------------------------------------------------------------------------------
PeakBackground::PeakBackground(IPeaksWorkspace_const_sptr peaksWS, const double &radiusEstimate,
                               const double &thresholdSignal, const Mantid::API::MDNormalization normalisation,
                               const SpecialCoordinateSystem coordinates)
    : HardThresholdBackground(thresholdSignal, normalisation), m_peaksWS(std::move(peaksWS)),
      m_radiusEstimate(radiusEstimate), m_mdCoordinates(coordinates) {
 
  if (m_mdCoordinates == QLab) {
    m_coordFunction = &IPeak::getQLabFrame;
  } else if (m_mdCoordinates == QSample) {
    m_coordFunction = &IPeak::getQSampleFrame;
  } else if (m_mdCoordinates == Mantid::Kernel::HKL) {
    m_coordFunction = &IPeak::getHKL;
  } else {
    throw std::invalid_argument("Unknown CoordinateSystem provided to PeakBackground");
  }
}
 
PeakBackground *PeakBackground::clone() const { return new PeakBackground(*this); }
 
bool PeakBackground::isBackground(Mantid::API::IMDIterator *iterator) const {
  if (!HardThresholdBackground::isBackground(iterator)) {
    const VMD &center = iterator->getCenter();
    V3D temp(center[0], center[1], center[2]); // This assumes dims 1, 2, and 3
                                               // in the workspace correspond to
                                               // positions.
 
    for (int i = 0; i < m_peaksWS->getNumberPeaks(); ++i) {
      const IPeak &peak = m_peaksWS->getPeak(i);
      V3D coords = m_coordFunction(&peak);
      if (coords.distance(temp) < m_radiusEstimate) {
        return false;
      }
    }
  }
  return true;
}
 
void PeakBackground::configureIterator(Mantid::API::IMDIterator *const /*iterator*/) const {}
 
} // namespace Mantid::Crystal