#154: Added return figure. Removed sample figure

pylhc · Oct 15, 2018 · abc0e5f · abc0e5f
1 parent 6412767
commit abc0e5f
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/figure_editor.py b/figure_editor.py
@@ -31,10 +31,12 @@ def main(fig=None):
     form = MainWindow(fig)
     form.show()
     app.exec_()
+    return form.get_figure()
 
 
 # Script Mode #################################################################
 
 
 if __name__ == "__main__":
-    main(_quick_figure())
+    # main(_quick_figure())
+    main()
diff --git a/plotshop/fig_editor/main_window.py b/plotshop/fig_editor/main_window.py
@@ -374,3 +374,7 @@ def update_figure(self, figure=None):
             self.canvas.set_pickers(self.picktol)
         self.canvas.draw()
 
+    def get_figure(self):
+        """ Return the currently used figure. """
+        return self.canvas.figure
+