Adding evolved stars

bmorris3 · Dec 4, 2018 · 268801e · 268801e
1 parent fb08d30
commit 268801e
Show file tree

Hide file tree

Showing 6 changed files with 94 additions and 71 deletions.
diff --git a/arcesetc/data/archive.hdf5 b/arcesetc/data/archive.hdf5
diff --git a/arcesetc/data/sptype_dict.json b/arcesetc/data/sptype_dict.json
@@ -1 +1 @@
-{"G2IV": "51 Peg", "K4V": "HD 79555", "K7": "GJ 9781A", "K0": "HD 145742", "G0V": "HD39587", "G3VaHdel1": "HD86728", "B6IV": "HR 6943", "K0Ve": "HD87884", "K6VeFe-1": "HD 88230", "G8V": "HD62613", "G8IVk": "HD 222107", "K2": "GJ 705", "M2.0Ve": "GJ4099", "K5Ve": "EQ Vir", "G5V:": "HD134319", "K2III-IV": "HD 6497", "G7V": "HD 67767", "K3V": "HR 8832", "G1.5V": "HD 129333", "G0Vs": "KIC9652680", "G2V": "HD98230", "G3Va": "HD 10697", "K5V": "HD149957", "K7.5Ve": "HD151288", "K7V": "61CygB", "G5": "Kepler-63", "K0V": "sigma Draconis", "K2V": "HD 73667", "K2.5V": "HD200560", "G3V": "HD68017", "G5V": "Kepler-17", "G9V": "HD 42250", "K1V": "HD221639", "M0.5V": "HD 209290", "WN8h": "WR124", "K3.5V": "HD47752", "sdO2VIIIHe5": "BD+28 4211", "F7V": "HD 89744", "K6V": "HD148467", "G1.5IV-VFe-1": "HD34411", "B3V": "HR 3454", "B9.5V": "HR5501", "K2Ve": "HD45088"}
+{"M2V": "GJ4099", "G8IVk": "HD 222107", "K1V": "HD221639", "G2IV": "51 Peg", "G1.5V": "HD 129333", "K2III-IV": "HD 6497", "G0V": "KIC 9652680", "F7V": "HD 89744", "K5V": "EQ Vir", "M0V": "HD 209290", "K3.5V": "HD47752", "B6IV": "HR 6943", "K2V": "HD 73667", "K7V": "61CygB", "K8V": "HD151288", "G2V": "HD98230",  "G8V": "HD62613", "K5V": "HD149957", "WN8h": "WR124", "G3V": "HD 10697", "B9V": "HR5501", "G0V": "HD39587", "sdO2VIIIHe5": "BD+28 4211", "K0V": "HD87884", "K7": "GJ 9781A", "G7V": "HD 67767", "G5": "Kepler-63", "G1.5IV-VFe-1": "HD34411", "K6V": "HD148467", "K4V": "HD 79555", "K0V": "sigma Draconis", "K3V": "HR 8832", "F3V": "KIC8462852", "G9V": "HD 42250", "B3V": "HR 3454", "K0": "HD 145742", "G5V": "Kepler-17", "G3V": "HD86728"}
diff --git a/arcesetc/util.py b/arcesetc/util.py
@@ -15,7 +15,7 @@
 sptype_to_temp = load(open(os.path.join(directory, 'data',
                                         'sptype_to_temp.json'), 'r'))
 spectral_types = [key for key in sptype_to_temp.keys() if key in sptypes]
-temps = np.array([sptype_to_temp[key] for key in spectral_types])
+temps = np.array([sptype_to_temp[key] for key in spectral_types if key in sptype_to_temp])
 
 
 def closest_sptype(sptype):
@@ -32,6 +32,8 @@ def closest_sptype(sptype):
     closest_spectral_type : str
         Closest spectral type available in the archive
     """
+    if sptype in sptypes:
+        return sptype
     return spectral_types[np.argmin(np.abs(sptype_to_temp[sptype] - temps))]
 
 
@@ -64,7 +66,7 @@ def available_sptypes():
     sptypes : list
         List of available spectral types
     """
-    return sorted(spectral_types)
+    return sorted(sptypes.keys())
 
 
 def get_closest_order(matrix, wavelength):

diff --git a/notebooks/create_archive.ipynb b/notebooks/create_archive.ipynb
@@ -60,7 +60,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 5,
+   "execution_count": 4,
    "metadata": {},
    "outputs": [
     {
@@ -69,7 +69,6 @@
      "text": [
       "WR124\n",
       "KIC 9652680\n",
-      "KIC9652680\n",
       "GJ 9781A\n",
       "HAT P-11\n",
       "HD 209290\n",
@@ -148,12 +147,11 @@
       "HD 89744\n",
       "EPIC 211928486\n",
       "EPIC 211966629\n",
-      "EPIC211928486\n",
-      "EPIC211966629\n",
       "HR 3454\n",
       "HR 6943\n",
       "HR5501\n",
-      "BD28_4211\n"
+      "BD28_4211\n",
+      "KIC8462852\n"
      ]
     }
    ],
@@ -209,30 +207,46 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 6,
+   "execution_count": 19,
    "metadata": {},
    "outputs": [],
    "source": [
-    "archive = h5py.File('../arcesetc/data/archive.hdf5', 'r+')\n",
-    "sptype_dictionary = dict()\n",
+    "# archive = h5py.File('../arcesetc/data/archive.hdf5', 'r+')\n",
+    "# sptype_dictionary = dict()\n",
     "\n",
-    "for target in archive:\n",
-    "    dset = archive[target]\n",
-    "    sptype = dset.attrs['SP_TYPE']\n",
-    "    if len(sptype) > 0:  \n",
-    "        sptype_dictionary[sptype] = dset.attrs['NAME']\n",
-    "    elif target == '61CygA': \n",
-    "        sptype_dictionary['K5V'] = dset.attrs['NAME']\n",
-    "    elif target == '61CygB': \n",
-    "        sptype_dictionary['K7V'] = dset.attrs['NAME']\n",
+    "# for target in archive:\n",
+    "#     dset = archive[target]\n",
+    "#     sptype = dset.attrs['SP_TYPE']\n",
+    "#     if target == '61CygA': \n",
+    "#         sptype_dictionary['K5V'] = dset.attrs['NAME']\n",
+    "#     elif target == '61CygB': \n",
+    "#         sptype_dictionary['K7V'] = dset.attrs['NAME']\n",
+    "#     elif target == 'GJ4099' or target == 'GJ 4099': \n",
+    "#         sptype_dictionary['M2V'] = dset.attrs['NAME']\n",
+    "#     elif target == 'HD86728': \n",
+    "#         sptype_dictionary['G3V'] = dset.attrs['NAME']\n",
+    "#     elif target == 'HD200560': \n",
+    "#         sptype_dictionary['G3V'] = dset.attrs['NAME']\n",
+    "#     elif target == 'HD151288': \n",
+    "#         sptype_dictionary['K8V'] = dset.attrs['NAME']\n",
+    "#     elif target == 'HD 88230': \n",
+    "#         sptype_dictionary['K6V'] = dset.attrs['NAME']\n",
+    "#     elif target == 'KIC 9652680': \n",
+    "#         sptype_dictionary['G0V'] = dset.attrs['NAME']\n",
+    "#     elif target == 'HD 209290':\n",
+    "#         sptype_dictionary['M0V'] = dset.attrs['NAME']\n",
+    "#     elif target == 'HD 10697': \n",
+    "#         sptype_dictionary['']\n",
+    "#     else: \n",
+    "#         sptype_dictionary[sptype] = dset.attrs['NAME']\n",
     "\n",
-    "from json import dump\n",
-    "dump(sptype_dictionary, open('../arcesetc/data/sptype_dict.json', 'w'))"
+    "# from json import dump\n",
+    "# dump(sptype_dictionary, open('../arcesetc/data/sptype_dict.json', 'w'))"
    ]
   },
   {
    "cell_type": "code",
-   "execution_count": 7,
+   "execution_count": 13,
    "metadata": {},
    "outputs": [
     {
@@ -246,8 +260,6 @@
       "EK Dra [7.61] G1.5V\n",
       "EPIC 211928486 [13.52] K5V\n",
       "EPIC 211966629 [13.26] K4V\n",
-      "EPIC211928486 [13.52] K5V\n",
-      "EPIC211966629 [13.26] K4V\n",
       "EQ Vir [9.372] K5Ve\n",
       "GJ 4099 [11.36] M2.0Ve\n",
       "GJ 705 [8.17] K2\n",
@@ -305,7 +317,7 @@
       "HR 8832 [5.57] K3V\n",
       "HR5501 [5.665] B9.5V\n",
       "KIC 9652680 [11.45] G0Vs\n",
-      "KIC9652680 [11.45] G0Vs\n",
+      "KIC8462852 [12.01] F3V\n",
       "Kepler-17 [14.] G5V\n",
       "Kepler-63 [12.02] G5\n",
       "WR124 [11.5] WN8h\n",

diff --git a/notebooks/example.ipynb b/notebooks/example.ipynb
@@ -149,7 +149,7 @@
      "name": "stderr",
      "output_type": "stream",
      "text": [
-      "../arcesetc/plots.py:141: RuntimeWarning: invalid value encountered in sqrt\n",
+      "../arcesetc/plots.py:178: RuntimeWarning: invalid value encountered in sqrt\n",
       "  sn = flux / np.sqrt(flux)\n"
      ]
     },
@@ -195,35 +195,49 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 5,
+   "execution_count": 9,
    "metadata": {},
    "outputs": [
     {
      "data": {
       "text/plain": [
        "['B3V',\n",
+       " 'B6IV',\n",
+       " 'B9V',\n",
+       " 'F3V',\n",
        " 'F7V',\n",
        " 'G0V',\n",
+       " 'G1.5IV-VFe-1',\n",
+       " 'G1.5V',\n",
+       " 'G2IV',\n",
        " 'G2V',\n",
        " 'G3V',\n",
+       " 'G5',\n",
        " 'G5V',\n",
        " 'G7V',\n",
+       " 'G8IVk',\n",
        " 'G8V',\n",
        " 'G9V',\n",
+       " 'K0',\n",
        " 'K0V',\n",
        " 'K1V',\n",
+       " 'K2III-IV',\n",
        " 'K2V',\n",
+       " 'K3.5V',\n",
        " 'K3V',\n",
        " 'K4V',\n",
        " 'K5V',\n",
        " 'K6V',\n",
+       " 'K7',\n",
        " 'K7V',\n",
-       " 'M0.5V',\n",
+       " 'K8V',\n",
+       " 'M0V',\n",
+       " 'M2V',\n",
        " 'WN8h',\n",
        " 'sdO2VIIIHe5']"
       ]
      },
-     "execution_count": 5,
+     "execution_count": 9,
      "metadata": {},
      "output_type": "execute_result"
     }
@@ -270,6 +284,41 @@
     "signal_to_noise_to_exp_time(sptype, wavelength, V, signal_to_noise)"
    ]
   },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "#### We have limited support for evolved stars where available: "
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 7,
+   "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/latex": [
+       "$2343.9419 \\; \\mathrm{s}$"
+      ],
+      "text/plain": [
+       "<Quantity 2343.94190525 s>"
+      ]
+     },
+     "execution_count": 7,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "source": [
+    "sptype = 'B6IV'\n",
+    "wavelength = 6562 * u.Angstrom\n",
+    "signal_to_noise = 30\n",
+    "V = 12\n",
+    "\n",
+    "signal_to_noise_to_exp_time(sptype, wavelength, V, signal_to_noise)"
+   ]
+  },
   {
    "cell_type": "code",
    "execution_count": null,

diff --git a/notebooks/target_paths.txt b/notebooks/target_paths.txt
@@ -1,17 +1,5 @@
 /Users/bmmorris/data/arcesetc/wr124.0002.wfrmcpc.fits
 /Users/bmmorris/data/freckles/UT160703_KIC9652680.0025.wfrmcpc.fits
-/Users/bmmorris/data/freckles/UT160703_KIC9652680.0026.wfrmcpc.fits
-/Users/bmmorris/data/freckles/UT160703_KIC9652680.0027.wfrmcpc.fits
-/Users/bmmorris/data/freckles/UT160703_KIC9652680.0028.wfrmcpc.fits
-/Users/bmmorris/data/freckles/UT160705_KIC9652680.0027.wfrmcpc.fits
-/Users/bmmorris/data/freckles/UT160705_KIC9652680.0028.wfrmcpc.fits
-/Users/bmmorris/data/freckles/UT160705_KIC9652680.0029.wfrmcpc.fits
-/Users/bmmorris/data/freckles/UT160705_KIC9652680.0030.wfrmcpc.fits
-/Users/bmmorris/data/freckles/UT160706_KIC9652680.0030.wfrmcpc.fits
-/Users/bmmorris/data/freckles/UT160706_KIC9652680.0031.wfrmcpc.fits
-/Users/bmmorris/data/freckles/UT160707_KIC9652680.0030.wfrmcpc.fits
-/Users/bmmorris/data/freckles/UT160707_KIC9652680.0031.wfrmcpc.fits
-/Users/bmmorris/data/freckles/UT160707_KIC9652680.0032.wfrmcpc.fits
 /Users/bmmorris/data/freckles/UT160918/gj9781a.0036.wfrmcpc.fits
 /Users/bmmorris/data/freckles/UT160918/hatp11.0024.wfrmcpc.fits
 /Users/bmmorris/data/freckles/UT160918/hatp11.0025.wfrmcpc.fits
@@ -90,24 +78,10 @@
 /Users/bmmorris/data/Q3UW03/UT170911/HD14039.0023.wfrmcpc.fits
 /Users/bmmorris/data/Q3UW03/UT170911/HD14039.0024.wfrmcpc.fits
 /Users/bmmorris/data/Q3UW03/UT170911/HD182488.0021.wfrmcpc.fits
-/Users/bmmorris/data/Q3UW03/UT170911/HD182488.0022.wfrmcpc.fits
 /Users/bmmorris/data/Q3UW03/UT170911/HD221639.0019.wfrmcpc.fits
 /Users/bmmorris/data/Q3UW03/UT170911/HD221639.0020.wfrmcpc.fits
 /Users/bmmorris/data/Q3UW03/UT170911/KIC9652680.0015.wfrmcpc.fits
-/Users/bmmorris/data/Q3UW03/UT170911/KIC9652680.0026.wfrmcpc.fits
 /Users/bmmorris/data/Q3UW04/UT170411/HAT-P-11.0026.wfrmcpc.fits
-/Users/bmmorris/data/Q3UW04/UT170411/HAT-P-11.0027.wfrmcpc.fits
-/Users/bmmorris/data/Q3UW04/UT170411/HAT-P-11.0028.wfrmcpc.fits
-/Users/bmmorris/data/Q3UW04/UT170411/HAT-P-11.0029.wfrmcpc.fits
-/Users/bmmorris/data/Q3UW04/UT170411/HAT-P-11.0030.wfrmcpc.fits
-/Users/bmmorris/data/Q3UW04/UT170411/HAT-P-11.0031.wfrmcpc.fits
-/Users/bmmorris/data/Q3UW04/UT170411/HAT-P-11.0032.wfrmcpc.fits
-/Users/bmmorris/data/Q3UW04/UT170411/HAT-P-11.0033.wfrmcpc.fits
-/Users/bmmorris/data/Q3UW04/UT170612/HAT-P-11.0001.wfrmcpc.fits
-/Users/bmmorris/data/Q3UW04/UT170612/HAT-P-11.0002.wfrmcpc.fits
-/Users/bmmorris/data/Q3UW04/UT170612/HAT-P-11.0009.wfrmcpc.fits
-/Users/bmmorris/data/Q3UW04/UT170612/HAT-P-11.0010.wfrmcpc.fits
-/Users/bmmorris/data/Q3UW04/UT170612/HAT-P-11.0011.wfrmcpc.fits
 /Users/bmmorris/data/Q3UW04/UT170615/HD113827.0058.wfrmcpc.fits
 /Users/bmmorris/data/Q3UW04/UT170615/HD122120.0060.wfrmcpc.fits
 /Users/bmmorris/data/Q3UW04/UT170615/HD127506.0063.wfrmcpc.fits
@@ -141,26 +115,12 @@
 /Users/bmmorris/data/Q4UW09/UT171128/EPIC211928486.0019.wfrmcpc.fits
 /Users/bmmorris/data/Q4UW09/UT171128/EPIC211928486.0020.wfrmcpc.fits
 /Users/bmmorris/data/Q4UW09/UT171128/header_sky.0028.wfrmcpc.fits
-/Users/bmmorris/data/Q4UW09/UT171204/EPIC211928486.0003.wfrmcpc.fits
-/Users/bmmorris/data/Q4UW09/UT171204/EPIC211966629.0002.wfrmcpc.fits
 /Users/bmmorris/data/Q4UW09/UT171204/HD42250.0006.wfrmcpc.fits
 /Users/bmmorris/data/Q4UW09/UT171204/HD42250.0007.wfrmcpc.fits
 /Users/bmmorris/git/nephelion/data/wavelength_calibrated/stack_EPIC211928486_2017-11-28.wfrmcpc.fits
-/Users/bmmorris/git/nephelion/data/wavelength_calibrated/stack_EPIC211928486_2017-12-04.wfrmcpc.fits
-/Users/bmmorris/git/nephelion/data/wavelength_calibrated/stack_EPIC211928486_2017-12-08.wfrmcpc.fits
-/Users/bmmorris/git/nephelion/data/wavelength_calibrated/stack_EPIC211928486_2017-12-11.wfrmcpc.fits
-/Users/bmmorris/git/nephelion/data/wavelength_calibrated/stack_EPIC211928486_2017-12-14.wfrmcpc.fits
-/Users/bmmorris/git/nephelion/data/wavelength_calibrated/stack_EPIC211928486_2017-12-29.wfrmcpc.fits
-/Users/bmmorris/git/nephelion/data/wavelength_calibrated/stack_EPIC211928486_2017-12-31.wfrmcpc.fits
-/Users/bmmorris/git/nephelion/data/wavelength_calibrated/stack_EPIC211928486_2018-02-13.wfrmcpc.fits
 /Users/bmmorris/git/nephelion/data/wavelength_calibrated/stack_EPIC211966629_2017-12-04.wfrmcpc.fits
-/Users/bmmorris/git/nephelion/data/wavelength_calibrated/stack_EPIC211966629_2017-12-08.wfrmcpc.fits
-/Users/bmmorris/git/nephelion/data/wavelength_calibrated/stack_EPIC211966629_2017-12-11.wfrmcpc.fits
-/Users/bmmorris/git/nephelion/data/wavelength_calibrated/stack_EPIC211966629_2017-12-14.wfrmcpc.fits
-/Users/bmmorris/git/nephelion/data/wavelength_calibrated/stack_EPIC211966629_2017-12-29.wfrmcpc.fits
-/Users/bmmorris/git/nephelion/data/wavelength_calibrated/stack_EPIC211966629_2017-12-31.wfrmcpc.fits
-/Users/bmmorris/git/nephelion/data/wavelength_calibrated/stack_EPIC211966629_2018-02-13.wfrmcpc.fits
 /Users/bmmorris/data/freckles/UT161202/HR3454.0016.wfrmcpc.fits
 /Users/bmmorris/data/freckles/UT160918/hr6943.0018.wfrmcpc.fits
 /Users/bmmorris/data/Q1UW09/UT170321/HR5501.0024.wfrmcpc.fits
 /Users/bmmorris/data/Q3UW04/UT160707/BD28_4211.0026.wfrmcpc.fits
+/Users/bmmorris/data/Q2NM02/UT170520/KIC8462852.0001.wfrmcpc.fits