HR-HIP-Bokeh-App-2.ipynb

{
 "cells": [
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "### Astro Lab Using Hertzsprung-Russell Diagrams\n",
    "\n",
    "Based on UNL's Flash-based [HR Diagram explorer](https://astro.unl.edu/naap/hr/animations/hr.html) from their [Astronomy page](https://astro.unl.edu).\n",
    "\n",
    "This is a [Bokeh app](https://docs.bokeh.org/en/latest/) showing HR Diagrams of sets of stars (e.g., 10,000 brightest and 10,000 closest stars) from the Hipparcos catalog.\n",
    "\n",
    "Although I probably can't help you debug your program I would love to hear if you use this animation (or derivative) in your astronomy lessons or observations.\n",
    "\n",
    "Good luck,\n",
    "\n",
    "Stephen Shadle 🌌\n",
    "\n",
    "swshadle@gmail.com"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "HR-HIP-Bokeh-App-2: Changing to t. eff. on x-axis. Isoradial lines are straight now.\n",
    "\n",
    "Published on GitHub https://github.com/swshadle/physics/blob/master/HR-HIP-Bokeh-App-2.ipynb"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "At the terminal/console, run with<br>\n",
    "`bokeh serve --show HR-HIP-Bokeh-App-2.ipynb`"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 1,
   "metadata": {},
   "outputs": [],
   "source": [
    "import pandas as pd\n",
    "import numpy as np\n",
    "from bokeh.layouts import column, row\n",
    "from bokeh.models import (Button, LinearColorMapper, LogColorMapper, ColumnDataSource,\n",
    "                          Div, HoverTool, Label, SingleIntervalTicker, Slider, DataRange1d,\n",
    "                          WheelZoomTool, RadioGroup, ColorBar, LabelSet, CheckboxGroup,\n",
    "                          Select, LogTicker, FuncTickFormatter, Paragraph, CustomJS)\n",
    "from bokeh.palettes import Blues256, Oranges256, Turbo256, Greys256\n",
    "from bokeh.plotting import figure, curdoc, show"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "Enable MathJax equation numbering\n",
    "\n",
    "https://github.com/ipython-contrib/jupyter_contrib_nbextensions/tree/master/src/jupyter_contrib_nbextensions/nbextensions/equation-numbering<br>\n",
    "(Really, just enabling `\\tag{#}` for manual equation numbering.)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "metadata": {},
   "outputs": [],
   "source": [
    "# %%javascript\n",
    "# MathJax.Hub.Config({\n",
    "#     TeX: { equationNumbers: { autoNumber: \"AMS\" } }\n",
    "# });\n",
    "# MathJax.Hub.Queue(\n",
    "#   [\"resetEquationNumbers\", MathJax.InputJax.TeX],\n",
    "#   [\"PreProcess\", MathJax.Hub],\n",
    "#   [\"Reprocess\", MathJax.Hub]\n",
    "# );"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "metadata": {},
   "outputs": [],
   "source": [
    "# check that data files are in the data subdirectory under the current working directory\n",
    "# import os\n",
    "# arr = os.listdir('./data')\n",
    "\n",
    "# print('should find \"I_239_selection.tsv\" & \"named_stars.csv\"')\n",
    "# for f in arr:\n",
    "#     print(f)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 4,
   "metadata": {},
   "outputs": [],
   "source": [
    "# imported the hipparcos catalog from http://balbuceosastropy.blogspot.com/2014/03/construction-of-hertzsprung-russell.html\n",
    "# read in hipparcos data and selected named stars\n",
    "filename = './data/I_239_selection.tsv'\n",
    "df = pd.read_table(filename, skiprows=49, sep=';', header=None, index_col=0,\n",
    "                   names = ['HIP',  # Hipparcos catalog number\n",
    "                            'Vmag', # visual magnitude\n",
    "                            'RA',   # right ascension\n",
    "                            'Dec',  # declination\n",
    "                            'Plx',  # parallax in milliarcseconds\n",
    "                            'B-V',  # color in B-V\n",
    "                            'SpType'], # spectral type\n",
    "                   skipfooter=1,\n",
    "                   engine='python')\n",
    "filename = './data/named_stars.csv'\n",
    "named_stars = pd.read_table(filename, skiprows=1, sep=',', header=None, index_col=1,\n",
    "                   names = ['Name', 'HIP', 'NotUsed'], skipfooter=1, engine='python')\n",
    "named_stars.drop(['NotUsed'], axis=1, inplace=True) # drop 3rd (NotUsed) column from named_stars"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 5,
   "metadata": {},
   "outputs": [],
   "source": [
    "# change any field with spaces to NaN values so we can easily clean all stars with blank fields\n",
    "df = df.applymap(lambda x: np.nan if isinstance(x, str)\n",
    "                       and (not x or x.isspace()) else x)\n",
    "\n",
    "# discard rows with any missing data\n",
    "df = df.dropna()\n",
    "\n",
    "# merge hipparcos (df) with named_stars--we can do this now that we've gotten rid of missing data in df\n",
    "df = df.merge(named_stars, how='left', on='HIP')\n",
    "\n",
    "# now that df has a few named stars, fill in the hipparcos catalog # for stars with no names\n",
    "HIPidx = pd.Index.to_series(df.index)\n",
    "names = HIPidx.map('HIP {:d}'.format)\n",
    "values = {'Name': names}\n",
    "df.fillna(value=values, inplace=True)\n",
    "\n",
    "# numeric data is currently stored as strings. convert to floating point numbers.\n",
    "df['Vmag'] = df['Vmag'].astype(np.float)\n",
    "df['RA']   = df['RA'].astype(np.float)\n",
    "df['Dec']  = df['Dec'].astype(np.float)\n",
    "df['Plx']  = df['Plx'].astype(np.float)\n",
    "df['B-V']  = df['B-V'].astype(np.float)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 6,
   "metadata": {
    "scrolled": false
   },
   "outputs": [],
   "source": [
    "# produce a comma-separated list of named stars with commented names for use in the next step\n",
    "# for n in named_stars.index:\n",
    "#     print(f'{n}, # {named_stars[\"Name\"][n]}')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 7,
   "metadata": {},
   "outputs": [],
   "source": [
    "# all named stars are on the following list. any stars can be added or removed from the list\n",
    "# of displayed simply by commenting or uncommenting them.\n",
    "select_HIPs_2b_labeled = [#0, # The sun\n",
    "#                 664, # Belenos         \n",
    "#                 677, # Alpheratz\n",
    "#                 746, # Caph\n",
    "#                 1067, # Algenib\n",
    "#                 1547, # Citadelle         \n",
    "#                 2081, # Ankaa\n",
    "#                 2247, # Felixvarela       \n",
    "#                 2920, # Fulu              \n",
    "#                 3179, # Shedir\n",
    "#                 3419, # Diphda\n",
    "#                 3479, # Cocibolca         \n",
    "#                 3527, # Musica            \n",
    "#                 3821, # Achird            \n",
    "                3829, # Van Maanen 2\n",
    "#                 4422, # Castula           \n",
    "#                 4427, # γ-Cassiopeiae\n",
    "#                 4780, # Solaris           \n",
    "#                 5054, # Nenque            \n",
    "#                 5348, # Wurren            \n",
    "#                 5447, # Mirach\n",
    "#                 5529, # Emiw              \n",
    "#                 5737, # Revati            \n",
    "#                 6411, # Adhil             \n",
    "#                 6686, # Ruchbah           \n",
    "#                 7097, # Alpherg           \n",
    "#                 7251, # Bosona            \n",
    "#                 7513, # Titawin           \n",
    "#                 7588, # Achernar\n",
    "#                 7607, # Nembus            \n",
    "#                 8198, # Torcular          \n",
    "#                 8375, # Itonda            \n",
    "#                 8645, # Baten Kaitos      \n",
    "#                 8796, # Mothallah         \n",
    "#                 8832, # Mesarthim         \n",
    "#                 8886, # Segin             \n",
    "#                 8903, # Sheratan          \n",
    "#                 9487, # Alrescha          \n",
    "#                 9640, # Almaak\n",
    "#                 9884, # Hamal\n",
    "#                 10813, # Lionrock          \n",
    "#                 10826, # Mira\n",
    "                11767, # Polaris\n",
    "#                 12191, # Buna              \n",
    "#                 12706, # Kaffaljidhma      \n",
    "#                 12961, # Koeia             \n",
    "#                 13061, # Lilii Borea       \n",
    "#                 13192, # Nushagak          \n",
    "#                 13209, # Bharani           \n",
    "#                 13268, # Miram             \n",
    "#                 13288, # Angetenar         \n",
    "#                 13357, # Helvetios         \n",
    "#                 13701, # Azha              \n",
    "#                 13847, # Acamar\n",
    "#                 13993, # Ayeyarwady        \n",
    "#                 14135, # Menkar\n",
    "#                 14322, # Ebla              \n",
    "#                 14576, # Algol\n",
    "#                 14668, # Misam             \n",
    "#                 14838, # Botein            \n",
    "#                 14879, # Dalim             \n",
    "#                 15197, # Zibal             \n",
    "#                 15578, # Intan             \n",
    "#                 15863, # Mirphak\n",
    "#                 16076, # Veritate          \n",
    "#                 16084, # Poerava           \n",
    "#                 16537, # Ran               \n",
    "#                 17096, # Tupi              \n",
    "#                 17448, # Atik              \n",
    "#                 17489, # Celaeno           \n",
    "#                 17499, # Electra           \n",
    "#                 17531, # Taygeta           \n",
    "#                 17573, # Maia              \n",
    "#                 17579, # Asterope          \n",
    "#                 17608, # Merope            \n",
    "#                 17702, # Alcyone\n",
    "#                 17847, # Atlas             \n",
    "#                 17851, # Pleione\n",
    "#                 18543, # Zaurak\n",
    "#                 18614, # Menkib            \n",
    "#                 19587, # Beid              \n",
    "#                 19849, # Keid              \n",
    "#                 20205, # Prima Hyadum      \n",
    "#                 20455, # Secunda Hyadum    \n",
    "#                 20535, # Beemim            \n",
    "#                 20889, # Ain               \n",
    "#                 20894, # Chamukuy          \n",
    "#                 21109, # Hoggar            \n",
    "#                 21393, # Theemin           \n",
    "                21421, # Aldebaran\n",
    "#                 21594, # Sceptrum          \n",
    "#                 22449, # Tabit             \n",
    "#                 22491, # Mouhoun           \n",
    "#                 23015, # Hassaleh          \n",
    "#                 23416, # Almaaz            \n",
    "#                 23453, # Saclateni         \n",
    "#                 23767, # Haedus            \n",
    "#                 23875, # Cursa             \n",
    "#                 24003, # Mago              \n",
    "                24186, # Kapteyn's Star\n",
    "                24436, # Rigel\n",
    "                24608, # Capella\n",
    "                25336, # Bellatrix\n",
    "#                 25428, # Alnath\n",
    "#                 25606, # Nihal\n",
    "#                 25930, # Mintaka\n",
    "#                 25985, # Arneb\n",
    "#                 26207, # Meissa            \n",
    "#                 26241, # Hatysa            \n",
    "#                 26311, # Alnilam\n",
    "#                 26380, # Bubup             \n",
    "#                 26451, # Tianguan          \n",
    "#                 26634, # Phact             \n",
    "#                 26727, # Alnitak\n",
    "                27366, # Saiph\n",
    "#                 27628, # Wazn              \n",
    "                27989, # Betelgeuse\n",
    "#                 28360, # Menkalinan\n",
    "#                 28380, # Mahasim           \n",
    "#                 29034, # Elkurud           \n",
    "#                 29550, # Amadioha          \n",
    "#                 29655, # Propus            \n",
    "#                 30089, # Red Rectangle\n",
    "#                 30122, # Furud             \n",
    "#                 30324, # Mirzam\n",
    "#                 30343, # Tejat\n",
    "                30438, # Canopus\n",
    "#                 30860, # Lucilinburhuc     \n",
    "#                 30905, # Lusitania         \n",
    "#                 31681, # Alhena\n",
    "#                 31895, # Nosaxa            \n",
    "#                 32246, # Mebsuta           \n",
    "                32349, # Sirius\n",
    "#                 32362, # Alzirr            \n",
    "#                 32916, # Nervia            \n",
    "#                 33579, # Adhara\n",
    "#                 33719, # Citala            \n",
    "#                 33856, # Unurgunite        \n",
    "#                 34045, # Muliphein         \n",
    "#                 34088, # Mekbuda           \n",
    "#                 34444, # Wezen (δ-Canis Majoris)\n",
    "#                 35550, # Wasat             \n",
    "#                 35904, # Aludra (η-Canis Majoris)\n",
    "#                 36188, # Gomeisa           \n",
    "                36208, # Luyten's Star\n",
    "#                 36850, # Castor\n",
    "#                 37265, # Jishui            \n",
    "                37279, # Procyon\n",
    "#                 37284, # Ceibo\n",
    "                37826, # Pollux\n",
    "#                 38041, # Tapecue           \n",
    "#                 38170, # Azmidi            \n",
    "#                 39429, # Naos (ζ-Puppis)\n",
    "#                 39757, # Tureis            \n",
    "#                 39953, # γ²-Vel\n",
    "#                 40167, # Tegmine           \n",
    "#                 40526, # Tarf              \n",
    "#                 40687, # Nasti             \n",
    "#                 40881, # Piautos           \n",
    "#                 41037, # Avior\n",
    "#                 41075, # Alsciaukat        \n",
    "#                 41704, # Muscida           \n",
    "#                 42402, # Minchir           \n",
    "#                 42446, # Gakyid            \n",
    "#                 42556, # Meleph            \n",
    "#                 42806, # Asellus Borealis  \n",
    "#                 42911, # Asellus Australis \n",
    "#                 42913, # Alsephina (δ-Velorum)\n",
    "#                 43109, # Ashlesha          \n",
    "#                 43587, # Copernicus        \n",
    "#                 43674, # Stribor           \n",
    "#                 44066, # Acubens           \n",
    "#                 44127, # Talitha           \n",
    "#                 44471, # Alkaphrah         \n",
    "#                 44816, # Suhail (λ-Vela)\n",
    "#                 44946, # Nahn              \n",
    "#                 45238, # Miaplacidus\n",
    "#                 45556, # Aspidiske (ι-Carina)\n",
    "#                 45941, # κ-Vel\n",
    "#                 46390, # Alphard\n",
    "#                 46471, # Intercrus         \n",
    "#                 46750, # Alterf            \n",
    "#                 47087, # Illyrian          \n",
    "#                 47202, # Kalausi           \n",
    "#                 47431, # Ukdah             \n",
    "#                 47508, # Subra             \n",
    "#                 48235, # Natasha           \n",
    "#                 48356, # Zhang             \n",
    "#                 48455, # Rasalas           \n",
    "#                 48615, # Felis             \n",
    "#                 48711, # Bibha             \n",
    "                49669, # Regulus\n",
    "#                 50335, # Adhafera          \n",
    "#                 50372, # Tania Borealis    \n",
    "#                 50583, # Algieba\n",
    "#                 50801, # Tania Australis   \n",
    "#                 52521, # Macondo           \n",
    "#                 53229, # Praecipua         \n",
    "#                 53721, # Chalawan          \n",
    "#                 53740, # Alkes             \n",
    "#                 53910, # Merak\n",
    "#                 54061, # Dubhe\n",
    "#                 54158, # Dingolay          \n",
    "#                 54872, # Zosma (δ-Leo)\n",
    "#                 54879, # Chertan           \n",
    "#                 55174, # Hunahpu           \n",
    "#                 55203, # Alula Australis   \n",
    "#                 55219, # Alula Borealis    \n",
    "#                 55664, # Shama             \n",
    "#                 56211, # Giausar           \n",
    "#                 56508, # Formosa           \n",
    "#                 56572, # Sagarmatha        \n",
    "#                 57291, # Uklun             \n",
    "#                 57370, # Flegetonte        \n",
    "#                 57399, # Taiyangshou       \n",
    "#                 57632, # Denebola\n",
    "#                 57757, # Zavijava          \n",
    "#                 57820, # Aniara            \n",
    "#                 57939, # Groombridge 1830\n",
    "#                 58001, # Phad\n",
    "#                 58952, # Tonatiuh          \n",
    "#                 59196, # δ-Centauri\n",
    "#                 59199, # Alchiba           \n",
    "#                 59747, # Imai              \n",
    "#                 59774, # Megrez\n",
    "#                 59803, # Gienah (γ-Corvi)\n",
    "#                 60129, # Zaniah            \n",
    "#                 60260, # Ginan             \n",
    "#                 60644, # Tupa              \n",
    "#                 60718, # Acrux (α-Crucis)\n",
    "#                 60936, # 3C 273\n",
    "#                 60965, # Algorab           \n",
    "#                 61084, # γ-Crucis\n",
    "#                 61177, # Funi              \n",
    "#                 61317, # Chara             \n",
    "#                 61359, # Kraz              \n",
    "#                 61932, # γ-Centauri\n",
    "#                 61941, # Porrima           \n",
    "#                 62223, # La Superba        \n",
    "#                 62423, # Tianyi            \n",
    "#                 62434, # Mimosa\n",
    "#                 62956, # Alioth\n",
    "#                 63076, # Taiyi             \n",
    "#                 63090, # Minelauva         \n",
    "#                 63125, # Cor Caroli\n",
    "#                 63608, # Vindemiatrix\n",
    "#                 64241, # Diadem            \n",
    "#                 65378, # Mizar\n",
    "                65474, # Spica\n",
    "#                 65477, # Alcor\n",
    "#                 66047, # Dofida            \n",
    "#                 66192, # Liesma            \n",
    "#                 66249, # Heze              \n",
    "#                 66657, # ε-Centauri\n",
    "#                 67301, # Alkaid\n",
    "#                 67927, # Muphrid           \n",
    "#                 68002, # ζ-Centauri\n",
    "#                 68702, # Agena\n",
    "#                 68756, # Thuban\n",
    "#                 68933, # Menkent (θ-Centauri)\n",
    "#                 69427, # Kang              \n",
    "                69673, # Arcturus\n",
    "#                 69701, # Syrma             \n",
    "#                 69732, # Xuange            \n",
    "#                 69974, # Khambalia         \n",
    "#                 70755, # Elgafar           \n",
    "                70890, # Proxima\n",
    "#                 71075, # Seginus           \n",
    "#                 71352, # η-Centauri\n",
    "#                 71681, # α-Centauri\n",
    "                71683, # Rigil Kent\n",
    "#                 71860, # α-Lupi\n",
    "#                 72105, # Izar\n",
    "#                 72339, # Monch             \n",
    "#                 72487, # Merga             \n",
    "#                 72607, # Kocab\n",
    "#                 72622, # Zubenelgenubi     \n",
    "#                 72845, # Arcalis           \n",
    "#                 73136, # Baekdu            \n",
    "#                 73555, # Nekkar            \n",
    "#                 73714, # Brachium          \n",
    "#                 74785, # Zubeneschamali    \n",
    "#                 74961, # Nikawiy           \n",
    "#                 75097, # Pherkad           \n",
    "#                 75411, # Alkalurops        \n",
    "#                 75458, # Edasich           \n",
    "#                 75695, # Nusakan           \n",
    "#                 76267, # Alphekka\n",
    "#                 76333, # Zubenelhakrabi    \n",
    "#                 76351, # Karaka            \n",
    "#                 77070, # Unukalhai\n",
    "#                 77450, # Gudja             \n",
    "#                 78104, # Iklil             \n",
    "#                 78265, # Fang              \n",
    "#                 78401, # Dschubba (δ-Scorpii)\n",
    "#                 78820, # Acrab             \n",
    "#                 79043, # Marsic            \n",
    "#                 79219, # Kamuy             \n",
    "#                 79374, # Jabbah            \n",
    "#                 79431, # Sharjah           \n",
    "#                 79593, # Yed Prior         \n",
    "#                 79882, # Yed Posterior     \n",
    "#                 80076, # Hunor             \n",
    "#                 80112, # Alniyat           \n",
    "#                 80331, # Athebyne          \n",
    "#                 80463, # Cujam             \n",
    "#                 80687, # Timir             \n",
    "                80763, # Antares\n",
    "#                 80816, # Kornephoros       \n",
    "#                 80838, # Ogma              \n",
    "#                 80883, # Marfik            \n",
    "#                 81022, # Rosaliadecastro   \n",
    "#                 81266, # Paikauhale        \n",
    "#                 82273, # Atria\n",
    "#                 82396, # Larawag           \n",
    "#                 82514, # Xamidimura        \n",
    "#                 82545, # Pipirima          \n",
    "#                 82651, # Mahsati           \n",
    "#                 83547, # Rapeto            \n",
    "#                 83608, # Alrakis           \n",
    "#                 83895, # Aldhibah          \n",
    "#                 84012, # Sabik             \n",
    "#                 84345, # Rasalgethi\n",
    "#                 84379, # Sarin             \n",
    "#                 84405, # Guniibuu          \n",
    "#                 84787, # Inquill           \n",
    "#                 85670, # Rastaban          \n",
    "#                 85693, # Maasym            \n",
    "#                 85696, # Lesath            \n",
    "#                 85822, # Yildun            \n",
    "#                 85927, # Shaula\n",
    "#                 86032, # Rasalhague\n",
    "#                 86228, # Sargas (θ-Scorpii)\n",
    "#                 86614, # Dziban            \n",
    "#                 86742, # Cebalrai          \n",
    "#                 86782, # Alruba            \n",
    "#                 86796, # Cervantes         \n",
    "#                 87261, # Fuyue             \n",
    "#                 87585, # Grumium           \n",
    "#                 87833, # Etamin\n",
    "                87937, # Barnard's Star\n",
    "#                 88414, # Pincoya           \n",
    "#                 88635, # Alnasl            \n",
    "#                 89341, # Polis             \n",
    "#                 89931, # Kaus Media        \n",
    "#                 90004, # Alasia            \n",
    "#                 90185, # Kaus Australis\n",
    "#                 90344, # Fafnir            \n",
    "#                 90496, # Kaus Borealis     \n",
    "                91262, # Vega\n",
    "#                 91852, # Xihe              \n",
    "#                 92420, # Sheliak\n",
    "#                 92761, # Ainalrami         \n",
    "#                 92855, # Nunki\n",
    "#                 92895, # Kaveh             \n",
    "#                 92946, # Alya              \n",
    "#                 93194, # Sulafat           \n",
    "#                 93506, # Ascella           \n",
    "#                 93747, # Okab              \n",
    "#                 94114, # Meridiana         \n",
    "#                 94141, # Albaldah          \n",
    "#                 94376, # Altais            \n",
    "#                 94481, # Aladfar           \n",
    "#                 94645, # Gumala            \n",
    "#                 95124, # Belel             \n",
    "#                 95241, # Arkab Prior       \n",
    "#                 95262, # Sika              \n",
    "#                 95294, # Arkab Posterior   \n",
    "#                 95347, # Rukbat            \n",
    "#                 95771, # Anser             \n",
    "#                 95947, # Albireo\n",
    "#                 96078, # Uruk              \n",
    "#                 96100, # Alsafi            \n",
    "#                 96295, # Campbell's Star\n",
    "#                 96757, # Sham              \n",
    "#                 97165, # Fawaris           \n",
    "#                 97278, # Tarazed\n",
    "#                 97649, # Altair\n",
    "#                 97938, # Libertas          \n",
    "#                 98036, # Alshain\n",
    "#                 98066, # Terebellum        \n",
    "#                 98298, # Cyg X-1\n",
    "#                 99711, # Phoenicia         \n",
    "#                 99894, # Chechia           \n",
    "#                 100064, # Algedi            \n",
    "#                 100310, # Alshat            \n",
    "#                 100345, # Dabih             \n",
    "#                 100453, # Sadr              \n",
    "#                 100751, # Peacock\n",
    "#                 101421, # Aldulfin          \n",
    "#                 101769, # Rotanev           \n",
    "#                 101958, # Sualocin          \n",
    "                102098, # Deneb\n",
    "#                 102488, # Aljanah           \n",
    "#                 102618, # Albali            \n",
    "#                 104382, # Polaris Australis \n",
    "#                 104987, # Kitalpha          \n",
    "#                 105199, # Alderamin\n",
    "#                 106032, # Alfirk            \n",
    "#                 106278, # Sadalsuud         \n",
    "#                 106786, # Bunda             \n",
    "#                 106824, # Samaya            \n",
    "#                 106985, # Nashira           \n",
    "#                 107136, # Azelfafage        \n",
    "#                 107315, # Enif\n",
    "#                 107556, # Deneb Algedi      \n",
    "#                 108085, # Aldhanab          \n",
    "#                 108917, # Kurhah            \n",
    "#                 109074, # Sadalmelik\n",
    "#                 109268, # Alnair\n",
    "#                 109427, # Biham             \n",
    "#                 110003, # Ancha             \n",
    "#                 110395, # Sadachbia         \n",
    "#                 110893, # Kruger 60\n",
    "#                 111710, # Situla            \n",
    "#                 112029, # Homam             \n",
    "#                 112122, # Tiaki             \n",
    "#                 112158, # Matar             \n",
    "#                 112247, # Babcock's Star\n",
    "#                 112748, # Sadalbari         \n",
    "#                 113136, # Skat              \n",
    "                113368, # Fomalhaut\n",
    "#                 113881, # Scheat\n",
    "#                 113889, # Fumalsamakah      \n",
    "#                 113963, # Markab\n",
    "#                 115250, # Salm              \n",
    "#                 115623, # Alkarab           \n",
    "#                 116727, # Errai\n",
    "               ]"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 8,
   "metadata": {},
   "outputs": [],
   "source": [
    "stars_2b_labeled = named_stars[named_stars.index.isin(select_HIPs_2b_labeled)]"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 9,
   "metadata": {},
   "outputs": [],
   "source": [
    "# manually add two named star that are not in the Hipparcos catalog: the sun and wolf 359\n",
    "stars_2b_labeled = stars_2b_labeled.append({'Name': 'The Sun'}, ignore_index=True)\n",
    "stars_2b_labeled = stars_2b_labeled.append({'Name': 'Wolf 359'}, ignore_index=True)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 10,
   "metadata": {},
   "outputs": [],
   "source": [
    "# how many named stars?\n",
    "# print(len(df[df['Name'].str.contains('HIP')==False]), 'named stars') # 423 named stars"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 11,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Named stars with unusable data (nonpositive parallax):\n",
      "Campbell's Star\n"
     ]
    }
   ],
   "source": [
    "# discarding stars with negative parallax: do we lose any named stars?\n",
    "print('Named stars with unusable data (nonpositive parallax):')\n",
    "count = 0\n",
    "for name in df.loc[(df['Name'].str.contains('HIP')==False) & (df['Plx'] <= 0.0)]['Name']:\n",
    "    count += 1\n",
    "    print(name)\n",
    "if not count:\n",
    "    print('None')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 12,
   "metadata": {},
   "outputs": [],
   "source": [
    "df = df[df.index!=62223] # dumping \"outlier\" La Superba for being too red and messing up my colorbar"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 13,
   "metadata": {},
   "outputs": [],
   "source": [
    "# discard stars with negative parallax\n",
    "df = df[df['Plx'] > 0.0]"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "Solving for absolute magnitude, $M_V$, from apparent magnitude, $m_v$, which we have in data column `'Vmag'`.<br>\n",
    "We'll use the formula for the distance modulus, $m_v - M_V$:\n",
    "$$m_v - M_V = 5\\cdot\\log{d} - 5\\tag{1}$$<br>\n",
    "$$m_v - 5\\cdot\\log{d} + 5 = M_V$$<br>\n",
    "Note that $d = 1/\\omega$, where $d$ is distance measured in parsecs and $\\omega$ is parallax measured in arcseconds.<br>\n",
    "Our data column, `Plx`, is parallax measured in milliarcseconds, so $\\omega = $ `Plx` $/1000$ and $d = 1000/$`Plx`<br>\n",
    "$$m_v - 5\\cdot\\log{\\frac{1}{\\omega}} + 5 = M_V$$<br>\n",
    "$$m_v - 5\\cdot\\log{\\frac{1000}{\\mathtt{Plx}}} + 5 = M_V$$<br>\n",
    "$$m_v - 5(\\log{1000} - \\log{\\mathtt{Plx}}) + 5 = M_V$$<br>\n",
    "$$m_v - 5(3 - \\log{\\mathtt{Plx}}) + 5 = M_V$$<br>\n",
    "$$m_v - 15 + 5\\cdot\\log{\\mathtt{Plx}} + 5 = M_V$$<br>\n",
    "$$m_v + 5\\cdot\\log{\\mathtt{Plx}} - 10 = M_V\\tag{2}$$"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 14,
   "metadata": {},
   "outputs": [],
   "source": [
    "# calculate absolute magnitude and add a new column ('M_V')\n",
    "df['M_V'] = df['Vmag'] + 5*np.log10(df['Plx'])-10"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "Calculate distance in LY and add a new data column<br>\n",
    "$d = 1/p$ when distance ($d$) is in parsecs and parallax ($p$) is in arcseconds, or $d = 1000/\\mathtt{Plx}$<br>\n",
    "Note that Hipparcos parallax data (column `'Plx'`) is in milliarcseconds (MAS), so we multiply\n",
    "by 1000 to get arcseconds).<br>\n",
    "Finally, there are 3.26156 light-years (LY) in one parsec, so the final formula is $LY = 3261.56/\\mathtt{Plx}$"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 15,
   "metadata": {},
   "outputs": [],
   "source": [
    "df['LY'] = 3261.56/df['Plx']"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 16,
   "metadata": {},
   "outputs": [],
   "source": [
    "sun_ly = 0.000015639\n",
    "sun_plx = 3261.56/sun_ly"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "Now that we have absolute magnitude (column `'M_V'`), we can solve for luminosity in terms of multiples of luminosity of the sun, or $\\frac{L}{L_{sun}}$, using:\n",
    "$$M_V = M_{Vsun} - 2.5 \\cdot log\\frac{L}{L_{sun}}\\tag{3}$$<br>\n",
    "$$2.5 \\cdot log\\frac{L}{L_{sun}} = M_{Vsun} - M_V$$<br>\n",
    "$$log\\frac{L}{L_{sun}} = \\frac{M_{Vsun} - M_V}{2.5}$$<br>\n",
    "$$\\frac{L}{L_{sun}} = 10^{\\frac{M_{Vsun} - M_V}{2.5}}\\tag{4}$$<br>\n",
    "http://hosting.astro.cornell.edu/academics/courses/astro201/mag_absolute.htm"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 17,
   "metadata": {},
   "outputs": [],
   "source": [
    "# add a new column ('L_sun') for luminosity in terms of multiples of solar luminosity\n",
    "df['L_sun'] = np.power(10,[(4.83-m)/2.5 for m in df['M_V']]) # M_V_☉ = 4.83"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "Solving for effective temperature in Kelvin, $T_K$, using:\n",
    "$$T_K = \\frac{5601 K}{(B-V+0.4)^{2/3}}\\tag{5}$$<br>\n",
    "Note that this formula breaks when $B-V = -0.4$<br>\n",
    "http://astro.physics.uiowa.edu/ITU/labs/professional-labs/photometry-of-a-globular/part-2-finding-temperature.html"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 18,
   "metadata": {},
   "outputs": [],
   "source": [
    "# print(\"We'll have to lose\", len(df[df['B-V']<=-0.4]), \"star(s) because of T_K formula (B-V can't be <= 0.4)\")\n",
    "# print('# of stars before:', len(df))"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 19,
   "metadata": {},
   "outputs": [],
   "source": [
    "df = df[df['B-V']>-0.4]"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 20,
   "metadata": {},
   "outputs": [],
   "source": [
    "# print('# of stars after:', len(df))"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 21,
   "metadata": {},
   "outputs": [],
   "source": [
    "# add a new column ('T_K') for effective temperature in K from color index ('B-V')\n",
    "df['T_K'] = [5601/np.power(c+0.4,2/3) for c in df['B-V']]"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "We want to show slider values with a marker on the HR plot. The sliders' temperature and luminosity can be converted to the HR plot's color index (x-axis) and abs. magnitude (y-axis) using equation (3) for `M_V` and solving equation (5) for `B-V`.<br>\n",
    "$$M_V = M_{Vsun} - 2.5 \\cdot log\\frac{L}{L_{sun}}\\tag{3}$$<br>\n",
    "$$T_K = \\frac{5601 K}{(B-V+0.4)^{2/3}}\\tag{5}$$<br>\n",
    "$$(B-V+0.4)^{2/3} = \\frac{5601 K}{T_K}$$<br>\n",
    "$$B-V+0.4 = \\left(\\frac{5601 K}{T_K}\\right)^{3/2}$$<br>\n",
    "$$B-V = \\left( \\frac{5601 K}{T_K} \\right) ^{3/2} - 0.4\\tag{6}$$"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "Now that we have $T_K$ and $\\frac{L}{L_{sun}}$, we can solve for $\\frac{R}{R_{sun}}$, using<br>\n",
    "$$\\frac{R}{R_{sun}} = \\frac{\\sqrt{L/L_{sun}}}{(T_K/T_{sun})^2}\\tag{7}$$"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 22,
   "metadata": {},
   "outputs": [],
   "source": [
    "# temperature in K, B-V color index, and abs. magnitude of the sun\n",
    "T_sun = 5780\n",
    "BV_sun = np.power(5601/T_sun, 1.5) - 0.4\n",
    "M_sun = 4.83"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 23,
   "metadata": {
    "scrolled": true
   },
   "outputs": [],
   "source": [
    "# add a new column ('R_sun') for radius in units of solar radii from luminosity and effective temperature in K\n",
    "df['R_sun']=np.around(np.sqrt(df['L_sun'])/(df['T_K']/T_sun)**2, decimals=2) # T_⊙ = 5780"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 24,
   "metadata": {},
   "outputs": [
    {
     "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>Vmag</th>\n",
       "      <th>RA</th>\n",
       "      <th>Dec</th>\n",
       "      <th>Plx</th>\n",
       "      <th>B-V</th>\n",
       "      <th>SpType</th>\n",
       "      <th>Name</th>\n",
       "      <th>M_V</th>\n",
       "      <th>LY</th>\n",
       "      <th>L_sun</th>\n",
       "      <th>T_K</th>\n",
       "      <th>R_sun</th>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>HIP</th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>89341</th>\n",
       "      <td>3.84</td>\n",
       "      <td>273.440870</td>\n",
       "      <td>-21.058830</td>\n",
       "      <td>0.11</td>\n",
       "      <td>0.195</td>\n",
       "      <td>B2III:</td>\n",
       "      <td>Polis</td>\n",
       "      <td>-10.953037</td>\n",
       "      <td>29650.545455</td>\n",
       "      <td>2.056907e+06</td>\n",
       "      <td>7917.491313</td>\n",
       "      <td>764.34</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>80763</th>\n",
       "      <td>1.06</td>\n",
       "      <td>247.351948</td>\n",
       "      <td>-26.431946</td>\n",
       "      <td>5.40</td>\n",
       "      <td>1.865</td>\n",
       "      <td>M1Ib + B2.5V</td>\n",
       "      <td>Antares</td>\n",
       "      <td>-5.278031</td>\n",
       "      <td>603.992593</td>\n",
       "      <td>1.104619e+04</td>\n",
       "      <td>3247.531970</td>\n",
       "      <td>332.93</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "       Vmag          RA        Dec   Plx    B-V        SpType  \\\n",
       "HIP                                                             \n",
       "89341  3.84  273.440870 -21.058830  0.11  0.195        B2III:   \n",
       "80763  1.06  247.351948 -26.431946  5.40  1.865  M1Ib + B2.5V   \n",
       "\n",
       "                     Name        M_V            LY         L_sun          T_K  \\\n",
       "HIP                                                                             \n",
       "89341  Polis              -10.953037  29650.545455  2.056907e+06  7917.491313   \n",
       "80763             Antares  -5.278031    603.992593  1.104619e+04  3247.531970   \n",
       "\n",
       "        R_sun  \n",
       "HIP            \n",
       "89341  764.34  \n",
       "80763  332.93  "
      ]
     },
     "execution_count": 24,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "# more data cleaning. which named stars have the largest calculated solar radii?\n",
    "df.nlargest(400,'R_sun').loc[(df['Name'].str.contains('HIP')==False)] #Polis 748.67 Antares 326.10"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 25,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "332.93"
      ]
     },
     "execution_count": 25,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "# what is Antares's radius?\n",
    "float(df[df['Name']=='Antares']['R_sun'])"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 26,
   "metadata": {},
   "outputs": [
    {
     "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>Vmag</th>\n",
       "      <th>RA</th>\n",
       "      <th>Dec</th>\n",
       "      <th>Plx</th>\n",
       "      <th>B-V</th>\n",
       "      <th>SpType</th>\n",
       "      <th>Name</th>\n",
       "      <th>M_V</th>\n",
       "      <th>LY</th>\n",
       "      <th>L_sun</th>\n",
       "      <th>T_K</th>\n",
       "      <th>R_sun</th>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>HIP</th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>80763</th>\n",
       "      <td>1.06</td>\n",
       "      <td>247.351948</td>\n",
       "      <td>-26.431946</td>\n",
       "      <td>5.4</td>\n",
       "      <td>1.865</td>\n",
       "      <td>M1Ib + B2.5V</td>\n",
       "      <td>Antares</td>\n",
       "      <td>-5.278031</td>\n",
       "      <td>603.992593</td>\n",
       "      <td>11046.189271</td>\n",
       "      <td>3247.53197</td>\n",
       "      <td>332.93</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "       Vmag          RA        Dec  Plx    B-V        SpType     Name  \\\n",
       "HIP                                                                     \n",
       "80763  1.06  247.351948 -26.431946  5.4  1.865  M1Ib + B2.5V  Antares   \n",
       "\n",
       "            M_V          LY         L_sun         T_K   R_sun  \n",
       "HIP                                                            \n",
       "80763 -5.278031  603.992593  11046.189271  3247.53197  332.93  "
      ]
     },
     "execution_count": 26,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "df[df['R_sun']==float(df[df['Name']=='Antares']['R_sun'])] # horrible way to find a star but a good way to limit a database to R_suns <= Antares's"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 27,
   "metadata": {},
   "outputs": [
    {
     "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>Vmag</th>\n",
       "      <th>RA</th>\n",
       "      <th>Dec</th>\n",
       "      <th>Plx</th>\n",
       "      <th>B-V</th>\n",
       "      <th>SpType</th>\n",
       "      <th>Name</th>\n",
       "      <th>M_V</th>\n",
       "      <th>LY</th>\n",
       "      <th>L_sun</th>\n",
       "      <th>T_K</th>\n",
       "      <th>R_sun</th>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>HIP</th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>80763</th>\n",
       "      <td>1.06</td>\n",
       "      <td>247.351948</td>\n",
       "      <td>-26.431946</td>\n",
       "      <td>5.40</td>\n",
       "      <td>1.865</td>\n",
       "      <td>M1Ib + B2.5V</td>\n",
       "      <td>Antares</td>\n",
       "      <td>-5.278031</td>\n",
       "      <td>603.992593</td>\n",
       "      <td>1.104619e+04</td>\n",
       "      <td>3247.531970</td>\n",
       "      <td>332.93</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>89341</th>\n",
       "      <td>3.84</td>\n",
       "      <td>273.440870</td>\n",
       "      <td>-21.058830</td>\n",
       "      <td>0.11</td>\n",
       "      <td>0.195</td>\n",
       "      <td>B2III:</td>\n",
       "      <td>Polis</td>\n",
       "      <td>-10.953037</td>\n",
       "      <td>29650.545455</td>\n",
       "      <td>2.056907e+06</td>\n",
       "      <td>7917.491313</td>\n",
       "      <td>764.34</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "       Vmag          RA        Dec   Plx    B-V        SpType  \\\n",
       "HIP                                                             \n",
       "80763  1.06  247.351948 -26.431946  5.40  1.865  M1Ib + B2.5V   \n",
       "89341  3.84  273.440870 -21.058830  0.11  0.195        B2III:   \n",
       "\n",
       "                     Name        M_V            LY         L_sun          T_K  \\\n",
       "HIP                                                                             \n",
       "80763             Antares  -5.278031    603.992593  1.104619e+04  3247.531970   \n",
       "89341  Polis              -10.953037  29650.545455  2.056907e+06  7917.491313   \n",
       "\n",
       "        R_sun  \n",
       "HIP            \n",
       "80763  332.93  \n",
       "89341  764.34  "
      ]
     },
     "execution_count": 27,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "df.loc[(df['Name'].str.contains('HIP')==False) & (df['R_sun']>323.86)]"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 28,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "110714"
      ]
     },
     "execution_count": 28,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "len(df) # length of dataframe before"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 29,
   "metadata": {},
   "outputs": [],
   "source": [
    "# just suspicious of calculated radii for the largest stars. i'm cutting out stars larger than Antares\n",
    "# but you might choose to leave them in\n",
    "df = df[df['R_sun']<=float(df[df['Name']=='Antares']['R_sun'])]"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 30,
   "metadata": {},
   "outputs": [],
   "source": [
    "# add the sun's and wolf 359's particulars to the dataframe\n",
    "df.loc[0] ={'Vmag': -26.74,\n",
    "           'RA': 0.0,\n",
    "           'Dec': 0.0,\n",
    "           'Plx': sun_plx,\n",
    "           'B-V': BV_sun, # 0.656\n",
    "           'SpType': 'G2V',\n",
    "           'Name': 'The Sun',\n",
    "           'M_V': M_sun, # 4.83\n",
    "           'LY': sun_ly,\n",
    "           'L_sun': 1.0,\n",
    "           'T_K': float(T_sun),#5780.0,\n",
    "           'R_sun': 1.0}\n",
    "\n",
    "df = df.append({'Vmag': 13.45,\n",
    "            'RA': 10.945,\n",
    "            'Dec': 7.053,\n",
    "            'Plx': 418,\n",
    "            'B-V': 2.034,\n",
    "            'SpType': 'M6Ve',\n",
    "            'Name': 'Wolf 359',\n",
    "            'M_V': 16.6,\n",
    "            'LY': 7.795,\n",
    "            'L_sun': 0.002,\n",
    "            'T_K': 2800,\n",
    "            'R_sun': 0.21}, ignore_index=True)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 31,
   "metadata": {},
   "outputs": [
    {
     "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>Vmag</th>\n",
       "      <th>RA</th>\n",
       "      <th>Dec</th>\n",
       "      <th>Plx</th>\n",
       "      <th>B-V</th>\n",
       "      <th>SpType</th>\n",
       "      <th>Name</th>\n",
       "      <th>M_V</th>\n",
       "      <th>LY</th>\n",
       "      <th>L_sun</th>\n",
       "      <th>T_K</th>\n",
       "      <th>R_sun</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>110389</th>\n",
       "      <td>-26.74</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>2.085530e+08</td>\n",
       "      <td>0.553908</td>\n",
       "      <td>G2V</td>\n",
       "      <td>The Sun</td>\n",
       "      <td>4.83</td>\n",
       "      <td>0.000016</td>\n",
       "      <td>1.0</td>\n",
       "      <td>5780.0</td>\n",
       "      <td>1.0</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "         Vmag   RA  Dec           Plx       B-V SpType     Name   M_V  \\\n",
       "110389 -26.74  0.0  0.0  2.085530e+08  0.553908    G2V  The Sun  4.83   \n",
       "\n",
       "              LY  L_sun     T_K  R_sun  \n",
       "110389  0.000016    1.0  5780.0    1.0  "
      ]
     },
     "execution_count": 31,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "df[df['Name'].str.contains('Sun')==True] # looking for the sun"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 32,
   "metadata": {},
   "outputs": [
    {
     "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>Vmag</th>\n",
       "      <th>RA</th>\n",
       "      <th>Dec</th>\n",
       "      <th>Plx</th>\n",
       "      <th>B-V</th>\n",
       "      <th>SpType</th>\n",
       "      <th>Name</th>\n",
       "      <th>M_V</th>\n",
       "      <th>LY</th>\n",
       "      <th>L_sun</th>\n",
       "      <th>T_K</th>\n",
       "      <th>R_sun</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>110390</th>\n",
       "      <td>13.45</td>\n",
       "      <td>10.945</td>\n",
       "      <td>7.053</td>\n",
       "      <td>418.0</td>\n",
       "      <td>2.034</td>\n",
       "      <td>M6Ve</td>\n",
       "      <td>Wolf 359</td>\n",
       "      <td>16.6</td>\n",
       "      <td>7.795</td>\n",
       "      <td>0.002</td>\n",
       "      <td>2800.0</td>\n",
       "      <td>0.21</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "         Vmag      RA    Dec    Plx    B-V SpType      Name   M_V     LY  \\\n",
       "110390  13.45  10.945  7.053  418.0  2.034   M6Ve  Wolf 359  16.6  7.795   \n",
       "\n",
       "        L_sun     T_K  R_sun  \n",
       "110390  0.002  2800.0   0.21  "
      ]
     },
     "execution_count": 32,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "df[df['Name'].str.contains('Wolf')==True] # looking for wolf 359"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 33,
   "metadata": {},
   "outputs": [],
   "source": [
    "# scale marker sizes based on star sizes\n",
    "sizemax = df['R_sun'].max()\n",
    "sizemin = df['R_sun'].min()\n",
    "\n",
    "df['markersizes'] = [50*(r-sizemin)/(sizemax-sizemin)+2 for r in df['R_sun']]"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 34,
   "metadata": {},
   "outputs": [],
   "source": [
    "df_close = df.nsmallest(10000,'LY') # get 10k of the closest stars\n",
    "df_bright = df.nsmallest(10000,'Vmag') # get 10k of the brightest stars\n",
    "df_either = pd.merge(df_bright, df_close, how='outer') # stars that are either bright or close\n",
    "df_both   = pd.merge(df_bright, df_close, how='inner') # stars that are both bright and close\n",
    "df_named        = df[df['Name'].str.contains('HIP')==False] # get all stars with common names (without HIP #--hipparcos number designators)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 35,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "10000 bright stars\n",
      "10000 close stars\n",
      "17997 stars that are either bright or close\n",
      "2003 stars that are both bright and close\n",
      "422 total named stars\n"
     ]
    }
   ],
   "source": [
    "df_bright_labels = pd.merge(df_bright, stars_2b_labeled, how='inner')[['T_K', 'M_V', 'Name']]\n",
    "df_close_labels  = pd.merge(df_close,  stars_2b_labeled, how='inner')[['T_K', 'M_V', 'Name']]\n",
    "df_either_labels = pd.merge(df_either, stars_2b_labeled, how='inner')[['T_K', 'M_V', 'Name']]\n",
    "df_both_labels   = pd.merge(df_both,   stars_2b_labeled, how='inner')[['T_K', 'M_V', 'Name']]\n",
    "df_named_labels  = pd.merge(df_named,  stars_2b_labeled, how='inner')[['T_K', 'M_V', 'Name']]\n",
    "\n",
    "print(len(df_bright), 'bright stars')\n",
    "print(len(df_close),  'close stars')\n",
    "print(len(df_either), 'stars that are either bright or close')\n",
    "print(len(df_both),   'stars that are both bright and close')\n",
    "print(len(df_named),  'total named stars')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 36,
   "metadata": {},
   "outputs": [],
   "source": [
    "# make a label for the bottom-center of the star plot showing which set of star. useful when saving a view to disk.\n",
    "label = Label(x=350, y=0, x_units='screen', y_units='screen', text_align='center',#'data'\n",
    "              background_fill_color='black', background_fill_alpha=1, text='')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 37,
   "metadata": {},
   "outputs": [],
   "source": [
    "# callback for the radiobutton group\n",
    "def zero():\n",
    "#     print('button zero')\n",
    "    near_plot.visible         = True\n",
    "    bright_plot.visible       = False\n",
    "    both_plot.visible         = False\n",
    "    either_plot.visible       = False\n",
    "    named_plot.visible        = False\n",
    "\n",
    "    close_lum_labels.visible  = 1 in checkboxgroup.active\n",
    "    bright_lum_labels.visible = False\n",
    "    both_lum_labels.visible   = False\n",
    "    either_lum_labels.visible = False\n",
    "    named_lum_labels.visible  = False\n",
    "\n",
    "    if 0 in checkboxgroup.active:\n",
    "        close_labels.visible         = True\n",
    "        close_labels_shadow.visible  = True\n",
    "        bright_labels.visible        = False\n",
    "        bright_labels_shadow.visible = False\n",
    "        both_labels.visible          = False\n",
    "        both_labels_shadow.visible   = False\n",
    "        either_labels.visible        = False\n",
    "        either_labels_shadow.visible = False\n",
    "        named_labels.visible         = False\n",
    "        named_labels_shadow.visible  = False\n",
    "        \n",
    "    plot.x_range = DataRange1d(only_visible=True, range_padding=0.1, range_padding_units='percent') # 'absolute'\n",
    "    plot.y_range = DataRange1d(only_visible=True, range_padding=0.1, range_padding_units='percent', flipped=True)\n",
    "    \n",
    "    return f'{len(df_close)} nearest stars '\n",
    "\n",
    "def one():\n",
    "#     print('button one')\n",
    "    near_plot.visible         = False\n",
    "    bright_plot.visible       = True\n",
    "    both_plot.visible         = False\n",
    "    either_plot.visible       = False\n",
    "    named_plot.visible        = False\n",
    "    \n",
    "    close_lum_labels.visible  = False\n",
    "    bright_lum_labels.visible = 1 in checkboxgroup.active\n",
    "    both_lum_labels.visible   = False\n",
    "    either_lum_labels.visible = False\n",
    "    named_lum_labels.visible  = False\n",
    "\n",
    "    if 0 in checkboxgroup.active:\n",
    "        close_labels.visible         = False\n",
    "        close_labels_shadow.visible  = False\n",
    "        bright_labels.visible        = True\n",
    "        bright_labels_shadow.visible = True\n",
    "        both_labels.visible          = False\n",
    "        both_labels_shadow.visible   = False\n",
    "        either_labels.visible        = False\n",
    "        either_labels_shadow.visible = False\n",
    "        named_labels.visible         = False\n",
    "        named_labels_shadow.visible  = False\n",
    "\n",
    "    plot.x_range = DataRange1d(only_visible=True, range_padding=0.1, range_padding_units='percent') # 'absolute'\n",
    "    plot.y_range = DataRange1d(only_visible=True, range_padding=1.0, range_padding_units='absolute', flipped=True)\n",
    "\n",
    "    return f'{len(df_bright)} brightest stars '\n",
    "\n",
    "def two():\n",
    "#     print('button two')\n",
    "    near_plot.visible         = False\n",
    "    bright_plot.visible       = False\n",
    "    both_plot.visible         = True\n",
    "    either_plot.visible       = False\n",
    "    named_plot.visible        = False\n",
    "\n",
    "    close_lum_labels.visible  = False\n",
    "    bright_lum_labels.visible = False\n",
    "    both_lum_labels.visible   = 1 in checkboxgroup.active\n",
    "    either_lum_labels.visible = False\n",
    "    named_lum_labels.visible  = False\n",
    "\n",
    "    if 0 in checkboxgroup.active:\n",
    "        close_labels.visible         = False\n",
    "        close_labels_shadow.visible  = False\n",
    "        bright_labels.visible        = False\n",
    "        bright_labels_shadow.visible = False\n",
    "        both_labels.visible          = True\n",
    "        both_labels_shadow.visible   = True\n",
    "        either_labels.visible        = False\n",
    "        either_labels_shadow.visible = False\n",
    "        named_labels.visible         = False\n",
    "        named_labels_shadow.visible  = False\n",
    "\n",
    "    plot.x_range = DataRange1d(only_visible=True, range_padding=0.1, range_padding_units='percent') # 'absolute'\n",
    "    plot.y_range = DataRange1d(only_visible=True, range_padding=1.0, range_padding_units='absolute', flipped=True)\n",
    "\n",
    "    return f'{len(df_both)} near and bright stars '\n",
    "\n",
    "def three():\n",
    "#     print('button three')\n",
    "    near_plot.visible         = False\n",
    "    bright_plot.visible       = False\n",
    "    both_plot.visible         = False\n",
    "    either_plot.visible       = True\n",
    "    named_plot.visible        = False\n",
    "\n",
    "    close_lum_labels.visible  = False\n",
    "    bright_lum_labels.visible = False\n",
    "    both_lum_labels.visible   = False\n",
    "    either_lum_labels.visible = 1 in checkboxgroup.active\n",
    "    named_lum_labels.visible  = False\n",
    "\n",
    "    if 0 in checkboxgroup.active:\n",
    "        close_labels.visible         = False\n",
    "        close_labels_shadow.visible  = False\n",
    "        bright_labels.visible        = False\n",
    "        bright_labels_shadow.visible = False\n",
    "        both_labels.visible          = False\n",
    "        both_labels_shadow.visible   = False\n",
    "        either_labels.visible        = True\n",
    "        either_labels_shadow.visible = True\n",
    "        named_labels.visible         = False\n",
    "        named_labels_shadow.visible  = False\n",
    "\n",
    "    plot.x_range = DataRange1d(only_visible=True, range_padding=0.1, range_padding_units='percent') # 'absolute'\n",
    "    plot.y_range = DataRange1d(only_visible=True, range_padding=1.0, range_padding_units='absolute', flipped=True)\n",
    "\n",
    "    return f'{len(df_either)} either near or bright stars '\n",
    "\n",
    "def four():\n",
    "#     print('button four')\n",
    "    near_plot.visible         = False\n",
    "    bright_plot.visible       = False\n",
    "    both_plot.visible         = False\n",
    "    either_plot.visible       = False\n",
    "    named_plot.visible        = True\n",
    "\n",
    "    close_lum_labels.visible  = False\n",
    "    bright_lum_labels.visible = False\n",
    "    both_lum_labels.visible   = False\n",
    "    either_lum_labels.visible = False\n",
    "    named_lum_labels.visible  = 1 in checkboxgroup.active\n",
    "\n",
    "    if 0 in checkboxgroup.active:\n",
    "        close_labels.visible         = False\n",
    "        close_labels_shadow.visible  = False\n",
    "        bright_labels.visible        = False\n",
    "        bright_labels_shadow.visible = False\n",
    "        both_labels.visible          = False\n",
    "        both_labels_shadow.visible   = False\n",
    "        either_labels.visible        = False\n",
    "        either_labels_shadow.visible = False\n",
    "        named_labels.visible         = True\n",
    "        named_labels_shadow.visible  = True\n",
    "\n",
    "    plot.x_range = DataRange1d(only_visible=True, range_padding=0.1, range_padding_units='percent') # 'absolute'\n",
    "    plot.y_range = DataRange1d(only_visible=True, range_padding=1.0, range_padding_units='absolute', flipped=True)\n",
    "\n",
    "    return f'{len(df_named)} stars with common names '\n",
    "\n",
    "def radiogroup_callback(new):\n",
    "    switcher = {\n",
    "        0: zero,\n",
    "        1: one,\n",
    "        2: two,\n",
    "        3: three,\n",
    "        4: four\n",
    "    }\n",
    "    func = switcher.get(new, lambda: 'Invalid selection') # get the function from switcher dictionary\n",
    "    label.text = func() # execute the function\n",
    "\n",
    "def checkboxgroup_callback(new):\n",
    "    if 0 in new:\n",
    "#         print('item 1 is checked')\n",
    "        if radiogroup.active==0:\n",
    "            close_labels.visible         = True\n",
    "            close_labels_shadow.visible  = True\n",
    "        if radiogroup.active==1:\n",
    "            bright_labels.visible        = True\n",
    "            bright_labels_shadow.visible = True\n",
    "        if radiogroup.active==2:\n",
    "            both_labels.visible          = True\n",
    "            both_labels_shadow.visible   = True\n",
    "        if radiogroup.active==3:\n",
    "            either_labels.visible        = True\n",
    "            either_labels_shadow.visible = True\n",
    "        if radiogroup.active==4:\n",
    "            named_labels.visible         = True\n",
    "            named_labels_shadow.visible  = True\n",
    "    else:\n",
    "        close_labels.visible         = False\n",
    "        close_labels_shadow.visible  = False\n",
    "        bright_labels.visible        = False\n",
    "        bright_labels_shadow.visible = False\n",
    "        both_labels.visible          = False\n",
    "        both_labels_shadow.visible   = False\n",
    "        either_labels.visible        = False\n",
    "        either_labels_shadow.visible = False\n",
    "        named_labels.visible         = False\n",
    "        named_labels_shadow.visible  = False\n",
    "            \n",
    "    if 1 in new:\n",
    "#         print('item 2 is checked')\n",
    "        isoradial_lines.visible = True\n",
    "\n",
    "        if radiogroup.active==0:\n",
    "            close_lum_labels.visible  = True\n",
    "        if radiogroup.active==1:\n",
    "            bright_lum_labels.visible = True\n",
    "        if radiogroup.active==2:\n",
    "            both_lum_labels.visible   = True\n",
    "        if radiogroup.active==3:\n",
    "            either_lum_labels.visible = True\n",
    "        if radiogroup.active==4:\n",
    "            named_lum_labels.visible  = True\n",
    "    else:\n",
    "        isoradial_lines.visible = False\n",
    "\n",
    "        close_lum_labels.visible  = False\n",
    "        bright_lum_labels.visible = False\n",
    "        both_lum_labels.visible   = False\n",
    "        either_lum_labels.visible = False\n",
    "        named_lum_labels.visible  = False"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 38,
   "metadata": {},
   "outputs": [],
   "source": [
    "# CheckboxGroup widget\n",
    "checkboxgroup_labels = ['Show star names', 'Show isoradial lines']\n",
    "checkboxgroup = CheckboxGroup(labels=checkboxgroup_labels, active=[0, 1])\n",
    "checkboxgroup.on_click(checkboxgroup_callback)\n",
    "\n",
    "# RadioGroup widget\n",
    "radiogroup_labels = ['Near stars', 'Bright stars', 'Both near and bright', 'Either near or bright', 'Named stars']\n",
    "radiogroup = RadioGroup(labels=radiogroup_labels, active=0)\n",
    "radiogroup.on_click(radiogroup_callback)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 39,
   "metadata": {},
   "outputs": [],
   "source": [
    "source_close  = ColumnDataSource(df_close) ; label.text = f'{len(df_close)} nearest stars '\n",
    "source_close_labels = ColumnDataSource(df_close_labels)\n",
    "\n",
    "source_bright = ColumnDataSource(df_bright)#; label.text = f'{len(df_bright)} brightest stars '\n",
    "source_bright_labels = ColumnDataSource(df_bright_labels)\n",
    "\n",
    "source_both   = ColumnDataSource(df_both)  #; label.text = f'{len(df_both)} near and bright stars '\n",
    "source_both_labels = ColumnDataSource(df_both_labels)\n",
    "\n",
    "source_either = ColumnDataSource(df_either)#; label.text = f'{len(df_either)} either bright or near stars '\n",
    "source_either_labels = ColumnDataSource(df_either_labels)\n",
    "\n",
    "source_named  = ColumnDataSource(df_named) #; label.text = f'{len(df_named)} stars with common names '\n",
    "source_named_labels = ColumnDataSource(df_named_labels)\n",
    "\n",
    "source = source_close"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 40,
   "metadata": {},
   "outputs": [],
   "source": [
    "plot = figure(title=f'HR Diagram',\n",
    "              tools='pan,zoom_in,zoom_out,save,reset',\n",
    "              active_drag='pan', x_axis_type='log',\n",
    "              plot_width=650, plot_height=650)\n",
    "plot.yaxis.ticker = SingleIntervalTicker(interval=1)\n",
    "plot.xaxis.axis_label = 'T. eff. in K'\n",
    "plot.yaxis.axis_label = 'Mᵥ'\n",
    "plot.title.align = 'left'\n",
    "plot.title.offset = .6\n",
    "plot.title.text_font_size = '12pt'\n",
    "plot.title.text_font = \"times\"\n",
    "plot.title.text_font_style = \"italic\"\n",
    "\n",
    "wheel_zoom = WheelZoomTool(zoom_on_axis=True)\n",
    "plot.add_tools(wheel_zoom)\n",
    "plot.toolbar.active_scroll = wheel_zoom\n",
    "\n",
    "plot.background_fill_color = 'black'\n",
    "plot.grid.grid_line_color = None\n",
    "plot.toolbar.autohide = True\n",
    "plot.toolbar.logo=None\n",
    "plot.toolbar_location='above'\n",
    "\n",
    "plot.x_range = DataRange1d(only_visible=True, range_padding=0.1, range_padding_units='percent', flipped=True) # 'absolute'\n",
    "plot.y_range = DataRange1d(only_visible=True, range_padding=0.08, range_padding_units='percent', flipped=True)\n",
    "\n",
    "plot.add_layout(label, place='left')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 41,
   "metadata": {},
   "outputs": [],
   "source": [
    "# make a color scheme for color index (B-V) colorbar\n",
    "repeat_white = 10 # of colors, out of 256, reserved for white in the middle of the colorbar (makes white strip bigger)\n",
    "skew_left_pct = 35 # percent of skew toward the left of the colorbar (bottom if vertical) (moves white strip left)\n",
    "skew_left = int(256*(skew_left_pct/100)) # resulting # of colors to skew left out of 256\n",
    "\n",
    "blu_or = [Blues256[b] for b in np.linspace(0, 255, (256-repeat_white)//2-skew_left, dtype=int, endpoint=True)] + ['white']*repeat_white \\\n",
    "     + [Oranges256[b] for b in np.linspace(255, 0, (256-repeat_white)//2+skew_left, dtype=int, endpoint=True)]"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 42,
   "metadata": {},
   "outputs": [],
   "source": [
    "select_mapper = Select(title='Select color bar', width=160, value='Color index (B-V)',\n",
    "                       options=['Color index (B-V)', 'Effective temperature', 'Luminosity'])\n",
    "\n",
    "def update_mapper(attrname, old, new):\n",
    "    if select_mapper.value == 'Color index (B-V)':\n",
    "        colorbar_color.visible = True\n",
    "        colorbar_temperature.visible = False\n",
    "        colorbar_luminosity.visible = False\n",
    "        near_plot.glyph.fill_color   = dict(field='B-V', transform=bv_color_mapper)\n",
    "        bright_plot.glyph.fill_color = dict(field='B-V', transform=bv_color_mapper)\n",
    "        both_plot.glyph.fill_color   = dict(field='B-V', transform=bv_color_mapper)\n",
    "        either_plot.glyph.fill_color = dict(field='B-V', transform=bv_color_mapper)\n",
    "        named_plot.glyph.fill_color  = dict(field='B-V', transform=bv_color_mapper)\n",
    "    elif select_mapper.value == 'Effective temperature':\n",
    "        colorbar_color.visible = False\n",
    "        colorbar_temperature.visible = True\n",
    "        colorbar_luminosity.visible = False\n",
    "        near_plot.glyph.fill_color   = dict(field='T_K', transform=color_mapper_temperature)\n",
    "        bright_plot.glyph.fill_color = dict(field='T_K', transform=color_mapper_temperature)\n",
    "        both_plot.glyph.fill_color   = dict(field='T_K', transform=color_mapper_temperature)\n",
    "        either_plot.glyph.fill_color = dict(field='T_K', transform=color_mapper_temperature)\n",
    "        named_plot.glyph.fill_color  = dict(field='T_K', transform=color_mapper_temperature)\n",
    "    else: # Luminosity\n",
    "        colorbar_color.visible = False\n",
    "        colorbar_temperature.visible = False\n",
    "        colorbar_luminosity.visible = True\n",
    "        near_plot.glyph.fill_color   = dict(field='L_sun', transform=color_mapper_luminosity)\n",
    "        bright_plot.glyph.fill_color = dict(field='L_sun', transform=color_mapper_luminosity)\n",
    "        both_plot.glyph.fill_color   = dict(field='L_sun', transform=color_mapper_luminosity)\n",
    "        either_plot.glyph.fill_color = dict(field='L_sun', transform=color_mapper_luminosity)\n",
    "        named_plot.glyph.fill_color  = dict(field='L_sun', transform=color_mapper_luminosity)\n",
    "\n",
    "select_mapper.on_change('value', update_mapper)"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "To show isoradial lines, we combine equations (4), (5), and (7) to find absolute magnitude (`M_V`) on the y-axis in terms of color index (`B-V`) on the x-axis versus. For a set of lines where $\\frac{R}{R_{sun}} = 0.01, 0.1, 1, 10, 100, 1000$, we'll set equation (7) equal to some factor, *F*, for orders of radial magnitude. For convenience, we'll make F an exponent because we'll end up taking the log.<br><br>\n",
    "$$\\frac{R}{R_{sun}} = 10^F = \\frac{\\sqrt{L/L_{sun}}}{(T_K/T_{sun})^2}$$<br>\n",
    "$$10^F = \\frac{\\sqrt{L/L_{sun}}}{(T_K/T_{sun})^2}$$<br>\n",
    "Subbing in equation (5), $\\frac{L}{L_{sun}} = 10^{\\frac{M_{Vsun} - M_V}{2.5}}$, we get<br><br>\n",
    "$$10^F = \\frac{\\sqrt{10^{\\frac{M_{Vsun} - M_V}{2.5}}}}{\\left(T_K/T_{sun}\\right)^2}$$<br>\n",
    "$$10^F\\cdot\\big(T_K/T_{sun}\\big)^2 = \\sqrt{10^{\\frac{M_{Vsun} - M_V}{2.5}}}$$<br>\n",
    "Squaring both sides, we get $$10^{2F}\\cdot\\big(T_K/T_{sun}\\big)^4 = 10^{\\frac{M_{Vsun} - M_V}{2.5}}$$<br>\n",
    "Taking the $\\log_{10}$ of both sides, we get $$\\log{\\Big(10^{2F}\\cdot\\big(T_K/T_{sun}\\big)^4\\Big)} = \\log{10^{\\frac{M_{Vsun} - M_V}{2.5}}}$$<br>\n",
    "$$2F+4\\log{\\big(T_K/T_{sun}\\big)} = \\frac{M_{Vsun} - M_V}{2.5}$$<br>\n",
    "$$5F+10\\log{\\big(T_K/T_{sun}\\big)} = M_{Vsun} - M_V$$<br>\n",
    "$$M_V = M_{Vsun} - 5F-10\\log{\\big(T_K/T_{sun}\\big)},\\; F = \\big\\{n \\in \\mathbb{Z}\\; \\bigm\\lvert -3\\le n\\le4\\big\\}\\tag{8}$$<br>"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 43,
   "metadata": {},
   "outputs": [],
   "source": [
    "# make dotted lines showing equal radius\n",
    "F = [-3, -2, -1, 0, 1, 2, 3, 4]\n",
    "x = [x for x in np.arange(int(df['T_K'].max()), int(df['T_K'].min()), -1000)]\n",
    "xs = [x, x, x, x, x, x, x, x]\n",
    "ys = [[M_sun - 5*f - 10*np.log10(t/T_sun) for t in x] for f in F]\n",
    "\n",
    "rad_source = ColumnDataSource(data=dict(xs=xs, ys=ys))\n",
    "\n",
    "isoradial_lines = plot.multi_line(\n",
    "    xs='xs',\n",
    "    ys='ys',\n",
    "    line_alpha=0.9,\n",
    "    line_cap='butt',\n",
    "    line_color='gray',\n",
    "    line_dash='dashed',\n",
    "    line_dash_offset=0,\n",
    "    line_join='bevel',\n",
    "    line_width=1,\n",
    "    visible=True,\n",
    "    source=rad_source)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "# starting and ending slider values\n",
    "start_lum, end_lum  = np.log10(0.00001), np.log10(10000000)\n",
    "start_bv, end_bv = df['B-V'].min(), df_either['B-V'].max() # df_either['B-V'].min()\n",
    "start_t, end_t = 27889, 2328\n",
    "\n",
    "# ColorBar from B-V (color index)\n",
    "bv_color_mapper = LinearColorMapper(palette=blu_or, low=start_bv, high=end_bv)#low=df['B-V'].min(), high=df['B-V'].max()) #Turbo256\n",
    "\n",
    "# Categorical colors from list of unique eff. temperatures, rounded to nearest 1000\n",
    "color_mapper_temperature = LogColorMapper(palette=Turbo256, low=df_close['T_K'].min(),   high=df_close['T_K'].max())\n",
    "color_mapper_luminosity  = LogColorMapper(palette=Greys256, low=df_close['L_sun'].min(), high=df_close['L_sun'].max())"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "line_color='#7c7e71'\n",
    "\n",
    "def scatter_factory(source, visible, name):\n",
    "    return plot.scatter(x='T_K', y='M_V', size='markersizes', line_color=line_color,\n",
    "                        fill_color={'field': 'B-V', 'transform': bv_color_mapper}, # Color index from B-V\n",
    "                        line_width=0.25, line_alpha=0.75, fill_alpha=1,\n",
    "                        source=source, visible=visible, name=name)\n",
    "\n",
    "near_plot   = scatter_factory(source_close,  True,  'stars') # the name 'stars' lets the hovertips activate for stars only and ignore other glyphs on the plot, such as the slider indicator and solar radius labels\n",
    "bright_plot = scatter_factory(source_bright, False, 'stars')\n",
    "both_plot   = scatter_factory(source_both,   False, 'stars')\n",
    "either_plot = scatter_factory(source_either, False, 'stars')\n",
    "named_plot  = scatter_factory(source_named,  False, 'stars')\n",
    "\n",
    "# adding orange-red asterisk to indicate slider values after the star plots so it shows up on top\n",
    "# (under name labels is ok)\n",
    "xy_source = ColumnDataSource(data=dict(x=[T_sun], y=[M_sun])) # marker starts at the sun's x- and y-coordinates\n",
    "render_marker = plot.scatter(x='x', y='y', size=15, marker='asterisk', line_color='orangered',#'azure','black'\n",
    "                             source=xy_source)\n",
    "\n",
    "# declaring range renderers allows the plot to ignore solar radius lines and labels from affecting the automatic range limits\n",
    "plot.x_range.renderers = [near_plot, bright_plot, both_plot, either_plot, named_plot, render_marker]\n",
    "plot.y_range.renderers = [near_plot, bright_plot, both_plot, either_plot, named_plot, render_marker]"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "def label_locations(lum, lt, rt, bm): # note luminosity (lum) is a power of 10\n",
    "#     try for label on right within margins\n",
    "    mag = M_sun - 5*lum - 10*np.log10(rt/T_sun)\n",
    "    if mag>bm: # go with label on left instead\n",
    "        return (lt, M_sun - 5*lum - 10*np.log10(lt/T_sun))\n",
    "    else:\n",
    "        return (rt, mag)\n",
    "\n",
    "def lum_label_factory(dataframe, visible):\n",
    "    left_margin   = dataframe['T_K'].max()\n",
    "    right_margin  = dataframe['T_K'].min()\n",
    "    bottom_margin = dataframe['M_V'].max()\n",
    "\n",
    "    x = np.zeros(8)\n",
    "    y = np.zeros(8)\n",
    "    for i, lum in enumerate(F):\n",
    "        x[i], y[i] = label_locations(lum, left_margin, right_margin, bottom_margin)\n",
    "\n",
    "    label_source = ColumnDataSource(data=dict(x=x, y=y,\n",
    "                                              text=['0.001 R⊙', '0.01 R⊙', '0.1 R⊙', '1 R⊙',\n",
    "                                                    '10 R⊙', '100 R⊙', '1000 R⊙', '10,000 R⊙']))\n",
    "    return LabelSet(x='x', y='y', text='text', level='glyph', x_offset=20, y_offset=-15, render_mode='canvas',\n",
    "                    text_color='white', text_font_size='10px', text_align='right',\n",
    "                    visible=visible, source=label_source)\n",
    "\n",
    "close_lum_labels  = lum_label_factory(df_close,  True)\n",
    "bright_lum_labels = lum_label_factory(df_bright, False)\n",
    "both_lum_labels   = lum_label_factory(df_both,   False)\n",
    "either_lum_labels = lum_label_factory(df_either, False)\n",
    "named_lum_labels  = lum_label_factory(df_named,  False)\n",
    "\n",
    "# placing isoradial luminosity labels at 'annotation' level ensures stars won't cover them up\n",
    "close_lum_labels.level  = 'annotation' # 'image', 'underlay', 'glyph', 'guide', 'annotation' or 'overlay'\n",
    "bright_lum_labels.level = 'annotation'\n",
    "both_lum_labels.level   = 'annotation'\n",
    "either_lum_labels.level = 'annotation'\n",
    "named_lum_labels.level  = 'annotation'\n",
    "\n",
    "plot.add_layout(close_lum_labels)\n",
    "plot.add_layout(bright_lum_labels)\n",
    "plot.add_layout(both_lum_labels)\n",
    "plot.add_layout(either_lum_labels)\n",
    "plot.add_layout(named_lum_labels)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "text_color='lightcyan'\n",
    "\n",
    "def labelset_factory(source, visible):\n",
    "    return (LabelSet(x='T_K', y='M_V', text='Name', x_offset=2, y_offset=0,\n",
    "                  text_font_size='11pt', text_color=text_color, text_align='center',\n",
    "                  source=source, visible=visible),\n",
    "            LabelSet(x='T_K', y='M_V', text='Name', x_offset=3, y_offset=-1,\n",
    "                  text_font_size='11pt', text_color='black', text_align='center',\n",
    "                  source=source, visible=visible))\n",
    "\n",
    "close_labels, close_labels_shadow   = labelset_factory(source_close_labels,  True)\n",
    "bright_labels, bright_labels_shadow = labelset_factory(source_bright_labels, False)\n",
    "both_labels, both_labels_shadow     = labelset_factory(source_both_labels,   False)\n",
    "either_labels, either_labels_shadow = labelset_factory(source_either_labels, False)\n",
    "named_labels, named_labels_shadow   = labelset_factory(source_named_labels,  False)\n",
    "\n",
    "plot.add_layout(close_labels_shadow)\n",
    "plot.add_layout(close_labels)\n",
    "plot.add_layout(bright_labels_shadow)\n",
    "plot.add_layout(bright_labels)\n",
    "plot.add_layout(both_labels_shadow)\n",
    "plot.add_layout(both_labels)\n",
    "plot.add_layout(either_labels_shadow)\n",
    "plot.add_layout(either_labels)\n",
    "plot.add_layout(named_labels_shadow)\n",
    "plot.add_layout(named_labels)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "tooltips='''\n",
    "        <HTML>\n",
    "        <HEAD>\n",
    "        <style>\n",
    "        .bk-tooltip {\n",
    "            background-color: black !important;\n",
    "            }\n",
    "        </style>\n",
    "        </HEAD>\n",
    "        <BODY>\n",
    "        <div>\n",
    "            <div>\n",
    "                <span style=\"color:blue; font-size: 12px; \"><b>@Name</b></span><br>\n",
    "                <span style=\"color:blue; font-size: 10px; \">Abs. mag.: @M_V{0.0}</span><br>\n",
    "                <span style=\"color:blue; font-size: 10px; \">Radius: @{R_sun}{0.0} R⊙</span><br>\n",
    "                <span style=\"color:blue; font-size: 10px; \">Luminosity: @{L_sun}{0.0} L⊙</span><br>\n",
    "                <span style=\"color:blue; font-size: 10px; \">Distance: @{LY}{0.0} LY</span><br>\n",
    "                <span style=\"color:blue; font-size: 10px; \">Spectral type: @SpType</span><br>\n",
    "                <span style=\"color:blue; font-size: 10px; \">Color index: @{B-V}</span><br>\n",
    "                <span style=\"color:blue; font-size: 10px; \">Eff. temp.: @{T_K}{0.0} K</span><br>\n",
    "            </div>\n",
    "        </div>\n",
    "        </BODY>\n",
    "        </HTML>'''\n",
    "\n",
    "plot.add_tools(HoverTool(tooltips=tooltips,\n",
    "                         show_arrow=False,\n",
    "                         point_policy='snap_to_data',#'follow_mouse',\n",
    "                         names=['stars'],\n",
    "                         toggleable=True\n",
    "                        ))"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "# making human-readable ticks for the log-scaled luminosity colorbar\n",
    "logformatter = FuncTickFormatter(code=\"\"\"\n",
    "if (tick >= 0.1 && tick <=10) {\n",
    "  return tick;\n",
    "} else {\n",
    "return 10 + (Math.log10(tick).toString()\n",
    "             .split('')\n",
    "             .map(function (d) { return d === '-' ? '⁻' : '⁰¹²³⁴⁵⁶⁷⁸⁹'[+d]; })\n",
    "             .join(''));\n",
    "}\n",
    "\"\"\")\n",
    "\n",
    "colorbar_color       = ColorBar(color_mapper=bv_color_mapper, location=(0,0), label_standoff=8,\n",
    "                                title = 'B-V', padding=10, width = 20, visible=True)\n",
    "colorbar_temperature = ColorBar(color_mapper=color_mapper_temperature, location=(0,0), label_standoff=8,\n",
    "                                title = 'T in K', padding=5, width = 20,\n",
    "                                ticker=LogTicker(desired_num_ticks=10, max_interval=10000, min_interval=10),\n",
    "                                visible=False)\n",
    "colorbar_luminosity  = ColorBar(color_mapper=color_mapper_luminosity, location=(0,0), label_standoff=8,\n",
    "                                title = 'L⊙', padding=10, width = 20, formatter=logformatter,\n",
    "                                ticker=LogTicker(desired_num_ticks=10, max_interval=100, min_interval=10),\n",
    "                                visible=False)\n",
    "\n",
    "plot.add_layout(colorbar_color,       'right')\n",
    "plot.add_layout(colorbar_temperature, 'right')\n",
    "plot.add_layout(colorbar_luminosity,  'right')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "# html Div widgets\n",
    "div_title = Div(text='''<b>Astro Lab Using HR Diagrams, <i>v. 1.1</i></b>''', width=400, height=20)\n",
    "\n",
    "div_spacer = Div(text='', height=50)\n",
    "\n",
    "div_hints = Div(text='''Things you can try: you can zoom in/out and pan on both plots with the mouse or gestures\n",
    "on a mobile device. Scroll with the mouse wheel while hovering over an axis to zoom vert. or horiz. only (star\n",
    "plot only). If the stars' tooltip pop-ups are distracting, they can be turned off with the <b>Hover</b> tool in\n",
    "the toolbar above the plot. There are buttons to <b>Reset</b> and <b>Save</b> the plot there as well. Below the\n",
    "star plot are options for which sets of stars to plot and which color scheme to use.''',\n",
    "          width=400, height=140)\n",
    "\n",
    "div_links = Div(text='''In appreciation of UNL's <a href=\"https://astro.unl.edu\" rel=\"noopener noreferrer\"\n",
    "target=\"_blank\">Astronomy page</a> and great but soon-to-expire Flash-based\n",
    "<a href=\"https://astro.unl.edu/naap/hr/animations/hr.html\" rel=\"noopener noreferrer\"target=\"_blank\"\n",
    ">HR Diagram explorer</a>.<br>Check out my <a href=\"https://skyfield-orbits.herokuapp.com/skyfield-orbits\" \n",
    "rel=\"noopener noreferrer\"target=\"_blank\">solar system animation</a>, now with retrograde! See my\n",
    "<a href=\"https://github.com/swshadle/physics\"rel=\"noopener noreferrer\"target=\"_blank\">GitHub</a>\n",
    "for contact info and more projects.''', width=400, height=50)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "p2 = figure(plot_width=420, plot_height=400, active_drag='pan', background_fill_color='black',\n",
    "#             x_range=(-2.35, 2.1),\n",
    "            y_range=(-2.1, 2.1), tools='pan,zoom_in,zoom_out,reset')\n",
    "\n",
    "p2.x_range = DataRange1d(range_padding=1, range_padding_units='percent', flipped=False) # 'absolute'\n",
    "p2.add_tools(wheel_zoom)\n",
    "p2.toolbar.active_scroll = wheel_zoom\n",
    "p2.grid.grid_line_color = None\n",
    "p2.toolbar_location='above'\n",
    "p2.axis.visible=False\n",
    "p2.toolbar.autohide = True\n",
    "p2.toolbar.logo=None"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "# create a data source to enable refreshing of fill & text color\n",
    "bv_source = ColumnDataSource(data=dict(color=[BV_sun]))\n",
    "bv_source_sun = ColumnDataSource(data=dict(color=[BV_sun]))\n",
    "bv_color_mapper = LinearColorMapper(palette=blu_or, low=start_bv, high=end_bv)\n",
    "star_rad = 1\n",
    "render_circle = p2.circle(-0.25-star_rad, 0, radius=star_rad, alpha=1, line_color='black',\n",
    "                          fill_color={'field': 'color', 'transform': bv_color_mapper}, source=bv_source)\n",
    "p2.circle(1, 0, radius=1, alpha=1,  line_color='black',\n",
    "          fill_color={'field': 'color', 'transform': bv_color_mapper}, source=bv_source_sun)\n",
    "\n",
    "sun_label  = Label(x=p2.plot_width*0.90, y=10, x_units='screen', y_units='screen', text_align='right',\n",
    "                   background_fill_color='black', text_color='white', background_fill_alpha=.5, text=' sun ')\n",
    "star_label = Label(x=p2.plot_width*0.03, y=10, x_units='screen', y_units='screen', text_align='left',\n",
    "                   background_fill_color='black', text_color='white', background_fill_alpha=.5, text=' star ')\n",
    "\n",
    "p2.add_layout(sun_label)\n",
    "p2.add_layout(star_label)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "# under the hood, the temperature slider has color index (B-V) values\n",
    "bv_slider = Slider(title='Temperature', start=start_bv, end=end_bv, step=0.001, width=400,\n",
    "                   value=np.around(BV_sun, 3), # round starting value to 3 decimal places\n",
    "                   format=FuncTickFormatter(code=\"\"\"return (5601/Math.pow(tick+0.4,2/3)).toFixed(0) + ' K'\"\"\"))\n",
    "\n",
    "lum_slider = Slider(title='Luminosity', start=start_lum, end=end_lum, step=.1, width=400,\n",
    "                    value=0, format=FuncTickFormatter(code=\"\"\"return Math.pow(10,tick).toPrecision(4) + ' L⊙'\"\"\"))"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "We want to show slider values with a marker on the HR plot. The sliders' temperature and luminosity can be converted to the HR plot's color index (x-axis) and abs. magnitude (y-axis) using equation (3) for `M_V` and equation (5) for `T_K`.<br><br>\n",
    "$$M_V = M_V☉ - 2.5 \\cdot log\\frac{L}{L_{sun}}\\tag{3}$$<br>\n",
    "$$T_K = \\frac{5601 K}{(B-V+0.4)^{2/3}}\\tag{5}$$"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "def t_from_color(c):\n",
    "    return 5601/np.power(c+0.4,2/3)\n",
    "\n",
    "temp = t_from_color(bv_slider.value)\n",
    "\n",
    "from latex_label import LatexLabel\n",
    "latex = LatexLabel(\n",
    "    text = f'\\mathrm{{radius}} = \\\n",
    "    {{\\sqrt{{L/L_\\odot}} \\over (T/T_\\odot)^2}} = \\\n",
    "    {{\\sqrt{{1}} \\over ({{{T_sun}}}/{T_sun})^2}} = \\\n",
    "    {1} R_\\odot',\n",
    "    x=0,\n",
    "    y=-10,\n",
    "    x_units='screen',\n",
    "    y_units='screen',\n",
    "    render_mode='css', # Note: must set ``render_mode=\"css\"``\n",
    "    text_font_size='14px',\n",
    "    text_color='black',\n",
    "    background_fill_alpha=0,\n",
    ")\n",
    "p2.add_layout(latex)\n",
    "\n",
    "# debug_label = Label(x=10, y=p2.plot_height/2, x_units='screen', y_units='screen', text_align='left',#'data',\n",
    "#                     text_color='white', background_fill_color='black', text_font_size='16px',\n",
    "#                     background_fill_alpha=0.8, text='Debug')\n",
    "# p2.add_layout(debug_label)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "slider_callback_js = CustomJS(args=dict(bv_source=bv_source, bv_slider=bv_slider,\n",
    "                                        lum_slider=lum_slider, latex=latex,\n",
    "                                        render_circle=render_circle,\n",
    "#                                         label=debug_label, \n",
    "                                        render_marker=render_marker),\n",
    "                              code=\"\"\"\n",
    "    var lum = Math.pow(10,lum_slider.value).toPrecision(4); // + ' L⊙';\n",
    "    var temp = ~~5601/Math.pow(bv_slider.value+0.4,2/3);   // ).toFixed(0) + ' K' // or Math.round() if double bit-inversion (~~) doesn't work\n",
    "    var rad = Math.sqrt(lum)/(temp*temp/33408400); // 5780**2 = 33408400\n",
    "\n",
    "    const color = bv_source.data['color'];\n",
    "    color[0] = bv_slider.value;\n",
    "    bv_source.change.emit();\n",
    "    // label.text = 'BV value: ' + bv_slider.value.toPrecision(3) + ' lum value: ' + lum_slider.value.toPrecision(3);\n",
    "    render_circle.glyph.radius = rad;\n",
    "    render_circle.glyph.x = -0.25 - rad;\n",
    "    render_marker.glyph.x = 5601/Math.pow(bv_slider.value+0.4,2/3);\n",
    "    render_marker.glyph.y = parseFloat(4.83 - 2.5*lum_slider.value);\n",
    "\"\"\")\n",
    "\n",
    "def solar_radii_str(lu, te):\n",
    "    return np.format_float_positional(np.sqrt(lu)/(te/T_sun)**2, precision=3, unique=False, trim='-',\n",
    "                                      fractional=False, sign=False, pad_left=None, pad_right=None)\n",
    "\n",
    "def slider_callback(attr, old, new):\n",
    "    lum_str = np.format_float_positional(10.0**lum_slider.value, precision=3, unique=False,\n",
    "                                         trim='-', fractional=False, sign=False, pad_left=None,\n",
    "                                         pad_right=None)\n",
    "    temp = t_from_color(bv_slider.value)\n",
    "    rad_str = solar_radii_str(float(lum_str), temp)\n",
    "    latex.text = f'\\mathrm{{radius}} = {{\\sqrt{{L/L_\\odot}} \\over (T/T_\\odot)^2}} = \\\n",
    "        {{\\sqrt{{{lum_str}}} \\over ({{{int(temp)}}}/{T_sun})^2}} = {rad_str}\\: R_\\odot'\n",
    "\n",
    "bv_slider.js_on_change('value', slider_callback_js)\n",
    "bv_slider.on_change('value', slider_callback)\n",
    "lum_slider.js_on_change('value', slider_callback_js)\n",
    "lum_slider.on_change('value', slider_callback)\n",
    "\n",
    "button = Button(label='Reset to solar temperature and luminosity', button_type=\"success\")\n",
    "\n",
    "def button_callback():\n",
    "    latex.text = f'\\mathrm{{radius}} = \\\n",
    "    {{\\sqrt{{L/L_\\odot}} \\over (T/T_\\odot)^2}} = \\\n",
    "    {{\\sqrt{{1}} \\over ({{{T_sun}}}/{T_sun})^2}} = 1\\: R_\\odot'\n",
    "    \n",
    "button.on_click(button_callback)\n",
    "\n",
    "button.js_on_click(CustomJS(args=dict(bv_source=bv_source, bv_slider=bv_slider, latex=latex,\n",
    "                                      lum_slider=lum_slider, BV_sun=BV_sun,\n",
    "                                      render_circle=render_circle,\n",
    "#                                       label=debug_label,\n",
    "                                      render_marker=render_marker,\n",
    "                                      M_sun=M_sun),\n",
    "                            code=\"\"\"\n",
    "    // label.text = 'Attempting to set bv_slider to ' + BV_sun.toFixed(3);\n",
    "    render_circle.glyph.radius = 1\n",
    "    render_circle.glyph.x = -0.25 - 1\n",
    "    bv_slider.value = parseFloat(BV_sun.toFixed(3));\n",
    "    lum_slider.value = 0.0;\n",
    "    // label.text = 'BV value: ' + bv_slider.value.toPrecision(3) + ' lum value: ' + lum_slider.value.toPrecision(3);\n",
    "\n",
    "    const color = bv_source.data['color'];\n",
    "    color[0] = bv_slider.value;\n",
    "    bv_source.change.emit();\n",
    "    render_marker.glyph.x = T_sun;\n",
    "    render_marker.glyph.y = M_sun;\n",
    "\"\"\"))"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "curdoc().clear()\n",
    "curdoc().title = f'Hertzsprung-Russell'\n",
    "layout = row(\n",
    "    column(plot, Paragraph(text='Show stars', width=70, height=10), row(radiogroup, column(select_mapper, checkboxgroup))),\n",
    "    column(div_title, p2, div_spacer, bv_slider, lum_slider, button, div_hints, div_links))\n",
    "curdoc().add_root(layout)"
   ]
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3",
   "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.7.4"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 4
}