import matplotlib.cm as cm

all_temp_entropy_values = []
all_temp_entropy_values = []
all_temp_entropy_values = []

natural_mean_entropy_values = []
designed_mean_entropy_values_rosetta = []
designed_mean_entropy_values_evolved = []

protein_file_name = "graph_mean_data_natural.csv"
[
    natural_mean_RSA_values, natural_mean_entropy_values,
    natural_cor_entropy_RSA_values, natural_mean_split_KL_values,
    natural_cor_entropy_iwcn_values
] = af.get_mean_designed_data(protein_file_name)

protein_file_name = "graph_mean_data_rosetta.csv"
[
    designed_mean_RSA_values_rosetta, designed_mean_entropy_values_rosetta,
    designed_cor_entropy_RSA_values_rosetta, designed_mean_KL_values_rosetta,
    designed_cor_entropy_iwcn_values_rosetta
] = af.get_mean_designed_data(protein_file_name)

protein_file_name = "graph_mean_data_evolved.csv"
[
    designed_mean_RSA_values_evolved, designed_mean_entropy_values_evolved,
    designed_cor_entropy_RSA_values_evolved, designed_mean_KL_values_evolved,
    designed_cor_entropy_iwcn_values_evolved
] = af.get_mean_designed_data(protein_file_name)
all_method_data = []
all_method_mean_KL_data_array = [] 
for line in mean_KL_method_data:
    data = re.split("\t", line)
    data.pop(0)
    data_array = af.make_array(data)
    all_method_data.append(data_array)
    mean_KL_method_values_array = af.make_array(data)
all_method_mean_KL_data_array = array(all_method_data)

# effective number data
all_temp_entropy_values = []

protein_file_name = "graph_mean_data_natural.csv"
[natural_mean_RSA_values, natural_mean_entropy_values, natural_cor_entropy_RSA_values, natural_mean_split_KL_values, natural_cor_entropy_icn_values, natural_cor_entropy_iwcn_values] = af.get_mean_designed_data(protein_file_name)
   
protein_file_name = "graph_mean_data_rosetta.csv"
[designed_mean_RSA_values_rosetta, designed_mean_entropy_values_rosetta, designed_cor_entropy_RSA_values_rosetta, designed_mean_KL_values_rosetta, designed_cor_entropy_icn_values_rosetta, designed_cor_entropy_iwcn_values_rosetta] = af.get_mean_designed_data(protein_file_name)

  
protein_file_name = "graph_mean_data_evolved.csv"
[designed_mean_RSA_values_evolved, designed_mean_entropy_values_evolved, designed_cor_entropy_RSA_values_evolved, designed_mean_KL_values_evolved, designed_cor_entropy_icn_values_evolved, designed_cor_entropy_iwcn_values_evolved] = af.get_mean_designed_data(protein_file_name)


all_temp_entropy_values.append(designed_mean_entropy_values_rosetta)
all_temp_entropy_values.append(designed_mean_entropy_values_evolved)
all_temp_entropy_values.append(natural_mean_entropy_values)

all_temp_entropy_values_array = []
for data in natural_data:
    pdb_names.append(data[0])
    chain_names.append(data[1])
    natural_mean_RSA_values.append(data[2])
    natural_mean_entropy_values.append(data[3])
    natural_cor_entropy_RSA_values.append(data[4])
    natural_mean_split_KL_values.append(data[5])

natural_mean_RSA_values_array = analysis_functions.make_array(natural_mean_RSA_values)
natural_mean_entropy_values_array = analysis_functions.make_array(natural_mean_entropy_values)
natural_cor_entropy_RSA_values_array = analysis_functions.make_array(natural_cor_entropy_RSA_values)
natural_mean_split_KL_values_array = analysis_functions.make_array(natural_mean_split_KL_values)

protein_file_name = "graph_mean_data_0.0_noah.csv"
[designed_mean_RSA_values_00, designed_mean_entropy_values_00, designed_cor_entropy_RSA_values_00, designed_mean_KL_values_00] = analysis_functions.get_mean_designed_data(protein_file_name)

designed_mean_RSA_values_array_00 = array(designed_mean_RSA_values_00)
designed_mean_entropy_values_array_00 = array(designed_mean_entropy_values_00)
designed_cor_entropy_RSA_values_array_00 = array(designed_cor_entropy_RSA_values_00)  
designed_mean_KL_values_array_00 = array(designed_mean_KL_values_00)

protein_file_name = "graph_mean_data_0.3_noah.csv"
[designed_mean_RSA_values_03, designed_mean_entropy_values_03, designed_cor_entropy_RSA_values_03, designed_mean_KL_values_03] = analysis_functions.get_mean_designed_data(protein_file_name)

designed_mean_RSA_values_array_03 = array(designed_mean_RSA_values_03)
designed_mean_entropy_values_array_03 = array(designed_mean_entropy_values_03)
designed_cor_entropy_RSA_values_array_03 = array(designed_cor_entropy_RSA_values_03) 
designed_mean_KL_values_array_03 = array(designed_mean_KL_values_03)

protein_file_name = "graph_mean_data_0.6_noah.csv"