ProtonReconstructionAlgorithm.h

Go to the documentation of this file.

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

#ifndef RecoPPS_ProtonReconstruction_ProtonReconstructionAlgorithm_h
#define RecoPPS_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/PPSObjects/interface/LHCInterpolatedOpticalFunctionsSet.h"
#include "CondFormats/PPSObjects/interface/LHCInterpolatedOpticalFunctionsSetCollection.h"

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

#include <unordered_map>

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

class ProtonReconstructionAlgorithm {
public:
  ProtonReconstructionAlgorithm(bool fit_vtx_y,
                                bool improved_estimate,
                                const std::string &multiRPAlgorithm,
                                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_;
  enum { mrpaUndefined, mrpaChi2, mrpaNewton, mrpaAnalIter } multi_rp_algorithm_;
  bool initialized_;

  struct RPOpticsData {
    const LHCInterpolatedOpticalFunctionsSet *optics;
    std::shared_ptr<const TSpline3> s_x_d_vs_xi, s_L_x_vs_xi, 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);

  static double newtonGoalFcn(double xi, double x_N, double x_F, const RPOpticsData &i_N, const RPOpticsData &i_F);
};

#endif