get_riseset_tests.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "34383513",
   "metadata": {},
   "outputs": [],
   "source": [
    "import numpy as np\n",
    "import mags\n",
    "import obsSlots\n",
    "import hp2np \n",
    "import hexalate\n",
    "\n",
    "\n",
    "import decam2hp\n",
    "import jsonMaker\n",
    "from os import getenv\n",
    "import pandas as pd\n",
    "import sys\n",
    "import  matplotlib.pyplot as plt"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 5,
   "id": "955b6591",
   "metadata": {},
   "outputs": [],
   "source": [
    "def get_hexinfo(ra, dec, prob, expTime, filt, mjd, best=False, camera=\"decam\"):\n",
    "    \"\"\"\n",
    "    Gets information about given hexes. Based on Jim's simplicity.calc\n",
    "    code.\n",
    "\n",
    "    Parameters:\n",
    "    -----------\n",
    "    ra: np.vector\n",
    "        vector of hex ra\n",
    "    dec: np.vector\n",
    "        vector of hex dec\n",
    "    prob: np.vector\n",
    "        vector of hex summed prob from LIGO map\n",
    "    expTime: np.vector\n",
    "        vector of exposure times\n",
    "    filt: str\n",
    "        telescope filt.\n",
    "    mjd: float\n",
    "        modified julian date.\n",
    "    best: bool\n",
    "        True/False if use best or not\n",
    "    camera: str\n",
    "        telescope camera name to take in account.\n",
    "        \n",
    "    Outputs:\n",
    "    --------\n",
    "    hexinfo: pd.dataframe\n",
    "        dataframe with each row representing a hex. contains ra, dec,\n",
    "        prob, and rise and set times in MJD (minutes), and is sorted by \n",
    "        probability.\n",
    "    \"\"\"\n",
    "\n",
    "    # find the highest prob hex\n",
    "    best_ix = np.argmax(prob)\n",
    "\n",
    "    # find night statistics\n",
    "    night, sunset, sunrise = mags.findNightDuration(mjd, camera)\n",
    "    night = np.float64(night)*24.\n",
    "    sunset = np.float64(sunset)\n",
    "    sunrise = np.float64(sunrise)\n",
    "\n",
    "    # work every minute from sunset to sunrise\n",
    "    mjd_list = np.arange(sunset, sunrise+1./1440., 1./1440.)\n",
    "    \n",
    "    # calculate the limiting magnitudes\n",
    "    limit_mag = []\n",
    "    best_limit_mag = []\n",
    "    moon_sep = []\n",
    "    moon_phase = []\n",
    "    print(sunset)\n",
    "    obs = mags.observed(ra,dec,prob, sunset, doMaps=False, verbose=False)\n",
    "    for mjd in mjd_list :\n",
    "        obs.resetTime(mjd)\n",
    "        obs.limitMag(filt,exposure=expTime)\n",
    "        limit_mag.append(obs.maglim)\n",
    "        best_limit_mag.append(obs.maglim[best_ix])\n",
    "        moon_sep.append(obs.moonSep[0]*360./2/np.pi)\n",
    "        moon_phase.append(obs.moonPhase)\n",
    "    limit_mag = np.vstack(limit_mag)\n",
    "    ix = limit_mag < 0; limit_mag[ix] = 0\n",
    "    \n",
    "    #copied from Jim's code, left it here in case someone wants moon information\n",
    "    moon_sep = np.array(moon_sep)\n",
    "    moon_phase = np.array(moon_phase)\n",
    "    # moon phase: where 0 = full, 90 equals half, and  180 = new\n",
    "    # convert to %full\n",
    "    moon_phase = ((180.-moon_phase)/180.)*100.\n",
    "    bins = np.arange(21,25.5,0.5)\n",
    "    \n",
    "    set_list = [ ]\n",
    "    rise_list = [ ]\n",
    "\n",
    "    for i in range(len(limit_mag)):\n",
    "        for j in range(len(limit_mag[i])):\n",
    "            if limit_mag[i][j] !=0 and limit_mag[i-1][j] == 0:\n",
    "                rise_list.append([mjd_list[i], ra[j], dec[j]])\n",
    "            elif limit_mag[i-1][j] !=0 and limit_mag[i][j] == 0:\n",
    "                set_list.append([mjd_list[i-1], ra[j], dec[j]])\n",
    "                #print(f'setting j={j}, i={i}, {ra[j]}, {mjd_list[i]}')\n",
    "    \n",
    "    set_times = []\n",
    "    rise_times = []\n",
    "    #print(rise_list)\n",
    "    \n",
    "    \n",
    "    for i in range(len(ra)):\n",
    "        sets =[]\n",
    "        rises=[]\n",
    "        \n",
    "        #find rise and set times corresponding to each ra and dec\n",
    "        for s in set_list:\n",
    "            #print(s[1])\n",
    "            if s[1]==ra[i] and s[2]==dec[i]:\n",
    "                #print(s[0], s[1], ra[i], dec[i])\n",
    "                sets.append(s[0])\n",
    "                #print(s[1], s[2], ra[i], dec[i], s[0])\n",
    "        #find rise time for each ra and dec\n",
    "        for r in rise_list:\n",
    "            if r[1]==ra[i] and r[2]==dec[i]:\n",
    "#                print(f'For {ra[i]}, have {r[0]}')\n",
    "                rises.append(r[0])\n",
    "\n",
    "        if len(sets) != 0:\n",
    "            set_times.append(np.max(sets))\n",
    "        elif len(sets) == 0:\n",
    "            set_times.append('N/A')\n",
    "        if len(rises) !=0:\n",
    "            rise_times.append(np.min(rises))\n",
    "        elif len(rises) == 0:\n",
    "            rise_times.append('N/A')\n",
    "\n",
    "    #print(rise_times, set_times)\n",
    "    rise_mins = 1440*(rise_times - sunset)\n",
    "    set_mins = 1440*(set_times - sunset)\n",
    "    #create the dataframe\n",
    "    d = {'RA (deg)': ra, 'DEC (deg)': dec, 'PROBABILITY': prob, 'RISES (MIN SINCE SUNSET)': rise_mins, 'SETS (MIN SINCE SUNSET)':set_mins, 'RISES (MJD)': rise_times, 'SETS (MJD)': set_times}\n",
    "    info = pd.DataFrame.from_dict(d)\n",
    "    \n",
    "    #sort by probability \n",
    "    hexinfo = info.sort_values(by='PROBABILITY', ascending=False)\n",
    "#    print(hexinfo)\n",
    "    \n",
    "    return hexinfo"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 6,
   "id": "7a210a5c",
   "metadata": {},
   "outputs": [],
   "source": [
    "dec = np.linspace(0, 30, num=10)\n",
    "ra = np.linspace(300, 340, num=10)\n",
    "probs=np.random.randint(10, size=(10))\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 8,
   "id": "6712c997",
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "60068.95138888889\n",
      "Loading dust map: .//data/plank-ebv-HFI_CompMap_ThermaDustModel.fits\n",
      "loading inverse stellar density = probability of recognition map\n",
      "\t ... the sun is up\n",
      "\t ... the sun is up\n"
     ]
    },
    {
     "data": {
      "text/html": [
       "<div>\n",
       "<style scoped>\n",
       "    .dataframe tbody tr th:only-of-type {\n",
       "        vertical-align: middle;\n",
       "    }\n",
       "\n",
       "    .dataframe tbody tr th {\n",
       "        vertical-align: top;\n",
       "    }\n",
       "\n",
       "    .dataframe thead th {\n",
       "        text-align: right;\n",
       "    }\n",
       "</style>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>RA (deg)</th>\n",
       "      <th>DEC (deg)</th>\n",
       "      <th>PROBABILITY</th>\n",
       "      <th>RISES (MIN SINCE SUNSET)</th>\n",
       "      <th>SETS (MIN SINCE SUNSET)</th>\n",
       "      <th>RISES (MJD)</th>\n",
       "      <th>SETS (MJD)</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>5</th>\n",
       "      <td>322.222222</td>\n",
       "      <td>16.666667</td>\n",
       "      <td>9</td>\n",
       "      <td>549.000002</td>\n",
       "      <td>700.000002</td>\n",
       "      <td>60069.332639</td>\n",
       "      <td>60069.4375</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>8</th>\n",
       "      <td>335.555556</td>\n",
       "      <td>26.666667</td>\n",
       "      <td>9</td>\n",
       "      <td>652.000002</td>\n",
       "      <td>700.000002</td>\n",
       "      <td>60069.404167</td>\n",
       "      <td>60069.4375</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>300.000000</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>8</td>\n",
       "      <td>407.000001</td>\n",
       "      <td>700.000002</td>\n",
       "      <td>60069.234028</td>\n",
       "      <td>60069.4375</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>313.333333</td>\n",
       "      <td>10.000000</td>\n",
       "      <td>8</td>\n",
       "      <td>489.000002</td>\n",
       "      <td>700.000002</td>\n",
       "      <td>60069.290972</td>\n",
       "      <td>60069.4375</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>304.444444</td>\n",
       "      <td>3.333333</td>\n",
       "      <td>6</td>\n",
       "      <td>434.000001</td>\n",
       "      <td>700.000002</td>\n",
       "      <td>60069.252778</td>\n",
       "      <td>60069.4375</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>317.777778</td>\n",
       "      <td>13.333333</td>\n",
       "      <td>6</td>\n",
       "      <td>519.000002</td>\n",
       "      <td>700.000002</td>\n",
       "      <td>60069.311806</td>\n",
       "      <td>60069.4375</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>7</th>\n",
       "      <td>331.111111</td>\n",
       "      <td>23.333333</td>\n",
       "      <td>5</td>\n",
       "      <td>615.000002</td>\n",
       "      <td>700.000002</td>\n",
       "      <td>60069.378472</td>\n",
       "      <td>60069.4375</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>9</th>\n",
       "      <td>340.000000</td>\n",
       "      <td>30.000000</td>\n",
       "      <td>3</td>\n",
       "      <td>693.000002</td>\n",
       "      <td>700.000002</td>\n",
       "      <td>60069.432639</td>\n",
       "      <td>60069.4375</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>308.888889</td>\n",
       "      <td>6.666667</td>\n",
       "      <td>2</td>\n",
       "      <td>461.000002</td>\n",
       "      <td>700.000002</td>\n",
       "      <td>60069.271528</td>\n",
       "      <td>60069.4375</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>6</th>\n",
       "      <td>326.666667</td>\n",
       "      <td>20.000000</td>\n",
       "      <td>1</td>\n",
       "      <td>581.000002</td>\n",
       "      <td>700.000002</td>\n",
       "      <td>60069.354861</td>\n",
       "      <td>60069.4375</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "     RA (deg)  DEC (deg)  PROBABILITY  RISES (MIN SINCE SUNSET)  \\\n",
       "5  322.222222  16.666667            9                549.000002   \n",
       "8  335.555556  26.666667            9                652.000002   \n",
       "0  300.000000   0.000000            8                407.000001   \n",
       "3  313.333333  10.000000            8                489.000002   \n",
       "1  304.444444   3.333333            6                434.000001   \n",
       "4  317.777778  13.333333            6                519.000002   \n",
       "7  331.111111  23.333333            5                615.000002   \n",
       "9  340.000000  30.000000            3                693.000002   \n",
       "2  308.888889   6.666667            2                461.000002   \n",
       "6  326.666667  20.000000            1                581.000002   \n",
       "\n",
       "   SETS (MIN SINCE SUNSET)   RISES (MJD)  SETS (MJD)  \n",
       "5               700.000002  60069.332639  60069.4375  \n",
       "8               700.000002  60069.404167  60069.4375  \n",
       "0               700.000002  60069.234028  60069.4375  \n",
       "3               700.000002  60069.290972  60069.4375  \n",
       "1               700.000002  60069.252778  60069.4375  \n",
       "4               700.000002  60069.311806  60069.4375  \n",
       "7               700.000002  60069.378472  60069.4375  \n",
       "9               700.000002  60069.432639  60069.4375  \n",
       "2               700.000002  60069.271528  60069.4375  \n",
       "6               700.000002  60069.354861  60069.4375  "
      ]
     },
     "execution_count": 8,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "get_hexinfo(ra, dec, probs, 90, 'r', 60069, best=False, camera=\"decam\")"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "a6b8348a",
   "metadata": {},
   "outputs": [],
   "source": []
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3 (ipykernel)",
   "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.9.16"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 5
}