GlobalMuonProducer.h

Go to the documentation of this file.

#ifndef RecoMuon_GlobalMuonProducer_GlobalMuonProducer_H
#define RecoMuon_GlobalMuonProducer_GlobalMuonProducer_H

#include "FWCore/Framework/interface/stream/EDProducer.h"

// Input and output collection
#include "DataFormats/TrackReco/interface/Track.h"
#include "DataFormats/TrackReco/interface/TrackFwd.h"

#include "DataFormats/MuonReco/interface/MuonTrackLinks.h"
#include "DataFormats/MuonReco/interface/MuonFwd.h"
#include "TrackingTools/PatternTools/interface/TrajTrackAssociation.h"
#include "DataFormats/TrackReco/interface/TrackToTrackMap.h"

namespace edm {
  class ParameterSet;
  class Event;
  class EventSetup;
}  // namespace edm

class MuonTrackFinder;
class MuonServiceProxy;

class GlobalMuonProducer : public edm::stream::EDProducer<> {
public:
  GlobalMuonProducer(const edm::ParameterSet&);

  ~GlobalMuonProducer() override;

  void produce(edm::Event&, const edm::EventSetup&) override;

private:
  edm::InputTag theSTACollectionLabel;
  edm::EDGetTokenT<reco::TrackCollection> staMuonsToken;
  edm::EDGetTokenT<std::vector<Trajectory> > staMuonsTrajToken;
  edm::EDGetTokenT<TrajTrackAssociationCollection> staAssoMapToken;
  edm::EDGetTokenT<reco::TrackToTrackMap> updatedStaAssoMapToken;

  std::unique_ptr<MuonTrackFinder> theTrackFinder;

  MuonServiceProxy* theService;

  std::string theAlias;

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

#endif