1 files changed, 123 insertions, 0 deletions
diff --git a/bayrate/bayrate.cpp b/bayrate/bayrate.cpp
new file mode 100644
index 0000000..f870e62
--- /dev/null
+++ b/bayrate/bayrate.cpp
@@ -0,0 +1,123 @@
+/*************************************************************************************
+
+	Copyright 2010 Philip Waldron
+	
+    This file is part of BayRate.
+
+    BayRate is free software: you can redistribute it and/or modify
+    it under the terms of the GNU General Public License as published by
+    the Free Software Foundation, either version 3 of the License, or
+    (at your option) any later version.
+
+    BayRate is distributed in the hope that it will be useful,
+    but WITHOUT ANY WARRANTY; without even the implied warranty of
+    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+    GNU General Public License for more details.
+
+    You should have received a copy of the GNU General Public License
+    along with BayRate.  If not, see <http://www.gnu.org/licenses/>.
+    
+***************************************************************************************/
+
+#include <iostream>
+#include <string>
+#include <vector>
+#include <algorithm>
+#include <mysql++/mysql++.h>
+#include <boost/date_time/gregorian/gregorian.hpp>
+#include "db.h"
+#include "tdListEntry.h"
+#include "collection.h"
+
+using namespace std;
+using namespace boost::gregorian;
+
+int main(int argc, char *argv[])
+{
+	
+ 	databaseConnection db;
+ 	vector<string> tournamentUpdateList;	// The list of tournaments after the tournamentCascadeDate
+	boost::gregorian::date tournamentCascadeDate;		
+ 	map<int, tdListEntry> tdList;
+ 	map<string, bool> argList;
+ 	collection c;
+
+	for (int i=1; i<argc; i++) 
+		argList[string(argv[i])] = true;
+
+ 	// Establish the database connection			
+	if ( !db.makeConnection() ) {	
+		std::cerr << "Fatal error:  db.makeConnection() failed." << std::endl;
+		exit (1);
+	}
+
+	db.excludeBogusGameData();
+
+	// Figure out which tournaments need rating
+	if ( !db.getTournamentUpdateList(tournamentUpdateList, tournamentCascadeDate) ) {
+		std::cerr << "Fatal error: db.getTournamentUpdateList() failed." << std::endl;
+		exit(1);
+	}
+	
+	if (tournamentUpdateList.size() == 0) {
+		cout << "No tournaments to update." << endl;
+		exit(1);
+	}	
+	else {
+		cout << "Updating all tournaments after " << to_iso_extended_string(tournamentCascadeDate) << endl;
+	}
+	
+	// Get the TDList immediately prior to the earliest unrated game
+	if (!db.getTDList(tournamentCascadeDate, tdList)) {
+		std::cerr << "Fatal error: db.getTDList() failed." << std::endl;
+		exit(1);
+	}
+	else {
+		cout << "Downloaded TDList" << endl;
+	}
+	
+	for (vector<string>::iterator It=tournamentUpdateList.begin(); It!=tournamentUpdateList.end(); It++) {
+		cout << "Processing " << *It << endl;
+		
+		c.reset();	
+		db.getTournamentInfo(*It, c);
+
+		if (c.gameList.size() == 0)
+			continue;
+		
+		c.initSeeding(tdList);
+		
+		// Start with the fast rating algorithm.  If it fails, then go for the simplex method as a backup. 
+		if (c.calc_ratings_fdf() != 0) {
+			if (c.calc_ratings() != 0) {
+				cerr << "Fatal error processing tournament " << *It << endl;
+				exit(1); 
+			}	
+		}
+		
+		// Copy the new ratings into the internal TDList for the next tournament update
+		for (map<int, player>::iterator It = c.playerHash.begin(); It != c.playerHash.end(); It++) {
+			tdList[It->second.id].id     = It->second.id;			
+			tdList[It->second.id].rating = It->second.rating;
+			tdList[It->second.id].sigma  = It->second.sigma;
+			tdList[It->second.id].lastRatingDate = c.tournamentDate;
+			tdList[It->second.id].ratingUpdated = true;
+		
+			cout << It->second.id << '\t' << It->second.rating << '\t' << It->second.sigma << endl;
+		}				
+	    cout << endl;
+	    
+		// Update database
+		if (argList.find(string("--commit")) != argList.end()) {
+			cout << "Committing results to database...";		
+			db.syncNewRatings(c);		
+			cout << "done." << endl;
+		}
+	}
+	cout << "Done ratings" << endl;
+	
+	for (map<int, tdListEntry>::iterator tdListIt = tdList.begin(); tdListIt != tdList.end(); tdListIt++) {
+		if (tdListIt->second.ratingUpdated)
+			cout << tdListIt->second.id << '\t' << tdListIt->second.rating << '\t' << tdListIt->second.sigma << endl;
+	}
+}