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	beforeNewRun (ldmx::RunHeader &header)
	Handle allowing producers to modify run headers before the run begins.

Public Member Functions inherited from framework::EventProcessor
	EventProcessor (const std::string &name, Process &process)
	Class constructor.

virtual	~EventProcessor ()
	Class destructor.

virtual void	onNewRun (const ldmx::RunHeader &runHeader)
	Callback for the EventProcessor to take any necessary action when the run being processed changes.

virtual void	onFileOpen (EventFile &eventFile)
	Callback for the EventProcessor to take any necessary action when a new event input ROOT file is opened.

virtual void	onFileClose (EventFile &eventFile)
	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 *	outFile_ {nullptr}

std::string	tag_ {"Events"}

std::string	outPath_ {"./ntuple.root"}

bool	writeTruth_ {true}

bool	writeEle_ {true}

bool	writeEcalSums_ {true}

bool	writeHcalSums_ {true}

Additional Inherited Members
Static Public Member Functions inherited from framework::EventProcessor
static void	declare (const std::string &classname, int classtype, EventProcessorMaker *)
	Internal function which is part of the PluginFactory machinery.

Static Public Attributes inherited from framework::Producer
static const int	CLASSTYPE {1}
	Constant used to track EventProcessor types by the PluginFactory.

Protected Member Functions inherited from framework::EventProcessor
void	abortEvent ()
	Abort the event immediately.

Protected Attributes inherited from framework::EventProcessor
HistogramHelper	histograms_
	Interface class for making and filling histograms.

NtupleManager &	ntuple_ {NtupleManager::getInstance()}
	Manager for any ntuples.

logging::logger	theLog_
	The logger for this EventProcessor.

Detailed Description

Null algorithm test.

Definition at line 16 of file NtupleWriter.h.

Constructor & Destructor Documentation

◆ NtupleWriter()

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

Definition at line 8 of file NtupleWriter.cxx.

9 : Producer(name, process) {}

framework::Producer::Producer

Producer(const std::string &name, Process &process)

Class constructor.

Definition EventProcessor.cxx:68

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 className equal to the EventProcessor class name.

For an example, look at MyProcessor.

Parameters

parameters Parameters for configuration.

Reimplemented from framework::EventProcessor.

Definition at line 11 of file NtupleWriter.cxx.

                                                          {
  outPath_ = ps.getParameter<std::string>("outPath");
}

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

◆ 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 222 of file NtupleWriter.cxx.

                                {
  outFile_->Write();
  outFile_->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 170 of file NtupleWriter.cxx.

                                  {
  // auto hdir = getHistoDirectory();
  outFile_ = new TFile(outPath_.c_str(), "recreate");
  outFile_->SetCompressionSettings(209);
  // 100*alg+level
  // 2=LZMA, 9 = max compression
  framework::NtupleManager& n{framework::NtupleManager::getInstance()};
  n.create(tag_);
 
  if (writeEle_) {
    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 (writeTruth_) {
    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 (writeEcalSums_) {
    n.addVar<vector<float> >(tag_, "Ecal_e_afterLayer");
    n.addVar<int>(tag_, "Ecal_e_nLayer");
  };
  if (writeHcalSums_) {
    n.addVar<vector<float> >(tag_, "Hcal_e_afterLayer");
    n.addVar<int>(tag_, "Hcal_e_nLayer");
  };
}

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 20 of file NtupleWriter.cxx.

                                                {
  framework::NtupleManager& n{framework::NtupleManager::getInstance()};
 
  std::string inTag;
  inTag = "TargetScoringPlaneHits";
  if (writeTruth_ && event.exists(inTag)) {
    const std::vector<ldmx::SimTrackerHit> hits =
        event.getCollection<ldmx::SimTrackerHit>(inTag);
    ldmx::SimTrackerHit h, hMaxEle;  // 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() > hMaxEle.getEnergy()))
          hMaxEle = hit;
      } else {
        continue;  // select one sp
      }
    }
    if (h.getPdgID() == 0)
      h = hMaxEle;  // 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());
  }
  inTag = "EcalScoringPlaneHits";
  if (writeTruth_ && event.exists(inTag)) {
    const std::vector<ldmx::SimTrackerHit> hits =
        event.getCollection<ldmx::SimTrackerHit>(inTag);
    ldmx::SimTrackerHit h, hMaxEle;  // 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() > hMaxEle.getEnergy()))
          hMaxEle = hit;
      } else {
        continue;  // select one sp
      }
    }
    if (h.getPdgID() == 0)
      h = hMaxEle;  // 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());
  }
 
  inTag = "ecalTrigSums";
  if (writeEcalSums_ && event.exists(inTag)) {
    const auto sums = event.getCollection<TrigEnergySum>(inTag);
    // const int nEcalLayers = 34;
    vector<float> energyAfterLayer;  // (nEcalLayers, 0.);
    for (const auto& sum : sums) {
      if (!(sum.energy() > 0)) continue;
      if (sum.layer() >= energyAfterLayer.size())
        energyAfterLayer.resize(sum.layer() + 1);
      for (int i = 0; i <= sum.layer(); i++) {
        energyAfterLayer[i] += sum.energy();
      }
    }
    n.setVar("Ecal_e_afterLayer", energyAfterLayer);
    n.setVar("Ecal_e_nLayer", int(energyAfterLayer.size()));
  }
  inTag = "hcalTrigQuadsBackLayerSums";
  if (writeHcalSums_ && event.exists(inTag)) {
    const auto sums = event.getCollection<TrigEnergySum>(inTag);
    vector<float> energyAfterLayer;
    for (const auto& sum : sums) {
      if (!(sum.hwEnergy() > 0)) continue;
      if (sum.layer() >= energyAfterLayer.size())
        energyAfterLayer.resize(sum.layer() + 1);
      for (int i = 0; i <= sum.layer(); i++) {
        energyAfterLayer[i] += sum.hwEnergy();
      }
    }
    n.setVar("Hcal_e_afterLayer", energyAfterLayer);
    n.setVar("Hcal_e_nLayer", int(energyAfterLayer.size()));
  }
 
  inTag = "trigElectrons";
  if (writeEle_ && event.exists(inTag)) {
    const auto eles = event.getCollection<TrigParticle>(inTag);
    const int nEle = eles.size();
    int maxE = -1;
    float maxEVal = 0;
    int maxPt = -1;
    float maxPtVal = 0;
    vector<float> v_e(nEle);
    vector<float> v_eC(nEle);
    vector<float> v_zC(nEle);
    vector<float> v_px(nEle);
    vector<float> v_py(nEle);
    vector<float> v_pz(nEle);
    vector<float> v_dx(nEle);
    vector<float> v_dy(nEle);
    vector<float> v_x(nEle);
    vector<float> v_y(nEle);
    vector<int> v_tp(nEle);
    vector<int> v_depth(nEle);
    for (unsigned int i = 0; i < nEle; i++) {
      if (eles[i].energy() > maxEVal) {
        maxEVal = eles[i].energy();
        maxE = i;
      }
      if (eles[i].pt() > maxPtVal) {
        maxPtVal = eles[i].pt();
        maxPt = i;
      }
      v_e[i] = prec(eles[i].energy());
      v_eC[i] = prec(eles[i].getClusEnergy());
      v_zC[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, nEle);
    n.setVar("maxE", maxE);
    n.setVar("maxPt", maxPt);
    n.setVar(coll + "_e", v_e);
    n.setVar(coll + "_eClus", v_eC);
    n.setVar(coll + "_zClus", v_zC);
    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(), framework::NtupleManager::getInstance(), and ldmx::SimTrackerHit::getPosition().

Member Data Documentation

◆ outFile_

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

private

Definition at line 25 of file NtupleWriter.h.

25{nullptr};

◆ outPath_

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

private

Definition at line 27 of file NtupleWriter.h.

27{"./ntuple.root"};

◆ tag_

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

private

Definition at line 26 of file NtupleWriter.h.

26{"Events"};

◆ writeEcalSums_

bool trigger::NtupleWriter::writeEcalSums_ {true}

private

Definition at line 30 of file NtupleWriter.h.

30{true};

◆ writeEle_

bool trigger::NtupleWriter::writeEle_ {true}

private

Definition at line 29 of file NtupleWriter.h.

29{true};

◆ writeHcalSums_

bool trigger::NtupleWriter::writeHcalSums_ {true}

private

Definition at line 31 of file NtupleWriter.h.

31{true};

◆ writeTruth_

bool trigger::NtupleWriter::writeTruth_ {true}

private

Definition at line 28 of file NtupleWriter.h.

28{true};

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

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

Public Member Functions

Private Attributes

Additional Inherited Members

Detailed Description

Constructor & Destructor Documentation

◆ NtupleWriter()

Member Function Documentation

◆ configure()

◆ onProcessEnd()

◆ onProcessStart()

◆ produce()

Member Data Documentation

◆ outFile_

◆ outPath_

◆ tag_

◆ writeEcalSums_

◆ writeEle_

◆ writeHcalSums_

◆ writeTruth_