#!/usr/bin/env python
# coding: utf-8

# In[2]:


# Note:
# The data shown below was retrieved with RESDEPTH set to 1008 instead of 3 as used in the paper(s).
# This more conservative backfilling resulted only in minor differences (see week 7).

import pickle
import matplotlib
import matplotlib.pyplot as plt

get_ipython().run_line_magic('matplotlib', 'notebook')
for week in [6, 7, 29]:
    with open("week{}_util.pkl".format(week), 'rb') as pickle_file:
        all_plot = pickle.load(pickle_file)
        # Workaround until sharing axes works with PICKLE
        all_plot[0].get_shared_x_axes().join(all_plot[0], all_plot[1])

    with open("week{}_wait.pkl".format(week), 'rb') as pickle_file:
        all_plot = pickle.load(pickle_file)
        # Workaround until sharing axes works with PICKLE
        all_plot[0].get_shared_x_axes().join(all_plot[0], all_plot[1])
        all_plot[0].get_shared_y_axes().join(all_plot[0], all_plot[1])

        if(len(all_plot) == 4): # Only for the plot with 3 axes
            all_plot[0].get_shared_x_axes().join(all_plot[0], all_plot[2])
            all_plot[0].get_shared_y_axes().join(all_plot[0], all_plot[2])

    with open("week{}_cdf.pkl".format(week), 'rb') as pickle_file:
        all_plot = pickle.load(pickle_file)

    with open("week{}_error.pkl".format(week), 'rb') as pickle_file:
        all_plot = pickle.load(pickle_file)

with open("sim_times.pkl", 'rb') as pickle_file:
    all_plot = pickle.load(pickle_file)
    # Workaround until sharing axes works with PICKLE
    all_plot[0].get_shared_x_axes().join(all_plot[0], all_plot[1])


# In[ ]: