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