Mostly messing with "forfax" AI

[progcomp2012.git] / samples / forfax / main.cpp

diff --git a/samples/forfax/main.cpp b/samples/forfax/main.cpp
index 2b31fee..f3678e8 100644 (file)
--- a/samples/forfax/main.cpp
+++ b/samples/forfax/main.cpp
@@ -1,3 +1,12 @@
+/**
+ * "forfax", a sample Stratego AI for the UCC Programming Competition 2012
+ * The main function for the "forfax" AI program
+ * @author Sam Moore (matches) [SZM]
+ * @website http://matches.ucc.asn.au/stratego
+ * @email [email protected] or [email protected]
+ * @git git.ucc.asn.au/progcomp2012.git
+ */
+
 #include <cstdlib>
 #include <iostream>
 
@@ -6,21 +15,73 @@
 using namespace std;
 
 #include <stdio.h>
-void quit();
+
+/**
+ * The AI
+ */
 Forfax forfax;
+
+/**
+ * The main function
+ * @param argc the number of arguments
+ * @param argv the arguments
+ * @returns exit code 0 for success, something else for error
+ * Do I really need to tell you this?
+ */
 int main(int argc, char ** argv)
 {
        setbuf(stdin, NULL);
        setbuf(stdout, NULL);
 
-       atexit(&quit);
 
-       if (!forfax.Setup())
-               exit(EXIT_SUCCESS);
+       Forfax::Status move = forfax.Setup();
+
+       
+       while (move == Forfax::OK)
+       {
+               move = forfax.MakeMove();
+       }
+
+       switch (move)
+       {
+               case Forfax::OK:
+                       cerr << argv[0] << " Error - Should never see this!\n";
+                       break;
+               case Forfax::NO_NEWLINE:
+                       cerr << argv[0] << " Error - Expected a new line!\n";
+                       break;
+               case Forfax::EXPECTED_ATTACKER:
+                       cerr << argv[0] << " Error - Attacking piece does not exist on board!\n";
+                       break;
+               case Forfax::UNEXPECTED_DEFENDER:
+                       cerr << argv[0] << " Error - Unexpected defending piece on board!\n";
+                       break;
+               case Forfax::NO_ATTACKER:
+                       cerr << argv[0] << " Error - Couldn't find attacker in list of pieces!\n";
+                       break;
+               case Forfax::NO_DEFENDER:
+                       cerr << argv[0] << " Error - Couldn't find defender in list of pieces!\n";
+                       break;
 
-       while (forfax.MakeMove());
+               case Forfax::COLOUR_MISMATCH:
+                       cerr << argv[0] << " Error - Colour of attacker and defender are the same!\n";
+                       break;
+               case Forfax::INVALID_QUERY:
+                       cerr << argv[0] << " Error - Query did not make sense\n";
+                       break;
+               case Forfax::VICTORY:
+                       cerr << argv[0] << " Game end - VICTORY!\n";
+                       break;          
+               case Forfax::BOARD_ERROR:
+                       cerr << argv[0] << " Error - An error occurred with the board!\n";
+                       break;
+       }
 
-       cerr << "Forfax threw a hissy fit, and exited!\n";
+       cerr << "Final board state:\n";
+       forfax.PrintBoard(cerr);
+
+       cerr << "Forfax is now exiting!\n";
+       
        
        exit(EXIT_SUCCESS);
        return 0;
@@ -28,7 +89,4 @@ int main(int argc, char ** argv)
        
 }
 
-void quit()
-{
-       cerr << " Forfax quit\n";
-}
+