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)
	~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 >	truthMatchingTool = nullptr
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 52 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 6 of file TruthMatchingTool.cxx.

                {
  TruthInfo ti;
  ti.truthProb = 0.;
  ti.trackID = -1;
  ti.pdgID = 0;
 
  for (std::unordered_map<unsigned int, unsigned int>::const_iterator it =
           trk_trackIDs.begin();
       it != trk_trackIDs.end(); it++) {
    double currentTruthProb = (double)it->second / (double)n_meas;
    if (currentTruthProb > ti.truthProb) {
      ti.truthProb = currentTruthProb;
      ti.trackID = it->first;
    }
  }
 
  if (ti.trackID > 0) ti.pdgID = map_[ti.trackID].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 54 of file TruthMatchingTool.cxx.

                          {
  // Map holding all tracksIds and their frequency
  std::unordered_map<unsigned int, unsigned int> trk_trackIDs;
 
  for (auto measID : trk.getMeasurementsIdxs()) {
    auto meas = measurements_.at(measID);
    if (debug_) {
      std::cout << "Getting measurement at ID:" << measID << std::endl;
      std::cout << meas << std::endl;
      std::cout << meas.getTrackIds().size() << std::endl;
    }
 
    for (auto trkId : meas.getTrackIds()) {
      if (trk_trackIDs.find(trkId) != trk_trackIDs.end()) {
        trk_trackIDs[trkId]++;
      } else {
        trk_trackIDs[trkId] = 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_trackIDs.begin();
           it != trk_trackIDs.end(); it++) {
        std::cout << it->first << " " << it->second << std::endl;
      }
    }
  }  // loop on measurements
 
  return Evaluate(trk_trackIDs, trk.getMeasurementsIdxs().size());
 
}  // Match Track

TruthMatch() [2/2]

TruthMatchingTool::TruthInfo tracking::sim::TruthMatchingTool::TruthMatch

(

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

vmeas

)

Definition at line 29 of file TruthMatchingTool.cxx.

                                             {
  std::unordered_map<unsigned int, unsigned int> trk_trackIDs;
 
  for (auto meas : vmeas) {
    for (auto trkId : meas.getTrackIds()) {
      if (trk_trackIDs.find(trkId) != trk_trackIDs.end())
        trk_trackIDs[trkId]++;
      else
        trk_trackIDs[trkId] = 1;
    }
  }  // loop on measurements
 
  return Evaluate(trk_trackIDs, 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 57 of file TruthMatchingTool.h.

{false};

map_

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

private

Definition at line 55 of file TruthMatchingTool.h.

measurements_

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

private

Definition at line 56 of file TruthMatchingTool.h.

truthMatchingTool

std::shared_ptr<tracking::sim::TruthMatchingTool> tracking::sim::TruthMatchingTool::truthMatchingTool = nullptr

private

Definition at line 58 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