From 02256124fbd57d88effd02c046093ecaf73b77e3 Mon Sep 17 00:00:00 2001
From: Justin Worthe <justin@worthe-it.co.za>
Date: Sat, 12 May 2018 22:37:48 +0200
Subject: Limited bot to run within the 2 second window

---
 src/strategy/monte_carlo.rs | 15 +++++++++++----
 src/strategy/sample.rs      |  4 +++-
 2 files changed, 14 insertions(+), 5 deletions(-)

(limited to 'src/strategy')

diff --git a/src/strategy/monte_carlo.rs b/src/strategy/monte_carlo.rs
index ba66d48..86960eb 100644
--- a/src/strategy/monte_carlo.rs
+++ b/src/strategy/monte_carlo.rs
@@ -6,21 +6,28 @@ use rand::{thread_rng, Rng};
 use std::process;
 const MAX_MOVES: u16 = 400;
 
+use time::{Duration, PreciseTime};
+
 // TODO Round start time here
-pub fn choose_move(settings: &GameSettings, state: &GameState) -> Command {
+pub fn choose_move(settings: &GameSettings, state: &GameState, start_time: &PreciseTime) -> Command {
     println!("Using MONTE_CARLO strategy");
+
+    //just under the max of 2 seconds, to avoid issues like overhead in the bot being run, and we still need to write the result of this to file
+    let max_time = Duration::milliseconds(1950);
     
     let mut rng = thread_rng();
     let mut command_scores = CommandScore::init_command_scores(settings, state);
 
-    // TODO Repeat this until time is out
-    for _ in 0..1000 {
+    loop {
         for mut score in &mut command_scores {
             simulate_to_endstate(score, settings, state, &mut rng);
         }
+        if start_time.to(PreciseTime::now()) > max_time {
+            break;
+        }
     }
 
-    println!("{:?}", command_scores);
+    println!("{:#?}", command_scores);
     let command = command_scores.iter().max_by_key(|&c| c.win_ratio());
     
     match command {
diff --git a/src/strategy/sample.rs b/src/strategy/sample.rs
index 2dad924..72ab66c 100644
--- a/src/strategy/sample.rs
+++ b/src/strategy/sample.rs
@@ -1,9 +1,11 @@
 use engine;
 use engine::command::*;
 
+use time::PreciseTime;
+
 use rand::{thread_rng, Rng};
 
-pub fn choose_move(settings: &engine::settings::GameSettings, state: &engine::GameState) -> Command {
+pub fn choose_move(settings: &engine::settings::GameSettings, state: &engine::GameState, _start_time: &PreciseTime) -> Command {
     let mut rng = thread_rng();
     
     if state.player.can_afford_defence_buildings(settings) {
-- 
cgit v1.2.3