CalculateReflectometryP.h

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 +
#pragma once
 
#include "MantidDataObjects/CalculateReflectometry.h"
 
namespace Mantid {
namespace DataObjects {
class CalculateReflectometryP : public DataObjects::CalculateReflectometry {
private:
  double m_sin_theta_i;
  double m_sin_theta_f;
 
public:
  CalculateReflectometryP(int version) : CalculateReflectometry(version), m_sin_theta_i(0.0), m_sin_theta_f(0.0) {}
 
  void updateThetaIncident(double thetaIncident) override { m_sin_theta_i = sin(to_radians_factor * thetaIncident); }
 
  void setThetaFinal(double thetaFinal) override { m_sin_theta_f = sin(to_radians_factor * thetaFinal); }
 
  void setTwoTheta(double twoTheta) override {
    if (m_version == 1)
      setThetaFinal(twoTheta);
    else
      setThetaFinal(twoTheta - m_theta_i);
  }
 
  double calculateDim0(double wavelength) const override {
    double wavenumber = 2.0 * M_PI / wavelength;
    double ki = wavenumber * m_sin_theta_i;
    double kf = wavenumber * m_sin_theta_f;
    return ki + kf;
  }
 
  double calculateDim1(double wavelength) const override {
    double wavenumber = 2.0 * M_PI / wavelength;
    double ki = wavenumber * m_sin_theta_i;
    double kf = wavenumber * m_sin_theta_f;
    return ki - kf;
  }
  Mantid::Geometry::Quadrilateral createQuad(double lamUpper, double lamLower, double thetaUpper,
                                             double thetaLower) override {
    setTwoTheta(thetaLower);
    auto dim1UpperRightVertex = calculateDim1(lamLower);
    auto dim0LowerLeftVertex = calculateDim0(lamUpper);
    // UPPER LEFT VERTEX
    const Mantid::Kernel::V2D secondVertex(calculateDim0(lamUpper), // highest qx
                                           calculateDim1(lamLower));
 
    setTwoTheta(thetaUpper);
    const Mantid::Kernel::V2D firstVertex(dim0LowerLeftVertex,
                                          calculateDim1(lamUpper)); // lowest qz
 
    const Mantid::Kernel::V2D thirdVertex(calculateDim0(lamLower),
                                          dim1UpperRightVertex); // highest qz
    // LOWER RIGHT VERTEX
    const Mantid::Kernel::V2D fourthVertex(calculateDim0(lamLower), // lowest qx
                                           calculateDim1(lamUpper));
 
    Mantid::Geometry::Quadrilateral quad(firstVertex, fourthVertex, thirdVertex, secondVertex);
    // Our lower-left vertex may not be in the right position
    // we keep shifting the vertices around in a clock-wise fashion
    // until the lower-left vertex is in the correct place.
    while ((quad.at(0).X() > quad.at(3).X()) || (quad.at(0).Y() > quad.at(1).Y())) {
      quad.shiftVertexesClockwise();
    }
    return quad;
  }
};
} // namespace DataObjects
} // namespace Mantid