Initial

plot_main.py

@@ -0,0 +1,68 @@
+#!/usr/bin/python3
+import sys
+from config_parse import CFG
+from constants import *
+from datetime import datetime, timedelta
+from frontend_utils import open_file
+from constants      import *
+
+import math
+import matplotlib
+matplotlib.use(CFG("use_gui_backend"))
+
+import matplotlib.pyplot as plt
+import matplotlib.dates
+import matplotlib.ticker as ticker
+
+import plot_graphutils
+import plot_imageutils
+import plot_timeutils
+
+
+def plot(datapoints,path=None,date1=None,date2=None):
+        plotname = "" if CFG("name_of_plot") == "None" else CFG("name_of_plot")
+        tup = [None,None,plot_timeutils.between_dates,plotname]
+        if CFG("enable_multicore_support"):
+                thread  = Process(target=__plot,args=(tup,datapoints,date1,date2))
+                thread.start()
+        else:
+                __plot(tup,datapoints,path,date1,date2)
+                        
+def __plot(tup,datapoints,path,date1=None,date2=None):
+        NO_SERIES  = True
+        x,y,ymin,ymax,unix_x,major_xticks = ( [] , [], -1 , -1 , [], [] )
+        lw = CFG("plot_line_width")
+        ls = CFG("plot_line_style")
+        tup[FIGURE],tup[AXIS] = plt.subplots(1, 1)
+
+        for g in datapoints.values():
+                #### GET AND CHECK TIMEFRAMES ####
+                x,y, = g.get_timeframe(tup[CALLBACK],date1,date2)
+                if len(x) <= 0 or len(y) <= 0:
+                    print("Warning: Empty series of data '%s' (wrong start/end time?)"%g.name)
+                    continue
+                else:
+                    NO_SERIES = False
+                    unix_x = list(map(plot_timeutils.unix,x))
+                    ymin,ymax = plot_graphutils.getlimits_y(y)
+
+                #### GET LINE STYLES ####
+                legend_label = plot_graphutils.legend_box_contents(g.name,y)
+                tup[AXIS].plot(unix_x, y,ls=ls,lw=lw,marker="None", label=legend_label, color=g.color)
+        
+        if NO_SERIES:
+                print("Error: no data, nothing to plot. cannot continue. exit.")
+                sys.exit(1)
+
+        ## GRID ##
+        plot_graphutils.general_background_setup(tup,ymin,ymax,x)
+
+        ## using unix_x relys on unix_x to be the same for all plots ##
+        if path == None:
+            path = open_file()
+        ## TODO function for picpathn for picpath
+        pic_path = path + ".png"
+        tup[FIGURE].savefig(pic_path,dpi=CFG("outfile_resolution_in_dpi"), bbox_inches='tight',transparent=CFG("transparent_background"))
+
+        ### do operations on the finished png ###
+        plot_imageutils.check_and_rotate(pic_path)