rfx_decode.c

 
#include <freerdp/config.h>
 
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
 
#include <winpr/stream.h>
#include <freerdp/primitives.h>
 
#include "rfx_types.h"
#include "rfx_rlgr.h"
#include "rfx_differential.h"
#include "rfx_quantization.h"
#include "rfx_dwt.h"
 
#include "rfx_decode.h"
 
WINPR_ATTR_NODISCARD
static inline BOOL rfx_decode_component(RFX_CONTEXT* WINPR_RESTRICT context,
                                        const UINT32* WINPR_RESTRICT quantization_values,
                                        size_t nrQuantValues, const BYTE* WINPR_RESTRICT data,
                                        size_t size, INT16* WINPR_RESTRICT buffer)
{
  BOOL res = FALSE;
  INT16* dwt_buffer = BufferPool_Take(context->priv->BufferPool, -1); /* dwt_buffer */
  WINPR_ASSERT(dwt_buffer);
 
  PROFILER_ENTER(context->priv->prof_rfx_decode_component)
  PROFILER_ENTER(context->priv->prof_rfx_rlgr_decode)
  WINPR_ASSERT(size <= UINT32_MAX);
 
  {
    const int rc = context->rlgr_decode(context->mode, data, (UINT32)size, buffer, 4096);
    if (rc < 0)
    {
      WLog_Print(context->priv->log, WLOG_ERROR, "context->rlgr_decode failed: %d", rc);
      goto fail;
    }
  }
 
  PROFILER_EXIT(context->priv->prof_rfx_rlgr_decode)
  PROFILER_ENTER(context->priv->prof_rfx_differential_decode)
  if (!rfx_differential_decode(buffer + 4032, 64))
    goto fail;
  PROFILER_EXIT(context->priv->prof_rfx_differential_decode)
  PROFILER_ENTER(context->priv->prof_rfx_quantization_decode)
  if (!context->quantization_decode(buffer, quantization_values, nrQuantValues))
    goto fail;
  PROFILER_EXIT(context->priv->prof_rfx_quantization_decode)
  PROFILER_ENTER(context->priv->prof_rfx_dwt_2d_decode)
  context->dwt_2d_decode(buffer, dwt_buffer);
  PROFILER_EXIT(context->priv->prof_rfx_dwt_2d_decode)
  PROFILER_EXIT(context->priv->prof_rfx_decode_component)
 
  res = TRUE;
fail:
  BufferPool_Return(context->priv->BufferPool, dwt_buffer);
  return res;
}
 
/* rfx_decode_ycbcr_to_rgb code now resides in the primitives library. */
 
/* stride is bytes between rows in the output buffer. */
BOOL rfx_decode_rgb(RFX_CONTEXT* WINPR_RESTRICT context, const RFX_TILE* WINPR_RESTRICT tile,
                    BYTE* WINPR_RESTRICT rgb_buffer, UINT32 stride)
{
  union
  {
    const INT16** cpv;
    INT16** pv;
  } cnv;
  BOOL rc = FALSE;
  BYTE* pBuffer = nullptr;
  INT16* pSrcDst[3];
  UINT32* y_quants = nullptr;
  UINT32* cb_quants = nullptr;
  UINT32* cr_quants = nullptr;
  static const prim_size_t roi_64x64 = { 64, 64 };
  const primitives_t* prims = primitives_get();
  PROFILER_ENTER(context->priv->prof_rfx_decode_rgb)
  y_quants = context->quants + (NR_QUANT_VALUES * tile->quantIdxY);
  cb_quants = context->quants + (NR_QUANT_VALUES * tile->quantIdxCb);
  cr_quants = context->quants + (NR_QUANT_VALUES * tile->quantIdxCr);
  pBuffer = (BYTE*)BufferPool_Take(context->priv->BufferPool, -1);
  pSrcDst[0] = WINPR_PACKED_ALIGN_CAST(
      INT16*, ((&pBuffer[((8192ULL + 32ULL) * 0ULL) + 16ULL]))); /* y_r_buffer */
  pSrcDst[1] = WINPR_PACKED_ALIGN_CAST(
      INT16*, ((&pBuffer[((8192ULL + 32ULL) * 1ULL) + 16ULL]))); /* cb_g_buffer */
  pSrcDst[2] = WINPR_PACKED_ALIGN_CAST(
      INT16*, ((&pBuffer[((8192ULL + 32ULL) * 2ULL) + 16ULL]))); /* cr_b_buffer */
  if (!rfx_decode_component(context, y_quants, NR_QUANT_VALUES, tile->YData, tile->YLen,
                            pSrcDst[0])) /* YData */
    goto fail;
  if (!rfx_decode_component(context, cb_quants, NR_QUANT_VALUES, tile->CbData, tile->CbLen,
                            pSrcDst[1])) /* CbData */
    goto fail;
  if (!rfx_decode_component(context, cr_quants, NR_QUANT_VALUES, tile->CrData, tile->CrLen,
                            pSrcDst[2])) /* CrData */
    goto fail;
  PROFILER_ENTER(context->priv->prof_rfx_ycbcr_to_rgb)
 
  cnv.pv = pSrcDst;
  if (prims->yCbCrToRGB_16s8u_P3AC4R(cnv.cpv, 64 * sizeof(INT16), rgb_buffer, stride,
                                     context->pixel_format, &roi_64x64) != PRIMITIVES_SUCCESS)
    goto fail;
 
  PROFILER_EXIT(context->priv->prof_rfx_ycbcr_to_rgb)
  PROFILER_EXIT(context->priv->prof_rfx_decode_rgb)
 
  rc = TRUE;
fail:
  BufferPool_Return(context->priv->BufferPool, pBuffer);
  return rc;
}