VideoCodec.h

/*
 *  Copyright (C) 2017-2018 Team Kodi
 *  This file is part of Kodi - https://kodi.tv
 *
 *  SPDX-License-Identifier: GPL-2.0-or-later
 *  See LICENSES/README.md for more information.
 */
 
#pragma once
 
#include "../AddonBase.h"
#include "../c-api/addon-instance/video_codec.h"
#include "inputstream/DemuxPacket.h"
#include "inputstream/StreamCodec.h"
#include "inputstream/StreamCrypto.h"
 
#ifdef __cplusplus
 
namespace kodi
{
namespace addon
{
 
class CInstanceVideoCodec;
 
//==============================================================================
class ATTR_DLL_LOCAL VideoCodecInitdata : public CStructHdl<VideoCodecInitdata, VIDEOCODEC_INITDATA>
{
  friend class CInstanceVideoCodec;
public:
 
 
  VIDEOCODEC_TYPE GetCodecType() const { return m_cStructure->codec; }
 
  STREAMCODEC_PROFILE GetCodecProfile() const { return m_cStructure->codecProfile; }
 
  std::vector<VIDEOCODEC_FORMAT> GetVideoFormats() const
  {
    std::vector<VIDEOCODEC_FORMAT> formats;
    unsigned int i = 0;
    while (i < VIDEOCODEC_FORMAT_MAXFORMATS &&
           m_cStructure->videoFormats[i] != VIDEOCODEC_FORMAT_UNKNOWN)
      formats.emplace_back(m_cStructure->videoFormats[i++]);
    if (formats.empty())
      formats.emplace_back(VIDEOCODEC_FORMAT_UNKNOWN);
    return formats;
  }
 
  uint32_t GetWidth() const { return m_cStructure->width; }
 
  uint32_t GetHeight() const { return m_cStructure->height; }
 
  const uint8_t* GetExtraData() const { return m_cStructure->extraData; }
 
  unsigned int GetExtraDataSize() const { return m_cStructure->extraDataSize; }
 
  kodi::addon::StreamCryptoSession GetCryptoSession() const { return &m_cStructure->cryptoSession; }
 
 
private:
  VideoCodecInitdata() = delete;
  VideoCodecInitdata(const VideoCodecInitdata& session) : CStructHdl(session) {}
  VideoCodecInitdata(const VIDEOCODEC_INITDATA* session) : CStructHdl(session) {}
  VideoCodecInitdata(VIDEOCODEC_INITDATA* session) : CStructHdl(session) {}
};
//------------------------------------------------------------------------------
 
//##############################################################################
 
//============================================================================
class ATTR_DLL_LOCAL CInstanceVideoCodec : public IAddonInstance
{
public:
  //============================================================================
  explicit CInstanceVideoCodec(const IInstanceInfo& instance) : IAddonInstance(instance)
  {
    if (CPrivateBase::m_interface->globalSingleInstance != nullptr)
      throw std::logic_error("kodi::addon::CInstanceVideoCodec: Creation of multiple together with "
                             "single instance way is not allowed!");
 
    SetAddonStruct(instance);
  }
  //----------------------------------------------------------------------------
 
  //============================================================================
  ~CInstanceVideoCodec() override = default;
  //----------------------------------------------------------------------------
 
  //============================================================================
  virtual bool Open(const kodi::addon::VideoCodecInitdata& initData) { return false; }
  //----------------------------------------------------------------------------
 
  //============================================================================
  virtual bool Reconfigure(const kodi::addon::VideoCodecInitdata& initData) { return false; }
  //----------------------------------------------------------------------------
 
  //============================================================================
  virtual bool AddData(const DEMUX_PACKET& packet) { return false; }
  //----------------------------------------------------------------------------
 
  //============================================================================
  virtual VIDEOCODEC_RETVAL GetPicture(VIDEOCODEC_PICTURE& picture) { return VC_ERROR; }
  //----------------------------------------------------------------------------
 
  //============================================================================
  virtual const char* GetName() { return nullptr; }
  //----------------------------------------------------------------------------
 
  //============================================================================
  virtual void Reset() {}
  //----------------------------------------------------------------------------
 
  //============================================================================
  bool GetFrameBuffer(VIDEOCODEC_PICTURE& picture)
  {
    return m_instanceData->toKodi->get_frame_buffer(m_instanceData->toKodi->kodiInstance, &picture);
  }
  //----------------------------------------------------------------------------
 
  //============================================================================
  void ReleaseFrameBuffer(void* buffer)
  {
    return m_instanceData->toKodi->release_frame_buffer(m_instanceData->toKodi->kodiInstance,
                                                        buffer);
  }
  //----------------------------------------------------------------------------
 
  //============================================================================
  bool GetFrameBufferPlatformHandle(KODI_HANDLE videoBufferHandle,
                                    VIDEOCODEC_PLATFORM_BUFFER& platformBuffer)
  {
    if (!m_instanceData->toKodi->get_frame_buffer_platform_handle)
      return false;
    return m_instanceData->toKodi->get_frame_buffer_platform_handle(
        m_instanceData->toKodi->kodiInstance, videoBufferHandle, &platformBuffer);
  }
  //----------------------------------------------------------------------------
 
private:
  void SetAddonStruct(KODI_ADDON_INSTANCE_STRUCT* instance)
  {
    instance->hdl = this;
    instance->videocodec->toAddon->open = ADDON_Open;
    instance->videocodec->toAddon->reconfigure = ADDON_Reconfigure;
    instance->videocodec->toAddon->add_data = ADDON_AddData;
    instance->videocodec->toAddon->get_picture = ADDON_GetPicture;
    instance->videocodec->toAddon->get_name = ADDON_GetName;
    instance->videocodec->toAddon->reset = ADDON_Reset;
 
    m_instanceData = instance->videocodec;
    m_instanceData->toAddon->addonInstance = this;
  }
 
  inline static bool ADDON_Open(const AddonInstance_VideoCodec* instance,
                                VIDEOCODEC_INITDATA* initData)
  {
    return static_cast<CInstanceVideoCodec*>(instance->toAddon->addonInstance)->Open(initData);
  }
 
  inline static bool ADDON_Reconfigure(const AddonInstance_VideoCodec* instance,
                                       VIDEOCODEC_INITDATA* initData)
  {
    return static_cast<CInstanceVideoCodec*>(instance->toAddon->addonInstance)
        ->Reconfigure(initData);
  }
 
  inline static bool ADDON_AddData(const AddonInstance_VideoCodec* instance,
                                   const DEMUX_PACKET* packet)
  {
    return static_cast<CInstanceVideoCodec*>(instance->toAddon->addonInstance)->AddData(*packet);
  }
 
  inline static VIDEOCODEC_RETVAL ADDON_GetPicture(const AddonInstance_VideoCodec* instance,
                                                   VIDEOCODEC_PICTURE* picture)
  {
    return static_cast<CInstanceVideoCodec*>(instance->toAddon->addonInstance)
        ->GetPicture(*picture);
  }
 
  inline static const char* ADDON_GetName(const AddonInstance_VideoCodec* instance)
  {
    return static_cast<CInstanceVideoCodec*>(instance->toAddon->addonInstance)->GetName();
  }
 
  inline static void ADDON_Reset(const AddonInstance_VideoCodec* instance)
  {
    return static_cast<CInstanceVideoCodec*>(instance->toAddon->addonInstance)->Reset();
  }
 
  AddonInstance_VideoCodec* m_instanceData;
};
 
} // namespace addon
} // namespace kodi
 
#endif /* __cplusplus */