Inheritance diagram for lcvn::LArNuCVNZlibMaker:

Public Types
using	ModuleType = EDAnalyzer

Public Member Functions
	LArNuCVNZlibMaker (fhicl::ParameterSet const &pset)

	~LArNuCVNZlibMaker ()

void	beginJob ()

void	endSubRun (art::SubRun const &sr)

void	analyze (const art::Event &evt)

void	reconfigure (const fhicl::ParameterSet &pset)

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)

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
template<class T >
void	write_files (LArTrainingData< T > td, std::string evtid)=delete

std::string const &	processName () const

bool	wantAllEvents () const noexcept

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

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

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

Protected Attributes
std::string	fOutputDir

std::string	fPixelMapInput

bool	fSetLog

std::vector< bool >	fReverseViews

unsigned int	fPlaneLimit

unsigned int	fTDCLimit

std::string	out_dir

CVNImageUtils	fImage

Private Member Functions
void	write_files (LArTrainingNuData td, std::string evtid)

Private Attributes
unsigned int	fTopologyHitsCut

std::string	fGenieGenModuleLabel

bool	fApplyFidVol

std::vector< double >	fFidMinCoords

std::vector< double >	fFidMaxCoords

TH1D *	hPOT

double	fPOT

int	fRun

int	fSubRun

Detailed Description

Definition at line 55 of file LArNuCVNZlibMaker_module.cc.

Member Typedef Documentation

using art::EDAnalyzer::ModuleType = EDAnalyzer

inherited

Definition at line 22 of file EDAnalyzer.h.

Constructor & Destructor Documentation

lcvn::LArNuCVNZlibMaker::LArNuCVNZlibMaker ( fhicl::ParameterSet const & pset )

explicit

Definition at line 81 of file LArNuCVNZlibMaker_module.cc.

References reconfigure().

                                                                     : ICVNZlibMaker(pset)
   {
     this->reconfigure(pset);
   }

lcvn::LArNuCVNZlibMaker::~LArNuCVNZlibMaker ( )

Definition at line 87 of file LArNuCVNZlibMaker_module.cc.

87 {}

Member Function Documentation

void lcvn::LArNuCVNZlibMaker::analyze ( const art::Event & evt )

Definition at line 127 of file LArNuCVNZlibMaker_module.cc.

   {
 
     // Get the pixel maps
     std::vector<art::Ptr<lcvn::PixelMap>> pixelmaps;
     art::InputTag itag1(fPixelMapInput, fPixelMapInput);
     auto h_pixelmaps = evt.getHandle<std::vector<lcvn::PixelMap>>(itag1);
     if (h_pixelmaps) art::fill_ptr_vector(pixelmaps, h_pixelmaps);
 
     // If no pixel maps, quit
     if (pixelmaps.size() == 0) return;
 
     InteractionType interaction = kOther;
 
     // MC information
     std::vector<art::Ptr<simb::MCTruth>> mctruth_list;
     auto h_mctruth = evt.getHandle<std::vector<simb::MCTruth>>(fGenieGenModuleLabel);
     if (h_mctruth) art::fill_ptr_vector(mctruth_list, h_mctruth);
 
     art::Ptr<simb::MCTruth> mctruth = mctruth_list[0];
     simb::MCNeutrino true_neutrino = mctruth->GetNeutrino();
 
     // Hard-coding event weight for now
     // Should probably fix this at some point
     double event_weight = 1.;
 
     AssignLabels labels;
 
     interaction = labels.GetInteractionType(true_neutrino);
     labels.GetTopology(mctruth, fTopologyHitsCut);
 
     // True lepton and neutrino energies
     float nu_energy = true_neutrino.Nu().E();
     float lep_energy = true_neutrino.Lepton().E();
 
     // Put a containment cut here
     // If outside the fiducial volume don't waste any time filling other variables
     if (fApplyFidVol) {
       // Get the interaction vertex from the end point of the neutrino. This is
       // because the start point of the lepton doesn't make sense for taus as they
       // are decayed by the generator and not GEANT
       TVector3 vtx = true_neutrino.Nu().EndPosition().Vect();
       bool isFid = (vtx.X() > fFidMinCoords[0] && vtx.X() < fFidMaxCoords[0]) &&
                    (vtx.Y() > fFidMinCoords[1] && vtx.Y() < fFidMaxCoords[1]) &&
                    (vtx.Z() > fFidMinCoords[2] && vtx.Z() < fFidMaxCoords[2]);
       if (!isFid) return;
     }
 
     TDNuInfo info;
     info.SetTruthInfo(nu_energy, lep_energy, 0., event_weight);
     info.SetTopologyInformation(labels.GetPDG(),
                                 labels.GetNProtons(),
                                 labels.GetNPions(),
                                 labels.GetNPizeros(),
                                 labels.GetNNeutrons(),
                                 labels.GetTopologyType(),
                                 labels.GetTopologyTypeAlt());
 
     LArTrainingNuData train(interaction, *pixelmaps[0], info);
 
     std::string evtid = "r" + std::to_string(evt.run()) + "_s" + std::to_string(evt.subRun()) +
                         "_e" + std::to_string(evt.event()) + "_h" + std::to_string(time(0));
     this->write_files(train, evtid);
   }

void lcvn::LArNuCVNZlibMaker::beginJob ( )

virtual

Reimplemented from lcvn::ICVNZlibMaker.

Definition at line 118 of file LArNuCVNZlibMaker_module.cc.

References lcvn::ICVNZlibMaker::beginJob(), and hPOT.

   {
     ICVNZlibMaker::beginJob();
 
     art::ServiceHandle<art::TFileService> tfs;
     hPOT = tfs->make<TH1D>("TotalPOT", "Total POT;; POT", 1, 0, 1);
   }

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

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

inherited

Definition at line 25 of file Analyzer.cc.

Referenced by art::detail::Analyzer::Analyzer().

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

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

inherited

Definition at line 68 of file Analyzer.cc.

References art::ModuleContext::scheduleID().

Referenced by art::detail::Analyzer::Analyzer().

   {
     ProcessingFrame const frame{mc.scheduleID()};
     beginRunWithFrame(std::as_const(rp).makeRun(mc), frame);
     return true;
   }

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

inherited

Definition at line 84 of file Analyzer.cc.

References art::ModuleContext::scheduleID().

Referenced by art::detail::Analyzer::Analyzer().

   {
     ProcessingFrame const frame{mc.scheduleID()};
     beginSubRunWithFrame(std::as_const(srp).makeSubRun(mc), frame);
     return true;
   }

void art::detail::Analyzer::doEndJob ( )

inherited

Definition at line 33 of file Analyzer.cc.

Referenced by art::detail::Analyzer::Analyzer().

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

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

inherited

Definition at line 76 of file Analyzer.cc.

References art::ModuleContext::scheduleID().

Referenced by art::detail::Analyzer::Analyzer().

   {
     ProcessingFrame const frame{mc.scheduleID()};
     endRunWithFrame(std::as_const(rp).makeRun(mc), frame);
     return true;
   }

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

inherited

Definition at line 92 of file Analyzer.cc.

References art::ModuleContext::scheduleID().

Referenced by art::detail::Analyzer::Analyzer().

   {
     ProcessingFrame const frame{mc.scheduleID()};
     endSubRunWithFrame(std::as_const(srp).makeSubRun(mc), frame);
     return true;
   }

bool art::detail::Analyzer::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 100 of file Analyzer.cc.

References e, and art::ModuleContext::scheduleID().

Referenced by art::detail::Analyzer::Analyzer().

   {
     auto const e = std::as_const(ep).makeEvent(mc);
     if (wantEvent(mc.scheduleID(), e)) {
       ++counts_run;
       ProcessingFrame const frame{mc.scheduleID()};
       analyzeWithFrame(e, frame);
       ++counts_passed;
     }
     return true;
   }

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

inherited

Definition at line 47 of file Analyzer.cc.

Referenced by art::detail::Analyzer::Analyzer().

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

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

inherited

Definition at line 61 of file Analyzer.cc.

Referenced by art::detail::Analyzer::Analyzer().

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

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

inherited

Definition at line 40 of file Analyzer.cc.

Referenced by art::detail::Analyzer::Analyzer().

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

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

inherited

Definition at line 54 of file Analyzer.cc.

Referenced by art::detail::Analyzer::Analyzer().

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

void lcvn::LArNuCVNZlibMaker::endSubRun ( art::SubRun const & sr )

virtual

Reimplemented from art::EDAnalyzer.

Definition at line 102 of file LArNuCVNZlibMaker_module.cc.

References fPOT, fRun, fSubRun, art::ProductRetriever::getByLabel(), hPOT, if(), art::SubRun::run(), art::SubRun::subRun(), and sumdata::POTSummary::totpot.

   {
 
     std::string fPOTModuleLabel = "generator";
     fRun = sr.run();
     fSubRun = sr.subRun();
 
     art::Handle<sumdata::POTSummary> potListHandle;
     if (sr.getByLabel(fPOTModuleLabel, potListHandle))
       fPOT = potListHandle->totpot;
     else
       fPOT = 0.;
     if (hPOT) hPOT->Fill(0.5, fPOT);
   }

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

Handle< TriggerResults > art::Observer::getTriggerResults ( Event const & e ) const

protectedinherited

Definition at line 75 of file Observer.cc.

References art::ProductRetriever::get(), and art::Observer::selectors_.

Referenced by art::OutputModule::doWriteEvent(), and art::Observer::wantAllEvents().

   {
     if (selectors_) {
       return selectors_->getOneTriggerResults(e);
     }
 
     // The following applies for cases where no SelectEvents entries
     // exist.
     Handle<TriggerResults> h;
     if (e.get(empty_process_name, h)) {
       return h;
     }
     return Handle<TriggerResults>{};
   }

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

string const & art::Observer::processName ( ) const

protectedinherited

Definition at line 57 of file Observer.cc.

References art::Observer::process_name_.

Referenced by art::FileDumperOutput::printPrincipal().

   {
     return process_name_;
   }

void lcvn::LArNuCVNZlibMaker::reconfigure ( const fhicl::ParameterSet & pset )

Definition at line 90 of file LArNuCVNZlibMaker_module.cc.

References fApplyFidVol, fFidMaxCoords, fFidMinCoords, fGenieGenModuleLabel, fTopologyHitsCut, fhicl::ParameterSet::get(), and lcvn::ICVNZlibMaker::reconfigure().

Referenced by LArNuCVNZlibMaker().

   {
     ICVNZlibMaker::reconfigure(pset);
 
     fTopologyHitsCut = pset.get<unsigned int>("TopologyHitsCut");
     fGenieGenModuleLabel = pset.get<std::string>("GenieGenModuleLabel");
     fApplyFidVol = pset.get<bool>("ApplyFidVol");
     fFidMinCoords = pset.get<std::vector<double>>("FidMinCoords");
     fFidMaxCoords = pset.get<std::vector<double>>("FidMaxCoords");
   }

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

bool art::Observer::wantAllEvents ( ) const

inlineprotectednoexceptinherited

Definition at line 31 of file Observer.h.

References e, art::Observer::getTriggerResults(), art::Observer::wantAllEvents_, and art::Observer::wantEvent().

     {
       return wantAllEvents_;
     }

bool art::Observer::wantEvent	(	ScheduleID	id,
		Event const &	e
	)		const

protectedinherited

Definition at line 63 of file Observer.cc.

References art::Observer::rejectors_, art::Observer::selectors_, and art::Observer::wantAllEvents_.

Referenced by art::OutputModule::doEvent(), art::OutputModule::doWriteEvent(), and art::Observer::wantAllEvents().

   {
     if (wantAllEvents_) {
       return true;
     }
     bool const select_event = selectors_ ? selectors_->matchEvent(id, e) : true;
     bool const reject_event =
       rejectors_ ? rejectors_->matchEvent(id, e) : false;
     return select_event and not reject_event;
   }

void lcvn::LArNuCVNZlibMaker::write_files	(	LArTrainingNuData	td,
		std::string	evtid
	)

private

Definition at line 193 of file LArNuCVNZlibMaker_module.cc.

References lcvn::CVNImageUtils::ConvertPixelMapToPixelArray(), DEFINE_ART_MODULE, art::errors::FileOpenError, lcvn::ICVNZlibMaker::fImage, lcvn::LArTrainingData< T >::fInfo, lcvn::LArTrainingData< T >::fInt, lcvn::ICVNZlibMaker::fPlaneLimit, lcvn::LArTrainingData< T >::fPMap, lcvn::ICVNZlibMaker::fTDCLimit, lcvn::PixelMap::GetTotHits(), lcvn::PixelMap::NTdc(), lcvn::PixelMap::NWire(), lcvn::ICVNZlibMaker::out_dir, and lcvn::CVNImageUtils::SetPixelMapSize().

Referenced by analyze().

   {
     // cropped from 2880 x 500 to 500 x 500 here
     std::vector<unsigned char> pixel_array(3 * fPlaneLimit * fTDCLimit);
 
     fImage.SetPixelMapSize(td.fPMap.NWire(), td.fPMap.NTdc());
     fImage.ConvertPixelMapToPixelArray(td.fPMap, pixel_array);
 
     ulong src_len = 3 * fPlaneLimit * fTDCLimit; // pixelArray length
     ulong dest_len = compressBound(src_len);     // calculate size of the compressed data
     char* ostream = (char*)malloc(dest_len);     // allocate memory for the compressed data
 
     int res = compress((Bytef*)ostream, &dest_len, (Bytef*)&pixel_array[0], src_len);
 
     // Buffer error
 
     if (res == Z_BUF_ERROR) std::cout << "Buffer too small!" << std::endl;
 
     // Memory error
     else if (res == Z_MEM_ERROR)
       std::cout << "Not enough memory for compression!" << std::endl;
 
     // Compression ok
     else {
 
       // Create output files
       std::string image_file_name = out_dir + "/event_" + evtid + ".gz";
       std::string info_file_name = out_dir + "/event_" + evtid + ".info";
 
       std::ofstream image_file(image_file_name, std::ofstream::binary);
       std::ofstream info_file(info_file_name);
 
       if (image_file.is_open() && info_file.is_open()) {
 
         // Write compressed data to file
 
         image_file.write(ostream, dest_len);
 
         image_file.close(); // close file
 
         // Write records to file
 
         // Category
 
         info_file << td.fInt << std::endl;
         info_file << td.fInfo << std::endl;
         info_file << td.fPMap.GetTotHits() << std::endl;
 
         info_file.close(); // close file
       }
       else {
 
         if (image_file.is_open())
           image_file.close();
         else
           throw art::Exception(art::errors::FileOpenError)
             << "Unable to open file " << image_file_name << "!" << std::endl;
 
         if (info_file.is_open())
           info_file.close();
         else
           throw art::Exception(art::errors::FileOpenError)
             << "Unable to open file " << info_file_name << "!" << std::endl;
       }
     }
 
     free(ostream); // free allocated memory
 
   } // lcvn::LArNuCVNZlibMaker::write_files