day9.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "def parseInput():\n",
    "    filepath = 'day9_input'\n",
    "    allParts = []\n",
    "    with open(filepath) as file:\n",
    "        # data = [line.rstrip() for line in file]\n",
    "        data = file.read().rstrip()\n",
    "        allParts = data.split(',')\n",
    "        allParts = [ int(x) for x in allParts ]\n",
    "    return allParts\n",
    "\n"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "### Part 1+2"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "def boostProgram(inputSignal, allInts):\n",
    "    skipInc = False\n",
    "    ip = 0\n",
    "    ipAdd = 0\n",
    "    rb = 0\n",
    "    lastOut = 0\n",
    "    allInts = { i : allInts[i] for i in range(0, len(allInts) ) }\n",
    "    \n",
    "    while ip < len(allInts):\n",
    "\n",
    "        startint = str(allInts[ip])\n",
    "        parammode = startint.rjust(5,'0')\n",
    "        opcode = int(parammode[-2:])\n",
    "        parammode = parammode[:-2]\n",
    "        # reverse\n",
    "        parammode = parammode[::-1]\n",
    "\n",
    "\n",
    "        # print(\"**\",ip,opcode,'parammodes:',parammode,'relative base:',rb)\n",
    "\n",
    "        ##################### exit opcode\n",
    "        if opcode == 99:\n",
    "            return lastOut\n",
    "        \n",
    "        ##################### one parameter opcodes\n",
    "        ipAdd += 2\n",
    "        targetIndex = allInts[ip+1]\n",
    "        \n",
    "        if parammode[0] == \"2\":\n",
    "            targetIndex += rb\n",
    "            \n",
    "        \n",
    "        ## input\n",
    "        if opcode == 3:\n",
    "            allInts[targetIndex] = inputSignal\n",
    "        ## output\n",
    "        elif opcode == 4:\n",
    "            lastOut = allInts[targetIndex]\n",
    "            \n",
    "            print(\"  OUT  \",lastOut)\n",
    "        ## change relative base\n",
    "        elif opcode == 9:\n",
    "            if parammode[0] == \"1\":\n",
    "                rb += allInts[ip+1]\n",
    "            else:\n",
    "                if targetIndex not in allInts:\n",
    "                    allInts[targetIndex] = 0\n",
    "                rb += allInts[targetIndex]\n",
    "            \n",
    "\n",
    "        ##################### two and three parameter opcodes\n",
    "        else:\n",
    "            ipAdd += 2\n",
    "            \n",
    "            param1 = allInts[ip+1]\n",
    "            if parammode[0] == \"0\":\n",
    "                if param1 not in allInts:\n",
    "                    allInts[param1] = 0\n",
    "                param1 = allInts[param1]\n",
    "            elif parammode[0] == \"2\":\n",
    "                param1rb = param1+rb\n",
    "                if param1rb not in allInts:\n",
    "                    allInts[param1rb] = 0\n",
    "                param1 = allInts[param1rb]\n",
    "\n",
    "            param2 = allInts[ip+2]\n",
    "            if parammode[1] == \"0\":\n",
    "                if param2 not in allInts:\n",
    "                    allInts[param2] = 0\n",
    "                param2 = allInts[param2]\n",
    "            elif parammode[1] == \"2\":\n",
    "                param2rb = param2+rb\n",
    "                if param2rb not in allInts:\n",
    "                    allInts[param2rb] = 0\n",
    "                param2 = allInts[param2rb]\n",
    "\n",
    "            if ip+3 not in allInts:\n",
    "                allInts[ip+3] = 0\n",
    "            targetIndex = allInts[ip+3]\n",
    "            if parammode[2] != \"0\":\n",
    "                targetIndex += rb\n",
    "            \n",
    "            \n",
    "            # print(\"***\",ip,opcode,'parammodes:',parammode,'relative base:',rb,param1,param2,targetIndex)\n",
    "            \n",
    "            if   opcode == 1:\n",
    "                allInts[targetIndex] = param1 + param2\n",
    "            elif opcode == 2:\n",
    "                allInts[targetIndex] = param1 * param2\n",
    "            elif opcode == 5:\n",
    "                if param1 != 0:\n",
    "                    ip = param2\n",
    "                    skipInc = True\n",
    "                else:\n",
    "                    ipAdd -= 1\n",
    "            elif opcode == 6:\n",
    "                if param1 == 0:\n",
    "                    ip = param2\n",
    "                    skipInc = True\n",
    "                else:\n",
    "                    ipAdd -= 1\n",
    "            elif opcode == 7:\n",
    "                allInts[targetIndex] = 1 if param1 < param2 else 0\n",
    "            elif opcode == 8:\n",
    "                allInts[targetIndex] = 1 if param1 == param2 else 0\n",
    "\n",
    "            elif opcode == 0:\n",
    "                print(\"zero\")\n",
    "                break\n",
    "\n",
    "        if not skipInc:\n",
    "            ip += ipAdd\n",
    "        else:\n",
    "            skipInc = False\n",
    "        ipAdd = 0\n",
    "            \n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "allInts = parseInput()\n",
    "# allInts = [109,1,204,-1,1001,100,1,100,1008,100,16,101,1006,101,0,99]\n",
    "\n",
    "boostProgram(2, allInts)"
   ]
  }
 ],
 "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.7.3"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 2
}