X-Git-Url: http://git.scottworley.com/planeteer/blobdiff_plain/330093c150f51922dfd2edeca88aff1668e14979..2f4a9ae81a1a4802acbba08e84f645a5d7b0cd52:/planeteer.go

diff --git a/planeteer.go b/planeteer.go
index 7f343b9..4dbb27b 100644
--- a/planeteer.go
+++ b/planeteer.go
@@ -436,6 +436,59 @@ func FillStateTable1(data planet_data, dims []int, table []State) {
 	print("\n")
 }
 
+func FindBestState(data planet_data, dims []int, table []State) int {
+	addr := make([]int, NumDimensions)
+	addr[Edens] = *end_edens
+	addr[Cloaks] = dims[Cloaks] - 1
+	addr[NeedFighters] = dims[NeedFighters] - 1
+	addr[NeedShields] = dims[NeedShields] - 1
+	addr[Visit] = dims[Visit] - 1
+	// Fuel, Hold, UnusedCargo left at 0
+	var max_index int
+	max_value := 0
+	for addr[Location] = 0; addr[Location] < dims[Location]; addr[Location]++ {
+		index := EncodeIndex(dims, addr)
+		if table[index].value > max_value {
+			max_value = table[index].value
+			max_index = index
+		}
+	}
+	return max_index
+}
+
+func DescribePath(data planet_data, dims []int, table []State, start int) (description []string) {
+	for index := start; index > 0 && table[index].from > 0; index = table[index].from {
+		line := fmt.Sprintf("%10v", table[index].value)
+		addr := DecodeIndex(dims, index)
+		prev := DecodeIndex(dims, table[index].from)
+		if addr[Location] != prev[Location] {
+			from := data.i2p[prev[Location]]
+			to := data.i2p[addr[Location]]
+			if addr[Fuel] != prev[Fuel] {
+				line += fmt.Sprintf(" Jump from %v to %v (%v reactor units)", from, to, prev[Fuel]-addr[Fuel])
+			} else if addr[Edens] != prev[Edens] {
+				line += fmt.Sprintf(" Eden warp from %v to %v", from, to)
+			} else {
+				panic("Traveling without fuel?")
+			}
+		}
+		if addr[Hold] != prev[Hold] {
+			if addr[Hold] == 0 {
+				quantity := *hold - (prev[UnusedCargo] + prev[Edens] + prev[Cloaks])
+				line += fmt.Sprintf(" Sell %v %v", quantity, data.i2c[prev[Hold]])
+			} else if prev[Hold] == 0 {
+				quantity := *hold - (addr[UnusedCargo] + addr[Edens] + addr[Cloaks])
+				line += fmt.Sprintf(" Buy %v %v", quantity, data.i2c[addr[Hold]])
+			} else {
+				panic("Switched cargo?")
+			}
+
+		}
+		description = append(description, line)
+	}
+	return
+}
+
 // (Example of a use case for generics in Go)
 func IndexPlanets(m *map[string]Planet, start_at int) (map[string]int, []string) {
 	e2i := make(map[string]int, len(*m)+start_at)
@@ -468,6 +521,11 @@ func main() {
 	dims := DimensionSizes(data)
 	table := InitializeStateTable(data, dims)
 	FillStateTable1(data, dims, table)
-	print("Going to print state table...")
-	fmt.Printf("%v", table)
+	best := FindBestState(data, dims, table)
+	fmt.Printf("Best state: %v (%v) with $%v\n",
+		best, DecodeIndex(dims, best), table[best].value)
+	description := DescribePath(data, dims, table, best)
+	for i := len(description) - 1; i >= 0; i-- {
+		print(description[i], "\n")
+	}
 }