l1tVertexFinder::InputData Class Reference

#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 32 of file InputData.h.

Constructor & Destructor Documentation

InputData() [1/2]

l1tVertexFinder::InputData::InputData

(

)

Constructor and destructor.

Definition at line 20 of file InputData.cc.

{}

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 22 of file InputData.cc.

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().

                                                                                             {
    // 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() > 2) {
        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() > 2)
      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

~InputData()

l1tVertexFinder::InputData::~InputData

(

)

Definition at line 184 of file InputData.cc.

{}

Member Function Documentation

genPt()

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

inline

Definition at line 71 of file InputData.h.

References genPt_.

{ return genPt_; }

genPt_PU()

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

inline

Definition at line 72 of file InputData.h.

References genPt_PU_.

{ return genPt_PU_; }

getGenVertex()

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

inline

Get primary vertex information (vertex from gen particles)

Definition at line 63 of file InputData.h.

References genVertex_.

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

{ return genVertex_; }

getHepMCVertex()

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

inline

Get primary vertex information (vertex from HepMCProduct)

Definition at line 61 of file InputData.h.

References hepMCVertex_.

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

{ return hepMCVertex_; }

getPileUpVertices()

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

inline

Get pile-up vertices information.

Definition at line 67 of file InputData.h.

References vertices_.

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

{ return vertices_; }

getPrimaryVertex()

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

inline

Get primary vertex information (vertex from tracking particles)

Definition at line 65 of file InputData.h.

References vertex_.

{ return vertex_; }

getRecoPileUpVertices()

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

inline

Get reconstructable pile-up vertices information.

Definition at line 69 of file InputData.h.

References recoVertices_.

{ return recoVertices_; }

getTPPtrToRefMap()

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

inline

Get the TrackingParticle to TP translation map.

Definition at line 59 of file InputData.h.

References tpPtrToRefMap_.

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

{ return tpPtrToRefMap_; }

Member Data Documentation

genPt_

float l1tVertexFinder::InputData::genPt_

private

Definition at line 84 of file InputData.h.

Referenced by genPt().

genPt_PU_

float l1tVertexFinder::InputData::genPt_PU_

private

Definition at line 85 of file InputData.h.

Referenced by genPt_PU().

genVertex_

Vertex l1tVertexFinder::InputData::genVertex_

private

Definition at line 76 of file InputData.h.

Referenced by getGenVertex().

hepMCVertex_

Vertex l1tVertexFinder::InputData::hepMCVertex_

private

Definition at line 75 of file InputData.h.

Referenced by getHepMCVertex().

recoVertices_

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

private

Definition at line 79 of file InputData.h.

Referenced by getRecoPileUpVertices().

tpPtrToRefMap_

TPPtrToRefMap l1tVertexFinder::InputData::tpPtrToRefMap_

private

Definition at line 83 of file InputData.h.

Referenced by getTPPtrToRefMap().

vAllStubs_

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

private

Definition at line 81 of file InputData.h.

vertex_

Vertex l1tVertexFinder::InputData::vertex_

private

Definition at line 77 of file InputData.h.

Referenced by getPrimaryVertex().

vertices_

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

private

Definition at line 78 of file InputData.h.

Referenced by getPileUpVertices().