def save(self): atom_values = self.do_expressions() if atom_values is not None: filename = context.application.model.filename if filename is None: filename = "" else: filename = filename[:filename.rfind(".")] filename = ask_save_filename("Save shell data", filename) if filename is not None: if not filename.endswith(".txt"): filename += ".txt" f = file(filename, "w") for atom, value in zip(self.graph.nodes, atom_values): print >> f, atom.get_name(), value f.close()
def on_bu_save_clicked(self, button): filename = context.application.model.filename if filename is not None: filename = filename[:filename.rfind(".")] + "_" else: filename = "" filename += self.label.lower().replace(" ", "_") filename = ask_save_filename("Save distribution data", filename) if filename is not None: self.save_data("%s.txt" % filename) import pylab pylab.figure(0) self.mpl_widget.print_figure("%s.svg" % filename) self.mpl_widget.print_figure("%s.eps" % filename, orientation='landscape') self.mpl_widget.print_figure("%s.png" % filename, dpi=400)