diff options
Diffstat (limited to 'src/main.rs')
-rw-r--r-- | src/main.rs | 13 |
1 files changed, 10 insertions, 3 deletions
diff --git a/src/main.rs b/src/main.rs index e5dc3aa..3d3d980 100644 --- a/src/main.rs +++ b/src/main.rs @@ -1,4 +1,7 @@ extern crate zombot; +extern crate time; +use time::PreciseTime; + use zombot::*; use zombot::engine::command::Command; @@ -12,8 +15,8 @@ use std::fs::File; use std::io::prelude::*; use std::process; -fn choose_move(settings: &engine::settings::GameSettings, state: &engine::GameState) -> Command { - strategy::monte_carlo::choose_move(settings, state) +fn choose_move(settings: &engine::settings::GameSettings, state: &engine::GameState, start_time: &PreciseTime) -> Command { + strategy::monte_carlo::choose_move(settings, state, start_time) } @@ -25,6 +28,8 @@ fn write_command(filename: &str, command: Command) -> Result<(), Box<Error> > { fn main() { + let start_time = PreciseTime::now(); + println!("Reading in state.json file"); let (settings, state) = match json::read_state_from_file(STATE_PATH) { Ok(ok) => ok, @@ -33,7 +38,7 @@ fn main() { process::exit(1); } }; - let command = choose_move(&settings, &state); + let command = choose_move(&settings, &state, &start_time); match write_command(COMMAND_PATH, command) { Ok(()) => {} @@ -42,4 +47,6 @@ fn main() { process::exit(1); } } + + println!("Elapsed time: {}", start_time.to(PreciseTime::now())); } |