Add a flag to force the plots to close on exit

Author: jgray-19

src/madl_plot.mad

@@ -313,15 +313,26 @@ local function gnuplot_mv (fname, uname)
   end
 end
 
+local not_paused = true
+local function kill_plots(self)
+  if not_paused then -- Pause only once, independent of the number of streams
+    pause("Closing plots, ") 
+    not_paused = false
+  end
+  gplot:set_sid(self.sid)
+  gplot:printf("set term %s close", self.term)
+end
+
 local function teardown_cmd (self, file, fname)
-  local scrdump, output in self
+  local scrdump, output, close_plots in self
   file:close()
   if output ~= false then
     gplot:printf("load('%s')", fname)
     if is_string(scrdump)
     then gnuplot_mv(fname, scrdump)
     else gnuplot_rm(fname)
     end
+    if close_plots then atexit(\-> kill_plots(self)) end
   else
     if is_string(scrdump) then
       os.rename(fname, scrdump)
@@ -1005,6 +1016,7 @@ local plot = command "plot" {
 
   -- output
   output      = nil,        -- ("filename") -> pdf | number -> wid
+  close_plots = true,       -- boolean, close all plots on exit of MAD-NG or not
   scrdump     = nil,        -- ("filename")
 
   -- title