109 lines
4.2 KiB
C#
109 lines
4.2 KiB
C#
using System;
|
|
using System.Collections.Generic;
|
|
using System.Drawing;
|
|
using System.Linq;
|
|
using System.Text;
|
|
using System.Threading.Tasks;
|
|
|
|
namespace Test_Merge
|
|
{
|
|
public class Zone
|
|
{
|
|
private Rectangle _bounds;
|
|
private List<Zone> _zones;
|
|
private List<Window> _windows;
|
|
private Bitmap _image;
|
|
|
|
public Bitmap ZoneImage
|
|
{
|
|
get
|
|
{
|
|
//This little trickery lets you have the image that the zone sees
|
|
Bitmap sample = new Bitmap(Bounds.Width, Bounds.Height);
|
|
Graphics g = Graphics.FromImage(sample);
|
|
g.DrawImage(Image, new Rectangle(0, 0, sample.Width, sample.Height), Bounds, GraphicsUnit.Pixel);
|
|
return sample;
|
|
}
|
|
}
|
|
public Bitmap Image
|
|
{
|
|
get { return _image; }
|
|
set
|
|
{
|
|
//It automatically sets the image for the contained windows and zones
|
|
_image = Image;
|
|
foreach (Window w in Windows)
|
|
{
|
|
w.Image = ZoneImage;
|
|
}
|
|
foreach (Zone z in Zones)
|
|
{
|
|
z.Image = Image;
|
|
}
|
|
}
|
|
}
|
|
|
|
public Rectangle Bounds { get => _bounds; protected set => _bounds = value; }
|
|
public List<Zone> Zones { get => _zones; protected set => _zones = value; }
|
|
public List<Window> Windows { get => _windows; protected set => _windows = value; }
|
|
|
|
public Zone(Bitmap image, Rectangle bounds)
|
|
{
|
|
Windows = new List<Window>();
|
|
Zones = new List<Zone>();
|
|
|
|
//You cant set the image in the CTOR because the processing is impossible at first initiation
|
|
_image = image;
|
|
Bounds = bounds;
|
|
}
|
|
/// <summary>
|
|
/// Adds a zone to the list of zones
|
|
/// </summary>
|
|
/// <param name="zone">The zone you want to add</param>
|
|
public virtual void AddZone(Zone zone)
|
|
{
|
|
Zones.Add(zone);
|
|
}
|
|
/// <summary>
|
|
/// Add a window to the list of windows
|
|
/// </summary>
|
|
/// <param name="window">the window you want to add</param>
|
|
public virtual void AddWindow(Window window)
|
|
{
|
|
Windows.Add(window);
|
|
}
|
|
/// <summary>
|
|
/// Calls all the windows to do OCR and to give back the results so we can send them to the model
|
|
/// </summary>
|
|
/// <param name="driverList">A list of all the driver in the race to help with text recognition</param>
|
|
/// <returns>A driver data object that contains all the infos about a driver</returns>
|
|
public virtual async Task<DriverData> Decode(List<string> driverList)
|
|
{
|
|
DriverData result = new DriverData();
|
|
Parallel.ForEach(Windows, async w =>
|
|
{
|
|
// A switch would be prettier but I dont think its supported in this C# version
|
|
if (w is DriverNameWindow)
|
|
result.Name = (string)await (w as DriverNameWindow).DecodePng(driverList);
|
|
if (w is DriverDrsWindow)
|
|
result.DRS = (bool)await (w as DriverDrsWindow).DecodePng();
|
|
if (w is DriverGapToLeaderWindow)
|
|
result.GapToLeader = (int)await (w as DriverGapToLeaderWindow).DecodePng();
|
|
if (w is DriverLapTimeWindow)
|
|
result.LapTime = (int)await (w as DriverLapTimeWindow).DecodePng();
|
|
if (w is DriverPositionWindow)
|
|
result.Position = (int)await (w as DriverPositionWindow).DecodePng();
|
|
if (w is DriverSector1Window)
|
|
result.Sector1 = (int)await (w as DriverSector1Window).DecodePng();
|
|
if (w is DriverSector2Window)
|
|
result.Sector2 = (int)await (w as DriverSector2Window).DecodePng();
|
|
if (w is DriverSector3Window)
|
|
result.Sector3 = (int)await (w as DriverSector3Window).DecodePng();
|
|
if (w is DriverTyresWindow)
|
|
result.CurrentTyre = (Tyre)await (w as DriverTyresWindow).DecodePng();
|
|
});
|
|
return result;
|
|
}
|
|
}
|
|
}
|