csiborgtools/notebooks/flow/mock_carrick.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 1,
   "metadata": {},
   "outputs": [],
   "source": [
    "# Copyright (C) 2024 Richard Stiskalek\n",
    "# This program is free software; you can redistribute it and/or modify it\n",
    "# under the terms of the GNU General Public License as published by the\n",
    "# Free Software Foundation; either version 3 of the License, or (at your\n",
    "# option) any later version.\n",
    "#\n",
    "# This program is distributed in the hope that it will be useful, but\n",
    "# WITHOUT ANY WARRANTY; without even the implied warranty of\n",
    "# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General\n",
    "# Public License for more details.\n",
    "#\n",
    "# You should have received a copy of the GNU General Public License along\n",
    "# with this program; if not, write to the Free Software Foundation, Inc.,\n",
    "# 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.\n",
    "import numpy as np\n",
    "import matplotlib.pyplot as plt\n",
    "from tqdm import trange\n",
    "from joblib import dump\n",
    "from h5py import File\n",
    "\n",
    "import csiborgtools\n",
    "\n",
    "%matplotlib inline\n",
    "%load_ext autoreload\n",
    "%autoreload 2\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 5,
   "metadata": {},
   "outputs": [],
   "source": [
    "fpath = \"/mnt/extraspace/rstiskalek/catalogs/PV_compilation.hdf5\"\n",
    "\n",
    "with File(fpath, 'r') as f:\n",
    "    RA_2MTF = f[\"2MTF/RA\"][...]\n",
    "    DEC_2MTF = f[\"2MTF/DEC\"][...]"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 36,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Saving to  /mnt/extraspace/rstiskalek/csiborg_postprocessing/flow_mock/Carrick2MTFmock_seed0.hdf5\n"
     ]
    }
   ],
   "source": [
    "!rm /mnt/extraspace/rstiskalek/csiborg_postprocessing/flow_mock/*\n",
    "\n",
    "\n",
    "reader = csiborgtools.read.Carrick2015Field()\n",
    "velocity = reader.velocity_field()\n",
    "boxsize = csiborgtools.simname2boxsize(\"Carrick2015\")\n",
    "\n",
    "mean_mag = 10.31\n",
    "std_mag = 0.83\n",
    "\n",
    "kmax = 1\n",
    "seed = 53\n",
    "\n",
    "for k in trange(kmax):\n",
    "    mock, truths = csiborgtools.flow.mock_Carrick2MTF(\n",
    "        velocity, boxsize, RA_2MTF, DEC_2MTF,\n",
    "        mean_mag=mean_mag, std_mag=std_mag,\n",
    "        seed=seed, verbose=True)\n",
    "    seed += 1\n",
    "\n",
    "    fname = f\"/mnt/extraspace/rstiskalek/csiborg_postprocessing/flow_mock/Carrick2MTFmock_seed{k}.hdf5\"  # noqa\n",
    "    print(\"Saving to \", fname)\n",
    "    with File(fname, 'w') as f:\n",
    "        for key in mock.keys():\n",
    "            f.create_dataset(key, data=mock[key])\n",
    "\n",
    "    fname_truths = fname.replace(\".hdf5\", \"_truths.pkl\")\n",
    "    dump(truths, fname_truths)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 31,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "image/png": "iVBORw0KGgoAAAANSUhEUgAAAigAAAGdCAYAAAA44ojeAAAAOXRFWHRTb2Z0d2FyZQBNYXRwbG90bGliIHZlcnNpb24zLjguMywgaHR0cHM6Ly9tYXRwbG90bGliLm9yZy/H5lhTAAAACXBIWXMAAA9hAAAPYQGoP6dpAAAiaElEQVR4nO3df1jV9f3/8QeIgL84CMpBCpTM+SOblT8INadJ0nIVi1U2Zr+8ZBVUaktlSytnYa7SS2dSzbCudDa3NNNrdilOvZaIilmZRtowMT24ZpyTOBDl9f3Dr+ezk5Q/OofzAu+36zrXtfN+v8/bJ2/e17j3Pr9CjDFGAAAAFgkN9gAAAADfRqAAAADrECgAAMA6BAoAALAOgQIAAKxDoAAAAOsQKAAAwDoECgAAsE5YsAe4EPX19Tp48KDatWunkJCQYI8DAADOgTFG33zzjRISEhQa+v3XSJpkoBw8eFCJiYnBHgMAAFyAiooKXXrppd+7TZMMlHbt2kk69QNGRUUFeRoAAHAuPB6PEhMTvX/Hv0+TDJTTT+tERUURKAAANDHn8vIMXiQLAACsQ6AAAADrECgAAMA6BAoAALAOgQIAAKxDoAAAAOsQKAAAwDoECgAAsA6BAgAArEOgAAAA6xAoAADAOgQKAACwDoECAACsQ6AAAADrhAV7AADB12XyqoDuf9+MkQHdP4DmhysoAADAOgQKAACwDoECAACsQ6AAAADrECgAAMA6BAoAALAOgQIAAKxDoAAAAOsQKAAAwDoECgAAsA6BAgAArEOgAAAA6xAoAADAOgQKAACwDoECAACsQ6AAAADrECgAAMA6BAoAALAOgQIAAKxDoAAAAOsQKAAAwDoECgAAsA6BAgAArEOgAAAA6xAoAADAOgQKAACwDoECAACsQ6AAAADrECgAAMA6BAoAALDOeQfKxo0bdfPNNyshIUEhISFavny5d11dXZ0mTZqkK6+8Um3atFFCQoLuvvtuHTx40GcfR44cUVZWlqKiohQdHa0xY8bo6NGjP/iHAQAAzcN5B0p1dbX69OmjefPmnbHu2LFj2r59u6ZMmaLt27fr7bffVllZmW655Raf7bKysvTJJ59ozZo1WrlypTZu3Kjs7OwL/ykAAECzEmKMMRf84JAQLVu2TBkZGd+5zdatWzVgwAB98cUXSkpK0u7du9WrVy9t3bpV/fr1kyStXr1aN910kw4cOKCEhISz/rsej0cOh0Nut1tRUVEXOj6A/6/L5FUB3f++GSMDun8ATcP5/P0O+GtQ3G63QkJCFB0dLUkqLi5WdHS0N04kKS0tTaGhoSopKWlwH7W1tfJ4PD43AADQfAU0UGpqajRp0iTddddd3lJyuVyKi4vz2S4sLEwxMTFyuVwN7ic/P18Oh8N7S0xMDOTYAAAgyAIWKHV1dbrjjjtkjNH8+fN/0L7y8vLkdru9t4qKCj9NCQAAbBQWiJ2ejpMvvvhC69at83meKT4+XocPH/bZ/sSJEzpy5Iji4+Mb3F9ERIQiIiICMSoAALCQ36+gnI6TPXv2aO3atYqNjfVZn5qaqqqqKpWWlnqXrVu3TvX19UpJSfH3OAAAoAk67ysoR48e1d69e733y8vLtWPHDsXExKhTp076xS9+oe3bt2vlypU6efKk93UlMTExCg8PV8+ePXXjjTdq7NixKigoUF1dnXJzczVq1KhzegcPAABo/s47ULZt26Zhw4Z570+YMEGSdM899+ipp57SihUrJElXXXWVz+P+8Y9/aOjQoZKkRYsWKTc3V8OHD1doaKgyMzM1Z86cC/wRAABAc3PegTJ06FB930ennMvHqsTExGjx4sXn+08DAICLBN/FAwAArEOgAAAA6xAoAADAOgH5HBQA+F981w+A88UVFAAAYB0CBQAAWIeneAA0eTyFBDQ/XEEBAADWIVAAAIB1CBQAAGAdAgUAAFiHQAEAANYhUAAAgHUIFAAAYB0CBQAAWIdAAQAA1iFQAACAdQgUAABgHQIFAABYhy8LBICzCOSXEfJFhEDDuIICAACsQ6AAAADrECgAAMA6BAoAALAOgQIAAKxDoAAAAOsQKAAAwDoECgAAsA6BAgAArEOgAAAA6xAoAADAOgQKAACwDoECAACsQ6AAAADrECgAAMA6BAoAALAOgQIAAKxDoAAAAOsQKAAAwDoECgAAsA6BAgAArHPegbJx40bdfPPNSkhIUEhIiJYvX+6z3hijqVOnqlOnTmrVqpXS0tK0Z88en22OHDmirKwsRUVFKTo6WmPGjNHRo0d/0A8CAACaj/MOlOrqavXp00fz5s1rcP3MmTM1Z84cFRQUqKSkRG3atFF6erpqamq822RlZemTTz7RmjVrtHLlSm3cuFHZ2dkX/lMAAIBmJcQYYy74wSEhWrZsmTIyMiSdunqSkJCgxx57TL/5zW8kSW63W06nUwsXLtSoUaO0e/du9erVS1u3blW/fv0kSatXr9ZNN92kAwcOKCEh4az/rsfjkcPhkNvtVlRU1IWODzQpXSavCvYICIB9M0YGewSg0ZzP32+/vgalvLxcLpdLaWlp3mUOh0MpKSkqLi6WJBUXFys6OtobJ5KUlpam0NBQlZSU+HMcAADQRIX5c2cul0uS5HQ6fZY7nU7vOpfLpbi4ON8hwsIUExPj3ebbamtrVVtb673v8Xj8OTYAALBMk3gXT35+vhwOh/eWmJgY7JEAAEAA+TVQ4uPjJUmVlZU+yysrK73r4uPjdfjwYZ/1J06c0JEjR7zbfFteXp7cbrf3VlFR4c+xAQCAZfwaKMnJyYqPj1dRUZF3mcfjUUlJiVJTUyVJqampqqqqUmlpqXebdevWqb6+XikpKQ3uNyIiQlFRUT43AADQfJ33a1COHj2qvXv3eu+Xl5drx44diomJUVJSksaNG6fp06erW7duSk5O1pQpU5SQkOB9p0/Pnj114403auzYsSooKFBdXZ1yc3M1atSoc3oHDwAAaP7OO1C2bdumYcOGee9PmDBBknTPPfdo4cKFmjhxoqqrq5Wdna2qqioNHjxYq1evVmRkpPcxixYtUm5uroYPH67Q0FBlZmZqzpw5fvhxAABAc/CDPgclWPgcFFyM+BwUXAg+ZwU2CdrnoAAAAPgDgQIAAKxDoAAAAOsQKAAAwDoECgAAsA6BAgAArEOgAAAA6xAoAADAOgQKAACwDoECAACsQ6AAAADrECgAAMA6BAoAALAOgQIAAKxDoAAAAOsQKAAAwDoECgAAsA6BAgAArEOgAAAA6xAoAADAOgQKAACwDoECAACsQ6AAAADrECgAAMA6BAoAALBOWLAHABpLl8mrArr/fTNGBnT/AHAx4QoKAACwDoECAACsQ6AAAADrECgAAMA6BAoAALAO7+IB/CTQ7xICgIsJV1AAAIB1CBQAAGAdAgUAAFiHQAEAANYhUAAAgHUIFAAAYB0CBQAAWIdAAQAA1iFQAACAdQgUAABgHb8HysmTJzVlyhQlJyerVatW6tq1q37/+9/LGOPdxhijqVOnqlOnTmrVqpXS0tK0Z88ef48CAACaKL8HynPPPaf58+frj3/8o3bv3q3nnntOM2fO1Ny5c73bzJw5U3PmzFFBQYFKSkrUpk0bpaenq6amxt/jAACAJsjvXxa4adMm3XrrrRo5cqQkqUuXLvrzn/+sLVu2SDp19WT27Nl64okndOutt0qS3njjDTmdTi1fvlyjRo3y90gAAKCJ8fsVlIEDB6qoqEifffaZJOnDDz/UP//5T/30pz+VJJWXl8vlciktLc37GIfDoZSUFBUXF/t7HAAA0AT5/QrK5MmT5fF41KNHD7Vo0UInT57UM888o6ysLEmSy+WSJDmdTp/HOZ1O77pvq62tVW1trfe+x+Px99gAAMAifr+C8pe//EWLFi3S4sWLtX37dr3++ut6/vnn9frrr1/wPvPz8+VwOLy3xMREP04MAABs4/dAefzxxzV58mSNGjVKV155pUaPHq3x48crPz9fkhQfHy9Jqqys9HlcZWWld9235eXlye12e28VFRX+HhsAAFjE74Fy7NgxhYb67rZFixaqr6+XJCU
      "text/plain": [
       "<Figure size 640x480 with 1 Axes>"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    }
   ],
   "source": [
    "plt.figure()\n",
    "plt.hist(mock[\"DEC\"], bins=\"auto\")\n",
    "plt.show()"
   ]
  },
  {
   "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
}