Inheritance diagram for ctp::CTPEvaluator:

Public Member Functions
	CTPEvaluator (fhicl::ParameterSet const &pset)
	Constructor. More...

virtual	~CTPEvaluator ()
	Destructor. More...

void	beginJob ()

void	endJob ()

void	produce (art::Event &evt)

Public Member Functions inherited from art::EDProducer
	EDProducer (fhicl::ParameterSet const &pset)

template<typename Config >
	EDProducer (Table< Config > const &config)

std::string	workerType () const

Public Member Functions inherited from art::detail::Producer
virtual	~Producer () noexcept

	Producer (fhicl::ParameterSet const &)

	Producer (Producer const &)=delete

	Producer (Producer &&)=delete

Producer &	operator= (Producer const &)=delete

Producer &	operator= (Producer &&)=delete

void	doBeginJob (SharedResources const &resources)

void	doEndJob ()

void	doRespondToOpenInputFile (FileBlock const &fb)

void	doRespondToCloseInputFile (FileBlock const &fb)

void	doRespondToOpenOutputFiles (FileBlock const &fb)

void	doRespondToCloseOutputFiles (FileBlock const &fb)

bool	doBeginRun (RunPrincipal &rp, ModuleContext const &mc)

bool	doEndRun (RunPrincipal &rp, ModuleContext const &mc)

bool	doBeginSubRun (SubRunPrincipal &srp, ModuleContext const &mc)

bool	doEndSubRun (SubRunPrincipal &srp, ModuleContext const &mc)

bool	doEvent (EventPrincipal &ep, ModuleContext const &mc, std::atomic< std::size_t > &counts_run, std::atomic< std::size_t > &counts_passed, std::atomic< std::size_t > &counts_failed)

Public Member Functions inherited from art::Modifier
	~Modifier () noexcept

	Modifier ()

	Modifier (Modifier const &)=delete

	Modifier (Modifier &&)=delete

Modifier &	operator= (Modifier const &)=delete

Modifier &	operator= (Modifier &&)=delete

Public Member Functions inherited from art::ModuleBase
virtual	~ModuleBase () noexcept

	ModuleBase ()

ModuleDescription const &	moduleDescription () const

void	setModuleDescription (ModuleDescription const &)

std::array< std::vector< ProductInfo >, NumBranchTypes > const &	getConsumables () const

void	sortConsumables (std::string const &current_process_name)

template<typename T , BranchType BT>
ViewToken< T >	consumesView (InputTag const &tag)

template<typename T , BranchType BT>
ViewToken< T >	mayConsumeView (InputTag const &tag)

Private Attributes
fhicl::ParameterSet	fHelperPars

CTPHelper	fConvTrackPID

std::string	fParticleLabel

bool	fWriteTree

std::vector< float >	fMuonScoreVector

std::vector< float >	fPionScoreVector

std::vector< float >	fProtonScoreVector

std::vector< int >	fPDGVector

std::vector< unsigned int >	fCaloPoints

TTree *	fPIDTree

Additional Inherited Members
Public Types inherited from art::EDProducer
using	ModuleType = EDProducer

using	WorkerType = WorkerT< EDProducer >

Public Types inherited from art::detail::Producer
template<typename UserConfig , typename KeysToIgnore = void>
using	Table = Modifier::Table< UserConfig, KeysToIgnore >

Public Types inherited from art::Modifier
template<typename UserConfig , typename UserKeysToIgnore = void>
using	Table = ProducerTable< UserConfig, detail::ModuleConfig, UserKeysToIgnore >

Static Public Member Functions inherited from art::EDProducer
static void	commitEvent (EventPrincipal &ep, Event &e)

Protected Member Functions inherited from art::ModuleBase
ConsumesCollector &	consumesCollector ()

template<typename T , BranchType = InEvent>
ProductToken< T >	consumes (InputTag const &)

template<typename Element , BranchType = InEvent>
ViewToken< Element >	consumesView (InputTag const &)

template<typename T , BranchType = InEvent>
void	consumesMany ()

template<typename T , BranchType = InEvent>
ProductToken< T >	mayConsume (InputTag const &)

template<typename Element , BranchType = InEvent>
ViewToken< Element >	mayConsumeView (InputTag const &)

template<typename T , BranchType = InEvent>
void	mayConsumeMany ()

Detailed Description

CTPEvaluator class.

Definition at line 39 of file CTPEvaluator_module.cc.

Constructor & Destructor Documentation

ctp::CTPEvaluator::CTPEvaluator ( fhicl::ParameterSet const & pset )

Constructor.

Parameters

pset

Definition at line 109 of file CTPEvaluator_module.cc.

                                                         : art::EDProducer(pset),
 fHelperPars(pset.get<fhicl::ParameterSet>("ctpHelper")),
 fConvTrackPID(fHelperPars),
 fParticleLabel(pset.get<std::string>("particleLabel")),
 fWriteTree(pset.get<bool>("writeTree"))
 {
     produces<std::vector<ctp::CTPResult>>();
     produces<art::Assns<recob::Track,ctp::CTPResult>>();
 }

ctp::CTPEvaluator::~CTPEvaluator ( )

virtual

Destructor.

Definition at line 121 of file CTPEvaluator_module.cc.

122 {

123 }

Member Function Documentation

void ctp::CTPEvaluator::beginJob ( )

virtual

Reimplemented from art::EDProducer.

Definition at line 127 of file CTPEvaluator_module.cc.

 {
     if (!fWriteTree) return;
   
     art::ServiceHandle<art::TFileService const> tfs;
   
     fPIDTree = tfs->make<TTree>("pidTree","pidTree");
     fPIDTree->Branch("muonScores",&fMuonScoreVector);
     fPIDTree->Branch("pionScores",&fPionScoreVector);
     fPIDTree->Branch("protonScores",&fProtonScoreVector);
     fPIDTree->Branch("pdgCodes",&fPDGVector);
     fPIDTree->Branch("caloPoints",&fCaloPoints);
 }

void ctp::CTPEvaluator::endJob ( )

virtual

Reimplemented from art::EDProducer.

Definition at line 143 of file CTPEvaluator_module.cc.

144 {

145 }

void ctp::CTPEvaluator::produce ( art::Event & evt )

virtual

Implements art::EDProducer.

Definition at line 149 of file CTPEvaluator_module.cc.

 {
     // Define containers for the things we're going to produce
     std::unique_ptr< std::vector<ctp::CTPResult> > resultCol(new std::vector<ctp::CTPResult>);
     std::unique_ptr< art::Assns<recob::Track,ctp::CTPResult> > trackResultAssn(new art::Assns<recob::Track,ctp::CTPResult>);
 
     if (fWriteTree)
     {
         fMuonScoreVector.clear();
         fPionScoreVector.clear();
         fProtonScoreVector.clear();
         fPDGVector.clear();
         fCaloPoints.clear();
     }
 
     // Get all of the PFParticles
     const std::vector<art::Ptr<recob::PFParticle>> particles = dune_ana::DUNEAnaEventUtils::GetPFParticles(evt,fParticleLabel);
 
     // Get the tracks too
     const std::string trkLabel = fHelperPars.get<std::string>("TrackLabel");
     const std::vector<art::Ptr<recob::Track>> tracks = dune_ana::DUNEAnaEventUtils::GetTracks(evt,trkLabel);
 
     for (const art::Ptr<recob::PFParticle> &particle : particles)
     {
         int thisTrackID = -999;
       
         // Get the track if this particle is track-like
         unsigned int nCaloPoints = 0;
         if (dune_ana::DUNEAnaPFParticleUtils::IsTrack(particle,evt,fParticleLabel,trkLabel))
         {
             const art::Ptr<recob::Track> trk = dune_ana::DUNEAnaPFParticleUtils::GetTrack(particle,evt,fParticleLabel,trkLabel);
             const std::string caloLabel = fHelperPars.get<std::string>("CalorimetryLabel");
             const art::Ptr<anab::Calorimetry> calo = dune_ana::DUNEAnaTrackUtils::GetCalorimetry(trk,evt,trkLabel,caloLabel);
 
             thisTrackID = trk.key();
             nCaloPoints = calo->dEdx().size();
         }
         else continue;
 
         // Returns a dummy value if not a track or not suitable
         CTPResult thisPID = fConvTrackPID.RunConvolutionalTrackPID(particle,evt);
         resultCol->push_back(thisPID);
         auto const ptrMaker = art::PtrMaker<ctp::CTPResult>(evt);
         art::Ptr<ctp::CTPResult> ptrResult = ptrMaker(resultCol->size()-1);
         art::Ptr<recob::Track> thisTrack = tracks.at(thisTrackID);
 
 //        std::cout << "Making association between track " << thisTrack.key() << " and PID result " << ptrResult.key() << std::endl;
 
         // Now to make the association
         util::CreateAssn(evt,ptrResult,thisTrack,*trackResultAssn);
 
         if (fWriteTree)
         {
             if (!thisPID.IsValid()) continue;
             int pdg = 0;
             if(!evt.isRealData()){
                 pdg = fConvTrackPID.GetTruePDGCode(particle,evt);
             }
             std::cout << "Got a track PID for particle of type " << pdg << ": " << thisPID.GetMuonScore() << ", " << thisPID.GetPionScore() << ", " << thisPID.GetProtonScore() << std::endl;       
             fMuonScoreVector.push_back(thisPID.GetMuonScore());
             fPionScoreVector.push_back(thisPID.GetPionScore());
             fProtonScoreVector.push_back(thisPID.GetProtonScore());
             fPDGVector.push_back(pdg);
             fCaloPoints.push_back(nCaloPoints);
         }
     }
     if (fWriteTree) fPIDTree->Fill();
 
     evt.put(std::move(resultCol));
     evt.put(std::move(trackResultAssn));
 }

Member Data Documentation

std::vector<unsigned int> ctp::CTPEvaluator::fCaloPoints

private

Definition at line 69 of file CTPEvaluator_module.cc.

CTPHelper ctp::CTPEvaluator::fConvTrackPID

private

Definition at line 61 of file CTPEvaluator_module.cc.

fhicl::ParameterSet ctp::CTPEvaluator::fHelperPars

private

Definition at line 60 of file CTPEvaluator_module.cc.

std::vector<float> ctp::CTPEvaluator::fMuonScoreVector

private

Definition at line 65 of file CTPEvaluator_module.cc.

std::string ctp::CTPEvaluator::fParticleLabel

private

Definition at line 62 of file CTPEvaluator_module.cc.

std::vector<int> ctp::CTPEvaluator::fPDGVector

private

Definition at line 68 of file CTPEvaluator_module.cc.

TTree* ctp::CTPEvaluator::fPIDTree

private

Definition at line 72 of file CTPEvaluator_module.cc.

std::vector<float> ctp::CTPEvaluator::fPionScoreVector

private

Definition at line 66 of file CTPEvaluator_module.cc.

std::vector<float> ctp::CTPEvaluator::fProtonScoreVector

private

Definition at line 67 of file CTPEvaluator_module.cc.

bool ctp::CTPEvaluator::fWriteTree

private

Definition at line 63 of file CTPEvaluator_module.cc.

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

dunereco/dunereco/TrackPID/modules/CTPEvaluator_module.cc

Public Member Functions

Private Attributes

Additional Inherited Members

Detailed Description

Constructor & Destructor Documentation

Member Function Documentation

Member Data Documentation