#include <PFParticleVarExtractor.h>

Inheritance diagram for VLN::PFParticleVarExtractor:

Public Member Functions
	PFParticleVarExtractor (const std::string &prefix, calo::CalorimetryAlg &algCalorimetry, const std::string &labelPFPModule, const std::string &labelPFPTrack, const std::string &labelPFPShower, unsigned int plane=2)

	~PFParticleVarExtractor ()=default

Public Member Functions inherited from VLN::VarExtractorBase
	VarExtractorBase (const std::string &prefix, const std::vector< std::string > &scalarVars, const std::vector< std::string > &vectorVars)

virtual	~VarExtractorBase ()=default

virtual void	extract (const art::Event &evt, VarDict &vars)

Protected Member Functions
void	extractVars (const art::Event &evt, VarDict &vars) override

Protected Member Functions inherited from VLN::VarExtractorBase
void	setScalarVar (VarDict &vars, const std::string &name, double value) const

void	appendToVectorVar (VarDict &vars, const std::string &name, double value) const

void	initScalarVars (VarDict &vars, const std::vector< std::string > &names) const

void	initVectorVars (VarDict &vars, const std::vector< std::string > &names) const

Private Member Functions
void	extractShowerVars (const art::Event &evt, const art::Ptr< recob::PFParticle > &particle, VarDict &vars)

void	extractTrackVars (const art::Event &evt, const art::Ptr< recob::PFParticle > &particle, VarDict &vars)

void	extractBasicVars (const art::Event &evt, const std::vector< art::Ptr< recob::Hit >> &hits, VarDict &vars)

Private Attributes
calo::CalorimetryAlg &	algCalorimetry

std::string	labelPFPModule

std::string	labelPFPTrack

std::string	labelPFPShower

unsigned int	plane

Additional Inherited Members
Protected Attributes inherited from VLN::VarExtractorBase
std::string	prefix

std::vector< std::string >	scalarVars

std::vector< std::string >	vectorVars

Detailed Description

Definition at line 9 of file PFParticleVarExtractor.h.

Constructor & Destructor Documentation

VLN::PFParticleVarExtractor::PFParticleVarExtractor	(	const std::string &	prefix,
		calo::CalorimetryAlg &	algCalorimetry,
		const std::string &	labelPFPModule,
		const std::string &	labelPFPTrack,
		const std::string &	labelPFPShower,
		unsigned int	plane = `2`
	)

Definition at line 29 of file PFParticleVarExtractor.cxx.

   : VarExtractorBase(prefix, SCALAR_VARS, VECTOR_VARS),
     algCalorimetry(algCalorimetry),
     labelPFPModule(labelPFPModule),
     labelPFPTrack(labelPFPTrack),
     labelPFPShower(labelPFPShower),
     plane(plane)
 { }

VLN::PFParticleVarExtractor::~PFParticleVarExtractor ( )

default

Member Function Documentation

void VLN::PFParticleVarExtractor::extractBasicVars	(	const art::Event &	evt,
		const std::vector< art::Ptr< recob::Hit >> &	hits,
		VarDict &	vars
	)

private

Definition at line 105 of file PFParticleVarExtractor.cxx.

 {
     const auto chargeCalE = calcHitsChargeCalE(
         hits, evt, algCalorimetry, plane
     );
 
     appendToVectorVar(vars, "nHit",   hits.size());
     appendToVectorVar(vars, "charge", chargeCalE.first);
     appendToVectorVar(vars, "calE",   chargeCalE.second);
 }

void VLN::PFParticleVarExtractor::extractShowerVars	(	const art::Event &	evt,
		const art::Ptr< recob::PFParticle > &	particle,
		VarDict &	vars
	)

private

Definition at line 73 of file PFParticleVarExtractor.cxx.

 {
     auto shower = dune_ana::DUNEAnaPFParticleUtils::GetShower(
         particle, evt, labelPFPModule, labelPFPShower
     );
     auto start  = shower->ShowerStart();
     auto dir    = shower->Direction();
     auto energy = shower->Energy();
 
     appendToVectorVar(vars, "is_shower", 1);
     appendToVectorVar(vars, "length",    shower->Length());
     appendToVectorVar(vars, "start.x",   start.x());
     appendToVectorVar(vars, "start.y",   start.y());
     appendToVectorVar(vars, "start.z",   start.z());
     appendToVectorVar(vars, "dir.x",     dir.x());
     appendToVectorVar(vars, "dir.y",     dir.y());
     appendToVectorVar(vars, "dir.z",     dir.z());
     appendToVectorVar(
         vars, "energy",  (energy.size() < plane + 1) ? 0 : energy[plane]
     );
 
     const auto hits = dune_ana::DUNEAnaShowerUtils::GetHits(
         shower, evt, labelPFPShower
     );
 
     extractBasicVars(evt, hits, vars);
 }

void VLN::PFParticleVarExtractor::extractTrackVars	(	const art::Event &	evt,
		const art::Ptr< recob::PFParticle > &	particle,
		VarDict &	vars
	)

private

Definition at line 44 of file PFParticleVarExtractor.cxx.

 {
     auto track = dune_ana::DUNEAnaPFParticleUtils::GetTrack(
         particle, evt, labelPFPModule, labelPFPTrack
     );
     auto start = track->Start();
     auto dir   = track->StartDirection();
 
     appendToVectorVar(vars, "is_shower", 0);
     appendToVectorVar(vars, "length",    track->Length());
     appendToVectorVar(vars, "start.x",   start.x());
     appendToVectorVar(vars, "start.y",   start.y());
     appendToVectorVar(vars, "start.z",   start.z());
     appendToVectorVar(vars, "dir.x",     dir.x());
     appendToVectorVar(vars, "dir.y",     dir.y());
     appendToVectorVar(vars, "dir.z",     dir.z());
     appendToVectorVar(vars, "energy",    track->StartMomentum());
 
     const auto hits = dune_ana::DUNEAnaTrackUtils::GetHits(
         track, evt, labelPFPTrack
     );
 
     extractBasicVars(evt, hits, vars);
 }

void VLN::PFParticleVarExtractor::extractVars	(	const art::Event &	evt,
		VarDict &	vars
	)

overrideprotectedvirtual

Implements VLN::VarExtractorBase.

Definition at line 120 of file PFParticleVarExtractor.cxx.

 {
     const std::vector<art::Ptr<recob::PFParticle>> particles
         = dune_ana::DUNEAnaEventUtils::GetPFParticles(evt, labelPFPModule);
 
     for (const auto &particle : particles)
     {
         const bool isTrack = dune_ana::DUNEAnaPFParticleUtils::IsTrack(
             particle, evt, labelPFPModule, labelPFPTrack
         );
 
         const bool isShower = dune_ana::DUNEAnaPFParticleUtils::IsShower(
             particle, evt, labelPFPModule, labelPFPShower
         );
 
         if (isTrack == isShower) {
             /* TODO: Maybe handle this condition somehow? */
             continue;
         }
 
         if (isShower) {
             extractShowerVars(evt, particle, vars);
         }
         else if (isTrack) {
             extractTrackVars(evt, particle, vars);
         }
     }
 }

Member Data Documentation

calo::CalorimetryAlg& VLN::PFParticleVarExtractor::algCalorimetry

private

Definition at line 46 of file PFParticleVarExtractor.h.

std::string VLN::PFParticleVarExtractor::labelPFPModule

private

Definition at line 47 of file PFParticleVarExtractor.h.

std::string VLN::PFParticleVarExtractor::labelPFPShower

private

Definition at line 49 of file PFParticleVarExtractor.h.

std::string VLN::PFParticleVarExtractor::labelPFPTrack

private

Definition at line 48 of file PFParticleVarExtractor.h.

unsigned int VLN::PFParticleVarExtractor::plane

private

Definition at line 50 of file PFParticleVarExtractor.h.

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

dunereco/dunereco/VLNets/art/var_extractors/PFParticleVarExtractor.h
dunereco/dunereco/VLNets/art/var_extractors/PFParticleVarExtractor.cxx

Public Member Functions

Protected Member Functions

Private Member Functions

Private Attributes

Additional Inherited Members

Detailed Description

Constructor & Destructor Documentation

Member Function Documentation

Member Data Documentation