adventofcode/2019/day13/main.go

200 lines
3.6 KiB
Go
Raw Normal View History

package main
import (
"bufio"
"fmt"
"io/ioutil"
"os"
"strings"
"time"
intcode "git.bullercodeworks.com/brian/adventofcode/2019/intcode-processor"
helpers "git.bullercodeworks.com/brian/adventofcode/helpers"
)
var auto bool
func main() {
2019-12-15 14:57:44 +00:00
progFileName := "input"
if len(os.Args) > 1 && os.Args[1] == "-auto" {
auto = true
}
prog := readIntCodeFile(progFileName)
play(prog)
}
func play(prog []int) {
p := intcode.NewProgram(prog)
2019-12-13 14:51:59 +00:00
// For part 1, comment out this line
p.SetProgramValueAt(0, 2)
//p.EnableDebug()
maxX, maxY := 0, 0
screen := make(map[string]int)
var score int
go func() {
for {
for !p.NeedsOutput() && !p.NeedsInput() {
time.Sleep(1)
}
if p.NeedsOutput() {
// There should be three outputs
x := p.Output()
y := waitForOutput(p)
b := waitForOutput(p)
if x > maxX {
maxX = x
}
if y > maxY {
maxY = y
}
if x == -1 && y == 0 {
score = b
} else {
screen[coord(x, y)] = b
}
}
if p.NeedsInput() {
printScreen(screen, maxX, maxY, score)
if auto {
ballX, _ := findFirst(4, screen)
pddlX, _ := findFirst(3, screen)
if ballX < pddlX {
p.Input(-1)
} else if ballX > pddlX {
p.Input(1)
} else {
p.Input(0)
}
} else {
var gotInput bool
for !gotInput {
fmt.Print("Input (vimlike): ")
reader := bufio.NewReader(os.Stdin)
2019-12-13 14:51:59 +00:00
inp, err := reader.ReadString('\n')
if err != nil {
panic(err)
}
2019-12-13 14:51:59 +00:00
inp = strings.TrimSpace(inp)
switch inp {
case "h":
p.Input(-1)
gotInput = true
2019-12-13 14:51:59 +00:00
case "l":
p.Input(1)
gotInput = true
2019-12-13 14:51:59 +00:00
case "j", "k":
p.Input(0)
gotInput = true
}
}
}
}
switch p.State() {
case intcode.RET_ERR:
panic(p.Error())
case intcode.RET_DONE:
break
}
}
}()
ret := p.Run()
if ret == intcode.RET_DONE {
2019-12-13 14:51:59 +00:00
fmt.Println("DONE:", score)
} else if ret == intcode.RET_ERR {
fmt.Println("ERROR")
}
//part1(screen)
}
2019-12-13 14:51:59 +00:00
func lookForScore(p *intcode.Program, score int) {
var res []int
code := p.GetCode()
for k, v := range code {
if v == score {
res = append(res, k)
}
}
if len(res) == 1 {
fmt.Println("Score Position:", res[0])
os.Exit(0)
}
}
func part1(s map[string]int) {
var cnt int
for _, v := range s {
if v == 2 {
cnt++
}
}
fmt.Println(cnt, "blocks")
}
func printScreen(screen map[string]int, maxX, maxY, score int) {
2019-12-13 14:51:59 +00:00
//fmt.Print(helpers.CLEAR_SCREEN)
fmt.Println("Score:", score)
for y := 0; y <= maxY; y++ {
for x := 0; x <= maxX; x++ {
v, ok := screen[coord(x, y)]
if !ok {
v = 0
}
switch v {
case 0:
fmt.Print(" ")
case 1:
fmt.Print("▉")
case 2:
fmt.Print("░")
case 3:
fmt.Print("═")
case 4:
fmt.Print("o")
}
}
fmt.Println()
}
}
func findFirst(val int, screen map[string]int) (int, int) {
for k := range screen {
if screen[k] == val {
return revCoord(k)
}
}
return -1, -1
}
// This will block waiting for output
func waitForOutput(p *intcode.Program) int {
for !p.NeedsOutput() {
time.Sleep(1)
}
return p.Output()
}
func coord(x, y int) string {
return fmt.Sprintf("[%d, %d]", x, y)
}
func revCoord(c string) (int, int) {
var x, y int
fmt.Sscanf(c, "[%d, %d]", &x, &y)
return x, y
}
func readIntCodeFile(fn string) []int {
dat, err := ioutil.ReadFile(fn)
if err != nil {
fmt.Println("Error reading program file:", err.Error())
os.Exit(1)
}
var prog []int
stringDat := strings.TrimSpace(string(dat))
for _, v := range strings.Split(stringDat, ",") {
prog = append(prog, helpers.Atoi(v))
}
return prog
}