Changeset 2425 for TOOLS/ConsoGENCMIP6/bin/plot_store.py

Timestamp:

03/02/15 15:50:26 (9 years ago)

Author:

labetoulle

Message:

Save daily plots on WORK

File:

• TOOLS/ConsoGENCMIP6/bin/plot_store.py (modified) (4 diffs)

TOOLS/ConsoGENCMIP6/bin/plot_store.py

@@ -10 +10 @@
 import os.path
 import numpy as np
-import matplotlib.pyplot as plt
-from matplotlib.backends.backend_pdf import PdfPages
+# import matplotlib.pyplot as plt
+# from matplotlib.backends.backend_pdf import PdfPages
 
 # Application library imports
@@ -194 +194 @@
                       default="IPSLCM6",
                       help="plot the whole period")
+  parser.add_argument("-s", "--show", action="store_true",
+                      help="interactive mode")
+  parser.add_argument("-d", "--dods", action="store_true",
+                      help="copy output on dods")
 
   return parser.parse_args()
@@ -209 +213 @@
     print(args)
 
+  # ... Turn interactive mode off ...
+  # ---------------------------------
+  if not args.show:
+    import matplotlib
+    matplotlib.use('Agg')
+
+  import matplotlib.pyplot as plt
+  from matplotlib.backends.backend_pdf import PdfPages
+
+  if not args.show:
+    plt.ioff()
+
   # ... Files and directories ...
   # -----------------------------
-  file_param = get_last_file(DIR["DATA"], OUT["PARAM"])
-  file_utheo = get_last_file(DIR["DATA"], OUT["UTHEO"])
-  file_store = get_last_file(DIR["DATA"], OUT["STORE"])
+  file_param = get_last_file(DIR["SAVE"], OUT["PARAM"])
+  file_utheo = get_last_file(DIR["SAVE"], OUT["UTHEO"])
+  file_store = get_last_file(DIR["SAVE"], OUT["STORE"])
   img_name = "bilan.pdf"
 
@@ -287 +303 @@
   plot_save(os.path.join(DIR["PLOT"], img_name))
 
-  plt.show()
+  # ... Publish figure on dods ...
+  # ------------------------------
+  if args.dods:
+    dods_cp(os.path.join(DIR["PLOT"], img_name))
+
+  if args.show:
+    plt.show()
+
   exit()