# -*- coding: utf-8 -*-
import matplotlib.pyplot as plt
import numpy as np

#read data

H_E, H_coh, H_incoh, H_photo, H_nucl, H_elect, H_totwc, H_totwo = np.loadtxt('X-section-H.dat', skiprows=4, unpack=True)
C_E, C_coh, C_incoh, C_photo, C_nucl, C_elect, C_totwc, C_totwo = np.loadtxt('X-section-C.dat', skiprows=4, unpack=True)
Si_E, Si_coh, Si_incoh, Si_photo, Si_nucl, Si_elect, Si_totwc, Si_totwo = np.loadtxt('X-section-Si.dat', skiprows=4, unpack=True)
Fe_E, Fe_coh, Fe_incoh, Fe_photo, Fe_nucl, Fe_elect, Fe_totwc, Fe_totwo = np.loadtxt('X-section-Fe.dat', skiprows=4, unpack=True)
Pb_E, Pb_coh, Pb_incoh, Pb_photo, Pb_nucl, Pb_elect, Pb_totwc, Pb_totwo = np.loadtxt('X-section-Pb.dat', skiprows=4, unpack=True)
U_E, U_coh, U_incoh, U_photo, U_nucl, U_elect, U_totwc, U_totwo = np.loadtxt('X-section-U.dat', skiprows=4, unpack=True)


#fig, ((ax1, ax2, ax3), (ax4, ax5, ax6)) = plt.subplots(3, 2, sharex='col', sharey='row')
fig, ((ax1, ax2), (ax3, ax4), (ax5, ax6)) = plt.subplots(3, 2, sharex='col', sharey='row')
#fig, ax1 = plt.subplots()

ax1.set_xscale('log')
ax1.set_xlabel('E [MeV]')
ax1.set_ylabel(u'Verhältnis')
ax1.plot(H_E, H_photo/H_totwc, c='k', label='Photo')
ax1.plot(H_E, H_incoh/H_totwc, c='r', label='Comp.')
ax1.plot(H_E, H_nucl/H_totwc, c='b', label='el. P.')
ax1.plot(H_E, H_elect/H_totwc, c='b', ls = 'dashed', label='nu. P.')
ax1.title.set_text('H')
ax1.legend(loc='center right')

ax2.set_xscale('log')
ax2.set_xlabel('E [MeV]')
ax2.set_ylabel(u'Verhältnis')
ax2.plot(C_E, C_photo/C_totwc, c='k', label='Photo')
ax2.plot(C_E, C_incoh/C_totwc, c='r', label='Comp.')
ax2.plot(C_E, C_nucl/C_totwc, c='b', label='nu. P.')
ax2.plot(C_E, C_elect/C_totwc, c='b', ls = 'dashed', label='el. P.')
ax2.title.set_text('C')
ax2.legend(loc='center right')

ax3.set_xscale('log')
ax3.set_xlabel('E [MeV]')
ax3.set_ylabel(u'Verhältnis')
ax3.plot(Si_E, Si_photo/Si_totwc, c='k', label='Photo')
ax3.plot(Si_E, Si_incoh/Si_totwc, c='r', label='Comp.')
ax3.plot(Si_E, Si_nucl/Si_totwc, c='b', label='nu. P.')
ax3.plot(Si_E, Si_elect/Si_totwc, c='b', ls = 'dashed', label='el. P.')
ax3.title.set_text('Si')
ax3.legend(loc='center right')

ax4.set_xscale('log')
ax4.set_xlabel('E [MeV]')
ax4.set_ylabel(u'Verhältnis')
ax4.plot(Fe_E, Fe_photo/Fe_totwc, c='k', label='Photo')
ax4.plot(Fe_E, Fe_incoh/Fe_totwc, c='r', label='Comp.')
ax4.plot(Fe_E, Fe_nucl/Fe_totwc, c='b', label='nu. P.')
ax4.plot(Fe_E, Fe_elect/Fe_totwc, c='b', ls = 'dashed', label='el. P.')
ax4.title.set_text('Fe')
ax4.legend(loc='center right')

ax5.set_xscale('log')
ax5.set_xlabel('E [MeV]')
ax5.set_ylabel(u'Verhältnis')
ax5.plot(Pb_E, Pb_photo/Pb_totwc, c='k', label='Photo')
ax5.plot(Pb_E, Pb_incoh/Pb_totwc, c='r', label='Comp.')
ax5.plot(Pb_E, Pb_nucl/Pb_totwc, c='b', label='nu. P.')
ax5.plot(Pb_E, Pb_elect/Pb_totwc, c='b', ls = 'dashed', label='el. P.')
ax5.title.set_text('Pb')
ax5.legend(loc='center right')

ax6.set_xscale('log')
ax6.set_xlabel('E [MeV]')
ax6.set_ylabel(u'Verhältnis')
ax6.plot(U_E, U_photo/U_totwc, c='k', label='Photo')
ax6.plot(U_E, U_incoh/U_totwc, c='r', label='Comp.')
ax6.plot(U_E, U_nucl/U_totwc, c='b', label='nu. P.')
ax6.plot(U_E, U_elect/U_totwc, c='b', ls = 'dashed', label='el. P.')
ax6.title.set_text('U')
ax6.legend(loc='center right')

plt.savefig('vergleich-querschnitte-2.png',dpi=300)
plt.show()