branchselection.BranchSelection Class Reference

Public Member Functions
def	__init__ (self, branchsel)
def	selectBranches (self, tree)

Private Attributes
	_ops

Detailed Description

Definition at line 9 of file branchselection.py.

Constructor & Destructor Documentation

__init__()

def branchselection.BranchSelection.__init__	(		self,
			branchsel
	)

Definition at line 10 of file branchselection.py.

References print(), and nano_mu_digi_cff.strip.

    def __init__(self, branchsel):
        comment = re.compile(r"#.*")
        ops = []

        if isinstance(branchsel, list):
            # branchsel is a list of commands
            lines = branchsel
        elif isinstance(branchsel, str):
            # branchsel is a filename
            lines=[]
            for line in open(branchsel, 'r'):                line = line.strip()
                if len(line) == 0 or line[0] == '#':
                    continue
                line = re.sub(comment, "", line)
                while line[-1] == "\\":
                    line = line[:-1] + " " + file.next().strip()
                    line = re.sub(comment, "", line)
                lines.append(line)
                
        for line in lines:
            try:
                (op, sel) = line.split()
                if op == "keep":
                    ops.append((sel, 1))
                elif op == "drop":
                    ops.append((sel, 0))
                elif op == "keepmatch":
                    ops.append((re.compile("(:?%s)$" % sel), 1))
                elif op == "dropmatch":
                    ops.append((re.compile("(:?%s)$" % sel), 0))
                else:
                    print("Error in branchsel: line '%s': "% (line)
                        + "it's not (keep|keepmatch|drop|dropmatch) "
                        + "<branch_pattern>"
                    )
            except ValueError as e:
                print("Error in branchsel: line '%s': " % (line)
                    + "it's not (keep|keepmatch|drop|dropmatch) "
                    + "<branch_pattern>"
                )
        self._ops = ops

Member Function Documentation

selectBranches()

def branchselection.BranchSelection.selectBranches	(		self,
			tree
	)

Definition at line 53 of file branchselection.py.

References branchselection.BranchSelection._ops.

    def selectBranches(self, tree):
        tree.SetBranchStatus("*", 1)
        branchNames = [b.GetName() for b in tree.GetListOfBranches()]
        for bre, stat in self._ops:
            if type(bre) == Pattern:
                for n in branchNames:
                    if re.match(bre, n):
                        tree.SetBranchStatus(n, stat)
            else:
                tree.SetBranchStatus(bre, stat)

Member Data Documentation

_ops

branchselection.BranchSelection._ops

private

Definition at line 51 of file branchselection.py.

Referenced by branchselection.BranchSelection.selectBranches().

---

Generated for CMSSW Reference Manual by  1.8.14