garana::MillG4Tree Class Reference

#include <MillG4Tree.h>

Inheritance diagram for garana::MillG4Tree:

Public Member Functions
	MillG4Tree ()
	MillG4Tree (TTree *treeIn, TTree *treeOut)
void	MillTrees () override
Public Member Functions inherited from garana::Mill
virtual	~Mill ()
bool	IsVerified () const
std::string	CharStarToString (const char *cstr)
Public Member Functions inherited from garana::FlatG4Tree
	FlatG4Tree ()
	FlatG4Tree (TTree *tree)
	FlatG4Tree (TTree *tree, char opt)
const UInt_t	NSim () const override
	number of particles More...
const UInt_t	NPoints (const UInt_t &iparticle) const override
	number of G4 steps (i.e. trajectory points) More...
const UInt_t	NRegions (const UInt_t &iparticle) const override
	number of regions traversed by particle More...
const Int_t	Region (const UInt_t &iparticle, const UInt_t &iregion) const override
	region number More...
const vector< const TLorentzVector * > *	SimMomEnter (const UInt_t &iparticle) const override
	particle 4-momentum at entry point, all regions More...
const vector< const TLorentzVector * > *	SimMomExit (const UInt_t &iparticle) const override
	particle 4-momentum at exit point, all regions More...
const vector< const TLorentzVector * > *	SimPosEnter (const UInt_t &iparticle) const override
	particle 4-position at entry point, all regions More...
const vector< const TLorentzVector * > *	SimPosExit (const UInt_t &iparticle) const override
	particle 4-position at exit point, all regions More...
const TLorentzVector *	SimMomEnter (const UInt_t &iparticle, const UInt_t &iregion) const override
	particle 4-momentum at entry point in region More...
const TLorentzVector *	SimMomExit (const UInt_t &iparticle, const UInt_t &iregion) const override
	particle 4-momentum at exit point in region More...
const TLorentzVector *	SimPosEnter (const UInt_t &iparticle, const UInt_t &iregion) const override
	particle 4-position at entry point in region More...
const TLorentzVector *	SimPosExit (const UInt_t &iparticle, const UInt_t &iregion) const override
	particle 4-position at exit point in region More...
const bool	IsPrimary (const UInt_t &iparticle) const override
	did particle come from generator? More...
const Int_t	PDG (const UInt_t &iparticle) const override
	particle PDG code More...
const int	ParentPDG (const UInt_t &iparticle) const override
	parent particle's PDG code More...
const int	ProgenitorPDG (const UInt_t &iparticle) const override
	PDG of primary that led this one. More...
const int	TrackID (const UInt_t &iparticle) const override
	G4 track ID (can be <0 if it fell below trking threshold) More...
const int	ParentTrackID (const UInt_t &iparticle) const override
	G4 track ID of parent particle. More...
const int	ProgenitorTrackID (const UInt_t &iparticle) const override
	G4 track ID of primary that led this one. More...
const Int_t	ProcessI (const UInt_t &iparticle) const override
	code for process that created this one More...
const Int_t	ProcessF (const UInt_t &iparticle) const override
	code for process that killed this one More...
void	GetEntry (const UInt_t &ientry) override
const UInt_t	NSubEntries () const
Public Member Functions inherited from garana::G4Tree
virtual	~G4Tree ()
UInt_t const	GetTruthIndex (UInt_t iparticle) const
	index in gen tree subentry to truth match to this More...
const UInt_t	NPrimary () const
bool	HasPassedTPC (const UInt_t &iparticle) const
	did the G4Particle pass through any TPC drift volume(s)? More...
bool	HasPassedCalo (const UInt_t &iparticle) const
	did the G4Particle pass through any active ECal volume(s)? More...
bool	IsStoppedTPC (const UInt_t &iparticle) const
	did the G4Particle stop/decay in any TPC drift volume(s)? More...
bool	IsStoppedCalo (const UInt_t &iparticle) const
	did the G4Particle stop/decay in any active ECal volume(s)? More...
bool	IsContainedTPC (const UInt_t &iparticle) const
	if the G4Particle was produced in any TPC drift volume, does it remain in either drift volume? More...
bool	IsContainedCalo (const UInt_t &iparticle) const
	if the G4Particle was produced in any active ECal volume, does it remain there? More...
bool	IsCathodeCrosser (const UInt_t &iparticle) const
	did the G4Particle cross the TPC central cathode? More...
bool	IsContainedTPCEvent () const
	do all particles produced in any TPC drift volume in this event remain in either volume? More...
bool	IsContainedTPCPrimaries () const
	do all primaries produced in any TPC drift volume in this event remain in either volume? More...
bool	IsContainedCaloEvent () const
	do all particles produced in any active ECal volume in this event remain there? More...
bool	IsContainedCaloPrimaries () const
	do all primaries produced in any active ECal volume in this event remain there? More...
const TLorentzVector *	SimMomBegin (const UInt_t &iparticle) const
const TLorentzVector *	SimMomEnd (const UInt_t &iparticle) const
const TLorentzVector *	SimPosBegin (const UInt_t &iparticle) const
const TLorentzVector *	SimPosEnd (const UInt_t &iparticle) const
Public Member Functions inherited from garana::TreeReader
virtual	~TreeReader ()
void	SetupRead (TTree *tree)
TTree *	GetInputTree ()
size_t	NEntries () const
Int_t	Event () const
const TObjArray *	GetBranchList () const
void	Fill ()
void	Write ()
void	CheckOpt (char opt)
bool	BlockWrite () const

Private Types
enum	G4Branch { kEvent, kG4Particles, kG4TruthIndex }

Private Member Functions
bool	VerifyBranches () override

Private Attributes
StructuredG4Tree *	fG4In = nullptr
std::map< std::string, G4Branch >	nameToG4Branch
std::map< G4Branch, bool >	branchToDrawOpt

Additional Inherited Members
Protected Member Functions inherited from garana::FlatG4Tree
bool	SetBranchAddresses () override
void	SetVecs ()
void	ClearVecs ()
void	SetLimits ()
void	FillIndexMap ()
const UInt_t	LocalToGlobalIndex (const UInt_t &iparticle) const
Protected Attributes inherited from garana::Mill
bool	fIsVerified = false
TTree *	fTreeOut = nullptr
TFile *	fOutfile = nullptr
Protected Attributes inherited from garana::FlatG4Tree
std::map< UInt_t, std::pair< UInt_t, UInt_t > >	fLimits
	for each array for a given particle More...
std::map< UInt_t, UInt_t >	fLocalToGlobalIndex
UInt_t	fNSim = UINT_MAX
	number of G4 particles per event More...
vector< UInt_t > *	fNPts = nullptr
	number of 4-vector "snapshots" (G4 steps) More...
vector< UInt_t > *	fNRegions = nullptr
	number of regions of interest cross by the particle More...
vector< Int_t > *	fRegions = nullptr
	region IDs More...
vector< Int_t > *	fTrkID = nullptr
	particle's G4 trackID More...
vector< Int_t > *	fPDG = nullptr
	particle's PDG code More...
vector< Int_t > *	fParentPdg = nullptr
	particle parent's PDG code More...
vector< Int_t > *	fProgenitorPdg = nullptr
	FS particle from gen stage that led to this one. More...
vector< Int_t > *	fParentTrackId = nullptr
	particle's parent's trackID More...
vector< Int_t > *	fProgenitorTrackId = nullptr
	FS particle from gen stage that led to this one. More...
vector< Int_t > *	fProcessI = nullptr
	process that produced the particle More...
vector< Int_t > *	fProcessF = nullptr
	process that killed the particle More...
vector< Float_t > *	fX = nullptr
	particle's x-position in lab frame [cm] More...
vector< Float_t > *	fY = nullptr
	particle's y-position in lab frame [cm] More...
vector< Float_t > *	fZ = nullptr
	particle's z-position in lab frame [cm] More...
vector< Float_t > *	fT = nullptr
	particle's time in lab frame [ns] More...
vector< Float_t > *	fPx = nullptr
	particle's x-momentum in lab frame [GeV/c] More...
vector< Float_t > *	fPy = nullptr
	particle's y-momentum in lab frame [GeV/c] More...
vector< Float_t > *	fPz = nullptr
	particle's z-momentum in lab frame [GeV/c] More...
vector< Float_t > *	fE = nullptr
	particle's total energy in lab frame [GeV] More...
TBranch *	b_NSim = nullptr
TBranch *	b_NPts = nullptr
TBranch *	b_NRegions = nullptr
TBranch *	b_Regions = nullptr
TBranch *	b_TrkID = nullptr
TBranch *	b_PDG = nullptr
TBranch *	b_ParentPdg = nullptr
TBranch *	b_ProgenitorPdg = nullptr
TBranch *	b_ParentTrackId = nullptr
TBranch *	b_ProgenitorTrackId = nullptr
TBranch *	b_ProcessI = nullptr
TBranch *	b_ProcessF = nullptr
TBranch *	b_X = nullptr
TBranch *	b_Y = nullptr
TBranch *	b_Z = nullptr
TBranch *	b_T = nullptr
TBranch *	b_Px = nullptr
TBranch *	b_Py = nullptr
TBranch *	b_Pz = nullptr
TBranch *	b_E = nullptr
Protected Attributes inherited from garana::G4Tree
vector< UInt_t > *	fG4TruthIndex = nullptr
vector< UInt_t > *	fG4FSIndex = nullptr
TBranch *	b_G4TruthIndex = nullptr
TBranch *	b_G4FSIndex = nullptr
Protected Attributes inherited from garana::TreeReader
const std::string	treename
char	fOpt = 'r'
TTree *	fTreeIn = nullptr
	pointer to the analyzed TTree or TChain More...
UInt_t	fCurrentEntry = UINT_MAX
TBranch *	b_Event = nullptr
Int_t	fEvent = -1
	event number for tree entry More...

Detailed Description

Definition at line 16 of file MillG4Tree.h.

Member Enumeration Documentation

enum garana::MillG4Tree::G4Branch

private

Enumerator
kEvent
kG4Particles
kG4TruthIndex

Definition at line 34 of file MillG4Tree.h.

                              {
                        kEvent,
                        kG4Particles,
                        kG4TruthIndex
                };//

Constructor & Destructor Documentation

garana::MillG4Tree::MillG4Tree ( )

inline

Definition at line 20 of file MillG4Tree.h.

{};

MillG4Tree::MillG4Tree	(	TTree *	treeIn,
		TTree *	treeOut
	)

Definition at line 20 of file MillG4Tree.cxx.

{
    fG4In = new StructuredG4Tree(treeIn);

    fTreeIn = treeOut;
    fOpt='w';
        this->VerifyBranches();
        this->SetVecs();
        this->SetBranchAddresses();
        this->MillTrees();

}//constructor()

Member Function Documentation

void MillG4Tree::MillTrees ( )

overridevirtual

Implements garana::Mill.

Definition at line 34 of file MillG4Tree.cxx.

                           {

    if(!IsVerified()){
        std::cerr << "WARNING(MillG4Tree::MillTrees): "
                          << "trying to mill trees that have not been verified"
                          << std::endl;
        return;
    }

    std::cout << "MillG4Tree: loop over " << fG4In->NEntries() << " gen entries" << std::endl;

    //loop over input g4Tree events (entries)
    for(size_t ientry=0; ientry<fG4In->NEntries(); ientry++){

        ClearVecs(); //clear previous entry's data (if any)
        fG4In->GetEntry(ientry);

        if(branchToDrawOpt[kEvent]) {
                fEvent = fG4In->Event();
        }

        if(branchToDrawOpt[kG4Particles]) {
            fNSim = fG4In->NSim();
                        std::cout << "MillG4: output NG4 = " << fNSim << std::endl;

                        //loop over g4 entries
                std::cout << "MillG4Tree: loop over " << fG4In->NSim() << " isim" <<std::endl;
                        for(size_t isim=0; isim<fG4In->NSim(); isim++){

                                auto posenter = fG4In->SimPosEnter(isim); // 4-position for particle at entry, all regions
                                auto momenter = fG4In->SimMomEnter(isim); // 4-momentum for particle at entry, all regions
                                auto posexit  = fG4In->SimPosExit(isim); // 4-position for particle at exit/end, all regions
                                auto momexit = fG4In->SimMomExit(isim); // 4-momentum for particle at exit, all regions

                                for(size_t ireg=0; ireg<fG4In->NRegions(isim); ireg++){
                                        for(size_t i=0; i<2; i++){
                                                fNPts              ->push_back(fG4In->NPoints(isim));
                                                fTrkID             ->push_back(fG4In->TrackID(isim));
                                                fPDG               ->push_back(fG4In->PDG(isim));
                                                fParentPdg         ->push_back(fG4In->ParentPDG(isim));
                                                fProgenitorPdg     ->push_back(fG4In->ProgenitorPDG(isim));
                                                fParentTrackId     ->push_back(fG4In->ParentTrackID(isim));
                                                fProgenitorTrackId ->push_back(fG4In->ProgenitorTrackID(isim));
                                                fProcessI          ->push_back(fG4In->ProcessI(isim));
                                                fProcessF          ->push_back(fG4In->ProcessF(isim));
                                                if(i==0) {
                                                        fX                 ->push_back(posenter->at(ireg)->X());
                                                        fY                 ->push_back(posenter->at(ireg)->Y());
                                                        fZ                 ->push_back(posenter->at(ireg)->Z());
                                                        fT                 ->push_back(posenter->at(ireg)->T());
                                                        fPx                ->push_back(momenter->at(ireg)->Px());
                                                        fPy                ->push_back(momenter->at(ireg)->Py());
                                                        fPz                ->push_back(momenter->at(ireg)->Pz());
                                                        fE                 ->push_back(momenter->at(ireg)->E());
                                                }
                                                else {
                                                        fX                 ->push_back(posexit->at(ireg)->X());
                                                        fY                 ->push_back(posexit->at(ireg)->Y());
                                                        fZ                 ->push_back(posexit->at(ireg)->Z());
                                                        fT                 ->push_back(posexit->at(ireg)->T());
                                                        fPx                ->push_back(momexit->at(ireg)->Px());
                                                        fPy                ->push_back(momexit->at(ireg)->Py());
                                                        fPz                ->push_back(momexit->at(ireg)->Pz());
                                                        fE                 ->push_back(momexit->at(ireg)->E());
                                                }
                                        }
                                }// for regions

                            if(branchToDrawOpt[kG4TruthIndex]) {
                                fG4TruthIndex->push_back(fG4In->GetTruthIndex(isim));
                            }

            }//for g4 entries
                }//if branch present

        fTreeIn->Fill(); //actually our fTreeOut, but it's called fTreeIn in FlatG4Tree.h, fill once per event

    }//for genTree entries

    fTreeIn->Write();

}//Mill()

bool MillG4Tree::VerifyBranches ( )

overrideprivatevirtual

Implements garana::Mill.

Definition at line 117 of file MillG4Tree.cxx.

                                {

    // get list of branches and check it matches what we expect
    const TObjArray* branches = fG4In->GetBranchList();
    std::cout << "got ObjArray of branches from fTreeIn" << std::endl;
    try {
        if(!branches || branches->GetEntries()==0 )
                throw branches;
    }
    catch(TObjArray* branches){
        std::cerr << "ERROR(MillG4Tree::VerifyBranches): no branches found in passed input tree"
                          << std::endl;
        return false;
    }

    if(branches->GetEntriesFast() != (Int_t)nameToG4Branch.size())
        std::cout << "WARNING(MillG4Tree::VerifyBranches): Mismatch in number of branches (expected "
                  << nameToG4Branch.size() << " but found " << branches->GetEntriesFast()
                  << ")" << std::endl;
    else
        std::cout << "found genTree with " << branches->GetEntriesFast() << " branches" << std::endl;

    // loop over branches
    TIter next(branches);
    TBranch* branch = nullptr;
    while( (branch=(TBranch*)next())) {

        // check if branch is expected
        if(nameToG4Branch.find(CharStarToString(branch->GetFullName()))!=nameToG4Branch.end()) {

                std::cout << "   chopping branch '" << branch->GetFullName() << "'" << std::endl;

                if(nameToG4Branch[CharStarToString(branch->GetFullName())] == kEvent)
                        branchToDrawOpt[kEvent] = true;

                if(nameToG4Branch[CharStarToString(branch->GetFullName())] == kG4Particles)
                        branchToDrawOpt[kG4Particles] = true;

                if(nameToG4Branch[CharStarToString(branch->GetFullName())] == kG4TruthIndex)
                        branchToDrawOpt[kG4TruthIndex] = true;

        }//endif known branch

        else{
                std::cout << "WARNING(MillG4Tree): ignoring unknown branch '"
                                  << branch->GetFullName() << "'" << std::endl;
        }//else

    }//for branches

    fIsVerified = true;
    return true;

}//VerifyBranches()

Member Data Documentation

std::map<G4Branch,bool> garana::MillG4Tree::branchToDrawOpt

private

Initial value:

= {
                    {kEvent,        false},
                        {kG4Particles,  false},
                        {kG4TruthIndex, false}
                }

Definition at line 47 of file MillG4Tree.h.

StructuredG4Tree* garana::MillG4Tree::fG4In = nullptr

private

Definition at line 30 of file MillG4Tree.h.

std::map<std::string,G4Branch> garana::MillG4Tree::nameToG4Branch

private

Initial value:

=       {
                        {"Event",        kEvent},
                        {"G4Particles",  kG4Particles},
                        {"G4TruthIndex", kG4TruthIndex}

                }

Definition at line 40 of file MillG4Tree.h.

---

The documentation for this class was generated from the following files:

• garana/garana/Processors/MillG4Tree.h
• garana/garana/Processors/MillG4Tree.cxx