from pylab import *

colors=['orange', 'g', 'b', 'r', 'purple', 'k']

phase_orig=loadtxt('phase_function_500nm_reff10.dat')

figure(1)
semilogy(phase_orig[:,0], phase_orig[:,1], 'grey', lw=2)

figure(2)
semilogy(phase_orig[:,0], phase_orig[:,1], 'grey', lw=2)

i=0
for nmom in ['8', '16', '32', '64', '128', '3000']: 
    
    figure(1)
    phase=loadtxt('phase'+nmom+'.dat')
    semilogy(phase[:,0], phase[:,1], colors[i], label=nmom)
    xlabel('theta [degress]')
    ylabel('phase function')
    
    figure(2)
    phase=loadtxt('phase'+nmom+'_ds.dat')
    semilogy(phase[:,0], phase[:,1], colors[i], label=nmom)
    xlabel('theta [degress]')
    ylabel('phase function')
    
    i+=1

figure(1)
legend()
savefig('phase_without_ds.png')

figure(2)
legend()
savefig('phase_with_ds.png')