FWTTreeCache.h

Go to the documentation of this file.

#ifndef Fireworks_Core_FWTTreeCache_h
#define Fireworks_Core_FWTTreeCache_h

#include "TTreeCache.h"

#include <set>
#include <string>

class FWTTreeCache : public TTreeCache
{
   std::set<std::string> m_branch_set;
   bool                  m_silent_low_level = false;

   static int            s_default_size;
   static bool           s_logging;
   static bool           s_prefetching;

protected:
   bool is_branch_in_cache(const char* name) { return (m_branch_set.find(name) != m_branch_set.end()); }

   bool start_learning() { if (fIsLearning) return true; fIsLearning = true; return false; }
   void stop_learning()  { fIsLearning = false; }

   struct LearnGuard
   {
      FWTTreeCache *f_tc;
      bool          f_was_learning;
      bool          f_was_silent_ll;

      LearnGuard(FWTTreeCache *tc) : f_tc(tc)
      {
         f_was_learning = f_tc->start_learning();
         f_was_silent_ll = f_tc->m_silent_low_level;
         f_tc->m_silent_low_level = true;
      }
      ~LearnGuard()
      {
         if ( ! f_was_learning) f_tc->stop_learning();
         f_tc->m_silent_low_level = f_was_silent_ll;
      }
   };
   
public:
   FWTTreeCache();
   FWTTreeCache(TTree *tree, Int_t buffersize=0);
   ~FWTTreeCache() override;

   static void LoggingOn();
   static void LoggingOff();
   static bool IsLogging();
   static void PrefetchingOn();
   static void PrefetchingOff();
   static bool IsPrefetching();
   static void SetDefaultCacheSize(int def_size);
   static int  GetDefaultCacheSize();

   Int_t AddBranchTopLevel (const char* bname);
   Int_t DropBranchTopLevel(const char* bname);

   void  BranchAccessCallIn(const TBranch *b);

   // virtuals from TTreeCache, just wrappers for info printouts
   Int_t AddBranch(TBranch *b, Bool_t subbranches = kFALSE) override;
   Int_t AddBranch(const char *branch, Bool_t subbranches = kFALSE) override;
   Int_t DropBranch(TBranch *b, Bool_t subbranches = kFALSE) override;
   Int_t DropBranch(const char *branch, Bool_t subbranches = kFALSE) override;
};

#endif