m-thesis-introduction/simulations/03_emitter_receiver_simulation.ipynb

{
 "cells": [
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "# Emitter/Receiver Simulation with Signals"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 1,
   "metadata": {},
   "outputs": [],
   "source": [
    "import numpy as np\n",
    "from functools import partial\n",
    "\n",
    "import matplotlib.pyplot as plt"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": []
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "#### Signal"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "metadata": {},
   "outputs": [],
   "source": [
    "from lib.TravelSignal import TravelSignal\n",
    "\n",
    "####\n",
    "from scipy.stats import norm\n",
    "\n",
    "sample_rate = 3e2 # Hz\n",
    "interp_sample_rate = sample_rate * 1/10 # Hz\n",
    "\n",
    "t_offset = 8\n",
    "periodic = False\n",
    "\n",
    "time   = t_offset + np.arange(0, 1, 1/sample_rate) #s\n",
    "time2  = t_offset + np.arange(-1.5, 1, 1/sample_rate) #s\n",
    "\n",
    "signal = norm.pdf(time, time[len(time)//2], (time[-1] - time[0])/10)\n",
    "\n",
    "if False:\n",
    "    mysignal = TravelSignal(signal, sample_rate, t_0 = t_offset, periodic=True)\n",
    "    mysignal2 = TravelSignal(signal, sample_rate, t_0 = t_offset, periodic=False)\n",
    "\n",
    "    fig, ax = plt.subplots(1, 1, figsize=(16,4))\n",
    "    ax.set_title(\"Raw and TravelSignal\")\n",
    "    ax.set_ylabel(\"Amplitude\")\n",
    "    ax.set_xlabel(\"Time\")\n",
    "\n",
    "    ax.plot(time,   signal,                    label='Raw signal')\n",
    "    ax.plot(time2,  mysignal(time2)+0.5,     '.-',     label='TravelSignal(periodic)+0.5')\n",
    "    ax.plot(time2,  mysignal2(time2)-0.5,     '.-',     label='TravelSignal-0.5')\n",
    "\n",
    "    ax.legend()\n",
    "\n",
    "    plt.show();"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "## New code"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "metadata": {},
   "outputs": [],
   "source": [
    "### Location\n",
    "class Location:\n",
    "    \"\"\"\n",
    "    A location is a point designated by a spatial coordinate x.\n",
    "    \"\"\"\n",
    "\n",
    "    def __init__(self, x):\n",
    "        self.x = np.asarray(x)        \n",
    "\n",
    "    def __repr__(self):\n",
    "        return \"Location({})\".format(repr(self.x))\n",
    "\n",
    "    def __getitem__(self, key):\n",
    "        return self.x[key]\n",
    "\n",
    "    def __setitem__(self, key, val):\n",
    "        self.x[key] = val\n",
    "\n",
    "    def __add__(self, other):\n",
    "        if isinstance(other, Location):\n",
    "            other = other.x\n",
    "\n",
    "        return self.__class__(self.x + other)\n",
    "\n",
    "    def __sub__(self, other):\n",
    "        if isinstance(other, Location):\n",
    "            other = other.x\n",
    "\n",
    "        return self.__class__(self.x - other)\n",
    "    \n",
    "    def __eq__(self, other):\n",
    "        if isinstance(other, Location):\n",
    "            other = other.x\n",
    "\n",
    "        return np.all(self.x == other)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 4,
   "metadata": {},
   "outputs": [],
   "source": [
    "### Receiver\n",
    "class Receiver(Location):\n",
    "    \"\"\"\n",
    "    A location able to trace a signal over time.\n",
    "    \n",
    "    Optionally applies a transformation to the traced signal.\n",
    "    \"\"\"\n",
    "    def __repr__(self):\n",
    "        return \"Receiver({})\".format(repr(self.x))\n",
    "    \n",
    "    def recv(self, travel_signal: TravelSignal) -> TravelSignal:\n",
    "        \"\"\"\n",
    "        Return a function that traces the signal as a function of time\n",
    "        at the receiver's location\n",
    "        \"\"\"\n",
    "        return partial(travel_signal, x_f=self.x)                       "
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 5,
   "metadata": {},
   "outputs": [],
   "source": [
    " ### Emitter\n",
    "class Emitter(Location):\n",
    "    \"\"\"\n",
    "    Emit a signal from position x_0 (and time t_0)\n",
    "    \"\"\"\n",
    "    def emit(self, travel_signal: TravelSignal) -> TravelSignal:\n",
    "        return partial(travel_signal, x_0=self.x)"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "## Testing"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 6,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "image/png": "iVBORw0KGgoAAAANSUhEUgAAAPIAAACqCAYAAACXtRI+AAAABHNCSVQICAgIfAhkiAAAAAlwSFlzAAALEgAACxIB0t1+/AAAADh0RVh0U29mdHdhcmUAbWF0cGxvdGxpYiB2ZXJzaW9uMy4xLjMsIGh0dHA6Ly9tYXRwbG90bGliLm9yZy+AADFEAAAQuklEQVR4nO3dfbRVdZ3H8fcHEVAkynhIVCBdpqIktUgrZ1KzHNEpnFU2miU5FqmZzZpaiY3NOGOm0zRrKpXMykkzdFyrYaLGHkwzKx9hYkxERsYQCIGLSmIqBnznj9/vxu54z72Xyz133/Pj81rrrnP2w9n7ux8+e//2Pufco4jAzNrbkLoLMLOd5yCbFcBBNiuAg2xWAAfZrAAOslkBHOSdIOlgSb+UtEnSBS2e16ckfa2V88jzGS7pYUmv6mG810q6u9X1VOb3AUk/H6j5tZteBVnSaZLuk/Q7Sevz8/MkqdUF7ghJKyS9bQBn+UngzogYFRFf6qKeOyW9IOnZyt93+zKjiPhsRHwwT3eypJA0tDKv/trRZwN3RcTaHup5ENgo6R39MM9+IWlkXse37uDrjpW0ulV1DYQegyzp48AXgX8GXgWMB84BjgaGtbS6flbd8fvJJGBJD+OcHxF7Vf4GzY5fVVk3Hwa+2cuXfSuPP1i8G9gMnCBpn7qLGVAR0fQPGA38DnhXD+MNBz4PrATWAdcAe1SGfwhYDjwFLAAmVIYFcB7wKLAJuBQ4ELgHeAa4BRhWGf/PgcXARuBu4LW5/zeBbcDzwLOks+XkPP2zc213Af8FfLSh/geBU5os2ztJYd0I3AkcmvvfAWwFXsjze00Xr70T+GCT6R4LrM51rgeeAE4BTgL+N6+rT1XGvwS4MT9fmZfr2fz3plzH1ty9saftUpn/hcDavP4m5vU3tDLfk4CH87b5DfCJyrB98/jDmyzjWcDS/NrHgA93sfwfryz/WZXhr8z7yjPA/Xm/+HkP++EdwGXAf1frzMNWAJ/I2/q3wL8DI4CReRm2VdbnBNJJbg7wf8CTpP1w7zytzv1qVl63G4C/bdhWtwA35GVfAkyvDO+c7qa8bv+ioda/yuvtaeCHwKTuljsiegzyicCW6oZtMt4X8krfGxgFfBe4PA97a17Q1+cd60pS060a5AXAy4DDSEfU24EDSAeSh4FZedzX541+FLBbXpErOnek/PxtlWl3rvAb8gbbA3gPcF9lnCPyhhrWxXK9hnQgezuwOyl0yzvHpZug9jLIW4C/y9P+ENABzMvr8DBSOA/oIsidy1UN3Ado2NF72C6d8/+nvF32AE4GljRM4wngT/PzVwCvbxj+DPlg2sUynkw6KAs4Bniu8/WV+f9jXv6T8vBX5OE3k8IwEjicdBBpGmTSQWgbMIV0cHiwiyDfTwrp3qSgnFM9qDSM/9fAvcB+ef18BbipYf1/Na+3I0j7bedB/pK87U4i7aeXA/dWpn0q2w8Wf0nax/bJw04h7WOHAkOBi4G7dzbI7wPWNvS7m3R2eh54S95IvwMOrIzzJuDX+fnXgc9Vhu0F/B6YXAny0ZXhi4ALK93/AnwhP/8ycGlDPcuAY3oI8gGVfsNJZ7uDcvfngblNlv/TwC2V7iF5hzp2B4L8XF5fnX+XVnae54HdcveoXOtRDevilL4EuRfb5VjgRWBEZfgZ1R0u91tJaj6/rMky/gZ4S087Wh73P4GPNSx/dRnWA28k7fy/Bw6pDPss3Qf5YmBxfj6B1Dp5XUOQ31fp/hxwTTdBXgocX+neJ9c0tLL+96sMvx84rbKtflwZNgV4vpvaFwMz8/PvA2c37HPP0cNZuadr5CeBMdVry4h4c0S8PA8bAowF9gQWSdooaSPwg9y/c6U+Xnn9s/m1+1bms67y/PkuuvfKzycBH++cT57X/nke3VlVmf9m0pH+fZKGAKfT/JqwsfZteVr7Nhm/KxdExMsrf5+uDHsyIrbm58/nx2bLvqN62i4AHRHxQqX7adIBpepdpDPL45J+KulNDcNHkQ5QLyFphqR7JT2V538SMKYyypMRsaXS/RxpeceSArOqMuxxuncm6ZqdiFgD/JTUYquq3sDrnFczk4D5lXW3lHRwGN/L6TUOG9GZI0lnSlpcmfbhbF8vk4AvVoY9RTood7vP9RTke0hNhpndjLOBtMMdVtlZR0dE50KtycWRF2Ik6frnNz3MuyurgMsagrFnRNyUh0eT1zX2v5509jkeeC4i7mnyusbaRTpw9KX2/tTVcjb262m7dPWaB4EDGg7cD0TETGAc6Yx6S+cwSRNINzyXNRYjaTjwbVKLZ3w++N9K2il70kFqdu9f6Tex2ciS3gwcBFwkaa2ktaTLr9N7eYOzq/W5CpjRsK+NiIid2vaSJpGa5OcDr8zr5SG2r5dVpHsJ1fnuERHdvtXXbZAjYiPwD8BcSe+WtJekIZKmka5dOs9SXwX+VdK4XOy+kv4sT2YecJakaXnjfpZ0jbpih9dCms85ko5SMlLSyZI6zyLrSNfW3crB3UZqtnd3h/YW4GRJx0vanXTttZl0eVGnDlL91WVdB+wnaRj0aru8RESsJt10PDKPP0zSGZJGR8TvSdfDWysvORa4I7dyGg0jXcZ0AFskzQBO6M3C5VbKfwCXSNpT0hReenatmgXcRmrCTst/h5NaJDN6Mct1wCslja70uwa4LAcPSWMldXdC662RpANHR57uWbnW6nwvknRYHj5a0qk9TbTHt58i4nPA37D97uo60oX/hWzfoS8kXaDfK+kZ4MfAwfn1t5OuNb9NunFyIHBaT/NtUstC0k2hq0jNwOWka8NOlwMX52bJJ3qY3A3AVODGbua3jHSf4ErSGe4dwDsi4sUdKPuqhveRF+3Aa5vV9Rzp7uwv8rK+kXTHdgmwVtKGPGrT7dKNrwDvr3S/H1iRX38OaX10OoO043VV4ybgAtLB8GngvaQbb711Pqmpuhb4BvBvXY0kaQTpBuaVEbG28vdr0kG6uwNAZ62PADcBj+X1OYH0lusC4EeSNpFufB21A/U3m9fDpBPIPaQsTQV+URk+n3QD8ua8zh+iFwcj5QvqXY6kM4HZEfEnddcymORW0y9JN3qe6Ga8qcC1EdF4zWw12CWDLGlP0hlsbkTcUHc9Zjtrl/usdb5G7CA1a+bVXI5Zv9glz8hmpdnlzshmJXKQzQrQ398GaktjxoyJyZMn112G9dGiRYs2RMTYnscsV1sHWdJ1pG9DrY+Iw3O/vUnfbJlM+nzteyLi6e6mM3nyZBYuXNjaYq1lJPX08c3itXvT+hukb2hVzQFuj4iDSN+imjPQRVnrzF08t+4SBqW2DnJE3EX6UHnVTNJnqcmPpwxoUdZSX/6fL9ddwqDU1kFuYnznJ5Ly47ia6zFruba+Rt4ZkmaT/j8VEyc2/WKNDQJzF8/9ozPx1OunAnDuEedy3rTz6iprUGn7D4RImgx8r3Kzaxnpi/9P5P/bdGdEdPtFgenTp4dvdrWHqddP5VezfvVH/SQtiojpNZU0KJTYtF7A9m+8zAK+U2MtZgOirYMs6SbS18EOlrRa0tnAFcDbJT1K+l9bV9RZo/Wvc484t+4SBqW2vkaOiNObDDp+QAuxAeNr4q619RnZzBIH2awADrJZARxkswI4yGYFcJDNCuAgmxXAQTYrgINsVgAH2awADrJZARxkswI4yGYFcJDNCuAgmxXAQTYrgINsVgAH2awADrJZARxkswI4yGYFcJDNCuAgmxXAQTYrgINsVgAH2awADrJZARxkswI4yGYFaOtfY+yOpBXAJmArsGVX/yFsK1uxQc6Oi4gNdRdh1mpuWpsVoOQgB/AjSYskza67GLNWKrlpfXRErJE0DrhN0iMRcVfnwBzu2QATJ06sq0azflHsGTki1uTH9cB84MiG4ddGxPSImD527Ng6SjTrN0UGWdJISaM6nwMnAA/VW5VZ65TatB4PzJcEaRnnRcQP6i3JrHWKDHJEPAYcUXcdZgOlyKa12a7GQTYrgINsVgAH2awAfQqypEslfazSfZmkC/qvLDPbEX09I38dmAUgaQh
      "text/plain": [
       "<Figure size 144x144 with 1 Axes>"
      ]
     },
     "metadata": {
      "needs_background": "light"
     },
     "output_type": "display_data"
    },
    {
     "data": {
      "image/png": "iVBORw0KGgoAAAANSUhEUgAAAvEAAAF1CAYAAABh8bWyAAAABHNCSVQICAgIfAhkiAAAAAlwSFlzAAALEgAACxIB0t1+/AAAADh0RVh0U29mdHdhcmUAbWF0cGxvdGxpYiB2ZXJzaW9uMy4xLjMsIGh0dHA6Ly9tYXRwbG90bGliLm9yZy+AADFEAAAgAElEQVR4nOzdeZhcZ3nn/e+t7la3pO6WbUleZVk2NjE2IQYLE15lsQ1J2MIyIcRmCUzCZTIZxhjDEAjJYCZAhkmGEPslMBrgzcJiglkCOMCw2MOYgEHyho1ZjBcsb1otdUvqbrX6fv84p6RSq6rX6q5q1fdzXbq66pznnLqrTkv69dP3OScyE0mSJEkLx6JmFyBJkiRpegzxkiRJ0gJjiJckSZIWGEO8JEmStMAY4iVJkqQFxhAvSZIkLTCGeEk6CkTE0oi4PiJ2RcQn5/i1LoyIu+byNeZaRNwUEa9pdh2SNFOGeEltKSIGq/6MRcS+quevaHZ9M/B7wHHAisy8dPzKiHhXROwf9763zeSFMvPGzDy3at+bI+LCqudnRsSCvwlJRLw2IjIi/t00t/tYRFw1R2VJEmCIl9SmMrO38gf4OfDbVcs+Pn58RHTOf5XTchrw48wcnWDMx6vfd2aunK/ipqOFPutXAzvKr5LUUgzxklRDOXP9qYj4ZEQMAK+MiGdGxHcj4vGIeCQiro6IrqptfjEivh4ROyLi0Yh4S7l8UUT8aUT8LCK2RcS1EXFsuW5pRHwiIraX+/1eRNQM1xFxbkT8n3LcDyLi+eXydwN/CryinGGfVuiMiM5yxvk/lDUORMQ7IuKs8v3uLj+HrnL8syPi/vLxJ4GTgS+Xr30l8K1yXWXG/+nl89dGxI8iYmdEfDkiTh33+n8cEfcAP6pR46KIuK78XB+PiBsj4klV6z9WHo8vl/V/JyJOr1r/nIj4cdlu9LdATPKZnAGsB14HPDciVlWte3ZE3B8Rb4mIrRHxcET8frnujyl+K/Kn5Xv/XLl8dUR8rhx/X0T8x6r9vav8fD9W1n5nRDytav3miLiyPOa7yrHd5boVEfGv5X53RsQXI+KUSQ+6pAXPEC9J9b0E+ASwHPgUMAq8AVhJEfCeQxHyiIjlwNeBLwInAU8Ebiz3cyXwfODXgNXAHuDqct2/B5aWy1cAfwwMjS8kIhYDXwKuB1YBbwQ+FRFnZubbgf/OoZn2f5jh+/0N4Lzyvb0d+DvgEopZ/qcCLxu/Qdm68zDw3PK131e+z+rfdnw/Il4K/GfgRWX9N1N8ttVeCDwd+MU69X0JOAs4EbgT+Kdx618O/DlFW9HPgb8AiIjjgeuAt1Icu83AMyb5LF4NfDczrwN+BoxvUVoNLKH4AeaPgA9GRH9m/h3F98p7yvf+kojoKGv/PnAKxef8nyPiWVX7e3H5fo4Bvsyh74+Kl5XbnQGcD7yqXL4I+F/AGorjtB/420nem6SjgCFekuq7KTO/mJljmbkvM7+fmTdn5mhm3gtsAH69HPtC4MHM/NvMHM7M3Zn5vXLd64A/zcyHMnMIuAp4WUQsoghdK4EzM/NAZm7MzMEatawHFgN/lZn7M/PrFGHvkmm8n5eXs9iVP18bt/69mTmQmXcAdwNfycz7M3Mn8FWKID9Tr6MItpWWn3cBF4ybNX5PZu7MzH3jNy6Pwd+X9VU+w/MjYlnVsOvKz28/8HGKH0gAXgDclpmfK9f9D2BrvUIjIihCcuWHjE9wZEvNEPCu8lh8ARim+MGtll8G+jPzPZk5kpn3AB/h8GP3fzLzq5l5gCLMnzduH+/PzEczczvFDwTnlZ/L1vJ97cvM3cB7OPQ9Keko1ip9h5LUih6sfhIRZ1MEwPMpZs87KWaUAU4F7qmznzXAFyNirGpZAscDf08xm/vPEdFPEeD+rEZv+8nAzzOz+oTRByhmdqfqE5n5mgnWP1b1eF+N58dM47XGOw34QNnKUjFGMaNdeZ0Hj9iqVM5m/yXwUoofeiqf5UqK32wAPFq1yV6gt3x8cvW+M3MsIjZPUOuvURzPfy6ffwL4rxHx5My8s1y2rQzctV5vvNOANRHxeNWyDg79pqZW7dU/nNRafxxA+UPM3wK/yaHj01enDklHEWfiJam+8VdY+Z8UbRxnZmY/8F841Fv9IPCEOvvZDPxGZh5T9aennFkdycyrMvNJwK9QtPDUujrOw8Cp5SxxxRrgoZm9tYYa/znVujLNg8AfjvsMlmTmzZNsV/H7wPOAiynam84sl0/Y2156hCKUFxsUvwFZPcH4V1P8/3hHRDwKfLus7fen8Fpw5Pt4EPjpuPfel5m/PcX9TeQtwOnABeX35MUN2KekBcAQL0lT1wfsAvaUJ1W+rmrdFyhmW18fEYsjoj8iLijXfQh4T0SsgaJHOyJeWD6+OCKeXAbL3RTtNdUzvBX/RtGT/6aI6IqIiylC7T/XGDvfHqPo1a7YAmR5cmjFh4C3V05GjYhjyj75qeqjaFnZTvFbkHdPY9svAedFxIuiuPLNGyn68o8QEUspZvv/kKJlpfLnjRQnN3dM4fXGfx7fAUYi4k0R0RMRHVGcBH3+NN5DPX0UM/M7I2IFxQ+WktqAIV6Spu5NFLO0AxSz8p+qrMjMXRQnHv4ORYj9CYd6k98HfAX4RhRXuvk3ihM4oWj1+CxFgL+L4uTYI27WlJnDwG9TnBi6jeLEx5dn5k+mUX/l6jXVf1ZMY/t63gO8s+yzvyIzByhaX24ul63LzE9TfA6fjojdwB3Ab03jNf4/it9GPEzxOf3bVDfMzMcorhjzVxQ/BKzhUBvUeP+O4vh+rPxNyaOZ+SjFyaNLKI7xZD4M/FJ5tZjrytao5wEXAPdTHL//CfRP9T1M4H0Uv5nYTvGZfLkB+5S0AMTh7ZWSJEmSWp0z8ZIkSdICY4iXJEmSFhhDvCRJkrTAGOIlSZKkBcYQXyov+XVrRHyp2bVIkiRJE/GOrYe8geI245Ne8mvlypW5du3aOS9IkiRJ7WvTpk3bMrPmfS0M8UBErAaeT3HzkCsnG7927Vo2btw453VJkiSpfUXEA/XW2U5TeD/FravHml2IdDTYtW8/L/vQd/jqXY82uxRJko5KbR/iI+IFwJbM3DTJuMsiYmNEbNy6des8VSctTN/52Xa+d/8O/tuXf9TsUiRJOiq1fYgH1gMvjIj7gWuBiyPiY+MHZeaGzFyXmetWrarZmiSpdOdDu4BiRl6SJDVe24f4zHxbZq7OzLXAJcA3M/OVTS5LWtAe2TUEwI49IwztP9DkaiRJOvq0fYiX1HhbBoYOPt6+Z6SJlUiSdHQyxFfJzBsz8wXNrkNa6LYODB98vH1weIKRkiRpJgzxkhpu2+AwZ5/YB8D2QWfiJUlqNEO8pIY6MJZs3zNyMMRvcyZekqSGM8RLaqjB4VEy4fSVvUBxcqskSWosQ7ykhhoYKi4reeLybjoWBQNDo02uSJKko48hXlJDDQ4Xob2vp4ve7k52D3mteEmSGs0QL6mhKjPvvd2d9C/pdCZekqQ5YIiX1FCDQ5WZ+E76ursOttdIkqTGMcRLaqiB4UMhvn9JJ7v3ORMvSVKjGeIlNVRl5r2vp4u+ni574iVJmgOGeEkNNVjVE9/XY0+8JElzwRAvqaEGhkZZFLB0cQf9zsRLkjQnDPGSGmpweJTe7k4igv6eTgaHRxkby2aXJUnSUcUQL6mhdg/tp6+nC4D+JV1kwuCILTWSJDWSIV5SQw0OjdLX0wkUffGVZZIkqXEM8ZIaamBo9GB4r8zIe3KrJEmNZYiX1FCDw4dm4itfveGTJEmNZYiX1FCDw6P0ljPwvZUQP+xMvCRJjTTrEB8RByLitqo/b53GtidHxHXl4/Mi4nlV6y6MiP9ntvVJml8DQ/sPttP0H5yJN8RLktRInQ3Yx77MPG8mG2bmw8BLy6fnAeuAfy2fXwgMAv821f1FRGdmmhakJhoYGj0Y3g/1xNtOI0lSI81ZO01E3B8R74mI70TExoh4WkR8NSJ+FhF/VI5ZGxF3RsRi4L8Cv1fO5v8J8EfAG8vnvxoRqyLiMxHx/fLP+nI
      "text/plain": [
       "<Figure size 864x432 with 4 Axes>"
      ]
     },
     "metadata": {
      "needs_background": "light"
     },
     "output_type": "display_data"
    }
   ],
   "source": [
    "if True:\n",
    "    sample_rate = 3e2 # Hz\n",
    "    periodic = False\n",
    "    \n",
    "    t_offset = 8\n",
    "    t_start = 0\n",
    "    t_end = 1\n",
    "    time   = t_offset + np.arange(t_start, t_end, 1/sample_rate) #s\n",
    "    \n",
    "    t_longstart = 0\n",
    "    t_longend = 30*t_end\n",
    "    longtime = np.arange(t_longstart, t_longend, 1/sample_rate) #s\n",
    "\n",
    "if False:\n",
    "    if True:\n",
    "        freq = sample_rate/8\n",
    "        signal = np.cos(2*np.pi*freq*time)\n",
    "    else: \n",
    "        from scipy.stats import norm\n",
    "        signal = norm.pdf(time, time[len(time)//2], (time[-1] - time[0])/10)\n",
    "\n",
    "\n",
    "#####\n",
    "# Setup Signal, Emitter and Antennae\n",
    "\n",
    "mysignal = TravelSignal(signal, sample_rate, t_0 = t_offset, periodic=periodic)\n",
    "\n",
    "source = Emitter([1,1])\n",
    "emitted = source.emit(mysignal)\n",
    "\n",
    "antennae = [\n",
    "    Receiver([2,3]),\n",
    "    Receiver([10,10]),\n",
    "    Receiver([-2,-3]),\n",
    "]\n",
    "    \n",
    "#####\n",
    "# Follow traces, and show geometry\n",
    "ylabel_kw = {\"rotation\": \"horizontal\", \"va\":\"center\", \"ha\":\"center\", \"labelpad\": 30}\n",
    "\n",
    "fig, axs = plt.subplots(1,1, figsize=(2,2))\n",
    "axs = [ axs ]\n",
    "\n",
    "### Geometry Plot\n",
    "i = 0\n",
    "axs[i].set_title(\"Geometry of Emitter(s) and Antennae\")\n",
    "axs[i].set_ylabel(\"y\", **ylabel_kw)\n",
    "axs[i].set_xlabel(\"x\")\n",
    "axs[i].plot(*source.x, '*', label=\"Emitter\")\n",
    "\n",
    "for j, ant in enumerate(antennae):\n",
    "    axs[i].plot(*ant.x, '+', label=\"Antenna {}\".format(j))\n",
    "\n",
    "### Plot Traces\n",
    "fig, axs = plt.subplots(1+len(antennae),1, sharex=True, figsize=(12,6))\n",
    "axs[0].set_title(\"Traces of Emitter and Antenna\")\n",
    "\n",
    "# Emitter\n",
    "i = 0\n",
    "axs[i].set_ylabel(\"Emitter\\n at ({},{})\".format(*source.x), **ylabel_kw)\n",
    "axs[i].plot(time, emitted(time))\n",
    "\n",
    "# Antenna\n",
    "for j, ant in enumerate(antennae):\n",
    "    i +=1\n",
    "    axs[i].set_ylabel(\"Antenna {}\\n at ({},{})\".format(j, *ant.x), **ylabel_kw)\n",
    "    axs[i].plot(longtime, ant.recv(emitted)(longtime), label=\"Antenna {}\".format(j))\n",
    "    "
   ]
  }
 ],
 "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.6"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 4
}