co2vmr, MM, temp, n_alts_trlo=n_alts_trlo, n_alts_trhi=n_top) hr_ref = all_coeffs_nlte[(atm, cco2, 'hr_nlte')] hr_ref[:n_alts_lte] = all_coeffs_nlte[(atm, cco2, 'hr_lte')][:n_alts_lte] ovmr = all_coeffs_nlte[(atm, cco2, 'o_vmr')] o2vmr = all_coeffs_nlte[(atm, cco2, 'o2_vmr')] n2vmr = all_coeffs_nlte[(atm, cco2, 'n2_vmr')] alt_fomi, hr_fomi = npl.old_param(alts, temp, pres, co2vmr, Oprof=ovmr, O2prof=o2vmr, N2prof=n2vmr) oldco = spline(alt_fomi, hr_fomi) hr_fomi = oldco(alts) ### FIGURE tit = 'co2: {} - atm: {}'.format(cco2, atm) xlab = 'CR (K/day)' ylab = 'Alt (km)' # labels = ['nlte_ref', 'new_param', 'np_wutop', 'np_all_wutop', 'np_aw_extended', 'np_noalpha', 'old param'] # hrs = [hr_ref, hr_calc, hr_calc_wutop, hr_calc_all, hr_calc_extended, hr_calc_alpha1, hr_fomi] labels = ['nlte_ref', 'new_param', 'old_param'] hrs = [hr_ref, hr_calc, hr_fomi]
hr_calc_extended = npl.recformula(alpha_dic[(n_top, 'L_esc_all_wutop', 'least_squares', cco2)], L_esc_all_wutop, lamb, hr_calc, co2vmr, MM, temp, n_alts_trlo=n_alts_trlo, n_alts_trhi=n_top) hr_ref = all_coeffs_nlte[(atm, cco2, 'hr_nlte')] hr_ref[:n_alts_lte] = all_coeffs_nlte[(atm, cco2, 'hr_lte')][:n_alts_lte] alt_fomi, hr_fomi = npl.old_param(alts, temp, pres, co2vmr) oldco = spline(alt_fomi, hr_fomi) hr_fomi = oldco(alts) tit = 'co2: {} - atm: {}'.format(cco2, atm) xlab = 'CR (K/day)' ylab = 'Alt (km)' # labels = ['nlte_ref', 'new_param', 'np_wutop', 'np_all_wutop', 'np_aw_extended', 'np_noalpha', 'old param'] # hrs = [hr_ref, hr_calc, hr_calc_wutop, hr_calc_all, hr_calc_extended, hr_calc_alpha1, hr_fomi] labels = [ 'nlte_ref', 'np_all_wutop', 'np_aw_extended', 'np_w_lfomi', 'old param' ] hrs = [hr_ref, hr_calc_all, hr_calc_extended, hr_calc_fom, hr_fomi] colors = np.array(npl.color_set(6))[np.arange(6) != 4]
tit = 'co2: {} - atm: {}'.format(cco2, atm) xlab = 'CR (K/day)' ylab = 'Alt (km)' hr_ref = all_coeffs[(atm, cco2, 'hr_ref')][:n_alts] hr_calc_uni = npl.hr_from_ab(acoeff_cco2_uni, bcoeff_cco2_uni, asurf_cco2_uni, bsurf_cco2_uni, temp, surf_temp)[:n_alts] hr_calc_var = npl.hr_from_ab(acoeff_cco2_var, bcoeff_cco2_var, asurf_cco2_var, bsurf_cco2_var, temp, surf_temp)[:n_alts] pres = atm_pt[(atm, 'pres')] print(np.median(co2pr)) alt_fomi, hr_fomi = npl.old_param(all_alts, temp, pres, co2pr) oldco = spline(alt_fomi, hr_fomi) hr_fomi = oldco(alts) hr_calc = npl.new_param_LTE(interp_coeffs, temp, co2pr, surf_temp=surf_temp)[:n_alts] hrs = [hr_ref, hr_fomi, hr_calc] labels = ['ref', 'old_param', 'new_param'] colors = ['black', 'steelblue', 'orange'] fig, a0, a1 = npl.manuel_plot(alts, hrs, labels, xlabel=xlab,