PerfStats.h

Go to the documentation of this file.

// Dear emacs, this is -*- c++ -*-
/*
  Copyright (C) 2002-2021 CERN for the benefit of the ATLAS collaboration
*/
#ifndef XAODCORE_TOOLS_PERFSTATS_H
#define XAODCORE_TOOLS_PERFSTATS_H
 
// ROOT include(s):
#include <TVirtualPerfStats.h>
 
// Athena include(s).
#include "CxxUtils/checker_macros.h"
 
// System include(s).
#include <mutex>
 
// Forward declaration(s):
class TTree;
 
namespace xAOD {

   class PerfStats : public ::TVirtualPerfStats {
 
   public:
      ~PerfStats();

      static PerfStats& instance();

      void start( bool clear = true );
      void stop();


      virtual void SimpleEvent( EEventType type );
      virtual void PacketEvent( const char *slave, const char *slavename,
                                const char *filename,
                                ::Long64_t eventsprocessed,
                                ::Double_t latency, ::Double_t proctime,
                                ::Double_t cputime, ::Long64_t bytesRead );
      virtual void FileEvent( const char *slave, const char *slavename,
                              const char *nodename, const char *filename,
                              ::Bool_t isStart );
      virtual void FileOpenEvent( ::TFile *file, const char *filename,
                                  ::Double_t start );
      virtual void FileReadEvent( ::TFile *file, ::Int_t len,
                                  ::Double_t start );
      virtual void UnzipEvent( ::TObject *tree, ::Long64_t pos,
                               ::Double_t start, ::Int_t complen,
                               ::Int_t objlen );
      virtual void RateEvent( ::Double_t proctime, ::Double_t deltatime,
                              ::Long64_t eventsprocessed,
                              ::Long64_t bytesRead );
      virtual void SetBytesRead( ::Long64_t num );
      virtual ::Long64_t GetBytesRead() const;
      virtual void SetNumEvents( ::Long64_t num );
      virtual ::Long64_t GetNumEvents() const;

      virtual void PrintBasketInfo(Option_t *option = "") const;

      virtual void SetLoaded(TBranch *b, size_t basketNumber);
      virtual void SetLoaded(size_t bi, size_t basketNumber);

      virtual void SetLoadedMiss(TBranch *b, size_t basketNumber);
      virtual void SetLoadedMiss(size_t bi, size_t basketNumber);

      virtual void SetMissed(TBranch *b, size_t basketNumber);
      virtual void SetMissed(size_t bi, size_t basketNumber);

      virtual void SetUsed(TBranch *b, size_t basketNumber);
      virtual void SetUsed(size_t bi, size_t basketNumber);

      virtual void UpdateBranchIndices(TObjArray *branches);

 
   protected:
      PerfStats();
 
   private:
#if ROOT_VERSION_CODE >= ROOT_VERSION( 6, 23, 2 )
      virtual void SetFile( TFile* file );
#endif // ROOT version

      static PerfStats* s_instance ATLAS_THREAD_SAFE;

      static std::mutex s_mutex;
      typedef std::lock_guard<std::mutex> lock_t;

      ::TVirtualPerfStats* m_otherPerfStats;

      bool m_running;
      ::Double_t m_startTime;

      ::TTree* m_tree;
      ::TFile* m_file;

      bool m_treeWarningPrinted;
 
      ClassDef( xAOD::PerfStats, 0 )
 
   }; // class PerfStats
 
} // namespace xAOD
 
#endif // XAODCORE_TOOLS_PERFSTATS_H