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)