MuonSpecialVariables Class Reference

Inheritance diagram for MuonSpecialVariables:

Public Member Functions
	MuonSpecialVariables (const edm::ParameterSet &iConfig)
void	setArbitratedTracker (pat::Muon &mu) const
void	setInnerTrackDxyz (pat::Muon &mu, const reco::Vertex &vtx) const
void	setNSegements (pat::Muon &mu) const
void	setTrkiso04 (pat::Muon &mu, const std::vector< reco::Track > tracks) const
	~MuonSpecialVariables () override
Public Member Functions inherited from edm::stream::EDProducer<>
	EDProducer ()=default
	EDProducer (const EDProducer &)=delete
bool	hasAbilityToProduceInBeginLumis () const final
bool	hasAbilityToProduceInBeginProcessBlocks () const final
bool	hasAbilityToProduceInBeginRuns () const final
bool	hasAbilityToProduceInEndLumis () const final
bool	hasAbilityToProduceInEndProcessBlocks () const final
bool	hasAbilityToProduceInEndRuns () const final
const EDProducer &	operator= (const EDProducer &)=delete

Private Member Functions
void	produce (edm::Event &iEvent, const edm::EventSetup &iSetup) override

Private Attributes
edm::EDGetToken	muonSrc_
edm::EDGetToken	trkSrc_
edm::EDGetToken	vertexSrc_

Additional Inherited Members
Public Types inherited from edm::stream::EDProducer<>
using	CacheTypes = CacheContexts< T... >
using	GlobalCache = typename CacheTypes::GlobalCache
using	HasAbility = AbilityChecker< T... >
using	InputProcessBlockCache = typename CacheTypes::InputProcessBlockCache
using	LuminosityBlockCache = typename CacheTypes::LuminosityBlockCache
using	LuminosityBlockContext = LuminosityBlockContextT< LuminosityBlockCache, RunCache, GlobalCache >
using	LuminosityBlockSummaryCache = typename CacheTypes::LuminosityBlockSummaryCache
using	RunCache = typename CacheTypes::RunCache
using	RunContext = RunContextT< RunCache, GlobalCache >
using	RunSummaryCache = typename CacheTypes::RunSummaryCache

Detailed Description

Definition at line 23 of file MuonSpecialVariables.cc.

Constructor & Destructor Documentation

MuonSpecialVariables()

MuonSpecialVariables::MuonSpecialVariables ( const edm::ParameterSet &  iConfig )

inlineexplicit

Definition at line 25 of file MuonSpecialVariables.cc.

      : muonSrc_(consumes<std::vector<pat::Muon>>(iConfig.getParameter<edm::InputTag>("muonSrc"))),
        vertexSrc_(consumes<std::vector<reco::Vertex>>(iConfig.getParameter<edm::InputTag>("vertexSrc"))),
        trkSrc_(consumes<reco::TrackCollection>(iConfig.getParameter<edm::InputTag>("trkSrc"))) {
    produces<std::vector<pat::Muon>>();
  }

~MuonSpecialVariables()

MuonSpecialVariables::~MuonSpecialVariables ( )

inlineoverride

Definition at line 31 of file MuonSpecialVariables.cc.

{}

Member Function Documentation

produce()

void MuonSpecialVariables::produce ( edm::Event &  iEvent, const edm::EventSetup &  iSetup  )

overrideprivate

Definition at line 83 of file MuonSpecialVariables.cc.

References mps_fire::i, iEvent, eostools::move(), amptDefaultParameters_cff::mu, pdwgLeptonRecoSkim_cfi::muonCollection, B2GDQM_cfi::muonSrc, muonSrc_, MillePedeFileConverter_cfg::out, setArbitratedTracker(), setInnerTrackDxyz(), setNSegements(), setTrkiso04(), trkSrc_, vertexSrc_, AlignmentTracksFromVertexSelector_cfi::vertices, and L1BJetProducer_cff::vtx.

                                                                                {
  edm::Handle<std::vector<pat::Muon>> muonSrc;
  iEvent.getByToken(muonSrc_, muonSrc);

  edm::Handle<std::vector<reco::Vertex>> vertices;
  iEvent.getByToken(vertexSrc_, vertices);

  edm::Handle<reco::TrackCollection> trkCollection;
  iEvent.getByToken(trkSrc_, trkCollection);

  std::unique_ptr<std::vector<pat::Muon>> muonCollection(new std::vector<pat::Muon>(*muonSrc));

  std::unique_ptr<std::vector<pat::Muon>> out(new std::vector<pat::Muon>());

  out->reserve(muonCollection->size());

  for (unsigned int i = 0; i < muonCollection->size(); i++) {
    pat::Muon &mu = (*muonCollection).at(i);
    setArbitratedTracker(mu);
    int good_vertex = 0;
    for (auto &vtx : *vertices) {
      if (vtx.isFake() || !vtx.isValid())
        continue;
      setInnerTrackDxyz(mu, vtx);
      good_vertex = 1;
      break;
    }
    mu.addUserInt("isGoodVertex", good_vertex);
    setTrkiso04(mu, *trkCollection);
    setNSegements(mu);
    out->push_back(mu);
  }

  iEvent.put(std::move(out));
}

setArbitratedTracker()

void MuonSpecialVariables::setArbitratedTracker

(

pat::Muon &

mu

)

const

Definition at line 46 of file MuonSpecialVariables.cc.

References muon::isGoodMuon(), amptDefaultParameters_cff::mu, and muon::TrackerMuonArbitrated.

Referenced by produce().

                                                                 {
  mu.addUserInt("isArbitratedTracker", muon::isGoodMuon(mu, muon::TrackerMuonArbitrated));
}

setInnerTrackDxyz()

void MuonSpecialVariables::setInnerTrackDxyz	(	pat::Muon &	mu,
		const reco::Vertex &	vtx
	)		const

Definition at line 50 of file MuonSpecialVariables.cc.

References amptDefaultParameters_cff::mu, and L1BJetProducer_cff::vtx.

Referenced by produce().

                                                                                     {
  float innerTrackDxy =
      mu.innerTrack().isNonnull() ? mu.innerTrack()->dxy(reco::TrackBase::Point(vtx.x(), vtx.y(), vtx.z())) : -999.9;
  float innerTrackDz =
      mu.innerTrack().isNonnull() ? mu.innerTrack()->dz(reco::TrackBase::Point(vtx.x(), vtx.y(), vtx.z())) : -999.9;
  mu.addUserFloat("innerTrackDxy", innerTrackDxy);
  mu.addUserFloat("innerTrackDz", innerTrackDz);
}

setNSegements()

void MuonSpecialVariables::setNSegements

(

pat::Muon &

mu

)

const

Definition at line 71 of file MuonSpecialVariables.cc.

References relativeConstraints::chamber, MuonSubdetId::CSC, MuonSubdetId::DT, amptDefaultParameters_cff::mu, and DetId::Muon.

Referenced by produce().

                                                          {
  int nsegments = 0;
  for (auto &chamber : mu.matches()) {
    if (chamber.id.det() != DetId::Muon)
      continue;
    if (chamber.id.subdetId() != MuonSubdetId::DT && chamber.id.subdetId() != MuonSubdetId::CSC)
      continue;
    nsegments += chamber.segmentMatches.size();
  }
  mu.addUserInt("nsegments", nsegments);
}

setTrkiso04()

void MuonSpecialVariables::setTrkiso04	(	pat::Muon &	mu,
		const std::vector< reco::Track >	tracks
	)		const

Definition at line 59 of file MuonSpecialVariables.cc.

References PbPb_ZMuSkimMuonDPG_cff::deltaR, hcalRecHitTable_cff::energy, amptDefaultParameters_cff::mu, and pwdgSkimBPark_cfi::tracks.

Referenced by produce().

                                                                                           {
  float energy = 0;
  for (const auto &trk : tracks) {
    if (deltaR(mu.eta(), mu.phi(), trk.eta(), trk.phi()) > 0.4)
      continue;
    energy += trk.pt();
  }
  float Trkiso04 = (energy - mu.pt()) / mu.pt();
  float relTrkiso4 = (Trkiso04 > 0) ? Trkiso04 : 0;
  mu.addUserFloat("relTrkiso4", relTrkiso4);
}

Member Data Documentation

muonSrc_

edm::EDGetToken MuonSpecialVariables::muonSrc_

private

Definition at line 41 of file MuonSpecialVariables.cc.

Referenced by produce().

trkSrc_

edm::EDGetToken MuonSpecialVariables::trkSrc_

private

Definition at line 43 of file MuonSpecialVariables.cc.

Referenced by produce().

vertexSrc_

edm::EDGetToken MuonSpecialVariables::vertexSrc_

private

Definition at line 42 of file MuonSpecialVariables.cc.

Referenced by produce().