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

Timestamp:

03/02/15 17:43:15 (9 years ago)

Author:

labetoulle

Message:

Move common functions to common module

File:

• TOOLS/ConsoGENCMIP6/bin/plot_login.py (modified) (9 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
 
 # Application library imports
@@ -26 +24 @@
   #---------------------------------------
   def fill_data(self, filein):
-    data = np.genfromtxt(
-      filein,
-      skip_header=1,
-      converters={0: string_to_date,
-                  1: str},
-      missing_values="nan",
-    )
+    """
+    """
+    try:
+      data = np.genfromtxt(
+        filein,
+        skip_header=1,
+        converters={0: string_to_date,
+                    1: str},
+        missing_values="nan",
+      )
+    except:
+      print("Empty file {}".format(filein))
+      exit(1)
 
     for date, login, conso in data:
@@ -87 +91 @@
   """
   """
-  print(ycoord)
-  print(consos)
-
   ax.barh(ycoord, consos, align="center", color="linen",
           linewidth=0.2, label="conso (heures)")
@@ -119 +120 @@
 
 
-########################################
-def plot_save(img_name):
-  """
-  """
-  dpi = 200.
-
-  with PdfPages(img_name) as pdf:
-    pdf.savefig(dpi=dpi)
-
-    # pdf file's metadata
-    d = pdf.infodict()
-    d["Title"]   = "Conso GENCMIP6 par login"
-    d["Author"]  = "plot_bilan.py"
-    # d["Subject"] = "Time spent over specific commands during create_ts \
-    #                 jobs at IDRIS and four configurations at TGCC"
-    # d["Keywords"] = "bench create_ts TGCC IDRIS ncrcat"
-    # d["CreationDate"] = dt.datetime(2009, 11, 13)
-    # d["ModDate"] = dt.datetime.today()
+# ########################################
+# def plot_save(img_name):
+#   """
+#   """
+#   dpi = 200.
+
+#   img_in  = os.path.join(DIR["PLOT"], "{}.pdf".format(img_name))
+
+#   with PdfPages(img_in) as pdf:
+#     pdf.savefig(dpi=dpi)
+
+#     # pdf file's metadata
+#     d = pdf.infodict()
+#     d["Title"]   = "Conso GENCMIP6 par login"
+#     d["Author"]  = "plot_bilan.py"
+#     # d["Subject"] = "Time spent over specific commands during create_ts \
+#     #                 jobs at IDRIS and four configurations at TGCC"
+#     # d["Keywords"] = "bench create_ts TGCC IDRIS ncrcat"
+#     # d["CreationDate"] = dt.datetime(2009, 11, 13)
+#     # d["ModDate"] = dt.datetime.today()
+
+#   if os.path.isdir(DIR["SAVEPLOT"]):
+#     img_out = os.path.join(DIR["SAVEPLOT"],
+#                            "{}_{}.pdf".format(img_name, today))
+#     shutil.copy(img_in, img_out)
 
 
@@ -173 +181 @@
 
   import matplotlib.pyplot as plt
-  from matplotlib.backends.backend_pdf import PdfPages
+  # from matplotlib.backends.backend_pdf import PdfPages
 
   if not args.show:
@@ -180 +188 @@
   # ... Files and directories ...
   # -----------------------------
-  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"
+  (file_param, file_utheo, file_data) = \
+      get_input_files(DIR["SAVEDATA"],
+                      [OUT["PARAM"], OUT["UTHEO"], OUT["LOGIN"]])
+
+  img_name = "login"
+  today = os.path.basename(file_param).strip(OUT["PARAM"])
 
   if args.verbose:
     print(file_param)
     print(file_utheo)
-    print(file_login)
+    print(file_data)
     print(img_name)
+    print(today)
 
   # .. Get project info ..
@@ -202 +213 @@
   # ----------------------
   logins = LoginDict()
-  logins.fill_data(file_login)
+  logins.fill_data(file_data)
 
   # .. Extract data depending on C.L. arguments ..
@@ -245 +256 @@
   # ... Save figure ...
   # -------------------
-  plot_save(os.path.join(DIR["PLOT"], img_name))
+  plot_save(img_name, today, "Conso GENCMIP6 par login")
 
   # ... Publish figure on dods ...
@@ -255 +266 @@
     plt.show()
 
-  exit()
+  exit(0)