from astropy.io import fits
import matplotlib.pyplot as plt
import matplotlib.gridspec as gridspec

plt.rcParams['xtick.direction'] = 'in'
plt.rcParams['ytick.direction'] = 'in'
plt.rcParams['xtick.top'] = True
plt.rcParams['ytick.right'] = True
plt.rcParams['ps.fonttype'] = 42
plt.rcParams['pdf.fonttype'] = 42

plt.rc('text', usetex=True)
plt.rc('font', family='serif')
plt.rcParams['font.size'] = 16



scale = 0.247
fig = plt.figure()
max_xpos = 36 * scale
fig.set_size_inches(5, 6)
gs = gridspec.GridSpec(2, 1, top=0.96, hspace=0, wspace = 0.0, left=0.15, right=0.98)
fitfile = '/home/chil/work/Data/pro/Binospec/lsb_psg_final/coma1/587741722286948519/results/sn010/ppxf_results_30_l15.fits'
hdul = fits.open(fitfile)
fit_data = hdul[1].data
ax_sig = fig.add_subplot(gs[1,0])
ax_vel = fig.add_subplot(gs[0,0])
ax_vel.plot((fit_data['XPOS'] - fit_data['XPOS'][0])*scale, fit_data['V'] - fit_data['V'][0], 'ro')
ax_sig.plot((fit_data['XPOS'] - fit_data['XPOS'][0])*scale, fit_data['SIG'], 'ro')
ax_sig.set_xlim(-max_xpos, max_xpos)
ax_vel.set_xlim(-max_xpos, max_xpos)
ax_vel.set_ylabel(r'$v, km/s$')
ax_sig.set_ylabel(r'$\sigma, km/s$')
ax_sig.set_xlabel('position, kpc')
plt.setp(ax_vel.get_xticklabels(),visible=False)
fig.savefig('519_hst.pdf')
print(fit_data['V'], fit_data['XPOS'])