298 lines
14 KiB
C#
298 lines
14 KiB
C#
using System;
|
|
using System.Collections.Generic;
|
|
using System.Linq;
|
|
using System.Text;
|
|
using System.Threading.Tasks;
|
|
using System.Drawing;
|
|
using System.IO;
|
|
using System.Text.Json;
|
|
using System.Windows.Forms;
|
|
|
|
namespace OCR_Decode
|
|
{
|
|
internal class Reader
|
|
{
|
|
private string _configFile;
|
|
private string _imagesFolder;
|
|
private List<string> _drivers;
|
|
private List<Zone> _mainZones;
|
|
private Bitmap FullImage;
|
|
public string ConfigFile { get => _configFile; private set => _configFile = value; }
|
|
public string ImagesFolder { get => _imagesFolder; private set => _imagesFolder = value; }
|
|
public List<string> Drivers { get => _drivers; private set => _drivers = value; }
|
|
public List<Zone> MainZones { get => _mainZones; set => _mainZones = value; }
|
|
|
|
const string DEFAULT_IMAGE_NAME = "screen_";
|
|
public const string DEBUG_DUMP_FOLDER = @"C:\Users\Moi\Desktop\imgDump\Decode\";
|
|
const int NUMBER_OF_DRIVERS = 20;
|
|
|
|
public Reader(string configFile, string imageFolder)
|
|
{
|
|
ConfigFile = configFile;
|
|
ImagesFolder = imageFolder;
|
|
Load(9);
|
|
}
|
|
private void Load(int imageNumber)
|
|
{
|
|
MainZones = new List<Zone>();
|
|
try
|
|
{
|
|
FullImage = (Bitmap)Image.FromFile(ImagesFolder + DEFAULT_IMAGE_NAME + imageNumber + ".png");
|
|
}
|
|
catch
|
|
{
|
|
MessageBox.Show("Trouble reaching the image");
|
|
//Maybe a bit to harsh, Ill see what I can do to soft this a bit
|
|
Application.Exit();
|
|
}
|
|
|
|
Zone MainZone;
|
|
try
|
|
{
|
|
using (var streamReader = new StreamReader(ConfigFile))
|
|
{
|
|
var jsonText = streamReader.ReadToEnd();
|
|
var jsonDocument = JsonDocument.Parse(jsonText);
|
|
|
|
var driversNames = jsonDocument.RootElement.GetProperty("Drivers");
|
|
Drivers = new List<string>();
|
|
|
|
foreach (var nameElement in driversNames.EnumerateArray())
|
|
{
|
|
Drivers.Add(nameElement.GetString());
|
|
}
|
|
|
|
var mainProperty = jsonDocument.RootElement.GetProperty("Main");
|
|
Point MainPosition = new Point(mainProperty.GetProperty("x").GetInt32(), mainProperty.GetProperty("y").GetInt32());
|
|
Size MainSize = new Size(mainProperty.GetProperty("width").GetInt32(), mainProperty.GetProperty("height").GetInt32());
|
|
Rectangle MainRectangle = new Rectangle(MainPosition, MainSize);
|
|
MainZone = new Zone(FullImage, MainRectangle);
|
|
|
|
var zones = mainProperty.GetProperty("Zones");
|
|
var driverZone = zones[0].GetProperty("DriverZone");
|
|
|
|
Point FirstZonePosition = new Point(driverZone.GetProperty("x").GetInt32(), driverZone.GetProperty("y").GetInt32());
|
|
Size FirstZoneSize = new Size(driverZone.GetProperty("width").GetInt32(), driverZone.GetProperty("height").GetInt32());
|
|
|
|
var windows = driverZone.GetProperty("Windows");
|
|
|
|
//var driverPosition = windows.GetProperty("Position");
|
|
var driverPosition = windows[0].GetProperty("Position");
|
|
Size driverPositionArea = new Size(driverPosition.GetProperty("width").GetInt32(), FirstZoneSize.Height);
|
|
Point driverPositionPosition = new Point(driverPosition.GetProperty("x").GetInt32(), driverPosition.GetProperty("y").GetInt32());
|
|
|
|
var driverGapToLeader = windows[0].GetProperty("GapToLeader");
|
|
Size driverGapToLeaderArea = new Size(driverGapToLeader.GetProperty("width").GetInt32(), FirstZoneSize.Height);
|
|
Point driverGapToLeaderPosition = new Point(driverGapToLeader.GetProperty("x").GetInt32(), driverGapToLeader.GetProperty("y").GetInt32());
|
|
|
|
var driverLapTime = windows[0].GetProperty("LapTime");
|
|
Size driverLapTimeArea = new Size(driverLapTime.GetProperty("width").GetInt32(), FirstZoneSize.Height);
|
|
Point driverLapTimePosition = new Point(driverLapTime.GetProperty("x").GetInt32(), driverLapTime.GetProperty("y").GetInt32());
|
|
|
|
|
|
var driverDrs = windows[0].GetProperty("Drs");
|
|
Size driverDrsArea = new Size(driverDrs.GetProperty("width").GetInt32(), FirstZoneSize.Height);
|
|
Point driverDrsPosition = new Point(driverDrs.GetProperty("x").GetInt32(), driverDrs.GetProperty("y").GetInt32());
|
|
|
|
var driverTyres = windows[0].GetProperty("Tyres");
|
|
Size driverTyresArea = new Size(driverTyres.GetProperty("width").GetInt32(), FirstZoneSize.Height);
|
|
Point driverTyresPosition = new Point(driverTyres.GetProperty("x").GetInt32(), driverTyres.GetProperty("y").GetInt32());
|
|
|
|
var driverName = windows[0].GetProperty("Name");
|
|
Size driverNameArea = new Size(driverName.GetProperty("width").GetInt32(), FirstZoneSize.Height);
|
|
Point driverNamePosition = new Point(driverName.GetProperty("x").GetInt32(), driverName.GetProperty("y").GetInt32());
|
|
|
|
var driverSector1 = windows[0].GetProperty("Sector1");
|
|
Size driverSector1Area = new Size(driverSector1.GetProperty("width").GetInt32(), FirstZoneSize.Height);
|
|
Point driverSector1Position = new Point(driverSector1.GetProperty("x").GetInt32(), driverSector1.GetProperty("y").GetInt32());
|
|
|
|
var driverSector2 = windows[0].GetProperty("Sector2");
|
|
Size driverSector2Area = new Size(driverSector2.GetProperty("width").GetInt32(), FirstZoneSize.Height);
|
|
Point driverSector2Position = new Point(driverSector2.GetProperty("x").GetInt32(), driverSector2.GetProperty("y").GetInt32());
|
|
|
|
var driverSector3 = windows[0].GetProperty("Sector3");
|
|
Size driverSector3Area = new Size(driverSector3.GetProperty("width").GetInt32(), FirstZoneSize.Height);
|
|
Point driverSector3Position = new Point(driverSector3.GetProperty("x").GetInt32(), driverSector3.GetProperty("y").GetInt32());
|
|
|
|
float offset = (((float)MainZone.ZoneImage.Height - (float)(Drivers.Count * FirstZoneSize.Height)) / (float)Drivers.Count);
|
|
|
|
for (int i = 0; i < NUMBER_OF_DRIVERS; i++)
|
|
{
|
|
Point tmpPos = new Point(0, FirstZonePosition.Y + i * FirstZoneSize.Height - Convert.ToInt32(i * offset) /*- (i* (FirstZoneSize.Height / 32))*/);
|
|
Zone newDriverZone = new Zone(MainZone.ZoneImage, new Rectangle(tmpPos, FirstZoneSize));
|
|
|
|
newDriverZone.AddWindow(new DriverPositionWindow(newDriverZone.ZoneImage, new Rectangle(driverPositionPosition, driverPositionArea)));
|
|
newDriverZone.AddWindow(new DriverGapToLeaderWindow(newDriverZone.ZoneImage, new Rectangle(driverGapToLeaderPosition, driverGapToLeaderArea)));
|
|
newDriverZone.AddWindow(new DriverLapTimeWindow(newDriverZone.ZoneImage, new Rectangle(driverLapTimePosition, driverLapTimeArea)));
|
|
newDriverZone.AddWindow(new DriverDrsWindow(newDriverZone.ZoneImage, new Rectangle(driverDrsPosition, driverDrsArea)));
|
|
newDriverZone.AddWindow(new DriverTyresWindow(newDriverZone.ZoneImage, new Rectangle(driverTyresPosition, driverTyresArea)));
|
|
newDriverZone.AddWindow(new DriverNameWindow(newDriverZone.ZoneImage, new Rectangle(driverNamePosition, driverNameArea)));
|
|
newDriverZone.AddWindow(new DriverSector1Window(newDriverZone.ZoneImage, new Rectangle(driverSector1Position, driverSector1Area)));
|
|
newDriverZone.AddWindow(new DriverSector2Window(newDriverZone.ZoneImage, new Rectangle(driverSector2Position, driverSector2Area)));
|
|
newDriverZone.AddWindow(new DriverSector3Window(newDriverZone.ZoneImage, new Rectangle(driverSector3Position, driverSector3Area)));
|
|
|
|
MainZone.AddZone(newDriverZone);
|
|
}
|
|
|
|
//MessageBox.Show("We have a main zone with " + MainZone.Zones.Count() + " Driver zones with " + MainZone.Zones[4].Windows.Count() + " windows each and we have " + Drivers.Count + " drivers");
|
|
MainZones.Add(MainZone);
|
|
}
|
|
}
|
|
catch (IOException ex)
|
|
{
|
|
MessageBox.Show("Error reading JSON file: " + ex.Message);
|
|
}
|
|
catch (JsonException ex)
|
|
{
|
|
MessageBox.Show("Invalid JSON format: " + ex.Message);
|
|
}
|
|
}
|
|
public void ChangeImage(int imageNumber)
|
|
{
|
|
Bitmap img = null;
|
|
string imagePath = ImagesFolder + DEFAULT_IMAGE_NAME + imageNumber + ".png";
|
|
try
|
|
{
|
|
img = (Bitmap)Image.FromFile(imagePath);
|
|
}
|
|
catch
|
|
{
|
|
MessageBox.Show("Unable to reach the image at " + imagePath);
|
|
}
|
|
if (img != null)
|
|
{
|
|
FullImage = img;
|
|
foreach (Zone z in MainZones)
|
|
{
|
|
z.Image = img;
|
|
}
|
|
}
|
|
}
|
|
public string Decode(int idImage)
|
|
{
|
|
string result = "";
|
|
ChangeImage(idImage);
|
|
List<List<Object>> mainResults = new List<List<Object>>();
|
|
|
|
//Decode
|
|
for (int mainZoneId = 0; mainZoneId < MainZones.Count;mainZoneId ++)
|
|
{
|
|
switch (mainZoneId)
|
|
{
|
|
case 0:
|
|
//Main Zone
|
|
foreach (Zone z in MainZones[mainZoneId].Zones)
|
|
{
|
|
mainResults.Add(z.Decode(Drivers));
|
|
}
|
|
break;
|
|
//Next there will be the title and laps added
|
|
}
|
|
|
|
}
|
|
|
|
//Display
|
|
foreach (List<Object> objects in mainResults)
|
|
{
|
|
for (int objId = 0; objId < objects.Count; objId++)
|
|
{
|
|
switch (objId)
|
|
{
|
|
case 0:
|
|
//Position
|
|
result += "Position : " + (int)objects[objId] + " ";
|
|
break;
|
|
case 1:
|
|
//Gap
|
|
result += "Gap to leader : " + ConvertMsToTime((int)objects[objId]) + " ";
|
|
break;
|
|
case 2:
|
|
//LapTime
|
|
result += "Lap time : " + ConvertMsToTime((int)objects[objId]) + " ";
|
|
break;
|
|
case 3:
|
|
//DRS
|
|
result += "DRS : " + (bool)objects[objId] + "";
|
|
break;
|
|
case 4:
|
|
//Tyres
|
|
Tyre tyre = (Tyre)objects[objId];
|
|
result += "Tyre : " + tyre.Coumpound + " laps with the tyre : " + tyre.NumberOfLaps + " ";
|
|
break;
|
|
case 5:
|
|
//Name
|
|
result += "Driver name : " + (string)objects[objId] + " ";
|
|
break;
|
|
case 6:
|
|
//Sector 1
|
|
result += "Sector 1 : " + ConvertMsToTime((int)objects[objId]) + " ";
|
|
break;
|
|
case 7:
|
|
//Sector 1
|
|
result += "Sector 2 : " + ConvertMsToTime((int)objects[objId]) + " ";
|
|
break;
|
|
case 8:
|
|
//Sector 1
|
|
result += "Sector 3 : " + ConvertMsToTime((int)objects[objId]) + " ";
|
|
break;
|
|
default:
|
|
result += "Unknown source : " + objects[objId].ToString();
|
|
break;
|
|
}
|
|
result += Environment.NewLine;
|
|
}
|
|
}
|
|
|
|
return result;
|
|
}
|
|
public static string ConvertMsToTime(int amountOfMs)
|
|
{
|
|
//Convert.ToInt32 would round upand I dont want that
|
|
int minuts = (int)((float)amountOfMs / (1000f * 60f));
|
|
int seconds = (int)((amountOfMs - (minuts*60f*1000f))/1000);
|
|
int ms = amountOfMs - ((minuts * 60 * 1000) + (seconds * 1000));
|
|
|
|
return minuts + ":" + seconds.ToString("00") + ":" + ms.ToString("000");
|
|
}
|
|
public Bitmap Draw(int idImage)
|
|
{
|
|
Bitmap result;
|
|
try
|
|
{
|
|
result = (Bitmap)Image.FromFile(ImagesFolder + DEFAULT_IMAGE_NAME + idImage + ".png");
|
|
}
|
|
catch
|
|
{
|
|
MessageBox.Show("Image could not be found");
|
|
return null;
|
|
}
|
|
|
|
Graphics g = Graphics.FromImage(result);
|
|
|
|
foreach (Zone z in MainZones)
|
|
{
|
|
int count = 0;
|
|
foreach (Zone zz in z.Zones)
|
|
{
|
|
string driverFolder = DEBUG_DUMP_FOLDER + "driver" + count + "\\";
|
|
if (!Directory.Exists(driverFolder))
|
|
Directory.CreateDirectory(driverFolder);
|
|
|
|
zz.ZoneImage.Save(driverFolder + "FullImage.png");
|
|
|
|
g.DrawRectangle(Pens.Red, z.Bounds);
|
|
foreach (Window w in zz.Windows)
|
|
{
|
|
g.DrawRectangle(Pens.Blue, new Rectangle(z.Bounds.X + zz.Bounds.X, z.Bounds.Y + zz.Bounds.Y, zz.Bounds.Width, zz.Bounds.Height));
|
|
}
|
|
|
|
count++;
|
|
}
|
|
}
|
|
|
|
return result;
|
|
}
|
|
}
|
|
}
|
|
|