#include <FWTTreeCache.h>

Inheritance diagram for FWTTreeCache:

Classes
struct	LearnGuard

Public Member Functions
Int_t	AddBranch (TBranch *b, Bool_t subbranches=kFALSE) override

Int_t	AddBranch (const char *branch, Bool_t subbranches=kFALSE) override

Int_t	AddBranchTopLevel (const char *bname)

void	BranchAccessCallIn (const TBranch *b)

Int_t	DropBranch (TBranch *b, Bool_t subbranches=kFALSE) override

Int_t	DropBranch (const char *branch, Bool_t subbranches=kFALSE) override

Int_t	DropBranchTopLevel (const char *bname)

	FWTTreeCache ()

	FWTTreeCache (TTree *tree, Int_t buffersize=0)

	~FWTTreeCache () override

Static Public Member Functions
static int	GetDefaultCacheSize ()

static bool	IsLogging ()

static void	LoggingOff ()

static void	LoggingOn ()

static void	SetDefaultCacheSize (int def_size)

Protected Member Functions
bool	is_branch_in_cache (const char *name)

bool	start_learning ()

void	stop_learning ()

Private Attributes
std::set< std::string >	m_branch_set

bool	m_silent_low_level = false

Static Private Attributes
static int	s_default_size = 50 * 1024 * 1024

static bool	s_logging = false

Detailed Description

Definition at line 9 of file FWTTreeCache.h.

Constructor & Destructor Documentation

FWTTreeCache::FWTTreeCache ( )

Definition at line 9 of file FWTTreeCache.cc.

Referenced by FWTTreeCache::LearnGuard::~LearnGuard().

                            :
   TTreeCache()
 {}

FWTTreeCache::FWTTreeCache	(	TTree *	tree,
		Int_t	buffersize = `0`
	)

Definition at line 13 of file FWTTreeCache.cc.

                                                         :
   TTreeCache(tree, buffersize)
 {}

FWTTreeCache::~FWTTreeCache ( )

override

Definition at line 17 of file FWTTreeCache.cc.

References s_default_size, and s_logging.

Referenced by FWTTreeCache::LearnGuard::~LearnGuard().

18 {}

Member Function Documentation

Int_t FWTTreeCache::AddBranch	(	TBranch *	b,
		Bool_t	subbranches = `kFALSE`
	)

override

Definition at line 98 of file FWTTreeCache.cc.

References m_silent_low_level, and s_logging.

Referenced by AddBranchTopLevel(), and FWTTreeCache::LearnGuard::~LearnGuard().

 {
    if (s_logging && ! m_silent_low_level)
       printf("FWTTreeCache::AddBranch by ptr '%s', subbp=%d\n", b->GetName(), subbranches);
    return TTreeCache::AddBranch(b, subbranches);
 }

Int_t FWTTreeCache::AddBranch	(	const char *	branch,
		Bool_t	subbranches = `kFALSE`
	)

override

Definition at line 105 of file FWTTreeCache.cc.

References m_silent_low_level, and s_logging.

 {
    if (s_logging)
       printf("FWTTreeCache::AddBranch by name '%s', subbp=%d\n", branch, subbranches);
    if (strcmp(branch,"*") == 0)
       m_silent_low_level = true;
    return TTreeCache::AddBranch(branch, subbranches);
 }

Int_t FWTTreeCache::AddBranchTopLevel ( const char * bname )

Definition at line 34 of file FWTTreeCache.cc.

References AddBranch(), is_branch_in_cache(), m_branch_set, and s_logging.

Referenced by BranchAccessCallIn(), and FWTTreeCache::LearnGuard::~LearnGuard().

 {
    if (bname == nullptr || bname[0] == 0)
    {
       printf("FWTTreeCache::AddBranchTopLevel Invalid branch name <%s>\n",
              bname == nullptr ? "nullptr" : "empty string");
       return -1;
    }
 
    Int_t ret = 0;
    if ( ! is_branch_in_cache(bname))
    {
       if (s_logging)
          printf("FWTTreeCache::AddBranchTopLevel '%s' -- adding\n", bname);
       {  LearnGuard _lg(this);
          ret = AddBranch(bname, true);
       }
       if (ret == 0)
          m_branch_set.insert(bname);
    }
    else
    {
       if (s_logging)
          printf("FWTTreeCache::AddBranchTopLevel '%s' -- already in cache\n", bname);
    }
 
    return ret;
 }

void FWTTreeCache::BranchAccessCallIn ( const TBranch * b )

Definition at line 88 of file FWTTreeCache.cc.

References AddBranchTopLevel(), and s_logging.

Referenced by FWTTreeCache::LearnGuard::~LearnGuard().

 {
    if (s_logging)
       printf("FWTTreeCache::BranchAccessCallIn '%s'\n", b->GetName());
 
    AddBranchTopLevel(b->GetName());
 }

Int_t FWTTreeCache::DropBranch	(	TBranch *	b,
		Bool_t	subbranches = `kFALSE`
	)

override

Definition at line 114 of file FWTTreeCache.cc.

References m_silent_low_level, and s_logging.

Referenced by DropBranchTopLevel(), and FWTTreeCache::LearnGuard::~LearnGuard().

 {
    if (s_logging && ! m_silent_low_level)
       printf("FWTTreeCache::DropBranch by ptr '%s', subbp=%d\n", b->GetName(), subbranches);
    return TTreeCache::DropBranch(b, subbranches);
 }

Int_t FWTTreeCache::DropBranch	(	const char *	branch,
		Bool_t	subbranches = `kFALSE`
	)

override

Definition at line 121 of file FWTTreeCache.cc.

References m_silent_low_level, and s_logging.

 {
    if (s_logging)
       printf("FWTTreeCache::DropBranch by name '%s', subbp=%d\n", branch, subbranches);
    Int_t ret = TTreeCache::DropBranch(branch, subbranches);
    if (strcmp(branch,"*") == 0)
       m_silent_low_level = false;
    return ret;
 }

Int_t FWTTreeCache::DropBranchTopLevel ( const char * bname )

Definition at line 63 of file FWTTreeCache.cc.

References DropBranch(), is_branch_in_cache(), m_branch_set, and s_logging.

Referenced by FWTTreeCache::LearnGuard::~LearnGuard().

 {
    if (bname == nullptr || bname[0] == 0)
    {
       printf("FWTTreeCache::AddBranchTopLevel Invalid branch name");
       return -1;
    }
 
    Int_t ret = 0;
    if (is_branch_in_cache(bname))
    {
       if (s_logging)
          printf("FWTTreeCache::DropBranchTopLevel '%s' -- dropping\n", bname);
       m_branch_set.erase(bname);
       LearnGuard _lg(this);
       ret = DropBranch(bname, true);
    }
    else
    {
       if (s_logging)
          printf("FWTTreeCache::DropBranchTopLevel '%s' -- not in cache\n", bname);
    }
    return ret;
 }