def __init__(self, *args, **kwargs):
super(GuardedTFile, self).__init__()
self._guarded_tfile = None
ROOT = import_ROOT()
if len(args) == 1 and isinstance(args[0], ROOT.TFile) and not kwargs:
self._guarded_tfile = args[0]
elif args or kwargs:
self._guarded_tfile = ROOT.TFile(*args, **kwargs)
def dump(cls, path, arr, *args, **kwargs):
ROOT = import_ROOT() # noqa: F841
import root_numpy
return root_numpy.array2root(arr, get_path(path), *args, **kwargs)
def load(cls, path, *args, **kwargs):
ROOT = import_ROOT() # noqa: F841
import root_numpy
return root_numpy.root2array(get_path(path), *args, **kwargs)
def Open(cls, *args, **kwargs):
ROOT = import_ROOT()
return cls(ROOT.TFile.Open(*args, **kwargs))
def dump(cls, path, df, *args, **kwargs):
ROOT = import_ROOT() # noqa: F841
# importing root_pandas adds the to_root() method to data frames
import root_pandas # noqa: F401
return df.to_root(get_path(path), *args, **kwargs)
def load(cls, path, *args, **kwargs):
ROOT = import_ROOT() # noqa: F841
import root_pandas
return root_pandas.read_root(get_path(path), *args, **kwargs)