Inheritance diagram for cvn::GCNZlibMakerProtoDUNE:

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

	~GCNZlibMakerProtoDUNE ()

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 Attributes
std::string	fOutputDir

std::string	fGraphLabel

unsigned int	fTopologyHitsCut

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 GCNZlibMakerProtoDUNE_module.cc.

Constructor & Destructor Documentation

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

explicit

Definition at line 59 of file GCNZlibMakerProtoDUNE_module.cc.

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

cvn::GCNZlibMakerProtoDUNE::~GCNZlibMakerProtoDUNE ( )

Definition at line 66 of file GCNZlibMakerProtoDUNE_module.cc.

67 { }

Member Function Documentation

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

override

Definition at line 97 of file GCNZlibMakerProtoDUNE_module.cc.

   {
 
     std::cout << "GCNZlibMakerProtoDUNE: looking for graphs with label " << fGraphLabel << std::endl;
 
     // Get the graphs
     std::vector<art::Ptr<cvn::GCNGraph>> graphs;
     auto h_graphs = evt.getHandle<std::vector<cvn::GCNGraph>>(fGraphLabel);
     if (h_graphs)
       art::fill_ptr_vector(graphs, h_graphs);
 
     // If no graphs, quit
     if (graphs.size() == 0) return;
 
     // MC information
     art::ServiceHandle< cheat::ParticleInventoryService > piService;
 
     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;
     unsigned int counter = 0;
     for(auto const graph : graphs){
       // If the graph has no nodes then give up
       if(graph->GetNumberOfNodes() == 0) return;
 
 //      std::cout << "GCNZlibMakerProtoDUNE: found graph with " << graph->GetNumberOfNodes() << " nodes" << std::endl;
 
 
       // Now write the zlib file using this information
       // We need to extract all of the information into a single vector to write
       // into the compressed file format
       const std::vector<float> vectorToWrite = graph->ConvertGraphToVector();
  
       ulong src_len = vectorToWrite.size() *sizeof(float);
       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 *) &vectorToWrite[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::stringstream image_file_name; 
         image_file_name << out_dir << "/gcn_event_" << evt.event() << "_" << counter << ".gz";
         std::stringstream info_file_name;
         info_file_name  << out_dir << "/gcn_event_" << evt.event() << "_" << counter << ".info";
 
         std::ofstream image_file (image_file_name.str(), std::ofstream::binary);
         std::ofstream info_file  (info_file_name.str());
 
         if(image_file.is_open() && info_file.is_open()) {
 
           // Write the graph to the file and close it
           image_file.write(ostream, dest_len);
           image_file.close(); // close file
 
           // Write the auxillary information to the text file
           info_file << beamParticleVtx.X() << std::endl;
           info_file << beamParticleVtx.Y() << std::endl;
           info_file << beamParticleVtx.Z() << std::endl;
           info_file << beamParticleEnergy << std::endl;
           info_file << beamParticleInteraction << std::endl; // Interaction type first
           info_file << beamParticlePDG << std::endl;
 
           // Number of nodes and node features is needed for unpacking
           info_file << graph->GetNumberOfNodes() << std::endl;
           info_file << graph->GetNode(0).GetNumberOfFeatures() << 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.str() << "!" << std::endl;
 
           if (info_file.is_open())
             info_file.close();
           else
             throw art::Exception(art::errors::FileOpenError)
               << "Unable to open file " << info_file_name.str() << "!" << std::endl;
         }
       }
       ++counter; 
       free(ostream);  // free allocated memory
     }
   }

void cvn::GCNZlibMakerProtoDUNE::beginJob ( )

overridevirtual

Reimplemented from art::EDAnalyzer.

Definition at line 80 of file GCNZlibMakerProtoDUNE_module.cc.

   {
     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::GCNZlibMakerProtoDUNE::reconfigure ( const fhicl::ParameterSet & pset )

Definition at line 70 of file GCNZlibMakerProtoDUNE_module.cc.

   {
     fOutputDir = pset.get<std::string>("OutputDir", "");
     fGraphLabel = pset.get<std::string>("GraphLabel");
     fTopologyHitsCut = pset.get<unsigned int>("TopologyHitsCut");
 
     fLArG4ModuleLabel = pset.get<std::string>("LArG4ModuleLabel");
   }