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

Timestamp:

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

Author:

labetoulle

Message:

Save daily plots on WORK

File:

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

TOOLS/ConsoGENCMIP6/bin/plot_login.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
@@ -147 +147 @@
                       help="plot all the logins" +
                            " (default: plot only non-zero)")
+  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()
@@ -159 +163 @@
   # ------------------------------
   args = get_arguments()
+  if args.verbose:
+    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_login = get_last_file(DIR["DATA"], OUT["LOGIN"])
+  file_param = get_last_file(DIR["SAVE"], OUT["PARAM"])
+  file_utheo = get_last_file(DIR["SAVE"], OUT["UTHEO"])
+  file_login = get_last_file(DIR["SAVE"], OUT["LOGIN"])
   img_name = "bilan.pdf"
 
@@ -229 +247 @@
   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()