bw_oscillation_graph.py

#!/usr/bin/python3
import sys
import numpy as np
import matplotlib.pyplot as plt
from scipy.signal import savgol_filter
from graphing.analysis.results_library import ResultsLibrary, TestResult
from python_utils.file_locations import results_dir

results = ResultsLibrary(results_dir)

#This is the filter function that match name of scheme?
def matching_func(test_result):
    print("Checking test result %s for %s" % (test_result.name, test_result.metadata["Scheme"]))
    return test_result.metadata["Scheme"] == sys.argv[1]

#test_result = results.get_all_results_matching("simple_1000pkt_queue", matching_func)[0]
test_result = results.get_all_results_matching("simple_10_to_20mbps_oscillation", matching_func)[0]
test_result.load()
print(test_result.flows)

other_flow_times = [float(event["Time"]) / 1000.0 for event in test_result.flows["flow_1"].data["Events"]]
other_flow_thpts = [float(event["Throughput"]) / 1000.0 for event in test_result.flows["flow_1"].data["Events"]]
other_flow_rates = [float(event["Target Rate"]) / 1000.0 for event in test_result.flows["flow_1"].data["Events"]]
plt.plot(other_flow_times, savgol_filter(other_flow_thpts, 13, 1), label= test_result.metadata["Scheme"])
plt.plot(other_flow_times, savgol_filter(other_flow_rates, 13, 1), label= test_result.metadata["Scheme"])

plt.legend()
plt.title("Oscillating Link (10 to 20mbps, 5s period)")
plt.xlabel("Time (s)")
plt.ylabel("Throughput (out of 30 Mbps)")
plt.show()