static void ScanIntoDB(string folderPath)
{
Console.WriteLine("Welcome to FIFA Statalyzer.");
Console.WriteLine("Processing screenshots located in " + folderPath);
string[] fileList = Directory.GetFiles(folderPath, "*.png");
SQLiteConnection dbConnection = DBManagement.InitializeDB();
foreach (string file in fileList)
{
string processedFile = ImageProcessing.ProcessScreenshot(file);
Dictionary <string, int> statsDict = new Dictionary <string, int>();
string[] legend = File.ReadAllLines(@".\cfg\legend.txt");
string ocrResult = (OCR.ReadImage(processedFile));
string ocrClean = OCR.CleanUp(ocrResult);
string[] ocrValues = ocrClean.Split(' ');
for (int i = 0; i < ocrValues.Length; i++)
{
statsDict.Add(legend[i], (int.Parse(ocrValues[i])));
}
int result = 0;
statsDict.TryGetValue("hGoals", out int hGoals);
statsDict.TryGetValue("aGoals", out int aGoals);
if (hGoals == aGoals)
{
result = 0;
} // Draw
else if (hGoals > aGoals)
{
result = 1;
} // Home win
else if (hGoals < aGoals)
{
result = 2;
} // Away win
statsDict.Add("result", result);
Console.WriteLine("Inserting these values into DB:");
foreach (var pair in statsDict)
{
Console.WriteLine("{0}: {1}", pair.Key, pair.Value);
}
StringBuilder commandBuilder = new StringBuilder("insert into fifa(");
foreach (var pair in statsDict)
{
commandBuilder.Append(pair.Key + ", ");
}
commandBuilder.Length -= 2;
commandBuilder.Append(") values (");
foreach (var pair in statsDict)
{
commandBuilder.Append(pair.Value + ", ");
}
commandBuilder.Length -= 2;
commandBuilder.Append(");");
string sqlCommand = commandBuilder.ToString();
DBManagement.ExecuteNonQuery(sqlCommand, dbConnection);
Console.WriteLine("Results inserted.");
}
dbConnection.Close();
Console.WriteLine("All images processed.");
}
static void Main(string[] args)
{
#if DEBUG
args = new[] { @"C:\test\active" };
#endif
// default path if none is specified
if (args == null || args.Length == 0)
{
args = new[] { @".\images\" };
}
if (args[0] == "dump" || args[0] == "DUMP")
{
DBManagement.DumpToText(DBManagement.InitializeDB());
}
else if (args[0] == "clear" || args[0] == "CLEAR")
{
DBManagement.ClearDB(DBManagement.InitializeDB());
}
else if (args[0] == "process" || args[0] == "PROCESS")
{
ImageProcessing.ProcessScreenshot(@"C:\test\img\img.png");
}
else
{
ScanIntoDB(args[0]);
}
DBManagement.DumpToText(DBManagement.InitializeDB());
DBManagement.ClearDB(DBManagement.InitializeDB());
Console.WriteLine("All done! Thanks for using Fifa Statalyzer! Press Enter to exit.");
Console.ReadLine();
}