
import numpy as np
from numpy import nan, inf

add_legend_handle = [
  'Data',
  'Analysis_CSS_FragNone_UENone.yoda.gz',
  'Analysis_Alaric_FragNone_UENone.yoda.gz'
]

xpoints = {
                                       'Data' : [2.500000e-02, 7.500000e-02, 1.250000e-01, 1.750000e-01, 2.250000e-01,
                                                 2.750000e-01, 3.250000e-01, 3.750000e-01, 4.250000e-01, 4.750000e-01],
       'Analysis_CSS_FragNone_UENone.yoda.gz' : [2.500000e-02, 7.500000e-02, 1.250000e-01, 1.750000e-01, 2.250000e-01,
                                                 2.750000e-01, 3.250000e-01, 3.750000e-01, 4.250000e-01, 4.750000e-01],
    'Analysis_Alaric_FragNone_UENone.yoda.gz' : [2.500000e-02, 7.500000e-02, 1.250000e-01, 1.750000e-01, 2.250000e-01,
                                                 2.750000e-01, 3.250000e-01, 3.750000e-01, 4.250000e-01, 4.750000e-01],
}

xedges = {
                                       'Data' : [0.000000e+00, 5.000000e-02, 1.000000e-01, 1.500000e-01, 2.000000e-01,
                                                 2.500000e-01, 3.000000e-01, 3.500000e-01, 4.000000e-01, 4.500000e-01,
                                                 5.000000e-01],
       'Analysis_CSS_FragNone_UENone.yoda.gz' : [0.000000e+00, 5.000000e-02, 1.000000e-01, 1.500000e-01, 2.000000e-01,
                                                 2.500000e-01, 3.000000e-01, 3.500000e-01, 4.000000e-01, 4.500000e-01,
                                                 5.000000e-01],
    'Analysis_Alaric_FragNone_UENone.yoda.gz' : [0.000000e+00, 5.000000e-02, 1.000000e-01, 1.500000e-01, 2.000000e-01,
                                                 2.500000e-01, 3.000000e-01, 3.500000e-01, 4.000000e-01, 4.500000e-01,
                                                 5.000000e-01],
}

ref_xerrs = [
  [abs(xpoints['Data'][i]   - xedges['Data'][i]) for i in range(len(xpoints['Data']))],
  [abs(xedges['Data'][i+1] - xpoints['Data'][i]) for i in range(len(xpoints['Data']))]
]

yvals = {
                                       'Data' : [1.418460e+01, 1.734490e+00, 1.072510e+00, 7.969180e-01, 4.632810e-01,
                                                 3.158580e-01, 3.577910e-01, 4.144960e-01, 3.961030e-01, 2.639410e-01],
       'Analysis_CSS_FragNone_UENone.yoda.gz' : [7.283902e+00, 4.619294e+00, 1.825879e+00, 2.639234e+00, 9.789865e-01,
                                                 3.998190e-01, -7.067739e-01, 9.569856e-01, 8.071658e-01, 1.195508e+00],
    'Analysis_Alaric_FragNone_UENone.yoda.gz' : [5.372494e+00, 8.216771e+00, 2.665406e+00, 2.578351e+00, 7.863272e-01,
                                                 -1.238451e-01, -7.051840e-01, 1.093520e+00, -2.616214e-01, 3.777809e-01],
}

xerrs = {
                                       'Data' : [
                                                   [2.500000e-02, 2.500000e-02, 2.500000e-02, 2.500000e-02, 2.500000e-02,
                                                    2.500000e-02, 2.500000e-02, 2.500000e-02, 2.500000e-02, 2.500000e-02],
                                                   [2.500000e-02, 2.500000e-02, 2.500000e-02, 2.500000e-02, 2.500000e-02,
                                                    2.500000e-02, 2.500000e-02, 2.500000e-02, 2.500000e-02, 2.500000e-02],
                                                ],
       'Analysis_CSS_FragNone_UENone.yoda.gz' : [
                                                   [2.500000e-02, 2.500000e-02, 2.500000e-02, 2.500000e-02, 2.500000e-02,
                                                    2.500000e-02, 2.500000e-02, 2.500000e-02, 2.500000e-02, 2.500000e-02],
                                                   [2.500000e-02, 2.500000e-02, 2.500000e-02, 2.500000e-02, 2.500000e-02,
                                                    2.500000e-02, 2.500000e-02, 2.500000e-02, 2.500000e-02, 2.500000e-02],
                                                ],
    'Analysis_Alaric_FragNone_UENone.yoda.gz' : [
                                                   [2.500000e-02, 2.500000e-02, 2.500000e-02, 2.500000e-02, 2.500000e-02,
                                                    2.500000e-02, 2.500000e-02, 2.500000e-02, 2.500000e-02, 2.500000e-02],
                                                   [2.500000e-02, 2.500000e-02, 2.500000e-02, 2.500000e-02, 2.500000e-02,
                                                    2.500000e-02, 2.500000e-02, 2.500000e-02, 2.500000e-02, 2.500000e-02],
                                                ],
}

yerrs = {
                                       'Data' : [
                                                   [1.502705e+00, 9.144475e-01, 5.053420e-01, 2.848183e-01, 2.177671e-01,
                                                    1.580750e-01, 1.148605e-01, 1.688193e-01, 1.599732e-01, 1.191735e-01],
                                                   [1.502705e+00, 9.144475e-01, 5.053420e-01, 2.848183e-01, 2.177671e-01,
                                                    1.580750e-01, 1.148605e-01, 1.688193e-01, 1.599732e-01, 1.191735e-01],
                                                ],
       'Analysis_CSS_FragNone_UENone.yoda.gz' : [
                                                   [2.182587e+00, 1.830287e+00, 1.164483e+00, 9.974935e-01, 1.243315e+00,
                                                    8.007908e-01, 8.103715e-01, 5.940806e-01, 8.384830e-01, 7.806217e-01],
                                                   [2.182587e+00, 1.830287e+00, 1.164483e+00, 9.974935e-01, 1.243315e+00,
                                                    8.007908e-01, 8.103715e-01, 5.940806e-01, 8.384830e-01, 7.806217e-01],
                                                ],
    'Analysis_Alaric_FragNone_UENone.yoda.gz' : [
                                                   [2.175975e+00, 2.548745e+00, 1.923143e+00, 1.627360e+00, 9.463378e-01,
                                                    1.049239e+00, 7.438148e-01, 8.309514e-01, 7.486303e-01, 9.800895e-01],
                                                   [2.175975e+00, 2.548745e+00, 1.923143e+00, 1.627360e+00, 9.463378e-01,
                                                    1.049239e+00, 7.438148e-01, 8.309514e-01, 7.486303e-01, 9.800895e-01],
                                                ],
}

variation_yvals = {
}



# lists for ratio plot
ratio0_yvals = {
                                       'Data' : [1.000000e+00, 1.000000e+00, 1.000000e+00, 1.000000e+00, 1.000000e+00,
                                                 1.000000e+00, 1.000000e+00, 1.000000e+00, 1.000000e+00, 1.000000e+00],
       'Analysis_CSS_FragNone_UENone.yoda.gz' : [5.135077e-01, 2.663200e+00, 1.702435e+00, 3.311801e+00, 2.113159e+00,
                                                 1.265819e+00, -1.975382e+00, 2.308793e+00, 2.037767e+00, 4.529452e+00],
    'Analysis_Alaric_FragNone_UENone.yoda.gz' : [3.787554e-01, 4.737284e+00, 2.485204e+00, 3.235403e+00, 1.697301e+00,
                                                 -3.920911e-01, -1.970938e+00, 2.638192e+00, -6.604883e-01, 1.431308e+00],
}

ratio0_yerrs = {
                                       'Data' : [
                                                   [1.059392e-01, 5.272141e-01, 4.711770e-01, 3.573998e-01, 4.700540e-01,
                                                    5.004622e-01, 3.210268e-01, 4.072881e-01, 4.038677e-01, 4.515157e-01],
                                                   [1.059392e-01, 5.272141e-01, 4.711770e-01, 3.573998e-01, 4.700540e-01,
                                                    5.004622e-01, 3.210268e-01, 4.072881e-01, 4.038677e-01, 4.515157e-01],
                                                ],
       'Analysis_CSS_FragNone_UENone.yoda.gz' : [
                                                   [1.538702e-01, 1.055231e+00, 1.085755e+00, 1.251689e+00, 2.683717e+00,
                                                    2.535287e+00, 2.264930e+00, 1.433260e+00, 2.116831e+00, 2.957561e+00],
                                                   [1.538702e-01, 1.055231e+00, 1.085755e+00, 1.251689e+00, 2.683717e+00,
                                                    2.535287e+00, 2.264930e+00, 1.433260e+00, 2.116831e+00, 2.957561e+00],
                                                ],
    'Analysis_Alaric_FragNone_UENone.yoda.gz' : [
                                                   [1.534040e-01, 1.469449e+00, 1.793124e+00, 2.042067e+00, 2.042686e+00,
                                                    3.321869e+00, 2.078909e+00, 2.004727e+00, 1.889989e+00, 3.713290e+00],
                                                   [1.534040e-01, 1.469449e+00, 1.793124e+00, 2.042067e+00, 2.042686e+00,
                                                    3.321869e+00, 2.078909e+00, 2.004727e+00, 1.889989e+00, 3.713290e+00],
                                                ],
}

ratio0_variation_vals = {
}

ratio_band_edges = {
}
