DAFTrackProducer.h

Go to the documentation of this file.

 
#ifndef DAFTrackProducer_h
#define DAFTrackProducer_h
 
#include "FWCore/Framework/interface/stream/EDProducer.h"
#include "RecoTracker/TrackProducer/interface/KfTrackProducerBase.h"
#include "RecoTracker/TrackProducer/interface/DAFTrackProducerAlgorithm.h"
#include "TrackingTools/PatternTools/interface/Trajectory.h"
#include "TrackingTools/PatternTools/interface/TrajAnnealing.h"
 
class DAFTrackProducer : public KfTrackProducerBase, public edm::stream::EDProducer<> {
public:
  typedef std::vector<Trajectory> TrajectoryCollection;
  //  typedef std::vector<TrajAnnealing> TrajAnnealingCollection;
  explicit DAFTrackProducer(const edm::ParameterSet& iConfig);
 
  // Implementation of produce method
  void produce(edm::Event&, const edm::EventSetup&) override;
 
private:
  DAFTrackProducerAlgorithm theAlgo;
  using TrackProducerBase<reco::Track>::getFromEvt;
  void getFromEvt(edm::Event&, edm::Handle<TrajTrackAssociationCollection>&, reco::BeamSpot&);
  void putInEvtTrajAnn(edm::Event& theEvent,
                       TrajAnnealingCollection& trajannResults,
                       std::unique_ptr<TrajAnnealingCollection>& selTrajAnn);
 
  bool TrajAnnSaving_;
  edm::EDGetToken srcTT_;
};
 
#endif