class to extract results from qualityEstimation More...

#include <EventInfoExtractor.h>

Inheritance diagram for EventInfoExtractor:

Public Member Functions
	EventInfoExtractor (std::vector< Named< float * > > &variableSet)
	Define names of variables that get extracted.

void	extractVariables (const StoreArray< RecoTrack > &recoTracks, const RecoTrack &thisRecoTrack)
	extract the actual variables and write into a variable set

Protected Member Functions
void	addVariable (const std::string &identifier, std::vector< Named< float * > > &variables)
	add a variable to the variable set

Protected Attributes
std::unordered_map< std::string, float >	m_variables
	unordered_map to associate float value with a string name

Detailed Description

class to extract results from qualityEstimation

Definition at line 24 of file EventInfoExtractor.h.

Constructor & Destructor Documentation

◆ EventInfoExtractor()

EventInfoExtractor ( std::vector< Named< float * > > & variableSet )

inlineexplicit

Define names of variables that get extracted.

Definition at line 28 of file EventInfoExtractor.h.

                                                                      :
      VariableExtractor()
    {
      addVariable("__experiment__", variableSet);
      addVariable("__run__", variableSet);
      addVariable("__event__", variableSet);
      addVariable("N_RecoTracks", variableSet);
      addVariable("N_PXDRecoTracks", variableSet);
      addVariable("N_SVDRecoTracks", variableSet);
      addVariable("N_CDCRecoTracks", variableSet);
      addVariable("N_diff_PXD_SVD_RecoTracks", variableSet);
      addVariable("N_diff_SVD_CDC_RecoTracks", variableSet);
      addVariable("RTs_Min_Mom_diff_Mag", variableSet);
      addVariable("RTs_Min_Mom_diff_Pt", variableSet);
      addVariable("RTs_Min_Pos_diff_Theta", variableSet);
      addVariable("RTs_Min_Pos_diff_Phi", variableSet);
    }

Member Function Documentation

◆ addVariable()

void addVariable	(	const std::string &	identifier,
		std::vector< Named< float * > > &	variables
	)

inlineprotectedinherited

add a variable to the variable set

Definition at line 27 of file VariableExtractor.h.

    {
      //todo: verify if it is faster to check explicitly or not
      auto value = m_variables.emplace(identifier, NAN).first;
      variables.emplace_back(identifier, &(value->second));
    }

◆ extractVariables()

void extractVariables	(	const StoreArray< RecoTrack > &	recoTracks,
		const RecoTrack &	thisRecoTrack
	)

inline

extract the actual variables and write into a variable set

Definition at line 47 of file EventInfoExtractor.h.

    {
      m_variables.at("N_RecoTracks") = recoTracks.getEntries();
 
      int n_pxdRecoTracks = 0;
      int n_svdRecoTracks = 0;
      int n_cdcRecoTracks = 0;
      float min_mom_diff_mag = std::numeric_limits<float>::max();
      float min_mom_diff_Pt = std::numeric_limits<float>::max();
      float min_pos_diff_Theta = std::numeric_limits<float>::max();
      float min_pos_diff_Phi = std::numeric_limits<float>::max();
 
      for (const RecoTrack& recoTrack : recoTracks) {
        if (recoTrack.hasPXDHits())
          n_pxdRecoTracks++;
        if (recoTrack.hasSVDHits())
          n_svdRecoTracks++;
        if (recoTrack.hasCDCHits())
          n_cdcRecoTracks++;
 
        if (recoTrack.getArrayIndex() == thisRecoTrack.getArrayIndex())
          continue;
 
        float mom_diff_mag = fabs(recoTrack.getMomentumSeed().R() - thisRecoTrack.getMomentumSeed().R());
        if (mom_diff_mag < min_mom_diff_mag) {
          min_mom_diff_mag = mom_diff_mag;
        }
        float mom_diff_Pt = fabs(recoTrack.getMomentumSeed().Rho() - thisRecoTrack.getMomentumSeed().Rho());
        if (mom_diff_Pt < min_mom_diff_Pt) {
          min_mom_diff_Pt = mom_diff_Pt;
        }
        float pos_diff_Theta = fabs(recoTrack.getPositionSeed().Theta() - thisRecoTrack.getPositionSeed().Theta());
        if (pos_diff_Theta < min_pos_diff_Theta) {
          min_pos_diff_Theta = pos_diff_Theta;
        }
        float pos_diff_Phi = fabs(recoTrack.getPositionSeed().Phi() - thisRecoTrack.getPositionSeed().Phi());
        if (pos_diff_Phi < min_pos_diff_Phi) {
          min_pos_diff_Phi = pos_diff_Phi;
        }
      }
 
      StoreObjPtr<EventMetaData> m_eventMetaData;
 
      m_variables.at("__experiment__") = m_eventMetaData->getExperiment();
      m_variables.at("__run__") = m_eventMetaData->getRun();
      m_variables.at("__event__") = m_eventMetaData->getEvent();
      m_variables.at("N_PXDRecoTracks") = n_pxdRecoTracks;
      m_variables.at("N_SVDRecoTracks") = n_svdRecoTracks;
      m_variables.at("N_CDCRecoTracks") = n_cdcRecoTracks;
      m_variables.at("N_diff_PXD_SVD_RecoTracks") = n_svdRecoTracks - n_pxdRecoTracks;
      m_variables.at("N_diff_SVD_CDC_RecoTracks") = n_cdcRecoTracks - n_svdRecoTracks;
      m_variables.at("RTs_Min_Mom_diff_Mag") = min_mom_diff_mag;
      m_variables.at("RTs_Min_Mom_diff_Pt") = min_mom_diff_Pt;
      m_variables.at("RTs_Min_Pos_diff_Theta") = min_pos_diff_Theta;
      m_variables.at("RTs_Min_Pos_diff_Phi") = min_pos_diff_Phi;
    }

Member Data Documentation

◆ m_variables

std::unordered_map<std::string, float> m_variables

protectedinherited

unordered_map to associate float value with a string name

Definition at line 35 of file VariableExtractor.h.

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

tracking/trackFitting/trackQualityEstimator/variableExtractors/include/EventInfoExtractor.h

Public Member Functions

Protected Member Functions

Protected Attributes