Move to old

scripts/fit_hmf.py

@@ -1,108 +0,0 @@
-# Copyright (C) 2022 Richard Stiskalek
-# This program is free software; you can redistribute it and/or modify it
-# under the terms of the GNU General Public License as published by the
-# Free Software Foundation; either version 3 of the License, or (at your
-# option) any later version.
-#
-# This program is distributed in the hope that it will be useful, but
-# WITHOUT ANY WARRANTY; without even the implied warranty of
-# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General
-# Public License for more details.
-#
-# You should have received a copy of the GNU General Public License along
-# with this program; if not, write to the Free Software Foundation, Inc.,
-# 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
-"""
-Script to calculate the HMF for CSIBORG and Quijote haloes.
-"""
-from argparse import ArgumentParser
-from datetime import datetime
-from distutils.util import strtobool
-
-import numpy
-from mpi4py import MPI
-
-from taskmaster import work_delegation
-from utils import get_nsims
-
-try:
-    import csiborgtools
-except ModuleNotFoundError:
-    import sys
-
-    sys.path.append("../")
-    import csiborgtools
-
-
-def get_counts(nsim, bins, paths, parser_args):
-    """
-    Calculate and save the number of haloes in each mass bin.
-    """
-    simname = parser_args.simname
-    paths = csiborgtools.read.Paths(**csiborgtools.paths_glamdring)
-    bounds = {"dist": (0, parser_args.Rmax)}
-
-    if simname == "csiborg":
-        cat = csiborgtools.read.CSiBORGHaloCatalogue(
-            nsim, paths, bounds=bounds, load_fitted=False, load_initial=False)
-        logmass = numpy.log10(cat["fof_totpartmass"])
-        counts = csiborgtools.number_counts(logmass, bins)
-    elif simname == "quijote":
-        cat0 = csiborgtools.read.QuijoteHaloCatalogue(
-            nsim, paths, nsnap=4, load_fitted=False, load_initial=False)
-        nmax = int(cat0.box.boxsize // (2 * parser_args.Rmax))**3
-        counts = numpy.full((nmax, len(bins) - 1), numpy.nan,
-                            dtype=numpy.float32)
-
-        for nobs in range(nmax):
-            cat = cat0.pick_fiducial_observer(nobs, rmax=parser_args.Rmax)
-            logmass = numpy.log10(cat["group_mass"])
-            counts[nobs, :] = csiborgtools.number_counts(logmass, bins)
-    elif simname == "quijote_full":
-        cat = csiborgtools.read.QuijoteHaloCatalogue(
-            nsim, paths, nsnap=4, load_fitted=False, load_initial=False,
-            load_backup=parser_args.from_quijote_backup)
-        logmass = numpy.log10(cat["group_mass"])
-        counts = csiborgtools.number_counts(logmass, bins)
-    else:
-        raise ValueError(f"Unknown simulation name `{simname}`.")
-
-    fout = paths.halo_counts(simname, nsim, parser_args.from_quijote_backup)
-    if parser_args.verbose:
-        print(f"{datetime.now()}: saving halo counts to `{fout}`.")
-    numpy.savez(fout, counts=counts, bins=bins, rmax=parser_args.Rmax)
-
-
-if __name__ == "__main__":
-    parser = ArgumentParser()
-    parser.add_argument("--simname", type=str,
-                        choices=["csiborg", "quijote", "quijote_full"],
-                        help="Simulation name.")
-    parser.add_argument("--nsims", type=int, nargs="+", default=None,
-                        help="Indices of simulations to cross. If `-1` all .")
-    parser.add_argument(
-        "--Rmax", type=float, default=155,
-        help="High-res region radius in Mpc / h. Ignored for `quijote_full`.")
-    parser.add_argument("--from_quijote_backup",
-                        type=lambda x: bool(strtobool(x)), default=False,
-                        help="Flag to indicate Quijote backup data.")
-    parser.add_argument("--lims", type=float, nargs="+", default=[11., 16.],
-                        help="Mass limits in Msun / h.")
-    parser.add_argument("--bw", type=float, default=0.2,
-                        help="Bin width in dex.")
-    parser.add_argument("--verbose", type=lambda x: bool(strtobool(x)),
-                        default=False, help="Verbosity flag.")
-    parser_args = parser.parse_args()
-
-    paths = csiborgtools.read.Paths(**csiborgtools.paths_glamdring)
-    nsims = get_nsims(parser_args, paths)
-
-    if len(parser_args.lims) != 2:
-        raise ValueError("Mass limits must be a pair of floats.")
-
-    bins = numpy.arange(*parser_args.lims, parser_args.bw, dtype=numpy.float32)
-
-    def do_work(nsim):
-        get_counts(nsim, bins, paths, parser_args)
-
-    work_delegation(do_work, nsims, MPI.COMM_WORLD)