Commit 1c20cf06 authored by Peter Jansweijer's avatar Peter Jansweijer

Extended 3D plot functions

parent 248cb513
......@@ -624,6 +624,8 @@ if __name__ == "__main__":
time_err = calc_alpha_error(alpha_3wlsel, alpha_3wl, crtt_fixed_lambda)
time_err_relative = time_err/crtt_fixed_lambda
#pdb.set_trace()
"""
# =====================================================
# Plot 3 wavelength alpha error
......@@ -683,7 +685,45 @@ if __name__ == "__main__":
alpha_file.write(str(itu_ch) + ", " + str(wavelength) + ", " + str(alpha_sellmeier[i]) + ", " + str (spec_alpha(alpha_sellmeier[i])) + ", " + str(alpha_tangent[i]) + ", " + str(alpha_sellmeier[i]- alpha_tangent[i]) + ", " + str(t_clean[i]) + "\n")
alpha_file.close()
# =====================================================
# Output 3 wavelength alpha_error values in file
# =====================================================
alpha_err_file = open(name+"_err.out","w")
alpha_err_file.write("file: " + insitu_file + "\n")
alpha_err_file.write("fixed_lambda: " + str(fixed_lambda) + "\n")
alpha_err_file.write("Sellmeier: \n")
alpha_err_file.write("A: " + str(popt[0]) + "\n")
alpha_err_file.write("B: " + str(popt[1]) + "\n")
alpha_err_file.write("C: " + str(popt[2]) + "\n")
alpha_err_file.write("D: " + str(popt[3]) + "\n")
alpha_err_file.write("E: " + str(popt[4]) + "\n")
alpha_err_file.write("min time_err: " + str(numpy.nanmin(time_err)) + "\n")
l1_idx,l2_idx = numpy.where(time_err==numpy.nanmin(time_err))
alpha_err_file.write(" @ l1: " + str(l2[l1_idx][0][0]) + "\n")
alpha_err_file.write(" @ l2: " + str(l2[l2_idx][0][0]) + "\n")
alpha_err_file.write("max time_err: " + str(numpy.nanmax(time_err)) + "\n")
l1_idx,l2_idx = numpy.where(time_err==numpy.nanmax(time_err))
alpha_err_file.write(" @ l1: " + str(l2[l1_idx][0][0]) + "\n")
alpha_err_file.write(" @ l2: " + str(l2[l2_idx][0][0]) + "\n")
rng = 1 # [ps]
alpha_err_file.write("time_err within range: " +str(rng) + " [ps]\n")
alpha_err_file.write(" @l1\tl2\n")
rng_sel = numpy.argwhere(numpy.logical_and(time_err<rng,time_err>-rng))
#pdb.set_trace()
for i in rng_sel:
alpha_err_file.write("l1: " + str(l2[i[0]][0]) + "l2: " + str(l2[i[1]][0]) + "\n")
alpha_err_file.write("============================\n")
alpha_err_file.write("l1[nm]\tl2[nm]\tcrtt_(l1)[ps]\tcrtt(l2)[ps]\talpha_sel(l1)\talpha_3wl(l1)\t time_err[ps]\ttime_relative_err[t_err ps/link_length ps]\n")
for l1_idx in range(len(x_clean)):
for l2_idx in range(len(x_clean)):
alpha_err_file.write(str(l1[l1_idx][l2_idx]) + "\t" + str(l2[l1_idx][l2_idx]) + "\t" + str(group_delay(l1[l1_idx][l2_idx],popt[0],popt[1],popt[2],popt[3],popt[4])) + "\t " + str(group_delay(l2[l1_idx][l2_idx],popt[0],popt[1],popt[2],popt[3],popt[4])) + "\t" + str(calc_alpha_sellmeier(l1[l1_idx][l2_idx]/1e9, fixed_lambda,popt[0],popt[1],popt[2],popt[3],popt[4])) + "\t " + str(alpha_3wl[l1_idx][l2_idx]) + "\t" + str(time_err[l1_idx][l2_idx]) + "\t" + str(time_err[l1_idx][l2_idx]/crtt_fixed_lambda) + "\n")
alpha_err_file.close()
#plt.subplot_tool()
plt.subplots_adjust(left=0.17)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment