diff options
-rw-r--r-- | fg21sim/extragalactic/clusters/psformalism.py | 14 | ||||
-rw-r--r-- | fg21sim/share.py | 2 |
2 files changed, 11 insertions, 5 deletions
diff --git a/fg21sim/extragalactic/clusters/psformalism.py b/fg21sim/extragalactic/clusters/psformalism.py index 4a65b8f..f28c0d4 100644 --- a/fg21sim/extragalactic/clusters/psformalism.py +++ b/fg21sim/extragalactic/clusters/psformalism.py @@ -44,7 +44,10 @@ class PSFormalism: self.datafile = self.configs.get_path(comp+"/ps_data") self.f_darkmatter = self.configs.getn(comp+"/f_darkmatter") self.Mmin_cluster = self.configs.getn(comp+"/mass_min") # [Msun] - self.Mmin_halo = self.Mmin_cluster * self.f_darkmatter + + @property + def Mmin_halo(self): + return self.Mmin_cluster * self.f_darkmatter def _load_data(self, filepath=None): """ @@ -116,7 +119,7 @@ class PSFormalism: dM = self.delta(self.masses) dMgrip, dzgrip = np.meshgrid(dM, dz) Mgrip, zgrip = np.meshgrid(self.masses, self.redshifts) - dVcgrip = COSMO.differential_comoving_volume(zgrip).value # [Mpc^3/sr] + dVcgrip = COSMO.dVc(zgrip) # [Mpc^3/sr] numgrid = self.densities * dVcgrip * dzgrip * dMgrip return numgrid @@ -193,7 +196,7 @@ class PSFormalism: i = 0 while i < counts: z = random.uniform(zmin, zmax) - M = random.uniform(self.mass_min, Mmax) + M = random.uniform(self.Mmin_halo, Mmax) r = random.random() zi1 = (self.redshifts < z).sum() zi2 = zi1 - 1 @@ -215,7 +218,10 @@ class PSFormalism: z_list.append(z) M_list.append(M) i += 1 - logger.info("Sampled %d (z, mass) pairs for each cluster" % counts) + if i % 100 == 0: + logger.debug("[%d/%d] %.1f%% done ..." % + (i, counts, 100*i/counts)) + logger.info("Sampled %d pairs of (z, mass) for each cluster" % counts) df = pd.DataFrame(np.column_stack([z_list, M_list]), columns=["z", "mass"]) diff --git a/fg21sim/share.py b/fg21sim/share.py index bbe52b4..72bc23d 100644 --- a/fg21sim/share.py +++ b/fg21sim/share.py @@ -8,7 +8,7 @@ NOTE: ``global`` and ``globals`` are both preserved by Python :-( """ from .configs import ConfigManager -from .utils import Cosmology +from .utils.cosmology import Cosmology # The globally shared `~ConfigManager` instance/object, that holds the |