{
 "cells": [
  {
   "cell_type": "markdown",
   "source": [
    "## Dispersion"
   ],
   "metadata": {
    "collapsed": false
   }
  },
  {
   "cell_type": "code",
   "execution_count": 1,
   "outputs": [
    {
     "data": {
      "text/plain": "<IPython.core.display.Markdown object>",
      "text/markdown": "|key|set0|set1|set2|set3|\n|-|-|-|-|-|\n|key1|1|10|11|100|\n|key2|2|20|22|200|\n|key3|3|30|33|300|\n|key4|4|40|44|400|\n|key5|5|50|55|500|"
     },
     "metadata": {},
     "output_type": "display_data"
    }
   ],
   "source": [
    "from src.utils import secure_fisher_yates_shuffle, matrix_to_list, list_to_matrix\n",
    "from src.user_interface import UserKeypad\n",
    "from IPython.display import Markdown, display\n",
    "from src.models import KeypadSize\n",
    "\n",
    "def keypad_md_table(interface: list[int], keypad_size: KeypadSize) -> str:\n",
    "    assert (keypad_size.numb_of_attrs == len(interface))\n",
    "    interface_keypad = list_to_matrix(interface, keypad_size.attrs_per_key)\n",
    "    table = \"|key|\" + \"\".join([f\"set{idx}|\" for idx in range(keypad_size.attrs_per_key)])\n",
    "    table += \"\\n|\" + \"\".join(\"-|\" for _ in range(keypad_size.attrs_per_key+1))\n",
    "\n",
    "    for key in range(keypad_size.numb_of_keys):\n",
    "        table += f\"\\n|key{key+1}|\"\n",
    "        table += \"|\".join([str(attr) for attr in interface_keypad[key]])\n",
    "        table += \"|\"\n",
    "    return table\n",
    "\n",
    "\n",
    "keypad_size = KeypadSize(numb_of_keys=5, attrs_per_key=4)\n",
    "attrs = [1, 10, 11, 100]\n",
    "keypad = []\n",
    "for key_numb in range(1,keypad_size.numb_of_keys+1):\n",
    "    keypad.extend([key_numb * attr for attr in attrs])\n",
    "\n",
    "demo_interface = UserKeypad(keypad_size=keypad_size, keypad=keypad)\n",
    "\n",
    "display(Markdown(keypad_md_table(demo_interface.keypad, keypad_size)))\n"
   ],
   "metadata": {
    "collapsed": false,
    "ExecuteTime": {
     "end_time": "2024-07-24T22:23:18.200797Z",
     "start_time": "2024-07-24T22:23:18.119660Z"
    }
   }
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "outputs": [
    {
     "data": {
      "text/plain": "<IPython.core.display.Markdown object>",
      "text/markdown": "|key|set0|set1|set2|set3|\n|-|-|-|-|-|\n|key1|3|30|33|300|\n|key2|1|10|11|100|\n|key3|5|50|55|500|\n|key4|2|20|22|200|\n|key5|4|40|44|400|"
     },
     "metadata": {},
     "output_type": "display_data"
    }
   ],
   "source": [
    "demo_interface_matrix = list_to_matrix(demo_interface.keypad, demo_interface.keypad_size.attrs_per_key)\n",
    "shuffled_keys = secure_fisher_yates_shuffle(demo_interface_matrix)\n",
    "shuffled_keys_list = matrix_to_list(shuffled_keys)\n",
    "display(Markdown(keypad_md_table(shuffled_keys_list, keypad_size)))\n"
   ],
   "metadata": {
    "collapsed": false,
    "ExecuteTime": {
     "end_time": "2024-07-24T22:23:18.201039Z",
     "start_time": "2024-07-24T22:23:18.185696Z"
    }
   }
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "outputs": [
    {
     "data": {
      "text/plain": "<IPython.core.display.Markdown object>",
      "text/markdown": "|key|set0|set1|set2|set3|\n|-|-|-|-|-|\n|key1|2|10|55|400|\n|key2|4|50|22|300|\n|key3|3|20|44|100|\n|key4|1|40|33|500|\n|key5|5|30|11|200|"
     },
     "metadata": {},
     "output_type": "display_data"
    }
   ],
   "source": [
    "attr_rotation = secure_fisher_yates_shuffle(list(range(keypad_size.numb_of_keys)))[:keypad_size.attrs_per_key]\n",
    "dispersed_interface = UserKeypad.random_attribute_rotation(\n",
    "    shuffled_keys,\n",
    "    attr_rotation\n",
    ")\n",
    "\n",
    "display(Markdown(keypad_md_table(matrix_to_list(dispersed_interface), keypad_size)))\n"
   ],
   "metadata": {
    "collapsed": false,
    "ExecuteTime": {
     "end_time": "2024-07-24T22:23:18.201164Z",
     "start_time": "2024-07-24T22:23:18.190198Z"
    }
   }
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 2
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython2",
   "version": "2.7.6"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 0
}