TrackDiff.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2022 CERN for the benefit of the ATLAS collaboration
*/

// TrackDiff.h
//   Header file for TrackDiff
// (c) ATLAS Detector software
// Sebastian.Fleischmann@cern.ch
 
#ifndef TRK_TRACKDIFF_H
#define TRK_TRACKDIFF_H
 
#include "AthenaBaseComps/AthAlgTool.h"
#include "TrkTrack/TrackStateOnSurface.h"
#include "TrkValInterfaces/ITrackDiff.h"
#include "TrkEventPrimitives/TrackStateDefs.h"
#include "TrkValEvent/TrackStateData.h"
#include "AthContainers/DataVector.h"
#include "StoreGate/ReadHandleKey.h"
#include "xAODEventInfo/EventInfo.h"
 
class AtlasDetectorID;
class TTree;
 
namespace Trk {
 
 class RIO_OnTrack;
 class Track;
 class Surface;
 
class TrackDiff : virtual public Trk::ITrackDiff, public AthAlgTool {
public:
 
    // standard AlgToolmethods
    TrackDiff(const std::string&,const std::string&,const IInterface*);
    ~TrackDiff();
 
    // standard Athena methods
    StatusCode initialize();
    StatusCode finalize();
 

    virtual StatusCode diff (
        const Trk::Track& referenceTrack,
        const Trk::Track& comparedTrack);
 
 
private:
 
    static const int s_maxHits  = 300;   // maximal number of hits per track
 
    TTree* m_nt; 
 
    AtlasDetectorID* m_idHelper;        
 
    // jobOptions
    std::string m_ntupleFileName;       
    std::string m_ntupleDirName;        
    std::string m_ntupleTreeName;       
    //bool m_compareMeasurements;         //!< jobOption: compare track states of type "measurement"
    bool m_compareOutliers;             
    bool m_compareAll;                  
    bool m_writeNtuple;                 
    bool m_writeCompetingROT;           
 
    int m_eventNumber;
   
    int m_nRefStates[Trk::TrackState::NumberOfMeasurementTypes];
    int m_nCompStates[Trk::TrackState::NumberOfMeasurementTypes];
    int m_fakes[Trk::TrackState::NumberOfMeasurementTypes];
 
    int m_missed[Trk::TrackState::NumberOfMeasurementTypes];
    int m_wrongType[Trk::TrackState::NumberOfMeasurementTypes];
    int m_missingType[Trk::TrackStateOnSurface::NumberOfTrackStateOnSurfaceTypes];
    int m_fakeType[Trk::TrackStateOnSurface::NumberOfTrackStateOnSurfaceTypes];
    int m_refType[Trk::TrackStateOnSurface::NumberOfTrackStateOnSurfaceTypes];
    int m_compareType[Trk::TrackStateOnSurface::NumberOfTrackStateOnSurfaceTypes];
    int m_PRD_Mismatches[Trk::TrackState::NumberOfMeasurementTypes];
    int m_driftCircleSignFlips;
    int m_noDriftTime;
 
    int m_nDiffs;
    std::vector<int>*   m_detectorType;
    std::vector<int>*   m_isFake;   
    std::vector<int>*   m_isMissing;
    std::vector<int>*   m_isPRD_Mismatch;
    std::vector<int>*   m_isFlippedSign;
    std::vector<int>*   m_isNoDriftTime;
    std::vector<int>*   m_refIsMeasurement;
    std::vector<int>*   m_refIsOutlier;
    std::vector<int>*   m_compIsMeasurement;
    std::vector<int>*   m_compIsOutlier;
    std::vector<float>* m_maxAssignProb;
    std::vector<float>* m_sumAssignProb;
    std::vector<float>* m_surfX;
    std::vector<float>* m_surfY;
    std::vector<float>* m_surfZ;
 
    float m_trackEta;
    float m_trackPhi;
 
 
    int m_nRefStatesSum[Trk::TrackState::NumberOfMeasurementTypes];
    int m_nCompStatesSum[Trk::TrackState::NumberOfMeasurementTypes];
    int m_fakesSum[Trk::TrackState::NumberOfMeasurementTypes];
    int m_missedSum[Trk::TrackState::NumberOfMeasurementTypes];
    int m_wrongTypeSum[Trk::TrackState::NumberOfMeasurementTypes];
    int m_PRD_MismatchesSum[Trk::TrackState::NumberOfMeasurementTypes];
    int m_trackSum;
 
    SG::ReadHandleKey<xAOD::EventInfo>    m_evt  {this, "EvtInfo", "EventInfo", "EventInfo name"};

    DataVector< const Trk::TrackStateData >* extractDataFromTrack( const Trk::Track& ) const;

    void resetVariables();

    Trk::TrackState::MeasurementType detectorType(const Trk::RIO_OnTrack& ROT,
                                                   std::string &detTypeName ) const;

    bool diffStateInfo(const Trk::TrackStateData* refTrackStateData,
                       const Trk::TrackStateData* compareTrackStateData);
 
};
 
 
} // end of namespace
 
#endif // TRK_TRACKDIFF_H