MuonInsideOutAnalysis.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2023 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef MUON_MUONINSIDEOUTANALYSIS_H
#define MUON_MUONINSIDEOUTANALYSIS_H
 
#include "MuonRecValidationNtuples/MuonInsideOutAnalysisPlots.h"
#include "MuonRecValidationNtuples/MuonInsideOutValidationNtuple.h"
#include "MuonRecHelperTools/TimePointBetaFitter.h"
#include <set>
 
class TTree;
class TDirectory;
 
namespace Muon {
 
  class MuonInsideOutAnalysis {
  public:
    enum MatchingStrategy {
      SameTrack = 0,
      NotSameTrack = 1,
      All = 2
    };

    MuonInsideOutAnalysis( TDirectory* dir );
    
    void analyse( TTree& tree );
    
  private:
    bool testStrategy( MuonInsideOutAnalysis::MatchingStrategy matchingStrategy, bool sameTrack ) const;
 
    unsigned int handleHits( MuonInsideOutValidationNtuple& ntuple, unsigned int trkIndex,
                             const std::vector<int>& indexVec, MuonInsideOutAnalysis::MatchingStrategy matchingStrategy,
                             HitPlots& plots );
    unsigned int handleHough( MuonInsideOutValidationNtuple& ntuple, unsigned int trkIndex,
                              const std::vector<int>& indexVec, MuonInsideOutAnalysis::MatchingStrategy matchingStrategy,
                              HoughPlots& plots );
    unsigned int handleSegments( MuonInsideOutValidationNtuple& ntuple, unsigned int trkIndex,
                                 const std::vector<int>& indexVec, int stage, MuonInsideOutAnalysis::MatchingStrategy matchingStrategy,
                                 SegmentPlots& segmentPlots );
 
    std::pair<unsigned int,unsigned int> handleCandidates( MuonInsideOutValidationNtuple& ntuple, unsigned int trkIndex,
                                                           const std::vector<int>& indexVec, 
                                                           StageSummaryPlots& plots, int stage );
 
    void handleTimes( MuonInsideOutValidationNtuple& ntuple, unsigned int trkIndex,
                      const std::vector<int>& indexVec, MuonInsideOutAnalysis::MatchingStrategy matchingStrategy, 
                      TrackPlots& trackPlots );
 
    TimePointBetaFitter::FitResult fitBeta( MuonInsideOutValidationNtuple& ntuple, const std::vector<int>& indexVec, const std::set<int>& type );
    
    MuonInsideOutAnalysisPlots m_plots;
  };
}
 
#endif