ConvBremPFTrackFinder.h

Go to the documentation of this file.

// user include files
#include "FWCore/Framework/interface/Frameworkfwd.h"

#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/MakerMacros.h"
#include "DataFormats/Common/interface/Handle.h"
#include "DataFormats/ParticleFlowReco/interface/PFRecTrack.h"
#include "DataFormats/ParticleFlowReco/interface/PFRecTrackFwd.h"
#include "DataFormats/ParticleFlowReco/interface/PFCluster.h"
#include "DataFormats/ParticleFlowReco/interface/PFClusterFwd.h"
#include "DataFormats/ParticleFlowReco/interface/GsfPFRecTrack.h"
#include "DataFormats/ParticleFlowReco/interface/GsfPFRecTrackFwd.h"
#include "TrackingTools/TransientTrack/interface/TransientTrackBuilder.h"
#include "DataFormats/VertexReco/interface/Vertex.h"
#include "DataFormats/VertexReco/interface/VertexFwd.h"
#include "DataFormats/ParticleFlowReco/interface/PFDisplacedTrackerVertex.h"
#include "DataFormats/ParticleFlowReco/interface/PFDisplacedVertexFwd.h"
#include "DataFormats/ParticleFlowReco/interface/PFDisplacedVertex.h"
#include "DataFormats/ParticleFlowReco/interface/PFConversionFwd.h"
#include "DataFormats/ParticleFlowReco/interface/PFConversion.h"
#include "DataFormats/ParticleFlowReco/interface/PFV0Fwd.h"
#include "DataFormats/ParticleFlowReco/interface/PFV0.h"
#include "TMVA/Reader.h"

#include "CondFormats/GBRForest/interface/GBRForest.h"
#include <memory>

#include "RecoParticleFlow/PFTracking/interface/ConvBremHeavyObjectCache.h"

class PFEnergyCalibration;

class ConvBremPFTrackFinder {
public:
  ConvBremPFTrackFinder(const TransientTrackBuilder& builder,
                        double mvaBremConvCutBarrelLowPt,
                        double mvaBremConvCutBarrelHighPt,
                        double mvaBremConvCutEndcapsLowPt,
                        double mvaBremConvCutEndcapsHighPt);
  ~ConvBremPFTrackFinder();

  bool foundConvBremPFRecTrack(const edm::Handle<reco::PFRecTrackCollection>& thePfRecTrackCol,
                               const edm::Handle<reco::VertexCollection>& primaryVertex,
                               const edm::Handle<reco::PFDisplacedTrackerVertexCollection>& pfNuclears,
                               const edm::Handle<reco::PFConversionCollection>& pfConversions,
                               const edm::Handle<reco::PFV0Collection>& pfV0,
                               const convbremhelpers::HeavyObjectCache* cache,
                               bool useNuclear,
                               bool useConversions,
                               bool useV0,
                               const reco::PFClusterCollection& theEClus,
                               const reco::GsfPFRecTrack& gsfpfrectk) {
    found_ = false;
    runConvBremFinder(thePfRecTrackCol,
                      primaryVertex,
                      pfNuclears,
                      pfConversions,
                      pfV0,
                      cache,
                      useNuclear,
                      useConversions,
                      useV0,
                      theEClus,
                      gsfpfrectk);
    return found_;
  };

  const std::vector<reco::PFRecTrackRef>& getConvBremPFRecTracks() { return pfRecTrRef_vec_; };

private:
  void runConvBremFinder(const edm::Handle<reco::PFRecTrackCollection>& thePfRecTrackCol,
                         const edm::Handle<reco::VertexCollection>& primaryVertex,
                         const edm::Handle<reco::PFDisplacedTrackerVertexCollection>& pfNuclears,
                         const edm::Handle<reco::PFConversionCollection>& pfConversions,
                         const edm::Handle<reco::PFV0Collection>& pfV0,
                         const convbremhelpers::HeavyObjectCache* cache,
                         bool useNuclear,
                         bool useConversions,
                         bool useV0,
                         const reco::PFClusterCollection& theEClus,
                         const reco::GsfPFRecTrack& gsfpfrectk);

  bool found_;
  TransientTrackBuilder builder_;
  double mvaBremConvCutBarrelLowPt_, mvaBremConvCutBarrelHighPt_, mvaBremConvCutEndcapsLowPt_,
      mvaBremConvCutEndcapsHighPt_;

  std::vector<reco::PFRecTrackRef> pfRecTrRef_vec_;
  float secR, secPout, ptRatioGsfKF, sTIP, Epout, detaBremKF, secPin;
  //int nHITS1;
  float nHITS1;
};