Inheritance diagram for l1tVertexFinder::VertexNTupler:

Classes
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::EDAnalyzer
void	callWhenNewProductsRegistered (std::function< void(BranchDescription const &)> const &func)

	EDAnalyzer ()

SerialTaskQueue *	globalLuminosityBlocksQueue ()

SerialTaskQueue *	globalRunsQueue ()

ModuleDescription const &	moduleDescription () const

std::string	workerType () const

	~EDAnalyzer () 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 &&)=default

	EDConsumerBase (EDConsumerBase const &)=delete

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 &	operator= (EDConsumerBase &&)=default

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

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_

bool	available_

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::unordered_map< std::string, std::vector< unsigned > >	l1Vertices_branchMap_numTracks_

std::unordered_map< std::string, std::vector< float > >	l1Vertices_branchMap_sumPt_

std::unordered_map< std::string, std::vector< float > >	l1Vertices_branchMap_z0_

std::unordered_map< std::string, std::vector< float > >	l1Vertices_branchMap_z0_etaWeighted_

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, edm::EDGetTokenT< std::vector< l1t::Vertex > > >	l1VerticesExtraTokenMap_

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::EDAnalyzer
typedef EDAnalyzer	ModuleType

Public Types inherited from edm::EDConsumerBase
typedef ProductLabels	Labels

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

static void	fillDescriptions (ConfigurationDescriptions &descriptions)

static void	prevalidate (ConfigurationDescriptions &)

static bool	wantsGlobalLuminosityBlocks ()

static bool	wantsGlobalRuns ()

static bool	wantsInputProcessBlocks ()

static bool	wantsProcessBlocks ()

static bool	wantsStreamLuminosityBlocks ()

static bool	wantsStreamRuns ()

Protected Member Functions inherited from edm::EDConsumerBase
EDGetToken	consumes (const TypeToGet &id, edm::InputTag const &tag)

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

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

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<Transition Tr = Transition::Event>
constexpr auto	esConsumes () noexcept

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

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

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

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

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

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

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 156 of file VertexNTupler.cc.

◆ TTTrackCollectionView

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

private

Definition at line 157 of file VertexNTupler.cc.

Constructor & Destructor Documentation

◆ VertexNTupler()

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

explicit

Definition at line 208 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"))),
         //outputTree_(fs_->make<TTree>("l1VertexReco", "L1 vertex-related info")),
         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"));
  
     edm::Service<TFileService> fs_;
     available_ = fs_.isAvailable();
     if (not available_)
       return;  // No ROOT file open.
  
     outputTree_ = fs_->make<TTree>("l1VertexReco", "L1 vertex-related info");
  
     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()) + ")");
  
     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);
     }
  
     std::vector<std::string>::const_iterator branchNameIt = vertexBranchNames.begin();
     std::vector<edm::InputTag>::const_iterator 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);
  
     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"));
     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 630 of file VertexNTupler.cc.

630 {}

Member Function Documentation

◆ analyze()

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

overrideprivatevirtual

Implements edm::EDAnalyzer.

Definition at line 399 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();
         }
       }
     }
  
     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::EDAnalyzer.

Definition at line 353 of file VertexNTupler.cc.

353 {}

◆ endJob()

void l1tVertexFinder::VertexNTupler::endJob ( void )

overrideprivatevirtual

Reimplemented from edm::EDAnalyzer.

Definition at line 628 of file VertexNTupler.cc.

628 {}

Member Data Documentation

◆ allMatchedTPsToken_

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

private

Definition at line 164 of file VertexNTupler.cc.

Referenced by analyze().

◆ available_

bool l1tVertexFinder::VertexNTupler::available_

private

Definition at line 205 of file VertexNTupler.cc.

Referenced by VertexNTupler().

◆ genJetsBranchData_

GenJetsBranchData l1tVertexFinder::VertexNTupler::genJetsBranchData_

private

Definition at line 186 of file VertexNTupler.cc.

Referenced by analyze(), and VertexNTupler().

◆ genJetsToken_

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

private

Definition at line 163 of file VertexNTupler.cc.

Referenced by analyze().

◆ genParticlesHardOutgoingBranchData_

GenParticlesBranchData l1tVertexFinder::VertexNTupler::genParticlesHardOutgoingBranchData_

private

Definition at line 189 of file VertexNTupler.cc.

Referenced by analyze(), and VertexNTupler().

◆ genParticlesToken_

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

private

Definition at line 162 of file VertexNTupler.cc.

Referenced by analyze().

◆ genVtxZ0_

float l1tVertexFinder::VertexNTupler::genVtxZ0_

private

Definition at line 183 of file VertexNTupler.cc.

Referenced by analyze(), and VertexNTupler().

◆ hepMCVtxZ0_

float l1tVertexFinder::VertexNTupler::hepMCVtxZ0_

private

Definition at line 183 of file VertexNTupler.cc.

Referenced by analyze(), and VertexNTupler().

◆ inputDataToken_

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

private

Definition at line 160 of file VertexNTupler.cc.

Referenced by analyze().

◆ l1TracksBranchData_

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

private

Definition at line 191 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 167 of file VertexNTupler.cc.

Referenced by analyze(), and VertexNTupler().

◆ l1TracksTokenMap_

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

private

Definition at line 166 of file VertexNTupler.cc.

Referenced by analyze(), and VertexNTupler().

◆ l1Vertices_branchMap_numTracks_

std::unordered_map<std::string, std::vector<unsigned> > l1tVertexFinder::VertexNTupler::l1Vertices_branchMap_numTracks_

private

Definition at line 195 of file VertexNTupler.cc.

◆ l1Vertices_branchMap_sumPt_

std::unordered_map<std::string, std::vector<float> > l1tVertexFinder::VertexNTupler::l1Vertices_branchMap_sumPt_

private

Definition at line 198 of file VertexNTupler.cc.

◆ l1Vertices_branchMap_z0_

std::unordered_map<std::string, std::vector<float> > l1tVertexFinder::VertexNTupler::l1Vertices_branchMap_z0_

private

Definition at line 196 of file VertexNTupler.cc.

◆ l1Vertices_branchMap_z0_etaWeighted_

std::unordered_map<std::string, std::vector<float> > l1tVertexFinder::VertexNTupler::l1Vertices_branchMap_z0_etaWeighted_

private

Definition at line 197 of file VertexNTupler.cc.

◆ l1Vertices_extra_numTracks_

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

private

Definition at line 200 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 203 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 201 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 202 of file VertexNTupler.cc.

Referenced by analyze(), and VertexNTupler().

◆ l1VerticesBranchData_

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

private

Definition at line 192 of file VertexNTupler.cc.

Referenced by analyze(), and VertexNTupler().

◆ l1VerticesExtraTokenMap_

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

private

Definition at line 169 of file VertexNTupler.cc.

◆ l1VerticesExtraTokens_

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

private

Definition at line 170 of file VertexNTupler.cc.

Referenced by analyze(), and VertexNTupler().

◆ l1VerticesInputMap_

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

private

Definition at line 193 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 168 of file VertexNTupler.cc.

Referenced by analyze(), and VertexNTupler().

◆ numPileupVertices_

int l1tVertexFinder::VertexNTupler::numPileupVertices_

private

Definition at line 184 of file VertexNTupler.cc.

Referenced by analyze(), and VertexNTupler().

◆ numTrueInteractions_

float l1tVertexFinder::VertexNTupler::numTrueInteractions_

private

Definition at line 183 of file VertexNTupler.cc.

Referenced by analyze(), and VertexNTupler().

◆ outputTree_

TTree* l1tVertexFinder::VertexNTupler::outputTree_

private

Definition at line 172 of file VertexNTupler.cc.

Referenced by analyze(), and VertexNTupler().

◆ pileupSummaryToken_

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

private

Definition at line 161 of file VertexNTupler.cc.

Referenced by analyze().

◆ printResults_

const bool l1tVertexFinder::VertexNTupler::printResults_

private

Definition at line 174 of file VertexNTupler.cc.

Referenced by analyze().

◆ settings_

AnalysisSettings l1tVertexFinder::VertexNTupler::settings_

private

Definition at line 177 of file VertexNTupler.cc.

Referenced by analyze().

◆ truePileUpVtxZ0_

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

private

Definition at line 188 of file VertexNTupler.cc.

Referenced by analyze(), and VertexNTupler().

◆ trueTracksBranchData_

TrueTracksBranchData l1tVertexFinder::VertexNTupler::trueTracksBranchData_

private

Definition at line 187 of file VertexNTupler.cc.

Referenced by analyze(), and VertexNTupler().

◆ vTPsToken_

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

private

Definition at line 165 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_

◆ available_

◆ genJetsBranchData_

◆ genJetsToken_

◆ genParticlesHardOutgoingBranchData_

◆ genParticlesToken_

◆ genVtxZ0_

◆ hepMCVtxZ0_

◆ inputDataToken_

◆ l1TracksBranchData_

◆ l1TracksMapTokenMap_

◆ l1TracksTokenMap_

◆ l1Vertices_branchMap_numTracks_

◆ l1Vertices_branchMap_sumPt_

◆ l1Vertices_branchMap_z0_

◆ l1Vertices_branchMap_z0_etaWeighted_

◆ l1Vertices_extra_numTracks_

◆ l1Vertices_extra_sumPt_

◆ l1Vertices_extra_z0_

◆ l1Vertices_extra_z0_etaWeighted_

◆ l1VerticesBranchData_

◆ l1VerticesExtraTokenMap_

◆ l1VerticesExtraTokens_

◆ l1VerticesInputMap_

◆ l1VerticesTokenMap_

◆ numPileupVertices_

◆ numTrueInteractions_

◆ outputTree_

◆ pileupSummaryToken_

◆ printResults_

◆ settings_

◆ truePileUpVtxZ0_

◆ trueTracksBranchData_

◆ vTPsToken_