from pylab import *
from matplotlib import pyplot


solar_spectrum=loadtxt('/local/libRadtran-1.5-beta/data/solar_flux/kurudz_1.0nm.dat')

# irradiance from 300 to 500 nm
sw_spectrum=loadtxt('/home/users/lainer/Strahlung/tmp_ozone/uv0..out')
pyplot.plot(sw_spectrum[:,0], sw_spectrum[:,1], label='irradiance ozone_column=0') # change 1 to desired variable

pyplot.title('irradiance spectrum')
pyplot.xlabel('wavelength [nm]')
pyplot.ylabel('irradiance [mW/ (m^2 nm)]')
pyplot.legend(loc=4)

sw_spectrum1=loadtxt('/home/users/lainer/Strahlung/tmp_ozone/uv100..out')
pyplot.plot(sw_spectrum1[:,0], sw_spectrum1[:,1], label='irradiance ozone_column=100') # change 1 to desired variable

pyplot.title('irradiance spectrum, sza=30, alb=0.3')
pyplot.xlabel('wavelength [nm]')
pyplot.ylabel('irradiance [mW/ (m^2 nm)]')
pyplot.legend(loc=4)


sw_spectrum3=loadtxt('/home/users/lainer/Strahlung/tmp_ozone/uv200..out')
pyplot.plot(sw_spectrum3[:,0], sw_spectrum3[:,1], label='irradiance ozone_column=200') # change 1 to desired variable

pyplot.title('irradiance spectrum')
pyplot.xlabel('wavelength [nm]')
pyplot.ylabel('irradiance [mW/ (m^2 nm)]')
pyplot.legend(loc=4)


sw_spectrum5=loadtxt('/home/users/lainer/Strahlung/tmp_ozone/uv300..out')
pyplot.plot(sw_spectrum5[:,0], sw_spectrum5[:,1], label='irradiance ozone_column=300') # change 1 to desired variable

pyplot.title('irradiance spectrum, sza=30, alb=0.3')
pyplot.xlabel('wavelength [nm]')
pyplot.ylabel('irradiance [mW/ (m^2 nm)]')
pyplot.legend(loc=4)


pyplot.savefig('solar_irrad_uv.png')
show()