Inheritance diagram for cvn::GCNZlibMaker:

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

	~GCNZlibMaker ()

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	fGenieGenModuleLabel

std::string	fEnergyNueLabel

std::string	fEnergyNumuLabel

std::string	fEnergyNutauLabel

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 37 of file GCNZlibMaker_module.cc.

Constructor & Destructor Documentation

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

explicit

Definition at line 63 of file GCNZlibMaker_module.cc.

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

cvn::GCNZlibMaker::~GCNZlibMaker ( )

Definition at line 70 of file GCNZlibMaker_module.cc.

71 { }

Member Function Documentation

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

override

Definition at line 104 of file GCNZlibMaker_module.cc.

   {
 
     std::cout << "GCNZlibMaker: 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
     std::cout << "Found " << graphs.size() << " graphs" << std::endl;
     if (graphs.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();
 
     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();
 
     // We only want interactions in the fiducial volume for training
     // 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 = (fabs(vtx.X())<310. && fabs(vtx.Y())<550. && vtx.Z()>50. && vtx.Z()<1244.);
     if(!isFid) return;
 
     float reco_nue_energy = 0.;
     float reco_numu_energy = 0.;
     float reco_nutau_energy = 0.;
 
     // Get nue info
     if (fEnergyNueLabel != "") {
       auto h_ereco = evt.getHandle<dune::EnergyRecoOutput>(fEnergyNueLabel);
       reco_nue_energy = h_ereco->fNuLorentzVector.E();
     }
 
     // Get numu info
     if (fEnergyNueLabel != "") {
       auto h_ereco = evt.getHandle<dune::EnergyRecoOutput>(fEnergyNumuLabel);
       reco_numu_energy = h_ereco->fNuLorentzVector.E();
     }
 
     // Get nutau info
     if (fEnergyNutauLabel != "") {
       auto h_ereco = evt.getHandle<dune::EnergyRecoOutput>(fEnergyNutauLabel);
       reco_nutau_energy = h_ereco->fNuLorentzVector.E();
     }
 
     float event_weight = -1.0; // We currently just set this to -1
 
     // If the graph has no nodes then give up
     for(unsigned int i = 0; i < graphs.size(); ++i){
       const art::Ptr<cvn::GCNGraph> g = graphs.at(i);
       
       std::cout << "GCNZlibMaker: found graph with " << g->GetNumberOfNodes() << " nodes" << std::endl;
       if(g->GetNumberOfNodes() == 0) continue;
   
       // 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 = g->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 {
 
         std::stringstream modifier;
         if(graphs.size() > 1){
           modifier << "_" << i;
         }
   
         // Create output files 
         std::stringstream image_file_name; 
         image_file_name << out_dir << "/event_" << evt.event() << modifier.str() << ".gz";
         std::stringstream info_file_name;
         info_file_name << out_dir << "/event_" << evt.event() << modifier.str() << ".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 << interaction << std::endl; // Interaction type first
   
           // True and reconstructed energy variables
           info_file << nu_energy << std::endl;
           info_file << lep_energy << std::endl;
           info_file << reco_nue_energy << std::endl;
           info_file << reco_numu_energy << std::endl;
           info_file << reco_nutau_energy << std::endl;
           info_file << event_weight << std::endl;
   
           info_file << labels.GetPDG() << std::endl;
           info_file << labels.GetNProtons() << std::endl;
           info_file << labels.GetNPions() << std::endl;
           info_file << labels.GetNPizeros() << std::endl;
           info_file << labels.GetNNeutrons() << std::endl;
           info_file << labels.GetTopologyType() << std::endl;
           info_file << labels.GetTopologyTypeAlt() << std::endl;
   
           // Number of nodes and node features is needed for unpacking
           info_file << g->GetNumberOfNodes() << std::endl;
           info_file << g->GetNumberOfNodeCoordinates() << std::endl;
           info_file << g->GetNumberOfNodeFeatures() << 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;
         }
       }
       free(ostream);  // free allocated memory
       ostream = nullptr;
     }
     
     return;
   }

void cvn::GCNZlibMaker::beginJob ( )

overridevirtual

Reimplemented from art::EDAnalyzer.

Definition at line 87 of file GCNZlibMaker_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::GCNZlibMaker::reconfigure ( const fhicl::ParameterSet & pset )

Definition at line 74 of file GCNZlibMaker_module.cc.

   {
     fOutputDir = pset.get<std::string>("OutputDir", "");
     fGraphLabel = pset.get<std::string>("GraphLabel");
     fTopologyHitsCut = pset.get<unsigned int>("TopologyHitsCut");
 
     fGenieGenModuleLabel = pset.get<std::string>("GenieGenModuleLabel");
     fEnergyNueLabel = pset.get<std::string>("EnergyNueLabel");
     fEnergyNumuLabel = pset.get<std::string>("EnergyNumuLabel");
     fEnergyNutauLabel = pset.get<std::string>("EnergyNutauLabel");
   }

Member Data Documentation

std::string cvn::GCNZlibMaker::fEnergyNueLabel

private

Definition at line 54 of file GCNZlibMaker_module.cc.

std::string cvn::GCNZlibMaker::fEnergyNumuLabel

private

Definition at line 55 of file GCNZlibMaker_module.cc.

std::string cvn::GCNZlibMaker::fEnergyNutauLabel

private

Definition at line 56 of file GCNZlibMaker_module.cc.

std::string cvn::GCNZlibMaker::fGenieGenModuleLabel

private

Definition at line 53 of file GCNZlibMaker_module.cc.

std::string cvn::GCNZlibMaker::fGraphLabel

private

Definition at line 50 of file GCNZlibMaker_module.cc.

std::string cvn::GCNZlibMaker::fOutputDir

private

Definition at line 49 of file GCNZlibMaker_module.cc.

unsigned int cvn::GCNZlibMaker::fTopologyHitsCut

private

Definition at line 51 of file GCNZlibMaker_module.cc.

std::string cvn::GCNZlibMaker::out_dir

private

Definition at line 58 of file GCNZlibMaker_module.cc.

The documentation for this class was generated from the following file:

dunereco/dunereco/CVN/art/GCNZlibMaker_module.cc

Public Member Functions

Private Attributes

Additional Inherited Members

Detailed Description

Constructor & Destructor Documentation

Member Function Documentation

Member Data Documentation