Inheritance diagram for nnet::ParticleDecayId:

Classes
struct	Config

Public Types
using	Parameters = art::EDProducer::Table< Config >

using	ModuleType = EDProducer

template<typename UserConfig , typename KeysToIgnore = void>
using	Table = Modifier::Table< UserConfig, KeysToIgnore >

Public Member Functions
	ParticleDecayId (Parameters const &p)

	ParticleDecayId (ParticleDecayId const &)=delete

	ParticleDecayId (ParticleDecayId &&)=delete

ParticleDecayId &	operator= (ParticleDecayId const &)=delete

ParticleDecayId &	operator= (ParticleDecayId &&)=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)

void	fillProductDescriptions ()

void	registerProducts (ProductDescriptions &productsToRegister)

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)

std::unique_ptr< Worker >	makeWorker (WorkerParams const &wp)

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

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

Protected Member Functions
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 ()

Private Member Functions
void	produce (art::Event &e) override

bool	DetectDecay (detinfo::DetectorClocksData const &clockData, detinfo::DetectorPropertiesData const &detProp, const std::vector< recob::Wire > &wires, const std::vector< art::Ptr< recob::Hit >> &hits, std::map< size_t, TVector3 > &spoints, std::vector< std::pair< TVector3, double >> &result)

Private Attributes
PointIdAlg	fPointIdAlg

art::InputTag	fWireProducerLabel

art::InputTag	fTrackModuleLabel

double	fRoiThreshold

double	fPointThreshold

int	fSkipView

Detailed Description

Definition at line 47 of file ParticleDecayId_module.cc.

Member Typedef Documentation

using art::EDProducer::ModuleType = EDProducer

inherited

Definition at line 17 of file EDProducer.h.

using nnet::ParticleDecayId::Parameters = art::EDProducer::Table<Config>

Definition at line 76 of file ParticleDecayId_module.cc.

template<typename UserConfig , typename KeysToIgnore = void>

using art::detail::Producer::Table = Modifier::Table<UserConfig, KeysToIgnore>

inherited

Definition at line 26 of file Producer.h.

Constructor & Destructor Documentation

nnet::ParticleDecayId::ParticleDecayId ( ParticleDecayId::Parameters const & config )

explicit

Definition at line 105 of file ParticleDecayId_module.cc.

References fPointIdAlg, fPointThreshold, fRoiThreshold, fSkipView, fTrackModuleLabel, and fWireProducerLabel.

     : EDProducer{config}
     , fPointIdAlg(config().PointIdAlg())
     , fWireProducerLabel(config().WireLabel())
     , fTrackModuleLabel(config().TrackModuleLabel())
     , fRoiThreshold(config().RoiThreshold())
     , fPointThreshold(config().PointThreshold())
     , fSkipView(config().SkipView())
   {
     produces<std::vector<recob::Vertex>>();
     produces<art::Assns<recob::Vertex, recob::Track>>();
   }

nnet::ParticleDecayId::ParticleDecayId ( ParticleDecayId const & )

delete

nnet::ParticleDecayId::ParticleDecayId ( ParticleDecayId && )

delete

Member Function Documentation

template<typename T , BranchType BT>

ProductToken< T > art::ModuleBase::consumes ( InputTag const & tag )

protectedinherited

Definition at line 61 of file ModuleBase.h.

References art::ModuleBase::collector_, and art::ConsumesCollector::consumes().

   {
     return collector_.consumes<T, BT>(tag);
   }

ConsumesCollector & art::ModuleBase::consumesCollector ( )

protectedinherited

Definition at line 57 of file ModuleBase.cc.

References art::ModuleBase::collector_.

   {
     return collector_;
   }

template<typename T , BranchType BT>

void art::ModuleBase::consumesMany ( )

protectedinherited

Definition at line 75 of file ModuleBase.h.

References art::ModuleBase::collector_, and art::ConsumesCollector::consumesMany().

   {
     collector_.consumesMany<T, BT>();
   }

template<typename Element , BranchType = InEvent>

ViewToken<Element> art::ModuleBase::consumesView ( InputTag const & )

protectedinherited

template<typename T , BranchType BT>

ViewToken<T> art::ModuleBase::consumesView ( InputTag const & tag )

inherited

Definition at line 68 of file ModuleBase.h.

References art::ModuleBase::collector_, and art::ConsumesCollector::consumesView().

   {
     return collector_.consumesView<T, BT>(tag);
   }

bool nnet::ParticleDecayId::DetectDecay	(	detinfo::DetectorClocksData const &	clockData,
		detinfo::DetectorPropertiesData const &	detProp,
		const std::vector< recob::Wire > &	wires,
		const std::vector< art::Ptr< recob::Hit >> &	hits,
		std::map< size_t, TVector3 > &	spoints,
		std::vector< std::pair< TVector3, double >> &	result
	)

private

Definition at line 179 of file ParticleDecayId_module.cc.

References c1, d, DEFINE_ART_MODULE, fPointIdAlg, fPointThreshold, fRoiThreshold, hits(), nnet::PointIdAlg::predictIdVector(), img::DataProviderAlg::setWireDriftData(), and util::size().

Referenced by produce().

   {
     const size_t nviews = 3;
 
     std::vector<art::Ptr<recob::Hit>> wire_drift[nviews];
     for (size_t i = 0; i < hits.size(); ++i) // split hits between views
     {
       wire_drift[hits[i]->View()].push_back(hits[i]);
     }
 
     std::vector<float> outputs[nviews];
     for (size_t v = 0; v < nviews;
          ++v) // calculate nn outputs for each view (hopefully not changing cryo/tpc many times)
     {
       outputs[v].resize(wire_drift[v].size(), 0);
       for (size_t i = 0; i < wire_drift[v].size(); ++i) {
         int tpc = wire_drift[v][i]->WireID().TPC;
         int cryo = wire_drift[v][i]->WireID().Cryostat;
 
         fPointIdAlg.setWireDriftData(clockData, detProp, wires, v, tpc, cryo);
 
         outputs[v][i] = fPointIdAlg.predictIdVector(wire_drift[v][i]->WireID().Wire,
                                                     wire_drift[v][i]->PeakTime())[0]; // p(decay)
       }
     }
 
     std::vector<std::pair<size_t, float>> candidates2d[nviews];
     std::vector<std::pair<TVector3, float>> candidates3d[nviews];
     for (size_t v = 0; v < nviews; ++v) {
       size_t idx = 0;
       while (idx < outputs[v].size()) {
         if (outputs[v][idx] > fRoiThreshold) {
           size_t ci = idx;
           float max = outputs[v][idx];
           ++idx;
 
           while ((idx < outputs[v].size()) && (outputs[v][idx] > fRoiThreshold)) {
             if (outputs[v][idx] > max) {
               max = outputs[v][idx];
               ci = idx;
             }
             ++idx;
           }
           candidates2d[v].emplace_back(ci, max);
           candidates3d[v].emplace_back(spoints[wire_drift[v][ci].key()], max);
         }
         else
           ++idx;
       }
     }
 
     double min_dist =
       2.0; // [cm], threshold for today to distinguish between two different candidates,
            // if belo threshold, then use 3D point corresponding to higher cnn output
 
     // need coincidence of high cnn out in two views, then look if there is another close candidate
     // and again select by cnn output value, would like to have few strong candidates
     bool found = false;
     for (size_t v = 0; v < nviews - 1; ++v) {
       for (size_t i = 0; i < candidates3d[v].size(); ++i) {
         TVector3 c0(candidates3d[v][i].first);
         float p0 = candidates3d[v][i].second;
 
         for (size_t u = v + 1; u < nviews; ++u) {
           for (size_t j = 0; j < candidates3d[v].size(); ++j) {
             TVector3 c1(candidates3d[v][j].first);
             float p1 = candidates3d[v][j].second;
 
             if ((c0 - c1).Mag() < min_dist) {
               TVector3 c(c0);
               if (p1 > p0) { c = c1; }
               double p = p0 * p1;
 
               if (p > fPointThreshold) {
                 double d, dmin = min_dist;
                 size_t kmin = 0;
                 for (size_t k = 0; k < result.size(); ++k) {
                   d = (result[k].first - c).Mag();
                   if (d < dmin) {
                     dmin = d;
                     kmin = k;
                   }
                 }
                 if (dmin < min_dist) {
                   if (result[kmin].second < p) // replace previously found point
                   {
                     result[kmin].first = c;
                     result[kmin].second = p;
                     found = true;
                   }
                 }
                 else // nothing close in the list, add new point
                 {
                   result.emplace_back(c, p);
                   found = true;
                 }
               } // if (p > fPointThreshold)
             }   // coincidence: points from views u and v are close
           }     // loop over points in view u
         }       // loop over views u
       }         // loop over points in view v
     }           // loop over views v
     return found;
   }

void art::detail::Producer::doBeginJob ( SharedResources const & resources )

inherited

Definition at line 22 of file Producer.cc.

References art::detail::Producer::beginJobWithFrame(), and art::detail::Producer::setupQueues().

   {
     setupQueues(resources);
     ProcessingFrame const frame{ScheduleID{}};
     beginJobWithFrame(frame);
   }

bool art::detail::Producer::doBeginRun	(	RunPrincipal &	rp,
		ModuleContext const &	mc
	)

inherited

Definition at line 65 of file Producer.cc.

References art::detail::Producer::beginRunWithFrame(), art::RangeSet::forRun(), art::RunPrincipal::makeRun(), r, art::RunPrincipal::runID(), and art::ModuleContext::scheduleID().

   {
     auto r = rp.makeRun(mc, RangeSet::forRun(rp.runID()));
     ProcessingFrame const frame{mc.scheduleID()};
     beginRunWithFrame(r, frame);
     r.commitProducts();
     return true;
   }

bool art::detail::Producer::doBeginSubRun	(	SubRunPrincipal &	srp,
		ModuleContext const &	mc
	)

inherited

Definition at line 85 of file Producer.cc.

References art::detail::Producer::beginSubRunWithFrame(), art::RangeSet::forSubRun(), art::SubRunPrincipal::makeSubRun(), art::ModuleContext::scheduleID(), and art::SubRunPrincipal::subRunID().

   {
     auto sr = srp.makeSubRun(mc, RangeSet::forSubRun(srp.subRunID()));
     ProcessingFrame const frame{mc.scheduleID()};
     beginSubRunWithFrame(sr, frame);
     sr.commitProducts();
     return true;
   }

void art::detail::Producer::doEndJob ( )

inherited

Definition at line 30 of file Producer.cc.

References art::detail::Producer::endJobWithFrame().

   {
     ProcessingFrame const frame{ScheduleID{}};
     endJobWithFrame(frame);
   }

bool art::detail::Producer::doEndRun	(	RunPrincipal &	rp,
		ModuleContext const &	mc
	)

inherited

Definition at line 75 of file Producer.cc.

References art::detail::Producer::endRunWithFrame(), art::RunPrincipal::makeRun(), r, art::ModuleContext::scheduleID(), and art::Principal::seenRanges().

   {
     auto r = rp.makeRun(mc, rp.seenRanges());
     ProcessingFrame const frame{mc.scheduleID()};
     endRunWithFrame(r, frame);
     r.commitProducts();
     return true;
   }

bool art::detail::Producer::doEndSubRun	(	SubRunPrincipal &	srp,
		ModuleContext const &	mc
	)

inherited

Definition at line 95 of file Producer.cc.

References art::detail::Producer::endSubRunWithFrame(), art::SubRunPrincipal::makeSubRun(), art::ModuleContext::scheduleID(), and art::Principal::seenRanges().

   {
     auto sr = srp.makeSubRun(mc, srp.seenRanges());
     ProcessingFrame const frame{mc.scheduleID()};
     endSubRunWithFrame(sr, frame);
     sr.commitProducts();
     return true;
   }

bool art::detail::Producer::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
	)

inherited

Definition at line 105 of file Producer.cc.

References art::detail::Producer::checkPutProducts_, e, art::EventPrincipal::makeEvent(), art::detail::Producer::produceWithFrame(), and art::ModuleContext::scheduleID().

   {
     auto e = ep.makeEvent(mc);
     ++counts_run;
     ProcessingFrame const frame{mc.scheduleID()};
     produceWithFrame(e, frame);
     e.commitProducts(checkPutProducts_, &expectedProducts<InEvent>());
     ++counts_passed;
     return true;
   }

void art::detail::Producer::doRespondToCloseInputFile ( FileBlock const & fb )

inherited

Definition at line 44 of file Producer.cc.

References art::detail::Producer::respondToCloseInputFileWithFrame().

   {
     ProcessingFrame const frame{ScheduleID{}};
     respondToCloseInputFileWithFrame(fb, frame);
   }

void art::detail::Producer::doRespondToCloseOutputFiles ( FileBlock const & fb )

inherited

Definition at line 58 of file Producer.cc.

References art::detail::Producer::respondToCloseOutputFilesWithFrame().

   {
     ProcessingFrame const frame{ScheduleID{}};
     respondToCloseOutputFilesWithFrame(fb, frame);
   }

void art::detail::Producer::doRespondToOpenInputFile ( FileBlock const & fb )

inherited

Definition at line 37 of file Producer.cc.

References art::detail::Producer::respondToOpenInputFileWithFrame().

   {
     ProcessingFrame const frame{ScheduleID{}};
     respondToOpenInputFileWithFrame(fb, frame);
   }

void art::detail::Producer::doRespondToOpenOutputFiles ( FileBlock const & fb )

inherited

Definition at line 51 of file Producer.cc.

References art::detail::Producer::respondToOpenOutputFilesWithFrame().

   {
     ProcessingFrame const frame{ScheduleID{}};
     respondToOpenOutputFilesWithFrame(fb, frame);
   }

void art::Modifier::fillProductDescriptions ( )

inherited

Definition at line 10 of file Modifier.cc.

References art::ProductRegistryHelper::fillDescriptions(), and art::ModuleBase::moduleDescription().

   {
     ProductRegistryHelper::fillDescriptions(moduleDescription());
   }

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

inherited

Definition at line 43 of file ModuleBase.cc.

References art::ModuleBase::collector_, and art::ConsumesCollector::getConsumables().

   {
     return collector_.getConsumables();
   }

std::unique_ptr< Worker > art::ModuleBase::makeWorker ( WorkerParams const & wp )

inherited

Definition at line 37 of file ModuleBase.cc.

References art::ModuleBase::doMakeWorker(), and art::NumBranchTypes.

   {
     return doMakeWorker(wp);
   }

template<typename T , BranchType BT>

ProductToken< T > art::ModuleBase::mayConsume ( InputTag const & tag )

protectedinherited

Definition at line 82 of file ModuleBase.h.

References art::ModuleBase::collector_, and art::ConsumesCollector::mayConsume().

   {
     return collector_.mayConsume<T, BT>(tag);
   }

template<typename T , BranchType BT>

void art::ModuleBase::mayConsumeMany ( )

protectedinherited

Definition at line 96 of file ModuleBase.h.

References art::ModuleBase::collector_, and art::ConsumesCollector::mayConsumeMany().

   {
     collector_.mayConsumeMany<T, BT>();
   }

template<typename Element , BranchType = InEvent>

ViewToken<Element> art::ModuleBase::mayConsumeView ( InputTag const & )

protectedinherited

template<typename T , BranchType BT>

ViewToken<T> art::ModuleBase::mayConsumeView ( InputTag const & tag )

inherited

Definition at line 89 of file ModuleBase.h.

References art::ModuleBase::collector_, and art::ConsumesCollector::mayConsumeView().

   {
     return collector_.mayConsumeView<T, BT>(tag);
   }

ModuleDescription const & art::ModuleBase::moduleDescription ( ) const

inherited

Definition at line 13 of file ModuleBase.cc.

References art::errors::LogicError.

Referenced by art::OutputModule::doRespondToOpenInputFile(), art::OutputModule::doWriteEvent(), art::Modifier::fillProductDescriptions(), art::OutputModule::makePlugins_(), art::OutputWorker::OutputWorker(), reco::shower::LArPandoraModularShowerCreation::produce(), art::Modifier::registerProducts(), and art::OutputModule::registerProducts().

   {
     if (md_.has_value()) {
       return *md_;
     }
 
     throw Exception{errors::LogicError,
                     "There was an error while calling moduleDescription().\n"}
       << "The moduleDescription() base-class member function cannot be called\n"
          "during module construction.  To determine which module is "
          "responsible\n"
          "for calling it, find the '<module type>:<module "
          "label>@Construction'\n"
          "tag in the message prefix above.  Please contact artists@fnal.gov\n"
          "for guidance.\n";
   }

ParticleDecayId& nnet::ParticleDecayId::operator= ( ParticleDecayId const & )

delete

ParticleDecayId& nnet::ParticleDecayId::operator= ( ParticleDecayId && )

delete

void nnet::ParticleDecayId::produce ( art::Event & e )

overrideprivatevirtual

Implements art::EDProducer.

Definition at line 119 of file ParticleDecayId_module.cc.

References DetectDecay(), art::EventID::event(), fTrackModuleLabel, fWireProducerLabel, art::ProductRetriever::getValidHandle(), hits(), art::Event::id(), and art::Event::put().

   {
     std::cout << std::endl << "event " << evt.id().event() << std::endl;
 
     auto vtxs = std::make_unique<std::vector<recob::Vertex>>();
     auto vtx2trk = std::make_unique<art::Assns<recob::Vertex, recob::Track>>();
 
     auto wireHandle = evt.getValidHandle<std::vector<recob::Wire>>(fWireProducerLabel);
     auto trkListHandle = evt.getValidHandle<std::vector<recob::Track>>(fTrackModuleLabel);
     auto spListHandle = evt.getValidHandle<std::vector<recob::SpacePoint>>(fTrackModuleLabel);
 
     art::FindManyP<recob::Hit> hitsFromTracks(trkListHandle, evt, fTrackModuleLabel);
     art::FindManyP<recob::SpacePoint> spFromTracks(trkListHandle, evt, fTrackModuleLabel);
     art::FindManyP<recob::Hit> hitsFromSPoints(spListHandle, evt, fTrackModuleLabel);
 
     auto const clockData = art::ServiceHandle<detinfo::DetectorClocksService const>()->DataFor(evt);
     auto const detProp =
       art::ServiceHandle<detinfo::DetectorPropertiesService const>()->DataFor(evt, clockData);
 
     std::vector<std::pair<TVector3, double>> decays;
     for (size_t i = 0; i < hitsFromTracks.size(); ++i) {
       auto hits = hitsFromTracks.at(i);
       auto spoints = spFromTracks.at(i);
       if (hits.empty()) continue;
 
       std::map<size_t, TVector3> trkSpacePoints;
       for (const auto& p : spoints) {
         auto sp_hits = hitsFromSPoints.at(p.key());
         for (const auto& h : sp_hits) {
           trkSpacePoints[h.key()] = TVector3(p->XYZ()[0], p->XYZ()[1], p->XYZ()[2]);
         }
       }
 
       DetectDecay(clockData, detProp, *wireHandle, hits, trkSpacePoints, decays);
     }
 
     double xyz[3];
     for (const auto& p3d : decays) {
 
       xyz[0] = p3d.first.X();
       xyz[1] = p3d.first.Y();
       xyz[2] = p3d.first.Z();
       std::cout << "   detected: [" << xyz[0] << ", " << xyz[1] << ", " << xyz[2]
                 << "] p:" << p3d.second << std::endl;
 
       size_t vidx = vtxs->size();
       vtxs->push_back(recob::Vertex(xyz, vidx));
 
       // to do: assn to eg. appropriate track
       // selected among set of connected tracks
       //art::Ptr<recob::Track> tptr(trkListHandle, i);
       //art::Ptr<recob::Vertex> vptr(vid, vidx, evt.productGetter(vid));
       //vtx2trk->addSingle(vptr, tptr);
     }
 
     evt.put(std::move(vtxs));
     evt.put(std::move(vtx2trk));
   }

void art::Modifier::registerProducts ( ProductDescriptions & productsToRegister )

inherited

Definition at line 16 of file Modifier.cc.

References art::ModuleBase::moduleDescription(), and art::ProductRegistryHelper::registerProducts().

   {
     ProductRegistryHelper::registerProducts(productsToRegister,
                                             moduleDescription());
   }

void art::ModuleBase::setModuleDescription ( ModuleDescription const & md )

inherited

Definition at line 31 of file ModuleBase.cc.

References art::ModuleBase::md_.

   {
     md_ = md;
   }

void art::ModuleBase::sortConsumables ( std::string const & current_process_name )

inherited

Definition at line 49 of file ModuleBase.cc.

References art::ModuleBase::collector_, and art::ConsumesCollector::sortConsumables().

   {
     // Now that we know we have seen all the consumes declarations,
     // sort the results for fast lookup later.
     collector_.sortConsumables(current_process_name);
   }