import numpy as np
import matplotlib.pyplot as plt
from exo_skryer.vert_Tp import MandS
from exo_skryer.data_constants import bar

# Create pressure grid
nlev = 100
p_bot = np.log10(100.0)
p_top = np.log10(1e-6)
p_lev = np.logspace(p_bot, p_top, nlev) * bar

# Example parameters
params = {
    "a1": 0.51,
    "a2": 0.51,
    "log_10_P1": -4.0,
    "log_10_P2": -2.0,
    "log_10_P3": 1.0,
    "T_ref": 1000.0
}
T_lev, T_lay = MandS(p_lev, params)

# Plot
fig, ax = plt.subplots(figsize=(10, 5))
ax.semilogy(T_lev, p_lev/bar, c='mediumvioletred')
ax.set_xlabel('Temperature [K]', fontsize=16)
ax.set_ylabel('pressure [bar]', fontsize=16)
ax.set_title('Madhusudhan & Seager (2009) T-p Profile', fontsize=14)
ax.tick_params(labelsize=14)
ax.invert_yaxis()
plt.tight_layout()