#include <MuonSimHitMatcher.h>

Inheritance diagram for MuonSimHitMatcher:

Public Member Functions
std::set< unsigned int >	chamberIds (int type=0) const

std::set< unsigned int >	detIds (int type=0) const

const edm::PSimHitContainer &	hitsInChamber (unsigned int) const

const edm::PSimHitContainer &	hitsInDetId (unsigned int) const

void	init (const edm::Event &e, const edm::EventSetup &eventSetup)
	initialize the event More...

void	match (const SimTrack &t, const SimVertex &v)
	do the matching More...

	MuonSimHitMatcher (const edm::ParameterSet &iPS, edm::ConsumesCollector &&iC)

const edm::PSimHitContainer &	simHits (int) const

GlobalVector	simHitsMeanMomentum (const edm::PSimHitContainer &sim_hits) const

GlobalPoint	simHitsMeanPosition (const edm::PSimHitContainer &sim_hits) const

GlobalPoint	simHitsMeanPositionStation (int n) const

	~MuonSimHitMatcher ()

Protected Member Functions
std::vector< unsigned int >	getIdsOfSimTrackShower (unsigned trk_id, const edm::SimTrackContainer &simTracks, const edm::SimVertexContainer &simVertices)

Protected Attributes
std::map< unsigned int, edm::PSimHitContainer >	chamber_to_hits_

std::map< unsigned int, edm::PSimHitContainer >	detid_to_hits_

bool	discardEleHits_

const TrackingGeometry *	geometry_

bool	hasGeometry_

edm::PSimHitContainer	hits_

edm::PSimHitContainer	no_hits_

edm::EDGetTokenT< edm::PSimHitContainer >	simHitInput_

edm::ParameterSet	simHitPSet_

edm::PSimHitContainer	simHits_

edm::Handle< edm::PSimHitContainer >	simHitsH_

bool	simMuOnly_

edm::EDGetTokenT< edm::SimTrackContainer >	simTrackInput_

edm::SimTrackContainer	simTracks_

edm::Handle< edm::SimTrackContainer >	simTracksH_

edm::EDGetTokenT< edm::SimVertexContainer >	simVertexInput_

edm::SimVertexContainer	simVertices_

edm::Handle< edm::SimVertexContainer >	simVerticesH_

std::vector< unsigned >	track_ids_

std::map< unsigned int, unsigned int >	trkid_to_index_

bool	verbose_

bool	verboseSimTrack_

Detailed Description

Description: Matching of muon SimHit to SimTrack

Author: Sven Dildick (TAMU), Tao Huang (TAMU)

Definition at line 33 of file MuonSimHitMatcher.h.

Constructor & Destructor Documentation

MuonSimHitMatcher::MuonSimHitMatcher	(	const edm::ParameterSet &	iPS,
		edm::ConsumesCollector &&	iC
	)

Definition at line 7 of file MuonSimHitMatcher.cc.

References edm::ParameterSet::getParameterSet(), muonSimHitMatcherPSet::simTrack, and ecalDetailedTimeRecHit_cfi::simVertex.

                                                                 {
   const auto& simVertex = ps.getParameterSet("simVertex");
   const auto& simTrack = ps.getParameterSet("simTrack");
   verboseSimTrack_ = simTrack.getParameter<int>("verbose");
 
   simVertexInput_ = iC.consumes<edm::SimVertexContainer>(
       simVertex.getParameter<edm::InputTag>("inputTag"));
   simTrackInput_ = iC.consumes<edm::SimTrackContainer>(
       simTrack.getParameter<edm::InputTag>("inputTag"));
 }

MuonSimHitMatcher::~MuonSimHitMatcher ( )

inline

Definition at line 39 of file MuonSimHitMatcher.h.

References chamberIds(), detIds(), MillePedeFileConverter_cfg::e, getIdsOfSimTrackShower(), hitsInChamber(), hitsInDetId(), init(), match(), gen::n, simHits(), simHitsMeanMomentum(), simHitsMeanPosition(), simHitsMeanPositionStation(), tkConvValidator_cfi::simTracks, HGCalValidator_cfi::simVertices, protons_cff::t, and findQualityFiles::v.

39 {}

Member Function Documentation

std::set<unsigned int> MuonSimHitMatcher::chamberIds ( int type = 0 ) const

Referenced by ~MuonSimHitMatcher().

std::set<unsigned int> MuonSimHitMatcher::detIds ( int type = 0 ) const

Referenced by ~MuonSimHitMatcher().

std::vector< unsigned int > MuonSimHitMatcher::getIdsOfSimTrackShower	(	unsigned	trk_id,
		const edm::SimTrackContainer &	simTracks,
		const edm::SimVertexContainer &	simVertices
	)

protected

Definition at line 51 of file MuonSimHitMatcher.cc.

References TrackValidation_cff::association, SimTrack::noVertex(), mps_fire::result, protons_cff::t, and SimTrack::vertIndex().

Referenced by ~MuonSimHitMatcher().

                                               {
   vector<unsigned int> result;
   result.push_back(initial_trk_id);
 
   if (!simMuOnly_) return result;
 
   for (const auto& t : simTracks_) {
     SimTrack last_trk = t;
     // if (std::abs(t.type()) != 13) continue;
     bool is_child = false;
     while (true) {
       if (last_trk.noVertex()) break;
       if (simVertices_[last_trk.vertIndex()].noParent()) break;
 
       unsigned parentId = simVertices_[last_trk.vertIndex()].parentIndex();
       if (parentId == initial_trk_id) {
         is_child = true;
         break;
       }
 
       const auto& association = trkid_to_index_.find(parentId);
       if (association == trkid_to_index_.end()) break;
 
       last_trk = simTracks_[association->second];
     }
     if (is_child) {
       result.push_back(t.trackId());
     }
   }
   return result;
 }

const edm::PSimHitContainer & MuonSimHitMatcher::hitsInChamber ( unsigned int detid ) const

Definition at line 95 of file MuonSimHitMatcher.cc.

Referenced by ME0SimHitMatcher::match(), RPCSimHitMatcher::match(), CSCSimHitMatcher::nLayersWithHitsInChamber(), DTSimHitMatcher::nSuperLayersWithHitsInChamber(), and ~MuonSimHitMatcher().

                               {
   if (chamber_to_hits_.find(detid) == chamber_to_hits_.end()) return no_hits_;
   return chamber_to_hits_.at(detid);
 }

const edm::PSimHitContainer & MuonSimHitMatcher::hitsInDetId ( unsigned int detid ) const

Definition at line 89 of file MuonSimHitMatcher.cc.

Referenced by RPCSimHitMatcher::hitStripsInDetId(), ME0SimHitMatcher::hitStripsInDetId(), CSCSimHitMatcher::hitStripsInDetId(), GEMSimHitMatcher::hitStripsInDetId(), CSCSimHitMatcher::hitWiregroupsInDetId(), DTSimHitMatcher::hitWiresInDTLayerId(), CSCSimHitMatcher::LocalBendingInChamber(), CSCSimHitMatcher::match(), GEMSimHitMatcher::match(), DTSimHitMatcher::match(), ME0SimHitMatcher::matchSimHitsToSimTrack(), GEMSimHitMatcher::matchSimHitsToSimTrack(), and ~MuonSimHitMatcher().

                               {
   if (detid_to_hits_.find(detid) == detid_to_hits_.end()) return no_hits_;
   return detid_to_hits_.at(detid);
 }

void MuonSimHitMatcher::init	(	const edm::Event &	e,
		const edm::EventSetup &	eventSetup
	)

initialize the event

Definition at line 20 of file MuonSimHitMatcher.cc.

References edm::Event::getByToken().

Referenced by CSCSimHitMatcher::init(), RPCSimHitMatcher::init(), ME0SimHitMatcher::init(), GEMSimHitMatcher::init(), DTSimHitMatcher::init(), and ~MuonSimHitMatcher().

                                                           {
   hasGeometry_ = true;
 
   iEvent.getByToken(simTrackInput_, simTracksH_);
   iEvent.getByToken(simVertexInput_, simVerticesH_);
   iEvent.getByToken(simHitInput_, simHitsH_);
 }

void MuonSimHitMatcher::match	(	const SimTrack &	t,
		const SimVertex &	v
	)

do the matching

Definition at line 30 of file MuonSimHitMatcher.cc.

References protons_cff::t, and CoreSimTrack::trackId().

Referenced by CSCSimHitMatcher::match(), RPCSimHitMatcher::match(), ME0SimHitMatcher::match(), GEMSimHitMatcher::match(), DTSimHitMatcher::match(), and ~MuonSimHitMatcher().

                                                                             {
   simTracks_ = *simTracksH_.product();
   simVertices_ = *simVerticesH_.product();
   simHits_ = *simHitsH_.product();
 
   // fill trkId2Index association:
   int no = 0;
   trkid_to_index_.clear();
   for (const auto& t : simTracks_) {
     trkid_to_index_[t.trackId()] = no;
     no++;
   }
 
   track_ids_ =
       getIdsOfSimTrackShower(track.trackId(), simTracks_, simVertices_);
   if (verboseSimTrack_) {
     edm::LogInfo("MuonSimHitMatcher") << "Printing track_ids" << std::endl;
     for (const auto& id : track_ids_) edm::LogInfo("MuonSimHitMatcher") << "id: " << id << std::endl;
   }
 }

const edm::PSimHitContainer & MuonSimHitMatcher::simHits ( int sub ) const

Definition at line 85 of file MuonSimHitMatcher.cc.

Referenced by ntupleDataFormat.TrackingParticle::bestMatchingTrack(), and ~MuonSimHitMatcher().

                                                                    {
   return hits_;
 }

GlobalVector MuonSimHitMatcher::simHitsMeanMomentum ( const edm::PSimHitContainer & sim_hits ) const

Definition at line 121 of file MuonSimHitMatcher.cc.

References h, gen::n, PV3DBase< T, PVType, FrameType >::x(), PV3DBase< T, PVType, FrameType >::y(), and PV3DBase< T, PVType, FrameType >::z().

Referenced by ~MuonSimHitMatcher().

                                                {
   if (sim_hits.empty()) return GlobalVector();  // point "zero"
 
   float sumx, sumy, sumz;
   sumx = sumy = sumz = 0.f;
   size_t n = 0;
   for (const auto& h : sim_hits) {
     const LocalVector& lv = h.momentumAtEntry();
     const GlobalVector& gv =
         geometry_->idToDet(h.detUnitId())->surface().toGlobal(lv);
     sumx += gv.x();
     sumy += gv.y();
     sumz += gv.z();
     ++n;
   }
   if (n == 0) return GlobalVector();
   return GlobalVector(sumx / n, sumy / n, sumz / n);
 }

GlobalPoint MuonSimHitMatcher::simHitsMeanPosition ( const edm::PSimHitContainer & sim_hits ) const

Definition at line 101 of file MuonSimHitMatcher.cc.

References runTauDisplay::gp, h, gen::n, PV3DBase< T, PVType, FrameType >::x(), PV3DBase< T, PVType, FrameType >::y(), and PV3DBase< T, PVType, FrameType >::z().

Referenced by CSCSimHitMatcher::LocalBendingInChamber(), RPCSimHitMatcher::match(), ME0SimHitMatcher::match(), GEMSimHitMatcher::match(), DTSimHitMatcher::match(), CSCSimHitMatcher::match(), and ~MuonSimHitMatcher().

                                                {
   if (sim_hits.empty()) return GlobalPoint();  // point "zero"
 
   float sumx, sumy, sumz;
   sumx = sumy = sumz = 0.f;
   size_t n = 0;
   for (const auto& h : sim_hits) {
     const LocalPoint& lp = h.entryPoint();
     const GlobalPoint& gp =
         geometry_->idToDet(h.detUnitId())->surface().toGlobal(lp);
     sumx += gp.x();
     sumy += gp.y();
     sumz += gp.z();
     ++n;
   }
   if (n == 0) return GlobalPoint();
   return GlobalPoint(sumx / n, sumy / n, sumz / n);
 }