ana::SBNOsc::Flatten Class Reference

#include <Flatten.h>

Inheritance diagram for ana::SBNOsc::Flatten:

Public Member Functions
void	FileSetup (TFile *f, TTree *eventTree)
void	Initialize (fhicl::ParameterSet *config)
void	ProcessEvent (const event::Event *event)
void	InitializeThread ()
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
Cuts	fCuts
std::vector< TFile * >	fOutputFiles
std::vector< TNtuple * >	fNtuples
std::vector < numu::flat::FlatInteraction >	fInteractions
std::vector< numu::MCType >	fMCTypes
std::vector< numu::RecoEvent * >	fRecoEvents

Additional Inherited Members
Protected Member Functions inherited from core::PostProcessorBase
virtual void	FileCleanup (TTree *eventTree)
unsigned	NWorkers ()
unsigned	WorkerID ()
Protected Attributes inherited from core::PostProcessorBase
ProviderManager *	fProviderManager
	Interface for provider access. More...
int	fConfigExperimentID

Detailed Description

Definition at line 30 of file Flatten.h.

Member Function Documentation

void ana::SBNOsc::Flatten::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 107 of file Flatten.cc.

                                                           {
  unsigned ind = WorkerID();

  eventTree->SetBranchAddress("reco_event", &fRecoEvents[ind]);

  TParameter<int> *file_type = (TParameter<int> *)f->Get("MCType");
  fMCTypes[ind] = (numu::MCType) file_type->GetVal();  

  fInteractions[ind].meta.mc_type =  fMCTypes[ind];
  fInteractions[ind].meta.detector = fConfigExperimentID; 
}

void ana::SBNOsc::Flatten::Finalize ( )

virtual

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

Reimplemented from core::PostProcessorBase.

Definition at line 198 of file Flatten.cc.

                                {
  TList *merge = new TList;
  fOutputFiles[0]->cd();
  for (unsigned i = 1; i < NWorkers(); i++) {
    merge->Add(fNtuples[i]);
  }
  fNtuples[0]->Merge(merge);
  fNtuples[0]->Write();
  delete merge;

  for (unsigned i = 1; i < NWorkers(); i++) {
    const char *name = fOutputFiles[i]->GetName();
    fOutputFiles[i]->Close();
    gSystem->Unlink(name);
  }
 
}

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

virtual

Perform user-level initialization.

Parameters

config

A configuration, as a JSON object.

Implements core::PostProcessorBase.

Definition at line 82 of file Flatten.cc.

                                                           {
  fCuts.Initialize(config->get<fhicl::ParameterSet>("Cuts"), fProviderManager->GetGeometryProvider());
  std::string output_name = config->get<std::string>("OutputFile", "output.root");

  fOutputFiles = std::vector<TFile *>(NWorkers());
  fNtuples = std::vector<TNtuple *>(NWorkers());
  fInteractions = std::vector<numu::flat::FlatInteraction>(NWorkers());
  fRecoEvents = std::vector<numu::RecoEvent *>(NWorkers(), 0);
  fMCTypes = std::vector<numu::MCType>(NWorkers());
  std::string ntuple = ClasstoNTupleName("numu::flat::FlatInteraction", "interaction");

  for (unsigned i = 0; i < NWorkers(); i++) {
    std::string name = output_name;
    if (i > 0) name += std::to_string(i);
    fOutputFiles[i] = new TFile(name.c_str(), "RECREATE");
    fOutputFiles[i]->cd();
    fNtuples[i] = new TNtuple("interaction", "interaction", ntuple.c_str());
  }
}

void ana::SBNOsc::Flatten::InitializeThread ( )

virtual

Perform user-level initialization per-thread. Only called when multiple workers are specified on the command line.

Reimplemented from core::PostProcessorBase.

Definition at line 102 of file Flatten.cc.

                                        {
  unsigned i = WorkerID();
  fOutputFiles[i]->cd();
}

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

virtual

Process one event.

Parameters

event

The sbncode event for the current event

Implements core::PostProcessorBase.

Definition at line 119 of file Flatten.cc.

                                                        {
  unsigned index = WorkerID();
  ana::SBNOsc::SetEvent(*fRecoEvents[index], *ev, fCuts, fMCTypes[index], true);
  for (const numu::RecoInteraction &interaction: fRecoEvents[index]->reco) {
    const numu::RecoTrack &primary_track = fRecoEvents[index]->tracks.at(interaction.slice.primary_track_index);
    // set stuff
    // 
    // Primary Track
    numu::flat::PrimaryTrack track;
    track.length = primary_track.length; 
    track.costh = primary_track.costh;
    track.range_momentum = numu::RangeMomentum(primary_track);
    track.mcs_momentum = numu::MCSMomentum(primary_track);
    track.crt_hit_distance = (primary_track.crt_match.hit_match.present) ? primary_track.crt_match.hit_match.distance : -1;
    track.crt_hit_time = primary_track.crt_match.hit_match.time;
    track.crt_track_angle = (primary_track.crt_match.track.present) ? primary_track.crt_match.track.angle: -1;
    track.crt_track_time = primary_track.crt_match.track.time;
    primary_track.start.GetXYZ(track.start);
    primary_track.end.GetXYZ(track.end);
    if (primary_track.match.has_match) {
      const numu::TrueParticle &truth = fRecoEvents[index]->particles.at(primary_track.match.mcparticle_id);
      truth.start_momentum.GetXYZ(track.truth.momentum);
      track.truth.wall_enter = truth.wall_enter;
      track.truth.wall_exit = truth.wall_exit;
      track.truth.time = truth.start_time;
      track.truth.is_cosmic = truth.is_cosmic;
      track.truth.pdgid = truth.pdgid;
      track.truth.is_contained = truth.is_contained;
    }
    else track.truth.pdgid = -1; 
   
    // Slice
    numu::flat::Slice slice;
    slice.flash_score = interaction.slice.flash_match.score;
    slice.flash_time = interaction.slice.flash_match.time;
  
    // True Neutrino
    numu::flat::TrueNeutrino neutrino;
    if (interaction.match.has_match) {
      neutrino.E = ev->truth[interaction.match.mctruth_track_id].neutrino.energy;
      neutrino.Q2 = ev->truth[interaction.match.mctruth_track_id].neutrino.Q2;
      ev->truth[interaction.match.mctruth_track_id].neutrino.position.GetXYZ(neutrino.vertex);
    }
    else neutrino.E = -1;

    // Event Info
    numu::flat::EventInfo event_info;
    for (unsigned i = 0; i < 10; i++) event_info.crt_hit_pes[i] = -1;
    // Event Info
    unsigned i = 0;
    for (const numu::CRTHit &hit: fRecoEvents[index]->in_time_crt_hits) {
      assert(i < 10);
      event_info.crt_hit_pes[i] = hit.pes; 
      event_info.crt_hit_times[i] = hit.time;
      i += 1;
    }
    event_info.pass_trig = fCuts.PassFlashTrigger(*fRecoEvents[index]);

    // Event Meta data -- set previously

    fNtuples[index]->Fill((float *)&fInteractions[index]);

    // overwrite normalization stuff so it isn't duplicated
    fInteractions[index].meta.pot = 0;
    fInteractions[index].meta.n_gen_events = 0;
  }

}

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

virtual

Reimplemented from core::PostProcessorBase.

Definition at line 193 of file Flatten.cc.

                                                           {
  unsigned index = WorkerID();
  fInteractions[index].meta.n_gen_events += meta->n_gen_events;
}

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

virtual

Reimplemented from core::PostProcessorBase.

Definition at line 188 of file Flatten.cc.

                                                         {
  unsigned index = WorkerID();
  fInteractions[index].meta.pot += subrun->totgoodpot; 
}

Member Data Documentation

Cuts ana::SBNOsc::Flatten::fCuts

private

Definition at line 43 of file Flatten.h.

std::vector<numu::flat::FlatInteraction> ana::SBNOsc::Flatten::fInteractions

private

Definition at line 46 of file Flatten.h.

std::vector<numu::MCType> ana::SBNOsc::Flatten::fMCTypes

private

Definition at line 47 of file Flatten.h.

std::vector<TNtuple *> ana::SBNOsc::Flatten::fNtuples

private

Definition at line 45 of file Flatten.h.

std::vector<TFile *> ana::SBNOsc::Flatten::fOutputFiles

private

Definition at line 44 of file Flatten.h.

std::vector<numu::RecoEvent *> ana::SBNOsc::Flatten::fRecoEvents

private

Definition at line 48 of file Flatten.h.

---

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

• Flatten.h
• Flatten.cc