ProtonReconstructionAlgorithm.h

Go to the documentation of this file.

/****************************************************************************
 * Authors:
 *   Jan Kašpar
 *   Laurent Forthomme
 ****************************************************************************/

#ifndef RecoCTPPS_ProtonReconstruction_ProtonReconstructionAlgorithm_h
#define RecoCTPPS_ProtonReconstruction_ProtonReconstructionAlgorithm_h

#include "FWCore/ParameterSet/interface/ParameterSet.h"

#include "DataFormats/CTPPSReco/interface/CTPPSLocalTrackLiteFwd.h"
#include "DataFormats/ProtonReco/interface/ForwardProtonFwd.h"

#include "CondFormats/RunInfo/interface/LHCInfo.h"
#include "CondFormats/CTPPSReadoutObjects/interface/LHCInterpolatedOpticalFunctionsSet.h"
#include "CondFormats/CTPPSReadoutObjects/interface/LHCInterpolatedOpticalFunctionsSetCollection.h"

#include "TSpline.h"
#include "Fit/Fitter.h"

#include <unordered_map>

//----------------------------------------------------------------------------------------------------

class ProtonReconstructionAlgorithm
{
  public:
    ProtonReconstructionAlgorithm(bool fit_vtx_y, bool improved_estimate, unsigned int verbosity);
    ~ProtonReconstructionAlgorithm() = default;

    void init(const LHCInterpolatedOpticalFunctionsSetCollection &opticalFunctions);
    void release();

    reco::ForwardProton reconstructFromSingleRP(const CTPPSLocalTrackLiteRef &track,
      const LHCInfo& lhcInfo, std::ostream& os) const;

    reco::ForwardProton reconstructFromMultiRP(const CTPPSLocalTrackLiteRefVector &tracks,
      const LHCInfo& lhcInfo, std::ostream& os) const;

  private:
    unsigned int verbosity_;
    bool fitVtxY_;
    bool useImprovedInitialEstimate_;
    bool initialized_;

    struct RPOpticsData
    {
      const LHCInterpolatedOpticalFunctionsSet *optics;
      std::shared_ptr<const TSpline3> s_xi_vs_x_d, s_y_d_vs_xi, s_v_y_vs_xi, s_L_y_vs_xi;
      double x0; 
      double y0; 
      double ch0; 
      double ch1; 
      double la0; 
      double la1; 
    };

    std::map<unsigned int, RPOpticsData> m_rp_optics_;

    class ChiSquareCalculator
    {
      public:
        ChiSquareCalculator() = default;

        double operator() (const double *parameters) const;

        const CTPPSLocalTrackLiteRefVector* tracks;
        const std::map<unsigned int, RPOpticsData>* m_rp_optics;
    };

    std::unique_ptr<ROOT::Fit::Fitter> fitter_;

    std::unique_ptr<ChiSquareCalculator> chiSquareCalculator_;

    static void doLinearFit(const std::vector<double> &vx, const std::vector<double> &vy, double &b, double &a);
};

#endif