tracking::sim::TruthMatchingTool Class Reference

Classes
struct	TruthInfo

Public Member Functions
	TruthMatchingTool (const std::map< int, ldmx::SimParticle > &particleMap, const std::vector< ldmx::Measurement > &measurements)
	Constructor.
void	setup (const std::map< int, ldmx::SimParticle > &particleMap, const std::vector< ldmx::Measurement > &measurements)
virtual	~TruthMatchingTool ()=default
	Destructor.
TruthInfo	truthMatch (const ldmx::Track &trk)
	Performs the truth matching by checking all the trackIDs associated to the measurements on track.
TruthInfo	evaluate (const std::unordered_map< unsigned int, unsigned int > &trk_trackIDs, int n_meas)
TruthInfo	truthMatch (const std::vector< ldmx::Measurement > &vmeas)
bool	configured ()

Private Attributes
std::map< int, ldmx::SimParticle >	map_
std::vector< ldmx::Measurement >	measurements_
bool	debug_ {false}
std::shared_ptr< tracking::sim::TruthMatchingTool >	truth_matching_tool_
bool	configured_ {false}

Detailed Description

Definition at line 11 of file TruthMatchingTool.h.

Constructor & Destructor Documentation

TruthMatchingTool()

tracking::sim::TruthMatchingTool::TruthMatchingTool ( const std::map< int, ldmx::SimParticle > & particleMap, const std::vector< ldmx::Measurement > & measurements )

inline

Constructor.

Parameters

particleMap	The map of all the simulated particles in the event.
measurements	All the measurements in the event.

Definition at line 28 of file TruthMatchingTool.h.

                                                                    {
    setup(particleMap, measurements);
  };

Member Function Documentation

configured()

bool tracking::sim::TruthMatchingTool::configured ( )

inline

Definition at line 51 of file TruthMatchingTool.h.

{ return configured_; }

evaluate()

TruthMatchingTool::TruthInfo tracking::sim::TruthMatchingTool::evaluate	(	const std::unordered_map< unsigned int, unsigned int > &	trk_trackIDs,
		int	n_meas )

Definition at line 8 of file TruthMatchingTool.cxx.

                {
  TruthInfo ti;
  ti.truth_prob_ = 0.;
  ti.track_id_ = -1;
  ti.pdg_id_ = 0;
 
  for (std::unordered_map<unsigned int, unsigned int>::const_iterator it =
           trk_trackIDs.begin();
       it != trk_trackIDs.end(); it++) {
    double current_truth_prob = (double)it->second / (double)n_meas;
    if (current_truth_prob > ti.truth_prob_) {
      ti.truth_prob_ = current_truth_prob;
      ti.track_id_ = it->first;
    }
  }
 
  if (ti.track_id_ > 0) ti.pdg_id_ = map_[ti.track_id_].getPdgID();
 
  return ti;
}

setup()

void tracking::sim::TruthMatchingTool::setup ( const std::map< int, ldmx::SimParticle > & particleMap, const std::vector< ldmx::Measurement > & measurements )

inline

Definition at line 33 of file TruthMatchingTool.h.

                                                             {
    map_ = particleMap;
    measurements_ = measurements;
    configured_ = true;
  }

truthMatch() [1/2]

TruthMatchingTool::TruthInfo tracking::sim::TruthMatchingTool::truthMatch

(

const ldmx::Track &

trk

)

Performs the truth matching by checking all the trackIDs associated to the measurements on track.

Parameters

trk	The track that needs to be truth matched

Returns

TruthInfo a struct containing the truth information regarding the track Each measurement weights the same.

Definition at line 56 of file TruthMatchingTool.cxx.

                          {
  // Map holding all tracksIds and their frequency
  std::unordered_map<unsigned int, unsigned int> trk_track_i_ds;
 
  for (auto meas_id : trk.getMeasurementsIdxs()) {
    auto meas = measurements_.at(meas_id);
    if (debug_) {
      std::cout << "Getting measurement at ID:" << meas_id << std::endl;
      std::cout << meas << std::endl;
      std::cout << meas.getTrackIds().size() << std::endl;
    }
 
    for (auto trk_id : meas.getTrackIds()) {
      if (trk_track_i_ds.find(trk_id) != trk_track_i_ds.end()) {
        trk_track_i_ds[trk_id]++;
      } else {
        trk_track_i_ds[trk_id] = 1;
      }
 
    }  // loop on measurements trackIDs
 
    if (debug_) {
      std::cout << "The trackIDs maps look like:" << std::endl;
      for (std::unordered_map<unsigned int, unsigned int>::iterator it =
               trk_track_i_ds.begin();
           it != trk_track_i_ds.end(); it++) {
        std::cout << it->first << " " << it->second << std::endl;
      }
    }
  }  // loop on measurements
 
  return evaluate(trk_track_i_ds, trk.getMeasurementsIdxs().size());
 
}  // Match Track

truthMatch() [2/2]

TruthMatchingTool::TruthInfo tracking::sim::TruthMatchingTool::truthMatch

(

const std::vector< ldmx::Measurement > &

vmeas

)

Definition at line 31 of file TruthMatchingTool.cxx.

                                             {
  std::unordered_map<unsigned int, unsigned int> trk_track_i_ds;
 
  for (auto meas : vmeas) {
    for (auto trk_id : meas.getTrackIds()) {
      if (trk_track_i_ds.find(trk_id) != trk_track_i_ds.end())
        trk_track_i_ds[trk_id]++;
      else
        trk_track_i_ds[trk_id] = 1;
    }
  }  // loop on measurements
 
  return evaluate(trk_track_i_ds, vmeas.size());
}

Member Data Documentation

configured_

bool tracking::sim::TruthMatchingTool::configured_ {false}

private

Definition at line 59 of file TruthMatchingTool.h.

{false};

debug_

bool tracking::sim::TruthMatchingTool::debug_ {false}

private

Definition at line 56 of file TruthMatchingTool.h.

{false};

map_

std::map<int, ldmx::SimParticle> tracking::sim::TruthMatchingTool::map_

private

Definition at line 54 of file TruthMatchingTool.h.

measurements_

std::vector<ldmx::Measurement> tracking::sim::TruthMatchingTool::measurements_

private

Definition at line 55 of file TruthMatchingTool.h.

truth_matching_tool_

std::shared_ptr<tracking::sim::TruthMatchingTool> tracking::sim::TruthMatchingTool::truth_matching_tool_

private

Initial value:

=
      nullptr

Definition at line 57 of file TruthMatchingTool.h.

---

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

• Tracking/include/Tracking/Reco/TruthMatchingTool.h
• Tracking/src/Tracking/Sim/TruthMatchingTool.cxx