loggercontroller.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "# !pip install pycapnp\n",
    "# !pip install typer\n",
    "# !pip install ipywidgets"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "import os\n",
    "import logging\n",
    "import numpy as np\n",
    "from pathlib import Path\n",
    "\n",
    "import asyncio\n",
    "import zmq.asyncio\n",
    "\n",
    "import capnp\n",
    "import typer\n",
    "import time\n",
    "from deepdrrzmq.utils.zmq_util import zmq_no_linger_context\n",
    "\n",
    "from deepdrrzmq.utils.typer_util import unwrap_typer_param\n",
    "from deepdrrzmq.utils.server_util import make_response, DeepDRRServerException, messages\n",
    "import random\n",
    "import string\n",
    "\n",
    "import threading\n",
    "from IPython.display import display\n",
    "import ipywidgets as widgets\n",
    "import time\n",
    "from threading import Event\n",
    "import asyncio"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "# rep_port=40100\n",
    "# pub_port=40101\n",
    "# sub_port=40102\n",
    "rep_port=40120\n",
    "pub_port=40121\n",
    "sub_port=40122"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "context = zmq.Context()\n",
    "context.__enter__()\n",
    "\n",
    "sub_socket = context.socket(zmq.SUB)\n",
    "sub_socket.hwm = 10000\n",
    "\n",
    "pub_socket = context.socket(zmq.PUB)\n",
    "pub_socket.hwm = 10000\n",
    "\n",
    "# addr = \"127.0.0.1\"\n",
    "addr = \"128.220.12.196\"\n",
    "# addr = \"172.16.92.40\"\n",
    "# addr = \"localhost\"\n",
    "pub_socket.connect(f\"tcp://{addr}:{pub_port}\")\n",
    "sub_socket.connect(f\"tcp://{addr}:{sub_port}\")\n",
    "\n",
    "sub_socket.subscribe(b\"/loggerd/\")"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "record_button = widgets.Button(description='New Session', disabled=True, button_style='', icon='circle')\n",
    "stop_button = widgets.Button(description='Stop', disabled=True, button_style='', icon='square')\n",
    "\n",
    "session_id = widgets.Text(value='None', placeholder='', description='Session ID:', disabled=True)\n",
    "record_status = widgets.Text(value='None', placeholder='', description='Recording:', disabled=True)\n",
    "\n",
    "async def f():\n",
    "    last_heard_time = 0\n",
    "\n",
    "    while True:\n",
    "        latest_msgs = {}\n",
    "\n",
    "        try:\n",
    "            ts = time.time()\n",
    "            for i in range(1000):\n",
    "                topic, data = sub_socket.recv_multipart(flags=zmq.NOBLOCK)\n",
    "                latest_msgs[topic] = data\n",
    "        except zmq.ZMQError:\n",
    "            pass\n",
    "\n",
    "        for topic, data in latest_msgs.items():\n",
    "            if topic == b\"/loggerd/status/\":\n",
    "                last_heard_time = time.time()\n",
    "                record_button.disabled = False\n",
    "                stop_button.disabled = False\n",
    "                with messages.LoggerStatus.from_bytes(data) as msg:\n",
    "                    session_id.value = msg.sessionId if msg.recording else \"no session\"\n",
    "                    record_status.value = str(msg.recording)\n",
    "                    record_button.button_style = 'danger' if msg.recording else ''\n",
    "\n",
    "        if time.time() - last_heard_time > 3:\n",
    "            session_id.value = \"Disconnected\"\n",
    "            record_status.value = \"Disconnected\"\n",
    "            record_button.disabled = True\n",
    "            stop_button.disabled = True\n",
    "\n",
    "        await asyncio.sleep(0.1)\n",
    "\n",
    "try:\n",
    "    ui_update_task.cancel()\n",
    "except:\n",
    "    pass\n",
    "ui_update_task = asyncio.create_task(f())\n",
    "\n",
    "def on_record_button_clicked(b):\n",
    "    pub_socket.send_multipart([b\"/loggerd/start/\", b\"\"])\n",
    "    \n",
    "\n",
    "def on_stop_button_clicked(b):\n",
    "    pub_socket.send_multipart([b\"/loggerd/stop/\", b\"\"])\n",
    "\n",
    "colortoggle = widgets.ToggleButtons(\n",
    "    options=[('Opaque', 0), ('Transparent' ,1)],\n",
    "    description='Patient Material:',\n",
    "    disabled=False,\n",
    "    button_style='', # 'success', 'info', 'warning', 'danger' or ''\n",
    ")\n",
    "\n",
    "viewindicator = widgets.ToggleButtons(\n",
    "    options=[('On',True), ('Off',False)],\n",
    "    description='View Indicator ON/OFF:',\n",
    "    disabled=False,\n",
    "    button_style='', # 'success', 'info', 'warning', 'danger' or ''\n",
    ")\n",
    "\n",
    "selfviewindicator = widgets.ToggleButtons(\n",
    "    options=[('On',True), ('Off',False)],\n",
    "    description='View Indicator Self-Select ON/OFF:',\n",
    "    disabled=False,\n",
    "    button_style='', # 'success', 'info', 'warning', 'danger' or ''\n",
    ")\n",
    "\n",
    "Flippatient = widgets.ToggleButtons(\n",
    "    options=[('Original',True), ('Flipped',False)],\n",
    "    description='Flip Patient:',\n",
    "    disabled=False,\n",
    "    button_style='', # 'success', 'info', 'warning', 'danger' or ''\n",
    ")\n",
    "\n",
    "kwireindicator = widgets.ToggleButtons(\n",
    "    options=[('On',True), ('Off',False)],\n",
    "    description='Kwire Error Indicator ON/OFF:',\n",
    "    disabled=False,\n",
    "    button_style='', # 'success', 'info', 'warning', 'danger' or ''\n",
    ")\n",
    "\n",
    "kwireerrorselection = widgets.ToggleButtons(\n",
    "    options=[('On',True), ('Off',False)],\n",
    "    description='Kwire Error Indicator Self-Selection',\n",
    "    disabled=False,\n",
    "    button_style='', # 'success', 'info', 'warning', 'danger' or ''\n",
    ")\n",
    "\n",
    "corridorselection= widgets.Select(\n",
    "    options=[('Left S1', 0), ('Left Ramus', 1), ('L Ramus Short', 2),('L Teardrop', 3),('Right S1', 4), ('Right Ramus', 5), ('Right Ramus Short', 6),('Right Teardrop', 7),('S 2', 8),('S 3', 9),('None',100)],\n",
    "    value=2,\n",
    "    description='Corridor:',\n",
    ")\n",
    "\n",
    "\n",
    "updatesetting = widgets.Button(\n",
    "    description='Update Setting',\n",
    "    disabled=False,\n",
    "    button_style='', # 'success', 'info', 'warning', 'danger' or ''\n",
    "    tooltip='Click me',\n",
    "    icon='check' # (FontAwesome names without the `fa-` prefix)\n",
    ")\n",
    "\n",
    "viewindicatorselect = widgets.SelectMultiple(\n",
    "    options=['Anteroposterior_View', 'Lateral_View', 'Inlet_View', 'Outlet_View', 'Obturator_Oblique_Left/Iliac_Oblique_Right', 'Obturator_Oblique_Right/Iliac_Oblique_Left', 'Obturator_Oblique_Inlet_Left/Iliac_Oblique_Inlet_Right','Obturator_Oblique_Outlet_Left/Iliac_Oblique_Outlet_Right','Obturator_Oblique_Inlet_Right/Iliac_Oblique_Inlet_Left','Obturator_Oblique_Outlet_Right/Iliac_Oblique_Outlet_Left', 'Teardrop_Left_View','Teardrop_Right_View','None'],\n",
    "    value=['Lateral_View'],\n",
    "    #rows=10,\n",
    "    description='View Indicator Selection:',\n",
    "    disabled=False\n",
    ")\n",
    "\n",
    "patientSwitch= widgets.Select(\n",
    "    options=[('100114', \"100114\"), ('100129', \"100129\"), ('100139', \"100139\"), ('100155', \"100155\"), ('100229', \"100229\")],\n",
    "    value=\"100114\",\n",
    "    description='Patient Switch:',\n",
    ")\n",
    "\n",
    "left_box = widgets.VBox([colortoggle,Flippatient,patientSwitch])\n",
    "middle_box = widgets.VBox([viewindicator,selfviewindicator,viewindicatorselect])\n",
    "right_box = widgets.VBox([kwireindicator,kwireerrorselection,corridorselection])\n",
    "display(widgets.HBox([left_box,middle_box ,right_box,updatesetting]))\n",
    "\n",
    "def opdatesetting_clicked(b):\n",
    "    response_topic = \"/mp/setting/webgui/SettingManager/\"\n",
    "    msg = messages.SyncedSetting.new_message()\n",
    "    msg.timestamp = msg.timestamp = float(np.float64(time.time()).item())\n",
    "    msg.clientId = 'webgui'\n",
    "    uiControlVar = msg.setting.init(\"uiControl\")\n",
    "    uiControlVar.patientMaterial = colortoggle.value\n",
    "    uiControlVar.flippatient = Flippatient.value #need add      \n",
    "    uiControlVar.carmIndicator = viewindicator.value\n",
    "    uiControlVar.viewIndicatorselfselect = selfviewindicator.value #need add\n",
    "    uiControlVar.annotationError = viewindicatorselect.value#for different view selection\n",
    "    uiControlVar.corridorIndicator =kwireindicator.value\n",
    "    uiControlVar.webcorridorerrorselect = kwireerrorselection.value\n",
    "    uiControlVar.webcorridorselection = corridorselection.value\n",
    "    uiControlVar.patientCaseId= patientSwitch.value\n",
    "    pub_socket.send_multipart([response_topic.encode(), msg.to_bytes()])\n",
    "    print(f\"Update!\")\n",
    "\n",
    "updatesetting.on_click(opdatesetting_clicked)\n",
    "record_button.on_click(on_record_button_clicked)\n",
    "record_button.on_click(opdatesetting_clicked)\n",
    "stop_button.on_click(on_stop_button_clicked)\n",
    "display(widgets.HBox([record_button, stop_button, session_id, record_status]))"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": []
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "deepdrr_zmq",
   "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.10.14"
  },
  "orig_nbformat": 4
 },
 "nbformat": 4,
 "nbformat_minor": 2
}