import pandas as pd
import numpy as np
from matplotlib import pyplot as plt

epochs = ['baseline','sdb','post']
nn = 50

def panel(csvpath):

    def process_phase( df, epoch ):
        dfe = df[df['epoch']==epoch]
        phase_median = dfe['phase'].median()
        phase_min = dfe['phase'].min()
        phase_max = dfe['phase'].max()

        return phase_median, phase_min, phase_max

    df = pd.read_csv(csvpath)
    colours = ['C0','C2','C1']
    my_zip = zip(epochs,colours,[0.2,0.5,0.8])
    plt.polar(0,1,color = [1,1,1])
    for epoch, clr, aa in my_zip:
        phase_median, phase_min, phase_max = process_phase(df,epoch)
        arc = np.linspace(phase_min,phase_max,nn)
        plt.polar(arc,[aa]*nn,
                  color = clr,
                  linewidth = 30,
                  alpha = 0.4)
        plt.polar(phase_median, aa, 'o',
                  color = clr)

    plt.show()



if __name__ == '__main__':
    panel('csv/model2_avgbp.csv')