# Routines to manipulate EZ-Web files

import numpy as np

# Read summary

def read_summary (file) :

    table = np.loadtxt(file)

    col_names = ['i', 't', 'M', 'log_L', 'log_R', 'log_T_s', 
                 'log_T_c', 'log_rho_c', 'log_P_c', 'Psi_c', 
                 'X_c', 'Y_c', 'X_Cc', 'X_Nc', 'X_Oc', 
                 'tau_dyn', 'tau_KH', 'tau_nuc', 'L_PP',
                 'L_CNO', 'L_3a', 'L_Z', 'L_nu', 'M_He',
                 'M_C', 'M_O', 'R_He', 'R_C', 'R_O']

    data = dict()

    for i in range(0,len(col_names)):
        data[col_names[i]] = table[:,i]

    return data
                 
# Read summary

def read_structure (file) :

    table = np.loadtxt(file)

    col_names = ['M_r', 'r', 'L_r', 'P', 'rho', 'T', 'U', 'S', 
                 'C_P', 'Gamma_1', 'nabla_ad', 'mu', 'n_e', 
                 'P_e', 'P_r', 'nabla_rad', 'nabla', 'v_c',
                 'kappa', 'epsilon_nuc', 'epsilon_PP', 
                 'epsilon_CNO', 'epsilon_3a', 'epsilon_nunuc',
                 'epsilon_nu', 'epsilon_grav', 'X', 'X_mol',
                 'X_+', 'Y', 'Y_+', 'Y_++', 'X_C', 'X_N', 
                 'X_O', 'Psi']

    data = dict()

    for i in range(0,len(col_names)):
        data[col_names[i]] = table[:,i]

    return data