adventofcode/2019/intcode-processor/processor.go

package intcodeprocessor

import (
	"errors"
	"fmt"
	"math"
)

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
)

const (
	MODE_POS = iota
	MODE_IMM
)

const (
	RET_ERR = iota - 1
	RET_OK
	RET_DONE
)

type Program struct {
	code []int
	ptr  int

	state int
	error error

	waitingForInput  bool
	input            chan int
	waitingForOutput bool
	output           chan int
}

func NewProgram(prog []int) *Program {
	p := new(Program)
	p.code = make([]int, len(prog))
	p.input = make(chan int)
	p.output = make(chan int)
	copy(p.code, prog)
	return p
}

func (p *Program) GetCode() []int {
	return p.code
}

func (p *Program) State() int {
	return p.state
}

func (p *Program) Error() error {
	return p.error
}

func (p *Program) Run() int {
	for {
		p.state = p.Step()
		if p.state != RET_OK {
			return p.state
		}
	}
}

func (p *Program) Step() int {
	if len(p.code) < p.ptr {
		p.error = errors.New("Pointer Exception")
		return RET_ERR
	}
	intcode := p.readNext()
	p.ptr++
	switch p.opCode(intcode) {
	case OP_ADD:
		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:
		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
	}
	return RET_ERR
}

func (p *Program) GetProgramValueAt(idx int) int {
	return p.code[idx]
}

func (p *Program) SetProgramValueAt(idx, val int) {
	p.code[idx] = val
}

func (p *Program) NeedsInput() bool {
	return p.waitingForInput
}

func (p *Program) NeedsOutput() bool {
	return p.waitingForOutput
}

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")
	}
	return p.code[p.ptr]
}

func (p *Program) readNextTwo() (int, int) {
	return p.code[p.ptr], p.code[p.ptr+1]
}

func (p *Program) readNextThree() (int, int, int) {
	return p.code[p.ptr], p.code[p.ptr+1], p.code[p.ptr+2]
}

func (p *Program) get(mode, v int) int {
	if mode == MODE_POS {
		return p.code[v]
	}
	return v
}

func (p *Program) Input(v int) {
	p.input <- v
	p.waitingForInput = false
}

func (p *Program) Output() int {
	v := <-p.output
	p.waitingForOutput = false
	return v
}

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.waitingForInput = true
	p.code[dest] = <-p.input
	p.waitingForInput = false
}

func (p *Program) opOut(intcode, val int) {
	valmd := p.paramMode(intcode, 0)
	ret := p.get(valmd, val)
	p.waitingForOutput = true
	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 {
	var ret string
	for k := range p.code {
		if k == p.ptr {
			ret = fmt.Sprintf("%s [%d]", ret, p.code[k])
		} else {
			ret = fmt.Sprintf("%s %d", ret, p.code[k])
		}
	}
	return ret
}