LHCInterpolatedOpticalFunctionsSet.h

Go to the documentation of this file.

// Original Author:  Jan Kašpar

#ifndef CondFormats_CTPPSReadoutObjects_LHCInterpolatedOpticalFunctionsSet_h
#define CondFormats_CTPPSReadoutObjects_LHCInterpolatedOpticalFunctionsSet_h

#include "CondFormats/CTPPSReadoutObjects/interface/LHCOpticalFunctionsSet.h"

#include "TSpline.h"

class CTPPSInterpolatedOpticalFunctionsESSource;
class CTPPSModifiedOpticalFunctionsESSource;

class LHCInterpolatedOpticalFunctionsSet : public LHCOpticalFunctionsSet
{
  public:
    LHCInterpolatedOpticalFunctionsSet() = default;

    LHCInterpolatedOpticalFunctionsSet(const LHCOpticalFunctionsSet &src) : LHCOpticalFunctionsSet(src) {}

    ~LHCInterpolatedOpticalFunctionsSet() = default;

    const std::vector<std::shared_ptr<const TSpline3>>& splines() const { return m_splines; }

    void initializeSplines();

    struct Kinematics
    {
      double x;     // physics vertex position (beam offset subtracted), cm
      double th_x;  // physics scattering angle (crossing angle subtracted), rad
      double y;     // physics vertex position, cm
      double th_y;  // physics scattering angle, rad
      double xi;    // relative momentum loss (positive for diffractive protons)
    };

    void transport(const Kinematics &input, Kinematics &output, bool calculateAngles = false) const;

  protected:
    friend CTPPSInterpolatedOpticalFunctionsESSource;
    friend CTPPSModifiedOpticalFunctionsESSource;

    std::vector<std::shared_ptr<const TSpline3>> m_splines;
};

#endif