using OpenCvSharp;
using OpenCvSharp.Extensions;

using Alturos.Yolo;

private void VideoObjectDetection()
{
  // YOLO setting
  int yoloWidth = 608, yoloHeight = 608;
  var configurationDetector = new ConfigurationDetector();
  var config = configurationDetector.Detect();
  YoloWrapper yoloWrapper = new YoloWrapper(config);
  
  // OpenCV & WPF setting
  VideoCapture videocapture;
  Mat image = new Mat();
  WriteableBitmap wb = new WriteableBitmap(yoloWidth, yoloHeight, 96, 96, PixelFormats.Bgr24, null);
  
  byte[] imageInBytes = new byte[(int)(yoloWidth * yoloHeight * image.Channels())];
  
  // Read a video file and run object detection over it!
  using (videocapture = new VideoCapture(address))
  {
    using(Mat imageOriginal = new Mat())
    {
      // read a single frame and convert the frame into a byte array
      videocapture.Read(imageOriginal);
      image = imageOriginal.Resize(new OpenCvSharp.Size(yoloWidth, yoloHeight));
      imageInBytes = image.ToBytes();
      
      // conduct object detection and display the result
      var items = yolowrapper.Detect(imageInBytes);
      foreach(var item in items)
      {
        var x = item.X;
        var y = item.Y;
        var width = item.Width;
        var height = item.Height;
        var type = item.Type;  // class name of the object
        
        // draw a bounding box for the detected object
        // you can set different colors for different classes
        Cv2.Rectangle(image, new OpenCvSharp.Rect(x, y, width, height), Scalar.Green, 3);
      }
      
      // display the detection result
      WriteableBitmapConverter.ToWriteableBitmap(image, wb);
      /* WPF component: videoViewer
      <Canvas Name="canvasYoloVideo" Height="608" Width="608">
        <Image Name="videoViewer" Height="608" Width="608" Stretch="Fill" />
      </Canvas>
      */
      videoViewer.Source = wb;
    }
  }
}