Inheritance diagram for cvn::GCNH5:

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

	~GCNH5 () noexcept

	GCNH5 (GCNH5 const &)=delete

	GCNH5 (GCNH5 &&)=delete

GCNH5 &	operator= (GCNH5 const &)=delete

GCNH5 &	operator= (GCNH5 &&)=delete

void	reconfigure (fhicl::ParameterSet const &p)

void	beginSubRun (art::SubRun const &sr) override

void	endSubRun (art::SubRun const &sr) override

void	analyze (art::Event const &e) 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
string	fGraphModuleLabel
	Name of graph producer module. More...

string	fGraphInstanceLabel
	Name of graph instance. More...

string	fTruthLabel
	Name of truth producer module. More...

string	fOutputName
	Output filename. More...

bool	fSaveEventTruth
	Whether to save event-level truth information. More...

bool	fSaveParticleTruth
	Whether to include particle truth information. More...

hep_hpc::hdf5::File	fFile
	Output HDF5 file. More...

hep_hpc::hdf5::Ntuple< Column< int, 1 >, Column< int, 1 >, Column< int, 1 >, Column< int, 1 >, Column< float, 1 >, Column< float, 1 >, Column< float, 1 >, Column< float, 1 >, Column< int, 1 >, Column< float, 1 >, Column< float, 1 >, Column< int, 1 >, Column< int, 1 > > *	fGraphNtuple
	graph ntuple More...

hep_hpc::hdf5::Ntuple< Column< int, 1 >, Column< int, 1 >, Column< int, 1 >, Column< int, 1 >, Column< float, 1 >, Column< float, 1 >, Column< float, 1 >, Column< float, 1 >, Column< float, 1 > > *	fEventNtuple
	Event ntuple. More...

hep_hpc::hdf5::Ntuple< Column< int, 1 >, Column< int, 1 >, Column< int, 1 >, Column< int, 1 >, Column< int, 1 >, Column< int, 1 >, Column< float, 1 >, Column< float, 1 >, Column< float, 1 >, Column< float, 1 >, Column< float, 1 >, Column< float, 1 >, Column< float, 1 >, Column< string, 1 >, Column< string, 1 > > *	fParticleNtuple
	Particle ntuple. More...

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 42 of file GCNH5_module.cc.

Constructor & Destructor Documentation

cvn::GCNH5::GCNH5 ( fhicl::ParameterSet const & p )

explicit

Definition at line 110 of file GCNH5_module.cc.

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

cvn::GCNH5::~GCNH5 ( )

inlinenoexcept

Definition at line 45 of file GCNH5_module.cc.

45 {};

cvn::GCNH5::GCNH5 ( GCNH5 const & )

delete

cvn::GCNH5::GCNH5 ( GCNH5 && )

delete

Member Function Documentation

void cvn::GCNH5::analyze ( art::Event const & e )

overridevirtual

Implements art::EDAnalyzer.

Definition at line 126 of file GCNH5_module.cc.

                                        {
 
     // Get the graphVector
     std::vector<art::Ptr<GCNGraph>> graphVector;
     art::InputTag itag1(fGraphModuleLabel, fGraphInstanceLabel);
     auto graphHandle = e.getHandle<std::vector<GCNGraph>>(itag1);
     if (!graphHandle) {
       throw art::Exception(art::errors::ProductNotFound)
         << "Could not find GCNGraph vector with module label "
         << fGraphModuleLabel << " and instance label "
         << fGraphInstanceLabel << "!" << endl;
     }
     art::fill_ptr_vector(graphVector, graphHandle);
 
     if (graphVector.size() > 1) throw art::Exception(art::errors::LogicError)
       << "There shouldn't be more than one GCNGraph per producer per event,"
       << " but here there are " << graphVector.size() << "." << endl;
 
     if (graphVector.empty()) return;
 
     int run = e.id().run();
     int subrun = e.id().subRun();
     int event = e.id().event();
 
     for (size_t itNode = 0; itNode < graphVector[0]->GetNumberOfNodes(); ++itNode) {
       GCNGraphNode node = graphVector[0]->GetNode(itNode);
 
       vector<float> pos = node.GetPosition();
       vector<float> feat = node.GetFeatures();
       vector<float> truth = node.GetGroundTruth();
 
       fGraphNtuple->insert(run, subrun, event, (int)round(abs(pos[0])),
         pos[1], pos[2], feat[0], feat[1], feat[4], feat[2], feat[3],
         (int)feat[5], truth[0]); 
     }
 
     // Event truth
     if (fSaveEventTruth) {
 
       // Get MC truth
       auto truthHandle = e.getHandle<vector<simb::MCTruth>>(fTruthLabel);
       if (!truthHandle || truthHandle->size() != 1) {
         throw art::Exception(art::errors::LogicError)
           << "Expected to find exactly one MC truth object!";
       }
       simb::MCNeutrino nutruth = truthHandle->at(0).GetNeutrino();
 
       // Fill variables
       fEventNtuple->insert(run, subrun, event,
         nutruth.CCNC() == simb::kCC,
         nutruth.Nu().E(),
         nutruth.Lepton().E(),
         nutruth.Nu().Momentum().Vect().Unit().X(),
         nutruth.Nu().Momentum().Vect().Unit().Y(),
         nutruth.Nu().Momentum().Vect().Unit().Z());
 
     } // if saving event truth
 
     // // Particle tree
     if (fSaveParticleTruth) {
       std::vector<ptruth> ptree = GCNFeatureUtils::GetParticleTree(graphVector[0].get());
       for (auto p : ptree) {
         fParticleNtuple->insert(run, subrun, event,
           std::get<0>(p), std::get<1>(p), std::get<2>(p),
           std::get<3>(p), std::get<4>(p), std::get<5>(p),
           std::get<6>(p), std::get<7>(p), std::get<8>(p),
           std::get<9>(p), std::get<10>(p), std::get<11>(p));
       }
     }
 
   } // cvn::GCNH5::analyze

void cvn::GCNH5::beginSubRun ( art::SubRun const & sr )

overridevirtual

Reimplemented from art::EDAnalyzer.

Definition at line 198 of file GCNH5_module.cc.

                                              {
 
     // Open HDF5 output
     boost::uuids::random_generator generator;
     boost::uuids::uuid uuid = generator();
     std::ostringstream fileName;
     fileName << fOutputName << "_r" << setfill('0') << setw(5) << sr.run()
       << "_r" << setfill('0') << setw(5) << sr.subRun() << "_" << uuid
       << ".h5";
 
     fFile = hep_hpc::hdf5::File(fileName.str(), H5F_ACC_TRUNC);
 
     fGraphNtuple = new hep_hpc::hdf5::Ntuple(
       make_ntuple({fFile, "graph_table", 1000},
       make_scalar_column<int>("run"),
       make_scalar_column<int>("subrun"),
       make_scalar_column<int>("event"),
       make_scalar_column<int>("plane"),
       make_scalar_column<float>("wire"),
       make_scalar_column<float>("time"),
       make_scalar_column<float>("integral"),
       make_scalar_column<float>("rms"),
       make_scalar_column<int>("rawplane"),
       make_scalar_column<float>("rawwire"),
       make_scalar_column<float>("rawtime"),
       make_scalar_column<int>("tpc"),
       make_scalar_column<int>("true_id")));
 
     if (fSaveEventTruth) {
       fEventNtuple = new hep_hpc::hdf5::Ntuple(
         make_ntuple({fFile, "event_table", 1000},
         make_scalar_column<int>("run"),
         make_scalar_column<int>("subrun"),
         make_scalar_column<int>("event"),
         make_scalar_column<int>("is_cc"),
         make_scalar_column<float>("nu_energy"),
         make_scalar_column<float>("lep_energy"),
         make_scalar_column<float>("nu_dir_x"),
         make_scalar_column<float>("nu_dir_y"),
         make_scalar_column<float>("nu_dir_z")));
     }
 
     if (fSaveParticleTruth) {
       fParticleNtuple = new hep_hpc::hdf5::Ntuple(
         make_ntuple({fFile, "particle_table", 1000},
         make_scalar_column<int>("run"),
         make_scalar_column<int>("subrun"),
         make_scalar_column<int>("event"),
         make_scalar_column<int>("id"),
         make_scalar_column<int>("type"),
         make_scalar_column<int>("parent_id"),
         make_scalar_column<float>("momentum"),
         make_scalar_column<float>("start_x"),
         make_scalar_column<float>("start_y"),
         make_scalar_column<float>("start_z"),
         make_scalar_column<float>("end_x"),
         make_scalar_column<float>("end_y"),
         make_scalar_column<float>("end_z"),
         make_scalar_column<string>("start_process"),
         make_scalar_column<string>("end_process")));
     }
   }

void cvn::GCNH5::endSubRun ( art::SubRun const & sr )

overridevirtual

Reimplemented from art::EDAnalyzer.

Definition at line 261 of file GCNH5_module.cc.

                                            {
     delete fGraphNtuple;
     if (fSaveEventTruth) delete fEventNtuple;
     if (fSaveParticleTruth) delete fParticleNtuple;
     fFile.close();
   }

GCNH5& cvn::GCNH5::operator= ( GCNH5 const & )

delete

GCNH5& cvn::GCNH5::operator= ( GCNH5 && )

delete

void cvn::GCNH5::reconfigure ( fhicl::ParameterSet const & p )

Definition at line 115 of file GCNH5_module.cc.

                                                     {
 
     fGraphModuleLabel   = p.get<string>("GraphModuleLabel");
     fGraphInstanceLabel = p.get<string>("GraphInstanceLabel");
     fTruthLabel         = p.get<string>("TruthLabel"),
     fOutputName         = p.get<string>("OutputName");
     fSaveEventTruth     = p.get<bool>("SaveEventTruth");
     fSaveParticleTruth  = p.get<bool>("SaveParticleTruth");
 
   } // cvn::GCNH5::reconfigure

Member Data Documentation

hep_hpc::hdf5::Ntuple<Column<int, 1>, Column<int, 1>, Column<int, 1>, Column<int, 1>, Column<float, 1>, Column<float, 1>, Column<float, 1>, Column<float, 1>, Column<float, 1> >* cvn::GCNH5::fEventNtuple

private

Event ntuple.

Definition at line 90 of file GCNH5_module.cc.

hep_hpc::hdf5::File cvn::GCNH5::fFile

private

Output HDF5 file.

Definition at line 67 of file GCNH5_module.cc.

string cvn::GCNH5::fGraphInstanceLabel

private

Name of graph instance.

Definition at line 61 of file GCNH5_module.cc.

string cvn::GCNH5::fGraphModuleLabel

private

Name of graph producer module.

Definition at line 60 of file GCNH5_module.cc.

hep_hpc::hdf5::Ntuple<Column<int, 1>, Column<int, 1>, Column<int, 1>, Column<int, 1>, Column<float, 1>, Column<float, 1>, Column<float, 1>, Column<float, 1>, Column<int, 1>, Column<float, 1>, Column<float, 1>, Column<int, 1>, Column<int, 1> >* cvn::GCNH5::fGraphNtuple

private

graph ntuple

Definition at line 80 of file GCNH5_module.cc.

string cvn::GCNH5::fOutputName

private

Output filename.

Definition at line 63 of file GCNH5_module.cc.

hep_hpc::hdf5::Ntuple<Column<int, 1>, Column<int, 1>, Column<int, 1>, Column<int, 1>, Column<int, 1>, Column<int, 1>, Column<float, 1>, Column<float, 1>, Column<float, 1>, Column<float, 1>, Column<float, 1>, Column<float, 1>, Column<float, 1>, Column<string, 1>, Column<string, 1> >* cvn::GCNH5::fParticleNtuple

private

Particle ntuple.

Definition at line 106 of file GCNH5_module.cc.

bool cvn::GCNH5::fSaveEventTruth

private

Whether to save event-level truth information.

Definition at line 64 of file GCNH5_module.cc.

bool cvn::GCNH5::fSaveParticleTruth

private

Whether to include particle truth information.

Definition at line 65 of file GCNH5_module.cc.

string cvn::GCNH5::fTruthLabel

private

Name of truth producer module.

Definition at line 62 of file GCNH5_module.cc.

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

dunereco/dunereco/CVN/art/GCNH5_module.cc

Public Member Functions

Private Attributes

Additional Inherited Members

Detailed Description

Constructor & Destructor Documentation

Member Function Documentation

Member Data Documentation