Inheritance diagram for l1tVertexFinder::VertexNTupler:

Classes
struct	EmulationVerticesBranchData

struct	GenJetsBranchData

struct	GenParticlesBranchData

struct	RecoTracksBranchData

struct	RecoVerticesBranchData

struct	TrueTracksBranchData

Public Member Functions
	VertexNTupler (const edm::ParameterSet &)

	~VertexNTupler () override

Public Member Functions inherited from edm::one::EDAnalyzer< edm::one::SharedResources >
	EDAnalyzer ()=default

	EDAnalyzer (const EDAnalyzer &)=delete

SerialTaskQueue *	globalLuminosityBlocksQueue () final

SerialTaskQueue *	globalRunsQueue () final

const EDAnalyzer &	operator= (const EDAnalyzer &)=delete

bool	wantsGlobalLuminosityBlocks () const final

bool	wantsGlobalRuns () const final

bool	wantsInputProcessBlocks () const final

bool	wantsProcessBlocks () const final

Public Member Functions inherited from edm::one::EDAnalyzerBase
void	callWhenNewProductsRegistered (std::function< void(BranchDescription const &)> const &func)

	EDAnalyzerBase ()

ModuleDescription const &	moduleDescription () const

bool	wantsStreamLuminosityBlocks () const

bool	wantsStreamRuns () const

	~EDAnalyzerBase () override

Public Member Functions inherited from edm::EDConsumerBase
std::vector< ConsumesInfo >	consumesInfo () const

void	convertCurrentProcessAlias (std::string const &processName)
	Convert "@currentProcess" in InputTag process names to the actual current process name. More...

	EDConsumerBase ()

	EDConsumerBase (EDConsumerBase const &)=delete

	EDConsumerBase (EDConsumerBase &&)=default

ESProxyIndex const *	esGetTokenIndices (edm::Transition iTrans) const

std::vector< ESProxyIndex > const &	esGetTokenIndicesVector (edm::Transition iTrans) const

std::vector< ESRecordIndex > const &	esGetTokenRecordIndicesVector (edm::Transition iTrans) const

ProductResolverIndexAndSkipBit	indexFrom (EDGetToken, BranchType, TypeID const &) const

void	itemsMayGet (BranchType, std::vector< ProductResolverIndexAndSkipBit > &) const

void	itemsToGet (BranchType, std::vector< ProductResolverIndexAndSkipBit > &) const

std::vector< ProductResolverIndexAndSkipBit > const &	itemsToGetFrom (BranchType iType) const

void	labelsForToken (EDGetToken iToken, Labels &oLabels) const

void	modulesWhoseProductsAreConsumed (std::array< std::vector< ModuleDescription const > , NumBranchTypes > &modulesAll, std::vector< ModuleProcessName > &modulesInPreviousProcesses, ProductRegistry const &preg, std::map< std::string, ModuleDescription const *> const &labelsToDesc, std::string const &processName) const

EDConsumerBase const &	operator= (EDConsumerBase const &)=delete

EDConsumerBase &	operator= (EDConsumerBase &&)=default

bool	registeredToConsume (ProductResolverIndex, bool, BranchType) const

bool	registeredToConsumeMany (TypeID const &, BranchType) const

void	selectInputProcessBlocks (ProductRegistry const &productRegistry, ProcessBlockHelperBase const &processBlockHelperBase)

ProductResolverIndexAndSkipBit	uncheckedIndexFrom (EDGetToken) const

void	updateLookup (BranchType iBranchType, ProductResolverIndexHelper const &, bool iPrefetchMayGet)

void	updateLookup (eventsetup::ESRecordsToProxyIndices const &)

virtual	~EDConsumerBase () noexcept(false)

Private Types
typedef TTTrackAssociationMap< Ref_Phase2TrackerDigi_ >	TTTrackAssMap

typedef edm::View< TTTrack< Ref_Phase2TrackerDigi_ > >	TTTrackCollectionView

Private Member Functions
void	analyze (const edm::Event &evt, const edm::EventSetup &setup) override

void	beginJob () override

void	endJob () override

Private Attributes
const edm::EDGetTokenT< std::vector< l1tVertexFinder::TP > >	allMatchedTPsToken_

GenJetsBranchData	genJetsBranchData_

const edm::EDGetTokenT< std::vector< reco::GenJet > >	genJetsToken_

GenParticlesBranchData	genParticlesHardOutgoingBranchData_

const edm::EDGetTokenT< edm::View< reco::GenParticle > >	genParticlesToken_

float	genVtxZ0_

float	hepMCVtxZ0_

const edm::EDGetTokenT< l1tVertexFinder::InputData >	inputDataToken_

std::map< std::string, RecoTracksBranchData >	l1TracksBranchData_

std::map< std::string, edm::EDGetTokenT< TTTrackAssociationMap< Ref_Phase2TrackerDigi_ > > >	l1TracksMapTokenMap_

std::map< std::string, edm::EDGetTokenT< TTTrackCollectionView > >	l1TracksTokenMap_

std::vector< std::vector< unsigned > >	l1Vertices_extra_numTracks_

std::vector< std::vector< float > >	l1Vertices_extra_sumPt_

std::vector< std::vector< float > >	l1Vertices_extra_z0_

std::vector< std::vector< float > >	l1Vertices_extra_z0_etaWeighted_

std::map< std::string, RecoVerticesBranchData >	l1VerticesBranchData_

std::map< std::string, EmulationVerticesBranchData >	l1VerticesEmulationBranchData_

std::map< std::string, edm::EDGetTokenT< std::vector< l1t::VertexWord > > >	l1VerticesEmulationTokenMap_

std::vector< edm::EDGetTokenT< std::vector< l1t::Vertex > > >	l1VerticesExtraTokens_

std::map< std::string, std::string >	l1VerticesInputMap_

std::map< std::string, edm::EDGetTokenT< std::vector< l1t::Vertex > > >	l1VerticesTokenMap_

int	numPileupVertices_

float	numTrueInteractions_

TTree *	outputTree_

const edm::EDGetTokenT< std::vector< PileupSummaryInfo > >	pileupSummaryToken_

const bool	printResults_

AnalysisSettings	settings_

std::vector< float >	truePileUpVtxZ0_

TrueTracksBranchData	trueTracksBranchData_

const edm::EDGetTokenT< edm::ValueMap< l1tVertexFinder::TP > >	vTPsToken_

Additional Inherited Members
Public Types inherited from edm::one::EDAnalyzerBase
typedef EDAnalyzerBase	ModuleType

Public Types inherited from edm::EDConsumerBase
typedef ProductLabels	Labels

Static Public Member Functions inherited from edm::one::EDAnalyzerBase
static const std::string &	baseType ()

static void	fillDescriptions (ConfigurationDescriptions &descriptions)

static void	prevalidate (ConfigurationDescriptions &descriptions)

Protected Member Functions inherited from edm::EDConsumerBase
template<typename ProductType , BranchType B = InEvent>
EDGetTokenT< ProductType >	consumes (edm::InputTag const &tag)

template<BranchType B = InEvent>
EDConsumerBaseAdaptor< B >	consumes (edm::InputTag tag) noexcept

EDGetToken	consumes (const TypeToGet &id, edm::InputTag const &tag)

template<BranchType B>
EDGetToken	consumes (TypeToGet const &id, edm::InputTag const &tag)

ConsumesCollector	consumesCollector ()
	Use a ConsumesCollector to gather consumes information from helper functions. More...

template<typename ProductType , BranchType B = InEvent>
void	consumesMany ()

void	consumesMany (const TypeToGet &id)

template<BranchType B>
void	consumesMany (const TypeToGet &id)

template<typename ESProduct , typename ESRecord , Transition Tr = Transition::Event>
auto	esConsumes ()

template<typename ESProduct , typename ESRecord , Transition Tr = Transition::Event>
auto	esConsumes (ESInputTag const &tag)

template<Transition Tr = Transition::Event>
constexpr auto	esConsumes ()

template<Transition Tr = Transition::Event>
auto	esConsumes (ESInputTag tag)

template<Transition Tr = Transition::Event>
ESGetTokenGeneric	esConsumes (eventsetup::EventSetupRecordKey const &iRecord, eventsetup::DataKey const &iKey)
	Used with EventSetupRecord::doGet. More...

template<typename ProductType , BranchType B = InEvent>
EDGetTokenT< ProductType >	mayConsume (edm::InputTag const &tag)

EDGetToken	mayConsume (const TypeToGet &id, edm::InputTag const &tag)

template<BranchType B>
EDGetToken	mayConsume (const TypeToGet &id, edm::InputTag const &tag)

void	resetItemsToGetFrom (BranchType iType)

Detailed Description

Definition at line 43 of file VertexNTupler.cc.

Member Typedef Documentation

◆ TTTrackAssMap

typedef TTTrackAssociationMap<Ref_Phase2TrackerDigi_> l1tVertexFinder::VertexNTupler::TTTrackAssMap

private

Definition at line 165 of file VertexNTupler.cc.

◆ TTTrackCollectionView

typedef edm::View<TTTrack<Ref_Phase2TrackerDigi_> > l1tVertexFinder::VertexNTupler::TTTrackCollectionView

private

Definition at line 166 of file VertexNTupler.cc.

Constructor & Destructor Documentation

◆ VertexNTupler()

l1tVertexFinder::VertexNTupler::VertexNTupler ( const edm::ParameterSet & iConfig )

explicit

Definition at line 209 of file VertexNTupler.cc.

       : inputDataToken_(consumes<l1tVertexFinder::InputData>(iConfig.getParameter<edm::InputTag>("inputDataInputTag"))),
         pileupSummaryToken_(consumes<std::vector<PileupSummaryInfo>>(edm::InputTag("addPileupInfo"))),
         genParticlesToken_(
             consumes<edm::View<reco::GenParticle>>(iConfig.getParameter<edm::InputTag>("genParticleInputTag"))),
         genJetsToken_(consumes<std::vector<reco::GenJet>>(iConfig.getParameter<edm::InputTag>("genJetsInputTag"))),
         allMatchedTPsToken_(
             consumes<std::vector<l1tVertexFinder::TP>>(iConfig.getParameter<edm::InputTag>("l1TracksTPInputTags"))),
         vTPsToken_(consumes<edm::ValueMap<l1tVertexFinder::TP>>(
             iConfig.getParameter<edm::InputTag>("l1TracksTPValueMapInputTags"))),
         printResults_(iConfig.getParameter<bool>("printResults")),
         settings_(iConfig) {
     const std::vector<std::string> trackBranchNames(
         iConfig.getParameter<std::vector<std::string>>("l1TracksBranchNames"));
     const std::vector<edm::InputTag> trackInputTags(
         iConfig.getParameter<std::vector<edm::InputTag>>("l1TracksInputTags"));
     const std::vector<edm::InputTag> trackMapInputTags(
         iConfig.getParameter<std::vector<edm::InputTag>>("l1TracksTruthMapInputTags"));
 
     if (trackBranchNames.size() != trackInputTags.size())
       throw cms::Exception("The number of track branch names (" + std::to_string(trackBranchNames.size()) +
                            ") specified in the config does not match the number of input tags (" +
                            std::to_string(trackInputTags.size()) + ")");
     if (trackBranchNames.size() != trackMapInputTags.size())
       throw cms::Exception("The number of track branch names (" + std::to_string(trackBranchNames.size()) +
                            ") specified in the config does not match the number of track map input tags (" +
                            std::to_string(trackMapInputTags.size()) + ")");
 
     const std::vector<std::string> vertexBranchNames(
         iConfig.getParameter<std::vector<std::string>>("l1VertexBranchNames"));
     const std::vector<edm::InputTag> vertexInputTags(
         iConfig.getParameter<std::vector<edm::InputTag>>("l1VertexInputTags"));
     const std::vector<std::string> vertexTrackNames(
         iConfig.getParameter<std::vector<std::string>>("l1VertexTrackInputs"));
 
     if (vertexBranchNames.size() != vertexInputTags.size())
       throw cms::Exception("The number of vertex branch names (" + std::to_string(vertexBranchNames.size()) +
                            ") specified in the config does not match the number of input tags (" +
                            std::to_string(vertexInputTags.size()) + ")");
     if (vertexBranchNames.size() != vertexTrackNames.size())
       throw cms::Exception(
           "The number of vertex branch names (" + std::to_string(vertexBranchNames.size()) +
           ") specified in the config does not match the number of associated input track collection names (" +
           std::to_string(vertexTrackNames.size()) + ")");
 
     const std::vector<std::string> emulationVertexBranchNames(
         iConfig.getParameter<std::vector<std::string>>("emulationVertexBranchNames"));
     const std::vector<edm::InputTag> emulationVertexInputTags(
         iConfig.getParameter<std::vector<edm::InputTag>>("emulationVertexInputTags"));
 
     const std::vector<edm::InputTag> extraVertexInputTags(
         iConfig.getParameter<std::vector<edm::InputTag>>("extraL1VertexInputTags"));
     const std::vector<std::string> extraVertexDescriptions(
         iConfig.getParameter<std::vector<std::string>>("extraL1VertexDescriptions"));
 
     usesResource(TFileService::kSharedResource);
     edm::Service<TFileService> fs;
     outputTree_ = fs->make<TTree>("l1VertexReco", "L1 vertex-related info");
 
     std::vector<std::string>::const_iterator branchNameIt = emulationVertexBranchNames.begin();
     std::vector<edm::InputTag>::const_iterator inputTagIt = emulationVertexInputTags.begin();
     for (; branchNameIt != emulationVertexBranchNames.end(); branchNameIt++, inputTagIt++) {
       l1VerticesEmulationTokenMap_[*branchNameIt] = consumes<std::vector<l1t::VertexWord>>(*inputTagIt);
       l1VerticesEmulationBranchData_[*branchNameIt] = EmulationVerticesBranchData();
       EmulationVerticesBranchData& branchData = l1VerticesEmulationBranchData_.at(*branchNameIt);
 
       outputTree_->Branch(("emulationVertices_" + *branchNameIt + "_numTracks").c_str(), &branchData.numTracks);
       outputTree_->Branch(("emulationVertices_" + *branchNameIt + "_z0").c_str(), &branchData.z0);
       outputTree_->Branch(("emulationVertices_" + *branchNameIt + "_sumPt").c_str(), &branchData.sumPt);
     }
 
     outputTree_->Branch("genJets_energy", &genJetsBranchData_.energy);
     outputTree_->Branch("genJets_pt", &genJetsBranchData_.pt);
     outputTree_->Branch("genJets_eta", &genJetsBranchData_.eta);
     outputTree_->Branch("genJets_phi", &genJetsBranchData_.phi);
     outputTree_->Branch("genParticles_hardProcOutgoing_energy", &genParticlesHardOutgoingBranchData_.energy);
     outputTree_->Branch("genParticles_hardProcOutgoing_pt", &genParticlesHardOutgoingBranchData_.pt);
     outputTree_->Branch("genParticles_hardProcOutgoing_eta", &genParticlesHardOutgoingBranchData_.eta);
     outputTree_->Branch("genParticles_hardProcOutgoing_phi", &genParticlesHardOutgoingBranchData_.phi);
     outputTree_->Branch("genParticles_hardProcOutgoing_pdgId", &genParticlesHardOutgoingBranchData_.pdgId);
     outputTree_->Branch("genParticles_hardProcOutgoing_status", &genParticlesHardOutgoingBranchData_.status);
     outputTree_->Branch("genVertex_z0", &genVtxZ0_);
     outputTree_->Branch("hepMCVertex_z0", &hepMCVtxZ0_);
     outputTree_->Branch("pileupSummary_trueNumInteractions", &numTrueInteractions_);
     outputTree_->Branch("pileupSummary_numPileupVertices", &numPileupVertices_);
 
     std::vector<std::string>::const_iterator trackBranchNameIt = trackBranchNames.begin();
     std::vector<edm::InputTag>::const_iterator trackInputTagIt = trackInputTags.begin();
     std::vector<edm::InputTag>::const_iterator trackMapInputTagIt = trackMapInputTags.begin();
     for (; trackBranchNameIt != trackBranchNames.end(); trackBranchNameIt++, trackInputTagIt++, trackMapInputTagIt++) {
       l1TracksTokenMap_[*trackBranchNameIt] = consumes<TTTrackCollectionView>(*trackInputTagIt);
       l1TracksMapTokenMap_[*trackBranchNameIt] =
           consumes<TTTrackAssociationMap<Ref_Phase2TrackerDigi_>>(*trackMapInputTagIt);
 
       RecoTracksBranchData& branchData = l1TracksBranchData_[*trackBranchNameIt];
 
       outputTree_->Branch(("recoTracks_" + *trackBranchNameIt + "_pt").c_str(), &branchData.pt);
       outputTree_->Branch(("recoTracks_" + *trackBranchNameIt + "_eta").c_str(), &branchData.eta);
       outputTree_->Branch(("recoTracks_" + *trackBranchNameIt + "_phi").c_str(), &branchData.phi);
       outputTree_->Branch(("recoTracks_" + *trackBranchNameIt + "_z0").c_str(), &branchData.z0);
       outputTree_->Branch(("recoTracks_" + *trackBranchNameIt + "_numStubs").c_str(), &branchData.numStubs);
       outputTree_->Branch(("recoTracks_" + *trackBranchNameIt + "_chi2dof").c_str(), &branchData.chi2dof);
       outputTree_->Branch(("recoTracks_" + *trackBranchNameIt + "_trueMatchIdx").c_str(), &branchData.trueMatchIdx);
       outputTree_->Branch(("recoTracks_" + *trackBranchNameIt + "_truthMap_matchIdx").c_str(),
                           &branchData.truthMapMatchIdx);
       outputTree_->Branch(("recoTracks_" + *trackBranchNameIt + "_truthMap_isGenuine").c_str(),
                           &branchData.truthMapIsGenuine);
       outputTree_->Branch(("recoTracks_" + *trackBranchNameIt + "_truthMap_isLooselyGenuine").c_str(),
                           &branchData.truthMapIsLooselyGenuine);
       outputTree_->Branch(("recoTracks_" + *trackBranchNameIt + "_truthMap_isCombinatoric").c_str(),
                           &branchData.truthMapIsCombinatoric);
       outputTree_->Branch(("recoTracks_" + *trackBranchNameIt + "_truthMap_isUnknown").c_str(),
                           &branchData.truthMapIsUnknown);
     }
 
     branchNameIt = vertexBranchNames.begin();
     inputTagIt = vertexInputTags.begin();
     std::vector<std::string>::const_iterator l1VertexTrackNameIt = vertexTrackNames.begin();
     for (; branchNameIt != vertexBranchNames.end(); branchNameIt++, inputTagIt++, l1VertexTrackNameIt++) {
       l1VerticesTokenMap_[*branchNameIt] = consumes<std::vector<l1t::Vertex>>(*inputTagIt);
       l1VerticesBranchData_[*branchNameIt] = RecoVerticesBranchData();
       RecoVerticesBranchData& branchData = l1VerticesBranchData_.at(*branchNameIt);
       l1VerticesInputMap_[*branchNameIt] = *l1VertexTrackNameIt;
 
       if (l1TracksTokenMap_.count(*l1VertexTrackNameIt) == 0)
         throw cms::Exception("Invalid track collection name '" + *l1VertexTrackNameIt +
                              "' specified as input to vertex collection '" + *branchNameIt + "'");
 
       outputTree_->Branch(("recoVertices_" + *branchNameIt + "_numTracks").c_str(), &branchData.numTracks);
       outputTree_->Branch(("recoVertices_" + *branchNameIt + "_trackIdxs").c_str(), &branchData.trackIdxs);
       outputTree_->Branch(("recoVertices_" + *branchNameIt + "_z0").c_str(), &branchData.z0);
       outputTree_->Branch(("recoVertices_" + *branchNameIt + "_sumPt").c_str(), &branchData.sumPt);
     }
 
     outputTree_->Branch("truePileUpVertices_z0", &truePileUpVtxZ0_);
     outputTree_->Branch("trueTracks_pt", &trueTracksBranchData_.pt);
     outputTree_->Branch("trueTracks_eta", &trueTracksBranchData_.eta);
     outputTree_->Branch("trueTracks_phi", &trueTracksBranchData_.phi);
     outputTree_->Branch("trueTracks_z0", &trueTracksBranchData_.z0);
     outputTree_->Branch("trueTracks_pdgId", &trueTracksBranchData_.pdgId);
     outputTree_->Branch("trueTracks_physCollision", &trueTracksBranchData_.physCollision);
     outputTree_->Branch("trueTracks_use", &trueTracksBranchData_.use);
     outputTree_->Branch("trueTracks_useForEff", &trueTracksBranchData_.useForEff);
     outputTree_->Branch("trueTracks_useForAlgEff", &trueTracksBranchData_.useForAlgEff);
     outputTree_->Branch("trueTracks_useForVtxReco", &trueTracksBranchData_.useForVertexReco);
 
     for (const auto& inputTag : extraVertexInputTags)
       l1VerticesExtraTokens_.push_back(consumes<std::vector<l1t::Vertex>>(inputTag));
     TObjArray* descriptionArray = new TObjArray();
     for (const auto& description : extraVertexDescriptions)
       descriptionArray->Add(new TObjString(description.c_str()));
     outputTree_->GetUserInfo()->Add(descriptionArray);
     outputTree_->Branch("vertices_extra_numTracks", &l1Vertices_extra_numTracks_);
     outputTree_->Branch("vertices_extra_z0", &l1Vertices_extra_z0_);
     outputTree_->Branch("vertices_extra_z0_etaWeighted", &l1Vertices_extra_z0_etaWeighted_);
     outputTree_->Branch("vertices_extra_sumPt", &l1Vertices_extra_sumPt_);
   }

◆ ~VertexNTupler()

l1tVertexFinder::VertexNTupler::~VertexNTupler ( )

override

Definition at line 669 of file VertexNTupler.cc.

669 {}

Member Function Documentation

◆ analyze()

void l1tVertexFinder::VertexNTupler::analyze	(	const edm::Event &	evt,
		const edm::EventSetup &	setup
	)

overrideprivatevirtual

Implements edm::one::EDAnalyzerBase.

Definition at line 413 of file VertexNTupler.cc.

                                                                                  {
     // Note useful info about MC truth particles and about reconstructed stubs
     edm::Handle<l1tVertexFinder::InputData> inputDataHandle;
     iEvent.getByToken(inputDataToken_, inputDataHandle);
     InputData inputData = *inputDataHandle;
 
     edm::Handle<edm::ValueMap<TP>> tpValueMapHandle;
     iEvent.getByToken(vTPsToken_, tpValueMapHandle);
     edm::ValueMap<TP> tpValueMap = *tpValueMapHandle;
 
     // Create collections to hold the desired objects
     std::map<std::string, std::vector<L1TrackTruthMatched>> l1TrackCollections;
     std::map<std::string, edm::Handle<TTTrackAssMap>> truthAssocMapHandles;
 
     // Create iterators for the maps
     auto tokenMapEntry = l1TracksTokenMap_.begin(), tokenMapEntryEnd = l1TracksTokenMap_.end();
     auto mapTokenMapEntry = l1TracksMapTokenMap_.begin(), mapTokenMapEntryEnd = l1TracksMapTokenMap_.end();
 
     // Iterate over the maps
     for (; (tokenMapEntry != tokenMapEntryEnd) && (mapTokenMapEntry != mapTokenMapEntryEnd);
          ++tokenMapEntry, ++mapTokenMapEntry) {
       edm::Handle<TTTrackCollectionView> l1TracksHandle;
       edm::Handle<TTTrackAssMap>& mcTruthTTTrackHandle = truthAssocMapHandles[mapTokenMapEntry->first];
       iEvent.getByToken(tokenMapEntry->second, l1TracksHandle);
       iEvent.getByToken(l1TracksMapTokenMap_.at(mapTokenMapEntry->first), mcTruthTTTrackHandle);
 
       std::vector<L1TrackTruthMatched>& l1Tracks = l1TrackCollections[tokenMapEntry->first];
       l1Tracks.reserve(l1TracksHandle->size());
       for (const auto& track : l1TracksHandle->ptrs()) {
         l1Tracks.push_back(L1TrackTruthMatched(track, inputData.getTPPtrToRefMap(), tpValueMap, mcTruthTTTrackHandle));
       }
     }
 
     // create a map for associating fat reco tracks with their underlying
     // TTTrack pointers
     std::map<std::string, std::map<const edm::Ptr<TTTrack<Ref_Phase2TrackerDigi_>>, const L1TrackTruthMatched*>>
         edmL1TrackMaps;
 
     // get a list of reconstructed tracks with references to their TPs
     for (const auto& entry : l1TrackCollections) {
       auto& edmL1Map = edmL1TrackMaps[entry.first];
       for (const auto& track : entry.second) {
         edmL1Map.insert(std::pair<const edm::Ptr<TTTrack<Ref_Phase2TrackerDigi_>>, const L1TrackTruthMatched*>(
             track.getTTTrackPtr(), &track));
       }
     }
 
 
     // Pile-up summary info
     edm::Handle<std::vector<PileupSummaryInfo>> pileupHandle;
     iEvent.getByToken(pileupSummaryToken_, pileupHandle);
 
     for (auto bxIt = pileupHandle->begin(); bxIt != pileupHandle->end(); bxIt++) {
       if (bxIt->getBunchCrossing() == 0) {
         numTrueInteractions_ = bxIt->getTrueNumInteractions();
         numPileupVertices_ = bxIt->getPU_NumInteractions();
       }
     }
 
     // True track info
     trueTracksBranchData_.clear();
     edm::Handle<std::vector<l1tVertexFinder::TP>> allMatchedTPsHandle;
     iEvent.getByToken(allMatchedTPsToken_, allMatchedTPsHandle);
 
     for (const auto& tp : *allMatchedTPsHandle) {
       trueTracksBranchData_.pt.push_back(tp->pt());
       trueTracksBranchData_.eta.push_back(tp->eta());
       trueTracksBranchData_.phi.push_back(tp->phi());
       trueTracksBranchData_.z0.push_back(tp->z0());
       trueTracksBranchData_.pdgId.push_back(tp->pdgId());
       trueTracksBranchData_.physCollision.push_back(tp.physicsCollision() ? 1.0 : 0.0);
       trueTracksBranchData_.use.push_back(tp.use() ? 1.0 : 0.0);
       trueTracksBranchData_.useForEff.push_back(tp.useForEff() ? 1.0 : 0.0);
       trueTracksBranchData_.useForAlgEff.push_back(tp.useForAlgEff() ? 1.0 : 0.0);
       trueTracksBranchData_.useForVertexReco.push_back(tp.useForVertexReco() ? 1.0 : 0.0);
     }
 
     // True pile-up vertex info
     truePileUpVtxZ0_.clear();
     for (const Vertex& vtx : inputData.getPileUpVertices())
       truePileUpVtxZ0_.push_back(vtx.z0());
 
     // Generator level vertex info
     hepMCVtxZ0_ = inputData.getHepMCVertex().vz();
     genVtxZ0_ = inputData.getGenVertex().vz();
 
     // Gen particles
     genParticlesHardOutgoingBranchData_.clear();
     edm::Handle<edm::View<reco::GenParticle>> genParticlesH;
     iEvent.getByToken(genParticlesToken_, genParticlesH);
     for (const auto& p : *genParticlesH) {
       genParticlesHardOutgoingBranchData_.energy.push_back(p.energy());
       genParticlesHardOutgoingBranchData_.pt.push_back(p.pt());
       genParticlesHardOutgoingBranchData_.eta.push_back(p.eta());
       genParticlesHardOutgoingBranchData_.phi.push_back(p.phi());
       genParticlesHardOutgoingBranchData_.pdgId.push_back(p.pdgId());
       genParticlesHardOutgoingBranchData_.status.push_back(p.status());
     }
 
     // Gen jet (AK4) branches
     genJetsBranchData_.clear();
     edm::Handle<std::vector<reco::GenJet>> genJetsHandle;
     iEvent.getByToken(genJetsToken_, genJetsHandle);
     for (const auto& genJet : *genJetsHandle) {
       genJetsBranchData_.energy.push_back(genJet.energy());
       genJetsBranchData_.pt.push_back(genJet.pt());
       genJetsBranchData_.eta.push_back(genJet.eta());
       genJetsBranchData_.phi.push_back(genJet.phi());
     }
 
     for (const auto& entry : l1TrackCollections) {
       const auto& l1Tracks = entry.second;
       RecoTracksBranchData& branchData = l1TracksBranchData_.at(entry.first);
 
       const TTTrackAssociationMap<Ref_Phase2TrackerDigi_>& truthAssocMap = *truthAssocMapHandles.at(entry.first);
 
       // Reco track branches
       branchData.clear();
       for (const L1TrackTruthMatched& track : l1Tracks) {
         branchData.pt.push_back(track.pt());
         branchData.eta.push_back(track.eta());
         branchData.phi.push_back(track.phi0());
         branchData.z0.push_back(track.z0());
         branchData.numStubs.push_back(track.getNumStubs());
         branchData.chi2dof.push_back(track.chi2dof());
         branchData.trueMatchIdx.push_back(track.getMatchedTPidx());
 
         edm::Ptr<TrackingParticle> matchedTP = truthAssocMap.findTrackingParticlePtr(track.getTTTrackPtr());
         if (matchedTP.isNull())
           branchData.truthMapMatchIdx.push_back(-1);
         else {
           auto it = std::find_if(allMatchedTPsHandle->begin(),
                                  allMatchedTPsHandle->end(),
                                  [&matchedTP](auto const& tp) { return tp.getTrackingParticle() == matchedTP; });
           assert(it != allMatchedTPsHandle->end());
           branchData.truthMapMatchIdx.push_back(std::distance(allMatchedTPsHandle->begin(), it));
         }
         branchData.truthMapIsGenuine.push_back(truthAssocMap.isGenuine(track.getTTTrackPtr()) ? 1.0 : 0.0);
         branchData.truthMapIsLooselyGenuine.push_back(truthAssocMap.isLooselyGenuine(track.getTTTrackPtr()) ? 1.0
                                                                                                             : 0.0);
         branchData.truthMapIsCombinatoric.push_back(truthAssocMap.isCombinatoric(track.getTTTrackPtr()) ? 1.0 : 0.0);
         branchData.truthMapIsUnknown.push_back(truthAssocMap.isUnknown(track.getTTTrackPtr()) ? 1.0 : 0.0);
       }
     }
 
     // Reco vertex branches
     for (const auto& tokenMapEntry : l1VerticesTokenMap_) {
       RecoVerticesBranchData& branchData = l1VerticesBranchData_.at(tokenMapEntry.first);
 
       edm::Handle<std::vector<l1t::Vertex>> handle;
       iEvent.getByToken(tokenMapEntry.second, handle);
       std::vector<std::shared_ptr<const RecoVertexWithTP>> recoVertices;
       recoVertices.reserve(handle->size());
       for (unsigned int i = 0; i < handle->size(); ++i) {
         recoVertices.push_back(std::shared_ptr<const RecoVertexWithTP>(
             new RecoVertexWithTP(handle->at(i), edmL1TrackMaps.at(l1VerticesInputMap_.at(tokenMapEntry.first)))));
       }
 
       branchData.clear();
       std::vector<L1TrackTruthMatched>& l1Tracks = l1TrackCollections.at(l1VerticesInputMap_.at(tokenMapEntry.first));
       for (const std::shared_ptr<const RecoVertexWithTP>& vtx : recoVertices) {
         branchData.numTracks.push_back(vtx->numTracks());
         branchData.trackIdxs.push_back(std::vector<unsigned>());
         for (const L1TrackTruthMatched* track : vtx->tracks())
           branchData.trackIdxs.back().push_back(track - l1Tracks.data());
         branchData.z0.push_back(vtx->z0());
         branchData.sumPt.push_back(vtx->pt());
       }
 
       if (printResults_) {
         edm::LogInfo("VertexNTupler") << "analyze::" << recoVertices.size() << " '" << tokenMapEntry.first
                                       << "' vertices were found ... ";
         for (const auto& vtx : recoVertices) {
           edm::LogInfo("VertexNTupler") << "analyze::"
                                         << "  * z0 = " << vtx->z0() << "; contains " << vtx->numTracks()
                                         << " tracks ...";
           for (const auto& trackPtr : vtx->tracks())
             edm::LogInfo("VertexNTupler") << "analyze::"
                                           << "     - z0 = " << trackPtr->z0() << "; pt = " << trackPtr->pt()
                                           << ", eta = " << trackPtr->eta() << ", phi = " << trackPtr->phi0();
         }
       }
     }
 
     // Emulation vertex branches
     for (const auto& tokenMapEntry : l1VerticesEmulationTokenMap_) {
       EmulationVerticesBranchData& branchData = l1VerticesEmulationBranchData_.at(tokenMapEntry.first);
       branchData.clear();
 
       edm::Handle<std::vector<l1t::VertexWord>> handle;
       iEvent.getByToken(tokenMapEntry.second, handle);
       for (const auto& vtx : *handle) {
         branchData.numTracks.push_back(vtx.multiplicity());
         branchData.z0.push_back(vtx.z0());
         branchData.sumPt.push_back(vtx.pt());
       }
 
       if (printResults_) {
         edm::LogInfo("VertexNTupler") << "analyze::" << handle->size() << " '" << tokenMapEntry.first
                                       << "' vertices were found ... ";
         for (const auto& vtx : *handle) {
           edm::LogInfo("VertexNTupler") << "analyze::"
                                         << "  * z0 = " << vtx.z0() << "; contains " << vtx.multiplicity()
                                         << " tracks ...";
         }
       }
     }
 
     // Extra vertex collections
     l1Vertices_extra_numTracks_.resize(l1VerticesExtraTokens_.size());
     l1Vertices_extra_z0_.resize(l1VerticesExtraTokens_.size());
     l1Vertices_extra_z0_etaWeighted_.resize(l1VerticesExtraTokens_.size());
     l1Vertices_extra_sumPt_.resize(l1VerticesExtraTokens_.size());
 
     for (size_t i = 0; i < l1VerticesExtraTokens_.size(); i++) {
       edm::Handle<std::vector<l1t::Vertex>> vertexHandle;
       iEvent.getByToken(l1VerticesExtraTokens_.at(i), vertexHandle);
 
       const std::vector<l1t::Vertex>& vertices = *vertexHandle;
 
       l1Vertices_extra_numTracks_.at(i).clear();
       l1Vertices_extra_z0_.at(i).clear();
       l1Vertices_extra_z0_etaWeighted_.at(i).clear();
       l1Vertices_extra_sumPt_.at(i).clear();
 
       for (const auto& vertex : vertices) {
         l1Vertices_extra_numTracks_.at(i).push_back(vertex.tracks().size());
         l1Vertices_extra_z0_.at(i).push_back(vertex.z0());
 
         float sumPt = 0.0;
         float etaWeightedSumZ0 = 0.0;
         float etaWeightSum = 0.0;
         for (const auto& track : vertex.tracks()) {
           sumPt += track->momentum().transverse();
           // const float zRes = 0.133616 * track->momentum().eta() * track->momentum().eta() - 0.0522353 * std::abs(track->momentum().eta()) + 0.109918;
           const float zRes = 0.223074 * track->momentum().eta() * track->momentum().eta() -
                              0.050231 * abs(track->momentum().eta()) + 0.209719;
           etaWeightedSumZ0 += track->POCA().z() / (zRes * zRes);
           etaWeightSum += 1.0 / (zRes * zRes);
         }
 
         l1Vertices_extra_sumPt_.at(i).push_back(sumPt);
         l1Vertices_extra_z0_etaWeighted_.at(i).push_back(etaWeightedSumZ0 / etaWeightSum);
       }
     }
 
     outputTree_->Fill();
 
     if (settings_.debug() > 2)
       edm::LogInfo("VertexNTupler") << "analyze::================ End of Event ==============";
   }

◆ beginJob()

void l1tVertexFinder::VertexNTupler::beginJob ( void )

overrideprivatevirtual

Reimplemented from edm::one::EDAnalyzerBase.

Definition at line 367 of file VertexNTupler.cc.

367 {}

◆ endJob()

void l1tVertexFinder::VertexNTupler::endJob ( void )

overrideprivatevirtual

Reimplemented from edm::one::EDAnalyzerBase.

Definition at line 667 of file VertexNTupler.cc.

667 {}

Member Data Documentation

◆ allMatchedTPsToken_

const edm::EDGetTokenT<std::vector<l1tVertexFinder::TP> > l1tVertexFinder::VertexNTupler::allMatchedTPsToken_

private

Definition at line 173 of file VertexNTupler.cc.

Referenced by analyze().

◆ genJetsBranchData_

GenJetsBranchData l1tVertexFinder::VertexNTupler::genJetsBranchData_

private

Definition at line 193 of file VertexNTupler.cc.

Referenced by analyze(), and VertexNTupler().

◆ genJetsToken_

const edm::EDGetTokenT<std::vector<reco::GenJet> > l1tVertexFinder::VertexNTupler::genJetsToken_

private

Definition at line 172 of file VertexNTupler.cc.

Referenced by analyze().

◆ genParticlesHardOutgoingBranchData_

GenParticlesBranchData l1tVertexFinder::VertexNTupler::genParticlesHardOutgoingBranchData_

private

Definition at line 196 of file VertexNTupler.cc.

Referenced by analyze(), and VertexNTupler().

◆ genParticlesToken_

const edm::EDGetTokenT<edm::View<reco::GenParticle> > l1tVertexFinder::VertexNTupler::genParticlesToken_

private

Definition at line 171 of file VertexNTupler.cc.

Referenced by analyze().

◆ genVtxZ0_

float l1tVertexFinder::VertexNTupler::genVtxZ0_

private

Definition at line 190 of file VertexNTupler.cc.

Referenced by analyze(), and VertexNTupler().

◆ hepMCVtxZ0_

float l1tVertexFinder::VertexNTupler::hepMCVtxZ0_

private

Definition at line 190 of file VertexNTupler.cc.

Referenced by analyze(), and VertexNTupler().

◆ inputDataToken_

const edm::EDGetTokenT<l1tVertexFinder::InputData> l1tVertexFinder::VertexNTupler::inputDataToken_

private

Definition at line 169 of file VertexNTupler.cc.

Referenced by analyze().

◆ l1TracksBranchData_

std::map<std::string, RecoTracksBranchData> l1tVertexFinder::VertexNTupler::l1TracksBranchData_

private

Definition at line 198 of file VertexNTupler.cc.

Referenced by analyze(), and VertexNTupler().

◆ l1TracksMapTokenMap_

std::map<std::string, edm::EDGetTokenT<TTTrackAssociationMap<Ref_Phase2TrackerDigi_> > > l1tVertexFinder::VertexNTupler::l1TracksMapTokenMap_

private

Definition at line 176 of file VertexNTupler.cc.

Referenced by analyze(), and VertexNTupler().

◆ l1TracksTokenMap_

std::map<std::string, edm::EDGetTokenT<TTTrackCollectionView> > l1tVertexFinder::VertexNTupler::l1TracksTokenMap_

private

Definition at line 175 of file VertexNTupler.cc.

Referenced by analyze(), and VertexNTupler().

◆ l1Vertices_extra_numTracks_

std::vector<std::vector<unsigned> > l1tVertexFinder::VertexNTupler::l1Vertices_extra_numTracks_

private

Definition at line 203 of file VertexNTupler.cc.

Referenced by analyze(), and VertexNTupler().

◆ l1Vertices_extra_sumPt_

std::vector<std::vector<float> > l1tVertexFinder::VertexNTupler::l1Vertices_extra_sumPt_

private

Definition at line 206 of file VertexNTupler.cc.

Referenced by analyze(), and VertexNTupler().

◆ l1Vertices_extra_z0_

std::vector<std::vector<float> > l1tVertexFinder::VertexNTupler::l1Vertices_extra_z0_

private

Definition at line 204 of file VertexNTupler.cc.

Referenced by analyze(), and VertexNTupler().

◆ l1Vertices_extra_z0_etaWeighted_

std::vector<std::vector<float> > l1tVertexFinder::VertexNTupler::l1Vertices_extra_z0_etaWeighted_

private

Definition at line 205 of file VertexNTupler.cc.

Referenced by analyze(), and VertexNTupler().

◆ l1VerticesBranchData_

std::map<std::string, RecoVerticesBranchData> l1tVertexFinder::VertexNTupler::l1VerticesBranchData_

private

Definition at line 199 of file VertexNTupler.cc.

Referenced by analyze(), and VertexNTupler().

◆ l1VerticesEmulationBranchData_

std::map<std::string, EmulationVerticesBranchData> l1tVertexFinder::VertexNTupler::l1VerticesEmulationBranchData_

private

Definition at line 201 of file VertexNTupler.cc.

Referenced by analyze(), and VertexNTupler().

◆ l1VerticesEmulationTokenMap_

std::map<std::string, edm::EDGetTokenT<std::vector<l1t::VertexWord> > > l1tVertexFinder::VertexNTupler::l1VerticesEmulationTokenMap_

private

Definition at line 178 of file VertexNTupler.cc.

Referenced by analyze(), and VertexNTupler().

◆ l1VerticesExtraTokens_

std::vector<edm::EDGetTokenT<std::vector<l1t::Vertex> > > l1tVertexFinder::VertexNTupler::l1VerticesExtraTokens_

private

Definition at line 179 of file VertexNTupler.cc.

Referenced by analyze(), and VertexNTupler().

◆ l1VerticesInputMap_

std::map<std::string, std::string> l1tVertexFinder::VertexNTupler::l1VerticesInputMap_

private

Definition at line 200 of file VertexNTupler.cc.

Referenced by analyze(), and VertexNTupler().

◆ l1VerticesTokenMap_

std::map<std::string, edm::EDGetTokenT<std::vector<l1t::Vertex> > > l1tVertexFinder::VertexNTupler::l1VerticesTokenMap_

private

Definition at line 177 of file VertexNTupler.cc.

Referenced by analyze(), and VertexNTupler().

◆ numPileupVertices_

int l1tVertexFinder::VertexNTupler::numPileupVertices_

private

Definition at line 191 of file VertexNTupler.cc.

Referenced by analyze(), and VertexNTupler().

◆ numTrueInteractions_

float l1tVertexFinder::VertexNTupler::numTrueInteractions_

private

Definition at line 190 of file VertexNTupler.cc.

Referenced by analyze(), and VertexNTupler().

◆ outputTree_

TTree* l1tVertexFinder::VertexNTupler::outputTree_

private

Definition at line 181 of file VertexNTupler.cc.

Referenced by analyze(), and VertexNTupler().

◆ pileupSummaryToken_

const edm::EDGetTokenT<std::vector<PileupSummaryInfo> > l1tVertexFinder::VertexNTupler::pileupSummaryToken_

private

Definition at line 170 of file VertexNTupler.cc.

Referenced by analyze().

◆ printResults_

const bool l1tVertexFinder::VertexNTupler::printResults_

private

Definition at line 183 of file VertexNTupler.cc.

Referenced by analyze().

◆ settings_

AnalysisSettings l1tVertexFinder::VertexNTupler::settings_

private

Definition at line 186 of file VertexNTupler.cc.

Referenced by analyze().

◆ truePileUpVtxZ0_

std::vector<float> l1tVertexFinder::VertexNTupler::truePileUpVtxZ0_

private

Definition at line 195 of file VertexNTupler.cc.

Referenced by analyze(), and VertexNTupler().

◆ trueTracksBranchData_

TrueTracksBranchData l1tVertexFinder::VertexNTupler::trueTracksBranchData_

private

Definition at line 194 of file VertexNTupler.cc.

Referenced by analyze(), and VertexNTupler().

◆ vTPsToken_

const edm::EDGetTokenT<edm::ValueMap<l1tVertexFinder::TP> > l1tVertexFinder::VertexNTupler::vTPsToken_

private

Definition at line 174 of file VertexNTupler.cc.

Referenced by analyze().

Classes

Public Member Functions

Private Types

Private Member Functions

Private Attributes

Additional Inherited Members

Detailed Description

Member Typedef Documentation

◆ TTTrackAssMap

◆ TTTrackCollectionView

Constructor & Destructor Documentation

◆ VertexNTupler()

◆ ~VertexNTupler()

Member Function Documentation

◆ analyze()

◆ beginJob()

◆ endJob()

Member Data Documentation

◆ allMatchedTPsToken_

◆ genJetsBranchData_

◆ genJetsToken_

◆ genParticlesHardOutgoingBranchData_

◆ genParticlesToken_

◆ genVtxZ0_

◆ hepMCVtxZ0_

◆ inputDataToken_

◆ l1TracksBranchData_

◆ l1TracksMapTokenMap_

◆ l1TracksTokenMap_

◆ l1Vertices_extra_numTracks_

◆ l1Vertices_extra_sumPt_

◆ l1Vertices_extra_z0_

◆ l1Vertices_extra_z0_etaWeighted_

◆ l1VerticesBranchData_

◆ l1VerticesEmulationBranchData_

◆ l1VerticesEmulationTokenMap_

◆ l1VerticesExtraTokens_

◆ l1VerticesInputMap_

◆ l1VerticesTokenMap_

◆ numPileupVertices_

◆ numTrueInteractions_

◆ outputTree_

◆ pileupSummaryToken_

◆ printResults_

◆ settings_

◆ truePileUpVtxZ0_

◆ trueTracksBranchData_

◆ vTPsToken_