ElectronSeedGenerator.h

Go to the documentation of this file.

#ifndef ElectronSeedGenerator_H
#define ElectronSeedGenerator_H
 
#include "RecoEgamma/EgammaElectronAlgos/interface/PixelHitMatcher.h"
 
#include "TrackingTools/TrajectoryState/interface/TrajectoryStateTransform.h"
#include "TrackingTools/PatternTools/interface/TrajectoryStateUpdator.h"
#include "TrackingTools/TransientTrackingRecHit/interface/TransientTrackingRecHit.h"
 
#include "RecoTracker/TransientTrackingRecHit/interface/TSiPixelRecHit.h"
#include "Geometry/TrackerGeometryBuilder/interface/TrackerGeometry.h"
 
#include "DataFormats/EgammaReco/interface/ElectronSeed.h"
#include "DataFormats/EgammaReco/interface/ElectronSeedFwd.h"
#include "DataFormats/EgammaReco/interface/SuperClusterFwd.h"
#include "DataFormats/TrajectorySeed/interface/TrajectorySeedCollection.h"
#include "DataFormats/BeamSpot/interface/BeamSpot.h"
#include "DataFormats/VertexReco/interface/Vertex.h"
#include "DataFormats/VertexReco/interface/VertexFwd.h"
 
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/ESHandle.h"
#include "FWCore/Framework/interface/EDConsumerBase.h"
 
#include "TrackingTools/KalmanUpdators/interface/KFUpdator.h"
 
class ElectronSeedGenerator {
public:
  struct Tokens {
    edm::EDGetTokenT<std::vector<reco::Vertex> > token_vtx;
    edm::EDGetTokenT<reco::BeamSpot> token_bs;
  };
 
  typedef edm::OwnVector<TrackingRecHit> PRecHitContainer;
  typedef TransientTrackingRecHit::ConstRecHitPointer ConstRecHitPointer;
  typedef TransientTrackingRecHit::RecHitPointer RecHitPointer;
  typedef TransientTrackingRecHit::RecHitContainer RecHitContainer;
 
  ElectronSeedGenerator(const edm::ParameterSet&, const Tokens&);
 
  void setupES(const edm::EventSetup& setup);
  void run(edm::Event&,
           const edm::EventSetup& setup,
           const reco::SuperClusterRefVector&,
           const std::vector<const TrajectorySeedCollection*>& seedsV,
           reco::ElectronSeedCollection&);
 
private:
  void seedsFromThisCluster(edm::Ref<reco::SuperClusterCollection> seedCluster,
                            reco::BeamSpot const& beamSpot,
                            std::vector<reco::Vertex> const* vertices,
                            reco::ElectronSeedCollection& out);
 
  const bool dynamicPhiRoad_;
  const edm::EDGetTokenT<std::vector<reco::Vertex> > verticesTag_;
  const edm::EDGetTokenT<reco::BeamSpot> beamSpotTag_;
 
  const float lowPtThresh_;
  const float highPtThresh_;
  const float nSigmasDeltaZ1_;     // first z window size if not using the reco vertex
  const float deltaZ1WithVertex_;  // first z window size when using the reco vertex
  const float sizeWindowENeg_;
 
  const float deltaPhi1Low_;
  const float deltaPhi1High_;
 
  // so that deltaPhi1 = dPhi1Coef1_ + dPhi1Coef2_/clusterEnergyT
  const double dPhi1Coef2_;
  const double dPhi1Coef1_;
 
  const std::vector<const TrajectorySeedCollection*>* initialSeedCollectionVector_ = nullptr;
 
  edm::ESHandle<MagneticField> magField_;
  edm::ESHandle<TrackerGeometry> trackerGeometry_;
  std::unique_ptr<PropagatorWithMaterial> propagator_;
 
  // keep cacheIds to get records only when necessary
  unsigned long long cacheIDMagField_ = 0;
  unsigned long long cacheIDCkfComp_ = 0;
  unsigned long long cacheIDTrkGeom_ = 0;
 
  const bool useRecoVertex_;
 
  const float deltaPhi2B_;
  const float deltaPhi2F_;
 
  const float phiMin2B_;
  const float phiMin2F_;
  const float phiMax2B_;
  const float phiMax2F_;
 
  PixelHitMatcher electronMatcher_;
  PixelHitMatcher positronMatcher_;
};
 
#endif  // ElectronSeedGenerator_H