X-Git-Url: http://git.scottworley.com/planeteer/blobdiff_plain/db81510d35a720dc85793c0684a9c13a5ae38237..a29aebecc25a3b2c40c86f11818d79e367b5ce30:/planeteer.go diff --git a/planeteer.go b/planeteer.go index 7c5b2e7..eb78bce 100644 --- a/planeteer.go +++ b/planeteer.go @@ -212,13 +212,13 @@ func bint(b bool) int { return 0 } -func DimensionSizes(data planet_data) []int { +func DimensionSizes(data planet_data) LogicalIndex { eden_capacity := data.Commodities["Eden Warp Units"].Limit if *start_edens > eden_capacity { eden_capacity = *start_edens } cloak_capacity := bint(*cloak) - dims := make([]int, NumDimensions) + dims := make(LogicalIndex, NumDimensions) dims[Edens] = eden_capacity + 1 dims[Cloaks] = cloak_capacity + 1 dims[UnusedCargo] = eden_capacity + cloak_capacity + 1 @@ -239,7 +239,11 @@ func DimensionSizes(data planet_data) []int { return dims } -func StateTableSize(dims []int) int { +type Value int32 +type PhysicalIndex int32 +type LogicalIndex []int + +func StateTableSize(dims LogicalIndex) int { product := 1 for _, size := range dims { product *= size @@ -248,7 +252,7 @@ func StateTableSize(dims []int) int { } type State struct { - value int32 + value Value from PhysicalIndex } @@ -260,9 +264,7 @@ const ( VALUE_RUBISH ) -type PhysicalIndex int32 - -func EncodeIndex(dims, addr []int) PhysicalIndex { +func EncodeIndex(dims, addr LogicalIndex) PhysicalIndex { index := addr[0] if addr[0] > dims[0] { panic(0) @@ -276,9 +278,9 @@ func EncodeIndex(dims, addr []int) PhysicalIndex { return PhysicalIndex(index) } -func DecodeIndex(dims []int, index PhysicalIndex) []int { +func DecodeIndex(dims LogicalIndex, index PhysicalIndex) LogicalIndex { scratch := int(index) - addr := make([]int, NumDimensions) + addr := make(LogicalIndex, NumDimensions) for i := NumDimensions - 1; i > 0; i-- { addr[i] = scratch % dims[i] scratch /= dims[i] @@ -287,14 +289,14 @@ func DecodeIndex(dims []int, index PhysicalIndex) []int { return addr } -func CreateStateTable(data planet_data, dims []int) []State { +func CreateStateTable(data planet_data, dims LogicalIndex) []State { table := make([]State, StateTableSize(dims)) for i := range table { table[i].value = VALUE_UNINITIALIZED table[i].from = FROM_UNINITIALIZED } - addr := make([]int, NumDimensions) + addr := make(LogicalIndex, NumDimensions) addr[Fuel] = *fuel addr[Edens] = *start_edens addr[Location] = data.p2i[*start] @@ -302,7 +304,7 @@ func CreateStateTable(data planet_data, dims []int) []State { addr[Hold] = data.c2i[*start_hold] } start_index := EncodeIndex(dims, addr) - table[start_index].value = int32(*funds) + table[start_index].value = Value(*funds) table[start_index].from = FROM_ROOT return table @@ -311,13 +313,13 @@ func CreateStateTable(data planet_data, dims []int) []State { /* CellValue fills in the one cell at address addr by looking at all * the possible ways to reach this cell and selecting the best one. */ -func Consider(data planet_data, dims []int, table []State, there []int, value_difference int, best_value *int32, best_source []int) { +func Consider(data planet_data, dims LogicalIndex, table []State, there LogicalIndex, value_difference int, best_value *Value, best_source LogicalIndex) { there_value := CellValue(data, dims, table, there) - if value_difference < 0 && int32(-value_difference) > there_value { + if value_difference < 0 && Value(-value_difference) > there_value { /* Can't afford this transition */ return } - possible_value := there_value + int32(value_difference) + possible_value := there_value + Value(value_difference) if possible_value > *best_value { *best_value = possible_value copy(best_source, there) @@ -326,7 +328,7 @@ func Consider(data planet_data, dims []int, table []State, there []int, value_di var cell_filled_count int -func CellValue(data planet_data, dims []int, table []State, addr []int) int32 { +func CellValue(data planet_data, dims LogicalIndex, table []State, addr LogicalIndex) Value { my_index := EncodeIndex(dims, addr) if table[my_index].value == VALUE_BEING_EVALUATED { panic("Circular dependency") @@ -336,9 +338,9 @@ func CellValue(data planet_data, dims []int, table []State, addr []int) int32 { } table[my_index].value = VALUE_BEING_EVALUATED - best_value := int32(VALUE_RUBISH) - best_source := make([]int, NumDimensions) - other := make([]int, NumDimensions) + best_value := Value(VALUE_RUBISH) + best_source := make(LogicalIndex, NumDimensions) + other := make(LogicalIndex, NumDimensions) copy(other, addr) planet := data.i2p[addr[Location]] @@ -543,8 +545,8 @@ func CellValue(data planet_data, dims []int, table []State, addr []int) int32 { return table[my_index].value } -func FinalState(dims []int) []int { - addr := make([]int, NumDimensions) +func FinalState(dims LogicalIndex) LogicalIndex { + addr := make(LogicalIndex, NumDimensions) addr[Edens] = *end_edens addr[Cloaks] = dims[Cloaks] - 1 addr[BuyFighters] = dims[BuyFighters] - 1 @@ -557,7 +559,7 @@ func FinalState(dims []int) []int { return addr } -func FindBestState(data planet_data, dims []int, table []State, addr []int) PhysicalIndex { +func FindBestState(data planet_data, dims LogicalIndex, table []State, addr LogicalIndex) PhysicalIndex { max_index := PhysicalIndex(-1) max_value := 0.0 max_fuel := 1 @@ -582,7 +584,7 @@ func FindBestState(data planet_data, dims []int, table []State, addr []int) Phys return max_index } -func Commas(n int32) (s string) { +func Commas(n Value) (s string) { if n < 0 { panic(n) } @@ -597,7 +599,7 @@ func Commas(n int32) (s string) { return } -func FighterAndShieldCost(data planet_data, dims []int, table []State, best PhysicalIndex) { +func FighterAndShieldCost(data planet_data, dims LogicalIndex, table []State, best PhysicalIndex) { if *drones == 0 && *batteries == 0 { return } @@ -618,7 +620,7 @@ func FighterAndShieldCost(data planet_data, dims []int, table []State, best Phys } } -func EndEdensCost(data planet_data, dims []int, table []State, best PhysicalIndex) { +func EndEdensCost(data planet_data, dims LogicalIndex, table []State, best PhysicalIndex) { if *end_edens == 0 { return } @@ -630,11 +632,11 @@ func EndEdensCost(data planet_data, dims []int, table []State, best PhysicalInde extra_funds := table[alt_best].value - table[best].value fmt.Println("\rUse", extra_edens, "extra edens, make an extra", Commas(extra_funds), "(", - Commas(extra_funds/int32(extra_edens)), "per eden)") + Commas(extra_funds/Value(extra_edens)), "per eden)") } } -func VisitCost(data planet_data, dims []int, table []State, best PhysicalIndex) { +func VisitCost(data planet_data, dims LogicalIndex, table []State, best PhysicalIndex) { if dims[Visit] == 1 { return } @@ -649,7 +651,7 @@ func VisitCost(data planet_data, dims []int, table []State, best PhysicalIndex) } } -func EndLocationCost(data planet_data, dims []int, table []State, best PhysicalIndex) { +func EndLocationCost(data planet_data, dims LogicalIndex, table []State, best PhysicalIndex) { if len(end()) == 0 { return } @@ -664,7 +666,7 @@ func EndLocationCost(data planet_data, dims []int, table []State, best PhysicalI *end_string = save_end_string } -func DescribePath(data planet_data, dims []int, table []State, start PhysicalIndex) (description []string) { +func DescribePath(data planet_data, dims LogicalIndex, table []State, start PhysicalIndex) (description []string) { for index := start; table[index].from > FROM_ROOT; index = table[index].from { if table[index].from == FROM_UNINITIALIZED { panic(index)