TrackProducerBase.h

Go to the documentation of this file.

#ifndef TrackProducerBase_h
#define TrackProducerBase_h

#include "AlgoProductTraits.h"

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

#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/Utilities/interface/InputTag.h"

#include "TrackingTools/PatternTools/interface/Trajectory.h"
#include "TrackingTools/TrackFitters/interface/TrajectoryFitter.h"

#include "DataFormats/TrackReco/interface/TrackExtra.h"
#include "DataFormats/TrackCandidate/interface/TrackCandidateCollection.h"
#include "DataFormats/BeamSpot/interface/BeamSpot.h"
#include "DataFormats/TrackerRecHit2D/interface/ClusterRemovalInfo.h"
#include <RecoTracker/MeasurementDet/interface/MeasurementTracker.h>

#include <tuple>

class Propagator;
class TrajectoryStateUpdator;
class MeasurementEstimator;
class TrackerGeometry;
class TrajectoryFitter;
class TransientTrackingRecHitBuilder;
class NavigationSchool;
class TrackerDigiGeometryRecord;
class IdealMagneticFieldRecord;
class TransientRecHitRecord;
class TrackingComponentsRecord;
class NavigationSchoolRecord;
class CkfComponentsRecord;

template <class T>
class TrackProducerBase : public AlgoProductTraits<T> {
public:
  using Base = AlgoProductTraits<T>;
  using TrackView = typename Base::TrackView;
  using TrackCollection = typename Base::TrackCollection;
  using AlgoProductCollection = typename Base::AlgoProductCollection;

public:
  TrackProducerBase(bool trajectoryInEvent = false) : trajectoryInEvent_(trajectoryInEvent) {}

  virtual ~TrackProducerBase() noexcept(false);

  virtual void getFromES(const edm::EventSetup&,
                         edm::ESHandle<TrackerGeometry>&,
                         edm::ESHandle<MagneticField>&,
                         edm::ESHandle<TrajectoryFitter>&,
                         edm::ESHandle<Propagator>&,
                         edm::ESHandle<MeasurementTracker>&,
                         edm::ESHandle<TransientTrackingRecHitBuilder>&);

  virtual void getFromEvt(edm::Event&, edm::Handle<TrackCandidateCollection>&, reco::BeamSpot&);
  virtual void getFromEvt(edm::Event&, edm::Handle<TrackView>&, reco::BeamSpot&);

  virtual void produce(edm::Event&, const edm::EventSetup&) = 0;

  void initTrackProducerBase(const edm::ParameterSet& conf, edm::ConsumesCollector cc, const edm::EDGetToken& src);

  void setAlias(std::string alias) {
    alias.erase(alias.size() - 6, alias.size());
    alias_ = alias;
  }

  void setSecondHitPattern(Trajectory* traj,
                           T& track,
                           const Propagator* prop,
                           const MeasurementTrackerEvent* measTk,
                           const TrackerTopology* ttopo);

  const edm::ParameterSet& getConf() const { return conf_; }

protected:
  edm::ParameterSet conf_;
  edm::EDGetToken src_;

protected:
  std::string alias_;
  bool trajectoryInEvent_;
  edm::OrphanHandle<TrackCollection> rTracks_;
  edm::EDGetTokenT<reco::BeamSpot> bsSrc_;
  edm::EDGetTokenT<MeasurementTrackerEvent> mteSrc_;

  edm::ESGetToken<TrackerGeometry, TrackerDigiGeometryRecord> trackGeomSrc_;
  edm::ESGetToken<MagneticField, IdealMagneticFieldRecord> mfSrc_;
  edm::ESGetToken<TrajectoryFitter, TrajectoryFitter::Record> fitterSrc_;
  edm::ESGetToken<Propagator, TrackingComponentsRecord> propagatorSrc_;
  edm::ESGetToken<TransientTrackingRecHitBuilder, TransientRecHitRecord> builderSrc_;
  edm::ESGetToken<MeasurementTracker, CkfComponentsRecord> measTkSrc_;
  edm::ESGetToken<NavigationSchool, NavigationSchoolRecord> schoolSrc_;

  edm::ESHandle<NavigationSchool> theSchool;
  bool useSchool_ = false;
};

#include "RecoTracker/TrackProducer/interface/TrackProducerBase.icc"

#endif