2019 Day 5 Complete
* Also: Move helpers to their own submodule, for importing
This commit is contained in:
@@ -3,11 +3,23 @@ package intcodeprocessor
|
||||
import (
|
||||
"errors"
|
||||
"fmt"
|
||||
"math"
|
||||
)
|
||||
|
||||
const (
|
||||
MODE_POS = iota
|
||||
MODE_IMM
|
||||
)
|
||||
|
||||
const (
|
||||
OP_ADD = 1
|
||||
OP_MLT = 2
|
||||
OP_INP = 3
|
||||
OP_OUT = 4
|
||||
OP_JIT = 5
|
||||
OP_JIF = 6
|
||||
OP_ILT = 7
|
||||
OP_IEQ = 8
|
||||
OP_EXT = 99
|
||||
)
|
||||
|
||||
@@ -23,11 +35,16 @@ type Program struct {
|
||||
|
||||
state int
|
||||
error error
|
||||
|
||||
input chan int
|
||||
output chan int
|
||||
}
|
||||
|
||||
func NewProgram(prog []int) *Program {
|
||||
p := new(Program)
|
||||
p.code = make([]int, len(prog))
|
||||
p.input = make(chan int, 2)
|
||||
p.output = make(chan int, 2)
|
||||
copy(p.code, prog)
|
||||
return p
|
||||
}
|
||||
@@ -58,19 +75,74 @@ func (p *Program) Step() int {
|
||||
p.error = errors.New("Pointer Exception")
|
||||
return RET_ERR
|
||||
}
|
||||
switch p.readNext() {
|
||||
intcode := p.readNext()
|
||||
p.ptr++
|
||||
switch p.opCode(intcode) {
|
||||
case OP_ADD:
|
||||
p.add(p.readNextThree())
|
||||
v1, v2, v3 := p.readNextThree()
|
||||
p.ptr = p.ptr + 3
|
||||
p.opAdd(intcode, v1, v2, v3)
|
||||
if p.error != nil {
|
||||
return RET_ERR
|
||||
}
|
||||
return RET_OK
|
||||
case OP_MLT:
|
||||
p.mult(p.readNextThree())
|
||||
v1, v2, v3 := p.readNextThree()
|
||||
p.ptr = p.ptr + 3
|
||||
p.opMult(intcode, v1, v2, v3)
|
||||
if p.error != nil {
|
||||
return RET_ERR
|
||||
}
|
||||
return RET_OK
|
||||
case OP_INP:
|
||||
v1 := p.readNext()
|
||||
p.ptr++
|
||||
p.opInp(intcode, v1)
|
||||
if p.error != nil {
|
||||
return RET_ERR
|
||||
}
|
||||
return RET_OK
|
||||
case OP_OUT:
|
||||
v1 := p.readNext()
|
||||
p.ptr++
|
||||
p.opOut(intcode, v1)
|
||||
if p.error != nil {
|
||||
return RET_ERR
|
||||
}
|
||||
return RET_OK
|
||||
case OP_JIT:
|
||||
v1, v2 := p.readNextTwo()
|
||||
p.ptr = p.ptr + 2
|
||||
p.opJumpIfTrue(intcode, v1, v2)
|
||||
if p.error != nil {
|
||||
return RET_ERR
|
||||
}
|
||||
return RET_OK
|
||||
case OP_JIF:
|
||||
v1, v2 := p.readNextTwo()
|
||||
p.ptr = p.ptr + 2
|
||||
p.opJumpIfFalse(intcode, v1, v2)
|
||||
if p.error != nil {
|
||||
return RET_ERR
|
||||
}
|
||||
return RET_OK
|
||||
case OP_ILT:
|
||||
v1, v2, dest := p.readNextThree()
|
||||
p.ptr = p.ptr + 3
|
||||
p.opIfLessThan(intcode, v1, v2, dest)
|
||||
if p.error != nil {
|
||||
return RET_ERR
|
||||
}
|
||||
return RET_OK
|
||||
case OP_IEQ:
|
||||
v1, v2, dest := p.readNextThree()
|
||||
p.ptr = p.ptr + 3
|
||||
p.opIfEqual(intcode, v1, v2, dest)
|
||||
if p.error != nil {
|
||||
return RET_ERR
|
||||
}
|
||||
return RET_OK
|
||||
|
||||
case OP_EXT:
|
||||
return RET_DONE
|
||||
}
|
||||
@@ -85,28 +157,116 @@ func (p *Program) SetProgramValueAt(idx, val int) {
|
||||
p.code[idx] = val
|
||||
}
|
||||
|
||||
func (p *Program) opCode(intcode int) int {
|
||||
return intcode % 100
|
||||
}
|
||||
|
||||
func (p *Program) paramMode(intcode, pNum int) int {
|
||||
plc := math.Pow10(pNum + 2)
|
||||
return ((intcode - p.opCode(intcode)) / int(plc)) % 10
|
||||
}
|
||||
|
||||
func (p *Program) readNext() int {
|
||||
if len(p.code) <= p.ptr {
|
||||
p.error = errors.New("Pointer Exception")
|
||||
}
|
||||
p.ptr++
|
||||
return p.code[p.ptr-1]
|
||||
return p.code[p.ptr]
|
||||
}
|
||||
|
||||
func (p *Program) readNextTwo() (int, int) {
|
||||
return p.readNext(), p.readNext()
|
||||
return p.code[p.ptr], p.code[p.ptr+1]
|
||||
}
|
||||
|
||||
func (p *Program) readNextThree() (int, int, int) {
|
||||
return p.readNext(), p.readNext(), p.readNext()
|
||||
return p.code[p.ptr], p.code[p.ptr+1], p.code[p.ptr+2]
|
||||
}
|
||||
|
||||
func (p *Program) add(a1, a2, dest int) {
|
||||
p.code[dest] = p.code[a1] + p.code[a2]
|
||||
func (p *Program) get(mode, v int) int {
|
||||
if mode == MODE_POS {
|
||||
return p.code[v]
|
||||
}
|
||||
return v
|
||||
}
|
||||
|
||||
func (p *Program) mult(a1, a2, dest int) {
|
||||
p.code[dest] = p.code[a1] * p.code[a2]
|
||||
func (p *Program) Input() chan int {
|
||||
return p.input
|
||||
}
|
||||
|
||||
func (p *Program) Output() chan int {
|
||||
return p.output
|
||||
}
|
||||
|
||||
func (p *Program) opAdd(intcode, a1, a2, dest int) {
|
||||
a1md, a2md, destmd := p.paramMode(intcode, 0), p.paramMode(intcode, 1), p.paramMode(intcode, 2)
|
||||
if destmd != MODE_POS {
|
||||
p.error = errors.New("Invalid Destination Mode")
|
||||
p.state = RET_ERR
|
||||
}
|
||||
p.code[dest] = p.get(a1md, a1) + p.get(a2md, a2)
|
||||
}
|
||||
|
||||
func (p *Program) opMult(intcode, a1, a2, dest int) {
|
||||
a1md, a2md, destmd := p.paramMode(intcode, 0), p.paramMode(intcode, 1), p.paramMode(intcode, 2)
|
||||
if destmd != MODE_POS {
|
||||
p.error = errors.New("Invalid Destination Mode")
|
||||
p.state = RET_ERR
|
||||
}
|
||||
p.code[dest] = p.get(a1md, a1) * p.get(a2md, a2)
|
||||
}
|
||||
|
||||
func (p *Program) opInp(intcode, dest int) {
|
||||
destmd := p.paramMode(intcode, 0)
|
||||
if destmd != MODE_POS {
|
||||
p.error = errors.New("Invalid Destination Mode")
|
||||
p.state = RET_ERR
|
||||
}
|
||||
p.code[dest] = <-p.input
|
||||
}
|
||||
|
||||
func (p *Program) opOut(intcode, val int) {
|
||||
valmd := p.paramMode(intcode, 0)
|
||||
ret := p.get(valmd, val)
|
||||
p.output <- ret
|
||||
}
|
||||
|
||||
func (p *Program) opJumpIfTrue(intcode, v1, v2 int) {
|
||||
v1md, v2md := p.paramMode(intcode, 0), p.paramMode(intcode, 1)
|
||||
if p.get(v1md, v1) != 0 {
|
||||
p.ptr = p.get(v2md, v2)
|
||||
}
|
||||
}
|
||||
|
||||
func (p *Program) opJumpIfFalse(intcode, v1, v2 int) {
|
||||
v1md, v2md := p.paramMode(intcode, 0), p.paramMode(intcode, 1)
|
||||
if p.get(v1md, v1) == 0 {
|
||||
p.ptr = p.get(v2md, v2)
|
||||
}
|
||||
}
|
||||
|
||||
func (p *Program) opIfLessThan(intcode, v1, v2, dest int) {
|
||||
v1md, v2md, destmd := p.paramMode(intcode, 0), p.paramMode(intcode, 1), p.paramMode(intcode, 2)
|
||||
if destmd != MODE_POS {
|
||||
p.error = errors.New("Invalid Destination Mode")
|
||||
p.state = RET_ERR
|
||||
}
|
||||
if p.get(v1md, v1) < p.get(v2md, v2) {
|
||||
p.code[dest] = 1
|
||||
} else {
|
||||
p.code[dest] = 0
|
||||
}
|
||||
}
|
||||
|
||||
func (p *Program) opIfEqual(intcode, v1, v2, dest int) {
|
||||
v1md, v2md, destmd := p.paramMode(intcode, 0), p.paramMode(intcode, 1), p.paramMode(intcode, 2)
|
||||
if destmd != MODE_POS {
|
||||
p.error = errors.New("Invalid Destination Mode")
|
||||
p.state = RET_ERR
|
||||
}
|
||||
if p.get(v1md, v1) == p.get(v2md, v2) {
|
||||
p.code[dest] = 1
|
||||
} else {
|
||||
p.code[dest] = 0
|
||||
}
|
||||
}
|
||||
|
||||
func (p Program) String() string {
|
||||
|
Reference in New Issue
Block a user