adventofcode/2019/day18/vault.go

76 lines
1.5 KiB
Go
Raw Normal View History

2019-12-18 14:59:20 +00:00
package main
import (
"bytes"
"errors"
"fmt"
helpers "git.bullercodeworks.com/brian/adventofcode/helpers"
)
type Vault struct {
maxX, maxY int
vault map[helpers.Coordinate]byte
keys map[helpers.Coordinate]byte
doors map[helpers.Coordinate]byte
}
func NewVault(file string) *Vault {
inp := helpers.FileToBytes(file)
wrk := bytes.Split(inp, []byte{'\n'})
v := Vault{
vault: make(map[helpers.Coordinate]byte),
keys: make(map[helpers.Coordinate]byte),
doors: make(map[helpers.Coordinate]byte),
}
v.maxY = len(wrk)
v.maxX = len(wrk[0])
for y, yv := range wrk {
for x, xv := range yv {
v.vault[*helpers.NewCoordinate(x, y)] = xv
if isKey(xv) {
v.keys[*helpers.NewCoordinate(x, y)] = xv
} else if isDoor(xv) {
v.doors[*helpers.NewCoordinate(x, y)] = xv
}
}
}
return &v
}
func (v *Vault) findKey(door byte) (helpers.Coordinate, error) {
var r helpers.Coordinate
if !isDoor(door) {
return r, errors.New("Invalid Door")
}
for k := range v.keys {
if v.keys[k] == doorToKey(door) {
return k, nil
}
}
return r, errors.New("No key for that door")
}
func (v *Vault) findDoor(key byte) (helpers.Coordinate, error) {
var r helpers.Coordinate
if !isKey(key) {
return r, errors.New("Invalid Key")
}
for k := range v.doors {
if v.doors[k] == keyToDoor(key) {
return k, nil
}
}
return r, errors.New("No door for that key")
}
func (v *Vault) Print() {
for y := 0; y < v.maxY; y++ {
for x := 0; x < v.maxX; x++ {
fmt.Print(string(v.vault[*helpers.NewCoordinate(x, y)]))
}
fmt.Println()
}
}