recon::TrackDeDxMassEstimator Class Reference

Public Member Functions
	TrackDeDxMassEstimator (const std::string &name, framework::Process &process)
virtual void	configure (framework::config::Parameters &ps) override
	Callback for the EventProcessor to configure itself from the given set of parameters.
virtual void	produce (framework::Event &event) override
	Process the event and put new data products into it.
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.
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.
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
float	fit_res_c_ {0.}
float	fit_res_k_ {-9999.}
std::string	track_collection_
std::string	out_collection_
std::string	simhit_collection_
std::string	input_pass_name_

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

Definition at line 21 of file TrackDeDxMassEstimator.h.

Constructor & Destructor Documentation

TrackDeDxMassEstimator()

recon::TrackDeDxMassEstimator::TrackDeDxMassEstimator ( const std::string & name, framework::Process & process )

inline

Definition at line 23 of file TrackDeDxMassEstimator.h.

      : framework::Producer(name, process) {}

Member Function Documentation

configure()

void recon::TrackDeDxMassEstimator::configure ( framework::config::Parameters & parameters )

overridevirtual

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 16 of file TrackDeDxMassEstimator.cxx.

                                                                    {
  fit_res_c_ = ps.get<double>("fit_res_c");
  fit_res_k_ = ps.get<double>("fit_res_k");
  input_pass_name_ = ps.get<std::string>("input_pass_name");
  track_collection_ = ps.get<std::string>("track_collection");
  out_collection_ = ps.get<std::string>("out_collection");
 
  ldmx_log(info) << "Track Collection used for TrackDeDxMassEstimator "
                 << track_collection_;
}

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

produce()

void recon::TrackDeDxMassEstimator::produce ( framework::Event & event )

overridevirtual

Process the event and put new data products into it.

Parameters

event

The Event to process.

Implements framework::Producer.

Definition at line 27 of file TrackDeDxMassEstimator.cxx.

                                                          {
  std::vector<ldmx::TrackDeDxMassEstimate> mass_estimates;
 
  if (!event.exists(track_collection_, input_pass_name_)) {
    ldmx_log(error) << "Track collection " << track_collection_ << "_"
                    << input_pass_name_ << " not in event, exiting...";
    event.add(out_collection_, mass_estimates);
    return;
  }
  const std::vector<ldmx::Track> tracks{
      event.getCollection<ldmx::Track>(track_collection_, input_pass_name_)};
 
  int track_type;
  std::string track_coll_str = track_collection_;
  std::transform(track_coll_str.begin(), track_coll_str.end(),
                 track_coll_str.begin(), ::tolower);
 
  bool is_truth = track_coll_str.find("truth") != std::string::npos;
 
  if (is_truth) {
    if (track_coll_str.find("tagger") != std::string::npos) {
      track_type = 1;
      simhit_collection_ = "TaggerSimHits";
    } else if (track_coll_str.find("recoil") != std::string::npos) {
      track_type = 2;
      simhit_collection_ = "RecoilSimHits";
    } else {
      track_type = 0;
      simhit_collection_ = "";
    }
  } else {
    // Reco tracks (e.g. RecoilTracks, RecoilTracksClean)
    track_type = 4;
    simhit_collection_ = "";
  }
 
  // Retrieve simhits only for truth tracks
  std::vector<ldmx::SimTrackerHit> simhits;
  if (is_truth) {
    if (!event.exists(simhit_collection_, input_pass_name_)) {
      ldmx_log(error) << " SimHit collection (" << simhit_collection_ << "_"
                      << input_pass_name_ << ") does not exists, exiting...";
      event.add(out_collection_, mass_estimates);
      return;
    }
    simhits = event.getCollection<ldmx::SimTrackerHit>(simhit_collection_,
                                                       input_pass_name_);
  }
 
  // Loop over the collection of tracks
  for (uint i = 0; i < tracks.size(); i++) {
    auto track = tracks.at(i);
    // If track momentum doen't exist, skip
    auto the_qop = track.getQoP();
    if (the_qop == 0) {
      ldmx_log(debug) << "Track " << i << "has zero q/p ";
      continue;
    }
 
    int pdg_id = track.getPdgID();
    float momentum = 1. / std::abs(the_qop) * 1000;  // unit: MeV
    ldmx_log(debug) << "Track " << i << " has momentum " << momentum;
 
    ldmx::TrackDeDxMassEstimate mass_est;
    float sum_dedx_inv2 = 0.;
    float dedx;
    int n_hits = 0;
 
    if (is_truth) {
      // Use simhits associated with the truth track
      for (auto hit : simhits) {
        if (hit.getTrackID() != track.getTrackID()) continue;
        if (hit.getEdep() >= 0 && hit.getPathLength() > 0) {
          dedx = hit.getEdep() / hit.getPathLength() * 10;  // unit: MeV/cm
          sum_dedx_inv2 += 1. / (dedx * dedx);
          n_hits++;
        }
      }
    } else {
      // Use dE/dx measurements stored on the reco track (in MeV/mm)
      for (auto dedx_meas : track.getDedxMeasurements()) {
        if (dedx_meas > 0) {
          dedx = dedx_meas * 10;  // convert MeV/mm to MeV/cm
          sum_dedx_inv2 += 1. / (dedx * dedx);
          n_hits++;
        }
      }
    }  // end of loop over measurements
 
    if (sum_dedx_inv2 == 0) {
      ldmx_log(debug) << "Track " << i << " has no dEdx measurements";
      continue;
    }
 
    // Ih = (1/N * sum_i^N(dE/dx_i)^-2)^-1/2
    float the_ih = 1. / sqrt(1. / n_hits * sum_dedx_inv2);
    // m^2 = (Ih - C) p^2 / K [MeV^2], for every track, negative when Ih < C
    float mass2 = (the_ih - fit_res_c_) * momentum * momentum / fit_res_k_;
 
    float mass = 0.;
    if (the_ih > fit_res_c_) {
      mass = momentum * sqrt((the_ih - fit_res_c_) / fit_res_k_);
    } else {
      ldmx_log(info) << "Track " << i << " has Ih " << the_ih
                     << " which is less than fit_res_C " << fit_res_c_;
      mass = -100.;
    }
 
    mass_est.setMomentum(momentum);
    mass_est.setNhits(n_hits);
    mass_est.setIh(the_ih);
    mass_est.setMass(mass);
    mass_est.setMassSquared(mass2);
    mass_est.setTrackIndex(i);
    mass_est.setTrackType(track_type);
    mass_est.setPdgId(pdg_id);
    mass_estimates.push_back(mass_est);
  }
 
  // Add the mass estimates to the event
  event.add(out_collection_, mass_estimates);
}

References framework::Event::exists(), ldmx::TrackDeDxMassEstimate::setIh(), ldmx::TrackDeDxMassEstimate::setMass(), ldmx::TrackDeDxMassEstimate::setMassSquared(), ldmx::TrackDeDxMassEstimate::setMomentum(), ldmx::TrackDeDxMassEstimate::setNhits(), ldmx::TrackDeDxMassEstimate::setPdgId(), ldmx::TrackDeDxMassEstimate::setTrackIndex(), and ldmx::TrackDeDxMassEstimate::setTrackType().

Member Data Documentation

fit_res_c_

float recon::TrackDeDxMassEstimator::fit_res_c_ {0.}

private

Definition at line 31 of file TrackDeDxMassEstimator.h.

{0.};

fit_res_k_

float recon::TrackDeDxMassEstimator::fit_res_k_ {-9999.}

private

Definition at line 32 of file TrackDeDxMassEstimator.h.

{-9999.};

input_pass_name_

std::string recon::TrackDeDxMassEstimator::input_pass_name_

private

Definition at line 44 of file TrackDeDxMassEstimator.h.

out_collection_

std::string recon::TrackDeDxMassEstimator::out_collection_

private

Definition at line 38 of file TrackDeDxMassEstimator.h.

simhit_collection_

std::string recon::TrackDeDxMassEstimator::simhit_collection_

private

Definition at line 41 of file TrackDeDxMassEstimator.h.

track_collection_

std::string recon::TrackDeDxMassEstimator::track_collection_

private

Definition at line 35 of file TrackDeDxMassEstimator.h.

---

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

• Recon/include/Recon/TrackDeDxMassEstimator.h
• Recon/src/Recon/TrackDeDxMassEstimator.cxx