TrackProducerBase.h

Go to the documentation of this file.

#ifndef TrackProducerBase_h
#define TrackProducerBase_h

#include "FWCore/Framework/interface/EDProducer.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 "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>

class Propagator;
class TrajectoryStateUpdator;
class MeasurementEstimator;
class TrackerGeometry;
class TrajectoryFitter;
class TransientTrackingRecHitBuilder;
class NavigationSchool;

template <class T>
class TrackProducerBase {
public:
  typedef std::vector<T> TrackCollection;
  typedef std::pair<Trajectory*, std::pair<T*,PropagationDirection> > AlgoProduct;
  typedef std::vector< AlgoProduct >  AlgoProductCollection;
public:
  TrackProducerBase(bool trajectoryInEvent = false):
     trajectoryInEvent_(trajectoryInEvent),
        rekeyClusterRefs_(false) {}

  virtual ~TrackProducerBase();
  
  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<TrackCollection>&, reco::BeamSpot&);

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

  void setConf(const edm::ParameterSet& conf){conf_=conf;}

  void setSrc(const edm::EDGetToken& src, const edm::EDGetTokenT<reco::BeamSpot>& bsSrc, const edm::EDGetTokenT<MeasurementTrackerEvent> &mteSrc) {
    src_ = src; bsSrc_ = bsSrc; mteSrc_ = mteSrc;
  }

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

  void setClusterRemovalInfo(const edm::InputTag &clusterRemovalInfo) {
    rekeyClusterRefs_ = true;
    clusterRemovalInfo_ = clusterRemovalInfo;
  }

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

  const edm::ParameterSet& getConf() const {return conf_;}
 private:
  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_;

  bool rekeyClusterRefs_;
  edm::InputTag clusterRemovalInfo_;

  edm::ESHandle<NavigationSchool> theSchool;

};

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

#endif