From 7ec48d0d454499177b63bc5bd512a3a2d6baa839 Mon Sep 17 00:00:00 2001 From: Justin Wernick Date: Tue, 19 Apr 2022 21:26:49 +0200 Subject: Refile for merging repos --- 2018-tower-defence/src/main.rs | 55 ++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 55 insertions(+) create mode 100644 2018-tower-defence/src/main.rs (limited to '2018-tower-defence/src/main.rs') diff --git a/2018-tower-defence/src/main.rs b/2018-tower-defence/src/main.rs new file mode 100644 index 0000000..4fa0366 --- /dev/null +++ b/2018-tower-defence/src/main.rs @@ -0,0 +1,55 @@ +extern crate zombot; +extern crate time; +use time::{PreciseTime, Duration}; + +use zombot::*; +use zombot::engine::constants::*; +use zombot::engine::command::Command; + +use std::error::Error; + +const STATE_PATH: &str = "state.json"; + +const COMMAND_PATH: &str = "command.txt"; + +use std::fs::File; +use std::io::prelude::*; +use std::process; + +fn write_command(filename: &str, command: Command) -> Result<(), Box > { + let mut file = File::create(filename)?; + write!(file, "{}", command)?; + Ok(()) +} + +fn main() { + let start_time = PreciseTime::now(); + let max_time = Duration::milliseconds(MAX_TIME_MILLIS); + + let state = match input::json::read_bitwise_state_from_file(STATE_PATH) { + Ok(ok) => ok, + Err(error) => { + println!("Error while parsing JSON file: {}", error); + process::exit(1); + } + }; + + let command = if cfg!(feature = "static-opening") && state.round < strategy::static_opening::STATIC_OPENING_LENGTH { + strategy::static_opening::choose_move(&state) + } else if cfg!(feature = "full-monte-carlo-tree") { + strategy::monte_carlo_tree::choose_move(&state, start_time, max_time) + } else { + strategy::monte_carlo::choose_move(&state, start_time, max_time) + }; + + match write_command(COMMAND_PATH, command) { + Ok(()) => {} + Err(error) => { + println!("Error while writing command file: {}", error); + process::exit(1); + } + } + + println!("Elapsed time: {}", start_time.to(PreciseTime::now())); +} + -- cgit v1.2.3