Inheritance diagram for cvn::CVNZlibMakerProtoDUNE:

Classes
struct	PrimaryTrainingInfo

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

	~CVNZlibMakerProtoDUNE ()

void	beginJob () override

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

void	reconfigure (const fhicl::ParameterSet &pset)

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 Member Functions
void	write_files (const PrimaryTrainingInfo &primary, const art::Ptr< cvn::PixelMap > pm, unsigned int n) const

Private Attributes
std::string	fOutputDir

std::string	fPixelMapInput

bool	fSetLog

std::vector< bool >	fReverseViews

std::string	fLArG4ModuleLabel

std::string	out_dir

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 36 of file CVNZlibMakerProtoDUNE_module.cc.

Constructor & Destructor Documentation

cvn::CVNZlibMakerProtoDUNE::CVNZlibMakerProtoDUNE ( fhicl::ParameterSet const & pset )

explicit

Definition at line 78 of file CVNZlibMakerProtoDUNE_module.cc.

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

cvn::CVNZlibMakerProtoDUNE::~CVNZlibMakerProtoDUNE ( )

Definition at line 85 of file CVNZlibMakerProtoDUNE_module.cc.

86 { }

Member Function Documentation

void cvn::CVNZlibMakerProtoDUNE::analyze ( const art::Event & evt )

override

Definition at line 131 of file CVNZlibMakerProtoDUNE_module.cc.

   {
 
     // Get the pixel maps
     std::vector<art::Ptr<cvn::PixelMap>> pixelmaps;
     art::InputTag itag1(fPixelMapInput, fPixelMapInput);
     auto h_pixelmaps = evt.getHandle<std::vector<cvn::PixelMap>>(itag1);
     if (h_pixelmaps)
       art::fill_ptr_vector(pixelmaps, h_pixelmaps);
 
     // If no pixel maps, quit
     if (pixelmaps.size() == 0) return;
 
     // MC information
     art::ServiceHandle< cheat::ParticleInventoryService > piService;
 
     // The truth information we need is the interaction vertex and interaction type
     AssignLabels labels;
     unsigned short beamParticleInteraction;
 
     // We only have one beam particle called "primary", so find it. It should be the
     // first particle, but best not to assume
     float beamParticleEnergy = 0;
     TVector3 beamParticleVtx; // This is the interaction vertex
     int beamParticlePDG = 0;
 
     bool gotPrimary = false;
     for(auto const m : piService->ParticleList()){
       const simb::MCParticle* particle = m.second;
       if(particle->Process().compare("primary")==0){
         beamParticleEnergy = particle->E();
         beamParticleVtx.SetXYZ(particle->EndX(),particle->EndY(),particle->EndZ());
         beamParticlePDG = particle->PdgCode();
         beamParticleInteraction = labels.GetProtoDUNEBeamInteractionType(*particle);
         gotPrimary = true;
         break;
       }
     }
 
     if(!gotPrimary) return;
 
     const PrimaryTrainingInfo beamPrimary(beamParticleVtx,beamParticleInteraction,beamParticlePDG,beamParticleEnergy);
 
     this->write_files(beamPrimary, pixelmaps.at(0), evt.event());
   }

void cvn::CVNZlibMakerProtoDUNE::beginJob ( )

overridevirtual

Reimplemented from art::EDAnalyzer.

Definition at line 99 of file CVNZlibMakerProtoDUNE_module.cc.

   {
     // Set the output directory.
     // First we look for CONDOR_DIR_INPUT and set it to the grid location.
     // If it isn't there, we look for a FHICL parameter.
     // Otherwise it just goes to the current directory.
 /*
     char const* grid_dir = getenv("CONDOR_DIR_INPUT");
     if (grid_dir != NULL) {
       char const* tmp_grid_dir = getenv("TMP");
       if (tmp_grid_dir == NULL)
         throw art::Exception(art::errors::NotFound)
           << "Could not find environment variable \"TMP\" which "
           << "is usually set for condor_lar environment." << std::endl;
       out_dir = std::string(tmp_grid_dir) + "/out";
     }
 */
     if (fOutputDir != "")
       out_dir = fOutputDir;
 
     else
       out_dir = ".";
 
     // Throw an error if the specified output directory doesn't exist
     if (!fs::exists(out_dir))
       throw art::Exception(art::errors::FileOpenError)
         << "Output directory " << out_dir << " does not exist!" << std::endl;
 
     // std::cout << "Writing files to output directory " << out_dir << std::endl;
   }

void cvn::CVNZlibMakerProtoDUNE::reconfigure ( const fhicl::ParameterSet & pset )

Definition at line 89 of file CVNZlibMakerProtoDUNE_module.cc.

   {
     fOutputDir = pset.get<std::string>("OutputDir", "");
     fPixelMapInput = pset.get<std::string>("PixelMapInput");
     fSetLog = pset.get<bool>("SetLog");
     fReverseViews = pset.get<std::vector<bool>>("ReverseViews");
     fLArG4ModuleLabel = pset.get<std::string>("LArG4ModuleLabel");
   }

void cvn::CVNZlibMakerProtoDUNE::write_files	(	const PrimaryTrainingInfo &	primary,
		const art::Ptr< cvn::PixelMap >	pm,
		unsigned int	n
	)		const

private

Definition at line 178 of file CVNZlibMakerProtoDUNE_module.cc.

   {
     CVNImageUtils image_utils;
     std::vector<unsigned char> pixel_array(3 * pm->NWire() * pm->NTdc());
 
     image_utils.DisableRegionSelection();
     image_utils.SetLogScale(fSetLog);
     image_utils.SetViewReversal(fReverseViews);
     image_utils.ConvertPixelMapToPixelArray(*(pm.get()),pixel_array);
 
     ulong src_len = 3 * pm->NWire() * pm->NTdc(); // 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 + "/cvn_event_" + std::to_string(n) + ".gz";
       std::string info_file_name = out_dir + "/cvn_event_" + std::to_string(n) + ".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 truth information
         
         info_file << primary.vertex.X() << std::endl;
         info_file << primary.vertex.Y() << std::endl;
         info_file << primary.vertex.Z() << std::endl;
         info_file << primary.energy << std::endl;
         info_file << primary.interaction << std::endl;
         info_file << primary.pdgCode << 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
 
   } // cvn::CVNZlibMakerProtoDUNE::write_files