Inheritance diagram for nnet::ParticleDecayId:

Classes
struct	Config

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

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 >

Public Member Functions
	ParticleDecayId (Parameters const &p)

	ParticleDecayId (ParticleDecayId const &)=delete

	ParticleDecayId (ParticleDecayId &&)=delete

ParticleDecayId &	operator= (ParticleDecayId const &)=delete

ParticleDecayId &	operator= (ParticleDecayId &&)=delete

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 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

Additional Inherited Members
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

Definition at line 36 of file ParticleDecayId_module.cc.

Member Typedef Documentation

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

Definition at line 65 of file ParticleDecayId_module.cc.

Constructor & Destructor Documentation

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

explicit

Definition at line 94 of file ParticleDecayId_module.cc.

     : 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

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 170 of file ParticleDecayId_module.cc.

   {
     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;
   }

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 109 of file ParticleDecayId_module.cc.

   {
     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));
   }

Member Data Documentation

PointIdAlg nnet::ParticleDecayId::fPointIdAlg

private

Definition at line 83 of file ParticleDecayId_module.cc.

double nnet::ParticleDecayId::fPointThreshold

private

Definition at line 88 of file ParticleDecayId_module.cc.

double nnet::ParticleDecayId::fRoiThreshold

private

Definition at line 88 of file ParticleDecayId_module.cc.

int nnet::ParticleDecayId::fSkipView

private

Definition at line 90 of file ParticleDecayId_module.cc.

art::InputTag nnet::ParticleDecayId::fTrackModuleLabel

private

Definition at line 86 of file ParticleDecayId_module.cc.

art::InputTag nnet::ParticleDecayId::fWireProducerLabel

private

Definition at line 85 of file ParticleDecayId_module.cc.

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

larrecodnn/larrecodnn/ImagePatternAlgs/Tensorflow/Modules/ParticleDecayId_module.cc

Classes

Public Types

Public Member Functions

Private Member Functions

Private Attributes

Additional Inherited Members

Detailed Description

Member Typedef Documentation

Constructor & Destructor Documentation

Member Function Documentation

Member Data Documentation