#!/usr/bin/env python3 # # (c) 2019-2022 Antmicro # License: Apache # import datetime import os import socket import subprocess import sys import time from common import * global gnuplot GNUPLOT_VERSION_EXPECTED = 5.0 # Run a command in a running gnuplot process def g(command): global gnuplot if not (gnuplot.poll() is None): print("Error: gnuplot not running!") return print ("gnuplot> %s" % command) try: command = b"%s\n" % command except: command = b"%s\n" % str.encode(command) gnuplot.stdin.write(b"%s\n" % command) gnuplot.stdin.flush() if command == b"quit\n": while gnuplot.poll() is None: time.sleep(0.25) # Check if the avaliable gnuplot has a required version p = run_process("gnuplot", "--version", stdout=subprocess.PIPE) version = scan(r"gnuplot (\S+)", float, p.stdout.readline().decode()) if version < GNUPLOT_VERSION_EXPECTED: print(f"Error: Gnuplot version too low. Need at least {GNUPLOT_VERSION_EXPECTED} found {version[0]}") sys.exit(1) START_DATE = "" END_DATE = "" AVERAGE_LOAD = 0.0 MAX_USED_RAM = 0 MAX_USED_FS = 0 TOTAL_RAM = 0 TOTAL_FS = 0 NAME_FS = "unknown" uname="unknown" cpus=0 cpu_name="unknown" duration = 0.0 labels = [] with open("data.txt", "r") as f: for line in f: value = None if len(line) <= 0: continue if line[0] != '#': if not START_DATE: START_DATE = scan("^(\S+)", str, line) END_DATE = scan("^(\S+)", str, line) value = scan("label: (.+)", str, line) if value is not None: key = scan("(\S+) label:", str, line) labels.append([key, value]) # Comments are not mixed with anything else, so skip continue value = scan("machine: ([^,]+)", str, line) if value is not None: uname = value value = scan("cpu count: ([^,]+)", int, line) if value is not None: cpus = value value = scan("cpu: ([^,\n]+)", str, line) if value is not None: cpu_name = value value = scan("observed disk: ([^,]+)", str, line) if value is not None: NAME_FS = value value = scan("total ram: (\S+)", stof, line) if value is not None: TOTAL_RAM = value value = scan("max ram used: (\S+)", stof, line) if value is not None: MAX_USED_RAM = value value = scan("total disk space: (\S+)", stof, line) if value is not None: TOTAL_FS = value value = scan("duration: (\S+)", stof, line) if value is not None: duration = value value = scan("max disk used: (\S+)", stof, line) if value is not None: MAX_USED_FS = value value = scan("average load: (\S+)", stof, line) if value is not None: AVERAGE_LOAD = value # Initialize the plot OUTPUT_TYPE="pngcairo" OUTPUT_EXT="png" try: if os.environ["SARGRAPH_OUTPUT_TYPE"] == "svg": OUTPUT_TYPE="svg" OUTPUT_EXT="svg" except: pass gnuplot = run_process("gnuplot", stdin=subprocess.PIPE, stdout=subprocess.PIPE) sdt = datetime.datetime.strptime(START_DATE, '%Y-%m-%d-%H:%M:%S') edt = datetime.datetime.strptime(END_DATE, '%Y-%m-%d-%H:%M:%S') seconds_between = (edt - sdt).total_seconds() if seconds_between < 100: seconds_between = 100 nsdt = sdt - datetime.timedelta(seconds = (seconds_between * 0.01)) nedt = edt + datetime.timedelta(seconds = (seconds_between * 0.01)) host = socket.gethostname() g("set ylabel 'cpu % load (user)'") g("set ylabel tc rgb 'white' font 'Courier-New,8'") g("set datafile commentschars '#'") g("set timefmt '%s'") g("set xdata time") g("set border lc rgb 'white'") g("set key tc rgb 'white'") g("set timefmt '%Y-%m-%d-%H:%M:%S'") g("set xtics format '%H:%M:%S'") g("set xtics font 'Courier-New,8' tc rgb 'white'") g("set ytics font 'Courier-New,8' tc rgb 'white'") g("set grid xtics ytics ls 12 lc rgb '#444444'") g("set style fill solid") g("set palette defined ( 0.2 '#00ff00', 0.8 '#ff0000' )") g("set cbrange [0:100]") g("unset colorbox") g("unset key") g("set rmargin 6") g(f"set terminal {OUTPUT_TYPE} size 1200,800 background '#222222' font 'Courier-New,8'") g(f"set output 'plot.{OUTPUT_EXT}'") g("set multiplot layout 3,1 title \"\\n\\n\\n\"") g(f"set title 'cpu load (average = {AVERAGE_LOAD:.2f} %%)'") g("set title tc rgb 'white' font 'Courier-New,8'") g(f"set xrange ['{nsdt.strftime('%Y-%m-%d-%H:%M:%S')}':'{nedt.strftime('%Y-%m-%d-%H:%M:%S')}']"); g("set label 101 at screen 0.02, screen 0.95 'Running on {/:Bold %s} \@ {/:Bold %s}, {/:Bold %d} threads x {/:Bold %s}, total ram: {/:Bold %.2f GB}, total disk space: {/:Bold %.2f GB}' tc rgb 'white'" % (host, uname, cpus, cpu_name, TOTAL_RAM, TOTAL_FS)) g("set label 102 at screen 0.02, screen 0.93 'duration: {/:Bold %s} .. {/:Bold %s} (%.2f minutes)' tc rgb 'white'" % (START_DATE, END_DATE, duration)) i = 0 for label in labels: i = i + 1 g(f"set arrow nohead from '{label[0]}', graph 0.01 to '{label[0]}', graph 0.87 front lc rgb 'red' dt 2") g(f"set object rect at '{label[0]}', graph 0.90 size char {len('%d' % i)+1}, char 1.5 fc rgb 'red'") g(f"set object rect at '{label[0]}', graph 0.0 size char 0.5, char 0.5 front fc rgb 'red'") g(f"set label at '{label[0]}', graph 0.90 '{i}' center tc rgb 'black' font 'Courier-New,7'") g(f"set label at '{label[0]}', graph 0.95 '{label[1][0:20]}' center tc rgb 'white' font 'Courier-New,7'") if i > 0: g("set yrange [0:119]") else: g("set yrange [0:100]") g("set object rectangle from graph 0, graph 0 to graph 2, graph 2 behind fillcolor rgb '#111111' fillstyle solid noborder") g(f"set object rectangle from '{START_DATE.replace(' ', '-')}', 0 to '{END_DATE.replace(' ', '-')}', 100 behind fillcolor rgb '#000000' fillstyle solid noborder") g("plot 'data.txt' using 1:2:2 title 'cpu' with boxes palette") g("set ylabel 'ram % usage'") g(f"set title 'ram usage (max = {MAX_USED_RAM:.2f} GB)'"); g("plot 'data.txt' using 1:3:3 title 'ram' with boxes palette") g("set ylabel '%s %% usage'" % NAME_FS) g(f"set title '{NAME_FS} usage (max = {MAX_USED_FS:.2f} MB)'"); g("plot 'data.txt' using 1:4:4 title 'fs' with boxes palette") g("unset multiplot") g("unset output") g("quit")