trigger::NtupleWriter Class Reference

Null algorithm test. More...

#include <NtupleWriter.h>

Public Member Functions
	NtupleWriter (const std::string &name, framework::Process &process)
virtual void	configure (framework::config::Parameters &)
	Callback for the EventProcessor to configure itself from the given set of parameters.
virtual void	produce (framework::Event &event)
	Process the event and put new data products into it.
virtual void	onProcessStart ()
	Callback for the EventProcessor to take any necessary action when the processing of events starts, such as creating histograms.
virtual void	onProcessEnd ()
	Callback for the EventProcessor to take any necessary action when the processing of events finishes, such as calculating job-summary quantities.
Public Member Functions inherited from framework::Producer
	Producer (const std::string &name, Process &process)
	Class constructor.
virtual void	process (Event &event) final
	Processing an event for a Producer is calling produce.
Public Member Functions inherited from framework::EventProcessor
	DECLARE_FACTORY (EventProcessor, EventProcessor *, const std::string &, Process &)
	declare that we have a factory for this class
	EventProcessor (const std::string &name, Process &process)
	Class constructor.
virtual	~EventProcessor ()=default
	Class destructor.
virtual void	beforeNewRun (ldmx::RunHeader &run_header)
	Callback for Producers to add parameters to the run header before conditions are initialized.
virtual void	onNewRun (const ldmx::RunHeader &run_header)
	Callback for the EventProcessor to take any necessary action when the run being processed changes.
virtual void	onFileOpen (EventFile &event_file)
	Callback for the EventProcessor to take any necessary action when a new event input ROOT file is opened.
virtual void	onFileClose (EventFile &event_file)
	Callback for the EventProcessor to take any necessary action when a event input ROOT file is closed.
template<class T >
const T &	getCondition (const std::string &condition_name)
	Access a conditions object for the current event.
TDirectory *	getHistoDirectory ()
	Access/create a directory in the histogram file for this event processor to create histograms and analysis tuples.
void	setStorageHint (framework::StorageControl::Hint hint)
	Mark the current event as having the given storage control hint from this module_.
void	setStorageHint (framework::StorageControl::Hint hint, const std::string &purposeString)
	Mark the current event as having the given storage control hint from this module and the given purpose string.
int	getLogFrequency () const
	Get the current logging frequency from the process.
int	getRunNumber () const
	Get the run number from the process.
std::string	getName () const
	Get the processor name.
void	createHistograms (const std::vector< framework::config::Parameters > &histos)
	Internal function which is used to create histograms passed from the python configuration @parma histos vector of Parameters that configure histograms to create.

Private Attributes
TFile *	out_file_ {nullptr}
std::string	tag_ {"Events"}
std::string	out_path_ {"./ntuple.root"}
std::string	target_sp_hits_event_passname_
std::string	target_sp_passname_
std::string	ecal_sp_hits_events_passname_
std::string	ecal_sp_passname_
std::string	ecal_trig_sums_event_passname_
std::string	ecal_trig_sums_passname_
std::string	hcal_trig_quads_events_passname_
std::string	hcal_trig_quads_passname_
std::string	trig_electrons_event_passname_
std::string	trig_electrons_passname_
bool	write_truth_ {true}
bool	write_ele_ {true}
bool	write_ecal_sums_ {true}
bool	write_hcal_sums_ {true}
bool	write_ecal_trig_mi_ps_ {true}
bool	write_hcal_trig_mi_ps_ {true}

Additional Inherited Members
Protected Member Functions inherited from framework::EventProcessor
void	abortEvent ()
	Abort the event immediately.
Protected Attributes inherited from framework::EventProcessor
HistogramPool	histograms_
	helper object for making and filling histograms
NtupleManager &	ntuple_ {NtupleManager::getInstance()}
	Manager for any ntuples.
logging::logger	the_log_
	The logger for this EventProcessor.

Detailed Description

Null algorithm test.

Definition at line 15 of file NtupleWriter.h.

Constructor & Destructor Documentation

NtupleWriter()

trigger::NtupleWriter::NtupleWriter	(	const std::string &	name,
		framework::Process &	process )

Definition at line 10 of file NtupleWriter.cxx.

    : Producer(name, process) {}

Member Function Documentation

configure()

void trigger::NtupleWriter::configure ( framework::config::Parameters & parameters )

virtual

Callback for the EventProcessor to configure itself from the given set of parameters.

The parameters a processor has access to are the member variables of the python class in the sequence that has class_name equal to the EventProcessor class name.

For an example, look at MyProcessor.

Parameters

parameters

Parameters for configuration.

Reimplemented from framework::EventProcessor.

Definition at line 13 of file NtupleWriter.cxx.

                                                          {
  out_path_ = ps.get<std::string>("out_path");
 
  target_sp_hits_event_passname_ =
      ps.get<std::string>("target_sp_hits_event_passname");
  target_sp_passname_ = ps.get<std::string>("target_sp_passname");
 
  ecal_sp_hits_events_passname_ =
      ps.get<std::string>("ecal_sp_hits_events_passname");
  ecal_sp_passname_ = ps.get<std::string>("ecal_sp_passname");
 
  ecal_trig_sums_event_passname_ =
      ps.get<std::string>("ecal_trig_sums_event_passname");
  ecal_trig_sums_passname_ = ps.get<std::string>("ecal_trig_sums_passname");
 
  trig_electrons_event_passname_ =
      ps.get<std::string>("trig_electrons_event_passname");
  trig_electrons_passname_ = ps.get<std::string>("trig_electrons_passname");
 
  hcal_trig_quads_events_passname_ =
      ps.get<std::string>("hcal_trig_quads_events_passname");
  hcal_trig_quads_passname_ = ps.get<std::string>("hcal_trig_quads_passname");
}

References framework::config::Parameters::get().

onProcessEnd()

void trigger::NtupleWriter::onProcessEnd ( )

virtual

Callback for the EventProcessor to take any necessary action when the processing of events finishes, such as calculating job-summary quantities.

Reimplemented from framework::EventProcessor.

Definition at line 312 of file NtupleWriter.cxx.

                                {
  out_file_->Write();
  out_file_->Close();
}

onProcessStart()

void trigger::NtupleWriter::onProcessStart ( )

virtual

Callback for the EventProcessor to take any necessary action when the processing of events starts, such as creating histograms.

Reimplemented from framework::EventProcessor.

Definition at line 249 of file NtupleWriter.cxx.

                                  {
  // auto hdir = getHistoDirectory();
  out_file_ = new TFile(out_path_.c_str(), "recreate");
  out_file_->SetCompressionSettings(209);
  // 100*alg+level
  // 2=LZMA, 9 = max compression
  framework::NtupleManager& n{framework::NtupleManager::getInstance()};
  n.create(tag_);
 
  if (write_ele_) {
    std::string coll = "Electron";
    n.addVar<int>(tag_, "n" + coll);
    n.addVar<int>(tag_, "maxE");
    n.addVar<int>(tag_, "maxPt");
    n.addVar<vector<float>>(tag_, coll + "_e");
    n.addVar<vector<float>>(tag_, coll + "_eClus");
    n.addVar<vector<float>>(tag_, coll + "_zClus");
    n.addVar<vector<float>>(tag_, coll + "_px");
    n.addVar<vector<float>>(tag_, coll + "_py");
    n.addVar<vector<float>>(tag_, coll + "_pz");
    n.addVar<vector<float>>(tag_, coll + "_dx");
    n.addVar<vector<float>>(tag_, coll + "_dy");
    n.addVar<vector<float>>(tag_, coll + "_x");  // at target
    n.addVar<vector<float>>(tag_, coll + "_y");
    n.addVar<vector<int>>(tag_, coll + "_tp");
    n.addVar<vector<int>>(tag_, coll + "_depth");
  }
  if (write_truth_) {
    n.addVar<float>(tag_, "Truth_x");
    n.addVar<float>(tag_, "Truth_y");
    n.addVar<float>(tag_, "Truth_px");
    n.addVar<float>(tag_, "Truth_py");
    n.addVar<float>(tag_, "Truth_pz");
    n.addVar<float>(tag_, "Truth_e");
    n.addVar<int>(tag_, "Truth_pdgId");
    n.addVar<float>(tag_, "TruthEcal_x");
    n.addVar<float>(tag_, "TruthEcal_y");
    n.addVar<float>(tag_, "TruthEcal_px");
    n.addVar<float>(tag_, "TruthEcal_py");
    n.addVar<float>(tag_, "TruthEcal_pz");
    n.addVar<float>(tag_, "TruthEcal_e");
    n.addVar<int>(tag_, "TruthEcal_pdgId");
  }
  if (write_ecal_trig_mi_ps_) {
    n.addVar<std::vector<int>>(tag_, "Ecal_mip_length");
    n.addVar<std::vector<int>>(tag_, "Ecal_mip_nHoles");
    n.addVar<std::vector<float>>(tag_, "Ecal_mip_isolationEnergy");
  }
  if (write_hcal_trig_mi_ps_) {
    n.addVar<std::vector<int>>(tag_, "Hcal_mip_length");
    n.addVar<std::vector<int>>(tag_, "Hcal_mip_nHoles");
    // n.addVar<std::vector<float>>(tag_, "Hcal_mip_isolationEnergy");
  }
  if (write_ecal_sums_) {
    n.addVar<vector<float>>(tag_, "Ecal_e_afterLayer");
    n.addVar<int>(tag_, "Ecal_e_nLayer");
  };
  if (write_hcal_sums_) {
    n.addVar<vector<float>>(tag_, "Hcal_e_afterLayer");
    n.addVar<int>(tag_, "Hcal_e_nLayer");
    n.addVar<float>(tag_, "SideHcal_e");
  };
}

References framework::NtupleManager::create(), and framework::NtupleManager::getInstance().

produce()

void trigger::NtupleWriter::produce ( framework::Event & event )

virtual

Process the event and put new data products into it.

Parameters

event

The Event to process.

Implements framework::Producer.

Definition at line 42 of file NtupleWriter.cxx.

                                                {
  framework::NtupleManager& n{framework::NtupleManager::getInstance()};
 
  std::string in_tag;
  in_tag = "TargetScoringPlaneHits";
  if (write_truth_ && event.exists(in_tag, target_sp_hits_event_passname_)) {
    const std::vector<ldmx::SimTrackerHit> hits =
        event.getCollection<ldmx::SimTrackerHit>(in_tag, target_sp_passname_);
 
    ldmx::SimTrackerHit h, h_max_ele;  // the desired truth hits
    for (const auto& hit : hits) {
      auto xyz = hit.getPosition();
      if (xyz[2] > 0 && xyz[2] < 1) {
        if (hit.getTrackID() == 1) h = hit;
        if (hit.getPdgID() == 11 && (hit.getEnergy() > h_max_ele.getEnergy()))
          h_max_ele = hit;
      } else {
        continue;  // select one sp
      }
    }
    if (h.getPdgID() == 0)
      h = h_max_ele;  // save max energy in case track1 isn't found (A')
    std::string coll = "Truth";
    n.setVar(coll + "_e", prec(h.getEnergy()));
    n.setVar(coll + "_x", prec(h.getPosition()[0]));
    n.setVar(coll + "_y", prec(h.getPosition()[1]));
    n.setVar(coll + "_px", prec(h.getMomentum()[0]));
    n.setVar(coll + "_py", prec(h.getMomentum()[1]));
    n.setVar(coll + "_pz", prec(h.getMomentum()[2]));
    n.setVar(coll + "_pdgId", h.getPdgID());
  }
  in_tag = "EcalScoringPlaneHits";
  if (write_truth_ && event.exists(in_tag, ecal_sp_hits_events_passname_)) {
    const std::vector<ldmx::SimTrackerHit> hits =
        event.getCollection<ldmx::SimTrackerHit>(in_tag, ecal_sp_passname_);
    ldmx::SimTrackerHit h, h_max_ele;  // the desired truth hits
    for (const auto& hit : hits) {
      auto xyz = hit.getPosition();
      if (xyz[2] > 239.99 && xyz[2] < 240.01) {
        if (hit.getTrackID() == 1) h = hit;
        if (hit.getPdgID() == 11 && (hit.getEnergy() > h_max_ele.getEnergy()))
          h_max_ele = hit;
      } else {
        continue;  // select one sp
      }
    }
    if (h.getPdgID() == 0)
      h = h_max_ele;  // save max energy in case track1 isn't found (A')
    std::string coll = "TruthEcal";
    n.setVar(coll + "_e", prec(h.getEnergy()));
    n.setVar(coll + "_x", prec(h.getPosition()[0]));
    n.setVar(coll + "_y", prec(h.getPosition()[1]));
    n.setVar(coll + "_px", prec(h.getMomentum()[0]));
    n.setVar(coll + "_py", prec(h.getMomentum()[1]));
    n.setVar(coll + "_pz", prec(h.getMomentum()[2]));
    n.setVar(coll + "_pdgId", h.getPdgID());
  }
 
  in_tag = "ecalTrigSums";
  if (write_ecal_sums_ &&
      event.exists(in_tag, ecal_trig_sums_event_passname_)) {
    const auto sums =
        event.getCollection<TrigEnergySum>(in_tag, ecal_trig_sums_passname_);
    // const int nEcalLayers = 34;
    vector<float> energy_after_layer;  // (nEcalLayers, 0.);
    for (const auto& sum : sums) {
      if (!(sum.energy() > 0)) continue;
      if (sum.layer() >= energy_after_layer.size())
        energy_after_layer.resize(sum.layer() + 1);
      for (int i = 0; i <= sum.layer(); i++) {
        energy_after_layer[i] += sum.energy();
      }
    }
    n.setVar("Ecal_e_afterLayer", energy_after_layer);
    n.setVar("Ecal_e_nLayer", int(energy_after_layer.size()));
  }
  in_tag = "hcalTrigQuadsBackLayerSums";
  if (write_hcal_sums_ &&
      event.exists(in_tag, hcal_trig_quads_events_passname_)) {
    const auto sums =
        event.getCollection<TrigEnergySum>(in_tag, hcal_trig_quads_passname_);
    vector<float> energy_after_layer;
    for (const auto& sum : sums) {
      if (!(sum.hwEnergy() > 0)) continue;
      if (sum.layer() >= energy_after_layer.size())
        energy_after_layer.resize(sum.layer() + 1);
      for (int i = 0; i <= sum.layer(); i++) {
        energy_after_layer[i] += sum.hwEnergy();
      }
    }
    n.setVar("Hcal_e_afterLayer", energy_after_layer);
    n.setVar("Hcal_e_nLayer", int(energy_after_layer.size()));
  }
 
  in_tag = "hcalTrigQuadsSideLayerSums";
  if (write_hcal_sums_ &&
      event.exists(in_tag, hcal_trig_quads_events_passname_)) {
    const auto sums =
        event.getCollection<TrigEnergySum>(in_tag, hcal_trig_quads_passname_);
    float energy_in_side_hcal = 0.f;
    for (const auto& sum : sums) {
      if (!(sum.hwEnergy() > 0)) continue;
      for (int i = 0; i <= sum.layer(); i++) {
        energy_in_side_hcal += sum.hwEnergy();
      }
    }
    n.setVar("SideHcal_e", energy_in_side_hcal);
  }
 
  in_tag = "ecalTrigMIPs";
  if (write_ecal_trig_mi_ps_ &&
      event.exists(in_tag, ecal_trig_sums_event_passname_)) {
    const auto mips =
        event.getCollection<TrigMip>(in_tag, ecal_trig_sums_passname_);
    std::vector<int> lengths;
    std::vector<int> n_holes;
    std::vector<float> iso_energies;
    for (const auto& mip : mips) {
      lengths.push_back(mip.length());
      n_holes.push_back(mip.nHoles());
      iso_energies.push_back(mip.sumEinIsolationRegion());
    }
    n.setVar("Ecal_mip_length", lengths);
    n.setVar("Ecal_mip_nHoles", n_holes);
    n.setVar("Ecal_mip_isolationEnergy", iso_energies);
  }
 
  in_tag = "hcalTrigMIPs";
  if (write_hcal_trig_mi_ps_ &&
      event.exists(in_tag, hcal_trig_quads_events_passname_)) {
    const auto mips =
        event.getCollection<TrigMip>(in_tag, hcal_trig_quads_passname_);
    std::vector<int> lengths;
    std::vector<int> n_holes;
    // std::vector<float> isoEnergies;
    for (const auto& mip : mips) {
      lengths.push_back(mip.length());
      n_holes.push_back(mip.nHoles());
      // isoEnergies.push_back(mip.SumEinIsolationRegion());
    }
    n.setVar("Hcal_mip_length", lengths);
    n.setVar("Hcal_mip_nHoles", n_holes);
    // n.setVar("Hcal_mip_isolationEnergy", isoEnergies);
  }
 
  in_tag = "trigElectrons";
  if (write_ele_ && event.exists(in_tag, trig_electrons_event_passname_)) {
    const auto eles =
        event.getCollection<TrigParticle>(in_tag, trig_electrons_passname_);
    const int n_ele = eles.size();
    int max_e = -1;
    float max_e_val = 0;
    int max_pt = -1;
    float max_pt_val = 0;
    vector<float> v_e(n_ele);
    vector<float> v_e_c(n_ele);
    vector<float> v_z_c(n_ele);
    vector<float> v_px(n_ele);
    vector<float> v_py(n_ele);
    vector<float> v_pz(n_ele);
    vector<float> v_dx(n_ele);
    vector<float> v_dy(n_ele);
    vector<float> v_x(n_ele);
    vector<float> v_y(n_ele);
    vector<int> v_tp(n_ele);
    vector<int> v_depth(n_ele);
    for (unsigned int i = 0; i < n_ele; i++) {
      if (eles[i].energy() > max_e_val) {
        max_e_val = eles[i].energy();
        max_e = i;
      }
      if (eles[i].pt() > max_pt_val) {
        max_pt_val = eles[i].pt();
        max_pt = i;
      }
      v_e[i] = prec(eles[i].energy());
      v_e_c[i] = prec(eles[i].getClusEnergy());
      v_z_c[i] = prec(eles[i].endz());
      v_px[i] = prec(eles[i].px());
      v_py[i] = prec(eles[i].py());
      v_pz[i] = prec(eles[i].pz());
      v_dx[i] = prec(eles[i].endx() - eles[i].vx());
      v_dy[i] = prec(eles[i].endy() - eles[i].vy());
      v_x[i] = prec(eles[i].vx());
      v_y[i] = prec(eles[i].vy());
      v_tp[i] = prec(eles[i].getClusTP());
      v_depth[i] = prec(eles[i].getClusDepth());
    }
    std::string coll = "Electron";
    n.setVar("n" + coll, n_ele);
    n.setVar("maxE", max_e);
    n.setVar("maxPt", max_pt);
    n.setVar(coll + "_e", v_e);
    n.setVar(coll + "_eClus", v_e_c);
    n.setVar(coll + "_zClus", v_z_c);
    n.setVar(coll + "_px", v_px);
    n.setVar(coll + "_py", v_py);
    n.setVar(coll + "_pz", v_pz);
    n.setVar(coll + "_dx", v_dx);
    n.setVar(coll + "_dy", v_dy);
    n.setVar(coll + "_x", v_x);
    n.setVar(coll + "_y", v_y);
    n.setVar(coll + "_tp", v_tp);
    n.setVar(coll + "_depth", v_depth);
  }
}

References framework::Event::exists(), and framework::NtupleManager::getInstance().

Member Data Documentation

ecal_sp_hits_events_passname_

std::string trigger::NtupleWriter::ecal_sp_hits_events_passname_

private

Definition at line 31 of file NtupleWriter.h.

ecal_sp_passname_

std::string trigger::NtupleWriter::ecal_sp_passname_

private

Definition at line 32 of file NtupleWriter.h.

ecal_trig_sums_event_passname_

std::string trigger::NtupleWriter::ecal_trig_sums_event_passname_

private

Definition at line 34 of file NtupleWriter.h.

ecal_trig_sums_passname_

std::string trigger::NtupleWriter::ecal_trig_sums_passname_

private

Definition at line 35 of file NtupleWriter.h.

hcal_trig_quads_events_passname_

std::string trigger::NtupleWriter::hcal_trig_quads_events_passname_

private

Definition at line 37 of file NtupleWriter.h.

hcal_trig_quads_passname_

std::string trigger::NtupleWriter::hcal_trig_quads_passname_

private

Definition at line 38 of file NtupleWriter.h.

out_file_

TFile* trigger::NtupleWriter::out_file_ {nullptr}

private

Definition at line 24 of file NtupleWriter.h.

{nullptr};

out_path_

std::string trigger::NtupleWriter::out_path_ {"./ntuple.root"}

private

Definition at line 26 of file NtupleWriter.h.

{"./ntuple.root"};

tag_

std::string trigger::NtupleWriter::tag_ {"Events"}

private

Definition at line 25 of file NtupleWriter.h.

{"Events"};

target_sp_hits_event_passname_

std::string trigger::NtupleWriter::target_sp_hits_event_passname_

private

Definition at line 28 of file NtupleWriter.h.

target_sp_passname_

std::string trigger::NtupleWriter::target_sp_passname_

private

Definition at line 29 of file NtupleWriter.h.

trig_electrons_event_passname_

std::string trigger::NtupleWriter::trig_electrons_event_passname_

private

Definition at line 40 of file NtupleWriter.h.

trig_electrons_passname_

std::string trigger::NtupleWriter::trig_electrons_passname_

private

Definition at line 41 of file NtupleWriter.h.

write_ecal_sums_

bool trigger::NtupleWriter::write_ecal_sums_ {true}

private

Definition at line 45 of file NtupleWriter.h.

{true};

write_ecal_trig_mi_ps_

bool trigger::NtupleWriter::write_ecal_trig_mi_ps_ {true}

private

Definition at line 47 of file NtupleWriter.h.

{true};

write_ele_

bool trigger::NtupleWriter::write_ele_ {true}

private

Definition at line 44 of file NtupleWriter.h.

{true};

write_hcal_sums_

bool trigger::NtupleWriter::write_hcal_sums_ {true}

private

Definition at line 46 of file NtupleWriter.h.

{true};

write_hcal_trig_mi_ps_

bool trigger::NtupleWriter::write_hcal_trig_mi_ps_ {true}

private

Definition at line 48 of file NtupleWriter.h.

{true};

write_truth_

bool trigger::NtupleWriter::write_truth_ {true}

private

Definition at line 43 of file NtupleWriter.h.

{true};

---

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

• Trigger/Algo/include/Trigger/NtupleWriter.h
• Trigger/Algo/src/Trigger/NtupleWriter.cxx