OCR_Decode/OCR_Decode/DriverTyresWindow.cs

using System;
using System.Collections.Generic;
using System.Diagnostics;
using System.Drawing;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
using Tesseract;

namespace OCR_Decode
{
    internal class DriverTyresWindow : Window
    {
        private static Random rnd = new Random();
        int seed = rnd.Next(0,10000);

        public static Color SOFT_TYRE_COLOR = Color.FromArgb(0xff, 0x00, 0x00);
        public static Color MEDIUM_TYRE_COLOR = Color.FromArgb(0xf5, 0xbf, 0x00);
        public static Color HARD_TYRE_COLOR = Color.FromArgb(0xa4, 0xa5, 0xa8);
        public static Color INTER_TYRE_COLOR = Color.FromArgb(0x00, 0xa4, 0x2e);
        public static Color WET_TYRE_COLOR = Color.FromArgb(0x27, 0x60, 0xa6);
        public static Color EMPTY_COLOR = Color.FromArgb(0x20,0x20,0x20);

        public DriverTyresWindow(Bitmap image, Rectangle bounds) : base(image, bounds)
        {
            Name = "Tyres";
        }
        public override async Task<object> DecodePng()
        {
            //WindowImage.Save(Reader.DEBUG_DUMP_FOLDER + "testTyres" + rnd.Next(0, 1000) + ".png");
            return await GetTyreInfos();
        }
        private async Task<Tyre> GetTyreInfos()
        {
            Bitmap tyreZone = GetSmallBitmapFromBigOne(WindowImage, FindTyreZone());
            //tyreZone.Save(Reader.DEBUG_DUMP_FOLDER + "NewTyreZone_"+seed+".png");
            Tyre.Type type = Tyre.Type.Undefined;
            type = GetTyreTypeFromColor(OcrImage.GetAvgColorFromBitmap(tyreZone));
            int laps = -1;
            
            string number = await GetStringFromPng(tyreZone, Engine, "0123456789", OcrImage.WindowType.Tyre);
            try
            {
                laps = Convert.ToInt32(number);
            }
            catch
            {
                //We could not convert the number so its a letter so its 0 laps old
                laps = 0;
            }
            //tyreZone.Save(Reader.DEBUG_DUMP_FOLDER + "Tyre" + type + "Laps" + laps + '#' + rnd.Next(0, 1000) + ".png");
            return new Tyre(type, laps);
        }
        private Rectangle FindTyreZone()
        {
            Bitmap bmp = WindowImage;

            //bmp.Save(Reader.DEBUG_DUMP_FOLDER + "OldTyreZone_" + seed + ".png");

            int currentPosition = bmp.Width;
            int height = bmp.Height / 2;
            Color limitColor = Color.FromArgb(0x50, 0x50, 0x50);
            Color currentColor = Color.FromArgb(0, 0, 0);

            Size newWindowSize = new Size(bmp.Height - Convert.ToInt32((float)bmp.Height / 100f * 25f), bmp.Height - Convert.ToInt32((float)bmp.Height / 100f * 35f));

            while (currentColor.R <= limitColor.R && currentColor.G <= limitColor.G && currentColor.B <= limitColor.B && currentPosition > 0)
            {
                currentPosition--;
                currentColor = bmp.GetPixel(currentPosition, height);
            }

            //Its here to let the new window include a little bit of the right 
            int CorrectedX = currentPosition - (newWindowSize.Width) + Convert.ToInt32((float)newWindowSize.Width / 100f * 10f);
            int CorrectedY = Convert.ToInt32((float)newWindowSize.Height / 100f * 35f);
            if (CorrectedX <= 0)
                return new Rectangle(0, 0, newWindowSize.Width, newWindowSize.Height);

            return new Rectangle(CorrectedX, CorrectedY, newWindowSize.Width, newWindowSize.Height);
        }
        //This method has been created with the help of chatGPT
        public Tyre.Type GetTyreTypeFromColor(Color inputColor)
        {
            Tyre.Type type = Tyre.Type.Undefined;
            List<Color> colors = new List<Color>();
            colors.Add(SOFT_TYRE_COLOR);
            colors.Add(MEDIUM_TYRE_COLOR);
            colors.Add(HARD_TYRE_COLOR);
            colors.Add(INTER_TYRE_COLOR);
            colors.Add(WET_TYRE_COLOR);
            colors.Add(EMPTY_COLOR);

            Color closestColor = colors[0];
            int closestDistance = int.MaxValue;
            foreach (Color color in colors)
            {
                int distance = Math.Abs(color.R - inputColor.R) + Math.Abs(color.G - inputColor.G) + Math.Abs(color.B - inputColor.B);
                if (distance < closestDistance)
                {
                    closestColor = color;
                    closestDistance = distance;
                }
            }

            //Bitmap colorBmp = new Bitmap(200, 100);
            //Graphics g = Graphics.FromImage(colorBmp);
            //g.FillRectangle(new SolidBrush(closestColor), new Rectangle(0, 0, colorBmp.Width, colorBmp.Height));
            //colorBmp.Save(Reader.DEBUG_DUMP_FOLDER+"AVGCOLOR_"+closestColor.ToString()+"_"+seed+".png");

            //We cant use a switch as the colors cant be constants ...
            if (closestColor == SOFT_TYRE_COLOR)
                type = Tyre.Type.Soft;
            if (closestColor == MEDIUM_TYRE_COLOR)
                type = Tyre.Type.Medium;
            if (closestColor == HARD_TYRE_COLOR)
                type = Tyre.Type.Hard;
            if (closestColor == INTER_TYRE_COLOR)
                type = Tyre.Type.Inter;
            if (closestColor == WET_TYRE_COLOR)
                type = Tyre.Type.Wet;
            if (closestColor == EMPTY_COLOR)
                return Tyre.Type.Undefined;

            return type;
        }
    }
}