Homogeneous Poisson.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 1,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "713264a70da74563ae2f6c8f552117b5",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "VBox(children=(HBox(children=(Button(description='next arrival', style=ButtonStyle()), Button(description='cle…"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    }
   ],
   "source": [
    "import numpy as np\n",
    "import plotly.graph_objs as go \n",
    "from ipywidgets import widgets\n",
    "\n",
    " \n",
    "import scipy.stats as spst\n",
    " \n",
    "\n",
    "next_arrival=widgets.Button(\n",
    "    description=\"next arrival\")\n",
    " \n",
    "\n",
    "clear=widgets.Button(\n",
    "    description=\"clear\")\n",
    "\n",
    " \n",
    "\n",
    "x=np.array([0,5])\n",
    "y=np.array([0,0])\n",
    "at_x=np.array([0])\n",
    "at_y=np.array([0])\n",
    "at_string=[\"\"]\n",
    "\n",
    "trace1=go.Scatter(x=x,y=y, \n",
    "              hoverinfo=\"text\",text=\"\",name=\"time between 0 and 5\",mode=\"lines\",marker=dict(size=10,color=\"yellow\") )\n",
    "\n",
    "trace2=go.Scatter(x=at_x[1:-1],y=at_x[1:-1], \n",
    "              hoverinfo=\"text\",text=at_string[1:-1],name=\"previous arrivals\",mode=\"markers\" ,marker=dict(size=10,color=\"blue\") )\n",
    "\n",
    "trace3=go.Scatter(x=[at_x[-1]],y=[at_x[-1]],hoverinfo=\"text\", text=[at_string[-1]],\n",
    "                  marker=dict(size=10,color=\"red\"),name=\"new arrival\",mode=\"markers\" )\n",
    "\n",
    "g = go.FigureWidget(data=[trace1],\n",
    "                        layout=go.Layout(\n",
    "                        title=dict(\n",
    "                        text='homogeneous Poisson arrival'\n",
    "                        ),\n",
    "                \n",
    "                        xaxis = dict(title = \" time space\", range = [-0.1,8],showticklabels=True),\n",
    "                        yaxis = dict(title = \"\", range = [-.1,0.5] ,showticklabels=False),\n",
    "                        hovermode=None,\n",
    "                        margin={'l': 0, 'r': 0, 't': 0, 'b': 0},width=1000, height=200  ),\n",
    "                   )\n",
    "g.update_layout(\n",
    "                     title_x=0.5,\n",
    "                 title_y=0.9,\n",
    "\n",
    "                 legend=dict(\n",
    "\n",
    "                    x=0.7,\n",
    "                    y=0.7,\n",
    "                    traceorder=\"normal\",\n",
    "                    font=dict(\n",
    "                        family=\"sans-serif\",\n",
    "                        size=12,\n",
    "                        color=\"black\"\n",
    "                    ))\n",
    "                   \n",
    ")\n",
    "\n",
    "lm=1\n",
    "def response2(change):\n",
    "    global at_x\n",
    "    global at_y\n",
    "    at_x=np.append(at_x,at_x[-1]-1/lm*np.log(np.random.rand()))\n",
    "    at_y=np.append(at_y,0)\n",
    "    at_string=np.round(at_x,2).tolist()\n",
    "    at_string=[\"arrival time:\"+str(item) for item in at_string]\n",
    "    if  at_x[-1]<5:\n",
    "        with g.batch_update():\n",
    "            g.data = [g.data[0]]\n",
    "            trace2.text=at_string[1:-1]\n",
    "            trace2.x=at_x[1:-1]\n",
    "            trace2.y=at_y[1:-1]\n",
    "            trace3.text=[at_string[-1]]\n",
    "            trace3.x=[at_x[-1]]\n",
    "            trace3.y=[at_y[-1]]\n",
    "            g.add_trace(trace2)\n",
    "            g.add_trace(trace3)\n",
    "    else:\n",
    "        next_arrival.disabled=True\n",
    "        with g.batch_update():\n",
    "            g.data = [g.data[0]]\n",
    "            trace2.text=at_string[1:-1]\n",
    "            trace2.x=at_x[1:-1]\n",
    "            trace2.y=at_y[1:-1]\n",
    "            trace3.text=[at_string[-1]]\n",
    "            trace3.x=[at_x[-1]]\n",
    "            trace3.y=[at_y[-1]]\n",
    "            trace3.marker.symbol=\"x-open\"\n",
    "            trace3.marker.color=\"black\"\n",
    "            trace3.name=\"rejected arrival\"\n",
    "\n",
    "            g.add_trace(trace2)\n",
    "            g.add_trace(trace3)\n",
    "        \n",
    "def response1(change):\n",
    "    global at_x\n",
    "    at_x=np.array([0])\n",
    "    global at_y\n",
    "    at_y=np.array([0])\n",
    "    next_arrival.disabled=False\n",
    "    trace3.marker.symbol=\"circle\"\n",
    "    trace3.marker.color=\"red\"\n",
    "    trace3.name=\"new arrival\"\n",
    "\n",
    "    g.data = [g.data[0]]\n",
    " \n",
    "\n",
    "        \n",
    "    \n",
    "container1 = widgets.HBox([next_arrival,clear]) \n",
    "\n",
    "container2 = widgets.HBox([g ])\n",
    "\n",
    " \n",
    "next_arrival.on_click(response2)\n",
    "clear.on_click(response1)\n",
    "\n",
    "Widget1=widgets.VBox([container1,container2] )\n",
    "Widget1"
   ]
  },
  {
   "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.10"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 4
}