FastTSGFromL2Muon

Macros

Line Code
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50
#ifndef FastSimulation_Muons_FastTSGFromL2Muon_H
#define FastSimulation_Muons_FastTSGFromL2Muon_H

#include "FWCore/Framework/interface/stream/EDProducer.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "DataFormats/TrackReco/interface/TrackFwd.h"
#include "DataFormats/TrajectorySeed/interface/TrajectorySeedCollection.h"
#include "SimDataFormats/Track/interface/SimTrackContainer.h"

#include <vector>
namespace edm {
  class ParameterSet;
  class Event;
  class EventSetup;
}  // namespace edm

class MuonServiceProxy;
class MuonTrackingRegionBuilder;
class RectangularEtaPhiTrackingRegion;
class SimTrack;

//
// generate seeds corresponding to L2 muons
//

class FastTSGFromL2Muon : public edm::stream::EDProducer<> {
public:
  FastTSGFromL2Muon(const edm::ParameterSet& cfg);
  ~FastTSGFromL2Muon() override = default;
  void produce(edm::Event& ev, const edm::EventSetup& es) override;

private:
  bool clean(reco::TrackRef muRef,
             RectangularEtaPhiTrackingRegion* region,
             const BasicTrajectorySeed* aSeed,
             const SimTrack& theSimTrack);

private:
  const double thePtCut;
  const edm::InputTag theL2CollectionLabel;
  const std::vector<edm::InputTag> theSeedCollectionLabels;
  const edm::InputTag theSimTrackCollectionLabel;

  const edm::EDGetTokenT<edm::SimTrackContainer> simTrackToken_;
  const edm::EDGetTokenT<reco::TrackCollection> l2TrackToken_;
  std::vector<edm::EDGetTokenT<edm::View<TrajectorySeed> > > seedToken_;

  std::unique_ptr<MuonTrackingRegionBuilder> theRegionBuilder;
};
#endif