Day 14 & 15 progress

2019/day14/main.go

@@ -0,0 +1,153 @@
+package main
+
+import (
+	"fmt"
+	"strings"
+
+	helpers "git.bullercodeworks.com/brian/adventofcode/helpers"
+)
+
+func main() {
+	inp := helpers.StdinToStringSlice()
+	part1(inp)
+	part2(inp)
+}
+
+var allReactions map[string]*Reaction
+
+func part1(inp []string) {
+	allReactions = make(map[string]*Reaction)
+	for _, v := range inp {
+		r := NewReaction(v)
+		allReactions[r.out] = r
+	}
+	factory := NewFactory(allReactions)
+	factory.setInfinite("ORE", true)
+	factory.generate("FUEL")
+	fmt.Println("# Part 1")
+	fmt.Println("Ore Used:", factory.generated["ORE"])
+}
+
+func part2(inp []string) {
+	allReactions = make(map[string]*Reaction)
+	for _, v := range inp {
+		r := NewReaction(v)
+		allReactions[r.out] = r
+	}
+	factory := NewFactory(allReactions)
+	oreStart := 1000000000000
+	factory.setStock("ORE", oreStart)
+	fmt.Println("\n# Part 2")
+	fmt.Println("Ore in stock", factory.stock["ORE"])
+	var out1 string
+	for factory.generate("FUEL") {
+		clearStr := strings.Repeat("", len(out1))
+		out1 = fmt.Sprintf("%s%3d%% (%d)", clearStr, ((factory.stock["ORE"] * 100) / oreStart), factory.generated["FUEL"])
+		fmt.Print(out1)
+	}
+}
+
+type Factory struct {
+	stock     map[string]int
+	recipes   map[string]*Reaction
+	generated map[string]int
+	used      map[string]int
+	infinite  map[string]bool
+}
+
+func NewFactory(recipes map[string]*Reaction) *Factory {
+	f := Factory{
+		stock:     make(map[string]int),
+		recipes:   recipes,
+		generated: make(map[string]int),
+		used:      make(map[string]int),
+		infinite:  make(map[string]bool),
+	}
+	return &f
+}
+
+func (f *Factory) setStock(chem string, amt int) {
+	f.stock[chem] = amt
+}
+
+func (f *Factory) setInfinite(chem string, is bool) {
+	f.infinite[chem] = is
+}
+
+func (f *Factory) count(chem string) int {
+	return f.stock[chem]
+}
+
+func (f *Factory) use(chem string, amt int) bool {
+	if f.count(chem) < amt {
+		return false
+	}
+	f.stock[chem] = f.stock[chem] - amt
+	f.used[chem] = f.used[chem] + amt
+	return true
+}
+
+func (f *Factory) generate(chem string) bool {
+	if f.infinite[chem] {
+		f.stock[chem] = f.stock[chem] + 1
+		f.generated[chem] = f.generated[chem] + 1
+		return true
+	}
+	react, ok := f.recipes[chem]
+	if !ok {
+		return false
+	}
+	for k, v := range react.in {
+		for f.count(k) < v {
+			f.generate(k)
+		}
+		f.use(k, v)
+	}
+	f.stock[chem] = f.stock[chem] + react.outQty
+	f.generated[chem] = f.generated[chem] + react.outQty
+	return true
+}
+
+type Reaction struct {
+	in     map[string]int
+	out    string
+	outQty int
+
+	ingredients []*Reaction
+}
+
+func NewReaction(inp string) *Reaction {
+	pts := strings.Split(inp, " => ")
+	if len(pts) != 2 {
+		panic("Invalid Reaction Input: " + inp)
+	}
+	ret := Reaction{
+		in: make(map[string]int),
+	}
+	inps := strings.Split(pts[0], ", ")
+	for k := range inps {
+		var qty int
+		var prod string
+		_, err := fmt.Sscanf(inps[k], "%d %s", &qty, &prod)
+		if err != nil {
+			panic("Invalid Reaction Input: " + inp)
+		}
+		ret.in[prod] = qty
+	}
+	_, err := fmt.Sscanf(pts[1], "%d %s", &ret.outQty, &ret.out)
+	if err != nil {
+		panic("Invalid Reaction Input: " + inp)
+	}
+	return &ret
+}
+
+func (r Reaction) String() string {
+	var ret string
+	for k, v := range r.in {
+		if len(ret) > 0 {
+			ret = ret + ","
+		}
+		ret = fmt.Sprintf("%s %d %s", ret, v, k)
+	}
+	return fmt.Sprintf("%s => %d %s", ret, r.outQty, r.out)
+}