DescribePath()

[planeteer] / planeteer.go

diff --git a/planeteer.go b/planeteer.go
index 7f343b963f051764ccc9e110636013c1efbc0d32..4dbb27b28c3cb5ece9e5380d83a910f90fc163f0 100644 (file)
--- 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")
+       }
 }