1 files changed, 14 insertions, 2 deletions
diff --git a/main.cpp b/main.cpp
index c53f5ae..61cd573 100644
--- a/main.cpp
+++ b/main.cpp
@@ -2,18 +2,28 @@
 #include <cstdlib>
 #include <sys/time.h>
 #include "board.h"
-#include "mc.h"
+#include "ai_mc.h"
+#include "ai_rand.h"
 #include "ui.h"
 #include "util.h"
 
 using namespace std;
 
+#define STR_(x) #x
+#define STR(x) STR_(x)
+
+#ifndef AI
+#define AI MC
+#endif
+
 
 int main() {
 	struct timeval tv;
 	gettimeofday(&tv, nullptr);
 	srandom(tv.tv_sec * 1000000U + tv.tv_usec);
 
+	cerr << "Using AI " << STR(AI) << endl;
+
 	Board bd = Board::makeEmpty();
 	cerr << "Initial stone at " << Idx(BSZ * BMID + BMID) << endl;
 	bd.put(BSZ * BMID + BMID, bd.bag.drawRandom());
@@ -30,7 +40,7 @@ int main() {
 			cout << "YOUR TURN." << endl;
 			idx = UI::getMove(bd);
 		} else {
-			idx = MC::calcMove(bd, onturn);
+			idx = AI::calcMove(bd, onturn);
 		}
 
 		uint8_t clr = bd.bag.drawRandom();
@@ -45,4 +55,6 @@ int main() {
 
 		onturn = NEXTTURN(onturn);
 	}
+
+	cout << "TIE" << endl;
 }