Merge pull request #327 from YBCS/dev-buda-development-nogui

figures generated by `pynml-modchananalysis` are saved by default

Author: sanjayankur31

man/man1/pynml-modchananalysis.1

@@ -3,7 +3,7 @@
 .SH NAME
 pynml-modchananalysis \- manual page for pynml-modchananalysis v1.2.8
 .SH DESCRIPTION
-usage: pynml\-modchananalysis [\-h] [\-v] [\-nogui] [\-minV <min v>]
+usage: pynml\-modchananalysis [\-h] [\-v] [\-nogui] [\-savePlots] [\-minV <min v>]
 .TP
 [\-maxV <max v>] [\-stepV <step v>]
 [\-dt <time step>] [\-duration <duration>]
@@ -30,6 +30,9 @@ Verbose output
 \fB\-nogui\fR
 Supress plotting of variables and only save to file
 .TP
+\fB\-savePlots\fR
+Export image (.png) of plotted figures to device
+.TP
 \fB\-minV\fR <min v>
 Minimum voltage to test (integer, mV)
 .TP

pyneuroml/neuron/analysis/HHanalyse.py

@@ -44,6 +44,13 @@ def process_args():
         help="Supress plotting of variables and only save to file",
     )
 
+    parser.add_argument(
+        "-savePlots",
+        action="store_true",
+        default=False,
+        help="Export image (.png) of plotted figures to device",
+    )
+
     parser.add_argument(
         "-minV",
         type=int,
@@ -535,6 +542,16 @@ def main():
             file.close()
             print("Written info to file: %s" % file_name)
 
+    if args.savePlots:
+        figs = pylab.get_fignums()
+
+        for fig in figs:
+            pylab.figure(fig)
+            window_title = pylab.get_current_fig_manager().get_window_title()
+            file_name = f"{window_title}.png"
+            plt.savefig(file_name)
+            print("Exported img to file: %s" % file_name)
+
     if not args.nogui:
         pylab.show()