diff --git a/bindings/pyroot_experimental/PyROOT/python/ROOT/pythonization/_tchain.py b/bindings/pyroot_experimental/PyROOT/python/ROOT/pythonization/_tchain.py new file mode 100644 index 0000000000000000000000000000000000000000..7b21fe25ed0bb6504ad4fbf64dcd8be470b58194 --- /dev/null +++ b/bindings/pyroot_experimental/PyROOT/python/ROOT/pythonization/_tchain.py @@ -0,0 +1,18 @@ + +from ROOT import pythonization + +from _ttree import _SetBranchAddress as TTreeSetBranchAddress + +# Pythonizor function +@pythonization +def pythonize_tchain(klass, name): + # Parameters: + # klass: class to be pythonized + # name: string containing the name of the class + + if name == 'TChain': + # SetBranchAddress + klass._OriginalSetBranchAddress = klass.SetBranchAddress + klass.SetBranchAddress = TTreeSetBranchAddress + + return True