diff --git a/maps/sinkholes.go b/maps/sinkholes.go new file mode 100644 index 0000000..80c8ceb --- /dev/null +++ b/maps/sinkholes.go @@ -0,0 +1,88 @@ +package maps + +import ( + "math" + + "github.com/BattlesnakeOfficial/rules" +) + +type SinkholesMap struct{} + +func init() { + globalRegistry.RegisterMap("sinkholes", SinkholesMap{}) +} + +func (m SinkholesMap) ID() string { + return "sinkholes" +} + +func (m SinkholesMap) Meta() Metadata { + return Metadata{ + Name: "Sinkholes", + Description: "Spawns a rounded sinkhole on the board that grows every N turns, layering additional hazard squares over previously spawned ones.", + Author: "Battlesnake", + Version: 1, + MinPlayers: 1, + MaxPlayers: 8, + BoardSizes: FixedSizes(Dimensions{7, 7}, Dimensions{11, 11}, Dimensions{19, 19}), + } +} + +func (m SinkholesMap) SetupBoard(initialBoardState *rules.BoardState, settings rules.Settings, editor Editor) error { + return (StandardMap{}).SetupBoard(initialBoardState, settings, editor) +} + +func (m SinkholesMap) UpdateBoard(lastBoardState *rules.BoardState, settings rules.Settings, editor Editor) error { + err := StandardMap{}.UpdateBoard(lastBoardState, settings, editor) + if err != nil { + return err + } + + currentTurn := lastBoardState.Turn + startTurn := 1 + spawnEveryNTurns := 10 + if settings.RoyaleSettings.ShrinkEveryNTurns > 0 { + spawnEveryNTurns = settings.RoyaleSettings.ShrinkEveryNTurns + } + maxRings := 5 + if lastBoardState.Width == 7 { + maxRings = 3 + } else if lastBoardState.Width == 19 { + maxRings = 7 + } + + spawnLocation := rules.Point{X: lastBoardState.Width / 2, Y: lastBoardState.Height / 2} + + if currentTurn == startTurn { + editor.AddHazard(spawnLocation) + return nil + } + + // Are we at max size, if so stop try to generate hazards + if currentTurn > spawnEveryNTurns*maxRings { + return nil + } + + // Is this a turn to grow the sinkhole? + if (currentTurn-startTurn)%spawnEveryNTurns != 0 { + return nil + } + + offset := int(math.Floor(float64(currentTurn-startTurn) / float64(spawnEveryNTurns))) + + if offset > 0 && offset <= maxRings { + for x := spawnLocation.X - offset; x <= spawnLocation.X+offset; x++ { + for y := spawnLocation.Y - offset; y <= spawnLocation.Y+offset; y++ { + // don't draw in the corners of the square so we get a rounded effect + if !(x == spawnLocation.X-offset && y == spawnLocation.Y-offset) && + !(x == spawnLocation.X+offset && y == spawnLocation.Y-offset) && + !(x == spawnLocation.X-offset && y == spawnLocation.Y+offset) && + !(x == spawnLocation.X+offset && y == spawnLocation.Y+offset) { + editor.AddHazard(rules.Point{X: x, Y: y}) + } + } + } + } + + return nil +} diff --git a/maps/sinkholes_test.go b/maps/sinkholes_test.go new file mode 100644 index 0000000..4fa06a7 --- /dev/null +++ b/maps/sinkholes_test.go @@ -0,0 +1,57 @@ +package maps_test + +import ( + "fmt" + "testing" + + "github.com/BattlesnakeOfficial/rules" + "github.com/BattlesnakeOfficial/rules/maps" + "github.com/stretchr/testify/require" +) + +func TestSinkholesMap(t *testing.T) { + + tests := []struct { + boardSize int + expectedHazards int + expectedHazardsInCenter int + }{ + {7, 27, 3}, + {11, 149, 5}, + {19, 431, 7}, + } + + for _, tc := range tests { + + t.Run(fmt.Sprintf("%dx%d", tc.boardSize, tc.boardSize), func(t *testing.T) { + m := maps.SinkholesMap{} + state := rules.NewBoardState(tc.boardSize, tc.boardSize) + settings := rules.Settings{} + + // ensure the ring of hazards is added to the board at setup + editor := maps.NewBoardStateEditor(state) + require.Empty(t, state.Hazards) + err := m.SetupBoard(state, settings, editor) + require.NoError(t, err) + require.Empty(t, state.Hazards) + + totalTurns := 100 + for i := 0; i < totalTurns; i++ { + state.Turn = i + err = m.UpdateBoard(state, settings, editor) + require.NoError(t, err) + } + require.NotEmpty(t, state.Hazards) + require.Len(t, state.Hazards, tc.expectedHazards) + + centerPoint := rules.Point{X: tc.boardSize / 2, Y: tc.boardSize / 2} + numCenterHazards := 0 + for _, p := range state.Hazards { + if p == centerPoint { + numCenterHazards += 1 + } + } + require.Equal(t, numCenterHazards, tc.expectedHazardsInCenter) + }) + } +}