Inheritance diagram for cvn::CVNEventDump:

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

	~CVNEventDump ()

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

void	reconfigure (const fhicl::ParameterSet &pset)

void	beginJob () override

void	endJob () 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 Member Functions
void	WriteMapTH2 (const art::Event &evt, int slice, const PixelMap &pm)
	Function to extract TH2 from PixelMap and write to TFile. More...

Private Attributes
dunemva::MVAAlg	fMVAAlg

std::string	fPixelMapInput

std::string	fGenieGenModuleLabel

std::string	fEnergyNueLabel

std::string	fEnergyNumuLabel

std::string	fEnergyNutauLabel

bool	fGetEnergyOutput

bool	fGetEventWeight

bool	fWriteMapTH2

bool	fApplyFidVol

bool	fUseTopology

unsigned int	fTopologyHits

TrainingData *	fTrain

TTree *	fTrainTree

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 44 of file CVNEventDump_module.cc.

Constructor & Destructor Documentation

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

explicit

Definition at line 81 of file CVNEventDump_module.cc.

     : EDAnalyzer(pset),
     fMVAAlg(pset.get<fhicl::ParameterSet>("MVAAlg"))
   {
     this->reconfigure(pset);
   }

cvn::CVNEventDump::~CVNEventDump ( )

Definition at line 89 of file CVNEventDump_module.cc.

90 { }

Member Function Documentation

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

override

Definition at line 128 of file CVNEventDump_module.cc.

   {
 
     // Get the pixel maps
     std::vector< art::Ptr< cvn::PixelMap > > pixelmaplist;
     art::InputTag itag1(fPixelMapInput, fPixelMapInput);
     auto pixelmapListHandle = evt.getHandle< std::vector< cvn::PixelMap > >(itag1);
     if (pixelmapListHandle)
       art::fill_ptr_vector(pixelmaplist, pixelmapListHandle);
 
     // If we have no pixel map then return
     if(pixelmaplist.size() == 0) return;
 
     InteractionType interaction = kOther;
 
     // * monte carlo
     std::vector<art::Ptr<simb::MCTruth> > mclist;
     auto mctruthListHandle = evt.getHandle< std::vector<simb::MCTruth> >(fGenieGenModuleLabel);
     if (mctruthListHandle)
       art::fill_ptr_vector(mclist, mctruthListHandle);
 
     //unsigned short nmclist=  mclist.size();
 
     //std::cout<<"mctruth: "<<nmclist<<std::endl;
 
     art::Ptr<simb::MCTruth> truth = mclist[0];
     simb::MCNeutrino truthN=truth->GetNeutrino();
     //truth = mclist[0];
 
     AssignLabels labels;
 
     interaction = labels.GetInteractionType(truthN);
     if(fUseTopology){
       labels.GetTopology(truth,fTopologyHits);
 //      labels.PrintTopology();
     }
     float nuEnergy = 0;
     float lepEnergy = 0;
 //    if(truth.NeutrinoSet()){
       nuEnergy = truthN.Nu().E();
       lepEnergy = truthN.Lepton().E();
 //    }
 
     // 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 = truthN.Nu().EndPosition().Vect();
       bool isFid = (fabs(vtx.X())<310 && fabs(vtx.Y())<550 && vtx.Z()>50 && vtx.Z()<1244);
       if(!isFid) return;
     }
 
     float recoNueEnergy = 0.;
     float recoNumuEnergy = 0.;
     float recoNutauEnergy = 0.;
     // Should we use the EnergyReco_module reconstructed energies?
     if(fGetEnergyOutput){
       // Get the nue info
       if(fEnergyNueLabel != ""){
         auto energyRecoNueHandle = evt.getHandle<dune::EnergyRecoOutput>(fEnergyNueLabel);
         recoNueEnergy = energyRecoNueHandle->fNuLorentzVector.E();
       }
       // And the numu
       if(fEnergyNumuLabel != ""){
         auto energyRecoNumuHandle = evt.getHandle<dune::EnergyRecoOutput>(fEnergyNumuLabel);
         recoNumuEnergy = energyRecoNumuHandle->fNuLorentzVector.E();
       }
       // And the nutau
       if(fEnergyNutauLabel != ""){
         auto energyRecoNutauHandle = evt.getHandle<dune::EnergyRecoOutput>(fEnergyNutauLabel);
         recoNutauEnergy = energyRecoNutauHandle->fNuLorentzVector.E();
       }
     }
 
     // If we don't want to get the event weight then leave it as 1.0.
     double eventWeight = 1.;
     if(fGetEventWeight){
       double mvaResult = 0.; // We don't care about this, but need it for the call
       fMVAAlg.Run(evt,mvaResult,eventWeight);
     }
 
     // Create the training data and add it to the tree
     TrainingData train(interaction, nuEnergy, lepEnergy, recoNueEnergy,
       recoNumuEnergy, recoNutauEnergy, eventWeight, *pixelmaplist[0]);
     // Set the topology information
     int topPDG     = labels.GetPDG();
     int nprot      = labels.GetNProtons();
     int npion      = labels.GetNPions();
     int npi0       = labels.GetNPizeros();
     int nneut      = labels.GetNNeutrons();
     int toptype    = labels.GetTopologyType();
     int toptypealt = labels.GetTopologyTypeAlt();
     if(fUseTopology){
       train.SetTopologyInformation(topPDG, nprot, npion, npi0, nneut, toptype, toptypealt);
     }
     fTrain = &train;
     fTrainTree->Fill();
 
     // Make a plot of the pixel map if required
     if (fWriteMapTH2) WriteMapTH2(evt, 0, train.fPMap);
 
   }

void cvn::CVNEventDump::beginJob ( )

overridevirtual

Reimplemented from art::EDAnalyzer.

Definition at line 109 of file CVNEventDump_module.cc.

   {
 
     art::ServiceHandle<art::TFileService> tfs;
 
     fTrainTree = tfs->make<TTree>("CVNTrainTree", "Training records");
     fTrainTree->Branch("train", "cvn::TrainingData", &fTrain);
 
   }

void cvn::CVNEventDump::endJob ( )

overridevirtual

Reimplemented from art::EDAnalyzer.

Definition at line 122 of file CVNEventDump_module.cc.

   {
 
   }

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

Definition at line 93 of file CVNEventDump_module.cc.

   {
     fPixelMapInput  = pset.get<std::string>("PixelMapInput");
     fGenieGenModuleLabel  = pset.get<std::string>("GenieGenModuleLabel");
     fWriteMapTH2    = pset.get<bool>       ("WriteMapTH2");
     fApplyFidVol    = pset.get<bool>("ApplyFidVol");
     fEnergyNueLabel = pset.get<std::string> ("EnergyNueLabel");  
     fEnergyNumuLabel = pset.get<std::string> ("EnergyNumuLabel");
     fEnergyNutauLabel = pset.get<std::string> ("EnergyNutauLabel");
     fGetEnergyOutput = pset.get<bool> ("GetEnergyOutput");
     fGetEventWeight = pset.get<bool> ("GetEventWeight");
     fUseTopology  = pset.get<bool>("UseTopology");
     fTopologyHits = pset.get<unsigned int>("TopologyHitsCut");    
   }

void cvn::CVNEventDump::WriteMapTH2	(	const art::Event &	evt,
		int	slice,
		const PixelMap &	pm
	)

private

Function to extract TH2 from PixelMap and write to TFile.

Definition at line 236 of file CVNEventDump_module.cc.

   {
       std::stringstream name;
       name << "PixelMap_r" << evt.run() << "_s" << evt.subRun()<< "_e" << evt.event() << "_sl" << slice;
       std::stringstream nameL;
       nameL << "PixelTruthMap_r" << evt.run() << "_s" << evt.subRun()<< "_e" << evt.event() << "_sl" << slice;
       std::stringstream nameX;
       nameX << "PixelMap_X_r" << evt.run() << "_s" << evt.subRun()<< "_e" << evt.event() << "_sl" << slice;
       std::stringstream nameY;
       nameY << "PixelMap_Y_r" << evt.run() << "_s" << evt.subRun()<< "_e" << evt.event() << "_sl" << slice;
       std::stringstream nameZ;
       nameZ << "PixelMap_Z_r" << evt.run() << "_s" << evt.subRun()<< "_e" << evt.event() << "_sl" << slice;
       TH2F* hist  = pm.ToTH2();
       TH2F* histL = pm.ToLabTH2();
       TH2F* histX = pm.SingleViewToTH2(0);
       TH2F* histY = pm.SingleViewToTH2(1);
       TH2F* histZ = pm.SingleViewToTH2(2);
       hist->SetName(name.str().c_str());
       histL->SetName(nameL.str().c_str());
       histX->SetName(nameX.str().c_str());
       histY->SetName(nameY.str().c_str());
       histZ->SetName(nameZ.str().c_str());
 
       art::ServiceHandle<art::TFileService> tfs;
 
       TH2F* histWrite = tfs->make<TH2F>(*hist);
       histWrite->Write();
       TH2F* histWriteL = tfs->make<TH2F>(*histL);
       histWriteL->GetZaxis()->SetRangeUser(0,10);
       histWriteL->Write();
       TH2F* histWriteX = tfs->make<TH2F>(*histX);
       histWriteX->Write();
       TH2F* histWriteY = tfs->make<TH2F>(*histY);
       histWriteY->Write();
       TH2F* histWriteZ = tfs->make<TH2F>(*histZ);
       histWriteZ->Write();
 
       delete hist;
       delete histWrite;
       delete histL;
       delete histWriteL;
       delete histX;
       delete histWriteX;
       delete histY;
       delete histWriteY;
       delete histZ;
       delete histWriteZ;
 
   }

Member Data Documentation

bool cvn::CVNEventDump::fApplyFidVol

private

Definition at line 67 of file CVNEventDump_module.cc.

std::string cvn::CVNEventDump::fEnergyNueLabel

private

Definition at line 61 of file CVNEventDump_module.cc.

std::string cvn::CVNEventDump::fEnergyNumuLabel

private

Definition at line 62 of file CVNEventDump_module.cc.

std::string cvn::CVNEventDump::fEnergyNutauLabel

private

Definition at line 63 of file CVNEventDump_module.cc.

std::string cvn::CVNEventDump::fGenieGenModuleLabel

private

Definition at line 60 of file CVNEventDump_module.cc.

bool cvn::CVNEventDump::fGetEnergyOutput

private

Definition at line 64 of file CVNEventDump_module.cc.

bool cvn::CVNEventDump::fGetEventWeight

private

Definition at line 65 of file CVNEventDump_module.cc.

dunemva::MVAAlg cvn::CVNEventDump::fMVAAlg

private

Definition at line 57 of file CVNEventDump_module.cc.

std::string cvn::CVNEventDump::fPixelMapInput

private

Definition at line 59 of file CVNEventDump_module.cc.

unsigned int cvn::CVNEventDump::fTopologyHits

private

Definition at line 69 of file CVNEventDump_module.cc.

TrainingData* cvn::CVNEventDump::fTrain

private

Definition at line 72 of file CVNEventDump_module.cc.

TTree* cvn::CVNEventDump::fTrainTree

private

Definition at line 73 of file CVNEventDump_module.cc.

bool cvn::CVNEventDump::fUseTopology

private

Definition at line 68 of file CVNEventDump_module.cc.

bool cvn::CVNEventDump::fWriteMapTH2

private

Definition at line 66 of file CVNEventDump_module.cc.

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

dunereco/dunereco/CVN/art/CVNEventDump_module.cc

Public Member Functions

Private Member Functions

Private Attributes

Additional Inherited Members

Detailed Description

Constructor & Destructor Documentation

Member Function Documentation

Member Data Documentation