rf_attributes.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 338,
   "id": "e59dc75e",
   "metadata": {},
   "outputs": [],
   "source": [
    "import paho.mqtt.client as mqtt\n",
    "import time\n",
    "import numpy as np\n",
    "\n",
    "class RF_attributes():\n",
    "    def __init__(self):\n",
    "        super(RF_attributes, self).__init__()\n",
    "        \n",
    "        self.collection_time = 3 #sec\n",
    "        self.start_time = time.time()\n",
    "            \n",
    "        self.rssi = {0.0: [], 1.0: [],2.0: []}\n",
    "        self.tof = {0.0: [], 1.0: [],2.0: []}\n",
    "        self.pde = {0.0: [], 1.0: [],2.0: []}\n",
    "        \n",
    "        self.cols = []\n",
    "\n",
    "        for i in ['tof', 'rssi', 'pde']:\n",
    "            for k in ['mean', 'min', 'max', 'std', 'var']:\n",
    "                for j in range(3):\n",
    "                    self.cols.append(i+ '_' + k +'_(peripheral_' + str(j) + ')')\n",
    "\n",
    "\n",
    "    def on_connect(self, client, userdata, flags, rc):\n",
    "          client.subscribe(\"fromBLE\")\n",
    "\n",
    "    def on_message(self, client, userdata, msg):\n",
    "        global deviceIdCount, label, logger, dataCollectionTime\n",
    "\n",
    "        data = msg.payload.decode()\n",
    "        data = bytes.fromhex(data) \n",
    "        data = data.decode(\"utf-8\") \n",
    "        data = data.split(\":\")\n",
    "        data = data[1].split(\",\")\n",
    "\n",
    "        device_id = float(data[0])\n",
    "\n",
    "        self.tof[device_id].insert(0, float(data[1]))\n",
    "        self.rssi[device_id].insert(0, float(data[2]))\n",
    "        self.pde[device_id].insert(0, float(data[3]))\n",
    "        \n",
    "        if self.done():\n",
    "            client.disconnect()\n",
    "\n",
    "    def done(self):\n",
    "        if time.time() - self.start_time < self.collection_time:\n",
    "            return False\n",
    "        \n",
    "        for key in [0.0, 1.0, 2.0]: \n",
    "            if not bool(self.rssi[key]):\n",
    "                return False    #if value from any peripheral is missing\n",
    "        return True\n",
    "            \n",
    "    def collect_rf_attributes(self, rf_runtime):\n",
    "        self.collection_time = rf_runtime\n",
    "        \n",
    "        client = mqtt.Client(\"P2\")\n",
    "\n",
    "        client.connect(\"localhost\",1883,60)\n",
    "        client.on_connect = self.on_connect\n",
    "        client.on_message = self.on_message\n",
    "        client.loop_forever()\n",
    "\n",
    "    def get_rf_feature(self):\n",
    "        data = [self.tof, self.rssi, self.pde]\n",
    "\n",
    "        feature = []\n",
    "        for d in data:\n",
    "            mean = []\n",
    "            min = []\n",
    "            max = []\n",
    "            std = []\n",
    "            var = []\n",
    "\n",
    "            for v in list(d.values()):\n",
    "                mean.append(np.mean(v))\n",
    "                min.append(np.min(v))\n",
    "                max.append(np.max(v))\n",
    "                std.append(np.std(v))\n",
    "                var.append(np.var(v))\n",
    "\n",
    "            feature += mean + min + max + std + var\n",
    "\n",
    "        feature = pd.DataFrame([feature], columns = self.cols)\n",
    "\n",
    "        return feature"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "ecc7447f",
   "metadata": {},
   "outputs": [],
   "source": []
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3 (ipykernel)",
   "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.8.16"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 5
}