close figures after drawing/saving/showing

1 files changed, 17 insertions, 3 deletions

diff --git a/mkcuts_dvpT.py b/mkcuts_dvpT.py
index fb3e256..ce6bed6 100644
--- a/mkcuts_dvpT.py
+++ b/mkcuts_dvpT.py
@@ -234,9 +234,23 @@ def main():
                 f7.savefig('cut_Txy_%05d.png' % iout)
                 f8.savefig('cut_Tzy_%05d.png' % iout)
 
-    # Show Figures?
-    if args.show:
-        plt.show()
+        # Show Figures?
+        if args.show:
+            plt.show()
+
+        # Close Figures
+        if args.d:
+            plt.close(f1)
+            plt.close(f2)
+        if args.v:
+            plt.close(f3)
+            plt.close(f4)
+        if args.p:
+            plt.close(f5)
+            plt.close(f6)
+        if args.T:
+            plt.close(f7)
+            plt.close(f8)
 
 """
 Jump into Main().