csiborgtools/scripts_plots/paper_environment.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 1,
   "metadata": {},
   "outputs": [],
   "source": [
    "from os.path import join\n",
    "\n",
    "import csiborgtools\n",
    "import healpy\n",
    "import matplotlib.pyplot as plt\n",
    "import numpy\n",
    "import scienceplots\n",
    "from cache_to_disk import cache_to_disk, delete_disk_caches_for_function\n",
    "from h5py import File\n",
    "\n",
    "import plt_utils\n",
    "\n",
    "\n",
    "%load_ext autoreload\n",
    "%autoreload 2"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 8,
   "metadata": {},
   "outputs": [],
   "source": [
    "@cache_to_disk(30)\n",
    "def _plot_sky_projected_density(nsim, simname, grid, nside, MAS, survey_name,\n",
    "                                dmin, dmax):\n",
    "    paths = csiborgtools.read.Paths(**csiborgtools.paths_glamdring)\n",
    "    boxsize = csiborgtools.simname2boxsize(simname)\n",
    "\n",
    "    if simname == \"csiborg1\":\n",
    "        reader = csiborgtools.read.CSiBORG1Field(nsim, paths)\n",
    "    elif \"csiborg2\" in simname:\n",
    "        kind = simname.split(\"_\")[-1]\n",
    "        reader = csiborgtools.read.CSiBORG2Field(nsim, kind, paths)\n",
    "    else:\n",
    "        raise ValueError(f\"Unknown simname `{simname}`.\")\n",
    "\n",
    "    field = reader.density_field(MAS, grid)\n",
    "\n",
    "    if survey_name == \"2M++\":\n",
    "        survey = csiborgtools.read.TwoMPPGalaxies()\n",
    "        dist = survey[\"ZCMB\"] * 3e5 / 100\n",
    "        ra, dec = survey[\"RA\"], survey[\"DEC\"]\n",
    "        ra *= numpy.pi / 180\n",
    "        dec *= numpy.pi / 180\n",
    "    elif survey_name == \"SDSS\":\n",
    "        survey = csiborgtools.SDSS()()\n",
    "        dist = survey[\"DIST\"]\n",
    "        ra, dec = survey[\"RA\"], survey[\"DEC\"]\n",
    "        ra *= numpy.pi / 180\n",
    "        dec *= numpy.pi / 180\n",
    "    elif survey_name == \"csiborg2\":\n",
    "        cat = csiborgtools.read.CSiBORG2Catalogue(\n",
    "            nsim, 99, \"main\", paths, bounds={\"totmass\": (1e12, None)})\n",
    "        coord = cat[\"spherical_pos\"]\n",
    "        dist, ra, dec = coord[:, 0], coord[:, 1], coord[:, 2]\n",
    "        ra *= numpy.pi / 180\n",
    "        dec *= numpy.pi / 180\n",
    "    else:\n",
    "        raise ValueError(f\"Unknown survey name `{survey_name}`.\")\n",
    "    \n",
    "    mask = (dist > dmin) & (dist < dmax)\n",
    "    ra = ra[mask]\n",
    "    dec = dec[mask]\n",
    "\n",
    "\n",
    "    angpos = csiborgtools.field.nside2radec(nside)\n",
    "    dist = numpy.linspace(dmin, dmax, 1000)\n",
    "    dmap = csiborgtools.field.make_sky(field, angpos=angpos, dist=dist,\n",
    "                                       boxsize=boxsize)\n",
    "    return dmap, ra, dec"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 9,
   "metadata": {},
   "outputs": [],
   "source": [
    "def plot_sky_projected_density(nsim, simname, grid, nside, MAS=\"PCS\",\n",
    "                               dmin=0, dmax=220, survey_name=\"2M++\",\n",
    "                               ext=\"png\", to_save=False):\n",
    "    dmap, ra, dec = _plot_sky_projected_density(\n",
    "        nsim, simname, grid, nside, MAS, survey_name, dmin, dmax)\n",
    "    \n",
    "    with plt.style.context(plt_utils.mplstyle):\n",
    "        healpy.mollview(numpy.log10(dmap), fig=0, title=\"\", unit=\"\", rot=90)\n",
    "\n",
    "        healpy.projscatter(numpy.pi / 2 - dec, ra, s=0.05, c=\"red\",\n",
    "                           label=\"2M++ galaxies\")\n",
    "\n",
    "        if to_save:\n",
    "            fout = join(plt_utils.fout, f\"sky_density_{simname}_{nsim}_from_{dmin}_to_{dmax}.{ext}\")  # noqa\n",
    "            print(f\"Saving to `{fout}`.\")\n",
    "            plt.savefig(fout, dpi=plt_utils.dpi, bbox_inches=\"tight\")\n",
    "        plt.show()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "metadata": {},
   "outputs": [
    {
     "ename": "NameError",
     "evalue": "name 'hp' is not defined",
     "output_type": "error",
     "traceback": [
      "\u001b[0;31m---------------------------------------------------------------------------\u001b[0m",
      "\u001b[0;31mNameError\u001b[0m                                 Traceback (most recent call last)",
      "Cell \u001b[0;32mIn[3], line 1\u001b[0m\n\u001b[0;32m----> 1\u001b[0m \u001b[43mhp\u001b[49m\n",
      "\u001b[0;31mNameError\u001b[0m: name 'hp' is not defined"
     ]
    }
   ],
   "source": []
  },
  {
   "cell_type": "code",
   "execution_count": 10,
   "metadata": {},
   "outputs": [
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "100%|██████████| 196608/196608 [00:22<00:00, 8765.57it/s]\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Saving to `../plots/sky_density_csiborg2_main_15517_from_100_to_125.png`.\n"
     ]
    },
    {
     "data": {
      "image/png": "iVBORw0KGgoAAAANSUhEUgAAAzoAAAHdCAYAAADVbwynAAAAOXRFWHRTb2Z0d2FyZQBNYXRwbG90bGliIHZlcnNpb24zLjcuMiwgaHR0cHM6Ly9tYXRwbG90bGliLm9yZy8pXeV/AAAACXBIWXMAAA9hAAAPYQGoP6dpAAEAAElEQVR4nOz9d3hcW3reC/523pWQAwESIFIhMEcwn8PDczooWp62Wm2lVrKCr8cejz0ehyt7rhzmXluWPXMtW5YlWS211MqyYofTPIE5ZxJAFRJJgCAyUHHnPX+sXQWyLdvSWLKlVr3PwwcAUbX32muvXfi+9b3v+0lhGIbUUEMNNdRQQw011FBDDTV8FUH+nz2AGmqooYYaaqihhhpqqKGGP27UEp0aaqihhhpqqKGGGmqo4asOtUSnhhpqqKGGGmqooYYaaviqQy3RqaGGGmqooYYaaqihhhq+6lBLdGqooYYaaqihhhpqqKGGrzrUEp0aaqihhhpqqKGGGmqo4asO6v/sAdRQQw011PDHg3fO/NPq976pkN+hv/b7xKIHgGL7AISyBIDdsPWnIPmsJL7xAgBK3Ynq72Q3RC371Z/Vgive32yK3zviPUrZq75G8gMkN6j+HGqv7K/JEqEkVX+02gwAAlX8n7EhjiN7W++XvJDglWM4Da//GatcE4Cvie/ji454bb14rZ7zXnuPbyhYTUr15+uf/VvUUEMNNdTwZx9SrY9ODTXUUMOfDrz5Nf8HAPqmA8HWR7MUhEhWFJwrUSDvi9/79SZuSqu+NlTA10UikJgTScuriYHsBoRbuQBBTAT/oSzh1G0lDbIXwlZ+gWr5SK+MyU1uvXazRyO+tJUAxZZd3FSUVGy4+LGtJKIyNgA97+KZytbxnVdOKEuvJS1Og4bsivOvpzWan9jV3+W7dKSt02Nu+Gz2ivMnX2z9QpIDzjx6zP26HvbnZrnZOYQvK6h5kbAFhoL3ylhVy68mR/FoLitjCjUFp2Fr3gmAaLgXfvfvUEMNNdRQw/981BKdGmqooYY/AXzswA/j1YlKh1KMAum4Fv3s4MfE97LjEWoiuJZLTvQ6HSXw2ec+4+a2NL6s4Bsydp1IEty4iKjrZ13CKPHRN8Q5QkV6rcqhFrYCfcWOKi6WB4H43kuJKorkBUhR8iRFfxbcpFZNkkIz5MTsBLebB8i3i/doZfG6QAHfEOOQAqpJhxT9dVHLQfS7EN+QsRrEMeun7eg8Yrxa0cOLi7nQCh5OnZijSoWncg7PkFCcMLpeSD0Vx8nvFONKzjl4CXGcYrsaXXvIcNMTcjGDN29PUzAMnra2klkaJNcjXtPyoEwQJWK+oVTnRfbCrXkDXpxO0jJhc2R5kgdGD0EQzVE0zur91pXqnIa6XK1sVRLGStKkFB2+ePdHqKGGGmqo4Y8XtUSnhhpqqOEPiY8d+OHq937SqH4fGArqRlRhUOUq7UuRAg6UnnKnvg9fVghfqWb4hoIS+BxeneRKn6gsKIHPoc1spTDASmOclGNxvWuQM+PjIMHNzgFGn05VEyDJByXwObqY5XbLAL4sAnRz1a2ey028Tu+qBO4QVXiUrcpJsV1DCXyOLGe5sH8QTxHHe/PRGAXdpN4qgSdzZ1sfh15Oc6dp65yvVlUqiQhsJU4AbnxrDsJYwOh8lhvb0yj5rf+vJEYAamnroKX2Vyoor8DXthKffLeMuRpG17ZF1wPwEyFHF7OovkdncZWnjW1c2bELgFzv1vnbbou5UwKfE9NPuFsn7p9vbs3jkY0sRdUk4VlcGN5d/f/UbOkPHON/CeX2GMaaWDteXIUo+QlUCXNh61ivUv7evfYP/0jnqKGGGmr484paolNDDTXUAHz02I8QqNFufkxFLYjqipfSX9GeuASGCHZ9Q0F2RBAu+WGUpExzfWCQg6szvH9wmP1Whpxp0jZn8VBOA1BqVSi3imD21MwYjqtzZH6Knzr1NgdzEwy8XEQpSjxtacWXZW61DbLPH6d/YRFtU8YMHK72p2nKlyi3hlweHuCtSzNYrkmDVSLXA9MvB/FlBbch4ORUlhsdIikiBC1KIqwGGUkJGJ3LcKcpjeyFHH2Z5cPjA3z35Q+42d2PoTrc04bFNaoBx2cyyEHAckecYzNZrvemaXtW5lrnMFIIxqYv5mF9invxXpG8uUG1MuMbEmoxwIuSnZNPx1hpS5B0LR5oA9Xqh7FUxqs3ove8khhVKiZWgN24VakJtK1Kj10vvk8sBdVKkFbwcRPRcSQJzfE4spTlVlsaX1IIjYCjL7PcbE/j6uK4bgKOt90iHzNo/EDjninun2Z7HJub4H59D7vcOR5qO/FlheJ2A8UW49dzXpUemO/SacyUASi3GySn8wDYbfHX6ImKFa2lV/6vktzIZa+aAEl+iJsS2ivZD16rHskVbZQq86XrtWSohhpqqKGW6NRQQw1/LnDu3P8bNQoKCUMCTalWMvy4ilr0kAk4tDHFnab+Kh3J6ZD5jvsf8OOfeoNj01Psm3jBz55+g8Nz08gOfOnsIFJR5UxmnEJM41PXrvOrh4+hJWzupIYZnctwtT8NtorX7HJiJsPl4QGChgDV8/mh3/4AbzXO81g7F4eHOTE7jrkZcrl3GE9WUK0Q1fM5upQBCb749X289f4schCwWWdw/FkWpRQy3taFVS9jleMoTRYXB4c5NTNGsByj3ioRyDI3tqdxDBUvLqbhxPwYGymTOttCKUEuafLmnSx32vvYvzHNLwycA+BgLsu9ZlHBud3Rx8GVae609XH02TQ3O9MoBalahdnnTeIVNepKZW60DhHKEmGUYyhWgGJ51QBetVyOLE1wN9WLZxg49TpazonuiYYcUb1kx8ep11GLUSAvgdOooxWinwOqlDenXsVqkIitiveGilStbnlxBd+QKLZt6XAAzmTHKGgmicDiRuswTkqsC7PkcfRFlms703iKwsaugI+fz5DXTWKaxbsf6UOfE0lH2wOPve4k17sGib0AX5fQ8wFK4HMwP8X9WFTVkyHQJJTQ58jyZDW50iKKoeyFeDExP5L/evVNsf2qhkixApHolMQcSK5fXc+S6+O0JLYocoqE1aSSmLei9Q9fvvy//rcfmhpqqKGGP+OoJTo11FDDn3mc+Qv/AhAaFMX2UUoebt2W45hi+VXhuFbwsNpVTk6O84WvG+T49CTmCnz+42neGJtgcGqZuOvyk6ffId8V8td+/zxSv0ffnWUcTcXWFFJzAePt25EDmGlu50J6hLo5j08/eI9re3s4cW+Gn999jgARlK6PyNQ99/iue+9xZbiPhONyqW8ExYY3ph7TtbHM88ZWLvTvRiuE2A0iYFXLoBZD9EKk7QjBiSoT+SGf/8fEf6LxRYFYnU++0eSHB76FU8+eAHCzYTeE8MaDDG5TwEY8TsqyuLp9BH1DzItheRyfzXKnaQDZDTmylOX2tj4OLU9zp7WPg6szlDtCVhrjnHmc5VpPmqZVi0cMAKBvuqLK4odVZzRZCjmYm+ZOQz++JGO36BgrDrneGAANEwWiF1aTTX25CAiBv1SqUAAV8HxCfYuuFsQ1lLwI1t22pLi3RRevUuGwfaRoHHaziZeQCVSJ5YMSqu/zkQ+zfHBY0PFU3+cjl8ZwUxJ36wY4MSmof1ajwpGlSW50pnE1BadeIrYSku+OKkU5h0/f/ICfP3gWz9NIrHnstSdF4mnGSUgWl/pGcBOgluD09BheziTpWlzpHoboL+4BZxw3b5JyLG43DAHg61DsjIwkXgTENl2OLE9y3xTVMbdOwYsqXHaDREPWida3SHa8uIoWOeGFkkQYVSjVvE0oSci2W/0/rzFWdbyzmivPho8fk6tr7MbPR+5zrgvnz8Pbb4P2B9MHa6ihhhr+NKKW6NRQQw1/JvDG1/1zAMotKooTorghsisoY0eWsvze1w3jKQpBo8MbjyeIzypc6x5m7YwIBmOPRKDtJkPeuT5OQTM5vJwl16AhpTxmlR0oQcDZzGOeJ1qYbOvg/YPDJMtl/vFnf5P/17d+I0eezrJv/AW/lD7D4YUZFDvgVvsAh19O894xQfF64/4ENzvSlNt
      "text/plain": [
       "<Figure size 850x540 with 2 Axes>"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    }
   ],
   "source": [
    "# delete_disk_caches_for_function(\"_plot_sky_projected_density\")\n",
    "plot_sky_projected_density(15517, \"csiborg2_main\", 1024, 128, \"SPH\", dmin=100,\n",
    "                           dmax=125, survey_name=\"csiborg2\", to_save=True)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": []
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": []
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": []
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": []
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": []
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "venv_csiborg",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.11.4"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 2
}