10#include "GaudiKernel/SystemOfUnits.h"
49 float result(std::numeric_limits<float>::quiet_NaN());
52 if (truthMatchProbabilityAcc.isAvailable(trackParticle)) {
53 result = truthMatchProbabilityAcc(trackParticle);
60 safelyGetEta(
const T& pTrk,
const float safePtThreshold = 0.1) {
61 return (pTrk->pt() > safePtThreshold) ? (pTrk->eta()) : std::nan(
"");
67 inRange(
const T& value,
const T& minVal,
const T& maxVal) {
68 return not ((value < minVal)or(value > maxVal));
73 inRange(
const T& value,
const T& absoluteMax) {
74 return not (std::abs(value) > absoluteMax);
82 binIndex(
const T& val,
const std::vector<T>& partitions) {
89 return nf ?
i :
i - 1;
94 const float x(vtx->
x()),
y(vtx->
y()),
z(vtx->
z());
95 const float vrtR2 = (
x *
x +
y *
y);
97 return inRange(
z, 500.f) and not (vrtR2 > 1);
101 2.7, 3.5, std::numeric_limits<float>::infinity()
108 const IInterface* parent) :
136 (
nullptr,
static_cast<std::string
>(
m_dirName) +
static_cast<std::string
>(
m_folder),
147 std::vector<std::string> trk_decorations;
163 std::vector<std::string> required_float_track_decorations {
"d0",
"hitResiduals_residualLocX",
"d0err"};
164 std::vector<std::string> required_int_track_decorations {};
165 std::vector<std::string> required_float_truth_decorations {
"d0"};
166 std::vector<std::string> required_int_truth_decorations {};
167 std::vector<std::string> required_int_jet_decorations {
"HadronConeExclTruthLabelID"};
172 m_datfile <<
"EvtNumber,PdgID,Px,Py,Pz,E,Pt,Eta,Phi,Mass,numPU,numvtx,MatchProb"<<std::endl;
174 std::string empty_prefix;
188 return StatusCode::SUCCESS;
329 const EventContext& ctx = Gaudi::Hive::currentContext();
333 if (not trackHandle.
isValid()) {
335 return StatusCode::FAILURE;
346 ATH_MSG_WARNING(
"Shouldn't happen - EventInfo is buggy, setting mu to 0");
354 return StatusCode::SUCCESS;
358 std::vector<const xAOD::TruthParticle*> truthParticlesVec =
getTruthParticles(ctx);
367 unsigned int truthMu = 0;
370 truthMu =
static_cast<int>( truthPileupEventContainer->size() );
372 if(pie.
isValid()) actualMu = pie->actualInteractionsPerCrossing();
375 float puEvents = truthMu>0 ? truthMu : actualMu;
378 unsigned int nVertices = 0;
379 float beamSpotWeight = 1;
382 ATH_MSG_DEBUG(
"Getting number of pu interactings per event");
388 nVertices = not vertices->empty() ? vertices->size() : 0;
389 beamSpotWeight = pie->beamSpotWeight();
390 ATH_MSG_DEBUG(
"beamSpotWeight is equal to " << beamSpotWeight);
394 if(readDecorHandle.
isAvailable()) prwWeight = readDecorHandle(*pie);
395 ATH_MSG_DEBUG(
"Applying pileup weight equal to " << prwWeight);
396 beamSpotWeight *= prwWeight;
399 if (vertices.
isValid() and not vertices->empty()) {
400 ATH_MSG_DEBUG(
"Number of vertices retrieved for this event " << vertices->size());
406 ATH_MSG_DEBUG(
"Failed to find a hard scatter vertex in this event.");
412 std::pair<std::vector<const xAOD::TruthVertex*>, std::vector<const xAOD::TruthVertex*>> truthVertices =
getTruthVertices(ctx);
413 std::vector<const xAOD::TruthVertex*> truthHSVertices = truthVertices.first;
414 std::vector<const xAOD::TruthVertex*> truthPUVertices = truthVertices.second;
421 m_monPlots->fill(*vertices, primaryvertex, truthHSVertices, truthPUVertices, actualMu, beamSpotWeight);
425 m_monPlots->fill(*vertices, truthMu, actualMu, beamSpotWeight);
442 const auto& stdVertexContainer = truthVrt->stdcont();
444 auto findVtx = std::find_if(stdVertexContainer.rbegin(), stdVertexContainer.rend(), acceptTruthVertex);
445 truthVertex = (findVtx == stdVertexContainer.rend()) ? nullptr : *findVtx;
449 if (not truthVertex)
ATH_MSG_INFO (
"Truth vertex did not pass cuts");
454 unsigned int nSelectedTruthTracks(0), nSelectedRecoTracks(0), nSelectedMatchedTracks(0), nAssociatedTruth(0), nMissingAssociatedTruth(0), nTruths(0);
463 std::map<const xAOD::TruthParticle*, std::vector<const xAOD::TrackParticle*>> cacheTruthMatching {};
465 std::vector<const xAOD::TrackParticle*> selectedTracks {};
466 selectedTracks.reserve(tracks->
size());
467 unsigned int nTrackTOT = 0;
468 unsigned int nTrackCentral = 0;
469 unsigned int nTrackPt1GeV = 0;
470 for (
const auto *
const thisTrack: *tracks) {
476 selectedTracks.push_back(thisTrack);
478 nSelectedRecoTracks++;
482 if (thisTrack->pt() >= (1 * Gaudi::Units::GeV))
484 if (std::abs(thisTrack->eta()) < 2.5)
487 m_monPlots->fill(*thisTrack, puEvents, nVertices, beamSpotWeight);
489 float prob = getMatchingProbability(*thisTrack);
492 if (associatedTruth) {
503 nSelectedMatchedTracks++;
504 bool truthIsFromB =
false;
508 m_monPlots->fill(*thisTrack, *associatedTruth, truthIsFromB, puEvents, beamSpotWeight);
512 const bool isAssociatedTruth = associatedTruth !=
nullptr;
513 const bool isFake = not std::isnan(prob) ? (prob <
m_lowProb) :
true;
515 if(!isAssociatedTruth) nMissingAssociatedTruth++;
516 m_monPlots->fillFakeRate(*thisTrack, isFake, isAssociatedTruth, puEvents, beamSpotWeight);
522 if (isAssociatedTruth) {
527 auto cachedAssoc = cacheTruthMatching.find(associatedTruth);
529 if (cachedAssoc == cacheTruthMatching.end()) {
531 cacheTruthMatching[associatedTruth] = {thisTrack};
535 cachedAssoc->second.push_back(thisTrack);
540 m_monPlots->fillNtuple(*thisTrack, primaryvertex);
548 for (
auto& cachedAssoc: cacheTruthMatching) {
555 std::sort(cachedAssoc.second.begin(), cachedAssoc.second.end(),
561 for (
int itrack = 0; itrack < (int) cachedAssoc.second.size(); itrack++) {
565 m_monPlots->fillNtuple(*thisTrack, *thisTruth, primaryvertex, itrack);
571 m_monPlots->fill(nTrackTOT, nTrackCentral, nTrackPt1GeV, truthMu, actualMu, nVertices, beamSpotWeight);
577 m_truthCutFlow.merge(std::move(tmp_truth_cutflow));
583 for (
int itruth = 0; itruth < (int) truthParticlesVec.size(); itruth++) {
590 ++nSelectedTruthTracks;
591 bool isEfficient(
false);
592 float matchingProbability{};
596 auto cachedAssoc = cacheTruthMatching.find(thisTruth);
598 if (cachedAssoc == cacheTruthMatching.end()) {
600 cacheTruthMatching[thisTruth] = {};
602 m_monPlots->fillDuplicate(*thisTruth, cacheTruthMatching[thisTruth], beamSpotWeight);
609 for (
const auto& thisTrack: selectedTracks) {
611 if (associatedTruth && associatedTruth == thisTruth) {
612 float prob = getMatchingProbability(*thisTrack);
613 if (not std::isnan(prob) && prob >
m_lowProb) {
614 matchingProbability = prob;
616 matchedTrack = thisTrack;
623 <<thisTruth->
py()/ Gaudi::Units::GeV<<
","<<thisTruth->
pz()/ Gaudi::Units::GeV<<
","
624 <<thisTruth->
e()/ Gaudi::Units::GeV<<
","<<thisTruth->
pt()/ Gaudi::Units::GeV<<
","
625 <<thisTruth->
eta()<<
","<<thisTruth->
phi()<<
","<<thisTruth->
m()/ Gaudi::Units::GeV<<
","
626 <<puEvents<<
","<<nVertices<<
","<<matchingProbability<<std::endl;
629 ATH_MSG_ERROR(
"An error occurred: Truth particle for tracking efficiency calculation is a nullptr");
631 else if (isEfficient && !matchedTrack){
632 ATH_MSG_ERROR(
"Something went wrong - we log a truth particle as reconstructed, but the reco track is a nullptr! Bailing out... ");
635 ATH_MSG_DEBUG(
"Filling efficiency plots info monitoring plots");
636 m_monPlots->fillEfficiency(*thisTruth, matchedTrack, isEfficient, truthMu, actualMu, beamSpotWeight);
638 ATH_MSG_DEBUG(
"Filling technical efficiency plots info monitoring plots");
642 m_monPlots->fillTechnicalEfficiency(*thisTruth, isEfficient,
643 truthMu, actualMu, beamSpotWeight);
646 ATH_MSG_DEBUG(
"Cannot fill technical efficiency. Missing si hit information for truth particle.");
664 if (nSelectedRecoTracks == nMissingAssociatedTruth) {
669 ATH_MSG_DEBUG(nAssociatedTruth <<
" tracks out of " << tracks->
size() <<
" had associated truth.");
688 return StatusCode::SUCCESS;
714 ATH_MSG_DEBUG(
"Selected by pileup switch decoration requested from a truth particle but not available");
721 for (
const auto& thisTruth: truthParticles) {
730 std::vector<HistData> hists =
m_monPlots->retrieveBookedHistograms();
731 for (
const auto& hist : hists) {
735 std::vector<EfficiencyData> effs =
m_monPlots->retrieveBookedEfficiencies();
736 for (
auto& eff : effs) {
743 std::vector<TreeData> trees =
m_monPlots->retrieveBookedTrees();
744 for (
auto& t : trees) {
749 return StatusCode::SUCCESS;
774 return StatusCode::SUCCESS;
777const std::vector<const xAOD::TruthParticle*>
780 std::vector<const xAOD::TruthParticle*> tempVec {};
787 if (not truthParticleContainer.
isValid()) {
790 tempVec.insert(tempVec.begin(), truthParticleContainer->begin(), truthParticleContainer->end());
800 const auto& links =
event->truthParticleLinks();
801 tempVec.reserve(event->nTruthParticles());
802 for (
const auto& link : links) {
804 tempVec.push_back(*link);
813 if (truthPileupEventContainer.
isValid()) {
814 const unsigned int nPileup = truthPileupEventContainer->size();
815 tempVec.reserve(nPileup * 200);
816 for (
unsigned int i(0); i != nPileup; ++i) {
817 const auto *eventPileup = truthPileupEventContainer->at(i);
819 int ntruth = eventPileup->nTruthParticles();
820 ATH_MSG_VERBOSE(
"Adding " << ntruth <<
" truth particles from TruthPileupEvents container");
821 const auto& links = eventPileup->truthParticleLinks();
822 for (
const auto& link : links) {
824 tempVec.push_back(*link);
839std::pair<const std::vector<const xAOD::TruthVertex*>,
const std::vector<const xAOD::TruthVertex*>>
842 std::vector<const xAOD::TruthVertex*> truthHSVertices = {};
843 truthHSVertices.reserve(5);
844 std::vector<const xAOD::TruthVertex*> truthPUVertices = {};
845 truthPUVertices.reserve(100);
868 if (truthEventContainer.
isValid()) {
869 for (
const auto *
const evt : *truthEventContainer) {
870 truthVtx = evt->signalProcessVertex();
872 truthHSVertices.push_back(truthVtx);
886 if (truthPileupEventContainer.
isValid()) {
887 for (
const auto *
const evt : *truthPileupEventContainer) {
892 size_t i_vtx = 0;
size_t n_vtx = evt->nTruthVertices();
893 while(!truthVtx && i_vtx<n_vtx){
894 truthVtx = evt->truthVertex(i_vtx);
899 truthPUVertices.push_back(truthVtx);
909 return std::make_pair<const std::vector<const xAOD::TruthVertex*>,
const std::vector<const xAOD::TruthVertex*>>((
const std::vector<const xAOD::TruthVertex*>)truthHSVertices, (
const std::vector<const xAOD::TruthVertex*>)truthPUVertices);
920 std::vector<int>& cutFlow) {
922 if (cutFlow.empty()) {
923 names.emplace_back(
"preCut");
924 cutFlow.push_back(0);
925 for (
unsigned int i = 0; i != accept.getNCuts(); ++i) {
926 cutFlow.push_back(0);
927 names.push_back((std::string) accept.getCutName(i));
932 bool cutPositive =
true;
933 for (
unsigned int i = 0; i != (accept.getNCuts() + 1); ++i) {
937 if (accept.getCutResult(i)) {
946 double absEta = std::abs(truth.
eta());
949 ATH_MSG_INFO(
"Requesting cut value outside of configured eta range: clamping eta = "
950 << std::abs(truth.
eta()) <<
" to eta= " << absEta);
953 const auto pVal = std::lower_bound(
m_etaBins.value().begin(),
m_etaBins.value().end(), absEta);
954 const int bin = std::distance(
m_etaBins.value().begin(), pVal) - 1;
955 ATH_MSG_DEBUG(
"Checking (abs(eta)/bin) = (" << absEta <<
"," <<
bin <<
")");
961 const std::vector<const xAOD::TruthParticle*>& truthParticles,
964 float beamSpotWeight) {
969 if (truthParticles.empty()) {
970 ATH_MSG_WARNING(
"No entries in TruthParticles truth particle container. Skipping jet plots.");
971 return StatusCode::SUCCESS;
977 return StatusCode::SUCCESS;
982 for (
const xAOD::Jet *
const thisJet: *jets) {
990 isBjet = (btagLabel(*thisJet) == 5);
998 if (not el.isValid())
continue;
1007 if(!accept)
continue;
1009 bool isEfficient(
false);
1011 for (
const auto *thisTrack: tracks) {
1017 if (associatedTruth and associatedTruth == truth) {
1018 float prob = getMatchingProbability(*thisTrack);
1019 if (not std::isnan(prob) && prob >
m_lowProb) {
1026 bool truthIsFromB =
false;
1028 truthIsFromB =
true;
1030 m_monPlots->fillEfficiency(*truth, *thisJet, isEfficient, isBjet, truthIsFromB, beamSpotWeight);
1040 if (thisJet->p4().DeltaR(thisTrack->p4()) >
m_maxTrkJetDR) {
1044 float prob = getMatchingProbability(*thisTrack);
1045 if(std::isnan(prob)) prob = 0.0;
1048 const bool unlinked = (associatedTruth==
nullptr);
1049 const bool isFake = (associatedTruth && prob <
m_lowProb);
1050 bool truthIsFromB =
false;
1052 truthIsFromB =
true;
1054 m_monPlots->fill(*thisTrack, *thisJet, isBjet, isFake, unlinked, truthIsFromB, beamSpotWeight);
1055 if (associatedTruth){
1056 m_monPlots->fillFakeRate(*thisTrack, *thisJet, isFake, isBjet, truthIsFromB, beamSpotWeight);
1062 return StatusCode::SUCCESS;
1067 const float jetPt =
jet.pt();
1068 const float jetEta = std::abs(
jet.eta());
#define ATH_CHECK
Evaluate an expression and check for errors.
#define ATH_MSG_VERBOSE(x)
#define ATH_MSG_WARNING(x)
Helper class to provide constant type-safe access to aux data.
virtual void lock()=0
Interface to allow an object to lock itself when made const in SG.
header file for class of same name
bool inRange(const double *boundaries, const double value, const double tolerance=0.02)
ServiceHandle< StoreGateSvc > & evtStore()
size_type size() const noexcept
Returns the number of elements in the collection.
ElementLink implementation for ROOT usage.
static void neededTrackParticleDecorations(std::vector< std::string > &decorations)
const xAOD::TruthParticle * getTruth(const xAOD::TrackParticle *const trackParticle)
Helper class to provide constant type-safe access to aux data.
bool isAvailable(const ELT &e) const
Test to see if this variable exists in the store.
Handle class for reading a decoration on an object.
bool isAvailable()
Test to see if this variable exists in the store, for the referenced object.
virtual bool isValid() override final
Can the handle be successfully dereferenced?
const_pointer_type cptr()
Dereference the pointer.
const_pointer_type get() const
Dereference the pointer, but don't cache anything.
@ IS_SIMULATION
true: simulation, false: data
uint64_t eventNumber() const
The current event's event number.
virtual double m() const override final
The mass of the particle.
int pdgId() const
PDG ID code.
float px() const
The x component of the particle's momentum.
virtual double e() const override final
The total energy of the particle.
virtual double pt() const override final
The transverse momentum ( ) of the particle.
float py() const
The y component of the particle's momentum.
virtual double eta() const override final
The pseudorapidity ( ) of the particle.
virtual double phi() const override final
The azimuthal angle ( ) of the particle.
virtual FourMom_t p4() const override final
The full 4-momentum of the particle.
float pz() const
The z component of the particle's momentum.
float z() const
Vertex longitudinal distance along the beam line form the origin.
float y() const
Vertex y displacement.
float x() const
Vertex x displacement.
void addReadDecoratorHandleKeys(T_Parent &parent, const SG::ReadHandleKey< T_Cont > &container_key, const std::string &prefix, const std::vector< std::string > &decor_names, std::vector< SG::ReadDecorHandleKey< T_Cont > > &decor_out)
float safelyGetEta(const T &pTrk, const float safePtThreshold=0.1)
Safely get eta.
unsigned int binIndex(const T &val, const std::vector< T > &partitions)
general utility function to return bin index given a value and the upper endpoints of each bin
HardScatterType classifyHardScatter(const xAOD::VertexContainer &vxContainer)
void sort(typename DataModel_detail::iterator< DVL > beg, typename DataModel_detail::iterator< DVL > end)
Specialization of sort for DataVector/List.
Jet_v1 Jet
Definition of the current "jet version".
EventInfo_v1 EventInfo
Definition of the latest event info version.
TruthVertex_v1 TruthVertex
Typedef to implementation.
TrackParticle_v1 TrackParticle
Reference the current persistent version:
Vertex_v1 Vertex
Define the latest version of the vertex class.
TruthEvent_v1 TruthEvent
Typedef to implementation.
TruthParticle_v1 TruthParticle
Typedef to implementation.
TrackParticleContainer_v1 TrackParticleContainer
Definition of the current "TrackParticle container version".
JetContainer_v1 JetContainer
Definition of the current "jet container version".
implementation file for function of same name
helper struct - steer the configuration from the parent tool's side
bool doTrkInJetPlots_matched_bjets
bool doTrkInJetPlots_unlinked_bjets
bool doHitsFakeTracksPlots
int detailLevel
detail level (kept for compatibility)
bool doNtupleTruthToReco
Ntuple functionality.
bool doEfficienciesPerAuthor
per author plots
bool doMissingTruthFakePlots
bool doHitsRecoTracksPlots
bool doHitsRecoTracksPlotsPerAuthor
bool doEffPlots
Efficiency and duplicate plots - require truth, optionally matching reco.
bool doTrkInJetPlots_bjets
bool doTrkInJetPlots_matched
bool doFakePlots
Fake plots (and unlinked).
bool doResolutionPlotPrim
Resolution and "matched track" plots - filled if both reco and truth exist.
bool doTrkInJetPlots_fake_bjets
bool doVertexTruthMatchingPlots
Vertexing plots - truth requirement.
bool doVertexPlots
Vertexing plots - no truth requirement.
bool doHardScatterVertexTruthMatchingPlots
bool doTrkInJetPlots_truthFromB
bool doTrkInJetPlots_fake
bool doHitsUnlinkedTracksPlots
bool doTrackParametersPerAuthor
bool doTrkInJetPlots_unlinked
bool doTrackParameters
Plots for (selected) tracks, not necessarily truth matched.
bool doTrkInJetPlots
Plots for tracks in jets.
bool doResolutionPlotPrim_truthFromB
bool doHardScatterVertexPlots
bool doResolutionPlotSecd
bool doResolutionsPerAuthor
bool doHitsMatchedTracksPlots