# TODO: How long does it take to download an average track?
#   TODO: How large is an average mezzanine asset?
#       - since always WAV should be determinable from track time
#   TODO: How long does it take to download X bytes?
#       - can be gleaned from daemon asset copy experience?
# TODO: How many workers are there?
# TODO: Graph number of waiting jobs
# TODO: Can we differentiate ingestions?
# TODO: Model overhead with database transactions
# TODO: Add monitoring for the download time
# TODO: Add better labels to plot
from sims.dac.plotting import plot_histogram, plot_weekly_max, plot_weekly_mean
from sims.dac.simulation import run_simulations

if __name__ == '__main__':
    worker_counts = range(10, 100, 10)
    download_speeds = [500000, 1000000, 1500000, 2000000, 2500000]
    sims = run_simulations('run', worker_counts, download_speeds, [10], './data/dac_enhanced_year.pkl')

    for download_speed in [1000000, 2000000]:
        filtered_sims = [sim for sim in sims
                if sim.worker_count in [30, 60, 90]
                and sim.download_bytes_per_second == download_speed]
        plot_weekly_max(download_speed, filtered_sims)
        plot_weekly_mean(download_speed, filtered_sims)
        plot_histogram(download_speed, filtered_sims)