check point

VisionBuilder.UI.Ringbuffer/VisionBuilderRingbuffer.cs

@@ -0,0 +1,127 @@
+using Inspectron.Fastbuffer;
+using Inspectron.Fastbuffer.Filesystems;
+using System.Drawing;
+using System.Drawing.Imaging;
+using OpenCvSharp;
+using VisionBuilder.UI.Common.Commands;
+using VisionBuilder.UI.Common.RecipeProcessing;
+
+namespace Lindt.Colorballs.Duo.Utils;
+
+public class VisionBuilderRingbuffer
+{
+    private readonly VisionBuilderRingbufferSettings _settings;
+    private readonly IRecognitionControl _recognitionControl;
+
+    public VisionBuilderRingbuffer(VisionBuilderRingbufferSettings settings, IRecognitionControl recognitionControl)
+    {
+        _settings = settings;
+        _recognitionControl = recognitionControl;
+        _recognitionControl.SessionStarted += _recognitionControl_SessionStarted;
+        _recognitionControl.ImageProcessed += _recognitionControl_ImageProcessed;
+    }
+
+    private void _recognitionControl_ImageProcessed(VisionBuilder.UI.Common.Commands.ImageProcessedEvent obj)
+    {
+        if (obj.HasError)
+        {
+            WriteBadResult(obj);
+            return;
+        }
+        WriteGoodResult(obj);
+    }
+
+    private void _recognitionControl_SessionStarted(VisionBuilder.UI.Common.Commands.SessionStartedEvent obj)
+    {
+        NewProduct(obj.RecipeName);
+    }
+
+    private IsolatedRingbuffer _goodRingbufferIsolated;
+    private string _path;
+    private DateTime _startTime;
+    private string _product;
+    
+
+    public void Dispose()
+    {
+
+    }
+
+
+    public void WriteGoodResult(ImageProcessedEvent result)
+    {
+        if (result.ImageOriginal == null) return;
+        var filename = PathSettingsUtils.ConvertVariables(_settings.FilenameTemplate, DateTime.Now, _startTime, _product,cameraName: _settings.CameraName) + ".png";
+        try
+        {
+            var image = result.ImageOriginal;
+            // save to memory stream
+            _goodRingbufferIsolated.Write(image.ToBytes(), filename);
+        }
+        catch (Exception e)
+        {
+            Console.WriteLine($"Error writing good result at {filename}: " + e.Message);
+        }
+    }
+
+    Dictionary<string, IsolatedRingbuffer> _badRingbuffers = new Dictionary<string, IsolatedRingbuffer>();
+
+    private IsolatedRingbuffer GetBadRingbuffer(string defectName)
+    {
+        if (_badRingbuffers.ContainsKey(defectName))
+        {
+            return _badRingbuffers[defectName];
+        }
+        var path = PathSettingsUtils.ConvertVariables(Path.Combine(_settings.Root, _settings.BadImagesPathTemplate), _startTime, _startTime, _product, cameraName: _settings.CameraName);
+        var ringbuffer = new IsolatedRingbuffer(path, _settings.MaxBad, new AsyncFilesystem());
+        _badRingbuffers[defectName] = ringbuffer;
+        return ringbuffer;
+    }
+
+    private void SaveResultImage(Mat image, string defectName, DateTime time, bool isAnalysis)
+    {
+        var filename = PathSettingsUtils.ConvertVariables(_settings.FilenameTemplate,time,_startTime,_product,defectName,_settings.CameraName) + (isAnalysis ? "_analysis" : "") + ".png";
+        var ringbuffer = GetBadRingbuffer(defectName);
+        try
+        {
+
+            
+            ringbuffer.Write(image.ToBytes(), filename);
+        }
+        catch (Exception e)
+        {
+            Console.WriteLine($"Error writing bad result for {filename}: " + e.Message);
+        }
+    }
+
+    public void WriteBadResult(ImageProcessedEvent result)
+    {
+        var defectName = "unknown";
+
+        if (result.ErrorNames.Any())
+        {
+            defectName = result.ErrorNames.First();
+        }
+
+        var time = DateTime.Now;
+        SaveResultImage(result.ImageOriginal, defectName, time, false);
+        SaveResultImage(result.ImageAnalysis, defectName, time, true);
+    }
+
+    public void NewProduct(string product)
+    {
+        _startTime = DateTime.Now;
+        _product = product;
+        _path = PathSettingsUtils.ConvertVariables(Path.Combine(_settings.Root,_settings.GoodImagesPathTemplate), _startTime, _startTime, _product,cameraName:_settings.CameraName);
+        _goodRingbufferIsolated?.Dispose();
+        _goodRingbufferIsolated = new IsolatedRingbuffer(_path, _settings.MaxGood, new AsyncFilesystem());
+
+        foreach (var ringbuffer in _badRingbuffers)
+        {
+            ringbuffer.Value.Dispose();
+        }
+
+        _badRingbuffers.Clear();
+
+    }
+}