classy-szfast 0.0.25.post5__py3-none-any.whl → 0.0.25.post7__py3-none-any.whl

classy_szfast/classy_szfast.py

@@ -283,9 +283,38 @@ class Class_szfast(object):
         params_values['Omega0_r'] = params_values['Omega0_ur']+params_values['Omega0_g']
         params_values['Omega0_m_nonu'] = params_values['Omega0_m'] - params_values['Omega0_ncdm']
         params_values['Omega0_cb'] = params_values['Omega0_m_nonu'] 
+        H0 = params_values['H0']*1.e3/Const._c_
+        params_values['Rho_crit_0'] = (3./(8.*self.pi*Const._G_*Const._M_sun_))*pow(Const._Mpc_over_m_,1)*self.pow(Const._c_,2)*self.pow(H0,2)/self.pow(params_values['h'],2)
         return params_values
 
 
+    def get_rho_crit_at_z(self,z,params_values_dict=None):
+        params_values = self.get_all_relevant_params(params_values_dict)
+        self.calculate_hubble(**params_values_dict)
+        H = self.hz_interp(z)
+        rho_crit = (3./(8.*self.pi*Const._G_*Const._M_sun_))*pow(Const._Mpc_over_m_,1)*self.pow(Const._c_,2)*self.pow(H,2)/self.pow(params_values['h'],2)
+        return rho_crit
+
+    def get_r_delta_of_m_delta_at_z(self,delta,m_delta,z,params_values_dict=None):
+        if params_values_dict:
+            params_values = params_values_dict.copy()
+        else:
+            params_values = self.params_for_emulators
+        rho_crit = self.get_rho_crit_at_z(z,params_values_dict=params_values_dict)
+        return (m_delta*3./4./self.pi/delta/rho_crit)**(1./3.)
+
+    def get_nu_at_z_and_m(self,z,m,params_values_dict=None):
+        if params_values_dict:
+            params_values = params_values_dict.copy()
+        else:
+            params_values = self.params_for_emulators
+
+        delta_c =  (3./20.)*self.pow(12.*self.pi,2./3.) # this is = 1.686470199841145
+        # note here we dont use matter dependent delta_c
+        # which would be multiplied by (1.+0.012299*log10(pvecback[pba->index_bg_Omega_m]));
+        sigma = 1. 
+        return (delta_c/sigma)**2 
+
 
     def find_As(self,params_cp):
 

classy_szfast/config.py

@@ -4,7 +4,7 @@ import get_cosmopower_emus
 import class_sz_data
 
 def get_cosmopower_path():
-    get_cosmopower_emus.set()
+    get_cosmopower_emus.set_path()
     return os.getenv('PATH_TO_CLASS_SZ_DATA')
 
 path_to_class_sz_data = get_cosmopower_path()

classy_szfast/cosmopower_jax.py

@@ -22,8 +22,8 @@ cp_s8_nn_jax = {}
 
 
 class CosmoPowerJAX_custom(CPJ):
-    def __init__(self, *args, **kwargs):
-        super().__init__(*args, **kwargs)
+    def __init__(self, verbose=False, *args, **kwargs):
+        super().__init__(verbose=verbose, *args, **kwargs)
         self.ten_to_predictions = True
         if 'ten_to_predictions' in kwargs.keys():
             self.ten_to_predictions = kwargs['ten_to_predictions']

classy_szfast/utils.py

@@ -69,4 +69,5 @@ class Const:
 
     # parameters entering in Stefan-Boltzmann constant sigma_B 
     _k_B_ = 1.3806504e-23
-    _h_P_ = 6.62606896e-34
+    _h_P_ = 6.62606896e-34
+    _M_sun_ =  1.98855e30 # solar mass in kg

{classy_szfast-0.0.25.post5.dist-info → classy_szfast-0.0.25.post7.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: classy_szfast
-Version: 0.0.25.post5
+Version: 0.0.25.post7
 Summary: The accelerator of the class_sz code from https://github.com/CLASS-SZ
 Maintainer-email: Boris Bolliet <bb667@cam.ac.uk>
 License: MIT

{classy_szfast-0.0.25.post5.dist-info → classy_szfast-0.0.25.post7.dist-info}/RECORD

@@ -1,20 +1,20 @@
 classy_szfast/__init__.py,sha256=E2thrL0Z9oXFfdzwcsu-xbOytudLFTlRlPqVFGlPPPg,279
 classy_szfast/classy_sz.py,sha256=QmbwrSXInQLMvCDqsr7KPmtaU0KOiOt1Rb-cTKuulZw,22240
-classy_szfast/classy_szfast.py,sha256=Fajoc14gdED3LpY0dMFhh1A_NzRqFVCCIof5pElCA4Y,39791
-classy_szfast/config.py,sha256=cd7Z62-qnX_4FJWfUNqcyJVh-AdBiXrF8DcQGpyAUZM,274
+classy_szfast/classy_szfast.py,sha256=bYxvdGz94Nk9zPwtMrEp0CGcn-nPayrZDH0q47HAKbg,41296
+classy_szfast/config.py,sha256=v6DGcBHmfn5JtuO48dKyXCh-Dmn0uwOF_izvVOJFnqw,279
 classy_szfast/cosmopower.py,sha256=ooYK2BDOZSo3XtGHfPtjXHxr5UW-yVngLPkb5gpvTx8,2351
-classy_szfast/cosmopower_jax.py,sha256=NvwY1a9YoizOV0zKp0SEtiHQkhWRO-seGFzO3FIzLl0,5436
+classy_szfast/cosmopower_jax.py,sha256=NqU_Sw5x0qw16DNHrN4ipCZP8HVdQvUiCq_Y04h8zoo,5468
 classy_szfast/cosmosis_classy_szfast_interface.py,sha256=zAnxvFtn73a5yS7jgs59zpWFEYKCIQyraYPs5hQ4Le8,11483
 classy_szfast/emulators_meta_data.py,sha256=mXG5LQuJw9QBNE_kxXW8Kx0AUCWpbV6uRO9BaBbIfHo,10732
 classy_szfast/pks_and_sigmas.py,sha256=drtuujE1HhlrYY1hY92DyY5lXlYS1uE15MSuVI4uo6k,6625
 classy_szfast/restore_nn.py,sha256=DqA9thhTRiGBDVb9zjhqcbF2W4V0AU0vrjJFhnLboU4,21075
 classy_szfast/suppress_warnings.py,sha256=6wIBml2Sj9DyRGZlZWhuA9hqvpxqrNyYjuz6BPK_a6E,202
-classy_szfast/utils.py,sha256=Et90Yo3EOjSaOTorqPmhLE3kVMF7qc2ZJHOD5NRY5vU,1916
+classy_szfast/utils.py,sha256=pDZSIgmII3J5bMUYAD8ErvngyeJEEhtla12kCIAWkMg,1962
 classy_szfast/custom_bias/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 classy_szfast/custom_bias/custom_bias.py,sha256=aR2t5RTIwv7P0m2bsEU0Eq6BTkj4pG10AebH6QpG4qM,486
 classy_szfast/custom_profiles/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 classy_szfast/custom_profiles/custom_profiles.py,sha256=4LZwb2XoqwCyWNmW2s24Z7AJdmgVdaRG7yYaBYe-d9Q,1188
-classy_szfast-0.0.25.post5.dist-info/METADATA,sha256=nhGFJzuwxdDWzS3UQpZMNk736flVpLyjjRB68r-wxbY,548
-classy_szfast-0.0.25.post5.dist-info/WHEEL,sha256=P9jw-gEje8ByB7_hXoICnHtVCrEwMQh-630tKvQWehc,91
-classy_szfast-0.0.25.post5.dist-info/top_level.txt,sha256=hRgqpilUck4lx2KkaWI2y9aCDKqF6pFfGHfNaoPFxv0,14
-classy_szfast-0.0.25.post5.dist-info/RECORD,,
+classy_szfast-0.0.25.post7.dist-info/METADATA,sha256=sIqiryp6dVpLkUbgautRNa5_Xzb6IRvaUaSNWGVsnZ4,548
+classy_szfast-0.0.25.post7.dist-info/WHEEL,sha256=P9jw-gEje8ByB7_hXoICnHtVCrEwMQh-630tKvQWehc,91
+classy_szfast-0.0.25.post7.dist-info/top_level.txt,sha256=hRgqpilUck4lx2KkaWI2y9aCDKqF6pFfGHfNaoPFxv0,14
+classy_szfast-0.0.25.post7.dist-info/RECORD,,