Changeset 2421 for TOOLS/ConsoGENCMIP6/bin/plot_bilan.py

Timestamp:

03/02/15 14:19:08 (9 years ago)

Author:

labetoulle

Message:

copy plots on dods

File:

• TOOLS/ConsoGENCMIP6/bin/plot_bilan.py (modified) (5 diffs)

TOOLS/ConsoGENCMIP6/bin/plot_bilan.py

@@ -11 +11 @@
 import datetime as dt
 import numpy as np
-import matplotlib.pyplot as plt
-from matplotlib.backends.backend_pdf import PdfPages
 
 # Application library imports
@@ -252 +250 @@
   parser = ArgumentParser()
   parser.add_argument("-v", "--verbose", action="store_true",
-                      help="Verbose mode")
+                      help="verbose mode")
   parser.add_argument("-f", "--full", action="store_true",
                       help="plot the whole period")
@@ -263 +261 @@
   parser.add_argument("-m", "--max", action="store_true",
                       help="plot with y_max = allocation")
+  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()
@@ -272 +274 @@
   # .. Initialization ..
   # ====================
+
   # ... Command line arguments ...
   # ------------------------------
   args = get_arguments()
 
+  # ... 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_bilan = get_last_file(DIR["DATA"], OUT["BILAN"])
+  file_param = get_last_file(DIR["SAVE"], OUT["PARAM"])
+  file_utheo = get_last_file(DIR["SAVE"], OUT["UTHEO"])
+  file_bilan = get_last_file(DIR["SAVE"], OUT["BILAN"])
   img_name = "bilan.pdf"
 
@@ -373 +388 @@
   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()