TrackJetQAMaker/src/TrksInJetQATrkManager.h

0001 // ----------------------------------------------------------------------------
0002 // 'TrksInJetQATrkManager.h'
0003 // Derek Anderson
0004 // 03.25.2024
0005 //
0006 // A submodule for the TrksInJetQA module to generate
0007 // QA plots for tracks
0008 // ----------------------------------------------------------------------------
0009
0010 #ifndef TRKSINJETQATRKMANAGER_H
0011 #define TRKSINJETQATRKMANAGER_H
0012
0013 // c++ utilities
0014 #include <limits>
0015 #include <vector>
0016 #include <utility>
0017 // root libraries
0018 #include <TH1.h>
0019 #include <TH2.h>
0020 // tracking libraries
0021 #include <trackbase_historic/SvtxTrack.h>
0022 // submodule definitions
0023 #include "TrksInJetQABaseManager.h"
0024
0025
0026
0027 // TrksInJetQATrkManager definition -------------------------------------------
0028
0029 class TrksInJetQATrkManager : public TrksInJetQABaseManager {
0030
0031   public:
0032
0033     // histogram accessors
0034     //   - TODO split tracks into seed types
0035     enum Type {All};
0036     enum H1D  {Eta, Phi, Pt, Qual};
0037     enum H2D  {EtaVsPhi, PtVsQual};
0038
0039     // histogram content
0040     struct TrackQAContent {
0041       double eta  = std::numeric_limits<double>::max();
0042       double phi  = std::numeric_limits<double>::max();
0043       double pt   = std::numeric_limits<double>::max();
0044       double qual = std::numeric_limits<double>::max();
0045     };
0046
0047     // ctor/dtor
0048     using TrksInJetQABaseManager::TrksInJetQABaseManager;
0049     ~TrksInJetQATrkManager() {};
0050
0051     // public methods
0052     void GetInfo(SvtxTrack* track);
0053
0054   private:
0055
0056     // private methods
0057     void FillHistograms(const int type, TrackQAContent& content);
0058
0059     // inherited private methods
0060     void DefineHistograms() override;
0061
0062 };  // end TrksInJetQATrkManager
0063
0064 #endif
0065
0066 // end ------------------------------------------------------------------------
0067