From: Scott Worley Date: Thu, 21 Jun 2012 07:45:34 +0000 (-0700) Subject: Calculate visit costs X-Git-Url: http://git.scottworley.com/planeteer/commitdiff_plain/b7a6e28b4e3619a9b8ff18285bba9876de52549c Calculate visit costs Also, add a test for Commas. It was fine, but I got confused after passing it a negative number. --- diff --git a/planeteer.go b/planeteer.go index a8e6516..82650b9 100644 --- a/planeteer.go +++ b/planeteer.go @@ -480,8 +480,7 @@ func CellValue(data planet_data, dims []int, table []State, addr []int) int32 { } /* Visit this planet */ - var i uint - for i = 0; i < uint(len(visit())); i++ { + for i := uint(0); i < uint(len(visit())); i++ { if addr[Visit]&(1< 0 { @@ -711,6 +713,8 @@ func main() { fmt.Println(description[i]) } + // Ok, that was the important stuff. Now some fun stuff. + // Calculate total cost of fighters and shields if *drones > 0 || *batteries > 0 { fmt.Println() @@ -719,14 +723,14 @@ func main() { final_state[BuyFighters] = 0 alt_best := FindBestState(data, dims, table, final_state) cost := table[alt_best].value - table[best].value - fmt.Println("Drones were", float64(cost)/float64(*drones), "each") + fmt.Println("\rDrones were", float64(cost)/float64(*drones), "each") final_state[BuyFighters] = 1 } if *batteries > 0 { final_state[BuyShields] = 0 alt_best := FindBestState(data, dims, table, final_state) cost := table[alt_best].value - table[best].value - fmt.Println("Batteries were", float64(cost)/float64(*batteries), "each") + fmt.Println("\rBatteries were", float64(cost)/float64(*batteries), "each") final_state[BuyShields] = 1 } @@ -742,4 +746,19 @@ func main() { Commas(extra_funds), "(", Commas(extra_funds/int32(extra_edens)), "per eden)") } + final_state[Edens] = *end_edens + + // Cost of visiting places + if dims[Visit] > 1 { + fmt.Println() + } + for i := uint(0); i < uint(len(visit())); i++ { + all_bits := dims[Visit] - 1 + final_state[Visit] = all_bits & ^(1 << i) + alt_best := FindBestState(data, dims, table, final_state) + cost := table[alt_best].value - table[best].value + fmt.Println("\r", Commas(cost), "Cost to visit", visit()[i]) + } + final_state[Visit] = dims[Visit] - 1 + } diff --git a/planeteer_test.go b/planeteer_test.go index 7661ad6..42baae0 100644 --- a/planeteer_test.go +++ b/planeteer_test.go @@ -17,3 +17,24 @@ func TestEncodeDecode(t *testing.T) { } } } + +func TestCommas(t *testing.T) { + cases := map[int32]string{ + 1: "1", + 10: "10", + 100: "100", + 1000: "1,000", + 10000: "10,000", + 100000: "100,000", + 1000000: "1,000,000", + 1234567: "1,234,567", + 1000567: "1,000,567", + 1234000: "1,234,000", + 525000: "525,000", + } + for n, s := range cases { + if Commas(n) != s { + t.Error(n, "not", s) + } + } +}