FlavorTagDiscriminants::TruthDecoratorHelpers Namespace Reference

Namespaces
	TruthSource
	TruthType

Functions
bool	sort_particles (const xAOD::IParticle *particle_A, const xAOD::IParticle *particle_B)
int	get_truth_type (const xAOD::TruthParticle *truth_particle)
int	get_source_type (const int origin)
bool	is_bc_hadron (const xAOD::TruthParticle *truth_particle, int flavour)
bool	is_weakly_decaying_hadron (const xAOD::TruthParticle *truth_particle)
bool	is_weakly_decaying_hadron (const xAOD::TruthParticle *truth_particle, int flavour)
const xAOD::TruthParticle *	get_parent_hadron (const xAOD::TruthParticle *truth_particle, bool user_called=true, int depth=0)
const xAOD::TruthVertex *	get_truth_vertex (const xAOD::TruthParticle *truth)
float	get_distance (const xAOD::TruthVertex *vertex_A, const xAOD::TruthVertex *vertex_B)
int	get_vertex_index (const xAOD::TruthVertex *vertex, const xAOD::TruthVertex *truth_PV, std::vector< const xAOD::TruthVertex * > &seen_vertices, const float truthVertexMergeDistance)

Function Documentation

get_distance()

float FlavorTagDiscriminants::TruthDecoratorHelpers::get_distance	(	const xAOD::TruthVertex *	vertex_A,
		const xAOD::TruthVertex *	vertex_B
	)

Definition at line 29 of file TruthDecoratorHelpers.cxx.

                                                                                             {
            if ( !vertex_A || !vertex_B ) { return 999.0; }
            return (vertex_A->v4().Vect() - vertex_B->v4().Vect()).Mag();
        }

get_parent_hadron()

const xAOD::TruthParticle * FlavorTagDiscriminants::TruthDecoratorHelpers::get_parent_hadron	(	const xAOD::TruthParticle *	truth_particle,
		bool	user_called = true,
		int	depth = 0
	)

Definition at line 54 of file TruthDecoratorHelpers.cxx.

                                                                                                                         {
            // get the weakly decaying parent hadron of truth_particle
            // check for sensible input
            if ( truth_particle == nullptr ) { return nullptr; }
            // loop protection
            if (depth>30) { return nullptr; }
            // if this is the weakly decaying hadron, stop here
            if ( !user_called && is_weakly_decaying_hadron(truth_particle) )  {
                return truth_particle;
            }
            for (unsigned int p = 0; p < truth_particle->nParents(); p++) {
                const auto parent = truth_particle->parent(p);
                if(parent == truth_particle) continue;// avoid infinite recursion
                const auto parent_hadron = get_parent_hadron(parent, false, depth+1);
                if ( parent_hadron != nullptr ) {
                    return parent_hadron;
                }
            }
            return nullptr;
        }

get_source_type()

int FlavorTagDiscriminants::TruthDecoratorHelpers::get_source_type

(

const int

origin

)

Definition at line 88 of file TruthDecoratorHelpers.cxx.

                                              {
            /* this label gives information about the origin of secondary
            particles (material interactions, gamme conversions, etc.)*/
 
            if (!(InDet::TrkOrigin::isSecondary(origin))) return TruthSource::Label::NotSecondary;
            if ( InDet::TrkOrigin::isHadronicInteraction(origin) ) return TruthSource::Label::HadronicInteraction;
            if ( InDet::TrkOrigin::isStrangeMesonDecay(origin) ) return TruthSource::Label::StrangeMesonDecay;
            if ( InDet::TrkOrigin::isStrangeBaryonDecay(origin) ) return TruthSource::Label::StrangeBaryonDecay;
            if ( InDet::TrkOrigin::isGammaConversion(origin) ) return TruthSource::Label::GammaConversion;
            // For simulation tracks not included in the above categories
            return TruthSource::Label::Other;
        }

get_truth_type()

int FlavorTagDiscriminants::TruthDecoratorHelpers::get_truth_type

(

const xAOD::TruthParticle *

truth_particle

)

Definition at line 75 of file TruthDecoratorHelpers.cxx.

                                                                    {
            if (!truth_particle) return TruthType::Label::NoTruth;
            // simple pdgid check for pion based on
            // PhysicsAnalysis/MCTruthClassifier/Root/MCTruthClassifierGen.cxx#L1159
            if (std::abs(truth_particle->pdgId()) == 211) return TruthType::Label::Pion * truth_particle->charge();
            if (truth_particle->isStrangeMeson()) return TruthType::Label::Kaon * truth_particle->charge();
            if (std::abs(truth_particle->pdgId()) == 3122) return TruthType::Label::Lambda;
            if (truth_particle->isElectron()) return TruthType::Label::Electron * truth_particle->charge() * -1;
            if (truth_particle->isMuon()) return TruthType::Label::Muon * truth_particle->charge() * -1;
            if (truth_particle->isPhoton()) return TruthType::Label::Photon;
            return TruthType::Label::Other;
        }

get_truth_vertex()

const xAOD::TruthVertex * FlavorTagDiscriminants::TruthDecoratorHelpers::get_truth_vertex

(

const xAOD::TruthParticle *

truth

)

Definition at line 16 of file TruthDecoratorHelpers.cxx.

                                                                                {
            // no truth
            if ( not truth ) { return nullptr; }
 
            // no vertex
            const xAOD::TruthVertex* truth_vertex = truth->prodVtx();
            if ( not truth_vertex || truth_vertex->perp() > 440.0 ) {
                return nullptr;
            }
 
            return truth_vertex;
        }

get_vertex_index()

int FlavorTagDiscriminants::TruthDecoratorHelpers::get_vertex_index	(	const xAOD::TruthVertex *	vertex,
		const xAOD::TruthVertex *	truth_PV,
		std::vector< const xAOD::TruthVertex * > &	seen_vertices,
		const float	truthVertexMergeDistance
	)

Definition at line 101 of file TruthDecoratorHelpers.cxx.

                                                                   {
            // no vertex
            if (!this_vertex) {
                return -2;
            }
            // primary vertex
            if (get_distance(this_vertex, truth_PV) < truthVertexMergeDistance) {
                return 0;
            }
            // have we already seen this vertex?
            for ( size_t i = 0; i != seen_vertices.size(); i++) {
                float dr = get_distance(seen_vertices.at(i), this_vertex);
                if ( dr < truthVertexMergeDistance ) {
                    // a vertex is nearby, reuse it
                    return i + 1;
                }
            }
            seen_vertices.push_back(this_vertex);
            return seen_vertices.size();
        }

is_bc_hadron()

bool FlavorTagDiscriminants::TruthDecoratorHelpers::is_bc_hadron	(	const xAOD::TruthParticle *	truth_particle,
		int	flavour
	)

Definition at line 34 of file TruthDecoratorHelpers.cxx.

                                                                                {
            if( truth_particle == nullptr ) { return false; }
            if( flavour == 5 && truth_particle->isBottomHadron() ) { return true; }
            if( flavour == 4 && truth_particle->isCharmHadron()  ) { return true; }
            return false;
        }

is_weakly_decaying_hadron() [1/2]

bool FlavorTagDiscriminants::TruthDecoratorHelpers::is_weakly_decaying_hadron

(

const xAOD::TruthParticle *

truth_particle

)

Definition at line 50 of file TruthDecoratorHelpers.cxx.

                                                                                {
            return is_weakly_decaying_hadron(truth_particle, 5) || is_weakly_decaying_hadron(truth_particle, 4);
        }

is_weakly_decaying_hadron() [2/2]

bool FlavorTagDiscriminants::TruthDecoratorHelpers::is_weakly_decaying_hadron	(	const xAOD::TruthParticle *	truth_particle,
		int	flavour
	)

Definition at line 41 of file TruthDecoratorHelpers.cxx.

                                                                                             {
          if (!is_bc_hadron(truth_particle, flavour)) return false;
          if (!truth_particle->hasDecayVtx() ) return false;
          for ( const auto out_part: truth_particle->decayVtx()->particles_out()) {
             if ( is_bc_hadron(out_part, flavour) ) return  false;
          }
          return true;
        }

sort_particles()

bool FlavorTagDiscriminants::TruthDecoratorHelpers::sort_particles	(	const xAOD::IParticle *	particle_A,
		const xAOD::IParticle *	particle_B
	)

Definition at line 12 of file TruthDecoratorHelpers.cxx.

                                                                                              {
            return particle_A->pt() < particle_B->pt();
        }