from numpy import *
import numpy as np
from matplotlib.pyplot import *
import matplotlib as mpl
mpl.rcParams['xtick.major.size'] = 8
mpl.rcParams['xtick.minor.size'] = 4
mpl.rcParams['xtick.major.width'] = 1.5
mpl.rcParams['xtick.minor.width'] = 1.5
mpl.rcParams['ytick.major.size'] = 8
mpl.rcParams['ytick.minor.size'] = 4
mpl.rcParams['ytick.major.width'] = 1.5
mpl.rcParams['ytick.minor.width'] = 1.5

ticksize=20
labelsize=22

data=loadtxt('hstof_1998_quiet_times.dat')

tof=data[:,0]
energy=data[:,1]
realenergy=data[:,2]
realmass=data[:,3]



H, xedges, yedges = np.histogram2d(tof,energy,bins=[np.arange(0.5,400.5,1),np.arange(0.5,200.5,1)])
H = np.rot90(H)
H = np.flipud(H)
Hmasked = np.ma.masked_where(H==0,H)


figure(figsize=(14,5))
subplot(121)
pcolormesh(xedges,yedges,Hmasked)


ylim(1,200)
xlim(1,400)
#yscale('log')
#xscale('log')

xticks(fontsize=ticksize)
yticks(fontsize=ticksize)
xlabel(r'$\tau_{\#}$',fontsize=labelsize)
ylabel('$E_{SSD\#}$',fontsize=labelsize)
tick_params(right='off',top='off',which='both')
#title('HSTOF',fontsize=labelsize)
show()
######################################################################################################


H, xedges, yedges = np.histogram2d(realenergy,realmass,bins=[np.logspace(1,4,100),np.logspace(-1,2,100)])
H = np.rot90(H)
H = np.flipud(H)
Hmasked = np.ma.masked_where(H==0,H)

subplot(122)
pcolormesh(xedges,yedges,Hmasked)


ylim(1e-1,1e2)
xlim(1e1,1e4)
#yscale('log')
#xscale('log')

xticks(fontsize=ticksize)
yticks(fontsize=ticksize)
xlabel('$E_{SSD}$ [keV]',fontsize=labelsize)
ylabel('Mass [amu]',fontsize=labelsize)
tick_params(right='off',top='off',which='both')
#title('HSTOF',fontsize=labelsize)
show()




tight_layout(pad=0.4,h_pad=0.01,w_pad=0.001)