day7.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "def parseInput():\n",
    "    filepath = 'day7_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"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "def ampProgram(phaseSetting, inputSignal, allInts):\n",
    "    phaseSet = False\n",
    "    remip = 0\n",
    "    skipInc = False\n",
    "    ip = 0\n",
    "    \n",
    "    lastOut = 0\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,startint,opcode,parammode)\n",
    "\n",
    "        if opcode == 99:\n",
    "            return lastOut\n",
    "        \n",
    "        targetIndex = ip+1 if parammode[0] == \"1\" else allInts[ip+1]\n",
    "        if opcode == 3:\n",
    "            if not phaseSet:\n",
    "                allInts[targetIndex] = phaseSetting\n",
    "                phaseSet = True\n",
    "            else:\n",
    "                allInts[targetIndex] = inputSignal\n",
    "            remip = 2\n",
    "        elif opcode == 4:\n",
    "            lastOut = allInts[targetIndex]\n",
    "            print(\"signal to next amp:\", lastOut)\n",
    "            remip = 2\n",
    "\n",
    "        else:\n",
    "            param1 = allInts[ip+1]\n",
    "            if parammode[0] == \"0\":\n",
    "                param1 = allInts[param1]\n",
    "\n",
    "            param2 = allInts[ip+2]\n",
    "            if parammode[1] == \"0\":\n",
    "                param2 = allInts[param2] \n",
    "\n",
    "            targetIndex = allInts[ip+3]\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",
    "                    remip = 1\n",
    "            elif opcode == 6:\n",
    "                if param1 == 0:\n",
    "                    ip = param2\n",
    "                    skipInc = True\n",
    "                else:\n",
    "                    remip = 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 += 4 -remip\n",
    "            remip = 0\n",
    "        else:\n",
    "            skipInc = False\n",
    "            \n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "\n",
    "# allInts = [3,31,3,32,1002,32,10,32,1001,31,-2,31,1007,31,0,33,1002,33,7,33,1,33,31,31,1,32,31,31,4,31,99,0,0,0]\n",
    "# phases = [0,0,4,3,2]\n",
    "\n",
    "allInts = parseInput()\n",
    "allIntsCopy = list(allInts)\n",
    "phases = [-1,1,2,3,4]\n",
    "largestOut = 0\n",
    "\n",
    "for allSteps in range(5**5):\n",
    "    carry = True\n",
    "    nextIn = 0\n",
    "    \n",
    "    for curAmp in range(5):\n",
    "        if carry:\n",
    "            carry = False\n",
    "            phases[curAmp] += 1\n",
    "            if phases[curAmp] == 5:\n",
    "                phases[curAmp] = 0\n",
    "                carry = True\n",
    "        \n",
    "        allInts = list(allIntsCopy)\n",
    "        nextIn = ampProgram(phases[curAmp], nextIn, allInts)\n",
    "        \n",
    "    if nextIn > largestOut:\n",
    "        hasDouble = False\n",
    "        for i in range(5):\n",
    "            if i not in phases:\n",
    "                hasDouble = True\n",
    "                break\n",
    "        if not hasDouble:\n",
    "            largestOut = nextIn\n",
    "            print(\"***\",nextIn)\n",
    "    \n",
    "print(largestOut)\n",
    "    \n",
    "    \n"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "### Part 2"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "def ampStepProgram(phaseSetting, inputSignal, allInts):\n",
    "    phaseSet = True if phaseSetting < 0 else False\n",
    "    remip = 0\n",
    "    skipInc = False\n",
    "    ip = inputSignal[1]\n",
    "    \n",
    "    lastOut = 0\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,startint,opcode,parammode)\n",
    "\n",
    "        if opcode == 99:\n",
    "            return[lastOut,-1,allInts]\n",
    "        \n",
    "        targetIndex = ip+1 if parammode[0] == \"1\" else allInts[ip+1]\n",
    "        if opcode == 3:\n",
    "            if not phaseSet:\n",
    "                allInts[targetIndex] = phaseSetting\n",
    "                phaseSet = True\n",
    "            else:\n",
    "                allInts[targetIndex] = inputSignal[0]\n",
    "            remip = 2\n",
    "        elif opcode == 4:\n",
    "            lastOut = allInts[targetIndex]\n",
    "            return[lastOut,ip+2,allInts]\n",
    "            remip = 2\n",
    "\n",
    "        else:\n",
    "            param1 = allInts[ip+1]\n",
    "            if parammode[0] == \"0\":\n",
    "                param1 = allInts[param1]\n",
    "\n",
    "            param2 = allInts[ip+2]\n",
    "            if parammode[1] == \"0\":\n",
    "                param2 = allInts[param2] \n",
    "\n",
    "            targetIndex = allInts[ip+3]\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",
    "                    remip = 1\n",
    "            elif opcode == 6:\n",
    "                if param1 == 0:\n",
    "                    ip = param2\n",
    "                    skipInc = True\n",
    "                else:\n",
    "                    remip = 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 += 4 -remip\n",
    "            remip = 0\n",
    "        else:\n",
    "            skipInc = False\n",
    "            \n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "\n",
    "# allInts = [3,52,1001,52,-5,52,3,53,1,52,56,54,1007,54,5,55,1005,55,26,1001,54,-5,54,1105,1,12,1,53,54,53,1008,54,0,55,1001,55,1,55,2,53,55,53,4,53,1001,56,-1,56,1005,56,6,99,0,0,0,0,10]\n",
    "\n",
    "allInts = parseInput()\n",
    "allIntsCopy = list(allInts)\n",
    "\n",
    "phases = [4,6,7,8,9]\n",
    "# phases = [8,7,8,5,6]\n",
    "\n",
    "largestOut = 0\n",
    "nextIn = [0,0]\n",
    "\n",
    "for allSteps in range(5**5):\n",
    "    carry = True\n",
    "\n",
    "    # produce next phase sequence\n",
    "    for curAmp in range(5):\n",
    "        if carry:\n",
    "            carry = False\n",
    "            phases[curAmp] += 1\n",
    "            if phases[curAmp] == 10:\n",
    "                phases[curAmp] = 5\n",
    "                carry = True\n",
    "      \n",
    "    # check phase sequence for validity\n",
    "    hasDouble = False\n",
    "    for i in range(5,10):\n",
    "        if i not in phases:\n",
    "            hasDouble = True\n",
    "            break\n",
    "    if hasDouble:\n",
    "        continue\n",
    "\n",
    "    print('next PHASES:',phases)\n",
    "    ips = [0,0,0,0,0]\n",
    "    ints = [list(allIntsCopy),list(allIntsCopy),list(allIntsCopy),list(allIntsCopy),list(allIntsCopy)]\n",
    "    ticks = 0\n",
    "    while ticks < 200:\n",
    "        ticks += 1\n",
    "        print(ips)\n",
    "        for curAmp in range(5):\n",
    "            nextIn[1] = ips[curAmp]\n",
    "            phaseParam = -1 if ips[curAmp] > 0 else phases[curAmp]\n",
    "            nextIn = ampStepProgram(phaseParam, nextIn, ints[curAmp])\n",
    "            if nextIn[0] > largestOut:\n",
    "                largestOut = nextIn[0]\n",
    "                print(\"***\",nextIn[0])\n",
    "            ips[curAmp] = nextIn[1]\n",
    "            ints[curAmp] = nextIn[2]\n",
    "\n",
    "        \n",
    "\n",
    "        if nextIn[1]<0:\n",
    "            break\n",
    "    \n",
    "print(largestOut)\n",
    "    \n",
    "    \n"
   ]
  }
 ],
 "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
}