import h5py
import matplotlib.pyplot as plt
import pandas as pd
import seaborn as sns

with h5py.File("Multiple-Fig-Data/synthetic_observed_samples/o1o2o3_mass_c_iid_mag_two_comp_iid_tilt_powerlaw_redshift_full_posterior.hdf5", "r") as ff:
    data = pd.DataFrame(ff["samples"][:])

sns.jointplot(data=data, x="S190521g_mass_1", y="mmax", kind="kde")
plt.savefig('190521.png')
plt.plot()
plt.close()

data["S190924h_mass_2"] = data["S190924h_mass_1"] * data["S190924h_mass_ratio"]
sns.jointplot(data=data, x="S190924h_mass_2", y="mmin", kind="kde")
plt.plot()
plt.savefig('190924.png')
plt.close()