Belle2::RootIOUtilities Namespace Reference

Some constants and helpers common to the RootInput and RootOutput modules. More...

Classes
class	RootFileInfo
	Helper class to factorize some necessary tasks when working with Belle2 output files. More...

Functions
std::set< std::string >	filterBranches (const std::set< std::string > &branchesToFilter, const std::vector< std::string > &branches, const std::vector< std::string > &excludeBranches, int durability, bool quiet=false)
	Given a list of input branches and lists of branches to include/exclude, returns a list of branches that are accepted. More...
size_t	setBranchStatus (TBranch *branch, bool process)
	Set Branch to be read or not. More...
std::vector< std::string >	expandWordExpansions (const std::vector< std::string > &filenames)
	Performs wildcard expansion using wordexp(), returns matches.
long	getEntryNumberWithEvtRunExp (TTree *tree, long event, long run, long experiment)
	return entry number with given (event, run, experiment) from tree. More...
void	buildIndex (TTree *tree)
	Build TTreeIndex on tree (assumes EventMetaData branch exists there).
bool	hasStreamer (const TClass *cl)
	Returns true if and only if 'cl' or one of its bases has I/O streamers. More...
bool	hasCustomStreamer (const TClass *cl)
	Returns true if and only if 'cl' has a user-defined streamer. More...
void	setCreationData (FileMetaData &metadata)
	Fill the creation info of a file meta data: site, user, data.
std::string	getCommitID ()
	Return git SHA1 hashes taking into account local & central release. More...

Variables
const std::string	c_treeNames [] = { "tree", "persistent" }
	Names of trees.
const std::string	c_SteerBranchNames [] = { "branchNames", "branchNamesPersistent" }
	Steering parameter names for m_branchNames.
const std::string	c_SteerExcludeBranchNames [] = { "excludeBranchNames", "excludeBranchNamesPersistent" }
	Steering parameter names for m_excludeBranchNames.
const std::string	c_SteerAdditionalBranchNames [] = { "additionalBranchNames", "additionalBranchNamesPersistent" }
	Steering parameter names for m_additionalBranchNames.

Detailed Description

Some constants and helpers common to the RootInput and RootOutput modules.

Function Documentation

filterBranches()

std::set< std::string > filterBranches	(	const std::set< std::string > &	branchesToFilter,
		const std::vector< std::string > &	branches,
		const std::vector< std::string > &	excludeBranches,
		int	durability,
		bool	quiet = false
	)

Given a list of input branches and lists of branches to include/exclude, returns a list of branches that are accepted.

More precisely, an item b from 'branchesToFilter' will be in the returned set if b not in excludeBranches and (b in branches or empty(branches) or b in relationsBetweenAcceptedBranches) and b not in relationsInvolvingExcludedBranches

Parameters

branchesToFilter	input
branches	if not empty, the list of branches to be accepted
excludeBranches	branches that should never end up in output (takes precedence over everything else)
durability	Durability being filtered (used for messages only)
quiet	If true don't warn about branches which are missing/extra/duplicate in the lists

Definition at line 32 of file RootIOUtilities.cc.

{
  std::set<std::string> branchSet, excludeBranchSet;
  for (const std::string& b : branches) {
    if (branchesToFilter.count(b) == 0 and not quiet)
      B2WARNING("The branch " << b << " given in " << c_SteerBranchNames[durability] << " does not exist.");
    if (!branchSet.insert(b).second and not quiet)
      B2WARNING(c_SteerBranchNames[durability] << " has duplicate entry " << b);
  }
  for (const std::string& b : excludeBranches) {
    // FIXME: ProcessStatistics is excluded by default but not always present. We should switch that to not write it out
    // in the first place but the info message is meaningless for almost everyone
    if (branchesToFilter.count(b) == 0 and not quiet and b != "ProcessStatistics")
      B2INFO("The branch " << b << " given in " << c_SteerExcludeBranchNames[durability] << " does not exist.");
    if (!excludeBranchSet.insert(b).second and not quiet)
      B2WARNING(c_SteerExcludeBranchNames[durability] << " has duplicate entry " << b);
  }
 
  std::set<std::string> out, relations, excluderelations;
  for (const std::string& branch : branchesToFilter) {
    if (excludeBranchSet.count(branch))
      continue;
    if (branchSet.empty() or branchSet.count(branch))
      out.insert(branch);
  }
  if (!excludeBranchSet.empty()) {
    //remove relations for excluded things
    for (const std::string& from : branchesToFilter) {
      for (const std::string& to : branchesToFilter) {
        std::string branch = DataStore::relationName(from, to);
        if (out.count(branch) == 0)
          continue; //not selected
        if (excludeBranchSet.count(from) == 0 and excludeBranchSet.count(to) == 0)
          continue; //at least one side should be excluded
        if (branchSet.count(branch) != 0)
          continue; //specifically included
        excluderelations.insert(branch);
      }
    }
    for (const std::string& rel : excluderelations) {
      out.erase(rel);
    }
  }
  //add relations between accepted branches
  for (const std::string& from : out) {
    for (const std::string& to : out) {
      std::string branch = DataStore::relationName(from, to);
      if (branchesToFilter.count(branch) == 0)
        continue; //not in input
      if (excludeBranchSet.count(branch))
        continue;
      relations.insert(branch);
    }
  }
  out.insert(relations.begin(), relations.end());
  return out;
}

getCommitID()

std::string getCommitID

(

)

Return git SHA1 hashes taking into account local & central release.

ID is a combined hash $CENTRAL_SHA1[+$LOCAL_SHA1][-modified], or just SHA1[-modified] if only one release or they are on the same revision. Empty string denotes at least one untracked release directory.

Definition at line 207 of file RootIOUtilities.cc.

getEntryNumberWithEvtRunExp()

long getEntryNumberWithEvtRunExp	(	TTree *	tree,
		long	event,
		long	run,
		long	experiment
	)

return entry number with given (event, run, experiment) from tree.

Returns -1 if not found.

Definition at line 133 of file RootIOUtilities.cc.

hasCustomStreamer()

bool hasCustomStreamer

(

const TClass *

cl

)

Returns true if and only if 'cl' has a user-defined streamer.

In that case, TClonesArrays of this type should be written with BypassStreamer(false) and split-level -1 (no splitting).

Definition at line 178 of file RootIOUtilities.cc.

hasStreamer()

bool hasStreamer

(

const TClass *

cl

)

Returns true if and only if 'cl' or one of its bases has I/O streamers.

TObject is not considered to have any.

Definition at line 157 of file RootIOUtilities.cc.

setBranchStatus()

size_t setBranchStatus	(	TBranch *	branch,
		bool	process
	)

Set Branch to be read or not.

TTree::SetBranchAddress doesn't do what we want it to do because our branch names seem to not be conform to their expectations. But we don't need to use name matching, we know our structure and it's simple enough so we just recursively go through the branch and all children and set processing on or off.

Parameters

branch	The branch to change the status for
process	Whether or not to read/process this branch

Returns

the number of branches enabled/disabled

Definition at line 92 of file RootIOUtilities.cc.