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()