RecoTrackAccumulator.h

Go to the documentation of this file.

#ifndef RecoTrackAccumulator_h
#define RecoTrackAccumulator_h

#include "SimGeneral/MixingModule/interface/DigiAccumulatorMixMod.h"
#include "FWCore/Framework/interface/ESHandle.h"

#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/EventPrincipal.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "SimGeneral/MixingModule/interface/PileUpEventPrincipal.h"
#include "DataFormats/TrackReco/interface/TrackFwd.h"
#include "DataFormats/TrackReco/interface/TrackExtra.h"
#include "DataFormats/TrackReco/interface/TrackExtraFwd.h"


namespace edm {
  class ConsumesCollector;
  template<typename T> class Handle;
  namespace stream {
    class EDProducerBase;
  }
  class StreamID;
}


class RecoTrackAccumulator : public DigiAccumulatorMixMod 
{
 public:
  explicit RecoTrackAccumulator(const edm::ParameterSet& conf, edm::stream::EDProducerBase& mixMod, edm::ConsumesCollector& iC);
  virtual ~RecoTrackAccumulator();
  
  virtual void initializeEvent(edm::Event const& e, edm::EventSetup const& c) override;
  virtual void accumulate(edm::Event const& e, edm::EventSetup const& c) override;
  virtual void accumulate(PileUpEventPrincipal const& e, edm::EventSetup const& c, edm::StreamID const&) override;
  virtual void finalizeEvent(edm::Event& e, edm::EventSetup const& c) override;

  
 private:
  template<class T> void accumulateEvent(const T& e, edm::EventSetup const& c,const edm::InputTag & label);

  std::unique_ptr<reco::TrackCollection>  newTracks_;
  std::unique_ptr<reco::TrackExtraCollection> newTrackExtras_;
  std::unique_ptr<TrackingRecHitCollection> newHits_;

  reco::TrackRefProd rNewTracks;
  reco::TrackExtraRefProd rNewTrackExtras;
  TrackingRecHitRefProd rNewHits;

  edm::InputTag signalTracksTag;
  edm::InputTag pileUpTracksTag;

  std::string outputLabel;
  
};


#endif