Compare commits
2 Commits
c5b0319ac4
...
16c78c81ae
| Author | SHA1 | Date | |
|---|---|---|---|
| 16c78c81ae | |||
| a4dc6eb6da |
114
Cargo.lock
generated
114
Cargo.lock
generated
@ -26,6 +26,12 @@ dependencies = [
|
||||
"memchr",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "anes"
|
||||
version = "0.2.0"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "735d4f398ca57cfa2880225c2bf81c3b9af3be5bb22e44ae70118dad38713e84"
|
||||
|
||||
[[package]]
|
||||
name = "anstream"
|
||||
version = "0.6.15"
|
||||
@ -148,6 +154,7 @@ dependencies = [
|
||||
name = "battlesnake"
|
||||
version = "1.0.0"
|
||||
dependencies = [
|
||||
"criterion2",
|
||||
"enum-iterator",
|
||||
"env_logger",
|
||||
"iter_tools",
|
||||
@ -171,6 +178,12 @@ version = "2.6.0"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "b048fb63fd8b5923fc5aa7b340d8e156aec7ec02f0c78fa8a6ddc2613f6f71de"
|
||||
|
||||
[[package]]
|
||||
name = "bpaf"
|
||||
version = "0.9.14"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "8d4e5ca9929037866947af4b8b7418124f2ec7c411a8b9ee24e46ad2b8470497"
|
||||
|
||||
[[package]]
|
||||
name = "bytemuck"
|
||||
version = "1.17.1"
|
||||
@ -189,6 +202,12 @@ version = "1.7.1"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "8318a53db07bb3f8dca91a600466bdb3f2eaadeedfdbcf02e1accbad9271ba50"
|
||||
|
||||
[[package]]
|
||||
name = "cast"
|
||||
version = "0.3.0"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "37b2a672a2cb129a2e41c10b1224bb368f9f37a2b16b612598138befd7b37eb5"
|
||||
|
||||
[[package]]
|
||||
name = "cc"
|
||||
version = "1.1.15"
|
||||
@ -204,6 +223,33 @@ version = "1.0.0"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "baf1de4339761588bc0619e3cbc0120ee582ebb74b53b4efbf79117bd2da40fd"
|
||||
|
||||
[[package]]
|
||||
name = "ciborium"
|
||||
version = "0.2.2"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "42e69ffd6f0917f5c029256a24d0161db17cea3997d185db0d35926308770f0e"
|
||||
dependencies = [
|
||||
"ciborium-io",
|
||||
"ciborium-ll",
|
||||
"serde",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "ciborium-io"
|
||||
version = "0.2.2"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "05afea1e0a06c9be33d539b876f1ce3692f4afea2cb41f740e7743225ed1c757"
|
||||
|
||||
[[package]]
|
||||
name = "ciborium-ll"
|
||||
version = "0.2.2"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "57663b653d948a338bfb3eeba9bb2fd5fcfaecb9e199e87e1eda4d9e8b240fd9"
|
||||
dependencies = [
|
||||
"ciborium-io",
|
||||
"half",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "clone_dyn_types"
|
||||
version = "0.23.0"
|
||||
@ -227,6 +273,24 @@ dependencies = [
|
||||
"version_check",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "criterion2"
|
||||
version = "1.1.1"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "d6c2be2af5816b76f45745ccfb9959a9efd1322ae4c88fc3c9b2d6bcdabe5554"
|
||||
dependencies = [
|
||||
"anes",
|
||||
"bpaf",
|
||||
"cast",
|
||||
"ciborium",
|
||||
"num-traits",
|
||||
"oorandom",
|
||||
"rayon",
|
||||
"serde",
|
||||
"serde_json",
|
||||
"walkdir",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "crossbeam-deque"
|
||||
version = "0.8.5"
|
||||
@ -252,6 +316,12 @@ version = "0.8.20"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "22ec99545bb0ed0ea7bb9b8e1e9122ea386ff8a48c0922e43f36d45ab09e0e80"
|
||||
|
||||
[[package]]
|
||||
name = "crunchy"
|
||||
version = "0.2.2"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "7a81dae078cea95a014a339291cec439d2f232ebe854a9d672b796c6afafa9b7"
|
||||
|
||||
[[package]]
|
||||
name = "deranged"
|
||||
version = "0.3.11"
|
||||
@ -514,6 +584,16 @@ dependencies = [
|
||||
"tracing",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "half"
|
||||
version = "2.4.1"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "6dd08c532ae367adf81c312a4580bc67f1d0fe8bc9c460520283f4c0ff277888"
|
||||
dependencies = [
|
||||
"cfg-if",
|
||||
"crunchy",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "hashbrown"
|
||||
version = "0.14.5"
|
||||
@ -826,6 +906,12 @@ version = "1.19.0"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "3fdb12b2476b595f9358c5161aa467c2438859caa136dec86c26fdd2efe17b92"
|
||||
|
||||
[[package]]
|
||||
name = "oorandom"
|
||||
version = "11.1.4"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "b410bbe7e14ab526a0e86877eb47c6996a2bd7746f027ba551028c925390e4e9"
|
||||
|
||||
[[package]]
|
||||
name = "ordered-float"
|
||||
version = "4.3.0"
|
||||
@ -1187,6 +1273,15 @@ version = "1.0.18"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "f3cb5ba0dc43242ce17de99c180e96db90b235b8a9fdc9543c96d2209116bd9f"
|
||||
|
||||
[[package]]
|
||||
name = "same-file"
|
||||
version = "1.0.6"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "93fc1dc3aaa9bfed95e02e6eadabb4baf7e3078b0bd1b4d7b6b0b68378900502"
|
||||
dependencies = [
|
||||
"winapi-util",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "scoped-tls"
|
||||
version = "1.0.1"
|
||||
@ -1586,6 +1681,16 @@ version = "0.9.5"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "0b928f33d975fc6ad9f86c8f283853ad26bdd5b10b7f1542aa2fa15e2289105a"
|
||||
|
||||
[[package]]
|
||||
name = "walkdir"
|
||||
version = "2.5.0"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "29790946404f91d9c5d06f9874efddea1dc06c5efe94541a7d6863108e3a5e4b"
|
||||
dependencies = [
|
||||
"same-file",
|
||||
"winapi-util",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "want"
|
||||
version = "0.3.1"
|
||||
@ -1617,6 +1722,15 @@ version = "0.4.0"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "ac3b87c63620426dd9b991e5ce0329eff545bccbbb34f3be09ff6fb6ab51b7b6"
|
||||
|
||||
[[package]]
|
||||
name = "winapi-util"
|
||||
version = "0.1.9"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "cf221c93e13a30d793f7645a0e7762c55d169dbb0a49671918a2319d289b10bb"
|
||||
dependencies = [
|
||||
"windows-sys 0.59.0",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "winapi-x86_64-pc-windows-gnu"
|
||||
version = "0.4.0"
|
||||
|
||||
@ -26,8 +26,16 @@ enum-iterator = "2.1"
|
||||
iter_tools = "0.21"
|
||||
ordered-float = "4.3.0"
|
||||
|
||||
[dev-dependencies]
|
||||
# criterion = { version = "0.5.1", features = ["html_reports"] }
|
||||
criterion2 = "1.1.1"
|
||||
|
||||
[profile.release]
|
||||
lto = "fat"
|
||||
codegen-units = 1
|
||||
panic = "abort"
|
||||
strip = true
|
||||
|
||||
[[bench]]
|
||||
name = "simulation"
|
||||
harness = false
|
||||
|
||||
120
battlesnake/benches/simulation.rs
Normal file
120
battlesnake/benches/simulation.rs
Normal file
@ -0,0 +1,120 @@
|
||||
use battlesnake::{
|
||||
simulation::{self, SnakeToken},
|
||||
Coord,
|
||||
};
|
||||
use criterion::{black_box, criterion_group, criterion_main, Criterion};
|
||||
use rand::{rngs::StdRng, SeedableRng};
|
||||
|
||||
fn random_moves(board: &battlesnake::Board) -> Option<SnakeToken> {
|
||||
let token_map = SnakeToken::from_board(board);
|
||||
let mut board = simulation::Board::from_game_board(board, &token_map, 0, 15, 1, false);
|
||||
let mut rng = StdRng::seed_from_u64(0);
|
||||
|
||||
board.simulate_until(&mut rng, |board| board.alive_snakes() <= 1);
|
||||
|
||||
let winner = board.snakes().next();
|
||||
winner
|
||||
}
|
||||
|
||||
fn bench_duel_random_moves(c: &mut Criterion) {
|
||||
c.bench_function("duel random moves", |b| {
|
||||
b.iter(|| {
|
||||
random_moves(black_box(&battlesnake::Board {
|
||||
height: 11,
|
||||
width: 11,
|
||||
food: vec![Coord { x: 5, y: 5 }],
|
||||
snakes: vec![
|
||||
battlesnake::Battlesnake {
|
||||
id: "1".to_owned(),
|
||||
name: "1".to_owned(),
|
||||
health: 100,
|
||||
body: vec![Coord { x: 5, y: 1 }; 3],
|
||||
head: Coord { x: 5, y: 1 },
|
||||
length: 3,
|
||||
latency: "0".to_owned(),
|
||||
shout: None,
|
||||
squad: String::new(),
|
||||
},
|
||||
battlesnake::Battlesnake {
|
||||
id: "2".to_owned(),
|
||||
name: "2".to_owned(),
|
||||
health: 100,
|
||||
body: vec![Coord { x: 5, y: 9 }; 3],
|
||||
head: Coord { x: 5, y: 9 },
|
||||
length: 3,
|
||||
latency: "0".to_owned(),
|
||||
shout: None,
|
||||
squad: String::new(),
|
||||
},
|
||||
],
|
||||
hazards: vec![],
|
||||
}))
|
||||
});
|
||||
});
|
||||
}
|
||||
|
||||
fn bench_standard_random_moves(c: &mut Criterion) {
|
||||
c.bench_function("standard random moves", |b| {
|
||||
b.iter(|| {
|
||||
random_moves(black_box(&battlesnake::Board {
|
||||
height: 11,
|
||||
width: 11,
|
||||
food: vec![Coord { x: 5, y: 5 }],
|
||||
snakes: vec![
|
||||
battlesnake::Battlesnake {
|
||||
id: "1".to_owned(),
|
||||
name: "1".to_owned(),
|
||||
health: 100,
|
||||
body: vec![Coord { x: 5, y: 1 }; 3],
|
||||
head: Coord { x: 5, y: 1 },
|
||||
length: 3,
|
||||
latency: "0".to_owned(),
|
||||
shout: None,
|
||||
squad: String::new(),
|
||||
},
|
||||
battlesnake::Battlesnake {
|
||||
id: "2".to_owned(),
|
||||
name: "2".to_owned(),
|
||||
health: 100,
|
||||
body: vec![Coord { x: 5, y: 9 }; 3],
|
||||
head: Coord { x: 5, y: 9 },
|
||||
length: 3,
|
||||
latency: "0".to_owned(),
|
||||
shout: None,
|
||||
squad: String::new(),
|
||||
},
|
||||
battlesnake::Battlesnake {
|
||||
id: "3".to_owned(),
|
||||
name: "3".to_owned(),
|
||||
health: 100,
|
||||
body: vec![Coord { x: 1, y: 5 }; 3],
|
||||
head: Coord { x: 1, y: 5 },
|
||||
length: 3,
|
||||
latency: "0".to_owned(),
|
||||
shout: None,
|
||||
squad: String::new(),
|
||||
},
|
||||
battlesnake::Battlesnake {
|
||||
id: "4".to_owned(),
|
||||
name: "4".to_owned(),
|
||||
health: 100,
|
||||
body: vec![Coord { x: 9, y: 5 }; 3],
|
||||
head: Coord { x: 9, y: 5 },
|
||||
length: 3,
|
||||
latency: "0".to_owned(),
|
||||
shout: None,
|
||||
squad: String::new(),
|
||||
},
|
||||
],
|
||||
hazards: vec![],
|
||||
}))
|
||||
});
|
||||
});
|
||||
}
|
||||
|
||||
criterion_group!(
|
||||
benches,
|
||||
bench_duel_random_moves,
|
||||
bench_standard_random_moves
|
||||
);
|
||||
criterion_main!(benches);
|
||||
183
battlesnake/src/lib.rs
Normal file
183
battlesnake/src/lib.rs
Normal file
@ -0,0 +1,183 @@
|
||||
use enum_iterator::Sequence;
|
||||
use serde::{Deserialize, Serialize};
|
||||
|
||||
pub mod logic;
|
||||
pub mod simulation;
|
||||
|
||||
pub const MAX_HEALTH: i32 = 100;
|
||||
|
||||
// API and Response Objects
|
||||
// See https://docs.battlesnake.com/api
|
||||
|
||||
#[derive(
|
||||
Debug, PartialEq, Eq, PartialOrd, Ord, Hash, Clone, Copy, Deserialize, Serialize, Sequence,
|
||||
)]
|
||||
#[serde(rename_all = "lowercase")]
|
||||
pub enum Direction {
|
||||
/// Move left (-x)
|
||||
Left,
|
||||
/// Move up (+y)
|
||||
Up,
|
||||
/// Move right (+x)
|
||||
Right,
|
||||
/// Move down (-y)
|
||||
Down,
|
||||
}
|
||||
|
||||
#[derive(Debug, Deserialize, Serialize)]
|
||||
pub struct Action {
|
||||
/// In which direction the snake should move
|
||||
pub r#move: Direction,
|
||||
/// Say something to the other snakes
|
||||
#[serde(default, skip_serializing_if = "is_default")]
|
||||
pub shout: Option<String>,
|
||||
}
|
||||
|
||||
fn is_default<T: Default + PartialEq>(value: &T) -> bool {
|
||||
*value == T::default()
|
||||
}
|
||||
|
||||
#[derive(Deserialize, Serialize, Debug)]
|
||||
pub struct Game {
|
||||
/// A unique identifier for this Game
|
||||
pub id: String,
|
||||
/// Information about the ruleset being used to run this game
|
||||
pub ruleset: Ruleset,
|
||||
/// The name of the map being played on.
|
||||
pub map: String,
|
||||
/// How much time your snake has to respond to requests for this Game
|
||||
pub timeout: u32,
|
||||
/// The source of this game.
|
||||
///
|
||||
/// One of:
|
||||
/// - "tournament"
|
||||
/// - "league"
|
||||
/// - "arena"
|
||||
/// - "challenge"
|
||||
/// - "custom"
|
||||
///
|
||||
/// The values may change.
|
||||
pub source: String,
|
||||
}
|
||||
|
||||
#[derive(Debug, Deserialize, Serialize)]
|
||||
pub struct Ruleset {
|
||||
/// Name of the ruleset being used to run this game.
|
||||
pub name: String,
|
||||
/// The release version of the [Rules](https://github.com/BattlesnakeOfficial/rules) module used in this game.
|
||||
pub version: String,
|
||||
/// A collection of specific settings being used by the current game that control how the rules
|
||||
/// are applied.
|
||||
pub settings: RulesetSettings,
|
||||
}
|
||||
|
||||
#[derive(Debug, Deserialize, Serialize)]
|
||||
pub struct RulesetSettings {
|
||||
/// Percentage chance of spawning a new food every round.
|
||||
#[serde(rename = "foodSpawnChance")]
|
||||
pub food_spawn_chance: u8,
|
||||
/// Minimum food to keep on the board every turn.
|
||||
#[serde(rename = "minimumFood")]
|
||||
pub minimum_food: u8,
|
||||
/// Health damage a snake will take when ending its turn in a hazard. This stacks on top of the
|
||||
/// regular 1 damage a snake takes per turn.
|
||||
#[serde(rename = "hazardDamagePerTurn")]
|
||||
pub hazard_damage_per_turn: u8,
|
||||
/// rules for the royale mode
|
||||
pub royale: RulesetRoyale,
|
||||
/// rules for the squad mode
|
||||
pub squad: RulesetSquad,
|
||||
}
|
||||
|
||||
#[derive(Debug, Deserialize, Serialize)]
|
||||
pub struct RulesetRoyale {
|
||||
/// The number of turns between generating new hazards (shrinking the safe board space).
|
||||
#[serde(rename = "shrinkEveryNTurns")]
|
||||
pub shrink_every_n_turns: i32,
|
||||
}
|
||||
|
||||
#[allow(clippy::struct_excessive_bools)]
|
||||
#[derive(Debug, Deserialize, Serialize)]
|
||||
pub struct RulesetSquad {
|
||||
/// Allow members of the same squad to move over each other without dying.
|
||||
#[serde(rename = "allowBodyCollisions")]
|
||||
pub allow_body_collisions: bool,
|
||||
/// All squad members are eliminated when one is eliminated.
|
||||
#[serde(rename = "sharedElimination")]
|
||||
pub shared_elimination: bool,
|
||||
/// All squad members share health.
|
||||
#[serde(rename = "sharedHealth")]
|
||||
pub shared_health: bool,
|
||||
/// All squad members share length.
|
||||
#[serde(rename = "sharedLength")]
|
||||
pub shared_length: bool,
|
||||
}
|
||||
|
||||
#[derive(Deserialize, Serialize, Debug, Clone)]
|
||||
pub struct Board {
|
||||
/// The number of rows in the y-axis of the game board.
|
||||
pub height: i32,
|
||||
/// The number of columns in the x-axis of the game board.
|
||||
pub width: i32,
|
||||
/// Array of coordinates representing food locations on the game board.
|
||||
pub food: Vec<Coord>,
|
||||
/// Array of Battlesnakes representing all Battlesnakes remaining on the game board (including
|
||||
/// yourself if you haven't been eliminated).
|
||||
pub snakes: Vec<Battlesnake>,
|
||||
/// Array of coordinates representing hazardous locations on the game board.
|
||||
pub hazards: Vec<Coord>,
|
||||
}
|
||||
|
||||
#[derive(Deserialize, Serialize, Debug, Clone)]
|
||||
pub struct Battlesnake {
|
||||
/// Unique identifier for this Battlesnake in the context of the current Game
|
||||
pub id: String,
|
||||
/// Name given to this Battlesnake by its author
|
||||
pub name: String,
|
||||
/// Health value of this Battlesnake, between 0 and 100
|
||||
pub health: i32,
|
||||
/// Array of coordinates representing this Battlesnake's location on the game board. This array
|
||||
/// is ordered from head to tail
|
||||
pub body: Vec<Coord>,
|
||||
/// Coordinates for this Battlesnake's head. Equivalent to the first element of the body array.
|
||||
pub head: Coord,
|
||||
/// Length of this Battlesnake from head to tail. Equivalent to the length of the body array.
|
||||
pub length: i32,
|
||||
/// The previous response time of this Battlesnake, in milliseconds. If the Battlesnake timed
|
||||
/// out and failed to respond, the game timeout will be returned.
|
||||
pub latency: String,
|
||||
/// Message shouted by this Battlesnake on the previous turn.
|
||||
pub shout: Option<String>,
|
||||
/// The squad that the Battlesnake belongs to. Used to identify squad members in Squad Mode
|
||||
/// games.
|
||||
pub squad: String,
|
||||
// /// The collection of customizations that control how this Battlesnake is displayed.
|
||||
// customizations: {color, head, tail}
|
||||
}
|
||||
|
||||
#[derive(Debug, PartialEq, Eq, PartialOrd, Ord, Clone, Copy, Deserialize, Serialize)]
|
||||
pub struct Coord {
|
||||
pub x: i32,
|
||||
pub y: i32,
|
||||
}
|
||||
|
||||
impl Coord {
|
||||
#[must_use]
|
||||
pub const fn move_to(mut self, direction: Direction) -> Self {
|
||||
match direction {
|
||||
Direction::Left => self.x -= 1,
|
||||
Direction::Up => self.y += 1,
|
||||
Direction::Right => self.x += 1,
|
||||
Direction::Down => self.y -= 1,
|
||||
}
|
||||
self
|
||||
}
|
||||
}
|
||||
|
||||
#[derive(Deserialize, Serialize, Debug)]
|
||||
pub struct GameState {
|
||||
pub game: Game,
|
||||
pub turn: i32,
|
||||
pub board: Board,
|
||||
pub you: Battlesnake,
|
||||
}
|
||||
@ -19,6 +19,7 @@ use std::{
|
||||
|
||||
use log::{error, info};
|
||||
use ordered_float::OrderedFloat;
|
||||
use rand::thread_rng;
|
||||
use serde_json::{json, Value};
|
||||
|
||||
use crate::{
|
||||
@ -205,7 +206,9 @@ impl Node {
|
||||
) -> Result<Option<SnakeToken>, DeadlineError> {
|
||||
let winner = if self.statistic.played == 0 {
|
||||
// We didn't simulate a game for this node yet. Do that
|
||||
board.simulate_until(|board| board.alive_snakes() <= 1 || Instant::now() >= *deadline);
|
||||
board.simulate_until(&mut thread_rng(), |board| {
|
||||
board.alive_snakes() <= 1 || Instant::now() >= *deadline
|
||||
});
|
||||
if Instant::now() >= *deadline {
|
||||
return Err(DeadlineError);
|
||||
}
|
||||
@ -239,7 +242,7 @@ impl Node {
|
||||
if Instant::now() >= *deadline {
|
||||
return Err(DeadlineError);
|
||||
}
|
||||
board.simulate_actions(&actions);
|
||||
board.simulate_actions(&actions, &mut thread_rng());
|
||||
let winner = self
|
||||
.childs
|
||||
.entry(actions.clone())
|
||||
@ -287,7 +290,7 @@ impl Node {
|
||||
.snakes()
|
||||
.filter_map(|snake| Some((snake, board.snake_length(snake)?)))
|
||||
.collect();
|
||||
board.simulate_until(|board| {
|
||||
board.simulate_until(&mut thread_rng(), |board| {
|
||||
if Instant::now() >= *deadline {
|
||||
return true;
|
||||
}
|
||||
@ -333,7 +336,7 @@ impl Node {
|
||||
if Instant::now() >= *deadline {
|
||||
return Err(DeadlineError);
|
||||
}
|
||||
board.simulate_actions(&actions);
|
||||
board.simulate_actions(&actions, &mut thread_rng());
|
||||
let lengths = self
|
||||
.childs
|
||||
.entry(actions.clone())
|
||||
|
||||
@ -1,6 +1,6 @@
|
||||
#![allow(clippy::needless_pass_by_value)]
|
||||
|
||||
use enum_iterator::Sequence;
|
||||
use battlesnake::{logic, Action, GameState};
|
||||
use log::{error, info};
|
||||
use rocket::{
|
||||
fairing::AdHoc,
|
||||
@ -10,190 +10,9 @@ use rocket::{
|
||||
serde::{json::Json, Deserialize},
|
||||
tokio::task,
|
||||
};
|
||||
use serde::Serialize;
|
||||
use serde_json::Value;
|
||||
use std::env;
|
||||
|
||||
mod logic;
|
||||
mod simulation;
|
||||
|
||||
const MAX_HEALTH: i32 = 100;
|
||||
|
||||
// API and Response Objects
|
||||
// See https://docs.battlesnake.com/api
|
||||
|
||||
#[derive(
|
||||
Debug, PartialEq, Eq, PartialOrd, Ord, Hash, Clone, Copy, Deserialize, Serialize, Sequence,
|
||||
)]
|
||||
#[serde(rename_all = "lowercase")]
|
||||
enum Direction {
|
||||
/// Move left (-x)
|
||||
Left,
|
||||
/// Move up (+y)
|
||||
Up,
|
||||
/// Move right (+x)
|
||||
Right,
|
||||
/// Move down (-y)
|
||||
Down,
|
||||
}
|
||||
|
||||
#[derive(Debug, Deserialize, Serialize)]
|
||||
struct Action {
|
||||
/// In which direction the snake should move
|
||||
r#move: Direction,
|
||||
/// Say something to the other snakes
|
||||
#[serde(default, skip_serializing_if = "is_default")]
|
||||
shout: Option<String>,
|
||||
}
|
||||
|
||||
fn is_default<T: Default + PartialEq>(value: &T) -> bool {
|
||||
*value == T::default()
|
||||
}
|
||||
|
||||
#[derive(Deserialize, Serialize, Debug)]
|
||||
struct Game {
|
||||
/// A unique identifier for this Game
|
||||
id: String,
|
||||
/// Information about the ruleset being used to run this game
|
||||
ruleset: Ruleset,
|
||||
/// The name of the map being played on.
|
||||
map: String,
|
||||
/// How much time your snake has to respond to requests for this Game
|
||||
timeout: u32,
|
||||
/// The source of this game.
|
||||
///
|
||||
/// One of:
|
||||
/// - "tournament"
|
||||
/// - "league"
|
||||
/// - "arena"
|
||||
/// - "challenge"
|
||||
/// - "custom"
|
||||
///
|
||||
/// The values may change.
|
||||
source: String,
|
||||
}
|
||||
|
||||
#[derive(Debug, Deserialize, Serialize)]
|
||||
struct Ruleset {
|
||||
/// Name of the ruleset being used to run this game.
|
||||
name: String,
|
||||
/// The release version of the [Rules](https://github.com/BattlesnakeOfficial/rules) module used in this game.
|
||||
version: String,
|
||||
/// A collection of specific settings being used by the current game that control how the rules
|
||||
/// are applied.
|
||||
settings: RulesetSettings,
|
||||
}
|
||||
|
||||
#[derive(Debug, Deserialize, Serialize)]
|
||||
struct RulesetSettings {
|
||||
/// Percentage chance of spawning a new food every round.
|
||||
#[serde(rename = "foodSpawnChance")]
|
||||
food_spawn_chance: u8,
|
||||
/// Minimum food to keep on the board every turn.
|
||||
#[serde(rename = "minimumFood")]
|
||||
minimum_food: u8,
|
||||
/// Health damage a snake will take when ending its turn in a hazard. This stacks on top of the
|
||||
/// regular 1 damage a snake takes per turn.
|
||||
#[serde(rename = "hazardDamagePerTurn")]
|
||||
hazard_damage_per_turn: u8,
|
||||
/// rules for the royale mode
|
||||
royale: RulesetRoyale,
|
||||
/// rules for the squad mode
|
||||
squad: RulesetSquad,
|
||||
}
|
||||
|
||||
#[derive(Debug, Deserialize, Serialize)]
|
||||
struct RulesetRoyale {
|
||||
/// The number of turns between generating new hazards (shrinking the safe board space).
|
||||
#[serde(rename = "shrinkEveryNTurns")]
|
||||
shrink_every_n_turns: i32,
|
||||
}
|
||||
|
||||
#[allow(clippy::struct_excessive_bools)]
|
||||
#[derive(Debug, Deserialize, Serialize)]
|
||||
struct RulesetSquad {
|
||||
/// Allow members of the same squad to move over each other without dying.
|
||||
#[serde(rename = "allowBodyCollisions")]
|
||||
allow_body_collisions: bool,
|
||||
/// All squad members are eliminated when one is eliminated.
|
||||
#[serde(rename = "sharedElimination")]
|
||||
shared_elimination: bool,
|
||||
/// All squad members share health.
|
||||
#[serde(rename = "sharedHealth")]
|
||||
shared_health: bool,
|
||||
/// All squad members share length.
|
||||
#[serde(rename = "sharedLength")]
|
||||
shared_length: bool,
|
||||
}
|
||||
|
||||
#[derive(Deserialize, Serialize, Debug, Clone)]
|
||||
struct Board {
|
||||
/// The number of rows in the y-axis of the game board.
|
||||
height: i32,
|
||||
/// The number of columns in the x-axis of the game board.
|
||||
width: i32,
|
||||
/// Array of coordinates representing food locations on the game board.
|
||||
food: Vec<Coord>,
|
||||
/// Array of Battlesnakes representing all Battlesnakes remaining on the game board (including
|
||||
/// yourself if you haven't been eliminated).
|
||||
snakes: Vec<Battlesnake>,
|
||||
/// Array of coordinates representing hazardous locations on the game board.
|
||||
hazards: Vec<Coord>,
|
||||
}
|
||||
|
||||
#[derive(Deserialize, Serialize, Debug, Clone)]
|
||||
struct Battlesnake {
|
||||
/// Unique identifier for this Battlesnake in the context of the current Game
|
||||
id: String,
|
||||
/// Name given to this Battlesnake by its author
|
||||
name: String,
|
||||
/// Health value of this Battlesnake, between 0 and 100
|
||||
health: i32,
|
||||
/// Array of coordinates representing this Battlesnake's location on the game board. This array
|
||||
/// is ordered from head to tail
|
||||
body: Vec<Coord>,
|
||||
/// Coordinates for this Battlesnake's head. Equivalent to the first element of the body array.
|
||||
head: Coord,
|
||||
/// Length of this Battlesnake from head to tail. Equivalent to the length of the body array.
|
||||
length: i32,
|
||||
/// The previous response time of this Battlesnake, in milliseconds. If the Battlesnake timed
|
||||
/// out and failed to respond, the game timeout will be returned.
|
||||
latency: String,
|
||||
/// Message shouted by this Battlesnake on the previous turn.
|
||||
shout: Option<String>,
|
||||
/// The squad that the Battlesnake belongs to. Used to identify squad members in Squad Mode
|
||||
/// games.
|
||||
squad: String,
|
||||
// /// The collection of customizations that control how this Battlesnake is displayed.
|
||||
// customizations: {color, head, tail}
|
||||
}
|
||||
|
||||
#[derive(Debug, PartialEq, Eq, PartialOrd, Ord, Clone, Copy, Deserialize, Serialize)]
|
||||
struct Coord {
|
||||
x: i32,
|
||||
y: i32,
|
||||
}
|
||||
|
||||
impl Coord {
|
||||
const fn move_to(mut self, direction: Direction) -> Self {
|
||||
match direction {
|
||||
Direction::Left => self.x -= 1,
|
||||
Direction::Up => self.y += 1,
|
||||
Direction::Right => self.x += 1,
|
||||
Direction::Down => self.y -= 1,
|
||||
}
|
||||
self
|
||||
}
|
||||
}
|
||||
|
||||
#[derive(Deserialize, Serialize, Debug)]
|
||||
struct GameState {
|
||||
game: Game,
|
||||
turn: i32,
|
||||
board: Board,
|
||||
you: Battlesnake,
|
||||
}
|
||||
|
||||
#[get("/")]
|
||||
fn handle_index() -> Json<Value> {
|
||||
Json(logic::info())
|
||||
|
||||
@ -86,6 +86,10 @@ impl Board {
|
||||
}
|
||||
}
|
||||
|
||||
pub const fn turn(&self) -> i32 {
|
||||
self.turn
|
||||
}
|
||||
|
||||
#[allow(clippy::cast_sign_loss)]
|
||||
pub const fn spaces(&self) -> usize {
|
||||
self.height as usize * self.width as usize
|
||||
@ -103,7 +107,11 @@ impl Board {
|
||||
self.snakes.get(&token).map(|snake| snake.body.len())
|
||||
}
|
||||
|
||||
pub fn simulate_actions(&mut self, actions: &BTreeMap<SnakeToken, Direction>) {
|
||||
pub fn simulate_actions(
|
||||
&mut self,
|
||||
actions: &BTreeMap<SnakeToken, Direction>,
|
||||
rng: &mut impl Rng,
|
||||
) {
|
||||
// move snakes
|
||||
for (token, snake) in &mut self.snakes {
|
||||
snake.perform_action(actions.get(token).copied().unwrap_or(Direction::Up));
|
||||
@ -152,7 +160,7 @@ impl Board {
|
||||
|
||||
// spawn new food
|
||||
if self.food.len() < usize::from(self.min_food)
|
||||
|| rand::thread_rng().gen_ratio(u32::from(self.food_chance), 100)
|
||||
|| rng.gen_ratio(u32::from(self.food_chance), 100)
|
||||
{
|
||||
let free_fields = (0..self.width)
|
||||
.flat_map(|x| (0..self.height).map(move |y| Coord { x, y }))
|
||||
@ -164,7 +172,7 @@ impl Board {
|
||||
.any(|body| body == coord)
|
||||
})
|
||||
.filter(|coord| self.food.contains(coord));
|
||||
if let Some(field) = free_fields.choose(&mut rand::thread_rng()) {
|
||||
if let Some(field) = free_fields.choose(rng) {
|
||||
self.food.insert(field);
|
||||
}
|
||||
}
|
||||
@ -172,7 +180,7 @@ impl Board {
|
||||
self.turn += 1;
|
||||
}
|
||||
|
||||
pub fn simulate_until(&mut self, mut exit: impl FnMut(&Self) -> bool) {
|
||||
pub fn simulate_until(&mut self, rng: &mut impl Rng, mut exit: impl FnMut(&Self) -> bool) {
|
||||
while !exit(self) {
|
||||
let actions = self
|
||||
.possible_actions()
|
||||
@ -180,15 +188,11 @@ impl Board {
|
||||
.map(|(token, actions)| {
|
||||
(
|
||||
*token,
|
||||
actions
|
||||
.iter()
|
||||
.choose(&mut rand::thread_rng())
|
||||
.copied()
|
||||
.unwrap_or(Direction::Up),
|
||||
actions.iter().choose(rng).copied().unwrap_or(Direction::Up),
|
||||
)
|
||||
})
|
||||
.collect();
|
||||
self.simulate_actions(&actions);
|
||||
self.simulate_actions(&actions, rng);
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
Loading…
Reference in New Issue
Block a user