def load_from_save(self, save_file_path): f = open(save_file_path) lines = f.readlines() f.close() vmg = visual_structure.visual_motif_graph_from_topology(lines[0]) levels = [int(x) for x in lines[1].split(" ")[:-1]] for i, n in enumerate(vmg.mg.graph.nodes): n.level = levels[i] self.state.vmg = vmg max_level = 0 for l in levels: if l > max_level: max_level = l for i in range(max_level): self.state.vmg.mg.increase_level() self.functions[0].points_from_str(lines[2]) spl = lines[3].split(" ") highlighed_end = [int(spl[0])] vm = self.state.vmg.v_motifs[int(spl[0])] end_name = spl[1].rstrip() for end in vm.ends: if end.name() == end_name: highlighed_end.append(end) self.functions[0].highlighted_ends.append(highlighed_end) if max_level > 0: self.functions[0].built = 1
vmg = visual_structure.VMotifGraph(view_mode=2) if preset_name == "ttr": m = rm.manager.get_motif(name="GAAA_tetraloop", end_name="A229-A245") vmg.add_motif(m) return vmg if __name__ == '__main__': args = parse_args() gui_window = gui_window_new.get_default_window() if args.preset: vmg = parse_presets(args.preset) gui_window.set_vmg(vmg) elif args.s: gui_window.load_from_save(args.s) elif args.mg: f = open(args.mg) lines = f.readlines() f.close() vmg = visual_structure.visual_motif_graph_from_topology(lines[0]) gui_window.set_vmg(vmg) gui_window.setup() while 1: rate(100) gui_window.listen()