Inheritance diagram for nnet::WaveformRoiFinder:

Public Member Functions
	WaveformRoiFinder (fhicl::ParameterSet const &p)

	WaveformRoiFinder (WaveformRoiFinder const &)=delete

	WaveformRoiFinder (WaveformRoiFinder &&)=delete

WaveformRoiFinder &	operator= (WaveformRoiFinder const &)=delete

WaveformRoiFinder &	operator= (WaveformRoiFinder &&)=delete

void	produce (art::Event &e) override

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
art::InputTag	fRawProducerLabel

art::InputTag	fWireProducerLabel

std::vector< std::unique_ptr< wavrec_tool::IWaveformRecog > >	fWaveformRecogToolVec

int	fNPlanes

unsigned int	fWaveformSize

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

Definition at line 39 of file WaveformRoiFinder_module.cc.

Constructor & Destructor Documentation

nnet::WaveformRoiFinder::WaveformRoiFinder ( fhicl::ParameterSet const & p )

explicit

Definition at line 64 of file WaveformRoiFinder_module.cc.

   : EDProducer{p}
   , fRawProducerLabel(p.get<art::InputTag>("RawProducerLabel", ""))
   , fWireProducerLabel(p.get<art::InputTag>("WireProducerLabel", ""))
 {
   // use either raw waveform or recob waveform
   if (fRawProducerLabel.empty() && fWireProducerLabel.empty()) {
     throw cet::exception("WaveformRoiFinder")
       << "Both RawProducerLabel and WireProducerLabel are empty";
   }
 
   if ((!fRawProducerLabel.empty()) && (!fWireProducerLabel.empty())) {
     throw cet::exception("WaveformRoiFinder")
       << "Only one of RawProducerLabel and WireProducerLabel should be set";
   }
 
   auto const* geo = lar::providerFrom<geo::Geometry>();
   fNPlanes = geo->Nplanes();
 
   // Signal/Noise waveform recognition tool
   fWaveformRecogToolVec.reserve(fNPlanes);
   auto const tool_psets = p.get<std::vector<fhicl::ParameterSet>>("WaveformRecogs");
   fWaveformSize = tool_psets[0].get<unsigned int>("WaveformSize");
   for (auto const& pset : tool_psets) {
     fWaveformRecogToolVec.push_back(art::make_tool<wavrec_tool::IWaveformRecog>(pset));
   }
 
   produces<std::vector<recob::Wire>>();
 }

nnet::WaveformRoiFinder::WaveformRoiFinder ( WaveformRoiFinder const & )

delete

nnet::WaveformRoiFinder::WaveformRoiFinder ( WaveformRoiFinder && )

delete

Member Function Documentation

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

delete

WaveformRoiFinder& nnet::WaveformRoiFinder::operator= ( WaveformRoiFinder && )

delete

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

overridevirtual

Implements art::EDProducer.

Definition at line 95 of file WaveformRoiFinder_module.cc.

 {
   art::Handle<std::vector<raw::RawDigit>> rawListHandle;
   std::vector<art::Ptr<raw::RawDigit>> rawlist;
   if (e.getByLabel(fRawProducerLabel, rawListHandle)) art::fill_ptr_vector(rawlist, rawListHandle);
 
   art::Handle<std::vector<recob::Wire>> wireListHandle;
   std::vector<art::Ptr<recob::Wire>> wirelist;
   if (e.getByLabel(fWireProducerLabel, wireListHandle))
     art::fill_ptr_vector(wirelist, wireListHandle);
 
   std::unique_ptr<std::vector<recob::Wire>> outwires(new std::vector<recob::Wire>);
 
   auto const* geo = lar::providerFrom<geo::Geometry>();
 
   //##############################
   //### Looping over the wires ###
   //##############################
   for (unsigned int ich = 0; ich < (rawlist.empty() ? wirelist.size() : rawlist.size()); ++ich) {
 
     std::vector<float> inputsignal(fWaveformSize);
 
     int view = -1;
 
     if (!wirelist.empty()) {
       const auto& wire = wirelist[ich];
       const auto& signal = wire->Signal();
 
       view = wire->View();
 
       for (size_t itck = 0; itck < inputsignal.size(); ++itck) {
         inputsignal[itck] = signal[itck];
       }
     }
     else if (!rawlist.empty()) {
       const auto& digitVec = rawlist[ich];
 
       view = geo->View(rawlist[ich]->Channel());
 
       std::vector<short> rawadc(fWaveformSize);
       raw::Uncompress(digitVec->ADCs(), rawadc, digitVec->GetPedestal(), digitVec->Compression());
       for (size_t itck = 0; itck < rawadc.size(); ++itck) {
         inputsignal[itck] = rawadc[itck] - digitVec->GetPedestal();
       }
     }
 
     // ... use waveform recognition CNN to perform inference on each window
     std::vector<bool> inroi(fWaveformSize, false);
     inroi = fWaveformRecogToolVec[view]->findROI(inputsignal);
 
     std::vector<float> sigs;
     int lastsignaltick = -1;
     int roistart = -1;
 
     recob::Wire::RegionsOfInterest_t rois(fWaveformSize);
 
     for (size_t i = 0; i < fWaveformSize; ++i) {
       if (inroi[i]) {
         if (sigs.empty()) {
           sigs.push_back(inputsignal[i]);
           lastsignaltick = i;
           roistart = i;
         }
         else {
           if (int(i) != lastsignaltick + 1) {
             rois.add_range(roistart, std::move(sigs));
             sigs.clear();
             sigs.push_back(inputsignal[i]);
             lastsignaltick = i;
             roistart = i;
           }
           else {
             sigs.push_back(inputsignal[i]);
             lastsignaltick = i;
           }
         }
       }
     }
     if (!sigs.empty()) { rois.add_range(roistart, std::move(sigs)); }
     if (!wirelist.empty()) {
       outwires->emplace_back(recob::Wire(rois, wirelist[ich]->Channel(), wirelist[ich]->View()));
     }
     else if (!rawlist.empty()) {
       outwires->emplace_back(
         recob::Wire(rois, rawlist[ich]->Channel(), geo->View(rawlist[ich]->Channel())));
     }
   }
 
   e.put(std::move(outwires));
 }