ConversionTrackCandidateProducer.h

Go to the documentation of this file.

#ifndef RecoEgamma_EgammaPhotonProducers_ConversionTrackCandidateProducer_h
#define RecoEgamma_EgammaPhotonProducers_ConversionTrackCandidateProducer_h

#include "FWCore/Framework/interface/EDProducer.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "FWCore/Framework/interface/ESHandle.h"

#include "MagneticField/Engine/interface/MagneticField.h"
#include "RecoTracker/MeasurementDet/interface/MeasurementTracker.h"
#include "Geometry/CaloGeometry/interface/CaloGeometry.h"
#include "RecoCaloTools/MetaCollections/interface/CaloRecHitMetaCollections.h"
#include "TrackingTools/MeasurementDet/interface/LayerMeasurements.h"
#include "TrackingTools/DetLayers/interface/NavigationSetter.h"
#include "TrackingTools/DetLayers/interface/NavigationSchool.h"
#include "RecoTracker/TkNavigation/interface/SimpleNavigationSchool.h"
#include "RecoTracker/TkDetLayers/interface/GeometricSearchTracker.h"
#include "DataFormats/TrackCandidate/interface/TrackCandidateCollection.h"
#include "DataFormats/CaloRecHit/interface/CaloCluster.h"
#include "DataFormats/CaloRecHit/interface/CaloClusterFwd.h"
#include "DataFormats/Common/interface/View.h"
#include "DataFormats/CaloTowers/interface/CaloTowerCollection.h"
#include "RecoCaloTools/MetaCollections/interface/CaloRecHitMetaCollections.h"
#include "RecoLocalCalo/EcalRecAlgos/interface/EcalSeverityLevelAlgoRcd.h"
#include "RecoLocalCalo/EcalRecAlgos/interface/EcalSeverityLevelAlgo.h"
#include "TrackingTools/PatternTools/interface/TrajectoryBuilder.h"

class OutInConversionSeedFinder;
class InOutConversionSeedFinder;
class OutInConversionTrackFinder;
class InOutConversionTrackFinder;

// ConversionTrackCandidateProducer inherits from EDProducer, so it can be a module:
class ConversionTrackCandidateProducer : public edm::EDProducer {

 public:

  ConversionTrackCandidateProducer (const edm::ParameterSet& ps);
  ~ConversionTrackCandidateProducer();
  
  virtual void beginRun (edm::Run const&, edm::EventSetup const & es) override final;
  virtual void endRun (edm::Run const&, edm::EventSetup const & es) override final;
  virtual void produce(edm::Event& evt, const edm::EventSetup& es);

 private:

  int nEvt_;
  
  void setEventSetup( const edm::EventSetup& es ) ;

  std::string OutInTrackCandidateCollection_;
  std::string InOutTrackCandidateCollection_;


  std::string OutInTrackSuperClusterAssociationCollection_;
  std::string InOutTrackSuperClusterAssociationCollection_;
  
  edm::EDGetTokenT<edm::View<reco::CaloCluster> > bcBarrelCollection_;
  edm::EDGetTokenT<edm::View<reco::CaloCluster> > bcEndcapCollection_;
  edm::EDGetTokenT<edm::View<reco::CaloCluster> > scHybridBarrelProducer_;
  edm::EDGetTokenT<edm::View<reco::CaloCluster> > scIslandEndcapProducer_;
  edm::ParameterSet conf_;
  edm::EDGetTokenT<CaloTowerCollection> hcalTowers_;
  edm::EDGetTokenT<EcalRecHitCollection> barrelecalCollection_;
  edm::EDGetTokenT<EcalRecHitCollection> endcapecalCollection_;
 
  double hOverEConeSize_;
  double maxHOverE_;
  double minSCEt_;
  double isoConeR_   ;   
  double isoInnerConeR_ ;
  double isoEtaSlice_   ;
  double isoEtMin_      ;
  double isoEMin_       ;
  bool   vetoClusteredHits_ ;
  bool   useNumXtals_;

  std::vector<int> flagsexclEB_;
  std::vector<int> flagsexclEE_;
  std::vector<int> severitiesexclEB_;
  std::vector<int> severitiesexclEE_;

  double ecalIsoCut_offset_;
  double ecalIsoCut_slope_;


  edm::ESHandle<CaloGeometry> theCaloGeom_;  

  std::string                      trajectoryBuilderName_;
  edm::ESHandle<TrajectoryBuilder> theTrajectoryBuilder_;

  const NavigationSchool*     theNavigationSchool_;
  OutInConversionSeedFinder*  theOutInSeedFinder_;
  OutInConversionTrackFinder* theOutInTrackFinder_;
  InOutConversionSeedFinder*  theInOutSeedFinder_;
  InOutConversionTrackFinder* theInOutTrackFinder_;


  std::vector<edm::Ptr<reco::CaloCluster> > caloPtrVecOutIn_; 
  std::vector<edm::Ptr<reco::CaloCluster> > caloPtrVecInOut_; 

  std::vector<edm::Ref<reco::SuperClusterCollection> > vecOfSCRefForOutIn;  
  std::vector<edm::Ref<reco::SuperClusterCollection> > vecOfSCRefForInOut;  
  
  void buildCollections(bool detector, 
            const edm::Handle<edm::View<reco::CaloCluster> > & scHandle,
            const edm::Handle<edm::View<reco::CaloCluster> > & bcHandle,
            edm::Handle<EcalRecHitCollection> ecalRecHitHandle, 
            CaloRecHitMetaCollectionV* metaEcalRecHits,
            const EcalSeverityLevelAlgo* sevLev,
            //edm::ESHandle<EcalChannelStatus>  chStatus,
            const edm::Handle<CaloTowerCollection> & hcalTowersHandle,
            TrackCandidateCollection& outInTracks,
            TrackCandidateCollection& inOutTracks,
            std::vector<edm::Ptr<reco::CaloCluster> >& vecRecOI,
            std::vector<edm::Ptr<reco::CaloCluster> >& vecRecIO);

};
#endif