// Copyright (c) 2021 homuler
//
// Use of this source code is governed by an MIT-style
// license that can be found in the LICENSE file or at
// https://opensource.org/licenses/MIT.

using System;

namespace Mediapipe
{
  public class ImageFramePacket : Packet<ImageFrame>
  {
    /// <summary>
    ///   Creates an empty <see cref="ImageFramePacket" /> instance.
    /// </summary>
    public ImageFramePacket() : base(true) { }

    [UnityEngine.Scripting.Preserve]
    public ImageFramePacket(IntPtr ptr, bool isOwner = true) : base(ptr, isOwner) { }

    public ImageFramePacket(ImageFrame imageFrame) : base()
    {
      UnsafeNativeMethods.mp__MakeImageFramePacket__Pif(imageFrame.mpPtr, out var ptr).Assert();
      imageFrame.Dispose(); // respect move semantics

      this.ptr = ptr;
    }

    public ImageFramePacket(ImageFrame imageFrame, Timestamp timestamp) : base()
    {
      UnsafeNativeMethods.mp__MakeImageFramePacket_At__Pif_Rt(imageFrame.mpPtr, timestamp.mpPtr, out var ptr).Assert();
      GC.KeepAlive(timestamp);
      imageFrame.Dispose(); // respect move semantics

      this.ptr = ptr;
    }

    public ImageFramePacket At(Timestamp timestamp)
    {
      return At<ImageFramePacket>(timestamp);
    }

    public override ImageFrame Get()
    {
      UnsafeNativeMethods.mp_Packet__GetImageFrame(mpPtr, out var imageFramePtr).Assert();

      GC.KeepAlive(this);
      return new ImageFrame(imageFramePtr, false);
    }

    public override StatusOr<ImageFrame> Consume()
    {
      UnsafeNativeMethods.mp_Packet__ConsumeImageFrame(mpPtr, out var statusOrImageFramePtr).Assert();

      GC.KeepAlive(this);
      return new StatusOrImageFrame(statusOrImageFramePtr);
    }

    public override Status ValidateAsType()
    {
      UnsafeNativeMethods.mp_Packet__ValidateAsImageFrame(mpPtr, out var statusPtr).Assert();

      GC.KeepAlive(this);
      return new Status(statusPtr);
    }
  }
}