ParticleReader Struct Reference

Struct used for reading particles written out by the TrackFinderPerformanceWriter. More...

#include <TreeReader.h>

Public Member Functions
	ParticleReader (TTree *tree_, bool sortEvents)
std::vector< ParticleInfo >	getParticles (const std::uint32_t &eventNumber) const
Public Member Functions inherited from TreeReader
	TreeReader (TTree *tree_)
void	getEntry (unsigned int i) const

Public Attributes
ULong64_t	eventId = 0
ULong64_t	particleId = 0
Int_t	particleType = 0
float	vx = 0
float	vy = 0
float	vz = 0
float	vt = 0
float	px = 0
float	py = 0
float	pz = 0
float	m = 0
float	q = 0
UShort_t	nHits = 0
UShort_t	nTracks = 0
UShort_t	nTracksMajority = 0
Public Attributes inherited from TreeReader
TTree *	tree = nullptr

Additional Inherited Members
Protected Attributes inherited from TreeReader
std::vector< long long >	entryNumbers = {}
	The entry numbers for accessing events in increased order (there could be multiple entries corresponding to one event number)

Detailed Description

Struct used for reading particles written out by the TrackFinderPerformanceWriter.

Definition at line 405 of file TreeReader.h.

Constructor & Destructor Documentation

ParticleReader()

ParticleReader::ParticleReader ( TTree * tree_, bool sortEvents )

inline

Definition at line 410 of file TreeReader.h.

                                                : TreeReader(tree_) {
    tree->SetBranchAddress("event_id", &eventId);
    tree->SetBranchAddress("particle_id", &particleId);
    tree->SetBranchAddress("particle_type", &particleType);
    tree->SetBranchAddress("vx", &vx);
    tree->SetBranchAddress("vy", &vy);
    tree->SetBranchAddress("vz", &vz);
    tree->SetBranchAddress("vt", &vt);
    tree->SetBranchAddress("px", &px);
    tree->SetBranchAddress("py", &py);
    tree->SetBranchAddress("pz", &pz);
    tree->SetBranchAddress("m", &m);
    tree->SetBranchAddress("q", &q);
    tree->SetBranchAddress("nhits", &nHits);
    tree->SetBranchAddress("ntracks", &nTracks);
    tree->SetBranchAddress("ntracks_majority", &nTracksMajority);
 
    // It's not necessary if you just need to read one file, but please do it to
    // synchronize events if multiple root files are read
    if (sortEvents) {
      entryNumbers.resize(tree->GetEntries());
      tree->Draw("event_id", "", "goff");
      // Sort to get the entry numbers of the ordered events
      TMath::Sort(tree->GetEntries(), tree->GetV1(), entryNumbers.data(),
                  false);
    }
  }

Member Function Documentation

getParticles()

std::vector< ParticleInfo > ParticleReader::getParticles ( const std::uint32_t & eventNumber ) const

inline

Definition at line 439 of file TreeReader.h.

                                                                             {
    // Find the start entry and the batch size for this event
    std::string eventNumberStr = std::to_string(eventNumber);
    std::string findStartEntry = "event_id<" + eventNumberStr;
    std::string findParticlesSize = "event_id==" + eventNumberStr;
    std::size_t startEntry = tree->GetEntries(findStartEntry.c_str());
    std::size_t nParticles = tree->GetEntries(findParticlesSize.c_str());
    if (nParticles == 0) {
      throw std::invalid_argument(
          "No particles found. Please check the input file.");
    }
    std::vector<ParticleInfo> particles;
    particles.reserve(nParticles);
    for (unsigned int i = 0; i < nParticles; ++i) {
      getEntry(startEntry + i);
      auto pt = std::hypot(px, py);
      auto p = std::hypot(pt, pz);
      auto eta = std::atanh(pz / p * 1.);
      particles.push_back({particleId, eta, p, pt, nHits});
    }
    return particles;
  }

Member Data Documentation

eventId

ULong64_t ParticleReader::eventId = 0

Definition at line 463 of file TreeReader.h.

m

float ParticleReader::m = 0

Definition at line 469 of file TreeReader.h.

nHits

UShort_t ParticleReader::nHits = 0

Definition at line 471 of file TreeReader.h.

nTracks

UShort_t ParticleReader::nTracks = 0

Definition at line 472 of file TreeReader.h.

nTracksMajority

UShort_t ParticleReader::nTracksMajority = 0

Definition at line 473 of file TreeReader.h.

particleId

ULong64_t ParticleReader::particleId = 0

Definition at line 464 of file TreeReader.h.

particleType

Int_t ParticleReader::particleType = 0

Definition at line 465 of file TreeReader.h.

px

float ParticleReader::px = 0

Definition at line 468 of file TreeReader.h.

py

float ParticleReader::py = 0

Definition at line 468 of file TreeReader.h.

pz

float ParticleReader::pz = 0

Definition at line 468 of file TreeReader.h.

q

float ParticleReader::q = 0

Definition at line 470 of file TreeReader.h.

vt

float ParticleReader::vt = 0

Definition at line 467 of file TreeReader.h.

vx

float ParticleReader::vx = 0

Definition at line 466 of file TreeReader.h.

vy

float ParticleReader::vy = 0

Definition at line 466 of file TreeReader.h.

vz

float ParticleReader::vz = 0

Definition at line 466 of file TreeReader.h.

---

The documentation for this struct was generated from the following file:

• build/_deps/acts-src/Examples/Scripts/TrackingPerformance/TreeReader.h