Inheritance diagram for t0::DirectHitParticleAssns:

Classes
struct	TrackIDEinfo

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

	~DirectHitParticleAssns ()
	Destructor. More...

void	reconfigure (fhicl::ParameterSet const &pset) override

void	CreateHitParticleAssociations (art::Event &, HitParticleAssociations *) override
	This rebuilds the internal maps. More...

Private Attributes
art::InputTag	fHitModuleLabel

art::InputTag	fMCParticleModuleLabel

std::unordered_map< int, TrackIDEinfo >	fTrkIDECollector

Detailed Description

Definition at line 37 of file DirectHitParticleAssns_tool.cc.

Constructor & Destructor Documentation

t0::DirectHitParticleAssns::DirectHitParticleAssns ( fhicl::ParameterSet const & pset )

explicit

Constructor.

Parameters

pset	Constructor.

Arguments:

pset - Fcl parameters.

Definition at line 79 of file DirectHitParticleAssns_tool.cc.

References reconfigure().

 {
     reconfigure(pset);
     
     // Report.
     mf::LogInfo("DirectHitParticleAssns") << "Configured\n";
 }

t0::DirectHitParticleAssns::~DirectHitParticleAssns ( )

Destructor.

Definition at line 89 of file DirectHitParticleAssns_tool.cc.

90 {}

Member Function Documentation

void t0::DirectHitParticleAssns::CreateHitParticleAssociations	(	art::Event &	evt,
		HitParticleAssociations *	hitPartAssns
	)

overridevirtual

This rebuilds the internal maps.

Rebuild method -> rebuild the basic maps to get truth information

Arguments:

event - the art event used to extract all information

Implements t0::IHitParticleAssociations.

Definition at line 112 of file DirectHitParticleAssns_tool.cc.

References art::Assns< L, R, D >::addSingle(), DEFINE_ART_CLASS_TOOL, anab::BackTrackerHitMatchingData::energy, fHitModuleLabel, fMCParticleModuleLabel, fTrkIDECollector, art::DataViewImpl::getByLabel(), art::DataViewImpl::getValidHandle(), cheat::BackTrackerService::HitToTrackIDEs(), anab::BackTrackerHitMatchingData::ideFraction, anab::BackTrackerHitMatchingData::ideNFraction, anab::BackTrackerHitMatchingData::isMaxIDE, anab::BackTrackerHitMatchingData::isMaxIDEN, and anab::BackTrackerHitMatchingData::numElectrons.

 {
     // This function handles the "direct" creation of hit<-->MCParticle associations through use of the BackTracker
     //
     auto mcpartHandle = evt.getValidHandle< std::vector<simb::MCParticle> >(fMCParticleModuleLabel);
     
     art::Handle< std::vector<recob::Hit> > hitListHandle;
     evt.getByLabel(fHitModuleLabel,hitListHandle);
     
     if(!hitListHandle.isValid()){
         std::cerr << "Hit handle is not valid! Returning empty collection" << std::endl;
         return;
     }
     
     // Access art services...
     art::ServiceHandle<cheat::BackTrackerService> btService;
     art::ServiceHandle<cheat::ParticleInventoryService> piService;
 
     double maxe = -1;
     double tote = 0;
     // int    trkid = -1;
     int    maxtrkid = -1;
     double maxn = -1;
     double totn = 0;
     int maxntrkid = -1;
     anab::BackTrackerHitMatchingData bthmd;
     std::unordered_map<int,int> trkid_lookup; //indexed by geant4trkid, delivers MC particle location
     
     auto const& hitList(*hitListHandle);
     auto const& mcpartList(*mcpartHandle);
     
     for(size_t i_h=0; i_h<hitList.size(); ++i_h){
         art::Ptr<recob::Hit> hitPtr(hitListHandle, i_h);
         auto trkide_list = btService->HitToTrackIDEs(hitPtr);
         
         maxe = -1; tote = 0; maxtrkid = -1;
         maxn = -1; totn = 0; maxntrkid = -1;
         fTrkIDECollector.clear();
         
         //for(auto const& t : trkide_list){
         for(size_t i_t=0; i_t<trkide_list.size(); ++i_t){
             auto const& t(trkide_list[i_t]);
             fTrkIDECollector[t.trackID].E += t.energy;
             tote += t.energy;
             if(fTrkIDECollector[t.trackID].E>maxe) { maxe = fTrkIDECollector[t.trackID].E; maxtrkid = t.trackID; }
             fTrkIDECollector[t.trackID].NumElectrons += t.numElectrons;
             totn += t.numElectrons;
             if(fTrkIDECollector[t.trackID].NumElectrons > maxn) {
                 maxn = fTrkIDECollector[t.trackID].NumElectrons;
                 maxntrkid = t.trackID;
             }
             
             //if not found, find mc particle...
             if(trkid_lookup.find(t.trackID)==trkid_lookup.end()){
                 size_t i_p=0;
                 while(i_p<mcpartList.size()){
                     if(mcpartList[i_p].TrackId() == abs(t.trackID)) { trkid_lookup[t.trackID] = (int)i_p; break;}
                     ++i_p;
                 }
                 if(i_p==mcpartList.size()) trkid_lookup[t.trackID] = -1;
             }
             
         }
         //end loop on TrackIDs
         
         //now find the mcparticle and loop back through ...
         for(auto const& t : fTrkIDECollector){
             int mcpart_i = trkid_lookup[t.first];
             if(mcpart_i==-1) continue; //no mcparticle here
             art::Ptr<simb::MCParticle> mcpartPtr(mcpartHandle, mcpart_i);
             bthmd.ideFraction = t.second.E / tote;
             bthmd.isMaxIDE = (t.first==maxtrkid);
             bthmd.ideNFraction = t.second.NumElectrons / totn;
             bthmd.isMaxIDEN = ( t.first == maxntrkid );
             bthmd.energy = t.second.E;
             bthmd.numElectrons = t.second.NumElectrons;
             hitPartAssns->addSingle(mcpartPtr, hitPtr, bthmd);
         }
         
     }//end loop on hits
     
     return;
 }

void t0::DirectHitParticleAssns::reconfigure ( fhicl::ParameterSet const & pset )

overridevirtual

Reconfigure method.

Arguments:

pset - Fcl parameter set.

Implements t0::IHitParticleAssociations.

Definition at line 99 of file DirectHitParticleAssns_tool.cc.

References fHitModuleLabel, fMCParticleModuleLabel, and fhicl::ParameterSet::get().

Referenced by DirectHitParticleAssns().

 {
     fMCParticleModuleLabel = pset.get<art::InputTag>("MCParticleLabel");
     fHitModuleLabel        = pset.get<art::InputTag>("HitModuleLabel");
 }