8 #include "TTreeIndex.h"
9 #include "TTreeCache.h"
18 if (branch ==
nullptr) {
23 TBranch* getProductProvenanceBranch(TTree* tree,
BranchType const& branchType) {
30 unsigned int nIndexes,
31 unsigned int maxVirtualSize,
32 unsigned int cacheSize,
33 unsigned int learningEntries,
34 bool enablePrefetching,
39 branchType_(branchType),
40 auxBranch_(tree_ ? getAuxiliaryBranch(tree_, branchType_) :
nullptr),
44 rawTriggerTreeCache_(),
47 entries_(tree_ ? tree_->GetEntries() : 0),
54 rawTriggerSwitchOverEntry_(-1),
55 learningEntries_(learningEntries),
56 cacheSize_(cacheSize),
58 enablePrefetching_(enablePrefetching),
60 enableTriggerCache_(
false),
62 branchEntryInfoBranch_(metaTree_ ? getProductProvenanceBranch(metaTree_, branchType_) : (tree_ ? getProductProvenanceBranch(tree_, branchType_) : 0)),
68 long treeAutoFlush = (
tree_ ?
tree_->GetAutoFlush() : 0);
69 if (treeAutoFlush < 0) {
71 Long64_t averageEventSizeBytes =
tree_->GetZipBytes() / (
tree_->GetEntries()+1) + 1;
82 Int_t branchCount =
tree_->GetListOfBranches()->GetEntriesFast();
93 assert(index < entryNumberForIndex_->
size());
99 assert(index < entryNumberForIndex_->
size());
124 if(
tree_->GetBranch(oldBranchName.c_str()) ==
nullptr){
135 TBranch* branch =
tree_->GetBranch(oldBranchName.c_str());
145 branches_->insert(std::make_pair(key, info));
151 TBranch* branch =
tree_->GetBranch(oldBranchName.c_str());
152 if (branch !=
nullptr) {
153 TObjArray* leaves =
tree_->GetListOfLeaves();
154 int entries = leaves->GetEntries();
156 TLeaf*
leaf = (TLeaf*)(*leaves)[
i];
157 if (leaf ==
nullptr)
continue;
158 TBranch*
br = leaf->GetBranch();
159 if (br ==
nullptr)
continue;
160 if (br->GetMother() == branch) {
161 leaves->Remove(leaf);
165 tree_->GetListOfBranches()->Remove(branch);
166 tree_->GetListOfBranches()->Compress();
177 tree_->SetCacheSize(static_cast<Long64_t>(cacheSize));
186 if (treeMaxVirtualSize >= 0)
tree_->SetMaxVirtualSize(static_cast<Long64_t>(treeMaxVirtualSize));
248 TTree::TClusterIterator clusterIter =
tree_->GetClusterIterator(entryNumber);
255 tree_->SetCacheSize(static_cast<Long64_t>(5*1024*1024));
259 int branchCount = branches->GetEntriesFast();
264 for (
int i=0;
i<branchCount;
i++) {
265 TBranch *tmp_branch = (TBranch*)branches->UncheckedAt(
i);
287 tree_->SetCacheSize(static_cast<Long64_t>(5*1024*1024));
313 tree_->SetCacheSize(static_cast<Long64_t>(5*1024*1024));
337 TTreeCache *triggerCache =
NULL;
357 TTreeCache * cache =
selectCache(branch, entryNumber);
359 branch->GetEntry(entryNumber);
449 int branchCount = branches->GetEntriesFast();
450 for (
int i=0;
i<branchCount;
i++) {
451 TBranch *branch = (TBranch*)branches->UncheckedAt(
i);
452 if ((branchNames[0] ==
'*') || (strcmp(branchNames, branch->GetName()) == 0)) {
464 n = branch->GetEntry(entryNumber);
476 n = tree->GetEntry(entryNumber);
484 std::unique_ptr<TTreeCache>
487 tree->SetCacheSize(cacheSize);
488 std::unique_ptr<TTreeCache> treeCache(dynamic_cast<TTreeCache*>(file.
GetCacheRead()));
489 if (
nullptr != treeCache.get()) {
490 treeCache->StartLearningPhase();
491 treeCache->SetEntryRange(0, tree->GetEntries());
492 treeCache->AddBranch(branchNames, kTRUE);
493 treeCache->StopLearningPhase();
Int_t getEntry(TBranch *branch, EntryNumber entryNumber)
boost::shared_ptr< TTreeCache > treeCache_
std::string const & branchName() const
std::string const & BranchTypeToMetaDataTreeName(BranchType const &branchType)
std::string const & BranchTypeToAuxiliaryBranchName(BranchType const &branchType)
unsigned int learningEntries_
TTreeCache * checkTriggerCache(TBranch *branch, EntryNumber entryNumber) const
void addBranch(BranchKey const &key, BranchDescription const &prod, std::string const &oldBranchName)
void dropBranch(std::string const &oldBranchName)
RootTree(boost::shared_ptr< InputFile > filePtr, BranchType const &branchType, unsigned int nIndexes, unsigned int maxVirtualSize, unsigned int cacheSize, unsigned int learningEntries, bool enablePrefetching, InputType inputType)
std::vector< std::string > branchNames_
EntryNumber rawTriggerSwitchOverEntry_
roottree::BranchMap BranchMap
EntryNumber const & entries() const
std::unordered_set< TBranch * > trainedSet_
TBranch * branchEntryInfoBranch_
void setPresence(BranchDescription &prod, std::string const &oldBranchName)
boost::shared_ptr< TTreeCache > rawTreeCache_
void trainCache(char const *branchNames)
void setTreeMaxVirtualSize(int treeMaxVirtualSize)
std::map< BranchKey const, BranchInfo > BranchMap
boost::shared_ptr< TTreeCache > triggerTreeCache_
boost::shared_ptr< TTreeCache > rawTriggerTreeCache_
void insertEntryForIndex(unsigned int index)
TTreeCache * selectCache(TBranch *branch, EntryNumber entryNumber) const
unsigned long treeAutoFlush_
std::string const & BranchTypeToBranchEntryInfoBranchName(BranchType const &branchType)
void setDropped(bool isDropped)
std::string const & BranchTypeToProductTreeName(BranchType const &branchType)
IndexIntoFile::EntryNumber_t EntryNumber
TTreeCache * checkTriggerCacheImpl(TBranch *branch, EntryNumber entryNumber) const
EntryNumber const & entryNumber() const
std::string const & BranchTypeToInfoTreeName(BranchType const &branchType)
TBranch * provenanceBranch_
DelayedReader * rootDelayedReader() const
EntryNumber const & entryNumberForIndex(unsigned int index) const
void getEntry(TBranch *branch, EntryNumber entry) const
boost::shared_ptr< BranchMap > branches_
void addContext(std::string const &context)
void setCacheSize(unsigned int cacheSize)
bool performedSwitchOver_
std::unique_ptr< std::vector< EntryNumber > > entryNumberForIndex_
std::unordered_set< TBranch * > triggerSet_
std::string const & BranchTypeToAuxBranchName(BranchType const &branchType)
BranchMap const & branches() const
std::unique_ptr< DelayedReader > rootDelayedReader_
boost::shared_ptr< InputFile > filePtr_
std::string const & branchListIndexesBranchName()
volatile std::atomic< bool > shutdown_flag false
std::unique_ptr< TTreeCache > trainCache(TTree *tree, InputFile &file, unsigned int cacheSize, char const *branchNames)
EntryNumber switchOverEntry_
tuple size
Write out results.
T get(const Candidate &c)
void setEntryNumber(EntryNumber theEntryNumber)
roottree::EntryNumber EntryNumber