Outline of MCTS

1 files changed, 7 insertions, 2 deletions

diff --git a/src/main.rs b/src/main.rs
index d6d9a4c..c24565a 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -2,14 +2,19 @@ use std::io::prelude::*;
 use std::io::stdin;
 use std::path::Path;
 
+use time::{Duration, PreciseTime};
+
 use steam_powered_wyrm::command::Command;
 use steam_powered_wyrm::strategy::choose_move;
 use steam_powered_wyrm::json;
 use steam_powered_wyrm::game;
 
 fn main() {
+    let max_time = Duration::milliseconds(950);
     let mut game_board = None;
     for line in stdin().lock().lines() {
+        let start_time = PreciseTime::now();
+        
         let round_number = line.expect("Failed to read line from stdin: {}");
         
         let command =
@@ -18,13 +23,13 @@ fn main() {
                     match &mut game_board {
                         None => {
                             let new_board = game::GameBoard::new(json_state);
-                            let command = choose_move(&new_board);
+                            let command = choose_move(&new_board, &start_time, max_time);
                             game_board = Some(new_board);
                             command
                         },
                         Some(game_board) => {
                             game_board.update(json_state);
-                            choose_move(&game_board)
+                            choose_move(&game_board, &start_time, max_time)
                         }
                     }
                 },