#include <InputData.h>

Classes
struct	SortVertexByPt

struct	SortVertexByZ0

Public Member Functions
const float	genPt () const

const float	genPt_PU () const

const Vertex &	getGenVertex () const
	Get primary vertex information (vertex from gen particles) More...

const Vertex &	getHepMCVertex () const
	Get primary vertex information (vertex from HepMCProduct) More...

const std::vector< Vertex > &	getPileUpVertices () const
	Get pile-up vertices information. More...

const Vertex &	getPrimaryVertex () const
	Get primary vertex information (vertex from tracking particles) More...

const std::vector< Vertex > &	getRecoPileUpVertices () const
	Get reconstructable pile-up vertices information. More...

const TPPtrToRefMap &	getTPPtrToRefMap () const
	Get the TrackingParticle to TP translation map. More...

	InputData ()
	Constructor and destructor. More...

	InputData (const edm::Event &iEvent, const edm::EventSetup &iSetup, const AnalysisSettings &settings, const edm::EDGetTokenT< edm::HepMCProduct > hepMCToken, const edm::EDGetTokenT< edm::View< reco::GenParticle >> genParticlesToken, const edm::EDGetTokenT< edm::View< TrackingParticle >> tpToken, const edm::EDGetTokenT< edm::ValueMap< l1tVertexFinder::TP >> tpValueMapToken, const edm::EDGetTokenT< DetSetVec > stubToken, const edm::ESGetToken< TrackerTopology, TrackerTopologyRcd > tTopoToken, const edm::ESGetToken< TrackerGeometry, TrackerDigiGeometryRecord > tGeomToken)

	~InputData ()

Private Attributes
float	genPt_

float	genPt_PU_

Vertex	genVertex_

Vertex	hepMCVertex_

std::vector< Vertex >	recoVertices_

TPPtrToRefMap	tpPtrToRefMap_

std::vector< Stub >	vAllStubs_

Vertex	vertex_

std::vector< Vertex >	vertices_

Detailed Description

Definition at line 35 of file InputData.h.

Constructor & Destructor Documentation

◆ InputData() [1/2]

l1tVertexFinder::InputData::InputData ( )

Constructor and destructor.

Definition at line 21 of file InputData.cc.

21 {}

◆ InputData() [2/2]

l1tVertexFinder::InputData::InputData	(	const edm::Event &	iEvent,
		const edm::EventSetup &	iSetup,
		const AnalysisSettings &	settings,
		const edm::EDGetTokenT< edm::HepMCProduct >	hepMCToken,
		const edm::EDGetTokenT< edm::View< reco::GenParticle >>	genParticlesToken,
		const edm::EDGetTokenT< edm::View< TrackingParticle >>	tpToken,
		const edm::EDGetTokenT< edm::ValueMap< l1tVertexFinder::TP >>	tpValueMapToken,
		const edm::EDGetTokenT< DetSetVec >	stubToken,
		const edm::ESGetToken< TrackerTopology, TrackerTopologyRcd >	tTopoToken,
		const edm::ESGetToken< TrackerGeometry, TrackerDigiGeometryRecord >	tGeomToken
	)

Definition at line 23 of file InputData.cc.

                                                                                              {
     // Get the tracker geometry info needed to unpack the stub info.
     const TrackerTopology& tTopo = iSetup.getData(tTopoToken);
     const TrackerGeometry& tGeom = iSetup.getData(tGeomToken);
  
     // Get stub info, by looping over modules and then stubs inside each module.
     // Also get the association map from stubs to tracking particles.
     edm::Handle<DetSetVec> ttStubHandle;
     iEvent.getByToken(stubToken, ttStubHandle);
  
     std::set<DetId> lStubDetIds;
     for (DetSetVec::const_iterator p_module = ttStubHandle->begin(); p_module != ttStubHandle->end(); p_module++) {
       for (DetSet::const_iterator p_ttstub = p_module->begin(); p_ttstub != p_module->end(); p_ttstub++) {
         lStubDetIds.insert(p_ttstub->getDetId());
       }
     }
  
     std::map<DetId, DetId> stubGeoDetIdMap;
     for (auto gd = tGeom.dets().begin(); gd != tGeom.dets().end(); gd++) {
       DetId detid = (*gd)->geographicalId();
       if (detid.subdetId() != StripSubdetector::TOB && detid.subdetId() != StripSubdetector::TID)
         continue;  // only run on OT
       if (!tTopo.isLower(detid))
         continue;                             // loop on the stacks: choose the lower arbitrarily
       DetId stackDetid = tTopo.stack(detid);  // Stub module detid
  
       if (lStubDetIds.count(stackDetid) > 0) {
         assert(stubGeoDetIdMap.count(stackDetid) == 0);
         stubGeoDetIdMap[stackDetid] = detid;
       }
     }
     assert(lStubDetIds.size() == stubGeoDetIdMap.size());
  
     // Get TrackingParticle info
     edm::Handle<edm::View<TrackingParticle>> tpHandle;
     edm::Handle<edm::ValueMap<TP>> tpValueMapHandle;
     iEvent.getByToken(tpToken, tpHandle);
     iEvent.getByToken(tpValueMapToken, tpValueMapHandle);
     edm::ValueMap<TP> tpValueMap = *tpValueMapHandle;
  
     for (unsigned int i = 0; i < tpHandle->size(); i++) {
       if (tpValueMap[tpHandle->refAt(i)].use()) {
         tpPtrToRefMap_[tpHandle->ptrAt(i)] = tpHandle->refAt(i);
       }
     }
  
     // Find the various vertices
     genPt_ = 0.;
     genPt_PU_ = 0.;
     for (const auto& [edmPtr, edmRef] : tpPtrToRefMap_) {
       TP tp = tpValueMap[edmRef];
       if (tp.physicsCollision()) {
         genPt_ += tp->pt();
       } else {
         genPt_PU_ += tp->pt();
       }
       if (settings.debug() > 0) {
         edm::LogInfo("InputData") << "InputData::genPt in the event " << genPt_;
       }
  
       if (tp.useForAlgEff()) {
         vertex_.insert(tp);
       } else if (tp.useForVertexReco()) {
         bool found = false;
         for (unsigned int i = 0; i < vertices_.size(); ++i) {
           if (tp->vz() == vertices_[i].vz()) {
             vertices_[i].insert(tp);
             found = true;
             break;
           }
         }
         if (!found) {
           Vertex vertex(tp->vz());
           vertex.insert(tp);
           vertices_.push_back(vertex);
         }
       }
     }
  
     for (const Vertex& vertex : vertices_) {
       if (vertex.numTracks() >= settings.vx_minTracks())
         recoVertices_.push_back(vertex);
     }
  
     if (settings.debug() > 0)
       edm::LogInfo("InputData") << "InputData::" << vertices_.size() << " pileup vertices in the event, "
                                 << recoVertices_.size() << " reconstructable";
  
     vertex_.computeParameters();
     if (settings.debug() > 2)
       edm::LogInfo("InputData") << "InputData::Vertex " << vertex_.z0() << " containing " << vertex_.numTracks()
                                 << " total pT " << vertex_.pT();
  
     for (unsigned int i = 0; i < vertices_.size(); ++i) {
       vertices_[i].computeParameters();
     }
  
     for (unsigned int i = 0; i < recoVertices_.size(); ++i) {
       recoVertices_[i].computeParameters();
     }
  
     std::sort(vertices_.begin(), vertices_.end(), SortVertexByPt());
     std::sort(recoVertices_.begin(), recoVertices_.end(), SortVertexByPt());
  
     // Form the HepMC and GenParticle based vertices
     edm::Handle<edm::HepMCProduct> HepMCEvt;
     iEvent.getByToken(hepMCToken, HepMCEvt);
  
     edm::Handle<edm::View<reco::GenParticle>> GenParticleHandle;
     iEvent.getByToken(genParticlesToken, GenParticleHandle);
  
     if (!HepMCEvt.isValid() && !GenParticleHandle.isValid()) {
       throw cms::Exception("Neither the edm::HepMCProduct nor the generator particles are available.");
     }
     if (HepMCEvt.isValid()) {
       const HepMC::GenEvent* MCEvt = HepMCEvt->GetEvent();
       for (HepMC::GenEvent::vertex_const_iterator ivertex = MCEvt->vertices_begin(); ivertex != MCEvt->vertices_end();
            ++ivertex) {
         bool hasParentVertex = false;
  
         // Loop over the parents looking to see if they are coming from a production vertex
         for (HepMC::GenVertex::particle_iterator iparent = (*ivertex)->particles_begin(HepMC::parents);
              iparent != (*ivertex)->particles_end(HepMC::parents);
              ++iparent)
           if ((*iparent)->production_vertex()) {
             hasParentVertex = true;
             break;
           }
  
         // Reject those vertices with parent vertices
         if (hasParentVertex)
           continue;
         // Get the position of the vertex
         HepMC::FourVector pos = (*ivertex)->position();
         const double mm = 0.1;  // [mm] --> [cm]
         hepMCVertex_ = Vertex(pos.z() * mm);
         break;  // there should be one single primary vertex
       }         // end loop over gen vertices
     }
     if (GenParticleHandle.isValid()) {
       for (const auto& genpart : *GenParticleHandle) {
         if ((genpart.status() != 1) || (genpart.numberOfMothers() == 0))  // not stable or one of the incoming hadrons
           continue;
         genVertex_ = Vertex(genpart.vz());
         break;
       }
     }
     if ((hepMCVertex_.vz() == 0.0) && (genVertex_.vz() == 0.0)) {
       throw cms::Exception("Neither the HepMC vertex nor the generator particle vertex were found.");
     }
  
   }  // end InputData::InputData

References cms::cuda::assert(), l1tVertexFinder::AlgoSettings::debug(), TrackerGeometry::dets(), Exception, newFWLiteAna::found, edm::EventSetup::getData(), mps_fire::i, iEvent, TrackerTopology::isLower(), edm::HandleBase::isValid(), parents, jetUpdater_cfi::sort, TrackerTopology::stack(), DetId::subdetId(), StripSubdetector::TID, StripSubdetector::TOB, cmsswSequenceInfo::tp, bphysicsOniaDQM_cfi::vertex, HltBtagValidation_cff::Vertex, and l1tVertexFinder::AlgoSettings::vx_minTracks().

◆ ~InputData()

l1tVertexFinder::InputData::~InputData ( )

Definition at line 185 of file InputData.cc.

185 {}

Member Function Documentation

◆ genPt()

const float l1tVertexFinder::InputData::genPt ( ) const

inline

Definition at line 74 of file InputData.h.

74 { return genPt_; }

References genPt_.

◆ genPt_PU()

const float l1tVertexFinder::InputData::genPt_PU ( ) const

inline

Definition at line 75 of file InputData.h.

75 { return genPt_PU_; }

References genPt_PU_.

◆ getGenVertex()

const Vertex& l1tVertexFinder::InputData::getGenVertex ( ) const

inline

Get primary vertex information (vertex from gen particles)

Definition at line 66 of file InputData.h.

66 { return genVertex_; }

References genVertex_.

Referenced by l1tVertexFinder::VertexNTupler::analyze().

◆ getHepMCVertex()

const Vertex& l1tVertexFinder::InputData::getHepMCVertex ( ) const

inline

Get primary vertex information (vertex from HepMCProduct)

Definition at line 64 of file InputData.h.

64 { return hepMCVertex_; }

References hepMCVertex_.

Referenced by l1tVertexFinder::VertexNTupler::analyze().

◆ getPileUpVertices()

const std::vector<Vertex>& l1tVertexFinder::InputData::getPileUpVertices ( ) const

inline

Get pile-up vertices information.

Definition at line 70 of file InputData.h.

70 { return vertices_; }

References vertices_.

Referenced by l1tVertexFinder::VertexNTupler::analyze().

◆ getPrimaryVertex()

const Vertex& l1tVertexFinder::InputData::getPrimaryVertex ( ) const

inline

Get primary vertex information (vertex from tracking particles)

Definition at line 68 of file InputData.h.

68 { return vertex_; }

References vertex_.

◆ getRecoPileUpVertices()

const std::vector<Vertex>& l1tVertexFinder::InputData::getRecoPileUpVertices ( ) const

inline

Get reconstructable pile-up vertices information.

Definition at line 72 of file InputData.h.

72 { return recoVertices_; }

References recoVertices_.

◆ getTPPtrToRefMap()

const TPPtrToRefMap& l1tVertexFinder::InputData::getTPPtrToRefMap ( ) const

inline

Get the TrackingParticle to TP translation map.

Definition at line 62 of file InputData.h.

62 { return tpPtrToRefMap_; }

References tpPtrToRefMap_.

Referenced by l1tVertexFinder::VertexNTupler::analyze().

Member Data Documentation

◆ genPt_

float l1tVertexFinder::InputData::genPt_

private

Definition at line 87 of file InputData.h.

Referenced by genPt().

◆ genPt_PU_

float l1tVertexFinder::InputData::genPt_PU_

private

Definition at line 88 of file InputData.h.

Referenced by genPt_PU().

◆ genVertex_

Vertex l1tVertexFinder::InputData::genVertex_

private

Definition at line 79 of file InputData.h.

Referenced by getGenVertex().

◆ hepMCVertex_

Vertex l1tVertexFinder::InputData::hepMCVertex_

private

Definition at line 78 of file InputData.h.

Referenced by getHepMCVertex().

◆ recoVertices_

std::vector<Vertex> l1tVertexFinder::InputData::recoVertices_

private

Definition at line 82 of file InputData.h.

Referenced by getRecoPileUpVertices().

◆ tpPtrToRefMap_

TPPtrToRefMap l1tVertexFinder::InputData::tpPtrToRefMap_

private

Definition at line 86 of file InputData.h.

Referenced by getTPPtrToRefMap().

◆ vAllStubs_

std::vector<Stub> l1tVertexFinder::InputData::vAllStubs_

private

Definition at line 84 of file InputData.h.

◆ vertex_

Vertex l1tVertexFinder::InputData::vertex_

private

Definition at line 80 of file InputData.h.

Referenced by getPrimaryVertex().

◆ vertices_

std::vector<Vertex> l1tVertexFinder::InputData::vertices_

private

Definition at line 81 of file InputData.h.

Referenced by getPileUpVertices().

Classes

Public Member Functions

Private Attributes

Detailed Description

Constructor & Destructor Documentation

◆ InputData() [1/2]

◆ InputData() [2/2]

◆ ~InputData()

Member Function Documentation

◆ genPt()

◆ genPt_PU()

◆ getGenVertex()

◆ getHepMCVertex()

◆ getPileUpVertices()

◆ getPrimaryVertex()

◆ getRecoPileUpVertices()

◆ getTPPtrToRefMap()

Member Data Documentation

◆ genPt_

◆ genPt_PU_

◆ genVertex_

◆ hepMCVertex_

◆ recoVertices_

◆ tpPtrToRefMap_

◆ vAllStubs_

◆ vertex_

◆ vertices_