Inheritance diagram for nnet::EmTrackClusterId:

Classes
struct	Config

Public Types
typedef std::unordered_map< unsigned int, std::vector< size_t > >	view_keymap

typedef std::unordered_map< unsigned int, view_keymap >	tpc_view_keymap

typedef std::unordered_map< unsigned int, tpc_view_keymap >	cryo_tpc_view_keymap

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
	EmTrackClusterId (Parameters const &p)

	EmTrackClusterId (EmTrackClusterId const &)=delete

	EmTrackClusterId (EmTrackClusterId &&)=delete

EmTrackClusterId &	operator= (EmTrackClusterId const &)=delete

EmTrackClusterId &	operator= (EmTrackClusterId &&)=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	isViewSelected (int view) const

Private Attributes
size_t	fBatchSize

PointIdAlg	fPointIdAlg

anab::MVAWriter< 3 >	fMVAWriter

art::InputTag	fWireProducerLabel

art::InputTag	fHitModuleLabel

art::InputTag	fClusterModuleLabel

art::InputTag	fTrackModuleLabel

bool	fDoClusters

bool	fDoTracks

std::vector< int >	fViews

art::InputTag	fNewClustersTag

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 43 of file EmTrackClusterId3out_module.cc.

Member Typedef Documentation

typedef std::unordered_map<unsigned int, tpc_view_keymap> nnet::EmTrackClusterId::cryo_tpc_view_keymap

Definition at line 48 of file EmTrackClusterId3out_module.cc.

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

Definition at line 79 of file EmTrackClusterId3out_module.cc.

typedef std::unordered_map<unsigned int, view_keymap> nnet::EmTrackClusterId::tpc_view_keymap

Definition at line 47 of file EmTrackClusterId3out_module.cc.

typedef std::unordered_map<unsigned int, std::vector<size_t> > nnet::EmTrackClusterId::view_keymap

Definition at line 46 of file EmTrackClusterId3out_module.cc.

Constructor & Destructor Documentation

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

explicit

Definition at line 108 of file EmTrackClusterId3out_module.cc.

     : EDProducer{config}
     , fBatchSize(config().BatchSize())
     , fPointIdAlg(config().PointIdAlg())
     , fMVAWriter(producesCollector(), "emtrack")
     , fWireProducerLabel(config().WireLabel())
     , fHitModuleLabel(config().HitModuleLabel())
     , fClusterModuleLabel(config().ClusterModuleLabel())
     , fTrackModuleLabel(config().TrackModuleLabel())
     , fViews(config().Views())
     ,
 
     fNewClustersTag(config.get_PSet().get<std::string>("module_label"),
                     "",
                     art::ServiceHandle<art::TriggerNamesService const>()->getProcessName())
   {
     fMVAWriter.produces_using<recob::Hit>();
 
     if (!fClusterModuleLabel.label().empty()) {
       produces<std::vector<recob::Cluster>>();
       produces<art::Assns<recob::Cluster, recob::Hit>>();
 
       fMVAWriter.produces_using<recob::Cluster>();
       fDoClusters = true;
     }
     else {
       fDoClusters = false;
     }
 
     if (!fTrackModuleLabel.label().empty()) {
       fMVAWriter.produces_using<recob::Track>();
       fDoTracks = true;
     }
     else {
       fDoTracks = false;
     }
   }

nnet::EmTrackClusterId::EmTrackClusterId ( EmTrackClusterId const & )

delete

nnet::EmTrackClusterId::EmTrackClusterId ( EmTrackClusterId && )

delete

Member Function Documentation

bool nnet::EmTrackClusterId::isViewSelected ( int view ) const

private

Definition at line 406 of file EmTrackClusterId3out_module.cc.

   {
     if (fViews.empty())
       return true;
     else {
       for (auto k : fViews)
         if (k == view) { return true; }
       return false;
     }
   }

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

delete

EmTrackClusterId& nnet::EmTrackClusterId::operator= ( EmTrackClusterId && )

delete

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

overrideprivatevirtual

Implements art::EDProducer.

Definition at line 148 of file EmTrackClusterId3out_module.cc.

   {
     mf::LogVerbatim("EmTrackClusterId") << "next event: " << evt.run() << " / " << evt.id().event();
 
     auto wireHandle = evt.getValidHandle<std::vector<recob::Wire>>(fWireProducerLabel);
 
     unsigned int cryo, tpc, view;
 
     // ******************* get and sort hits ********************
     auto hitListHandle = evt.getValidHandle<std::vector<recob::Hit>>(fHitModuleLabel);
     std::vector<art::Ptr<recob::Hit>> hitPtrList;
     art::fill_ptr_vector(hitPtrList, hitListHandle);
 
     EmTrackClusterId::cryo_tpc_view_keymap hitMap;
     for (auto const& h : hitPtrList) {
       view = h->WireID().Plane;
       if (!isViewSelected(view)) continue;
 
       cryo = h->WireID().Cryostat;
       tpc = h->WireID().TPC;
 
       hitMap[cryo][tpc][view].push_back(h.key());
     }
 
     // ********************* classify hits **********************
     auto hitID = fMVAWriter.initOutputs<recob::Hit>(
       fHitModuleLabel, hitPtrList.size(), fPointIdAlg.outputLabels());
 
     auto const clockData = art::ServiceHandle<detinfo::DetectorClocksService const>()->DataFor(evt);
     auto const detProp =
       art::ServiceHandle<detinfo::DetectorPropertiesService const>()->DataFor(evt, clockData);
 
     std::vector<char> hitInFA(
       hitPtrList.size(),
       0); // tag hits in fid. area as 1, use 0 for hits close to the projectrion edges
     for (auto const& [cryo, tpcs] : hitMap) {
       for (auto const& [tpc, views] : tpcs) {
         for (auto const& pview : views) {
           view = pview.first;
           if (!isViewSelected(view)) continue; // should not happen, hits were selected
 
           fPointIdAlg.setWireDriftData(clockData, detProp, *wireHandle, view, tpc, cryo);
 
           // (1) do all hits in this plane ------------------------------------------------
           for (size_t idx = 0; idx < pview.second.size(); idx += fBatchSize) {
             std::vector<std::pair<unsigned int, float>> points;
             std::vector<size_t> keys;
             for (size_t k = 0; k < fBatchSize; ++k) {
               if (idx + k >= pview.second.size()) { break; } // careful about the tail
 
               size_t h = pview.second[idx + k]; // h is the Ptr< recob::Hit >::key()
               const recob::Hit& hit = *(hitPtrList[h]);
               points.emplace_back(hit.WireID().Wire, hit.PeakTime());
               keys.push_back(h);
             }
 
             auto batch_out = fPointIdAlg.predictIdVectors(points);
             if (points.size() != batch_out.size()) {
               throw cet::exception("EmTrackClusterId") << "hits processing failed" << std::endl;
             }
 
             for (size_t k = 0; k < points.size(); ++k) {
               size_t h = keys[k];
               fMVAWriter.setOutput(hitID, h, batch_out[k]);
               if (fPointIdAlg.isInsideFiducialRegion(points[k].first, points[k].second)) {
                 hitInFA[h] = 1;
               }
             }
           } // hits done ------------------------------------------------------------------
         }
       }
     }
 
     // (2) do clusters when hits are ready in all planes ----------------------------------------
     if (fDoClusters) {
       // **************** prepare for new clusters ****************
       auto clusters = std::make_unique<std::vector<recob::Cluster>>();
       auto clu2hit = std::make_unique<art::Assns<recob::Cluster, recob::Hit>>();
 
       // ************** get and sort input clusters ***************
       auto cluListHandle = evt.getValidHandle<std::vector<recob::Cluster>>(fClusterModuleLabel);
       std::vector<art::Ptr<recob::Cluster>> cluPtrList;
       art::fill_ptr_vector(cluPtrList, cluListHandle);
 
       EmTrackClusterId::cryo_tpc_view_keymap cluMap;
       for (auto const& c : cluPtrList) {
         view = c->Plane().Plane;
         if (!isViewSelected(view)) continue;
 
         cryo = c->Plane().Cryostat;
         tpc = c->Plane().TPC;
 
         cluMap[cryo][tpc][view].push_back(c.key());
       }
 
       auto cluID =
         fMVAWriter.initOutputs<recob::Cluster>(fNewClustersTag, fPointIdAlg.outputLabels());
 
       unsigned int cidx = 0; // new clusters index
       art::FindManyP<recob::Hit> hitsFromClusters(cluListHandle, evt, fClusterModuleLabel);
       std::vector<bool> hitUsed(hitPtrList.size(), false); // tag hits used in clusters
       for (auto const& pcryo : cluMap) {
         cryo = pcryo.first;
         for (auto const& ptpc : pcryo.second) {
           tpc = ptpc.first;
           for (auto const& pview : ptpc.second) {
             view = pview.first;
             if (!isViewSelected(view)) continue; // should not happen, clusters were pre-selected
 
             for (size_t c : pview.second) // c is the Ptr< recob::Cluster >::key()
             {
               auto v = hitsFromClusters.at(c);
               if (v.empty()) continue;
 
               for (auto const& hit : v) {
                 if (hitUsed[hit.key()]) {
                   mf::LogWarning("EmTrackClusterId") << "hit already used in another cluster";
                 }
                 hitUsed[hit.key()] = true;
               }
 
               auto vout = fMVAWriter.getOutput<recob::Hit>(
                 v, [&](art::Ptr<recob::Hit> const& ptr) { return (float)hitInFA[ptr.key()]; });
 
               float pvalue = vout[0] / (vout[0] + vout[1]);
               mf::LogVerbatim("EmTrackClusterId") << "cluster in tpc:" << tpc << " view:" << view
                                                   << " size:" << v.size() << " p:" << pvalue;
 
               clusters->emplace_back(recob::Cluster(0.0F,
                                                     0.0F,
                                                     0.0F,
                                                     0.0F,
                                                     0.0F,
                                                     0.0F,
                                                     0.0F,
                                                     0.0F,
                                                     0.0F,
                                                     0.0F,
                                                     0.0F,
                                                     0.0F,
                                                     0.0F,
                                                     0.0F,
                                                     0.0F,
                                                     0.0F,
                                                     0.0F,
                                                     0.0F,
                                                     v.size(),
                                                     0.0F,
                                                     0.0F,
                                                     cidx,
                                                     (geo::View_t)view,
                                                     v.front()->WireID().planeID()));
               util::CreateAssn(*this, evt, *clusters, v, *clu2hit);
               cidx++;
 
               fMVAWriter.addOutput(cluID, vout); // add copy of the input cluster
             }
 
             // (2b) make single-hit clusters --------------------------------------------
             for (size_t h : hitMap[cryo][tpc][view]) // h is the Ptr< recob::Hit >::key()
             {
               if (hitUsed[h]) continue;
 
               auto vout = fMVAWriter.getOutput<recob::Hit>(h);
               float pvalue = vout[0] / (vout[0] + vout[1]);
 
               mf::LogVerbatim("EmTrackClusterId")
                 << "single hit in tpc:" << tpc << " view:" << view
                 << " wire:" << hitPtrList[h]->WireID().Wire
                 << " drift:" << hitPtrList[h]->PeakTime() << " p:" << pvalue;
 
               art::PtrVector<recob::Hit> cluster_hits;
               cluster_hits.push_back(hitPtrList[h]);
               clusters->emplace_back(recob::Cluster(0.0F,
                                                     0.0F,
                                                     0.0F,
                                                     0.0F,
                                                     0.0F,
                                                     0.0F,
                                                     0.0F,
                                                     0.0F,
                                                     0.0F,
                                                     0.0F,
                                                     0.0F,
                                                     0.0F,
                                                     0.0F,
                                                     0.0F,
                                                     0.0F,
                                                     0.0F,
                                                     0.0F,
                                                     0.0F,
                                                     1,
                                                     0.0F,
                                                     0.0F,
                                                     cidx,
                                                     (geo::View_t)view,
                                                     hitPtrList[h]->WireID().planeID()));
               util::CreateAssn(*this, evt, *clusters, cluster_hits, *clu2hit);
               cidx++;
 
               fMVAWriter.addOutput(cluID, vout); // add single-hit cluster tagging unclutered hit
             }
             mf::LogVerbatim("EmTrackClusterId")
               << "...produced " << cidx - pview.second.size() << " single-hit clusters.";
           }
         }
       }
 
       evt.put(std::move(clusters));
       evt.put(std::move(clu2hit));
     } // all clusters done ----------------------------------------------------------------------
 
     // (3) do tracks when all hits in all cryo/tpc/plane are done -------------------------------
     if (fDoTracks) {
       auto trkListHandle = evt.getValidHandle<std::vector<recob::Track>>(fTrackModuleLabel);
       art::FindManyP<recob::Hit> hitsFromTracks(trkListHandle, evt, fTrackModuleLabel);
       std::vector<std::vector<art::Ptr<recob::Hit>>> trkHitPtrList(trkListHandle->size());
       for (size_t t = 0; t < trkListHandle->size(); ++t) {
         auto v = hitsFromTracks.at(t);
         size_t nh[3] = {0, 0, 0};
         for (auto const& hptr : v) {
           ++nh[hptr->View()];
         }
         size_t best_view = 2;                                       // collection
         if ((nh[0] >= nh[1]) && (nh[0] > 2 * nh[2])) best_view = 0; // ind1
         if ((nh[1] >= nh[0]) && (nh[1] > 2 * nh[2])) best_view = 1; // ind2
 
         size_t k = 0;
         while (!isViewSelected(best_view)) {
           best_view = (best_view + 1) % 3;
           if (++k > 3) {
             throw cet::exception("EmTrackClusterId") << "No views selected at all?" << std::endl;
           }
         }
 
         for (auto const& hptr : v) {
           if (hptr->View() == best_view) trkHitPtrList[t].emplace_back(hptr);
         }
       }
 
       auto trkID = fMVAWriter.initOutputs<recob::Track>(
         fTrackModuleLabel, trkHitPtrList.size(), fPointIdAlg.outputLabels());
       for (size_t t = 0; t < trkHitPtrList.size(); ++t) // t is the Ptr< recob::Track >::key()
       {
         auto vout =
           fMVAWriter.getOutput<recob::Hit>(trkHitPtrList[t], [&](art::Ptr<recob::Hit> const& ptr) {
             return (float)hitInFA[ptr.key()];
           });
         fMVAWriter.setOutput(trkID, t, vout);
       }
     }
     // tracks done ------------------------------------------------------------------------------
 
     fMVAWriter.saveOutputs(evt);
   }