IVideoEncoder.h

#pragma once
 
#include "IVideoDecoder.h"
 
#include <string>
 
namespace Cogs::Memory{
  class MemoryBuffer;
}
namespace Cogs::Core
{
  using VideoEncoderCallback = std::function<void(struct IVideoEncoder*, struct VideoEncoderPayload &payload)>;
 
  enum PictureType
  {
    UNKNOWN,
    P,
    B,
    I,
    IDR,
    BI,
    SKIPPED,
    INTRA_REFRESH,
    NONREF_P,
  };
  struct VideoEncoderPayload
  {
    PictureType type;
    uint32_t frameIdx;
    uint64_t timestamp;
    uint64_t duration;
    bool ltrFrame; // Long Term Reference
    uint32_t ltrFrameIdx;
    uint8_t *data;
    size_t size;
  };
  struct VideoEncoderDescription
  {
    Codec codec;
    uint32_t encodeWidth;
    uint32_t encodeHeight;
    uint32_t darWidth;
    uint32_t darHeight;
    uint32_t maxWidth;
    uint32_t maxHeight;
    uint32_t bitrate;
    uint32_t maxBitrate;
    uint32_t bFrameCount;
    uint32_t bFrameReference;
    float frameRate;
    bool lookahead;
    bool recordToDisk;
    bool useMSAA;
    std::string recordPath;
    VideoEncoderCallback callback;
  };
  struct VideoEncoderInput
  {
    Cogs::Core::TextureHandle texture;
    uint64_t timestamp = 0;
    uint64_t duration = 0;
    bool forceIntra = false;
    bool forceIDR = false;
    bool outputSPSPPS = false;
  };
  struct VideoEncoderStatus
  {
    uint64_t timestamp = 0;
  };
 
  struct IVideoEncoder
  {
    virtual ~IVideoEncoder() { }
    virtual Cogs::Core::TextureHandle getRenderTarget() = 0;
    virtual Cogs::Core::TextureHandle getCurrentInputBuffer() = 0;
    virtual Cogs::Core::TextureHandle getNextInputBuffer() = 0;
    virtual void encode(VideoEncoderInput &input) = 0;
    virtual void getBitstream() = 0;
    virtual void getSPSPPS(Cogs::Memory::MemoryBuffer& spspps) = 0;
    virtual VideoEncoderStatus getStatus() = 0;
    virtual void reconfigure(VideoEncoderDescription &desc, bool resetEncoder, bool forceIDR) = 0;
  };
  struct IVideoEncoderContext
  {
    virtual ~IVideoEncoderContext() { }
    virtual IVideoEncoder *createVideoEncoder(VideoEncoderDescription &desc) = 0;
    virtual void destroyVideoEncoder(IVideoEncoder*) = 0;
  };
}// namespace ...