PFCandConnector.h

Go to the documentation of this file.

#ifndef PFProducer_PFCandConnector_H_
#define PFProducer_PFCandConnector_H_

#include "DataFormats/ParticleFlowCandidate/interface/PFCandidate.h"
#include "DataFormats/ParticleFlowCandidate/interface/PFCandidateFwd.h"
#include "FWCore/MessageLogger/interface/MessageLogger.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/ParameterSet/interface/ParameterSetDescription.h"

// \author : M. Gouzevitch
// \date : May 2010


class PFCandConnector {
public:
  PFCandConnector() {
    bCorrect_ = false;
    bCalibPrimary_ = false;

    fConst_.push_back(1), fConst_.push_back(0);
    fNorm_.push_back(0), fNorm_.push_back(0);
    fExp_.push_back(0);

    dptRel_PrimaryTrack_ = 0.;
    dptRel_MergedTrack_ = 0.;
    ptErrorSecondary_ = 0.;
  }

  void setParameters(const edm::ParameterSet& iCfgCandConnector) {
    const auto& bCorrect = iCfgCandConnector.getParameter<bool>("bCorrect");
    const auto& bCalibPrimary = iCfgCandConnector.getParameter<bool>("bCalibPrimary");
    const auto& dptRel_PrimaryTrack = iCfgCandConnector.getParameter<double>("dptRel_PrimaryTrack");
    const auto& dptRel_MergedTrack = iCfgCandConnector.getParameter<double>("dptRel_MergedTrack");
    const auto& ptErrorSecondary = iCfgCandConnector.getParameter<double>("ptErrorSecondary");
    const auto& nuclCalibFactors = iCfgCandConnector.getParameter<std::vector<double> >("nuclCalibFactors");

    setParameters(bCorrect, bCalibPrimary, dptRel_PrimaryTrack, dptRel_MergedTrack, ptErrorSecondary, nuclCalibFactors);
  }

  void setParameters(bool bCorrect,
                     bool bCalibPrimary,
                     double dptRel_PrimaryTrack,
                     double dptRel_MergedTrack,
                     double ptErrorSecondary,
                     const std::vector<double>& nuclCalibFactors);

  static void fillPSetDescription(edm::ParameterSetDescription& iDesc);

  reco::PFCandidateCollection connect(reco::PFCandidateCollection& pfCand) const;

private:
  void analyseNuclearWPrim(reco::PFCandidateCollection&, std::vector<bool>&, unsigned int) const;

  void analyseNuclearWSec(reco::PFCandidateCollection&, std::vector<bool>&, unsigned int) const;

  bool isPrimaryNucl(const reco::PFCandidate& pf) const;

  bool isSecondaryNucl(const reco::PFCandidate& pf) const;

  double rescaleFactor(const double pt, const double cFrac) const;

  bool bCorrect_;

  bool bCalibPrimary_;
  std::vector<double> fConst_;
  std::vector<double> fNorm_;
  std::vector<double> fExp_;

  // Maximal accepatble uncertainty on primary tracks to usem them as MC truth for calibration
  double dptRel_PrimaryTrack_;
  double dptRel_MergedTrack_;
  double ptErrorSecondary_;

  static const double pion_mass2;
  static const reco::PFCandidate::Flags fT_TO_DISP_;
  static const reco::PFCandidate::Flags fT_FROM_DISP_;
};

#endif