Black Hole Class.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 1,
   "metadata": {},
   "outputs": [],
   "source": [
    "import rebound\n",
    "import numpy as np\n",
    "import math\n",
    "import matplotlib.pyplot as plt\n",
    "import csv\n",
    "from matplotlib import animation\n",
    "from matplotlib import colors as mcolors\n",
    "from IPython.display import HTML, Image\n",
    "\n",
    "class BlackHoleSimulation:\n",
    "    g = 4*np.pi**2 \n",
    "    units = ['Au', 'yr', 'Msun']\n",
    "    \n",
    "    #initialize Black Hole Class\n",
    "    def __init__(self, starA, starB, semi = 1, blackhole = [100,0,100], phase = 0, binary_init_v = 10):\n",
    "        self.sim = rebound.Simulation()\n",
    "        self.sim.G = self.g \n",
    "        self.semi = semi\n",
    "        self.phase = phase\n",
    "        self.initial_v = binary_init_v\n",
    "        self.sim.units = self.units\n",
    "        self.sim.add(m=starA, a = semi)\n",
    "        self.sim.add(m=starB)\n",
    "        self.sim.add(m=blackhole[0], x= -1*blackhole[1],y= -1*blackhole[2])\n",
    "        self.sim.particles[1].y += math.tan(phase*math.pi/180)*semi\n",
    "        print(\"Star A:\",self.sim.particles[0].x, self.sim.particles[0].y, \"Star B:\", self.sim.particles[1].x, self.sim.particles[1].y)\n",
    "        self.sim.particles[0].vx=-0.210945021 * self.initial_v\n",
    "        self.sim.particles[1].vx=-0.210945021 * self.initial_v\n",
    "        self.sim.move_to_com()\n",
    "        self.starA = self.sim.particles[0]\n",
    "        self.starB = self.sim.particles[1]\n",
    "        self.blackhole = self.sim.particles[2]\n",
    "        #stores data on stars\n",
    "        self.starA_x ,self.starA_y, self.starA_z = [],[],[]\n",
    "        self.starB_x, self.starB_y, self.starB_z = [],[],[]\n",
    "        self.starA_v, self.starA_a = [],[]\n",
    "        self.starB_v, self.starB_a = [],[]\n",
    "        \n",
    "    #converts Au to Km/s \n",
    "    def unit_converter(self,unconverted, unit): \n",
    "        if unit == \"AuToKms\":\n",
    "            converted = (unconverted * 4.74)\n",
    "        elif unit == \"AuToKms^2\":\n",
    "            converted = (unconverted * 1.50222861e-7)\n",
    "        return converted\n",
    "    \n",
    "    #distance calculator\n",
    "    def calculate_distance(self,star, blackhole):\n",
    "        distance = np.sqrt(np.square(star.x - blackhole.x) + np.square(star.y - blackhole.y))\n",
    "        return distance\n",
    "    \n",
    "    #integrate the simulation\n",
    "    def run_simulation(self, endtime, outputs):\n",
    "        times = np.linspace(0,endtime,outputs) #(start,finish, number of data points)\n",
    "        self.timeList = times.tolist() \n",
    "        self.outputs = outputs\n",
    "        self.endtime = endtime\n",
    "        for i,times in enumerate(times): #integrate simulation \n",
    "            self.starA_x.append(self.starA.x)\n",
    "            self.starA_y.append(self.starA.y) \n",
    "            self.starA_z.append(self.starA.z) \n",
    "            self.starB_x.append(self.starB.x)\n",
    "            self.starB_y.append(self.starB.y)\n",
    "            self.starB_z.append(self.starA.z)\n",
    "            self.starA_v.append(self.unit_converter(((self.starA.vx**2) + (self.starA.vy**2) + (self.starA.vz**2))**(.5),\"AuToKms\"))\n",
    "            self.starB_v.append(self.unit_converter(((self.starB.vx**2) + (self.starB.vy**2) + (self.starB.vz**2))**(.5),\"AuToKms\"))\n",
    "            self.starA_a.append(self.unit_converter(((self.starA.ax**2) + (self.starA.ay**2) + (self.starA.az**2))**(.5),\"AuToKms^2\"))\n",
    "            self.starB_a.append(self.unit_converter(((self.starB.ax**2) + (self.starB.ay**2) + (self.starB.az**2))**(.5),\"AuToKms^2\"))\n",
    "            self.sim.integrate(times)\n",
    "\n",
    "    #returns maximum velocities of stars\n",
    "    def get_velocity(self,command):\n",
    "        if command == \"i\":\n",
    "            return \"Star A initial velocity: \", self.starA_v[0], \"Star B initial velocity: \", self.starB_v[0]\n",
    "        elif command == \"f\":\n",
    "            return \"Star A final velocity: \", self.starA_v[self.outputs-1], \"Star B final velocity: \", self.starB_v[self.outputs-1]\n",
    "        elif command == \"max\":\n",
    "            return \"Star A max velocity: \", max(self.starA_v), \"Star B max velocity: \", max(self.starB_v)\n",
    "        else:\n",
    "            return \"Please enter valid command(i,f,max)\"\n",
    "    \n",
    "    #plots a x vs. y position graph for both stars\n",
    "    def get_position_graph(self):\n",
    "        #star A \n",
    "        plt.subplot(2,2,1)\n",
    "        plt.plot(self.starA_x,self.starA_y)\n",
    "        plt.plot(self.blackhole.x,self.blackhole.y,'b.')\n",
    "        plt.title(\"Position of Star A\")\n",
    "        plt.xlabel(\"X-position(Au)\")\n",
    "        plt.ylabel(\"Y-position(Au)\")\n",
    "\n",
    "        #star B\n",
    "        plt.subplot(2,2,2)\n",
    "        plt.plot(self.starB_x,self.starB_y)\n",
    "        plt.plot(self.blackhole.x,self.blackhole.y,'b.')\n",
    "        plt.title(\"Position of Star B\")\n",
    "        plt.xlabel(\"X-position(Au)\")\n",
    "        plt.ylabel(\"Y-position(Au)\")\n",
    "        plt.tight_layout()\n",
    "        plt.show()\n",
    "    \n",
    "    #creates scatter plot. Takes the min and max of each axis.\n",
    "    def create_scatter(self, xmin, xmax, ymin, ymax):\n",
    "        fig = plt.figure()\n",
    "        ax = plt.axes(xlim=(xmin,xmax), ylim=(ymin, ymax))#set axis\n",
    "        plt.plot(self.starA_x,self.starA_y)#plot star a\n",
    "        plt.plot(self.starB_x,self.starB_y)#plot star b\n",
    "        plt.plot(self.blackhole.x,self.blackhole.y,'b.')#plot black hole\n",
    "        plt.xlabel(\"X-pos(Au)\")\n",
    "        plt.ylabel(\"Y-pos(Au)\")\n",
    "        plt.show()\n",
    "        \n",
    "        fig.savefig(\"BH\" + str(self.sim.particles[2].m) + \"_Bi\" + str(self.sim.particles[0].m)+\"_a\"+ str(self.semi) +\"_p\" + str(self.phase) +\"_v\" + str(self.initial_v) + \".png\")\n",
    "\n",
    "    #creates csv file for data points collected\n",
    "    def create_CSV(self):\n",
    "        with open(\"BH\" + str(self.sim.particles[2].m) + \"_Bi\" + str(self.sim.particles[0].m)+\"_a\"+ str(self.semi) +\"_p\" + str(self.phase) + \"_v\" + str(self.initial_v) +\".csv\", 'w', newline='') as csvfile:\n",
    "            fieldnames = ['Star A X-pos(AU)', 'Star A Y-pos(AU)','Star B X-pos(AU)', 'Star B Y-pos(AU)', 'Star A Velocity(Km/s)','Star B Velocity(Km/s)', 'Star A Acceleration(Km/s^2)', 'Star B Acceleration(Km/s^2)', 'time(yr)']\n",
    "            writer = csv.DictWriter(csvfile, fieldnames=fieldnames)\n",
    "            writer.writeheader()\n",
    "            for i in range(0,self.outputs):\n",
    "                 writer.writerow({'Star A X-pos(AU)': str(self.starA_x[i]), 'Star A Y-pos(AU)': str(self.starA_y[i]), 'Star B X-pos(AU)': str(self.starB_x[i]), 'Star B Y-pos(AU)': str(self.starB_y[i]), 'Star A Velocity(Km/s)': str(self.starA_v[i]), 'Star B Velocity(Km/s)': str(self.starB_v[i]),'Star A Acceleration(Km/s^2)': str(self.starA_a[i]), 'Star B Acceleration(Km/s^2)': str(self.starB_a[i]), 'time(yr)': str(self.timeList[i])})    \n",
    "    \n",
    "    #creates animation. Specify start and end of animation\n",
    "    def animate_simulation(self, start, end, axis, title=\"Animation\", fps=60, option=0):\n",
    "        fig = plt.figure()\n",
    "        ax = plt.axes(xlim=(axis[0], axis[1]), ylim=(axis[2], axis[3]))\n",
    "        graph = plt.scatter([], [], s=2)\n",
    "        graph2 = plt.scatter([], [], s=2)\n",
    "        graph.set_facecolors('b')\n",
    "        graph2.set_facecolors('r')\n",
    "        time_text = ax.text(0.1, 0.9, '', transform=ax.transAxes)\n",
    "        starA_v_text = ax.text(0.1,0.85,'',transform=ax.transAxes)\n",
    "        starB_v_text = ax.text(0.1,0.80,'',transform=ax.transAxes)\n",
    "        plt.plot(self.blackhole.x,self.blackhole.y,'b.')\n",
    "        plt.title(\"Black Hole and Binary Star System\", bbox={'facecolor':'blue', 'alpha':0.3, 'pad':10})\n",
    "        plt.xlabel(\"X-position(Au)\")\n",
    "        plt.ylabel(\"Y-position(Au)\")\n",
    "        totalFrames = int(self.outputs*((end-start)/self.endtime))\n",
    "        startFrame = int(self.outputs/self.endtime*start)\n",
    "        print(\"total frames: \", totalFrames, \"start frame: \",startFrame)\n",
    "        \n",
    "        # animation function.  This is called sequentially\n",
    "        def animate(i):\n",
    "            #plt.plot(self.starA_x[startFrame:i+startFrame],self.starA_y[startFrame:i+startFrame])\n",
    "            #plt.plot(self.starB_x[startFrame:i+startFrame],self.starB_y[startFrame:i+startFrame])\n",
    "            time_text.set_text('Time(yr): ' + str(int(self.timeList[i+startFrame])))\n",
    "            starA_v_text.set_text('Star A Velocity: ' + str(int(self.starA_v[i+startFrame])) + ' km/s')\n",
    "            starB_v_text.set_text('Star B Velocity: ' + str(int(self.starB_v[i+startFrame])) + ' km/s')\n",
    "            if option == 0:\n",
    "                graph.set_offsets(np.vstack((self.starA_x[startFrame:i+startFrame], self.starA_y[startFrame:i+startFrame])).T)\n",
    "                graph2.set_offsets(np.vstack((self.starB_x[startFrame:i+startFrame], self.starB_y[startFrame:i+startFrame])).T)\n",
    "            elif option == 1:\n",
    "                plt.plot(self.starA_x[0:i*10],self.starA_y[0:i*10])\n",
    "                plt.plot(self.starB_x[0:i*10],self.starB_y[0:i*10])\n",
    "            return graph,\n",
    "        \n",
    "        # call the animator\n",
    "        anim = animation.FuncAnimation(fig, animate, frames=totalFrames, interval=20, blit=True)\n",
    "        anim.save(title+'.mp4', fps=fps, extra_args=['-vcodec', 'libx264'])\n",
    "        HTML(anim.to_html5_video())\n",
    "        "
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 50,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Star A: 0.01 0.0 Star B: 0.0 0.005773502691896256\n"
     ]
    }
   ],
   "source": [
    "sim = BlackHoleSimulation(.1,.1,semi = .01, blackhole = [100,0,100], phase = 30, binary_init_v=15) #(Star A mass, Star B mass, Semi, black hole[mass,x,y])"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 51,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "CPU times: user 30min 16s, sys: 16 s, total: 30min 32s\n",
      "Wall time: 31min 38s\n"
     ]
    }
   ],
   "source": [
    "%time sim.run_simulation(50,25000)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 52,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "image/png": "iVBORw0KGgoAAAANSUhEUgAAAYgAAAEKCAYAAAAIO8L1AAAABHNCSVQICAgIfAhkiAAAAAlwSFlz\nAAALEgAACxIB0t1+/AAAADl0RVh0U29mdHdhcmUAbWF0cGxvdGxpYiB2ZXJzaW9uIDIuMS4wLCBo\ndHRwOi8vbWF0cGxvdGxpYi5vcmcvpW3flQAAIABJREFUeJzt3Xl0VGWe//H3Nwkk7FvYt4QtISwi\noqCyKa6gKCICamvbPe3P6Z759XJmepme7czM70z39Mz0tNMz9jDd7datgAtKu4GiyA6yyJoEAgIJ\n+76HLPX9/VEXTeMlBFKVyvJ5nZOTuk89detb1zIf7vI819wdERGRiyUlugAREamdFBAiIhJKASEi\nIqEUECIiEkoBISIioRQQIiISKq4BYWY7zWyjmX1qZquDtrZm9r6ZbQt+twnazcyeNrMCM9tgZkPj\nWZuIiFSuJvYgbnH3Ie4+LFj+IbDA3fsCC4JlgLuBvsHPk8AzNVCbiIhcQiIOMd0HPB88fh64v0L7\nCx61AmhtZp0TUJ+IiAApcV6/A/PNzIH/cfcZQEd33wfg7vvMrEPQtytQWOG1RUHbvoorNLMnie5h\n0KxZs+uys7Pj/BFEROqXNWvWHHb39pfrF++AuNnd9wYh8L6Z5VXS10LavjQPSBAyMwCGDRvmq1ev\njk2lIiINhJntqkq/uB5icve9we+DwBzgBuDAhUNHwe+DQfcioHuFl3cD9sazPhERubS4BYSZNTOz\nFhceA3cAm4C5wONBt8eBN4PHc4HHgquZRgAnLhyKEhGRmhfPQ0wdgTlmduF9XnL398zsE2C2mX0d\n2A1MCfq/A4wHCoCzwBNxrE1ERC4jbgHh7juAa0LajwDjQtod+Fa86hERkSujkdQiIhJKASEiIqEU\nECIiEkoBISIioRQQIiISSgEhIiKhFBAiIhJKASEiIqEUECIiEkoBISIioRQQIiISSgEhIiKhFBAi\nIhJKASEiIqEUECIiEkoBISIioRQQIiISSgEhIiKhFBAiIhJKASEiIqEUECIiEkoBISIioRQQIiIS\nSgEhIiKhFBAiIhJKASEiIqEUECIiEkoBISIioRQQIiISSgEhIiKhFBAiIhJKASEiIqEUECIiEkoB\nISIioRQQIiISKu4BYWbJZrbOzN4KljPNbKWZbTOzWWbWOGhPDZYLgucz4l2biIhcWk3sQXwbyK2w\n/FPg5+7eFzgGfD1o/zpwzN37AD8P+omISILENSDMrBswAfh1sGzArcCrQZfngfuDx/cFywTPjwv6\ni4hIAsR7D+I/gO8DkWC5HXDc3cuC5SKga/C4K1AIEDx/Iuj/R8zsSTNbbWarDx06FM/aRUQatLgF\nhJndAxx09zUVm0O6ehWe+6LBfYa7D3P3Ye3bt49BpSIiEiYljuu+GZhoZuOBNKAl0T2K1maWEuwl\ndAP2Bv2LgO5AkZmlAK2Ao3GsT0REKhG3PQh3/5G7d3P3DGAa8KG7PwJ8BDwYdHsceDN4PDdYJnj+\nQ3f/0h6EiIjUjESMg/gB8D0zKyB6juE3QftvgHZB+/eAHyagNhERCcTzENPn3H0hsDB4vAO4IaRP\nMTClJuoREZHL00hqEREJpYAQEZFQCggREQmlgBARkVAKCBERCaWAEBGRUAoIEREJpYAQEZFQCggR\nEQmlgBARkVB1OiDOHNrFlpXzKS8ru3xnERG5IjUyF1O8NC09Rs67Uzj6bkv2tBqKDX6IfiMn0zg1\nLdGliYjUeVaXZ9Qeeu0Qn/FP36LtxmfpVbz58/bVLW4jacBEskfeT9PmrRJYoYhI7WNma9x92GX7\n1eWAGDZsmK9evRqA4nNn2LZoNudy59P3+GLacIpz3pjcZtdT1m8CWaOn0KpthwRXLCKSeA0uICoq\nKy0hf+U8Tq9/g8xDH9IhuDHdweRO7M2cTNdb/w/tu/Ss6XJFRGqFBh0QFXmknG2fLub8shkMOvz2\n5+15Kf051vNOut/0EN16D4h3qSIitYYCIoRHIuzKX8v+Fa/Qvmg+vct3APBZUgb7u95Ox+FTyMy5\nHkuq0xd3iYhUSgFRBXt35rN76Sxa7XyPrJItJJlTZJ0o6jiO1sMm0+/asSQlJ8ewYhGRxFNAXKHD\n+3ezY8ls0greJfvcOhpbOQdpy870MTS7ZhJZI+4mpVHjmLyXiEgiKSCq4cTxI2xb9ApJ+W+RfXoV\nTe08x2nOttajaDTwPrJvnkhak2Yxf18RkZqggIiRc2dOkbvkDSKb36TfyaW05CxnPZW8FiPw/veS\nNWoyzVu2jWsNIiKxpICIg5LzxeQvf5tzG96g19FFpHOcEk8ht+lQSvpMoM/oqbRp37nG6hERuRoK\niDgrLytj65oFnFj7Oj0OfkgXP0i5G3mpgyjpMYout/wJHbv2SkhtIiKVUUDUII9E2L5xGYc/eZUR\nRc9+3p6fksXRHnfS9caH6NF3UAIrFBH5ggIigXblrWPfildoVzSPvmUFAOyoMNail8ZaiEgCKSBq\nif27t7Jr8Uxa7JpH9vnNwViLzhR2Gkeb6ybT79oxGmshIjVKAVELHTlQyPbFs2hS8A7Z5z6lUTDW\nYkf6LTQfMons4XdqrIWIxJ0CopY7cewQBYtfJTn/D2SfXkWalXKMFmxtPZrUQfeRfdO9pDVpmugy\nRaQeUkDUIedOnyRv8WtEcv9A1ollNLdznPYm5LW8EcuZSPbIB2jWQve1EJHYUEDUUSXF58hb/hbn\nN8yhz7HFtOEkxd6ILc1uoCzrXrJGPUirtu0TXaaI1GEKiHqgvKyU/FXzObXu9c/va1HqyeQ2uZaz\nfcbTZ9RU0jt2S3SZIlLHKCDqmUh5OQWfLuLo6lfpvv99uvqB6MC8xgM5mXk3maOm0ql7n0SXKSJ1\ngAKiHvNIhJ1bVnFg5St02jOfjMhuAPJT+gUD86ZqYJ6IXJICogEp2raewmWzSS+cR9+ybcCFgXl3\n0HH4gxqYJyJ/JOEBYWZpwCIgFUgBXnX3vzOzTGAm0BZYC3zF3UvMLBV4AbgOOAJMdfedlb2HAuLL\nDuzeys4ls2ix873PB+YVWheKOo2j7bDowDyFhUjDVhsCwoBm7n7azBoBS4BvA98DXnf3mWb2K2C9\nuz9jZt8EBrv7U2Y2DZjk7lMrew8FROUO79/NjsWzaVLwNtnF62lk5RyhNQdbDcZHfJOs628nOSUl\n0WWKSA1LeEBcVExTogHxp8DbQCd3LzOzG4G/d/c7zWxe8Hi5maUA+4H2XkmBCoiqO3H0EFsXzabL\n5hl0LtlFkjmHaU1B27E0GTKJnBF306hxaqLLFJEaELOACPYEBgJdgHPAZnc/UsUikoE1QB/gv4Cf\nASvcvU/wfHfgXXcfaGabgLvcvSh4bjsw3N0PX7TOJ4EnAXr06HHdrl27qlKKVHD65DHyl7yGbZlL\n9qkVn98xL7/1KBoPvJ/+N0/UKG6ReqzaAWFmGcD3gbuAz4BDQBrQFzgO/Ar4XWX/wq+wrtbAHOBv\ngWcvCoh33H2QmW0G7rwoIG6oLIy0B1F9xWdPk7f0Dco2vUm/E0toyVlOexNyW96I5dxH/5GTNIpb\npJ6pakBUdgD6X4BngD9z98hFK+8MPAI8Djx3uTdx9+NmthAYAbQ2sxR3LwO6AXuDbkVAd6AoOMTU\nCjh6uXVL9aQ1bc6Q2x+F2x+ltKSYjcve4tyGN+h79GParPyQcyv+kjXNhlOefQ9Zox+iVWvdXlWk\noYjnSer2QGkQDk2A+cBPiYbKaxVOUm9w9/82s28BgyqcpH7A3R+q7D20BxE/5WWlbF017/NR3O05\nRomnsKXJdRT3nUC/0Q/RVrdXFamTYnkO4uGwdnd/6TKvGww8DyQDScBsd/8HM+vFF5e5rgMedffz\nwWWxLwLXEt1zmObuOyp7DwVEzYiUl1Ow9iOOrXmNHgc+oLMfpMyTyE0dzJneE6JTfnTpmegyRaSK\nYhkQz1RYTANuBda4+wPVK7H6FBA1zyMRdmxaxqGVr9Bl3/v0iOwh4kZ+4xyOZ9xNxsiH6NwzK9Fl\nikgl4naZq5m1AZ5z9/uutrhYUUAkmDu78tayd/ksOhbNo1dkJwDbUvpwpuMNtB37TU35IVILxTMg\nUoCN7t7/aouLFQVE7bJn+yYKl85i0Ge/ppmfBaJTfhzodgedRkwlI3uoRnGL1AKxPMQ0B7jQKQkY\nALzp7n9R7SqrSQFRe+3fvY1di1+m5c73yCrZQpI5u6wbe7vcTvsbptB70I0KC5EEiWVAjKuwWAbs\nutwcSTVFAVE3HN67i+1LZtJ8+9tkF28g2Zwi60RRx3G0GfYgfa8dS1KywkKkpsTzENNw4GF3//bV\nFhcrCoi65+jBPRQsmkVawdv0P7eORlbOftLZ2eFWWg59kKzrbyM5OTnRZYrUazENCDMbCDwMTCU6\nsO01d/+PaldZTQqIuu3EsUNsWzSblPy36H/mE1KtlMO0Znu7sTS7djLZw+8ipVHjRJcpUu/EYqqN\nXsA0osFwGpgFfNfde8Sy0OpQQNQfp08eJX9xMD/U6ZU0tfMcoyXb2o4h7ZoH6H/jBE0mKBIjsQiI\nCLAY+Ia7bw3adrh7r5hWWg0KiPrp3JlT5C1+lcjmuWSfXEYzK+YEzchvNZpGgzSZoEh1xSIgphDd\ng7gB+APRPYjn3D0zloVWhwKi/is+d4a8JW9Stul1+p1YSkvOcsqbkN/qZpIG3E/OqEmkNW2e6DJF\n6pRYXsXUAngAmA6MAX4NzHH3D2NRaHUoIBqWkvPF5C2by/n1c+h7fBGtOc0ZTyW3xU1YzkSyR03W\nzLMiVRCXq5jMLJ3oieqp7j66GvXFhAKi4SorOU/einc49+lr9D76MW05CcD21ByODfoa2aMn07yl\nZp4VCROLQ0xN3YPhsJd+k8v2iScFhACUl5WxdcVblK9+ns7H19KO45z3RmxpNoyyrIn0GzOVVq3b\nJbpMkVojFgHxIfAJ8Caw1t2Lg/YewFii5yeec/fZsSr6Sikg5GKRsjK2rvmAE2teI+PgAjpyhBJP\nIbfpUM73vZessdNo1bZDossUSahYBIQB9xK9MdDNQHMgAhQQva/0r919T8wqvgoKCKlMpLycbesW\ncuyTV+hxcAFd/CClnkxuk2uJ9BxJj9ue0j0tpEGK20jq2kQBIVXlkQgF6xdzeNUrDN03i1RKove0\nSLuGM33upc/oqaR37JboMkVqRCyvYhpB9K5vZ81sOtEb+vynuxfGptSrp4CQq1Hxnhbd9s6jm++j\n3I3c1Gs43XsCfUZPI71zrRkPKhJzsQyIDcA1wCDg90TvQT3R3cfEoM5qUUBIdXkkws7cTziwfCZd\n9s77/AZIeY0HUtx9JN3GPUWHrhmJLlMkpqoaEClVWFeZu7uZ3Qf8wt1/bWaPVL9EkcSzpCQyBwwn\nc8BwcGdn3hr2L5vJ0MJnabxjI5HtvyKvcX+OZ95D5pjpdOxaayYSEIm7quxBLAbmAt8gOlDuALDe\n3RN+qzDtQUg87cpby75lM+lYNI/M4G55uY1yOJ4xgYwxD9O5m8JC6qZYHmLqAjwKfOLuHwWXuY5z\n92djU+rVU0BITSnc+il7ls2kQ+G79CrfCUBuSn+OZd5DxshpdOnZJ7EFilyBWE/3nQ5cWNlqdz9c\nzfpiQgEhiVC0bT17lr5M+4vDImM8GaMeVlhIrRfLPYjJwM+JzuxqwE1Ep/2eE4tCq0MBIYm2p2AD\nRUtfpsPud8ks/wxQWEjtF8uAWA/c4e4HguWOwHx3vyYmlVaDAkJqk70FGyla+hLtLwqLoz3HkzH6\nYboqLKSWiGVAbKx4QtrMktBJapFK7du+kcIlL5G++116fR4W2RztOYHM0dqzkMSKZUD8G9AfeClo\nmgbkuftfVLvKalJASF2wb3t0z6LdrvfoVb4DCMIiYwKZOgwlCRDLgDBgCjCS6DmIRcCrXgvm6FBA\nSF2zb/tGCpe+TPqud/5oz+KLcxZ9E1yhNASxvoqpPdE7y0WIXsV0qPolVp8CQuqyfTs2RQ9DKSyk\nhsVyD+IJ4B+Aj4nuQYwE/tbdn49FodWhgJD6Yt+OTexeHD1n0Ts4DJV34Wqo0dPp3ENhIbETy4DI\nB0Ze2GsIxkQsdfesmFRaDQoIqY/2bt/0+QnuL8Iim+OZE+g5+hE6d++d4AqlrotlQHwI3OnupcFy\nI+A9dx8Xk0qrQQEh9d2e7RspXPIy7S/es8i8h15jH9bcUHJVYhkQzwEDgTcAB+4neqe5PAB3f7q6\nxV4tBYQ0JEUFGyla8hIdCt/5YgR3owGc6HUPvUY/rFlnpcpiGRD/WNnz7v43V1hbzCggpKEq3Pop\nRUtfpnPhO2REdkenKE8dyKne99JrzHTad9L9LOTS4nJHOTNLry3zMIECQgRgV9469i57iS5F79Iz\nUvjFzY/6TqTvmGm069A10SVKLROvgFjr7kOrVVkMKSBEKnBnZ+4n7F82k6573qW776XMk9iSdi3F\n/SbSb8x0Wqd3THSVUgvEKyDWufu1VezbHXgB6ER0/MQMd/+FmbUFZgEZwE7gIXc/FgzI+wUwHjgL\nfNXd11b2HgoIkXAeibBzyyoOLH+Z7nvfpasfoNST2dJkKOez7idr7DRatUlPdJmSINUOCDN7B/im\nu++s0Pbn7v6fVSygM9DZ3deaWQtgDdET3F8Fjrr7T8zsh0Abd/+BmY0H/pxoQAwneve64ZW9hwJC\n5PI8EmH7hqUcXvkyPffPp7MfosRT2Nzsesr7TyJ7zEM0b9km0WVKDYpFQDwE/BPwPPAvFy5zrUZB\nbwK/DH7Guvu+IEQWunuWmf1P8PjloH/+hX6XWqcCQuTKeCTCtnULObZyJpkH36cDRyn2RmxpfiOR\nAZPIGfMgTZu1THSZEmcxOcRkZs2AvwXuAl4keqgIAHf/9ysoJoPoHE4Dgd3u3rrCc8fcvY2ZvQX8\nxN2XBO0LgB+4++qL1vUk8CRAjx49rtu1a1dVyxCRCiLl5Wxd/QEnVs+mz6EPaMdxzngquS1vJmXg\n/WSPfpC0Js0SXabEQVUDIuUyz5cCZ4BUoAUVAuIKCmkOvAZ8x91PRk81hHcNaftSern7DGAGRPcg\nrrQeEYlKSk4me/idMPxOysvK2LziXc6se4WBR+bRdPmHsPz/sqrVXTS+ZjI5N99H49TURJcsNeyS\nAWFmdwH/DswFhrr72StdeTDq+jXg9+7+etB8wMw6VzjEdDBoLwK6V3h5N2Dvlb6niFy55JQUBoy8\nF0beS2nJeTYufYPi9XPIPv4xLRe9x7FF32Nd21toNnQK/UeMJznlcv+2lPqgsnMQi4Gn3H3zVa04\nuqvwPNET0t+p0P4z4EiFk9Rt3f37ZjYB+DO+OEn9tLvfUNl76ByESHydLz5L3pI3KN3wGjknFtPU\nznOY1hSkj6PV9dPIGjaOpOTkRJcpVygul7leYQEjid7HeiNfHJr6K2AlMBvoAewGprj70SBQfkn0\nfMdZ4ImLzz9cTAEhUnOKz54i9+NX8c2vk3NqOWlWyn7S2dnxdtrd+Ah9Bt+MJSUlukypgoQHRE1Q\nQIgkxumTx8hfOJOU3Dn0P7uaxlZOkXWmsOvddLrpUTJzrkt0iVIJBYSI1IiTRw6Qv/Almmx9k/7F\nn5Jszo6kDA70GE/3UY/SrfeARJcoF1FAiEiNO7x/N9sX/p5W2+eSXboFgK0p/TiaeS+9xj6mGWdr\nCQWEiCTUgd3b+OzjF0nf+RZ9yrcTcSM3dRCn+95P1i2Pal6oBFJAiEitsXvrp+xd/Du6Fr1Nd99L\niSezpen1lOZMov/YaTRv0fryK5GYUUCISK3jkQg7Ni3j0PKXyNz3Hh05wllPZUuLm0ga/CA5ox4g\nrUnTRJdZ7ykgRKRWi0718T4nP5lJ38MLaMNJTtKMvDZjaTp0Kv1vnKABeXGigBCROqOs5Dy5y/5A\n8bpZ5BxfRDMrjg7Ia38HbW98hL5DRmuMRQwpIESkTio+e5otH79C0qZXyDm9ksZWxh7rRGHXCXQd\n/Rjd+w1JdIl1ngJCROq8E8cOs/Wj39Mk/3VyiteTZE5Bcm+O9LqP3mMfI71rZqJLrJMUECJSrxzc\ns4vtC1+g3Y436Ve+jYgbeWmDOZs1iX63PkrL1u0TXWKdoYAQkXpr19b17Fn8It0/v2w2hdzmw/HB\nU+k/ejKpTZonusRaTQEhIvWeRyLkf7qEYyt+R9+D80jnOKdoQl6bW2ly3TRyRownSVdCfYkCQkQa\nlLLSErYse5vitTPJOf4xze0cB2nL9k530f6mx+g9cLiuhAooIESkwTp35hRbFs4iefOrDDizikZW\nzmdJPdjf8156jnmcLhlZiS4xoRQQIiLA8UP72PrRC7Tc9sbnEwhuaTSQU1mTyb71K7Rq2/BObisg\nREQusvezPHYvfJYuhX+gR2QPJZ7C5uY3wjXTyBkzmdTUJokusUYoIERELsEjEQo2LOXIshfoe3Ae\n7TjBcZqT3+42Wg7/CtnDbq3X5ysUECIiVVBWWsKWJW9Qum4mOScW08RKKLJOFHa7l+5jn6iXNzxS\nQIiIXKHTJ4+R9+HvaJL7Kv2Dkdt5Kf053m8y/cc9Rqt29eMeFgoIEZFqOFi0nR0fPkfHnW+QGdlN\niaewqfmN2JDpDBwzmUaN0xJd4lVTQIiIxIBHImzfuIzDS5+n38H3aMtJjtGSre3vJH3k4/QePBLM\nEl3mFVFAiIjEWGnJeTYvnkP5upcYeGoZqVbKrqQeHOg1iT7jvkbbzhmJLrFKFBAiInF0/MhBchc8\nT5utr5Jdlke5G7lNh1E+aBr9b5lO4ybNEl3iJSkgRERqyGf569n78W/ptfctOnOYUzRha7vbaXPT\n42ReW/sumVVAiIjUsLKyMjYufZvza37H4BMf09TOU5TUhX0ZD5B529dJ79Ir0SUCCggRkYQ6cewo\nmxe8SMv82Qws3UTEjS1NhlI2eDr9b32Y1LTEHYJSQIiI1BI7t21iz8Lf0GvPXDpzmJM0Iy/9Dtrc\n/AR9rhlV44egFBAiIrVMeXk5m5a+RcknLzDo5MekWSk7k3qwr9cU+t3xddp16FojdSggRERqsRPH\nDpP/wXO0zp9Fv7KtlHgy65uPImXY4wwaNZGUON7oSAEhIlJH7M5bzb6P/pesA2/TmlPspT3bu00i\n47Yn6Z7RN+bvp4AQEaljSs+fY/OHL9No/YsMKF5LuRvr04ZROvgrXDNuKmlpsZneQwEhIlKHHSrM\nZ+f7/0NG4Rza+1GO0IrcDvfQaeyT9MkZUq11KyBEROqBSFkp+UvfoPST58g5tYwUi7AxZRCncqYz\n+I7HaN68xRWvUwEhIlLPnDi4m23z/5cuO16hS2QfJ70pm9Pvot3oP6HfNTdXeT0KCBGResoj5Wz7\nZB6nl/2WAccXkmqlbEvuw9Hs6eTc8TVatGpb6eurGhBxG51hZr81s4NmtqlCW1sze9/MtgW/2wTt\nZmZPm1mBmW0ws6HxqktEpK6zpGT6DR/P0O++yvnv5rIq+4ckexnDN/8jSf/en5VPP0bBhmXVfp94\nDt97DrjrorYfAgvcvS+wIFgGuBvoG/w8CTwTx7pEROqNlq3bc8O0H5H51+vIv+d1ctuM5Zoj79Dn\n9bvZ+k/Xs+r1pzl35vRVrTtuAeHui4CjFzXfBzwfPH4euL9C+wsetQJobWad41WbiEh9Y0lJZA0b\nx7DvzOL8d3JZkfV9UiPnuGHD31Dys36sfn/WFa+zpueg7eju+wCC3x2C9q5AYYV+RUHbl5jZk2a2\n2sxWHzp0KK7FiojURa3atGfE9B/T4683sOWOl9je4gba9xp8xeuJ31juKxN2v77Qs+fuPgOYAdGT\n1PEsSkSkLrOkJHJumgA3Tbiq19f0HsSBC4eOgt8Hg/YioHuFft2AvTVcm4iIVFDTATEXeDx4/Djw\nZoX2x4KrmUYAJy4cihIRkcSI2yEmM3sZGAukm1kR8HfAT4DZZvZ1YDcwJej+DjAeKADOAk/Eqy4R\nEamauAWEu0+/xFPjQvo68K141SIiIleudt1JW0REag0FhIiIhFJAiIhIKAWEiIiEUkCIiEgoBYSI\niIRSQIiISCgFhIiIhFJAiIhIKAWEiIiEUkCIiEgoBYSIiIRSQIiISCgFhIiIhFJAiIhIKAWEiIiE\nUkCIiEgoBYSIiIRSQIiISCgFhIiIhFJAiIhIKAWEiIiEUkCIiEgoBYSIiIRSQIiISCgFhIiIhFJA\niIhIKAWEiIiEUkCIiEgoBYSIiIRSQIiISCgFhIiIhFJAiIhIKAWEiIiEqlUBYWZ3mVm+mRWY2Q8T\nXY+ISENWawLCzJKB/wLuBnKA6WaWk9iqREQarloTEMANQIG773D3EmAmcF+CaxIRqfOWL4d//ufo\n7yuREp9yrkpXoLDCchEw/OJOZvYk8CRAjx49aqYyEZE6avlyGDcOSkqgcWNYsKDqr61NAWEhbf6l\nBvcZwAwAMztlZvnxLqwOSAcOJ7qIWkLbIkrbIUrbga6doFNXgHPn3G+66cBeoGdVXlmbAqII6F5h\nuRuw9zKvyXf3YfErqW4ws9XaDlHaFlHaDlHaDl+4mm1Rm85BfAL0NbNMM2sMTAPmJrgmEZEGq9bs\nQbh7mZn9GTAPSAZ+6+6bE1yWiEiDVWsCAsDd3wHeuYKXzIhXLXWMtsMXtC2itB2itB2+cMXbwty/\ndB5YRESkVp2DEBGRWkQBISIioepMQJjZFDPbbGYRMxt20XM/CuZvyjezOyu01+u5ncxsiJmtMLNP\nzWy1md0QtJuZPR187g1mNjTRtcabmf158N96s5n9S4X20O9GfWdmf2FmbmbpwXKD+k6Y2c/MLC/4\nrHPMrHWF5xrUd6JafwfdvU78AP2BLGAhMKxCew6wHkgFMoHtRK+CSg4e9wIaB31yEv05YrxN5gN3\nB4/HAwsrPH6X6ODDEcDKRNca5+1wC/ABkBosd6jsu5Hoemtge3QnejXgLiC9gX4n7gBSgsc/BX7a\nEL8T1f07WGf2INw9193DRk3fB8x09/Pu/hlQQHRep4Ywt5MDLYPHrfhiYOF9wAsetQJobWadE1Fg\nDflT4Cfufh7A3Q8G7Zf6btR3Pwe+zx/PRNCgvhPuPt/dy4LFFUQH3kLD+05U6+9gnQmISoTN4dS1\nkvb65DvAz8ysEPhX4EdBe0OrPap+AAAEu0lEQVT47BX1A0aZ2Uoz+9jMrg/aG9p2wMwmAnvcff1F\nTzW4bVHB14juPUHD2w7V+ry1ahyEmX0AdAp56sfu/ualXhbS5oSHX527preybQKMA77r7q+Z2UPA\nb4DbqOK8VnXJZbZDCtCG6KGT64HZZtaLergd4LLb4q+IHl750stC2ur0tqjK3wsz+zFQBvz+wstC\n+tfp7XAZ1fq8tSog3P22q3hZZXM4XencTrVOZdvEzF4Avh0svgL8Onh8NfNa1WqX2Q5/Crzu0YOu\nq8wsQnSStnq3HeDS28LMBhE9rr7ezCD6edcGFy/Uu21xub8XZvY4cA8wLvhuQD3cDpdRrc9bHw4x\nzQWmmVmqmWUCfYFVNIy5nfYCY4LHtwLbgsdzgceCK1dGACfcfV8iCqwhbxD9/JhZP6In4w5z6e9G\nveTuG929g7tnuHsG0T8OQ919Pw3sO2FmdwE/ACa6+9kKTzWo7wTV/DtYq/YgKmNmk4D/BNoDb5vZ\np+5+p7tvNrPZwBaiu5Lfcvfy4DX1fW6nbwC/MLMUoJjgPhlEpysZT/QE3FngicSUV2N+C/zWzDYB\nJcDjwb8YL/ndaIAa2nfil0SvVHo/2Jta4e5PVfb3oj7yas5xp6k2REQkVH04xCQiInGggBARkVAK\nCBERCaWAEBGRUAoIEREJpYCQBsnMupvZZ2bWNlhuEyz3jNP7NQmmAUmu0PZdMys2s1ZVXMcHZtYm\nHvWJhFFASIPk7oXAM8BPgqafADPcfVec3vJrREd7V7zmfjrRgUyTqriOF4FvxrowkUtRQEhD9nNg\nhJl9BxgJ/NvFHczsOTP7lZktNrOtZnZP0J5mZs+a2UYzW2dmtwTtA8xsVXCPjg1m1jdY1SPAmxXW\n2xtoDvw10aC40P5VM/tlheW3zGxssDi3Yl+ReKszI6lFYs3dS83sL4H3gDuC6ZDDZBCd0qQ38JGZ\n9QG+FaxjkJllA/ODaT6eAn7h7r8PpjZIDn73cvedFdY5HXgZWAxkmVmHCtOUX6reY8EUEe3c/cjV\nfm6RqtIehDR0dwP7gIGV9Jnt7hF33wbsALKJ7nG8CODueURvztMPWA78lZn9AOjp7ueIThx4/KJ1\nTiN6X4II8DowpYr1HgS6VLGvSLUoIKTBMrMhwO1Epwn/rpl1NrP/Fxwe+rRC14vno3HCp1HG3V8C\nJgLngHlmdmvwOK3C+w4mOknc+2a2k2hYXDh0VMYf/3+Zxh9LC9YnEncKCGmQLDqD2zPAd9x9N/Az\n4F/d/cfuPsTdh1ToPsXMkoLzBr2AfGAR0fMKF2aQ7QHkB/eh2OHuTxM9ZzDY3Y8RPdR04Y/9dODv\nL8y66u5dgK7BFVQ7gSHB+3Wnwt3Ogpo7BX1E4k4BIQ3VN4Dd7v5+sPzfQLaZjQnpmw98TPSuZE+5\ne3HQP9nMNgKzgK8GtzydCmwK9kCygReCdcwnelgKonsMcy56jzlB+1LgM2Aj0bsErq3Q5zqis5KW\nIVIDNJurSCXM7DngLXd/tZrruRb4nrt/pRrr+AUw190XVKcWkarSHoRIDXD3dUSvgEq+bOdL26Rw\nkJqkPQgREQmlPQgREQmlgBARkVAKCBERCaWAEBGRUAoIEREJ9f8BXCvGI1e4LFgAAAAASUVORK5C\nYII=\n",
      "text/plain": [
       "<matplotlib.figure.Figure at 0x11bb33a90>"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    }
   ],
   "source": [
    "sim.create_scatter(-100,1,-1.5,500)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 53,
   "metadata": {},
   "outputs": [],
   "source": [
    "sim.create_CSV()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 23,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "('Star A final velocity: ',\n",
       " 72.77832879742856,\n",
       " 'Star B final velocity: ',\n",
       " 66.68495435562421)"
      ]
     },
     "execution_count": 23,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "sim.get_velocity(\"f\")"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "ansSet= False\n",
    "\n",
    "inputs = []\n",
    "\n",
    "def get_input():\n",
    "    A_mass = input(\"Star A Mass(Msun): \")\n",
    "    B_mass = input(\"Star B Mass(Msun): \")\n",
    "    Semi = input(\"Semi-Major Axis(Au): \")\n",
    "    bh_mass = input(\"Black Hole Mass(Msun): \")\n",
    "    bh_pos = input(\"Initial Coordinates of Binary System(x,y): \")\n",
    "    #starA = [float(A_info[0:A_info.find(\",\")]), float(A_info[A_info.find(\",\")+1:])]\n",
    "    #starB = [float(B_info[0:B_info.find(\",\")]), float(B_info[B_info.find(\",\")+1:])]\n",
    "    blackhole = [float(bh_mass), float(bh_pos[0:bh_pos.find(\",\")]), float(bh_pos[bh_pos.find(\",\")+1:])]\n",
    "    inputs.append(float(A_mass))\n",
    "    inputs.append(float(B_mass))\n",
    "    inputs.append(float(Semi))\n",
    "    inputs.append(blackhole)\n",
    "    return True "
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "ans = input(\"Would you like rewrite the previous parameters entered?(y/n): \")\n",
    "\n",
    "if(ans == \"y\"):\n",
    "    ansSet = get_input()\n",
    "    \n",
    "elif(ans == \"n\" and ansSet == False):\n",
    "    print(\"You have not entered any parameters\")\n",
    "    while(ansSet == False):\n",
    "        ansSet = getInput()\n",
    "sim = BlackHoleSimulation(inputs[0],inputs[1],inputs[2],inputs[3]) #(Star A mass, Star B mass, Semi, black hole[mass,x,y])"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": []
  }
 ],
 "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.6.3"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 2
}