#include <Selection.h>

Inheritance diagram for ana::SBNOsc::Selection:

Public Member Functions
void	FileCleanup (TTree *eventTree)

void	FileSetup (TFile f, TTree eventTree)

void	Initialize (fhicl::ParameterSet *config)

void	ProcessEvent (const event::Event *event)

void	ProcessSubRun (const SubRun *subrun)

void	ProcessFileMeta (const FileMeta *meta)

void	Finalize ()

Public Member Functions inherited from core::PostProcessorBase
	PostProcessorBase ()

virtual	~PostProcessorBase ()

void	Run (std::vector< std::string > filelist)

void	Initialize (char *config=NULL, const std::string &output_fname="", unsigned n_workers=1)

Private Attributes
sbnd::CRTGeoAlg *	fCRTGeo

Cuts	fCuts

ROC	fROC

Histograms	fHistograms

Histograms	fCosmicHistograms

Histograms *	fHistsToFill

CRTHistos	fCRTCosmicHistos

CRTHistos	fCRTNeutrinoHistos

Normalize	fNormalize

fhicl::ParameterSet	fCutConfig

double	fCRTHitDistance

bool	fDoNormalize

bool	fFillAllTracks

double	fGoalPOT

double	fNCosmicData

std::vector< numu::TrackSelector >	fTrackSelectors

std::vector< std::string >	fTrackSelectorNames

std::vector< numu::TrackFunction >	fTrackProfileValues

numu::RecoEvent *	fRecoEvent

TFile *	fOutputFile

std::string	fHistogramPostfix

bool	fUseCalorimetry

int	fFileIndex

numu::MCType	fFileType

Additional Inherited Members
Protected Member Functions inherited from core::PostProcessorBase
virtual void	InitializeThread ()

unsigned	NWorkers ()

unsigned	WorkerID ()

Protected Attributes inherited from core::PostProcessorBase
ProviderManager *	fProviderManager
	Interface for provider access. More...

int	fConfigExperimentID

Detailed Description

Definition at line 42 of file Selection.h.

Member Function Documentation

void ana::SBNOsc::Selection::FileCleanup ( TTree * eventTree )

inlinevirtual

Any cleanup needed per file

Parameters

eventTree the TTree associated with the sbncode event.

Reimplemented from core::PostProcessorBase.

Definition at line 45 of file Selection.h.

45 {}

void ana::SBNOsc::Selection::FileSetup	(	TFile *	f,
		TTree *	eventTree
	)

virtual

Setup anything needed per file

Parameters

f	the TFile being opened
eventTree	the TTree associated with the sbncode event. Use this TTree to set branch addresses for everything other than the sbncode event.

Files are guaranteed to be processed in the order they are specified on the command line for sbn-postprocess

Reimplemented from core::PostProcessorBase.

Definition at line 88 of file Selection.cc.

                                                       {
     eventTree->SetBranchAddress("reco_event", &fRecoEvent);
     fCuts.Initialize(fCutConfig, fProviderManager->GetGeometryProvider());
     TParameter<int> *file_type = (TParameter<int> *)f->Get("MCType");
     fFileType = (numu::MCType) file_type->GetVal();
 
     if (fDoNormalize) {
       if (fFileType == numu::fIntimeCosmic) {
         fHistsToFill = &fCosmicHistograms;
       }
       else if (fFileType == numu::fOverlay) {
         fHistsToFill = &fHistograms;
       }
       else assert(false);
     }
     else {
       fHistsToFill = &fHistograms;
     }
     fFileIndex += 1;
     
     CRTHistos crt_from_file;
     crt_from_file.Get(*f, "_all");
     if (fFileType == numu::fIntimeCosmic) {
       fCRTCosmicHistos.Add(crt_from_file);
     }
     else if (fFileType == numu::fOverlay) {
       fCRTNeutrinoHistos.Add(crt_from_file);
     }
   }

void ana::SBNOsc::Selection::Finalize ( )

virtual

Perform user-level finalization. Called after all events have been processed.

Reimplemented from core::PostProcessorBase.

Definition at line 168 of file Selection.cc.

                            {
     fOutputFile->cd();
     std::cout << "Finalizing!\n";
     if (fDoNormalize) {
       std::cout << "Normalizing!\n";
       // save scale per 1e20 POT
       TParameter<float> scale_neutrino("ScaleNeutrino", fNormalize.ScaleNeutrino(1.e20));
       TParameter<float> scale_cosmic("ScaleCosmic", fNormalize.ScaleCosmic(1.e20));
       scale_neutrino.Write();
       scale_cosmic.Write();
       std::cout << "Scale Neutrino: " << fNormalize.ScaleNeutrino(1.e20) << std::endl;
       std::cout << "Scale Cosmic: " << fNormalize.ScaleCosmic(1.e20) << std::endl;
 
       fHistograms.Scale(fNormalize.ScaleNeutrino(fGoalPOT));
       fCosmicHistograms.Scale(fNormalize.ScaleCosmic(fGoalPOT));
 
       fCRTCosmicHistos.Scale(fNormalize.ScaleNeutrino(fGoalPOT));
       fCRTNeutrinoHistos.Scale(fNormalize.ScaleCosmic(fGoalPOT));
 
       fHistograms.Add(fCosmicHistograms);
       fROC.Normalize(fNormalize.ScaleNeutrino(fGoalPOT), fNormalize.ScaleCosmic(fGoalPOT));  
     }
     fROC.BestCuts();
     fROC.Write();
     fHistograms.Write();
     fCRTCosmicHistos.Write();
     fCRTNeutrinoHistos.Write();
 
   }

void ana::SBNOsc::Selection::Initialize ( fhicl::ParameterSet * config )

virtual

Perform user-level initialization.

Parameters

config A configuration, as a JSON object.

Implements core::PostProcessorBase.

Definition at line 32 of file Selection.cc.

                                                       {
     std::cout << "Input Configuration:\n";
     std::cout << config->to_indented_string() << std::endl;
     fOutputFile = new TFile(config->get<std::string>("OutputFile", "output.root").c_str(), "CREATE");
     fOutputFile->cd();
     fCRTHitDistance = config->get<double>("CRTHitDistance", -1);
 
     fFileIndex = 0;
 
     fCutConfig = config->get<fhicl::ParameterSet>("Cuts");
 
     fDoNormalize = config->get<bool>("DoNormalize", false);
     fGoalPOT = config->get<double>("GoalPOT", 0.);
     fFillAllTracks = config->get<bool>("FillAllTracks", true);
 
     fUseCalorimetry = config->get<bool>("UseCalorimetry", true);
 
     fHistogramPostfix = config->get<std::string>("HistogramPostfix", "");
 
     std::vector<std::vector<std::string>> track_selector_strings = config->get<std::vector<std::vector<std::string>>>("TrackSelectors", {{""}});
     std::vector<std::vector<std::string>> track_selector_names = config->get<std::vector<std::vector<std::string>>>("TrackSelectorNames", {{""}});
 
     assert(track_selector_strings.size() == track_selector_names.size());
 
     fTrackSelectorNames = numu::MultiplyNames(track_selector_names, '/');
     fTrackSelectors = numu::MultiplyTrackSelectors(track_selector_strings);
 
     assert(fTrackSelectorNames.size() == fTrackSelectors.size());
 
     std::vector<std::string> track_profile_values = config->get<std::vector<std::string>>("TrackProfileValues", {});
     for (const std::string &source: track_profile_values) {
       fTrackProfileValues.push_back(uscript::compile<numu::RecoTrack, numu::TrueParticle, unsigned>("track", "particle", "mctype", source.c_str()));
     }
 
     std::vector<std::string> track_profile_value_names = config->get<std::vector<std::string>>("TrackProfileValueNames", {});
     std::vector<std::tuple<unsigned, float, float>> track_profile_xranges =  config->get<std::vector<std::tuple<unsigned, float, float>>>("TrackProfileXRanges", {});
 
     assert(track_profile_value_names.size() == track_profile_xranges.size());
     assert(fTrackProfileValues.size() == track_profile_xranges.size());
 
     if (fDoNormalize) {
       fNormalize.Initialize(config->get<fhicl::ParameterSet>("Normalize", {}));
     }
     fROC.Initialize();
     fRecoEvent = NULL;
 
     fCRTGeo = new sbnd::CRTGeoAlg(fProviderManager->GetGeometryProvider(), fProviderManager->GetAuxDetGeometryProvider());
 
     fCuts.Initialize(fCutConfig, fProviderManager->GetGeometryProvider());
 
     if (fDoNormalize) {
       fCosmicHistograms.Initialize(fProviderManager->GetGeometryProvider(), *fCRTGeo, fCuts, "Cosmic", fTrackSelectorNames, track_profile_value_names, track_profile_xranges);
     }
     fHistograms.Initialize(fProviderManager->GetGeometryProvider(), *fCRTGeo, fCuts, fHistogramPostfix,  fTrackSelectorNames, track_profile_value_names, track_profile_xranges); 
   }

void ana::SBNOsc::Selection::ProcessEvent ( const event::Event * event )

virtual

Process one event.

Parameters

event The sbncode event for the current event

Implements core::PostProcessorBase.

Definition at line 130 of file Selection.cc.

                                                            {
     std::cout << "Event: " << core_event->metadata.eventID << std::endl;
 
     // set stuff in the event
     SetEvent(*fRecoEvent, *core_event, fCuts, fFileType, fUseCalorimetry);
 
     for (const numu::RecoInteraction &reco: fRecoEvent->reco) {
       std::cout << "Event type: " << Histograms::mode2Str(reco.match.mode) << std::endl;
       const std::vector<size_t> &tracks = reco.slice.tracks;
       for (size_t ind: tracks) {
         const numu::RecoTrack &track = fRecoEvent->tracks.at(ind);
         const numu::RecoParticle &t_particle = reco.slice.particles.at(ind);
         std::cout << "Reco track: "; 
         std::cout << "length: " << track.length << " PID: " << track.match.match_pdg << " Chi2 m: " << track.chi2_muon << " Chi2 p: " << track.chi2_proton << " n daughter: " << t_particle.daughters.size() << std::endl;
         for (size_t d_ind: t_particle.daughters) {
           if (fRecoEvent->tracks.count(d_ind)) {
             std::cout << "Daughter track: ";
             const numu::RecoTrack &d_track = fRecoEvent->tracks.at(ind);
             const numu::RecoParticle &d_t_particle = reco.slice.particles.at(ind);
             std::cout << "length: " << d_track.length << " PID: " << d_track.match.match_pdg << " Chi2 m: " << d_track.chi2_muon << " Chi2 p: " << d_track.chi2_proton <<" n daughter: " << d_t_particle.daughters.size() << std::endl;
           }
           else {
             std::cout << "Shower Daughter\n";
           }
         }
       }
     }
 
     fROC.Fill(fCuts, *fRecoEvent, fFileType == numu::fIntimeCosmic);
     fHistsToFill->Fill(*fRecoEvent, *core_event, fCuts, fTrackSelectors, fTrackProfileValues, fFillAllTracks);
 
     if (fDoNormalize) {
       if (fFileType == numu::fIntimeCosmic) fNormalize.AddCosmicEvent(*core_event);
       else fNormalize.AddNeutrinoEvent(*core_event);
     }
 
   }

void ana::SBNOsc::Selection::ProcessFileMeta ( const FileMeta * meta )

virtual

Reimplemented from core::PostProcessorBase.

Definition at line 124 of file Selection.cc.

                                                       {
     if (fDoNormalize && fFileType == numu::fIntimeCosmic) {
       fNormalize.AddCosmicFile(*meta);
     }
   }

void ana::SBNOsc::Selection::ProcessSubRun ( const SubRun * subrun )

virtual

Reimplemented from core::PostProcessorBase.

Definition at line 118 of file Selection.cc.

                                                     {
     if (fDoNormalize && fFileType == numu::fOverlay) {
       fNormalize.AddNeutrinoSubRun(*subrun);
     }
   }

Member Data Documentation

Histograms ana::SBNOsc::Selection::fCosmicHistograms

private

Definition at line 58 of file Selection.h.

CRTHistos ana::SBNOsc::Selection::fCRTCosmicHistos

private

Definition at line 61 of file Selection.h.

sbnd::CRTGeoAlg* ana::SBNOsc::Selection::fCRTGeo

private

Definition at line 54 of file Selection.h.

double ana::SBNOsc::Selection::fCRTHitDistance

private

Definition at line 66 of file Selection.h.

CRTHistos ana::SBNOsc::Selection::fCRTNeutrinoHistos

private

Definition at line 62 of file Selection.h.

fhicl::ParameterSet ana::SBNOsc::Selection::fCutConfig

private

Definition at line 65 of file Selection.h.

Cuts ana::SBNOsc::Selection::fCuts

private

Definition at line 55 of file Selection.h.

bool ana::SBNOsc::Selection::fDoNormalize

private

Definition at line 67 of file Selection.h.

int ana::SBNOsc::Selection::fFileIndex

private

Definition at line 82 of file Selection.h.

numu::MCType ana::SBNOsc::Selection::fFileType

private

Definition at line 83 of file Selection.h.

bool ana::SBNOsc::Selection::fFillAllTracks

private

Definition at line 68 of file Selection.h.

double ana::SBNOsc::Selection::fGoalPOT

private

Definition at line 69 of file Selection.h.

std::string ana::SBNOsc::Selection::fHistogramPostfix

private

Definition at line 80 of file Selection.h.

Histograms ana::SBNOsc::Selection::fHistograms

private

Definition at line 57 of file Selection.h.

Histograms* ana::SBNOsc::Selection::fHistsToFill

private

Definition at line 59 of file Selection.h.

double ana::SBNOsc::Selection::fNCosmicData

private

Definition at line 70 of file Selection.h.

Normalize ana::SBNOsc::Selection::fNormalize

private

Definition at line 64 of file Selection.h.

TFile* ana::SBNOsc::Selection::fOutputFile

private

Definition at line 78 of file Selection.h.

numu::RecoEvent* ana::SBNOsc::Selection::fRecoEvent

private

Definition at line 77 of file Selection.h.

ROC ana::SBNOsc::Selection::fROC

private

Definition at line 56 of file Selection.h.

std::vector<numu::TrackFunction> ana::SBNOsc::Selection::fTrackProfileValues

private

Definition at line 75 of file Selection.h.

std::vector<std::string> ana::SBNOsc::Selection::fTrackSelectorNames

private

Definition at line 73 of file Selection.h.

std::vector<numu::TrackSelector> ana::SBNOsc::Selection::fTrackSelectors

private

Definition at line 72 of file Selection.h.

bool ana::SBNOsc::Selection::fUseCalorimetry

private

Definition at line 81 of file Selection.h.

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

Public Member Functions

Private Attributes

Additional Inherited Members

Detailed Description

Member Function Documentation

Member Data Documentation