#import AngularSeparation
from astropy import coordinates as coord
import time
import matplotlib
matplotlib.use('Agg')
import matplotlib.pyplot as p
import numpy as n
import os
import sys
import ClusterScalingRelations as clsr
from scipy.interpolate import interp1d
import StellarMass as sm
smhmr = sm.StellarMass()
scl = clsr.ClusterScalingRelations_Mantz2016()
cat = fits.open(os.path.join(os.environ['DATA_DIR'], 'spiders', 'cluster', 'CODEX-DR14-MergedSpectroscopicCatalogue_2018-02-05.fits'))[1].data
all_clus_id = n.array(list(set(cat['CLUS_ID'])))
all_clus_id.sort()
#spm26 = fits.open(os.path.join(os.environ['DATA_DIR'], 'spiders', 'cluster', 'CODEX-DR14-MergedSpectroscopicCatalogue_2018-02-05-spm26.fits'))[1].data
#spmV5 = fits.open(os.path.join(os.environ['DATA_DIR'], 'spiders', 'cluster', 'CODEX-DR14-MergedSpectroscopicCatalogue_2018-02-05-spmv5.fits'))[1].data
spmV5 = fits.open('/home/comparat/data/spm/firefly/FireflyGalaxyEbossDR14_Chabrier_MILES.fits')[1].data
p.xlim((9.5, 12.5))
p.ylim((-6,-2))
p.grid()
p.legend(loc=0, frameon=False)
p.savefig('/home/comparat/data/eRoMok/BO12_fig3.png')
p.clf()
sys.exit()
print " set up box, and redshift "
#MD 1 hlist_0.74980_SAM_Nb_0.fits
#MD 25 hlist_0.75440_SAM_Nb_10.fits
z = 1./0.74230 -1.
# set up the stellar mass computation
sm = StellarMass.StellarMass()
mhs = n.logspace(10,15,100)
Nhalo = len(mhs)
ratio = sm.SMHMr(mhs,0.)
stellar_mass = sm.meanSM(mhs,0.)
# set up the x ray lambda SAR
logMs = n.arange(6.5,12.5,0.01)
cdfs_interpolations = []
cdfs_interpolations_maxs = []
XXS = n.arange(32,36.1,0.1)
for jj,mass in enumerate(logMs):
pd = lambda ll : xr.psi_log(ll, logM=mass, z=z)
norming = quad( pd, 32, 36)[0]
cdfs_interpolations.append( interp1d(n.array([quad( pd, 32, X)[0] for X in XXS ])/norming, XXS) )