hevc.c

 
#include <freerdp/codec/hevc.h>
#include <freerdp/primitives.h>
#include <freerdp/log.h>
#include <freerdp/codec/region.h>
#include <freerdp/codec/h264.h>
 
#define TAG FREERDP_TAG("codec.hevc")
 
struct S_FREERDP_HEVC_CONTEXT
{
  BOOL Compressor;
  wLog* log;
  H264_CONTEXT* decoder;
};
 
WINPR_ATTR_NODISCARD
static H264_CONTEXT_OPTION hevc2h264option(FREERDP_HEVC_CONTEXT_OPTION option)
{
  switch (option)
  {
    case FREERDP_HEVC_CONTEXT_OPTION_PROFILE:
      return H264_CONTEXT_OPTION_QP;
    case FREERDP_HEVC_CONTEXT_OPTION_RATECONTROL:
      return H264_CONTEXT_OPTION_RATECONTROL;
    case FREERDP_HEVC_CONTEXT_OPTION_BITRATE:
      return H264_CONTEXT_OPTION_BITRATE;
    case FREERDP_HEVC_CONTEXT_OPTION_USAGETYPE:
      return H264_CONTEXT_OPTION_USAGETYPE;
    case FREERDP_HEVC_CONTEXT_OPTION_HW_ACCEL:
      return H264_CONTEXT_OPTION_HW_ACCEL;
    default:
      return 0;
  }
}
BOOL freerdp_hevc_context_set_option(WINPR_ATTR_UNUSED FREERDP_HEVC_CONTEXT* hevc,
                                     WINPR_ATTR_UNUSED FREERDP_HEVC_CONTEXT_OPTION option,
                                     WINPR_ATTR_UNUSED UINT32 value)
{
  WINPR_ASSERT(hevc);
  if (!hevc->Compressor)
    return h264_context_set_option(hevc->decoder, hevc2h264option(option), value);
  return FALSE;
}
 
UINT32 freerdp_hevc_context_get_option(WINPR_ATTR_UNUSED FREERDP_HEVC_CONTEXT* hevc,
                                       WINPR_ATTR_UNUSED FREERDP_HEVC_CONTEXT_OPTION option)
{
  WINPR_ASSERT(hevc);
  if (!hevc->Compressor)
    return h264_context_get_option(hevc->decoder, hevc2h264option(option));
  return 0;
}
 
INT32 freerdp_hevc_compress(WINPR_ATTR_UNUSED FREERDP_HEVC_CONTEXT* hevc,
                            WINPR_ATTR_UNUSED const BYTE* pSrcData,
                            WINPR_ATTR_UNUSED DWORD SrcFormat, WINPR_ATTR_UNUSED UINT32 nSrcStep,
                            WINPR_ATTR_UNUSED UINT32 nSrcWidth, WINPR_ATTR_UNUSED UINT32 nSrcHeight,
                            WINPR_ATTR_UNUSED const RECTANGLE_16* regionRect,
                            WINPR_ATTR_UNUSED BYTE** ppDstData, WINPR_ATTR_UNUSED UINT32* pDstSize,
                            WINPR_ATTR_UNUSED RDPGFX_H264_METABLOCK* meta)
{
  WINPR_ASSERT(hevc);
  WLog_Print(hevc->log, WLOG_ERROR,
             "This build does not support HEVC codec. Recompile with '-DWITH_AVM=ON' or "
             "'-DWITH_DHEVCD=ON'");
  return -1;
}
 
INT32 freerdp_hevc_decompress(WINPR_ATTR_UNUSED FREERDP_HEVC_CONTEXT* hevc,
                              WINPR_ATTR_UNUSED const BYTE* pSrcData,
                              WINPR_ATTR_UNUSED UINT32 SrcSize, WINPR_ATTR_UNUSED BYTE* pDstData,
                              WINPR_ATTR_UNUSED DWORD DstFormat, WINPR_ATTR_UNUSED UINT32 nDstStep,
                              WINPR_ATTR_UNUSED UINT32 nDstWidth,
                              WINPR_ATTR_UNUSED UINT32 nDstHeight,
                              WINPR_ATTR_UNUSED const RECTANGLE_16* regionRects,
                              WINPR_ATTR_UNUSED UINT32 numRegionRect)
{
  WINPR_ASSERT(hevc);
  return avc420_decompress(hevc->decoder, pSrcData, SrcSize, pDstData, DstFormat, nDstStep,
                           nDstWidth, nDstHeight, regionRects, numRegionRect);
}
 
BOOL freerdp_hevc_context_reset(WINPR_ATTR_UNUSED FREERDP_HEVC_CONTEXT* hevc,
                                WINPR_ATTR_UNUSED UINT32 width, WINPR_ATTR_UNUSED UINT32 height)
{
  WINPR_ASSERT(hevc);
  if (!hevc->Compressor)
    return h264_context_reset(hevc->decoder, width, height);
 
  WLog_Print(hevc->log, WLOG_WARN,
             "This build does not support HEVC codec. Recompile with '-DWITH_AVM=ON' or "
             "'-DWITH_DHEVCD=ON'");
  return FALSE;
}
 
void freerdp_hevc_context_free(WINPR_ATTR_UNUSED FREERDP_HEVC_CONTEXT* hevc)
{
  if (!hevc)
    return;
 
  h264_context_free(hevc->decoder);
  free(hevc);
}
 
FREERDP_HEVC_CONTEXT* freerdp_hevc_context_new(BOOL Compressor)
{
  FREERDP_HEVC_CONTEXT* ctx = calloc(1, sizeof(FREERDP_HEVC_CONTEXT));
  if (!ctx)
    return nullptr;
  ctx->Compressor = Compressor;
  ctx->log = WLog_Get(TAG);
  return ctx;
}