Inheritance diagram for Infill::MakeInfillTrainingData:

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

	MakeInfillTrainingData (MakeInfillTrainingData const &)=delete

	MakeInfillTrainingData (MakeInfillTrainingData &&)=delete

MakeInfillTrainingData &	operator= (MakeInfillTrainingData const &)=delete

MakeInfillTrainingData &	operator= (MakeInfillTrainingData &&)=delete

void	analyze (art::Event const &e) override

void	beginJob () override

void	endJob () override

Public Member Functions inherited from art::EDAnalyzer
	EDAnalyzer (fhicl::ParameterSet const &pset)

template<typename Config >
	EDAnalyzer (Table< Config > const &config)

std::string	workerType () const

Public Member Functions inherited from art::detail::Analyzer
virtual	~Analyzer () noexcept

	Analyzer (fhicl::ParameterSet const &pset)

template<typename Config >
	Analyzer (Table< Config > const &config)

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::Observer
	~Observer () noexcept

	Observer (Observer const &)=delete

	Observer (Observer &&)=delete

Observer &	operator= (Observer const &)=delete

Observer &	operator= (Observer &&)=delete

void	registerProducts (ProductDescriptions &, ModuleDescription const &)

void	fillDescriptions (ModuleDescription const &)

fhicl::ParameterSetID	selectorConfig () const

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
const geo::GeometryCore *	fGeom

art::ServiceHandle< art::TFileService >	tfs

std::set< raw::ChannelID_t >	fBadChannels

std::set< raw::ChannelID_t >	fNoisyChannels

std::set< raw::ChannelID_t >	fDeadChannels

std::set< readout::ROPID >	fActiveRops

std::string	fInputLabel

Additional Inherited Members
Public Types inherited from art::EDAnalyzer
using	WorkerType = WorkerT< EDAnalyzer >

using	ModuleType = EDAnalyzer

Protected Member Functions inherited from art::Observer
std::string const &	processName () const

bool	wantAllEvents () const noexcept

bool	wantEvent (ScheduleID id, Event const &e) const

Handle< TriggerResults >	getTriggerResults (Event const &e) const

	Observer (fhicl::ParameterSet const &config)

	Observer (std::vector< std::string > const &select_paths, std::vector< std::string > const &reject_paths, fhicl::ParameterSet const &config)

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 46 of file MakeInfillTrainingData_module.cc.

Constructor & Destructor Documentation

Infill::MakeInfillTrainingData::MakeInfillTrainingData ( fhicl::ParameterSet const & p )

explicit

Definition at line 80 of file MakeInfillTrainingData_module.cc.

   : EDAnalyzer{p},
   fInputLabel           (p.get<std::string> ("inputLabel"))
 {
   consumes<std::vector<raw::RawDigit>>(fInputLabel);
 }

Infill::MakeInfillTrainingData::MakeInfillTrainingData ( MakeInfillTrainingData const & )

delete

Infill::MakeInfillTrainingData::MakeInfillTrainingData ( MakeInfillTrainingData && )

delete

Member Function Documentation

void Infill::MakeInfillTrainingData::analyze ( art::Event const & e )

overridevirtual

Implements art::EDAnalyzer.

Definition at line 87 of file MakeInfillTrainingData_module.cc.

 {
   std::string sEvent = (
     "Ev" + std::to_string(e.id().event()) + "Run" + std::to_string(e.id().run()) + 
     "SRun" + std::to_string(e.id().subRun())
   );
   std::map<readout::ROPID, TH2F*> ropImages;
 
   auto const detProp = art::ServiceHandle<detinfo::DetectorPropertiesService>()->DataFor(e);
   // Networks expect a fixed image size
   if (detProp.NumberTimeSamples() != 6000) {
     std::cerr << "MakeInfillTrainingData_module.cc: Training data should have 6000 time ticks\n";
     std::abort();
   } 
 
   // Prepare TH2s
   for (const readout::ROPID& rop : fActiveRops) {
     std::string sTitle = (
       sEvent + "_TPCset" + std::to_string(rop.TPCset) + "_ROP" + std::to_string(rop.ROP)
     );
     ropImages[rop] = tfs->make<TH2F>(
       sTitle.c_str(), sTitle.c_str(), fGeom->Nchannels(rop), 0, fGeom->Nchannels(rop), 6000, 0, 6000
     );
   }
 
   auto digs = e.getHandle<std::vector<raw::RawDigit>>(fInputLabel);
 
   // Fill TH2s
   for(const raw::RawDigit& dig : *digs){ 
     readout::ROPID rop = fGeom->ChannelToROP(dig.Channel());
     if(ropImages.count(rop)){ // Ignores ROPs associated with only inactive TPCIDs
       raw::RawDigit::ADCvector_t adcs(dig.Samples());
       // raw::Uncompress(dig.ADCs(), adcs, dig.GetPedestal(), dig.Compression());
       raw::Uncompress(dig.ADCs(), adcs, dig.Compression());
 
       const raw::ChannelID_t firstCh = fGeom->FirstChannelInROP(rop);
       for(unsigned int tick = 0; tick < adcs.size(); ++tick){
         const int adc = adcs[tick] ? int(adcs[tick]) - dig.GetPedestal() : 0;
 
         ropImages[rop]->Fill(dig.Channel() - firstCh, tick, adc);
       }
     }
   }
 
   tfs->file().Write();
   tfs->file().Flush();
   for(auto it: ropImages) delete it.second;
 }

void Infill::MakeInfillTrainingData::beginJob ( )

overridevirtual

Reimplemented from art::EDAnalyzer.

Definition at line 136 of file MakeInfillTrainingData_module.cc.

 {
   fGeom = art::ServiceHandle<geo::Geometry>()->provider();
 
   // Get active ROPs (not facing a wall)
   geo::ROP_id_iterator iRop, rBegin(fGeom, geo::ROP_id_iterator::begin_pos), 
     rEnd(fGeom, geo::ROP_id_iterator::end_pos);
   for (iRop = rBegin; iRop != rEnd; ++iRop) { // Iterate over ROPs in detector
     for (const geo::TPCID tpcId : fGeom->ROPtoTPCs(*iRop)) {
       const geo::TPCGeo tpc = fGeom->TPC(tpcId);
       const TGeoVolume* tpcVol = tpc.ActiveVolume();
       
       if (tpcVol->Capacity() > 1000000) { // At least one of the ROP's TPCIDs needs to be active
         // Networks expect a fixed image size
         if(fGeom->SignalType(*iRop) == geo::kInduction && fGeom->Nchannels(*iRop) != 800) {
           std::cerr << "MakeInfillTrainingData_module.cc: Induction view training data should have 800 channels\n";
           std::abort();
         }
         if(fGeom->SignalType(*iRop) == geo::kCollection && fGeom->Nchannels(*iRop) != 480) {
           std::cerr << "MakeInfillTrainingData_module.cc: Collection view training data should have 480 channels\n";
           std::abort();
         }
 
         fActiveRops.insert(*iRop);
         break;
       }
     }
   }
 }

void Infill::MakeInfillTrainingData::endJob ( )

overridevirtual

Reimplemented from art::EDAnalyzer.

Definition at line 166 of file MakeInfillTrainingData_module.cc.

 {
   std::cout << "Getting dead channels..." << std::endl;
 
   fBadChannels = art::ServiceHandle<lariov::ChannelStatusService const>()->GetProvider().BadChannels();
   fNoisyChannels = art::ServiceHandle<lariov::ChannelStatusService const>()->GetProvider().NoisyChannels();
 
   std::merge(
     fBadChannels.begin(), fBadChannels.end(), fNoisyChannels.begin(), fNoisyChannels.end(), 
     std::inserter(fDeadChannels, fDeadChannels.begin())
   );
 
   // Print dead channels to terminal
   for(const readout::ROPID& rop : fActiveRops) {
     std::cout << "(TPCset " << rop.TPCset << ", ROP " << rop.ROP << "): ";
     const raw::ChannelID_t firstCh = fGeom->FirstChannelInROP(rop);
     for (const raw::ChannelID_t ch : fDeadChannels) {
       if (fGeom->ChannelToROP(ch) == rop) {
         std::cout << ch - firstCh << " ";
       }
     } // Could break early if ch > last ch in currentRop to save a bit of looping?
     std::cout << std::endl;
   }
 }