def save(wrps, filename_func, suffices = None):
"""
Saves passing wrps to disk, plus .info file with the wrapper infos.
:param wrps: Wrapper iterable
:param filename_func: callable that returns path and filename without
suffix.
:param suffices: list of suffices
**Example:** ::
save(
wrappers,
lambda wrp: OUTPUT_DIR + wrp.name,
[.root, .png] # DEFAULT: settings.rootfile_postfixes
)
"""
if not suffices: suffices = settings.rootfile_postfixes
for wrp in wrps:
filename = filename_func(wrp)
prim_obj = wrp.primary_object()
for suffix in suffices:
prim_obj.SaveAs(filename + suffix)
diskio.write(wrp, filename)
yield wrp
def save(wrps, filename_func, suffices=None):
"""
Saves passing wrps to disk, plus .info file with the wrapper infos.
:param wrps: Wrapper iterable
:param filename_func: callable that returns path and filename without
suffix.
:param suffices: list of suffices
**Example:** ::
save(
wrappers,
lambda wrp: OUTPUT_DIR + wrp.name,
[.root, .png] # DEFAULT: settings.rootfile_postfixes
)
"""
if not suffices: suffices = settings.rootfile_postfixes
for wrp in wrps:
filename = filename_func(wrp)
prim_obj = wrp.primary_object()
for suffix in suffices:
prim_obj.SaveAs(filename + suffix)
diskio.write(wrp, filename)
yield wrp
def save(wrps, filename_func, suffices=None, write_complete_wrp=False):
"""
Saves passing wrps to disk, plus .info file with the wrapper infos.
:param wrps: Wrapper iterable
:param filename_func: callable that returns path and filename without
suffix.
:param suffices: list of suffices
**Example:** ::
save(
wrappers,
lambda wrp: OUTPUT_DIR + wrp.name,
[.root, .png] # DEFAULT: settings.rootfile_postfixes
)
"""
if not suffices:
suffices = settings.rootfile_postfixes
for wrp in wrps:
filename = filename_func(wrp)
if write_complete_wrp:
diskio.write(wrp, filename, suffices)
else:
diskio.small_write(wrp, filename, suffices)
yield wrp
def finished(self):
res_file = self.plot_output_dir + "result"
if isinstance(self.result, wrappers.Wrapper):
self.result.name = self.name
if self.has_output_dir:
diskio.write(self.result, res_file)
settings.post_proc_dict[self.name] = self.result
elif isinstance(self.result, list) or isinstance(self.result, tuple):
if self.has_output_dir:
filenames = []
for i, wrp in enumerate(self.result):
num_str = "_%03d" % i
wrp.name = self.name + num_str
filenames.append(res_file + num_str)
diskio.write(wrp, res_file + num_str)
diskio.write(
wrappers.Wrapper(
name = self.name,
RESULT_WRAPPERS = filenames
),
res_file
)
settings.post_proc_dict[self.name] = self.result
self.time_fin = time.ctime() + "\n"
with open(self._info_file, "w") as f:
f.write(self.time_start)
f.write(self.time_fin)
super(PostProcTool, self).finished()
