struct_map_generator.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "struct_def = \"\"\"struct gate_argument_local_vars {\n",
    "        uint256 modulus;\n",
    "        // 0x20\n",
    "        uint256 theta;\n",
    "        // 0x40\n",
    "        uint256 constraint_eval;\n",
    "        // 0x60\n",
    "        uint256 gate_eval;\n",
    "        // 0x80\n",
    "        uint256[][] witness_evaluations;\n",
    "        // 0xa0\n",
    "        uint256[] selector_evaluations;\n",
    "        // 0xc0\n",
    "        uint256 eval_proof_witness_offset;\n",
    "        // 0xe0\n",
    "        uint256 eval_proof_selector_offset;\n",
    "        // 0x100\n",
    "        uint256 gates_evaluation;\n",
    "        // 0x120\n",
    "        uint256 theta_acc;\n",
    "        uint256[] witness_evaluations_offsets;\n",
    "        uint256 selector_evaluations_offset;\n",
    "        uint256 offset;\n",
    "    }\"\"\""
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {
    "scrolled": false
   },
   "outputs": [],
   "source": [
    "def is_field_def(line):\n",
    "    if not line or line.startswith('struct') or line.startswith('//') or line.startswith('{') or line.startswith('}'):\n",
    "        return False\n",
    "    return True\n",
    "\n",
    "def detect_field_size(line):\n",
    "    if '[]' in line:\n",
    "        return 0x20\n",
    "    \n",
    "    size_0x20_types_list = ['uint256', 'bytes32', 'int256', 'bytes']\n",
    "    if any(map(lambda x: line.startswith(x), size_0x20_types_list)):\n",
    "        return 0x20\n",
    "    \n",
    "    size_0x01_types_list = ['byte', 'bool', 'uint8', 'int8']\n",
    "    if any(map(lambda x: line.startswith(x), size_0x01_types_list)):\n",
    "        return 1\n",
    "\n",
    "def is_size_comment(line):\n",
    "    if line.startswith('// 0x'):\n",
    "        return True\n",
    "    return False\n",
    "    \n",
    "def get_field_name(line):\n",
    "    for subs in line.split():\n",
    "        if subs.endswith(';'):\n",
    "            return subs.replace(';', '').upper()\n",
    "\n",
    "def print_offset_constants(offset_map):\n",
    "    for k, v in offset_map.items():\n",
    "        print(f'uint256 constant {k} =', f'{v};')\n",
    "    \n",
    "        \n",
    "result_struct_def = ''\n",
    "current_offset = 0\n",
    "offset_map = dict()\n",
    "for raw_line in struct_def.splitlines():\n",
    "    line = raw_line.strip()\n",
    "    if is_field_def(line):\n",
    "        offset_map[get_field_name(line) + '_OFFSET'] = hex(current_offset)\n",
    "        result_struct_def += f'// {hex(current_offset)}\\n'\n",
    "        current_offset += detect_field_size(line)\n",
    "    if not is_size_comment(line):\n",
    "        result_struct_def += line + '\\n'\n",
    "\n",
    "print(result_struct_def)\n",
    "print_offset_constants(offset_map)"
   ]
  },
  {
   "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.10.4"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 4
}