rom_decompress_test.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 30,
   "metadata": {},
   "outputs": [],
   "source": [
    "%load_ext autoreload\n",
    "%autoreload 2"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 31,
   "metadata": {},
   "outputs": [],
   "source": [
    "from rom_tools import rom_manager\n",
    "from rom_tools import address\n",
    "from rom_tools import byte_ops\n",
    "from rom_tools.compress import decompress\n",
    "from rom_tools.compress import compress"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 32,
   "metadata": {},
   "outputs": [],
   "source": [
    "# Load the ROM\n",
    "rom_m = rom_manager.RomManager(\"../roms/sm_clean.smc\", \"../roms/sm_newlevel.smc\")"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 33,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "0x792b3\n",
      "b'\\x01\\x00\\x12\\x02\\x05\\x01p\\xa0\\x00\\xf9\\x92\\x12\\xe6\\x00\\xdf\\x92\\xe6\\xe5\\xe8\\xd6\\xc2\\x00\\x00\\x00\\xe0\\x80z\\x84\\xa1\\x80\\xc1\\xc1\\x00\\x00\\x00\\x00\\x00\\x00X\\x80\\x99\\xb8\\xd3\\x91\\xe8\\xd6\\xc2\\x00\\t\\x05\\xe0\\x80z\\x84\\xa1\\x80\\xc1\\xc1\\x00\\x00\\x00\\x00\\x00\\x00X\\x80\\x99\\xb8\\xd3\\x91F\\x89R\\x89'\n"
     ]
    }
   ],
   "source": [
    "# Get the room header for room x0792b3\n",
    "addr = address.Address(0x792b3)\n",
    "print(addr)\n",
    "room_header = rom_m.read_from_clean(addr, 74)\n",
    "print(room_header)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 34,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "12769000\n",
      "0xc2d6e8\n"
     ]
    }
   ],
   "source": [
    "# Get the leveldata as a PC address\n",
    "#TODO...\n",
    "leveldata_ptr_b = room_header[18:21]\n",
    "leveldata_ptr = int.from_bytes(leveldata_ptr_b, byteorder='little')\n",
    "print(leveldata_ptr)\n",
    "print(hex(leveldata_ptr))\n",
    "level_addr = address.Address(0x2156e8)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 35,
   "metadata": {},
   "outputs": [],
   "source": [
    "level_size = 0x100\n",
    "#compressed_level_data = romM.readFromRom(level_addr+2, 763)\n",
    "#compressed_level_data = romM.readFromRom(level_addr+2, 738)\n",
    "compressed_level_data = rom_m.read_from_clean(level_addr, 1244)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 46,
   "metadata": {
    "scrolled": true
   },
   "outputs": [
    {
     "data": {
      "text/plain": [
       "3842"
      ]
     },
     "execution_count": 46,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "level_data = decompress.decompress(compressed_level_data)\n",
    "len(level_data)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 47,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "821\n"
     ]
    }
   ],
   "source": [
    "#recompressed_level_data = compress.compress(level_data[0:1000], min_size=50, debug=True)\n",
    "recompressed_level_data = compress.greedy_compress(level_data, min_size=2)\n",
    "print(len(recompressed_level_data))"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 48,
   "metadata": {},
   "outputs": [],
   "source": [
    "re_level_data = decompress.decompress(recompressed_level_data)\n",
    "for i in range(len(level_data)):\n",
    "    if level_data[i] != re_level_data[i]:\n",
    "        print(i)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 8,
   "metadata": {},
   "outputs": [],
   "source": [
    "#cf = open(\"tmp.bin\", \"wb\")\n",
    "#cf.write(compressed_level_data)\n",
    "#cf.close()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 9,
   "metadata": {},
   "outputs": [],
   "source": [
    "#cf = open(\"tmp.bin\", \"wb\")\n",
    "#cf.write(recompressed_level_data)\n",
    "#cf.close()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 17,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "b'\\x00\\n'\n"
     ]
    },
    {
     "data": {
      "text/plain": [
       "2560"
      ]
     },
     "execution_count": 17,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "print(level_data[0:2])\n",
    "int.from_bytes(level_data[0:2], byteorder='little')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 10,
   "metadata": {},
   "outputs": [],
   "source": [
    "from world_rando import room_gen\n",
    "from world_rando import room_viz\n",
    "from world_rando.coord import *"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 11,
   "metadata": {},
   "outputs": [],
   "source": [
    "level = room_gen.level_from_bytes(level_data, Coord(80, 16))"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 12,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "image/png": "iVBORw0KGgoAAAANSUhEUgAABQAAAAEACAYAAADsq9vkAAAXRElEQVR4nO3dQZbbOIwAUGle+lC5yxxv7tKHysKzSJRUqqvKsigJBPD/ql+/qCwTAEVBtL0uy/JYAAAAAICS/if6BAAAAACA62gAAgAAAEBhGoAAAAAAUJgGIAAAAAAUpgEIAAAAAIVpAAIAAABAYRqAAAAAAFCYBiAAAAAAFKYBCAAAAACFaQACAAAAQGEagAAAAABQmAYgAAAAABSmAQgAAAAAhWkAAgAAAEBhGoAAAAAAUJgGIAAAAAAUpgEIAAAAAIVpAAIAAABAYRqAAAAAAFCYBiAAAAAAFKYBCAAAAACFaQACAAAAQGEagAAAAABQmAYgAAAAABSmAQgAAAAAhWkAAgAAAEBhGoAAAAAAUJgGIAAAAAAUpgEIAAAAAIVpAAIAAABAYRqAAAAAAFCYBiAAAAAAFKYBCAAAwPQej8fyeDyiTwMgpXVZFjMoAAAAu73aiFvX9dDx23Hv//1n/3/v62c3Ov5AP3YAAgAAMJ2vmnzb/9PYAthHAxAAAICp7NnhpwkIsJ8GIAAAANN45eO9moAA+2gAAgAAMIVXv9vv7b/VBAT4nAYgAAAA4Y40/zaagABf0wAEAAAg1Ejzb6MJCPA5DUAAAADCnNH822gCAnxMAxAAAIAQZzb/NpqAAP+lAQgAAMDtrmj+bTQBAf62Lsty/mwLAAAAT1zR/HtLAxDgJw1AAAAAXvJq4+6jRtzVzb+vXju7M8Yf6EUDEAAAgFvd1fzbaIAB3fkOQAAAAG5zd/Mv6jUBZmIHYHPZL4Tdn+RFx6/7+MOI6Podpf4hzuj8oX5jZY+f61cs459b9/qPPv/u7ABsLPvFY1lqvIejZnjvM5wDZFShdiq8B8jojNpTv3Gyx0/uxDL+ZK9/ORzLDsCTjH4J613Hb8dVK7xX31fU+I8eP2v8osef3tRvLPVPZt3nj+71K349r1+z5KHx7+msOGav/+7Xnyh2ADZS7eLx1vaeKhf2zPHrMP4wYub6HaX+4VpXzh/q93rZ41f5+pWB8ecz2evf9SeGBmATHS4elSeRDPGrPP4wIkP9jlL/cI075g/1e53s8etw/ZqZ8eeZ7PXv+nM/DcAGOl08Kk4imeJXcfxhRKb6HaX+4Vx3zh/q93zZ49fp+jUj489e2evf9edeGoDFdbx4VJpEMsav0vjDiIz1O0r9wzki5g/1e57s8et4/ZqJ8edV2evf9ec+GoCFdb54VJhEMsevwvjDiMz1O0r9w5jI+UP9jssev87XrxkYf47KXv+uP/fQACzKxSP3JFIhfpnHH0ZUqN9R6h+OmWH+UL/HZY/fDOffmfFnVPb6d/25ngZgQTMU7ywyTiKV4pdx/GFEpfodpf7hNTPNH+r3ddnjN9P5d2T8OUv2+nf9uZYGYDEzFe8sMk0iFeOXafxhRMX6HaX+YZ8Z5w/1u1/2+M14/p0Yf86Wvf5df66jAVjIjMU7iwyTSOX4ZRh/GFG5fkepf/jazPOH+n0ue/xmPv8OjD9XyV7/rj/XWJdlmS/aHDZj8c5k9gmkevxmH38YUb1+R6l/+Nzs84f6/Vr2+M1+/mebLZ+Nfy93xzt7/XfPl7NpAJ7k1cJ5n8ijxx/5G12dMXbid9wV409v6jcP9c9szB/7Vazf7vFbljznf6aZ8tD49xMR8+z1X/H6E0UDsIgsxTuL2SaAbvGbbfxhRLf6HaX+4Y9s84f6/Vv2+GU7/7PMksfGv6eouGev/+55cxbfAVhAtuKdwUxjNtO53KXje6Ymufw6YwY/ZayFjOd8lYxj8facM55/Jcafu2Wv/4znPCM7AAdJxNyinyR0z5/o8Y8m/uqvs+j4jxrNn+zvPzv1Hys6/8W/N/kXK3r8o3WP/6ju+TPKDsABihfGdK6hzu8dliV3DWQ+dwCArCLXYI/HI/0a8Fv0CZzlri9xnvmnsjnu3xc3wn5f3uXP7/948nd+5Y/nFn88Ho+X62q2Jz/mjzHqr68z6t/1HwDyiV7/Rb8+x925fvzodTLfv9oB+AKLf7jGVlOzTIxXMH/AxzLUv/oFAJjHHevHr9Z/GdavH9EA3MniH66VdRLdw/wBX5u5/tUvAMB8rlw/7ln/zbx+/YwG4A4W/3CPjJPoM+YP2GfG+le/AADzumL9+Mr6b8b161c0AJ+w+Id7ZZtEv2L+gNfMVP/qFwBgfmeuH4+s/2Zavz5T5kdArmDxz8sSFH0G2xerruuatv7MHwHUXwkz1L/6BYAkotd/0a/PsiznrB9H1n8zrF/3sAPwExb/ECvTk5T3zB8wJrL+1S8AQD4j68cz1n8Z7l/tAPyAxT+H7fwZePbJ8iTlLfNHIPVXSkT9q18ASCZ6/Rf9+vzlyPrxzPXf7PevdgC+Y/EPc8nwJGVj/oBz3Vn/6hcAIL9X1o9XrP9mvn/VAHzD4h/mNPMkujF/wDXuqH/1CwBQx57145Xrv1nvX8t8BHh0YC3+YW5vt1PPxvwB1/qq/l3/AQB4b8/68cr134z3r2UagK8G7qMgWPzD3LZJdEbmD7jWZ/Xv+g8AwEe+un+8Y/032/3ruiyLVe9i8d9VdDHKu2Oi4/aeOB4THUdxy+nsvInKg+j87079x4rOf/HvTf7Fih7/aN3jH+V93t0dh1ny3ncALooQspmpZmc6F+jgii9pBgCgrrdrvoj13yxrzvQfAZ5lIIF7zbCd2vwDMdQeAACviF4/znD/mnoHYHQAAQCOsIaJY+wBgI6m2QG4dzHm1/q4wmNZlmVvTq3r8r5vP3o8dKb+6Ortk+BX10Fv/8Yruh5v/Qi89++LX4X//d0K4vfRz+aVX/OP9cffotd/0cdDhFQ7AC3eAIBKtjVN9EdCKrN+BABI1AC0eONye26+vvo3o8dDZ+qPxjQBr2P9CDCx6PVf9PFwsxQNQIs3AKAyTcDzWT8CAPwxfQPQ4o1bbPm15wnPR7k4ejx0pv5gWRZNwDNZPwJMLnr9F308BJjmR0A+YvEGAHSy/TDIuq7WPwdZPwK38tAGSGLaHYAWb9zuq6c4e57ejB4Pnak/+M1OwOOsHwESiV7/RR8PN5tyB6DFGwDQmZ2Ar7N+BEI8m3M8zAEmMd0OQIs3Qn30FOeVpzejx0Nn6g/+YifgftaPAElFr/+ij4cbTdUAtHgDAPhDE/A560cAgOem+QiwxRvTeDx+PrV5e7P1Sl6OHg+dqT/4j7cfB35vtDFY5XjrR4DEotd/0cfDTaZpAC6LxRsAwEe2JuBH//8V7/9G9uOP/A0AgI7WZVmmWDVZvBEh+iNV8n6M+OUmfvC66LqZjTrOKTqP5U1v8i9W9PhH6x7/7qLzf4rvAFQEAADPWTP9YSwAAPb79mNw7fTPYAPT4g0AYL/PPg7cifUjANDNaP9uih2AAAAAAMA1zvsRkP97sRX5v72fXDOXf1/8Kszvy7svMf/9H0/+zq8dG7L/XOKXm/jBcdH1E/365CZ/iPRYlv2/1Lqu/8mf0eO7i67/6NenuaD+mR2AAAAAAFCYBiAAAAD97Pk+1a/+zejxADfSAAQAAACAwjQAAQAA6GX77rY9O/w++p630eMBbnbej4AAtvhnJ365iR8cF10/0a9PbvIH+oqu/+jXhxfYAQgAAEA/X+3i27N7b/R4gBvZAQhn2vkz8ExK/HITPzguun6iX5/c5A/0FV3/0a8PL7ADEAAAgJ4+2sX3yu690eMBbqIBCAAAAACFaQACAADQ19tdfEd2740eD3ADDUAAAAAAKGz98ViGHk38M/idlg9PRgi0Bn8pq/wfI365iR8cF10/0dRvbtH5K396k3+xosc/Wvf4dzea/z8G08cOQAAAAAAoTAMQAAAAAArTAAQAAACAwr6d9Ycey7L/l47Wden9yX9mM5q/8j+W+OUmfnBcdP1EH09u8odI/774Vfjf32XQ76Of5eCv7/ySf3+Lrv/o4+ktKn/sAAQAAACAws5tAO75RZPmv/rDxEbzV/7HEr/cxA+Oi66f6OPJTf5AX9H1H308vQXkjx2AAAAAAFDYeQ3A7fPLezrkez/rDHcZzV/5H0v8chM/OC66fqKPJzf5A31F13/08fQWlD+n/QgIAAAAtOIjnkAS534E+Ksupu43sxvNX/kfS/xyEz84Lrp+oo8nN/kDfUXXf/Tx9BaQP3YAAgAAwBHPbtDtEAQmcf6PgHzUxdT9JovR/JX/scQvN/GD46LrJ/p4cpM/0Fd0/UcfT283549fAQYAAACAwq5pAL7tYup+k81o/sr/WOKXm/jBcdH1E308uckf6Cu6/qOPp7cb88cOQAAAAAAobP3xWIZai/8MfqfpQ2ecQGvwl/LK/zHil5v4wXHR9RNN/eYWnb/ypzf5Fyt6/KN1j393o/n/YzB97AAEAAAAgMI0AAEAAACgMA1AAAAAACjs21l/6LEs+3+pZF2X3p/8Zzb/vvhVmN/fZfDvo5/VwK/P/Mv/c4lfbuIHx0XXT/Trk5v8IdLo/av73zHR9R/9+vQWNX/YAQgAAAAAhZ3bANzziybNf/UHAACACYzev7r/BY4KmD/sAAQAAACAws5rAG6fX97zhGTvZ50BAADgbKP3r+5/gaOC5o/TfgQEWGzxz078chM/OC66fqJfn9zkD/QVXf/Rrw8vOPcjwF91MT39AAAAYBaj96/uf4GjAuYPOwDhTDt/Bp5JiV9u4gfHRddP9OuTm/yBvqLrP/r14QXn/wjIR11MTz8AAACYzej9q/tf4Kib5w+/AgwAAAAAhV3TAHzbxfT0AwAAgFmN3r+6/wWOunH+sAMQAAAAAApbfzyWodbiP4PfafnwZIRAa/CXssr/MeKXm/jBcdH1E0395hadv/KnN/kXK3r8o3WPf3ej+f9jMH3sAAQAAACAwr5Fn0BmZz298BQAAAAAqEr/JJ4dgAeduXW5+zZoAAAAoCb9kzmctgPwsSz7f6lkXZfMIfso4V7tQr//G+u66mQHGs3fTvk/I/HLTfzguOj6iT6e3OQP9BVd/9HHd6J/8l9R+eMjwC/6rNt8Rhc6exIDAAAALIv+yWzObQCu6/MuZuLtmluSPh6P/yTsaAd7+5uSONBo/hbP/+mJX27iB8dF10/08eQmf6Cv6PqPPr44/ZMnAvLHdwDu9DZ5r7L9bZ9pBwAAADLSP5nTeQ3ALbDPOuRv/20SdyTvRhIHGc3fwvmfgvjlJn5wXHT9RB9PbvIH+oqu/+jjC9M/2SEof+wAfOLO5N2kTWIAAACgJf2TuZ3bAPyqi5mw+x2RvBtJHGA0f4vlfzril5v4wXHR9RN9PLnJH+gruv6jjy9G/+RFAfljB+AnIpN3kzKJAQAAgDb0T3I4vwH4URczWfd7huTdSOKbjeZvgfxPTfxyEz84Lrp+oo8nN/kDfUXXf/TxBeifDLg5f+wAfGem5N2kS2IAAACgNP2TXK5pAL7tYibqfs+YvBtJfKPR/E2a/2WIX27iB8dF10/08eQmf6Cv6PqPPj4p/ZOT3Jg/dgD+MnPyblIlMQAAAFCO/klO64/HMhSxfwbHcoaEyZoQM4xddtGxF8Mx4peb+MFx0fUTTf3mFp2/8qc3+RcrevyjVYh/1hjOMPajY/dj8C203wGYNXmXJfe5AwAAAHlk7kFkPvezfBvdwTdKEMaMjt8MXXA4Yoa5Y11XNQQAADtZO8ea4R4qUvb3P9q/a78DEMhnpol7pnMBAACAj3yLPgGAV3zUcHv1SeL7vzF6vJ2AQJR/X/wq5+/Lu/nv9388+Tu/5r33M3D06wMAsI8GIJDGZ7vtRnfhnbGLTxMQAACAWWkAAim8/an5s3fwnXH8uq6agAAAAEzJdwAC03vb/JvVdm6+ExAAAIDZaAACU8vQ/NtoAgIAADAjHwEGppWp+bfxcWAgpegHF9GvDwBQnB2AwJQyNv82dgICAAAwEzsAgelkbv5t7AQEUnk2T139QCP69QEAirMDEJhKhebfxk5AAAAAZqABCEyjUvNvowkIAABANA1AYAoVm38bTUAAAAAiaQAC4So3/zaagAAAAERZl2Wpe8fNU5UbLntEN2O6j/+yxMcgitjHx14MyCy6fqKp39yi81f+9Cb/6Cw6/4llByAQpvMFqPN7BwAA4F4agM11bkLM8N5nOAeIMEPuz3AOANzP/N/XDLGf4RzoSe6hAUjLiWCm9zzTucAdZsr5mc4FgPuY//uZKeYznQs9yDmWZVm+RZ8Ac1jXtc33Ucw4+XUaf3pTf3Cux7Isy978XdflfQVmPx5GmP/7sP6gsxnznxh2APJbh4lh5vc487nBGWbO8ZnPDYDrmP/rmznGM58bNcgx3tIAZFmWP79GVXmC2N7bjE/aOow/vak/uNCe3P3q32Q/Hg4y/9dn/UFnM+c/MTQA+a3yRSjD5Fd5/OlN/QEwK/N/XdYfdJYh/7mfBiB/qXgRyjT5VRx/elN/cLGttvbssPuoDrMfDycw/9dj/UFnmfKfe2kA8h+VLkIZJ79K409v6g+ALMz/dVh/0FnG/Oc+GoB8qMJFKPPkV2H86U39wY2+2kW3Z/dc9uPhJOb//Kw/6Cxz/nMPDUA+lfkiVGHyyzz+9Kb+AMjK/J+X9QedVch/rqcByJcyXoQqTX4Zx5/e1B8E+WgX3Su757IfDycy/+dj/UFnlfKfa2kA8lSmi1DFyS/T+NOb+gOgCvN/HtYfdFYx/7mOBiC7ZLgIVZ78Mow/vak/mMDbXXRHds9lPx5OZv6fn/UHnVXOf66hAchuM1+EOkx+M48/vak/AKoy/8/L+oPOOuQ/51uXZZExHDLLZNP1gjjL+I/oGruNGOZVIXbk1rX2Nmowt8z5K/fmkDmHRsg/lqVv/jPODkAOm2HimeEconR+78yhcw52fu8AnZn/43WOQef3zk9ygBHfok8AoCsX8NzWdfUknkPU/rgzxlD9xqiQ//IPgIzsAASAgyrcyHIvOTMPsbifMf/DWABwNzsAAWCAnYDs9dEN/6u58/5vjB4fLfr9q9/7zJZ7Z5B/AGTiR0AYEr1oqbiYfEX0+I/qHj9qyV6PXMt8Nzf1ey35/zX595ru+SRfeuue/4yxAxAABjwej2VdVzs5+NS2WN9y5a3oHXDRot+/+r3eV/mfnfwDIBPfAQgAg7Ybt2o3t4x72/xgTur3OvL/OfkHwF00AAHgBG7ieE/zIw/1ez75v5/8A+AOGoAAcBI3cWw0P/JRv+eR/6+TfwBcTQMQAE7kJg7Nj7zU7zj5f5z8A+BKGoAAcDI3cX1pfuSnfo+T/+PkHwBX0QAEgAu4ietH86MO9fs6+X8e+QfAFTQAAeAibuL60PyoR/3uJ//PJ/8AOJsGIABcyE1cfZofdanf5+T/deQfAGfSAASAi7mJq0vzoz71+zn5fz35B8BZ1mVZXLE5LHrB130xFD3+o7rHj76y1y4/mcN6Ur8/yf8Y8k/uyYHeuuc/Y+wABICbWbzlJ4Z9ib0xiGTsATjKDkAAAAAAKMwOQAAAAAAoTAMQAAAAAArTAAQAAACAwjQAAQAAAKAwDUAAAAAAKEwDEAAAAAAK0wAEAAAAgMI0AAEAAACgMA1AAAAAAChMAxAAAAAACtMABAAAAIDCNAABAAAAoDANQAAAAAAoTAMQAAAAAArTAAQAAACAwjQAAQAAAKAwDUAAAAAAKEwDEAAAAAAK0wAEAAAAgMI0AAEAAACgMA1AAAAAAChMAxAAAAAACtMABAAAAIDCNAABAAAAoDANQAAAAAAoTAMQAAAAAArTAAQAAACAwjQAAQAAAKAwDUAAAAAAKEwDEAAAAAAK0wAEAAAAgMI0AAEAAACgMA1AAAAAAChMAxAAAAAACtMABAAAAIDCNAABAAAAoDANQAAAAAAoTAMQAAAAAArTAAQAAACAwjQAAQAAAKAwDUAAAAAAKEwDEAAAAAAK0wAEAAAAgMI0AAEAAACgMA1AAAAAAChMAxAAAAAACtMABAAAAIDC/h8p0YoBvFpxRwAAAABJRU5ErkJggg==\n",
      "text/plain": [
       "<PIL.Image.Image image mode=RGBA size=1280x256 at 0x7F557791BB20>"
      ]
     },
     "execution_count": 12,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "room_viz.room_viz(level, \"test.png\", \"encoding/room_tiles\")"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 21,
   "metadata": {},
   "outputs": [],
   "source": [
    "from world_rando import room_dtypes"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 22,
   "metadata": {},
   "outputs": [
    {
     "ename": "AttributeError",
     "evalue": "module 'world_rando.room_dtypes' has no attribute 'mk_default_solid'",
     "output_type": "error",
     "traceback": [
      "\u001b[0;31m---------------------------------------------------------------------------\u001b[0m",
      "\u001b[0;31mAttributeError\u001b[0m                            Traceback (most recent call last)",
      "\u001b[0;32m<ipython-input-22-3736943316c4>\u001b[0m in \u001b[0;36m<module>\u001b[0;34m()\u001b[0m\n\u001b[1;32m      4\u001b[0m     \u001b[0;32mfor\u001b[0m \u001b[0my\u001b[0m \u001b[0;32min\u001b[0m \u001b[0;34m[\u001b[0m\u001b[0;36m0\u001b[0m\u001b[0;34m,\u001b[0m\u001b[0;36m1\u001b[0m\u001b[0;34m,\u001b[0m\u001b[0;36m14\u001b[0m\u001b[0;34m,\u001b[0m\u001b[0;36m15\u001b[0m\u001b[0;34m]\u001b[0m\u001b[0;34m:\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[1;32m      5\u001b[0m         \u001b[0mc\u001b[0m \u001b[0;34m=\u001b[0m \u001b[0mCoord\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0mx\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0my\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[0;32m----> 6\u001b[0;31m         \u001b[0mlvl\u001b[0m\u001b[0;34m[\u001b[0m\u001b[0mc\u001b[0m\u001b[0;34m]\u001b[0m \u001b[0;34m=\u001b[0m \u001b[0mroom_dtypes\u001b[0m\u001b[0;34m.\u001b[0m\u001b[0mmk_default_solid\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[0m\u001b[1;32m      7\u001b[0m \u001b[0;31m# sides\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[1;32m      8\u001b[0m \u001b[0;32mfor\u001b[0m \u001b[0my\u001b[0m \u001b[0;32min\u001b[0m \u001b[0mrange\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0;36m0\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mlvl\u001b[0m\u001b[0;34m.\u001b[0m\u001b[0mdimensions\u001b[0m\u001b[0;34m.\u001b[0m\u001b[0my\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m:\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n",
      "\u001b[0;31mAttributeError\u001b[0m: module 'world_rando.room_dtypes' has no attribute 'mk_default_solid'"
     ]
    }
   ],
   "source": [
    "lvl = room_dtypes.Level(Coord(80, 16))\n",
    "# top and bottom\n",
    "for x in range(0, lvl.dimensions.x):\n",
    "    for y in [0,1,14,15]:\n",
    "        c = Coord(x, y)\n",
    "        lvl[c] = room_dtypes.mk_default_solid()\n",
    "# sides\n",
    "for y in range(0, lvl.dimensions.y):\n",
    "    for x in [0,1,78,79]:\n",
    "        c = Coord(x, y)\n",
    "        lvl[c] = room_dtypes.mk_default_solid()\n",
    "# fill the interior\n",
    "lvl.missing_defaults(room_dtypes.mk_default_air)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 23,
   "metadata": {},
   "outputs": [
    {
     "ename": "KeyError",
     "evalue": "(0,0)",
     "output_type": "error",
     "traceback": [
      "\u001b[0;31m---------------------------------------------------------------------------\u001b[0m",
      "\u001b[0;31mKeyError\u001b[0m                                  Traceback (most recent call last)",
      "\u001b[0;32m<ipython-input-23-5a1e1bc14859>\u001b[0m in \u001b[0;36m<module>\u001b[0;34m()\u001b[0m\n\u001b[0;32m----> 1\u001b[0;31m \u001b[0mroom_viz\u001b[0m\u001b[0;34m.\u001b[0m\u001b[0mroom_viz\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0mlvl\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0;34m\"test.png\"\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0;34m\"encoding/room_tiles\"\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[0m",
      "\u001b[0;32m/mnt/d/Ross/Programming/sm door rando/sm_rando/world_rando/room_viz.py\u001b[0m in \u001b[0;36mroom_viz\u001b[0;34m(level, filename, room_dir)\u001b[0m\n\u001b[1;32m    138\u001b[0m     \u001b[0mimages\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mslopes\u001b[0m \u001b[0;34m=\u001b[0m \u001b[0mload_room_tiles\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0mroom_dir\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[1;32m    139\u001b[0m     \u001b[0;32mfor\u001b[0m \u001b[0mc\u001b[0m \u001b[0;32min\u001b[0m \u001b[0mlevel\u001b[0m\u001b[0;34m.\u001b[0m\u001b[0mitercoords\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m:\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[0;32m--> 140\u001b[0;31m         \u001b[0mrtile\u001b[0m \u001b[0;34m=\u001b[0m \u001b[0mlevel\u001b[0m\u001b[0;34m[\u001b[0m\u001b[0mc\u001b[0m\u001b[0;34m]\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[0m\u001b[1;32m    141\u001b[0m         \u001b[0mimg\u001b[0m \u001b[0;34m=\u001b[0m \u001b[0mfind_image\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0mrtile\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mimages\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mslopes\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[1;32m    142\u001b[0m         \u001b[0;32mif\u001b[0m \u001b[0mimg\u001b[0m \u001b[0;32mis\u001b[0m \u001b[0;32mnot\u001b[0m \u001b[0;32mNone\u001b[0m\u001b[0;34m:\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n",
      "\u001b[0;32m/mnt/d/Ross/Programming/sm door rando/sm_rando/world_rando/room_dtypes.py\u001b[0m in \u001b[0;36m__getitem__\u001b[0;34m(self, key)\u001b[0m\n\u001b[1;32m    185\u001b[0m     \u001b[0;31m# Behaves like a dictionary\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[1;32m    186\u001b[0m     \u001b[0;32mdef\u001b[0m \u001b[0m__getitem__\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0mself\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mkey\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m:\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[0;32m--> 187\u001b[0;31m         \u001b[0;32mreturn\u001b[0m \u001b[0mself\u001b[0m\u001b[0;34m.\u001b[0m\u001b[0mtiles\u001b[0m\u001b[0;34m[\u001b[0m\u001b[0mkey\u001b[0m\u001b[0;34m]\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[0m\u001b[1;32m    188\u001b[0m     \u001b[0;32mdef\u001b[0m \u001b[0m__setitem__\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0mself\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mkey\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mvalue\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m:\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[1;32m    189\u001b[0m         \u001b[0;32mif\u001b[0m \u001b[0mself\u001b[0m\u001b[0;34m.\u001b[0m\u001b[0min_bounds\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0mkey\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m:\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n",
      "\u001b[0;31mKeyError\u001b[0m: (0,0)"
     ]
    }
   ],
   "source": [
    "room_viz.room_viz(lvl, \"test.png\", \"encoding/room_tiles\")"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 24,
   "metadata": {},
   "outputs": [],
   "source": [
    "# Get the compressed bytes representation\n",
    "new_data = lvl.to_bytes()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 25,
   "metadata": {
    "scrolled": true
   },
   "outputs": [],
   "source": [
    "new_data_compressed = compress.greedy_compress(new_data)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 26,
   "metadata": {},
   "outputs": [],
   "source": [
    "ndd = decompress.decompress(new_data_compressed)\n",
    "assert ndd == new_data"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 28,
   "metadata": {},
   "outputs": [],
   "source": [
    "# Write it to the ROM\n",
    "rom_m.write_to_new(level_addr, new_data_compressed + b\"\\x00\\x00\")"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 31,
   "metadata": {},
   "outputs": [],
   "source": [
    "rom_m.save_rom()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": []
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python3.9",
   "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.9.2"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 2
}