/src/FreeRDP/libfreerdp/codec/progressive.c

Source (jump to first uncovered line)
/**
 * FreeRDP: A Remote Desktop Protocol Implementation
 * Progressive Codec Bitmap Compression
 *
 * Copyright 2014 Marc-Andre Moreau <marcandre.moreau@gmail.com>
 * Copyright 2019 Armin Novak <armin.novak@thincast.com>
 * Copyright 2019 Thincast Technologies GmbH
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#include <freerdp/config.h>

#include <winpr/assert.h>
#include <winpr/crt.h>
#include <winpr/print.h>
#include <winpr/bitstream.h>

#include <freerdp/primitives.h>
#include <freerdp/codec/color.h>
#include <freerdp/codec/progressive.h>
#include <freerdp/codec/region.h>
#include <freerdp/log.h>

#include "rfx_differential.h"
#include "rfx_quantization.h"
#include "rfx_dwt.h"
#include "rfx_rlgr.h"
#include "rfx_types.h"
#include "progressive.h"

#define TAG FREERDP_TAG("codec.progressive")

typedef struct
{
  BOOL nonLL;
  wBitStream* srl;
  wBitStream* raw;

  /* SRL state */

  UINT32 kp;
  int nz;
  BOOL mode;
} RFX_PROGRESSIVE_UPGRADE_STATE;

static INLINE BOOL progressive_write_tile_simple(PROGRESSIVE_CONTEXT* progressive, wStream* s,
                                                 const RFX_TILE* tile);

static const char* progressive_get_block_type_string(UINT16 blockType)
{
  switch (blockType)
  {
    case PROGRESSIVE_WBT_SYNC:
      return "PROGRESSIVE_WBT_SYNC";

    case PROGRESSIVE_WBT_FRAME_BEGIN:
      return "PROGRESSIVE_WBT_FRAME_BEGIN";

    case PROGRESSIVE_WBT_FRAME_END:
      return "PROGRESSIVE_WBT_FRAME_END";

    case PROGRESSIVE_WBT_CONTEXT:
      return "PROGRESSIVE_WBT_CONTEXT";

    case PROGRESSIVE_WBT_REGION:
      return "PROGRESSIVE_WBT_REGION";

    case PROGRESSIVE_WBT_TILE_SIMPLE:
      return "PROGRESSIVE_WBT_TILE_SIMPLE";

    case PROGRESSIVE_WBT_TILE_FIRST:
      return "PROGRESSIVE_WBT_TILE_FIRST";

    case PROGRESSIVE_WBT_TILE_UPGRADE:
      return "PROGRESSIVE_WBT_TILE_UPGRADE";

    default:
      return "PROGRESSIVE_WBT_UNKNOWN";
  }
}

static INLINE void progressive_component_codec_quant_read(wStream* s,
                                                          RFX_COMPONENT_CODEC_QUANT* quantVal)
{
  BYTE b;
  Stream_Read_UINT8(s, b);
  quantVal->LL3 = b & 0x0F;
  quantVal->HL3 = b >> 4;
  Stream_Read_UINT8(s, b);
  quantVal->LH3 = b & 0x0F;
  quantVal->HH3 = b >> 4;
  Stream_Read_UINT8(s, b);
  quantVal->HL2 = b & 0x0F;
  quantVal->LH2 = b >> 4;
  Stream_Read_UINT8(s, b);
  quantVal->HH2 = b & 0x0F;
  quantVal->HL1 = b >> 4;
  Stream_Read_UINT8(s, b);
  quantVal->LH1 = b & 0x0F;
  quantVal->HH1 = b >> 4;
}

static INLINE void progressive_rfx_quant_ladd(RFX_COMPONENT_CODEC_QUANT* q, int val)
{
  q->HL1 += val; /* HL1 */
  q->LH1 += val; /* LH1 */
  q->HH1 += val; /* HH1 */
  q->HL2 += val; /* HL2 */
  q->LH2 += val; /* LH2 */
  q->HH2 += val; /* HH2 */
  q->HL3 += val; /* HL3 */
  q->LH3 += val; /* LH3 */
  q->HH3 += val; /* HH3 */
  q->LL3 += val; /* LL3 */
}

static INLINE void progressive_rfx_quant_add(const RFX_COMPONENT_CODEC_QUANT* q1,
                                             const RFX_COMPONENT_CODEC_QUANT* q2,
                                             RFX_COMPONENT_CODEC_QUANT* dst)
{
  dst->HL1 = q1->HL1 + q2->HL1; /* HL1 */
  dst->LH1 = q1->LH1 + q2->LH1; /* LH1 */
  dst->HH1 = q1->HH1 + q2->HH1; /* HH1 */
  dst->HL2 = q1->HL2 + q2->HL2; /* HL2 */
  dst->LH2 = q1->LH2 + q2->LH2; /* LH2 */
  dst->HH2 = q1->HH2 + q2->HH2; /* HH2 */
  dst->HL3 = q1->HL3 + q2->HL3; /* HL3 */
  dst->LH3 = q1->LH3 + q2->LH3; /* LH3 */
  dst->HH3 = q1->HH3 + q2->HH3; /* HH3 */
  dst->LL3 = q1->LL3 + q2->LL3; /* LL3 */
}

static INLINE void progressive_rfx_quant_lsub(RFX_COMPONENT_CODEC_QUANT* q, int val)
{
  q->HL1 -= val; /* HL1 */
  q->LH1 -= val; /* LH1 */
  q->HH1 -= val; /* HH1 */
  q->HL2 -= val; /* HL2 */
  q->LH2 -= val; /* LH2 */
  q->HH2 -= val; /* HH2 */
  q->HL3 -= val; /* HL3 */
  q->LH3 -= val; /* LH3 */
  q->HH3 -= val; /* HH3 */
  q->LL3 -= val; /* LL3 */
}

static INLINE void progressive_rfx_quant_sub(const RFX_COMPONENT_CODEC_QUANT* q1,
                                             const RFX_COMPONENT_CODEC_QUANT* q2,
                                             RFX_COMPONENT_CODEC_QUANT* dst)
{
  dst->HL1 = q1->HL1 - q2->HL1; /* HL1 */
  dst->LH1 = q1->LH1 - q2->LH1; /* LH1 */
  dst->HH1 = q1->HH1 - q2->HH1; /* HH1 */
  dst->HL2 = q1->HL2 - q2->HL2; /* HL2 */
  dst->LH2 = q1->LH2 - q2->LH2; /* LH2 */
  dst->HH2 = q1->HH2 - q2->HH2; /* HH2 */
  dst->HL3 = q1->HL3 - q2->HL3; /* HL3 */
  dst->LH3 = q1->LH3 - q2->LH3; /* LH3 */
  dst->HH3 = q1->HH3 - q2->HH3; /* HH3 */
  dst->LL3 = q1->LL3 - q2->LL3; /* LL3 */
}

static INLINE BOOL progressive_rfx_quant_lcmp_less_equal(const RFX_COMPONENT_CODEC_QUANT* q,
                                                         int val)
{
  if (q->HL1 > val)
    return FALSE; /* HL1 */

  if (q->LH1 > val)
    return FALSE; /* LH1 */

  if (q->HH1 > val)
    return FALSE; /* HH1 */

  if (q->HL2 > val)
    return FALSE; /* HL2 */

  if (q->LH2 > val)
    return FALSE; /* LH2 */

  if (q->HH2 > val)
    return FALSE; /* HH2 */

  if (q->HL3 > val)
    return FALSE; /* HL3 */

  if (q->LH3 > val)
    return FALSE; /* LH3 */

  if (q->HH3 > val)
    return FALSE; /* HH3 */

  if (q->LL3 > val)
    return FALSE; /* LL3 */

  return TRUE;
}

static INLINE BOOL progressive_rfx_quant_cmp_less_equal(const RFX_COMPONENT_CODEC_QUANT* q1,
                                                        const RFX_COMPONENT_CODEC_QUANT* q2)
{
  if (q1->HL1 > q2->HL1)
    return FALSE; /* HL1 */

  if (q1->LH1 > q2->LH1)
    return FALSE; /* LH1 */

  if (q1->HH1 > q2->HH1)
    return FALSE; /* HH1 */

  if (q1->HL2 > q2->HL2)
    return FALSE; /* HL2 */

  if (q1->LH2 > q2->LH2)
    return FALSE; /* LH2 */

  if (q1->HH2 > q2->HH2)
    return FALSE; /* HH2 */

  if (q1->HL3 > q2->HL3)
    return FALSE; /* HL3 */

  if (q1->LH3 > q2->LH3)
    return FALSE; /* LH3 */

  if (q1->HH3 > q2->HH3)
    return FALSE; /* HH3 */

  if (q1->LL3 > q2->LL3)
    return FALSE; /* LL3 */

  return TRUE;
}

static INLINE BOOL progressive_rfx_quant_lcmp_greater_equal(const RFX_COMPONENT_CODEC_QUANT* q,
                                                            int val)
{
  if (q->HL1 < val)
    return FALSE; /* HL1 */

  if (q->LH1 < val)
    return FALSE; /* LH1 */

  if (q->HH1 < val)
    return FALSE; /* HH1 */

  if (q->HL2 < val)
    return FALSE; /* HL2 */

  if (q->LH2 < val)
    return FALSE; /* LH2 */

  if (q->HH2 < val)
    return FALSE; /* HH2 */

  if (q->HL3 < val)
    return FALSE; /* HL3 */

  if (q->LH3 < val)
    return FALSE; /* LH3 */

  if (q->HH3 < val)
    return FALSE; /* HH3 */

  if (q->LL3 < val)
    return FALSE; /* LL3 */

  return TRUE;
}

static INLINE BOOL progressive_rfx_quant_cmp_greater_equal(const RFX_COMPONENT_CODEC_QUANT* q1,
                                                           const RFX_COMPONENT_CODEC_QUANT* q2)
{
  if (q1->HL1 < q2->HL1)
    return FALSE; /* HL1 */

  if (q1->LH1 < q2->LH1)
    return FALSE; /* LH1 */

  if (q1->HH1 < q2->HH1)
    return FALSE; /* HH1 */

  if (q1->HL2 < q2->HL2)
    return FALSE; /* HL2 */

  if (q1->LH2 < q2->LH2)
    return FALSE; /* LH2 */

  if (q1->HH2 < q2->HH2)
    return FALSE; /* HH2 */

  if (q1->HL3 < q2->HL3)
    return FALSE; /* HL3 */

  if (q1->LH3 < q2->LH3)
    return FALSE; /* LH3 */

  if (q1->HH3 < q2->HH3)
    return FALSE; /* HH3 */

  if (q1->LL3 < q2->LL3)
    return FALSE; /* LL3 */

  return TRUE;
}

static INLINE BOOL progressive_rfx_quant_cmp_equal(const RFX_COMPONENT_CODEC_QUANT* q1,
                                                   const RFX_COMPONENT_CODEC_QUANT* q2)
{
  if (q1->HL1 != q2->HL1)
    return FALSE; /* HL1 */

  if (q1->LH1 != q2->LH1)
    return FALSE; /* LH1 */

  if (q1->HH1 != q2->HH1)
    return FALSE; /* HH1 */

  if (q1->HL2 != q2->HL2)
    return FALSE; /* HL2 */

  if (q1->LH2 != q2->LH2)
    return FALSE; /* LH2 */

  if (q1->HH2 != q2->HH2)
    return FALSE; /* HH2 */

  if (q1->HL3 != q2->HL3)
    return FALSE; /* HL3 */

  if (q1->LH3 != q2->LH3)
    return FALSE; /* LH3 */

  if (q1->HH3 != q2->HH3)
    return FALSE; /* HH3 */

  if (q1->LL3 != q2->LL3)
    return FALSE; /* LL3 */

  return TRUE;
}

static INLINE BOOL progressive_set_surface_data(PROGRESSIVE_CONTEXT* progressive, UINT16 surfaceId,
                                                void* pData)
{
  ULONG_PTR key;
  key = ((ULONG_PTR)surfaceId) + 1;

  if (pData)
    return HashTable_Insert(progressive->SurfaceContexts, (void*)key, pData);

  HashTable_Remove(progressive->SurfaceContexts, (void*)key);
  return TRUE;
}

static INLINE PROGRESSIVE_SURFACE_CONTEXT*
progressive_get_surface_data(PROGRESSIVE_CONTEXT* progressive, UINT16 surfaceId)
{
  void* key = (void*)(((ULONG_PTR)surfaceId) + 1);

  if (!progressive)
    return NULL;

  return HashTable_GetItemValue(progressive->SurfaceContexts, key);
}

static void progressive_tile_free(RFX_PROGRESSIVE_TILE* tile)
{
  if (tile)
  {
    winpr_aligned_free(tile->sign);
    winpr_aligned_free(tile->current);
    winpr_aligned_free(tile->data);
    winpr_aligned_free(tile);
  }
}

static void progressive_surface_context_free(void* ptr)
{
  PROGRESSIVE_SURFACE_CONTEXT* surface = ptr;
  UINT32 index;

  if (!surface)
    return;

  if (surface->tiles)
  {
    for (index = 0; index < surface->tilesSize; index++)
    {
      RFX_PROGRESSIVE_TILE* tile = surface->tiles[index];
      progressive_tile_free(tile);
    }
  }

  winpr_aligned_free(surface->tiles);
  winpr_aligned_free(surface->updatedTileIndices);
  winpr_aligned_free(surface);
}

static INLINE RFX_PROGRESSIVE_TILE* progressive_tile_new(void)
{
  RFX_PROGRESSIVE_TILE* tile = winpr_aligned_calloc(1, sizeof(RFX_PROGRESSIVE_TILE), 32);
  if (!tile)
    goto fail;

  tile->width = 64;
  tile->height = 64;
  tile->stride = 4 * tile->width;

  size_t dataLen = 1ull * tile->stride * tile->height;
  tile->data = (BYTE*)winpr_aligned_malloc(dataLen, 16);
  if (!tile->data)
    goto fail;
  memset(tile->data, 0xFF, dataLen);

  size_t signLen = (8192 + 32) * 3;
  tile->sign = (BYTE*)winpr_aligned_calloc(signLen, sizeof(BYTE), 16);
  if (!tile->sign)
    goto fail;

  size_t currentLen = (8192 + 32) * 3;
  tile->current = (BYTE*)winpr_aligned_calloc(currentLen, sizeof(BYTE), 16);
  if (!tile->current)
    goto fail;

  return tile;

fail:
  progressive_tile_free(tile);
  return NULL;
}

static BOOL progressive_allocate_tile_cache(PROGRESSIVE_SURFACE_CONTEXT* surface, size_t min)
{
  size_t oldIndex = 0;

  WINPR_ASSERT(surface);
  WINPR_ASSERT(surface->gridSize > 0);

  if (surface->tiles)
  {
    oldIndex = surface->gridSize;
    while (surface->gridSize < min)
      surface->gridSize += 1024;
  }

  void* tmp = winpr_aligned_recalloc(surface->tiles, surface->gridSize,
                                     sizeof(RFX_PROGRESSIVE_TILE*), 32);
  if (!tmp)
    return FALSE;
  surface->tilesSize = surface->gridSize;
  surface->tiles = tmp;

  for (size_t x = oldIndex; x < surface->tilesSize; x++)
  {
    surface->tiles[x] = progressive_tile_new();
    if (!surface->tiles[x])
      return FALSE;
  }

  tmp =
      winpr_aligned_recalloc(surface->updatedTileIndices, surface->gridSize, sizeof(UINT32), 32);
  if (!tmp)
    return FALSE;

  surface->updatedTileIndices = tmp;

  return TRUE;
}

static PROGRESSIVE_SURFACE_CONTEXT* progressive_surface_context_new(UINT16 surfaceId, UINT32 width,
                                                                    UINT32 height)
{
  PROGRESSIVE_SURFACE_CONTEXT* surface = (PROGRESSIVE_SURFACE_CONTEXT*)winpr_aligned_calloc(
      1, sizeof(PROGRESSIVE_SURFACE_CONTEXT), 32);

  if (!surface)
    return NULL;

  surface->id = surfaceId;
  surface->width = width;
  surface->height = height;
  surface->gridWidth = (width + (64 - width % 64)) / 64;
  surface->gridHeight = (height + (64 - height % 64)) / 64;
  surface->gridSize = surface->gridWidth * surface->gridHeight;

  if (!progressive_allocate_tile_cache(surface, surface->gridSize))
  {
    progressive_surface_context_free(surface);
    return NULL;
  }

  return surface;
}

static BOOL progressive_surface_tile_replace(PROGRESSIVE_SURFACE_CONTEXT* surface,
                                             PROGRESSIVE_BLOCK_REGION* region,
                                             const RFX_PROGRESSIVE_TILE* tile, BOOL upgrade)
{
  RFX_PROGRESSIVE_TILE* t;

  size_t zIdx;
  if (!surface || !tile)
    return FALSE;

  zIdx = (tile->yIdx * surface->gridWidth) + tile->xIdx;

  if (zIdx >= surface->tilesSize)
  {
    WLog_ERR(TAG, "Invalid zIndex %" PRIuz, zIdx);
    return FALSE;
  }

  t = surface->tiles[zIdx];

  t->blockType = tile->blockType;
  t->blockLen = tile->blockLen;
  t->quantIdxY = tile->quantIdxY;
  t->quantIdxCb = tile->quantIdxCb;
  t->quantIdxCr = tile->quantIdxCr;
  t->xIdx = tile->xIdx;
  t->yIdx = tile->yIdx;
  t->flags = tile->flags;
  t->quality = tile->quality;
  t->x = tile->xIdx * t->width;
  t->y = tile->yIdx * t->height;

  if (upgrade)
  {
    t->ySrlLen = tile->ySrlLen;
    t->yRawLen = tile->yRawLen;
    t->cbSrlLen = tile->cbSrlLen;
    t->cbRawLen = tile->cbRawLen;
    t->crSrlLen = tile->crSrlLen;
    t->crRawLen = tile->crRawLen;
    t->ySrlData = tile->ySrlData;
    t->yRawData = tile->yRawData;
    t->cbSrlData = tile->cbSrlData;
    t->cbRawData = tile->cbRawData;
    t->crSrlData = tile->crSrlData;
    t->crRawData = tile->crRawData;
  }
  else
  {
    t->yLen = tile->yLen;
    t->cbLen = tile->cbLen;
    t->crLen = tile->crLen;
    t->tailLen = tile->tailLen;
    t->yData = tile->yData;
    t->cbData = tile->cbData;
    t->crData = tile->crData;
    t->tailData = tile->tailData;
  }

  if (region->usedTiles >= region->numTiles)
  {
    WLog_ERR(TAG, "Invalid tile count, only expected %" PRIu16 ", got %" PRIu16,
             region->numTiles, region->usedTiles);
    return FALSE;
  }

  region->tiles[region->usedTiles++] = t;
  if (!t->dirty)
  {
    if (surface->numUpdatedTiles >= surface->gridSize)
    {
      if (!progressive_allocate_tile_cache(surface, surface->numUpdatedTiles + 1))
        return FALSE;
    }

    surface->updatedTileIndices[surface->numUpdatedTiles++] = (UINT32)zIdx;
  }

  t->dirty = TRUE;
  return TRUE;
}

INT32 progressive_create_surface_context(PROGRESSIVE_CONTEXT* progressive, UINT16 surfaceId,
                                         UINT32 width, UINT32 height)
{
  PROGRESSIVE_SURFACE_CONTEXT* surface = progressive_get_surface_data(progressive, surfaceId);

  if (!surface)
  {
    surface = progressive_surface_context_new(surfaceId, width, height);

    if (!surface)
      return -1;

    if (!progressive_set_surface_data(progressive, surfaceId, (void*)surface))
    {
      progressive_surface_context_free(surface);
      return -1;
    }
  }

  return 1;
}

int progressive_delete_surface_context(PROGRESSIVE_CONTEXT* progressive, UINT16 surfaceId)
{
  progressive_set_surface_data(progressive, surfaceId, NULL);

  return 1;
}

/*
 * Band     Offset      Dimensions  Size
 *
 * HL1      0           31x33       1023
 * LH1      1023        33x31       1023
 * HH1      2046        31x31       961
 *
 * HL2      3007        16x17       272
 * LH2      3279        17x16       272
 * HH2      3551        16x16       256
 *
 * HL3      3807        8x9         72
 * LH3      3879        9x8         72
 * HH3      3951        8x8         64
 *
 * LL3      4015        9x9         81
 */

static INLINE void progressive_rfx_idwt_x(const INT16* pLowBand, size_t nLowStep,
                                          const INT16* pHighBand, size_t nHighStep, INT16* pDstBand,
                                          size_t nDstStep, size_t nLowCount, size_t nHighCount,
                                          size_t nDstCount)
{
  size_t i;
  INT16 L0;
  INT16 H0, H1;
  INT16 X0, X1, X2;

  for (i = 0; i < nDstCount; i++)
  {
    size_t j;
    const INT16* pL = pLowBand;
    const INT16* pH = pHighBand;
    INT16* pX = pDstBand;
    H0 = *pH++;
    L0 = *pL++;
    X0 = L0 - H0;
    X2 = L0 - H0;

    for (j = 0; j < (nHighCount - 1); j++)
    {
      H1 = *pH;
      pH++;
      L0 = *pL;
      pL++;
      X2 = L0 - ((H0 + H1) / 2);
      X1 = ((X0 + X2) / 2) + (2 * H0);
      pX[0] = X0;
      pX[1] = X1;
      pX += 2;
      X0 = X2;
      H0 = H1;
    }

    if (nLowCount <= (nHighCount + 1))
    {
      if (nLowCount <= nHighCount)
      {
        pX[0] = X2;
        pX[1] = X2 + (2 * H0);
      }
      else
      {
        L0 = *pL;
        pL++;
        X0 = L0 - H0;
        pX[0] = X2;
        pX[1] = ((X0 + X2) / 2) + (2 * H0);
        pX[2] = X0;
      }
    }
    else
    {
      L0 = *pL;
      pL++;
      X0 = L0 - (H0 / 2);
      pX[0] = X2;
      pX[1] = ((X0 + X2) / 2) + (2 * H0);
      pX[2] = X0;
      L0 = *pL;
      pL++;
      pX[3] = (X0 + L0) / 2;
    }

    pLowBand += nLowStep;
    pHighBand += nHighStep;
    pDstBand += nDstStep;
  }
}

static INLINE void progressive_rfx_idwt_y(const INT16* pLowBand, size_t nLowStep,
                                          const INT16* pHighBand, size_t nHighStep, INT16* pDstBand,
                                          size_t nDstStep, size_t nLowCount, size_t nHighCount,
                                          size_t nDstCount)
{
  size_t i;
  INT16 L0;
  INT16 H0, H1;
  INT16 X0, X1, X2;

  for (i = 0; i < nDstCount; i++)
  {
    size_t j;
    const INT16* pL = pLowBand;
    const INT16* pH = pHighBand;
    INT16* pX = pDstBand;
    H0 = *pH;
    pH += nHighStep;
    L0 = *pL;
    pL += nLowStep;
    X0 = L0 - H0;
    X2 = L0 - H0;

    for (j = 0; j < (nHighCount - 1); j++)
    {
      H1 = *pH;
      pH += nHighStep;
      L0 = *pL;
      pL += nLowStep;
      X2 = L0 - ((H0 + H1) / 2);
      X1 = ((X0 + X2) / 2) + (2 * H0);
      *pX = X0;
      pX += nDstStep;
      *pX = X1;
      pX += nDstStep;
      X0 = X2;
      H0 = H1;
    }

    if (nLowCount <= (nHighCount + 1))
    {
      if (nLowCount <= nHighCount)
      {
        *pX = X2;
        pX += nDstStep;
        *pX = X2 + (2 * H0);
      }
      else
      {
        L0 = *pL;
        X0 = L0 - H0;
        *pX = X2;
        pX += nDstStep;
        *pX = ((X0 + X2) / 2) + (2 * H0);
        pX += nDstStep;
        *pX = X0;
      }
    }
    else
    {
      L0 = *pL;
      pL += nLowStep;
      X0 = L0 - (H0 / 2);
      *pX = X2;
      pX += nDstStep;
      *pX = ((X0 + X2) / 2) + (2 * H0);
      pX += nDstStep;
      *pX = X0;
      pX += nDstStep;
      L0 = *pL;
      *pX = (X0 + L0) / 2;
    }

    pLowBand++;
    pHighBand++;
    pDstBand++;
  }
}

static INLINE size_t progressive_rfx_get_band_l_count(size_t level)
{
  return (64 >> level) + 1;
}

static INLINE size_t progressive_rfx_get_band_h_count(size_t level)
{
  if (level == 1)
    return (64 >> 1) - 1;
  else
    return (64 + (1 << (level - 1))) >> level;
}

static INLINE void progressive_rfx_dwt_2d_decode_block(INT16* WINPR_RESTRICT buffer,
                                                       INT16* WINPR_RESTRICT temp, size_t level)
{
  size_t nDstStepX;
  size_t nDstStepY;
  INT16 *WINPR_RESTRICT HL, *WINPR_RESTRICT LH;
  INT16 *WINPR_RESTRICT HH, *WINPR_RESTRICT LL;
  INT16 *WINPR_RESTRICT L, *WINPR_RESTRICT H, *WINPR_RESTRICT LLx;

  const size_t nBandL = progressive_rfx_get_band_l_count(level);
  const size_t nBandH = progressive_rfx_get_band_h_count(level);
  size_t offset = 0;

  HL = &buffer[offset];
  offset += (nBandH * nBandL);
  LH = &buffer[offset];
  offset += (nBandL * nBandH);
  HH = &buffer[offset];
  offset += (nBandH * nBandH);
  LL = &buffer[offset];
  nDstStepX = (nBandL + nBandH);
  nDstStepY = (nBandL + nBandH);
  offset = 0;
  L = &temp[offset];
  offset += (nBandL * nDstStepX);
  H = &temp[offset];
  LLx = &buffer[0];

  /* horizontal (LL + HL -> L) */
  progressive_rfx_idwt_x(LL, nBandL, HL, nBandH, L, nDstStepX, nBandL, nBandH, nBandL);

  /* horizontal (LH + HH -> H) */
  progressive_rfx_idwt_x(LH, nBandL, HH, nBandH, H, nDstStepX, nBandL, nBandH, nBandH);

  /* vertical (L + H -> LL) */
  progressive_rfx_idwt_y(L, nDstStepX, H, nDstStepX, LLx, nDstStepY, nBandL, nBandH,
                         nBandL + nBandH);
}

void rfx_dwt_2d_extrapolate_decode(INT16* WINPR_RESTRICT buffer, INT16* WINPR_RESTRICT temp)
{
  WINPR_ASSERT(buffer);
  WINPR_ASSERT(temp);
  progressive_rfx_dwt_2d_decode_block(&buffer[3807], temp, 3);
  progressive_rfx_dwt_2d_decode_block(&buffer[3007], temp, 2);
  progressive_rfx_dwt_2d_decode_block(&buffer[0], temp, 1);
}

static INLINE int progressive_rfx_dwt_2d_decode(PROGRESSIVE_CONTEXT* progressive, INT16* buffer,
                                                INT16* current, BOOL coeffDiff, BOOL extrapolate,
                                                BOOL reverse)
{
  const primitives_t* prims = primitives_get();

  if (!progressive || !buffer || !current)
    return -1;

  INT16 dst[4096] = { 0 };
  if (reverse)
    memcpy(buffer, current, sizeof(dst));
  else
  {
    if (coeffDiff)
    {
      prims->add_16s(buffer, current, dst, ARRAYSIZE(dst));
      memcpy(current, dst, sizeof(dst));
      memcpy(buffer, dst, sizeof(dst));
    }
    else
      memcpy(current, buffer, sizeof(dst));
  }

  INT16* temp = (INT16*)BufferPool_Take(progressive->bufferPool, -1); /* DWT buffer */

  if (!temp)
    return -2;

  if (!extrapolate)
  {
    progressive->rfx_context->dwt_2d_decode(buffer, temp);
  }
  else
  {
    WINPR_ASSERT(progressive->rfx_context->dwt_2d_extrapolate_decode);
    progressive->rfx_context->dwt_2d_extrapolate_decode(buffer, temp);
  }
  BufferPool_Return(progressive->bufferPool, temp);
  return 1;
}

static INLINE void progressive_rfx_decode_block(const primitives_t* prims, INT16* buffer,
                                                UINT32 length, UINT32 shift)
{
  if (!shift)
    return;

  WINPR_ASSERT(length <= 1024);
  INT16 dst[1024] = { 0 };
  prims->lShiftC_16s(buffer, shift, dst, length);
  memcpy(buffer, dst, length * sizeof(INT16));
}

static INLINE int progressive_rfx_decode_component(PROGRESSIVE_CONTEXT* progressive,
                                                   const RFX_COMPONENT_CODEC_QUANT* shift,
                                                   const BYTE* data, UINT32 length, INT16* buffer,
                                                   INT16* current, INT16* sign, BOOL coeffDiff,
                                                   BOOL subbandDiff, BOOL extrapolate)
{
  int status;
  const primitives_t* prims = primitives_get();

  status = progressive->rfx_context->rlgr_decode(RLGR1, data, length, buffer, 4096);

  if (status < 0)
    return status;

  CopyMemory(sign, buffer, 4096 * 2);
  if (!extrapolate)
  {
    rfx_differential_decode(buffer + 4032, 64);
    progressive_rfx_decode_block(prims, &buffer[0], 1024, shift->HL1);    /* HL1 */
    progressive_rfx_decode_block(prims, &buffer[1024], 1024, shift->LH1); /* LH1 */
    progressive_rfx_decode_block(prims, &buffer[2048], 1024, shift->HH1); /* HH1 */
    progressive_rfx_decode_block(prims, &buffer[3072], 256, shift->HL2);  /* HL2 */
    progressive_rfx_decode_block(prims, &buffer[3328], 256, shift->LH2);  /* LH2 */
    progressive_rfx_decode_block(prims, &buffer[3584], 256, shift->HH2);  /* HH2 */
    progressive_rfx_decode_block(prims, &buffer[3840], 64, shift->HL3);   /* HL3 */
    progressive_rfx_decode_block(prims, &buffer[3904], 64, shift->LH3);   /* LH3 */
    progressive_rfx_decode_block(prims, &buffer[3968], 64, shift->HH3);   /* HH3 */
    progressive_rfx_decode_block(prims, &buffer[4032], 64, shift->LL3);   /* LL3 */
  }
  else
  {
    progressive_rfx_decode_block(prims, &buffer[0], 1023, shift->HL1);    /* HL1 */
    progressive_rfx_decode_block(prims, &buffer[1023], 1023, shift->LH1); /* LH1 */
    progressive_rfx_decode_block(prims, &buffer[2046], 961, shift->HH1);  /* HH1 */
    progressive_rfx_decode_block(prims, &buffer[3007], 272, shift->HL2);  /* HL2 */
    progressive_rfx_decode_block(prims, &buffer[3279], 272, shift->LH2);  /* LH2 */
    progressive_rfx_decode_block(prims, &buffer[3551], 256, shift->HH2);  /* HH2 */
    progressive_rfx_decode_block(prims, &buffer[3807], 72, shift->HL3);   /* HL3 */
    progressive_rfx_decode_block(prims, &buffer[3879], 72, shift->LH3);   /* LH3 */
    progressive_rfx_decode_block(prims, &buffer[3951], 64, shift->HH3);   /* HH3 */
    rfx_differential_decode(&buffer[4015], 81);                           /* LL3 */
    progressive_rfx_decode_block(prims, &buffer[4015], 81, shift->LL3);   /* LL3 */
  }
  return progressive_rfx_dwt_2d_decode(progressive, buffer, current, coeffDiff, extrapolate,
                                       FALSE);
}

static INLINE int progressive_decompress_tile_first(PROGRESSIVE_CONTEXT* progressive,
                                                    RFX_PROGRESSIVE_TILE* tile,
                                                    PROGRESSIVE_BLOCK_REGION* region,
                                                    const PROGRESSIVE_BLOCK_CONTEXT* context)
{
  int rc;
  BOOL diff, sub, extrapolate;
  BYTE* pBuffer;
  INT16* pSign[3];
  INT16* pSrcDst[3];
  INT16* pCurrent[3];
  RFX_COMPONENT_CODEC_QUANT shiftY = { 0 };
  RFX_COMPONENT_CODEC_QUANT shiftCb = { 0 };
  RFX_COMPONENT_CODEC_QUANT shiftCr = { 0 };
  RFX_COMPONENT_CODEC_QUANT* quantY;
  RFX_COMPONENT_CODEC_QUANT* quantCb;
  RFX_COMPONENT_CODEC_QUANT* quantCr;
  RFX_COMPONENT_CODEC_QUANT* quantProgY;
  RFX_COMPONENT_CODEC_QUANT* quantProgCb;
  RFX_COMPONENT_CODEC_QUANT* quantProgCr;
  RFX_PROGRESSIVE_CODEC_QUANT* quantProgVal;
  static const prim_size_t roi_64x64 = { 64, 64 };
  const primitives_t* prims = primitives_get();

  tile->pass = 1;
  diff = tile->flags & RFX_TILE_DIFFERENCE;
  sub = context->flags & RFX_SUBBAND_DIFFING;
  extrapolate = region->flags & RFX_DWT_REDUCE_EXTRAPOLATE;

#if defined(WITH_DEBUG_CODECS)
  WLog_Print(progressive->log, WLOG_DEBUG,
             "ProgressiveTile%s: quantIdx Y: %" PRIu8 " Cb: %" PRIu8 " Cr: %" PRIu8
             " xIdx: %" PRIu16 " yIdx: %" PRIu16 " flags: 0x%02" PRIX8 " quality: %" PRIu8
             " yLen: %" PRIu16 " cbLen: %" PRIu16 " crLen: %" PRIu16 " tailLen: %" PRIu16 "",
             (tile->blockType == PROGRESSIVE_WBT_TILE_FIRST) ? "First" : "Simple",
             tile->quantIdxY, tile->quantIdxCb, tile->quantIdxCr, tile->xIdx, tile->yIdx,
             tile->flags, tile->quality, tile->yLen, tile->cbLen, tile->crLen, tile->tailLen);
#endif

  if (tile->quantIdxY >= region->numQuant)
  {
    WLog_ERR(TAG, "quantIdxY %" PRIu8 " > numQuant %" PRIu8, tile->quantIdxY, region->numQuant);
    return -1;
  }

  quantY = &(region->quantVals[tile->quantIdxY]);

  if (tile->quantIdxCb >= region->numQuant)
  {
    WLog_ERR(TAG, "quantIdxCb %" PRIu8 " > numQuant %" PRIu8, tile->quantIdxCb,
             region->numQuant);
    return -1;
  }

  quantCb = &(region->quantVals[tile->quantIdxCb]);

  if (tile->quantIdxCr >= region->numQuant)
  {
    WLog_ERR(TAG, "quantIdxCr %" PRIu8 " > numQuant %" PRIu8, tile->quantIdxCr,
             region->numQuant);
    return -1;
  }

  quantCr = &(region->quantVals[tile->quantIdxCr]);

  if (tile->quality == 0xFF)
  {
    quantProgVal = &(progressive->quantProgValFull);
  }
  else
  {
    if (tile->quality >= region->numProgQuant)
    {
      WLog_ERR(TAG, "quality %" PRIu8 " > numProgQuant %" PRIu8, tile->quality,
               region->numProgQuant);
      return -1;
    }

    quantProgVal = &(region->quantProgVals[tile->quality]);
  }

  quantProgY = &(quantProgVal->yQuantValues);
  quantProgCb = &(quantProgVal->cbQuantValues);
  quantProgCr = &(quantProgVal->crQuantValues);

  tile->yQuant = *quantY;
  tile->cbQuant = *quantCb;
  tile->crQuant = *quantCr;
  tile->yProgQuant = *quantProgY;
  tile->cbProgQuant = *quantProgCb;
  tile->crProgQuant = *quantProgCr;

  progressive_rfx_quant_add(quantY, quantProgY, &(tile->yBitPos));
  progressive_rfx_quant_add(quantCb, quantProgCb, &(tile->cbBitPos));
  progressive_rfx_quant_add(quantCr, quantProgCr, &(tile->crBitPos));
  progressive_rfx_quant_add(quantY, quantProgY, &shiftY);
  progressive_rfx_quant_lsub(&shiftY, 1); /* -6 + 5 = -1 */
  progressive_rfx_quant_add(quantCb, quantProgCb, &shiftCb);
  progressive_rfx_quant_lsub(&shiftCb, 1); /* -6 + 5 = -1 */
  progressive_rfx_quant_add(quantCr, quantProgCr, &shiftCr);
  progressive_rfx_quant_lsub(&shiftCr, 1); /* -6 + 5 = -1 */

  pSign[0] = (INT16*)((BYTE*)(&tile->sign[((8192 + 32) * 0) + 16])); /* Y/R buffer */
  pSign[1] = (INT16*)((BYTE*)(&tile->sign[((8192 + 32) * 1) + 16])); /* Cb/G buffer */
  pSign[2] = (INT16*)((BYTE*)(&tile->sign[((8192 + 32) * 2) + 16])); /* Cr/B buffer */

  pCurrent[0] = (INT16*)((BYTE*)(&tile->current[((8192 + 32) * 0) + 16])); /* Y/R buffer */
  pCurrent[1] = (INT16*)((BYTE*)(&tile->current[((8192 + 32) * 1) + 16])); /* Cb/G buffer */
  pCurrent[2] = (INT16*)((BYTE*)(&tile->current[((8192 + 32) * 2) + 16])); /* Cr/B buffer */

  pBuffer = (BYTE*)BufferPool_Take(progressive->bufferPool, -1);
  pSrcDst[0] = (INT16*)((BYTE*)(&pBuffer[((8192 + 32) * 0) + 16])); /* Y/R buffer */
  pSrcDst[1] = (INT16*)((BYTE*)(&pBuffer[((8192 + 32) * 1) + 16])); /* Cb/G buffer */
  pSrcDst[2] = (INT16*)((BYTE*)(&pBuffer[((8192 + 32) * 2) + 16])); /* Cr/B buffer */

  rc = progressive_rfx_decode_component(progressive, &shiftY, tile->yData, tile->yLen, pSrcDst[0],
                                        pCurrent[0], pSign[0], diff, sub, extrapolate); /* Y */
  if (rc < 0)
    goto fail;
  rc = progressive_rfx_decode_component(progressive, &shiftCb, tile->cbData, tile->cbLen,
                                        pSrcDst[1], pCurrent[1], pSign[1], diff, sub,
                                        extrapolate); /* Cb */
  if (rc < 0)
    goto fail;
  rc = progressive_rfx_decode_component(progressive, &shiftCr, tile->crData, tile->crLen,
                                        pSrcDst[2], pCurrent[2], pSign[2], diff, sub,
                                        extrapolate); /* Cr */
  if (rc < 0)
    goto fail;

  rc = prims->yCbCrToRGB_16s8u_P3AC4R((const INT16* const*)pSrcDst, 64 * 2, tile->data,
                                      tile->stride, progressive->format, &roi_64x64);
fail:
  BufferPool_Return(progressive->bufferPool, pBuffer);
  return rc;
}

static INLINE INT16 progressive_rfx_srl_read(RFX_PROGRESSIVE_UPGRADE_STATE* state, UINT32 numBits)
{
  UINT32 k;
  UINT32 bit;
  UINT32 max;
  UINT32 mag;
  UINT32 sign;
  wBitStream* bs = state->srl;

  if (state->nz)
  {
    state->nz--;
    return 0;
  }

  k = state->kp / 8;

  if (!state->mode)
  {
    /* zero encoding */
    bit = (bs->accumulator & 0x80000000) ? 1 : 0;
    BitStream_Shift(bs, 1);

    if (!bit)
    {
      /* '0' bit, nz >= (1 << k), nz = (1 << k) */
      state->nz = (1 << k);
      state->kp += 4;

      if (state->kp > 80)
        state->kp = 80;

      state->nz--;
      return 0;
    }
    else
    {
      /* '1' bit, nz < (1 << k), nz = next k bits */
      state->nz = 0;
      state->mode = 1; /* unary encoding is next */

      if (k)
      {
        bs->mask = ((1 << k) - 1);
        state->nz = ((bs->accumulator >> (32u - k)) & bs->mask);
        BitStream_Shift(bs, k);
      }

      if (state->nz)
      {
        state->nz--;
        return 0;
      }
    }
  }

  state->mode = 0; /* zero encoding is next */
  /* unary encoding */
  /* read sign bit */
  sign = (bs->accumulator & 0x80000000) ? 1 : 0;
  BitStream_Shift(bs, 1);

  if (state->kp < 6)
    state->kp = 0;
  else
    state->kp -= 6;

  if (numBits == 1)
    return sign ? -1 : 1;

  mag = 1;
  max = (1 << numBits) - 1;

  while (mag < max)
  {
    bit = (bs->accumulator & 0x80000000) ? 1 : 0;
    BitStream_Shift(bs, 1);

    if (bit)
      break;

    mag++;
  }

  return sign ? -1 * mag : mag;
}

static INLINE int progressive_rfx_upgrade_state_finish(RFX_PROGRESSIVE_UPGRADE_STATE* state)
{
  UINT32 pad;
  wBitStream* srl;
  wBitStream* raw;
  if (!state)
    return -1;

  srl = state->srl;
  raw = state->raw;
  /* Read trailing bits from RAW/SRL bit streams */
  pad = (raw->position % 8) ? (8 - (raw->position % 8)) : 0;

  if (pad)
    BitStream_Shift(raw, pad);

  pad = (srl->position % 8) ? (8 - (srl->position % 8)) : 0;

  if (pad)
    BitStream_Shift(srl, pad);

  if (BitStream_GetRemainingLength(srl) == 8)
    BitStream_Shift(srl, 8);

  return 1;
}

static INLINE int progressive_rfx_upgrade_block(RFX_PROGRESSIVE_UPGRADE_STATE* state, INT16* buffer,
                                                INT16* sign, UINT32 length, UINT32 shift,
                                                UINT32 bitPos, UINT32 numBits)
{
  UINT32 index;
  INT16 input;
  wBitStream* raw;

  if (!numBits)
    return 1;

  raw = state->raw;

  if (!state->nonLL)
  {
    for (index = 0; index < length; index++)
    {
      raw->mask = ((1 << numBits) - 1);
      input = (INT16)((raw->accumulator >> (32 - numBits)) & raw->mask);
      BitStream_Shift(raw, numBits);
      buffer[index] += (input << shift);
    }

    return 1;
  }

  for (index = 0; index < length; index++)
  {
    if (sign[index] > 0)
    {
      /* sign > 0, read from raw */
      raw->mask = ((1 << numBits) - 1);
      input = (INT16)((raw->accumulator >> (32 - numBits)) & raw->mask);
      BitStream_Shift(raw, numBits);
    }
    else if (sign[index] < 0)
    {
      /* sign < 0, read from raw */
      raw->mask = ((1 << numBits) - 1);
      input = (INT16)((raw->accumulator >> (32 - numBits)) & raw->mask);
      BitStream_Shift(raw, numBits);
      input *= -1;
    }
    else
    {
      /* sign == 0, read from srl */
      input = progressive_rfx_srl_read(state, numBits);
      sign[index] = input;
    }

    buffer[index] += (INT16)((UINT32)input << shift);
  }

  return 1;
}

static INLINE int progressive_rfx_upgrade_component(
    PROGRESSIVE_CONTEXT* progressive, const RFX_COMPONENT_CODEC_QUANT* shift,
    const RFX_COMPONENT_CODEC_QUANT* bitPos, const RFX_COMPONENT_CODEC_QUANT* numBits,
    INT16* buffer, INT16* current, INT16* sign, const BYTE* srlData, UINT32 srlLen,
    const BYTE* rawData, UINT32 rawLen, BOOL coeffDiff, BOOL subbandDiff, BOOL extrapolate)
{
  int rc;
  UINT32 aRawLen;
  UINT32 aSrlLen;
  wBitStream s_srl = { 0 };
  wBitStream s_raw = { 0 };
  RFX_PROGRESSIVE_UPGRADE_STATE state = { 0 };

  state.kp = 8;
  state.mode = 0;
  state.srl = &s_srl;
  state.raw = &s_raw;
  BitStream_Attach(state.srl, srlData, srlLen);
  BitStream_Fetch(state.srl);
  BitStream_Attach(state.raw, rawData, rawLen);
  BitStream_Fetch(state.raw);

  state.nonLL = TRUE;
  rc = progressive_rfx_upgrade_block(&state, &current[0], &sign[0], 1023, shift->HL1, bitPos->HL1,
                                     numBits->HL1); /* HL1 */
  if (rc < 0)
    return rc;
  rc = progressive_rfx_upgrade_block(&state, &current[1023], &sign[1023], 1023, shift->LH1,
                                     bitPos->LH1, numBits->LH1); /* LH1 */
  if (rc < 0)
    return rc;
  rc = progressive_rfx_upgrade_block(&state, &current[2046], &sign[2046], 961, shift->HH1,
                                     bitPos->HH1, numBits->HH1); /* HH1 */
  if (rc < 0)
    return rc;
  rc = progressive_rfx_upgrade_block(&state, &current[3007], &sign[3007], 272, shift->HL2,
                                     bitPos->HL2, numBits->HL2); /* HL2 */
  if (rc < 0)
    return rc;
  rc = progressive_rfx_upgrade_block(&state, &current[3279], &sign[3279], 272, shift->LH2,
                                     bitPos->LH2, numBits->LH2); /* LH2 */
  if (rc < 0)
    return rc;
  rc = progressive_rfx_upgrade_block(&state, &current[3551], &sign[3551], 256, shift->HH2,
                                     bitPos->HH2, numBits->HH2); /* HH2 */
  if (rc < 0)
    return rc;
  rc = progressive_rfx_upgrade_block(&state, &current[3807], &sign[3807], 72, shift->HL3,
                                     bitPos->HL3, numBits->HL3); /* HL3 */
  if (rc < 0)
    return rc;
  rc = progressive_rfx_upgrade_block(&state, &current[3879], &sign[3879], 72, shift->LH3,
                                     bitPos->LH3, numBits->LH3); /* LH3 */
  if (rc < 0)
    return rc;
  rc = progressive_rfx_upgrade_block(&state, &current[3951], &sign[3951], 64, shift->HH3,
                                     bitPos->HH3, numBits->HH3); /* HH3 */
  if (rc < 0)
    return rc;

  state.nonLL = FALSE;
  rc = progressive_rfx_upgrade_block(&state, &current[4015], &sign[4015], 81, shift->LL3,
                                     bitPos->LL3, numBits->LL3); /* LL3 */
  if (rc < 0)
    return rc;
  rc = progressive_rfx_upgrade_state_finish(&state);
  if (rc < 0)
    return rc;
  aRawLen = (state.raw->position + 7) / 8;
  aSrlLen = (state.srl->position + 7) / 8;

  if ((aRawLen != rawLen) || (aSrlLen != srlLen))
  {
    int pRawLen = 0;
    int pSrlLen = 0;

    if (rawLen)
      pRawLen = (int)((((float)aRawLen) / ((float)rawLen)) * 100.0f);

    if (srlLen)
      pSrlLen = (int)((((float)aSrlLen) / ((float)srlLen)) * 100.0f);

    WLog_Print(progressive->log, WLOG_WARN,
               "RAW: %" PRIu32 "/%" PRIu32 " %d%% (%" PRIu32 "/%" PRIu32 ":%" PRIu32
               ")\tSRL: %" PRIu32 "/%" PRIu32 " %d%% (%" PRIu32 "/%" PRIu32 ":%" PRIu32 ")",
               aRawLen, rawLen, pRawLen, state.raw->position, rawLen * 8,
               (rawLen * 8) - state.raw->position, aSrlLen, srlLen, pSrlLen,
               state.srl->position, srlLen * 8, (srlLen * 8) - state.srl->position);
    return -1;
  }

  return progressive_rfx_dwt_2d_decode(progressive, buffer, current, coeffDiff, extrapolate,
                                       TRUE);
}

static INLINE int progressive_decompress_tile_upgrade(PROGRESSIVE_CONTEXT* progressive,
                                                      RFX_PROGRESSIVE_TILE* tile,
                                                      PROGRESSIVE_BLOCK_REGION* region,
                                                      const PROGRESSIVE_BLOCK_CONTEXT* context)
{
  int status;
  BOOL coeffDiff, sub, extrapolate;
  BYTE* pBuffer;
  INT16* pSign[3] = { 0 };
  INT16* pSrcDst[3] = { 0 };
  INT16* pCurrent[3] = { 0 };
  RFX_COMPONENT_CODEC_QUANT shiftY = { 0 };
  RFX_COMPONENT_CODEC_QUANT shiftCb = { 0 };
  RFX_COMPONENT_CODEC_QUANT shiftCr = { 0 };
  RFX_COMPONENT_CODEC_QUANT yBitPos = { 0 };
  RFX_COMPONENT_CODEC_QUANT cbBitPos = { 0 };
  RFX_COMPONENT_CODEC_QUANT crBitPos = { 0 };
  RFX_COMPONENT_CODEC_QUANT yNumBits = { 0 };
  RFX_COMPONENT_CODEC_QUANT cbNumBits = { 0 };
  RFX_COMPONENT_CODEC_QUANT crNumBits = { 0 };
  RFX_COMPONENT_CODEC_QUANT* quantY;
  RFX_COMPONENT_CODEC_QUANT* quantCb;
  RFX_COMPONENT_CODEC_QUANT* quantCr;
  RFX_COMPONENT_CODEC_QUANT* quantProgY;
  RFX_COMPONENT_CODEC_QUANT* quantProgCb;
  RFX_COMPONENT_CODEC_QUANT* quantProgCr;
  RFX_PROGRESSIVE_CODEC_QUANT* quantProg;
  static const prim_size_t roi_64x64 = { 64, 64 };
  const primitives_t* prims = primitives_get();

  coeffDiff = tile->flags & RFX_TILE_DIFFERENCE;
  sub = context->flags & RFX_SUBBAND_DIFFING;
  extrapolate = region->flags & RFX_DWT_REDUCE_EXTRAPOLATE;

  tile->pass++;

#if defined(WITH_DEBUG_CODECS)
  WLog_Print(progressive->log, WLOG_DEBUG,
             "ProgressiveTileUpgrade: pass: %" PRIu16 " quantIdx Y: %" PRIu8 " Cb: %" PRIu8
             " Cr: %" PRIu8 " xIdx: %" PRIu16 " yIdx: %" PRIu16 " quality: %" PRIu8
             " ySrlLen: %" PRIu16 " yRawLen: %" PRIu16 " cbSrlLen: %" PRIu16 " cbRawLen: %" PRIu16
             " crSrlLen: %" PRIu16 " crRawLen: %" PRIu16 "",
             tile->pass, tile->quantIdxY, tile->quantIdxCb, tile->quantIdxCr, tile->xIdx,
             tile->yIdx, tile->quality, tile->ySrlLen, tile->yRawLen, tile->cbSrlLen,
             tile->cbRawLen, tile->crSrlLen, tile->crRawLen);
#endif

  if (tile->quantIdxY >= region->numQuant)
  {
    WLog_ERR(TAG, "quantIdxY %" PRIu8 " > numQuant %" PRIu8, tile->quantIdxY, region->numQuant);
    return -1;
  }

  quantY = &(region->quantVals[tile->quantIdxY]);

  if (tile->quantIdxCb >= region->numQuant)
  {
    WLog_ERR(TAG, "quantIdxCb %" PRIu8 " > numQuant %" PRIu8, tile->quantIdxCb,
             region->numQuant);
    return -1;
  }

  quantCb = &(region->quantVals[tile->quantIdxCb]);

  if (tile->quantIdxCr >= region->numQuant)
  {
    WLog_ERR(TAG, "quantIdxCr %" PRIu8 " > numQuant %" PRIu8, tile->quantIdxCr,
             region->numQuant);
    return -1;
  }

  quantCr = &(region->quantVals[tile->quantIdxCr]);

  if (tile->quality == 0xFF)
  {
    quantProg = &(progressive->quantProgValFull);
  }
  else
  {
    if (tile->quality >= region->numProgQuant)
    {
      WLog_ERR(TAG, "quality %" PRIu8 " > numProgQuant %" PRIu8, tile->quality,
               region->numProgQuant);
      return -1;
    }

    quantProg = &(region->quantProgVals[tile->quality]);
  }

  quantProgY = &(quantProg->yQuantValues);
  quantProgCb = &(quantProg->cbQuantValues);
  quantProgCr = &(quantProg->crQuantValues);

  if (!progressive_rfx_quant_cmp_equal(quantY, &(tile->yQuant)))
    WLog_Print(progressive->log, WLOG_WARN, "non-progressive quantY has changed!");

  if (!progressive_rfx_quant_cmp_equal(quantCb, &(tile->cbQuant)))
    WLog_Print(progressive->log, WLOG_WARN, "non-progressive quantCb has changed!");

  if (!progressive_rfx_quant_cmp_equal(quantCr, &(tile->crQuant)))
    WLog_Print(progressive->log, WLOG_WARN, "non-progressive quantCr has changed!");

  if (!(context->flags & RFX_SUBBAND_DIFFING))
    WLog_WARN(TAG, "PROGRESSIVE_BLOCK_CONTEXT::flags & RFX_SUBBAND_DIFFING not set");

  progressive_rfx_quant_add(quantY, quantProgY, &yBitPos);
  progressive_rfx_quant_add(quantCb, quantProgCb, &cbBitPos);
  progressive_rfx_quant_add(quantCr, quantProgCr, &crBitPos);
  progressive_rfx_quant_sub(&(tile->yBitPos), &yBitPos, &yNumBits);
  progressive_rfx_quant_sub(&(tile->cbBitPos), &cbBitPos, &cbNumBits);
  progressive_rfx_quant_sub(&(tile->crBitPos), &crBitPos, &crNumBits);
  progressive_rfx_quant_add(quantY, quantProgY, &shiftY);
  progressive_rfx_quant_lsub(&shiftY, 1); /* -6 + 5 = -1 */
  progressive_rfx_quant_add(quantCb, quantProgCb, &shiftCb);
  progressive_rfx_quant_lsub(&shiftCb, 1); /* -6 + 5 = -1 */
  progressive_rfx_quant_add(quantCr, quantProgCr, &shiftCr);
  progressive_rfx_quant_lsub(&shiftCr, 1); /* -6 + 5 = -1 */

  tile->yBitPos = yBitPos;
  tile->cbBitPos = cbBitPos;
  tile->crBitPos = crBitPos;
  tile->yQuant = *quantY;
  tile->cbQuant = *quantCb;
  tile->crQuant = *quantCr;
  tile->yProgQuant = *quantProgY;
  tile->cbProgQuant = *quantProgCb;
  tile->crProgQuant = *quantProgCr;

  pSign[0] = (INT16*)((BYTE*)(&tile->sign[((8192 + 32) * 0) + 16])); /* Y/R buffer */
  pSign[1] = (INT16*)((BYTE*)(&tile->sign[((8192 + 32) * 1) + 16])); /* Cb/G buffer */
  pSign[2] = (INT16*)((BYTE*)(&tile->sign[((8192 + 32) * 2) + 16])); /* Cr/B buffer */

  pCurrent[0] = (INT16*)((BYTE*)(&tile->current[((8192 + 32) * 0) + 16])); /* Y/R buffer */
  pCurrent[1] = (INT16*)((BYTE*)(&tile->current[((8192 + 32) * 1) + 16])); /* Cb/G buffer */
  pCurrent[2] = (INT16*)((BYTE*)(&tile->current[((8192 + 32) * 2) + 16])); /* Cr/B buffer */

  pBuffer = (BYTE*)BufferPool_Take(progressive->bufferPool, -1);
  pSrcDst[0] = (INT16*)((BYTE*)(&pBuffer[((8192 + 32) * 0) + 16])); /* Y/R buffer */
  pSrcDst[1] = (INT16*)((BYTE*)(&pBuffer[((8192 + 32) * 1) + 16])); /* Cb/G buffer */
  pSrcDst[2] = (INT16*)((BYTE*)(&pBuffer[((8192 + 32) * 2) + 16])); /* Cr/B buffer */

  status = progressive_rfx_upgrade_component(progressive, &shiftY, quantProgY, &yNumBits,
                                             pSrcDst[0], pCurrent[0], pSign[0], tile->ySrlData,
                                             tile->ySrlLen, tile->yRawData, tile->yRawLen,
                                             coeffDiff, sub, extrapolate); /* Y */

  if (status < 0)
    goto fail;

  status = progressive_rfx_upgrade_component(progressive, &shiftCb, quantProgCb, &cbNumBits,
                                             pSrcDst[1], pCurrent[1], pSign[1], tile->cbSrlData,
                                             tile->cbSrlLen, tile->cbRawData, tile->cbRawLen,
                                             coeffDiff, sub, extrapolate); /* Cb */

  if (status < 0)
    goto fail;

  status = progressive_rfx_upgrade_component(progressive, &shiftCr, quantProgCr, &crNumBits,
                                             pSrcDst[2], pCurrent[2], pSign[2], tile->crSrlData,
                                             tile->crSrlLen, tile->crRawData, tile->crRawLen,
                                             coeffDiff, sub, extrapolate); /* Cr */

  if (status < 0)
    goto fail;

  status = prims->yCbCrToRGB_16s8u_P3AC4R((const INT16* const*)pSrcDst, 64 * 2, tile->data,
                                          tile->stride, progressive->format, &roi_64x64);
fail:
  BufferPool_Return(progressive->bufferPool, pBuffer);
  return status;
}

static INLINE BOOL progressive_tile_read_upgrade(PROGRESSIVE_CONTEXT* progressive, wStream* s,
                                                 UINT16 blockType, UINT32 blockLen,
                                                 PROGRESSIVE_SURFACE_CONTEXT* surface,
                                                 PROGRESSIVE_BLOCK_REGION* region,
                                                 const PROGRESSIVE_BLOCK_CONTEXT* context)
{
  RFX_PROGRESSIVE_TILE tile = { 0 };
  const size_t expect = 20;

  if (!Stream_CheckAndLogRequiredLength(TAG, s, expect))
    return FALSE;

  tile.blockType = blockType;
  tile.blockLen = blockLen;
  tile.flags = 0;

  Stream_Read_UINT8(s, tile.quantIdxY);
  Stream_Read_UINT8(s, tile.quantIdxCb);
  Stream_Read_UINT8(s, tile.quantIdxCr);
  Stream_Read_UINT16(s, tile.xIdx);
  Stream_Read_UINT16(s, tile.yIdx);
  Stream_Read_UINT8(s, tile.quality);
  Stream_Read_UINT16(s, tile.ySrlLen);
  Stream_Read_UINT16(s, tile.yRawLen);
  Stream_Read_UINT16(s, tile.cbSrlLen);
  Stream_Read_UINT16(s, tile.cbRawLen);
  Stream_Read_UINT16(s, tile.crSrlLen);
  Stream_Read_UINT16(s, tile.crRawLen);

  tile.ySrlData = Stream_Pointer(s);
  if (!Stream_SafeSeek(s, tile.ySrlLen))
  {
    WLog_Print(progressive->log, WLOG_ERROR, " Failed to seek %" PRIu32 " bytes", tile.ySrlLen);
    return FALSE;
  }

  tile.yRawData = Stream_Pointer(s);
  if (!Stream_SafeSeek(s, tile.yRawLen))
  {
    WLog_Print(progressive->log, WLOG_ERROR, " Failed to seek %" PRIu32 " bytes", tile.yRawLen);
    return FALSE;
  }

  tile.cbSrlData = Stream_Pointer(s);
  if (!Stream_SafeSeek(s, tile.cbSrlLen))
  {
    WLog_Print(progressive->log, WLOG_ERROR, " Failed to seek %" PRIu32 " bytes",
               tile.cbSrlLen);
    return FALSE;
  }

  tile.cbRawData = Stream_Pointer(s);
  if (!Stream_SafeSeek(s, tile.cbRawLen))
  {
    WLog_Print(progressive->log, WLOG_ERROR, " Failed to seek %" PRIu32 " bytes",
               tile.cbRawLen);
    return FALSE;
  }

  tile.crSrlData = Stream_Pointer(s);
  if (!Stream_SafeSeek(s, tile.crSrlLen))
  {
    WLog_Print(progressive->log, WLOG_ERROR, " Failed to seek %" PRIu32 " bytes",
               tile.crSrlLen);
    return FALSE;
  }

  tile.crRawData = Stream_Pointer(s);
  if (!Stream_SafeSeek(s, tile.crRawLen))
  {
    WLog_Print(progressive->log, WLOG_ERROR, " Failed to seek %" PRIu32 " bytes",
               tile.crRawLen);
    return FALSE;
  }

  return progressive_surface_tile_replace(surface, region, &tile, TRUE);
}

static INLINE BOOL progressive_tile_read(PROGRESSIVE_CONTEXT* progressive, BOOL simple, wStream* s,
                                         UINT16 blockType, UINT32 blockLen,
                                         PROGRESSIVE_SURFACE_CONTEXT* surface,
                                         PROGRESSIVE_BLOCK_REGION* region,
                                         const PROGRESSIVE_BLOCK_CONTEXT* context)
{
  RFX_PROGRESSIVE_TILE tile = { 0 };
  size_t expect = simple ? 16 : 17;

  if (!Stream_CheckAndLogRequiredLength(TAG, s, expect))
    return FALSE;

  tile.blockType = blockType;
  tile.blockLen = blockLen;

  Stream_Read_UINT8(s, tile.quantIdxY);
  Stream_Read_UINT8(s, tile.quantIdxCb);
  Stream_Read_UINT8(s, tile.quantIdxCr);
  Stream_Read_UINT16(s, tile.xIdx);
  Stream_Read_UINT16(s, tile.yIdx);
  Stream_Read_UINT8(s, tile.flags);

  if (!simple)
    Stream_Read_UINT8(s, tile.quality);
  else
    tile.quality = 0xFF;
  Stream_Read_UINT16(s, tile.yLen);
  Stream_Read_UINT16(s, tile.cbLen);
  Stream_Read_UINT16(s, tile.crLen);
  Stream_Read_UINT16(s, tile.tailLen);

  tile.yData = Stream_Pointer(s);
  if (!Stream_SafeSeek(s, tile.yLen))
  {
    WLog_Print(progressive->log, WLOG_ERROR, " Failed to seek %" PRIu32 " bytes", tile.yLen);
    return FALSE;
  }

  tile.cbData = Stream_Pointer(s);
  if (!Stream_SafeSeek(s, tile.cbLen))
  {
    WLog_Print(progressive->log, WLOG_ERROR, " Failed to seek %" PRIu32 " bytes", tile.cbLen);
    return FALSE;
  }

  tile.crData = Stream_Pointer(s);
  if (!Stream_SafeSeek(s, tile.crLen))
  {
    WLog_Print(progressive->log, WLOG_ERROR, " Failed to seek %" PRIu32 " bytes", tile.crLen);
    return FALSE;
  }

  tile.tailData = Stream_Pointer(s);
  if (!Stream_SafeSeek(s, tile.tailLen))
  {
    WLog_Print(progressive->log, WLOG_ERROR, " Failed to seek %" PRIu32 " bytes", tile.tailLen);
    return FALSE;
  }

  return progressive_surface_tile_replace(surface, region, &tile, FALSE);
}

typedef struct
{
  PROGRESSIVE_CONTEXT* progressive;
  PROGRESSIVE_BLOCK_REGION* region;
  const PROGRESSIVE_BLOCK_CONTEXT* context;
  RFX_PROGRESSIVE_TILE* tile;
} PROGRESSIVE_TILE_PROCESS_WORK_PARAM;

static void CALLBACK progressive_process_tiles_tile_work_callback(PTP_CALLBACK_INSTANCE instance,
                                                                  void* context, PTP_WORK work)
{
  PROGRESSIVE_TILE_PROCESS_WORK_PARAM* param = (PROGRESSIVE_TILE_PROCESS_WORK_PARAM*)context;

  WINPR_UNUSED(instance);
  WINPR_UNUSED(work);

  switch (param->tile->blockType)
  {
    case PROGRESSIVE_WBT_TILE_SIMPLE:
    case PROGRESSIVE_WBT_TILE_FIRST:
      progressive_decompress_tile_first(param->progressive, param->tile, param->region,
                                        param->context);
      break;

    case PROGRESSIVE_WBT_TILE_UPGRADE:
      progressive_decompress_tile_upgrade(param->progressive, param->tile, param->region,
                                          param->context);
      break;
    default:
      WLog_Print(param->progressive->log, WLOG_ERROR, "Invalid block type %04" PRIx16 " (%s)",
                 param->tile->blockType,
                 progressive_get_block_type_string(param->tile->blockType));
      break;
  }
}

static INLINE SSIZE_T progressive_process_tiles(PROGRESSIVE_CONTEXT* progressive, wStream* s,
                                                PROGRESSIVE_BLOCK_REGION* region,
                                                PROGRESSIVE_SURFACE_CONTEXT* surface,
                                                const PROGRESSIVE_BLOCK_CONTEXT* context)
{
  int status = 0;
  size_t end;
  const size_t start = Stream_GetPosition(s);
  UINT16 index;
  UINT16 blockType;
  UINT32 blockLen;
  UINT32 count = 0;
  PTP_WORK* work_objects = NULL;
  PROGRESSIVE_TILE_PROCESS_WORK_PARAM* params = NULL;
  UINT16 close_cnt = 0;

  WINPR_ASSERT(progressive);
  WINPR_ASSERT(region);

  if (!Stream_CheckAndLogRequiredLength(TAG, s, region->tileDataSize))
    return -1;

  while ((Stream_GetRemainingLength(s) >= 6) &&
         (region->tileDataSize > (Stream_GetPosition(s) - start)))
  {
    const size_t pos = Stream_GetPosition(s);

    Stream_Read_UINT16(s, blockType);
    Stream_Read_UINT32(s, blockLen);

#if defined(WITH_DEBUG_CODECS)
    WLog_Print(progressive->log, WLOG_DEBUG, "%s",
               progressive_get_block_type_string(blockType));
#endif

    if (blockLen < 6)
    {
      WLog_Print(progressive->log, WLOG_ERROR, "Expected >= %" PRIu32 " remaining %" PRIuz, 6,
                 blockLen);
      return -1003;
    }
    if (!Stream_CheckAndLogRequiredLength(TAG, s, blockLen - 6))
      return -1003;

    switch (blockType)
    {
      case PROGRESSIVE_WBT_TILE_SIMPLE:
        if (!progressive_tile_read(progressive, TRUE, s, blockType, blockLen, surface,
                                   region, context))
          return -1022;
        break;

      case PROGRESSIVE_WBT_TILE_FIRST:
        if (!progressive_tile_read(progressive, FALSE, s, blockType, blockLen, surface,
                                   region, context))
          return -1027;
        break;

      case PROGRESSIVE_WBT_TILE_UPGRADE:
        if (!progressive_tile_read_upgrade(progressive, s, blockType, blockLen, surface,
                                           region, context))
          return -1032;
        break;
      default:
        WLog_ERR(TAG, "Invalid block type %04" PRIx16 " (%s)", blockType,
                 progressive_get_block_type_string(blockType));
        return -1039;
    }

    size_t rem = Stream_GetPosition(s);
    if ((rem - pos) != blockLen)
    {
      WLog_Print(progressive->log, WLOG_ERROR,
                 "Actual block read %" PRIuz " but expected %" PRIu32, rem - pos, blockLen);
      return -1040;
    }
    count++;
  }

  end = Stream_GetPosition(s);
  if ((end - start) != region->tileDataSize)
  {
    WLog_Print(progressive->log, WLOG_ERROR,
               "Actual total blocks read %" PRIuz " but expected %" PRIu32, end - start,
               region->tileDataSize);
    return -1041;
  }

  if (count != region->numTiles)
  {
    WLog_Print(progressive->log, WLOG_WARN,
               "numTiles inconsistency: actual: %" PRIu32 ", expected: %" PRIu16 "\n", count,
               region->numTiles);
    return -1044;
  }

  if (progressive->rfx_context->priv->UseThreads)
  {
    work_objects = (PTP_WORK*)winpr_aligned_calloc(region->numTiles, sizeof(PTP_WORK), 32);
    if (!work_objects)
      return -1;
  }

  params = (PROGRESSIVE_TILE_PROCESS_WORK_PARAM*)winpr_aligned_calloc(
      region->numTiles, sizeof(PROGRESSIVE_TILE_PROCESS_WORK_PARAM), 32);
  if (!params)
  {
    winpr_aligned_free(work_objects);
    return -1;
  }

  for (index = 0; index < region->numTiles; index++)
  {
    RFX_PROGRESSIVE_TILE* tile = region->tiles[index];
    PROGRESSIVE_TILE_PROCESS_WORK_PARAM* param = &params[index];
    param->progressive = progressive;
    param->region = region;
    param->context = context;
    param->tile = tile;

    if (progressive->rfx_context->priv->UseThreads)
    {
      if (!(work_objects[index] = CreateThreadpoolWork(
                progressive_process_tiles_tile_work_callback, (void*)&params[index],
                &progressive->rfx_context->priv->ThreadPoolEnv)))
      {
        WLog_ERR(TAG, "CreateThreadpoolWork failed.");
        status = -1;
        break;
      }

      SubmitThreadpoolWork(work_objects[index]);
      close_cnt = index + 1;
    }
    else
    {
      progressive_process_tiles_tile_work_callback(0, &params[index], 0);
    }

    if (status < 0)
    {
      WLog_Print(progressive->log, WLOG_ERROR, "Failed to decompress %s at %" PRIu16,
                 progressive_get_block_type_string(tile->blockType), index);
      goto fail;
    }
  }

  if (status < 0)
    WLog_Print(progressive->log, WLOG_ERROR,
               "Failed to create ThreadpoolWork for tile %" PRIu16, index);

  if (progressive->rfx_context->priv->UseThreads)
  {
    for (index = 0; index < close_cnt; index++)
    {
      WaitForThreadpoolWorkCallbacks(work_objects[index], FALSE);
      CloseThreadpoolWork(work_objects[index]);
    }
  }

fail:
  winpr_aligned_free(work_objects);
  winpr_aligned_free(params);

  if (status < 0)
    return -1;

  return (SSIZE_T)(end - start);
}

static INLINE BOOL progressive_write_wb_sync(PROGRESSIVE_CONTEXT* progressive, wStream* s)
{
  const UINT32 blockLen = 12;
  WINPR_ASSERT(progressive);
  WINPR_ASSERT(s);

  if (!Stream_EnsureRemainingCapacity(s, blockLen))
    return FALSE;

  Stream_Write_UINT16(s, PROGRESSIVE_WBT_SYNC); /* blockType (2 bytes) */
  Stream_Write_UINT32(s, blockLen);             /* blockLen (4 bytes) */
  Stream_Write_UINT32(s, 0xCACCACCA);           /* magic (4 bytes) */
  Stream_Write_UINT16(s, 0x0100);               /* version (2 bytes) */
  return TRUE;
}

static INLINE BOOL progressive_write_wb_context(PROGRESSIVE_CONTEXT* progressive, wStream* s)
{
  const UINT32 blockLen = 10;
  WINPR_ASSERT(progressive);
  WINPR_ASSERT(s);

  if (!Stream_EnsureRemainingCapacity(s, blockLen))
    return FALSE;

  Stream_Write_UINT16(s, PROGRESSIVE_WBT_CONTEXT); /* blockType (2 bytes) */
  Stream_Write_UINT32(s, blockLen);                /* blockLen (4 bytes) */
  Stream_Write_UINT8(s, 0);                        /* ctxId (1 byte) */
  Stream_Write_UINT16(s, 64);                      /* tileSize (2 bytes) */
  Stream_Write_UINT8(s, 0);                        /* flags (1 byte) */
  return TRUE;
}

static INLINE BOOL progressive_write_region(PROGRESSIVE_CONTEXT* progressive, wStream* s,
                                            const RFX_MESSAGE* msg)
{
  /* RFX_PROGRESSIVE_REGION */
  UINT32 blockLen = 18;
  UINT32 tilesDataSize = 0;
  const size_t start = Stream_GetPosition(s);

  WINPR_ASSERT(progressive);
  WINPR_ASSERT(s);
  WINPR_ASSERT(msg);

  blockLen += msg->numRects * 8;
  blockLen += msg->numQuant * 5;
  tilesDataSize = msg->numTiles * 22UL;
  for (UINT16 i = 0; i < msg->numTiles; i++)
  {
    const RFX_TILE* tile = msg->tiles[i];
    WINPR_ASSERT(tile);
    tilesDataSize += tile->YLen + tile->CbLen + tile->CrLen;
  }
  blockLen += tilesDataSize;

  if (!Stream_EnsureRemainingCapacity(s, blockLen))
    return FALSE;

  Stream_Write_UINT16(s, PROGRESSIVE_WBT_REGION); /* blockType (2 bytes) */
  Stream_Write_UINT32(s, blockLen);               /* blockLen (4 bytes) */
  Stream_Write_UINT8(s, 64);                      /* tileSize (1 byte) */
  Stream_Write_UINT16(s, msg->numRects);          /* numRects (2 bytes) */
  WINPR_ASSERT(msg->numQuant <= UINT8_MAX);
  Stream_Write_UINT8(s, (UINT8)msg->numQuant); /* numQuant (1 byte) */
  Stream_Write_UINT8(s, 0);                    /* numProgQuant (1 byte) */
  Stream_Write_UINT8(s, 0);                    /* flags (1 byte) */
  Stream_Write_UINT16(s, msg->numTiles);       /* numTiles (2 bytes) */
  Stream_Write_UINT32(s, tilesDataSize);       /* tilesDataSize (4 bytes) */

  for (UINT16 i = 0; i < msg->numRects; i++)
  {
    /* TS_RFX_RECT */
    const RFX_RECT* r = &msg->rects[i];
    Stream_Write_UINT16(s, r->x);      /* x (2 bytes) */
    Stream_Write_UINT16(s, r->y);      /* y (2 bytes) */
    Stream_Write_UINT16(s, r->width);  /* width (2 bytes) */
    Stream_Write_UINT16(s, r->height); /* height (2 bytes) */
  }

  /**
   * Note: The RFX_COMPONENT_CODEC_QUANT structure differs from the
   * TS_RFX_CODEC_QUANT ([MS-RDPRFX] section 2.2.2.1.5) structure with respect
   * to the order of the bands.
   *             0    1    2   3     4    5    6    7    8    9
   * RDPRFX:   LL3, LH3, HL3, HH3, LH2, HL2, HH2, LH1, HL1, HH1
   * RDPEGFX:  LL3, HL3, LH3, HH3, HL2, LH2, HH2, HL1, LH1, HH1
   */
  for (UINT16 i = 0; i < msg->numQuant; i++)
  {
    const UINT32* qv = &msg->quantVals[i * 10];
    /* RFX_COMPONENT_CODEC_QUANT */
    Stream_Write_UINT8(s, (UINT8)(qv[0] + (qv[2] << 4))); /* LL3 (4-bit), HL3 (4-bit) */
    Stream_Write_UINT8(s, (UINT8)(qv[1] + (qv[3] << 4))); /* LH3 (4-bit), HH3 (4-bit) */
    Stream_Write_UINT8(s, (UINT8)(qv[5] + (qv[4] << 4))); /* HL2 (4-bit), LH2 (4-bit) */
    Stream_Write_UINT8(s, (UINT8)(qv[6] + (qv[8] << 4))); /* HH2 (4-bit), HL1 (4-bit) */
    Stream_Write_UINT8(s, (UINT8)(qv[7] + (qv[9] << 4))); /* LH1 (4-bit), HH1 (4-bit) */
  }

  for (UINT16 i = 0; i < msg->numTiles; i++)
  {
    const RFX_TILE* tile = msg->tiles[i];
    if (!progressive_write_tile_simple(progressive, s, tile))
      return FALSE;
  }

  const size_t end = Stream_GetPosition(s);
  const size_t used = end - start;
  return (used == blockLen);
}

static INLINE BOOL progressive_write_frame_begin(PROGRESSIVE_CONTEXT* progressive, wStream* s,
                                                 const RFX_MESSAGE* msg)
{
  const UINT32 blockLen = 12;
  WINPR_ASSERT(progressive);
  WINPR_ASSERT(s);
  WINPR_ASSERT(msg);

  if (!Stream_EnsureRemainingCapacity(s, blockLen))
    return FALSE;

  Stream_Write_UINT16(s, PROGRESSIVE_WBT_FRAME_BEGIN); /* blockType (2 bytes) */
  Stream_Write_UINT32(s, blockLen);                    /* blockLen (4 bytes) */
  Stream_Write_UINT32(s, msg->frameIdx);               /* frameIndex (4 bytes) */
  Stream_Write_UINT16(s, 1);                           /* regionCount (2 bytes) */

  return TRUE;
}

static INLINE BOOL progressive_write_frame_end(PROGRESSIVE_CONTEXT* progressive, wStream* s)
{
  const UINT32 blockLen = 6;
  WINPR_ASSERT(progressive);
  WINPR_ASSERT(s);

  if (!Stream_EnsureRemainingCapacity(s, blockLen))
    return FALSE;

  Stream_Write_UINT16(s, PROGRESSIVE_WBT_FRAME_END); /* blockType (2 bytes) */
  Stream_Write_UINT32(s, blockLen);                  /* blockLen (4 bytes) */

  return TRUE;
}

INLINE BOOL progressive_write_tile_simple(PROGRESSIVE_CONTEXT* progressive, wStream* s,
                                          const RFX_TILE* tile)
{
  UINT32 blockLen;
  WINPR_ASSERT(progressive);
  WINPR_ASSERT(s);
  WINPR_ASSERT(tile);

  blockLen = 22 + tile->YLen + tile->CbLen + tile->CrLen;
  if (!Stream_EnsureRemainingCapacity(s, blockLen))
    return FALSE;

  Stream_Write_UINT16(s, PROGRESSIVE_WBT_TILE_SIMPLE); /* blockType (2 bytes) */
  Stream_Write_UINT32(s, blockLen);                    /* blockLen (4 bytes) */
  Stream_Write_UINT8(s, tile->quantIdxY);              /* quantIdxY (1 byte) */
  Stream_Write_UINT8(s, tile->quantIdxCb);             /* quantIdxCb (1 byte) */
  Stream_Write_UINT8(s, tile->quantIdxCr);             /* quantIdxCr (1 byte) */
  Stream_Write_UINT16(s, tile->xIdx);                  /* xIdx (2 bytes) */
  Stream_Write_UINT16(s, tile->yIdx);                  /* yIdx (2 bytes) */
  Stream_Write_UINT8(s, 0);                            /* flags (1 byte) */
  Stream_Write_UINT16(s, tile->YLen);                  /* YLen (2 bytes) */
  Stream_Write_UINT16(s, tile->CbLen);                 /* CbLen (2 bytes) */
  Stream_Write_UINT16(s, tile->CrLen);                 /* CrLen (2 bytes) */
  Stream_Write_UINT16(s, 0);                           /* tailLen (2 bytes) */
  Stream_Write(s, tile->YData, tile->YLen);            /* YData */
  Stream_Write(s, tile->CbData, tile->CbLen);          /* CbData */
  Stream_Write(s, tile->CrData, tile->CrLen);          /* CrData */

  return TRUE;
}

static INLINE SSIZE_T progressive_wb_sync(PROGRESSIVE_CONTEXT* progressive, wStream* s,
                                          UINT16 blockType, UINT32 blockLen)
{
  const UINT32 magic = 0xCACCACCA;
  const UINT16 version = 0x0100;
  PROGRESSIVE_BLOCK_SYNC sync;

  sync.blockType = blockType;
  sync.blockLen = blockLen;

  if (sync.blockLen != 12)
  {
    WLog_Print(progressive->log, WLOG_ERROR,
               "PROGRESSIVE_BLOCK_SYNC::blockLen = 0x%08" PRIx32 " != 0x%08" PRIx32,
               sync.blockLen, 12);
    return -1005;
  }

  if (!Stream_CheckAndLogRequiredLength(TAG, s, 6))
    return -1004;

#if defined(WITH_DEBUG_CODECS)
  WLog_Print(progressive->log, WLOG_DEBUG, "ProgressiveSync");
#endif

  Stream_Read_UINT32(s, sync.magic);
  Stream_Read_UINT16(s, sync.version);

  if (sync.magic != magic)
  {
    WLog_Print(progressive->log, WLOG_ERROR,
               "PROGRESSIVE_BLOCK_SYNC::magic = 0x%08" PRIx32 " != 0x%08" PRIx32, sync.magic,
               magic);
    return -1005;
  }

  if (sync.version != 0x0100)
  {
    WLog_Print(progressive->log, WLOG_ERROR,
               "PROGRESSIVE_BLOCK_SYNC::version = 0x%04" PRIx16 " != 0x%04" PRIu16,
               sync.version, version);
    return -1006;
  }

  if ((progressive->state & FLAG_WBT_SYNC) != 0)
    WLog_WARN(TAG, "Duplicate PROGRESSIVE_BLOCK_SYNC, ignoring");

  progressive->state |= FLAG_WBT_SYNC;
  return 0;
}

static INLINE SSIZE_T progressive_wb_frame_begin(PROGRESSIVE_CONTEXT* progressive, wStream* s,
                                                 UINT16 blockType, UINT32 blockLen)
{
  PROGRESSIVE_BLOCK_FRAME_BEGIN frameBegin;

  frameBegin.blockType = blockType;
  frameBegin.blockLen = blockLen;

  if (frameBegin.blockLen != 12)
  {
    WLog_Print(progressive->log, WLOG_ERROR,
               " RFX_PROGRESSIVE_FRAME_BEGIN::blockLen = 0x%08" PRIx32 " != 0x%08" PRIx32,
               frameBegin.blockLen, 12);
    return -1005;
  }

  if (!Stream_CheckAndLogRequiredLength(TAG, s, 6))
    return -1007;

  Stream_Read_UINT32(s, frameBegin.frameIndex);
  Stream_Read_UINT16(s, frameBegin.regionCount);

#if defined(WITH_DEBUG_CODECS)
  WLog_Print(progressive->log, WLOG_DEBUG,
             "ProgressiveFrameBegin: frameIndex: %" PRIu32 " regionCount: %" PRIu16 "",
             frameBegin.frameIndex, frameBegin.regionCount);
#endif

  /**
   * If the number of elements specified by the regionCount field is
   * larger than the actual number of elements in the regions field,
   * the decoder SHOULD ignore this inconsistency.
   */

  if ((progressive->state & FLAG_WBT_FRAME_BEGIN) != 0)
  {
    WLog_ERR(TAG, "Duplicate RFX_PROGRESSIVE_FRAME_BEGIN in stream, this is not allowed!");
    return -1008;
  }

  if ((progressive->state & FLAG_WBT_FRAME_END) != 0)
  {
    WLog_ERR(TAG, "RFX_PROGRESSIVE_FRAME_BEGIN after RFX_PROGRESSIVE_FRAME_END in stream, this "
                  "is not allowed!");
    return -1008;
  }

  progressive->state |= FLAG_WBT_FRAME_BEGIN;
  return 0;
}

static INLINE SSIZE_T progressive_wb_frame_end(PROGRESSIVE_CONTEXT* progressive, wStream* s,
                                               UINT16 blockType, UINT32 blockLen)
{
  PROGRESSIVE_BLOCK_FRAME_END frameEnd;

  frameEnd.blockType = blockType;
  frameEnd.blockLen = blockLen;

  if (frameEnd.blockLen != 6)
  {
    WLog_Print(progressive->log, WLOG_ERROR,
               " RFX_PROGRESSIVE_FRAME_END::blockLen = 0x%08" PRIx32 " != 0x%08" PRIx32,
               frameEnd.blockLen, 6);
    return -1005;
  }

  if (Stream_GetRemainingLength(s) != 0)
  {
    WLog_Print(progressive->log, WLOG_ERROR,
               "ProgressiveFrameEnd short %" PRIuz ", expected %" PRIuz,
               Stream_GetRemainingLength(s), 0);
    return -1008;
  }

#if defined(WITH_DEBUG_CODECS)
  WLog_Print(progressive->log, WLOG_DEBUG, "ProgressiveFrameEnd");
#endif

  if ((progressive->state & FLAG_WBT_FRAME_BEGIN) == 0)
    WLog_WARN(TAG, "RFX_PROGRESSIVE_FRAME_END before RFX_PROGRESSIVE_FRAME_BEGIN, ignoring");
  if ((progressive->state & FLAG_WBT_FRAME_END) != 0)
    WLog_WARN(TAG, "Duplicate RFX_PROGRESSIVE_FRAME_END, ignoring");

  progressive->state |= FLAG_WBT_FRAME_END;
  return 0;
}

static INLINE SSIZE_T progressive_wb_context(PROGRESSIVE_CONTEXT* progressive, wStream* s,
                                             UINT16 blockType, UINT32 blockLen)
{
  PROGRESSIVE_BLOCK_CONTEXT* context = &progressive->context;
  context->blockType = blockType;
  context->blockLen = blockLen;

  if (context->blockLen != 10)
  {
    WLog_Print(progressive->log, WLOG_ERROR,
               "RFX_PROGRESSIVE_CONTEXT::blockLen = 0x%08" PRIx32 " != 0x%08" PRIx32,
               context->blockLen, 10);
    return -1005;
  }

  if (!Stream_CheckAndLogRequiredLength(TAG, s, 4))
    return -1009;

  Stream_Read_UINT8(s, context->ctxId);
  Stream_Read_UINT16(s, context->tileSize);
  Stream_Read_UINT8(s, context->flags);

  if (context->ctxId != 0x00)
    WLog_WARN(TAG, "RFX_PROGRESSIVE_CONTEXT::ctxId != 0x00: %" PRIu8, context->ctxId);

  if (context->tileSize != 64)
  {
    WLog_ERR(TAG, "RFX_PROGRESSIVE_CONTEXT::tileSize != 0x40: %" PRIu16, context->tileSize);
    return -1010;
  }

  if ((progressive->state & FLAG_WBT_FRAME_BEGIN) != 0)
    WLog_WARN(TAG, "RFX_PROGRESSIVE_CONTEXT received after RFX_PROGRESSIVE_FRAME_BEGIN");
  if ((progressive->state & FLAG_WBT_FRAME_END) != 0)
    WLog_WARN(TAG, "RFX_PROGRESSIVE_CONTEXT received after RFX_PROGRESSIVE_FRAME_END");
  if ((progressive->state & FLAG_WBT_CONTEXT) != 0)
    WLog_WARN(TAG, "Duplicate RFX_PROGRESSIVE_CONTEXT received, ignoring.");

#if defined(WITH_DEBUG_CODECS)
  WLog_Print(progressive->log, WLOG_DEBUG, "ProgressiveContext: flags: 0x%02" PRIX8 "",
             context->flags);
#endif

  progressive->state |= FLAG_WBT_CONTEXT;
  return 0;
}

static INLINE SSIZE_T progressive_wb_read_region_header(PROGRESSIVE_CONTEXT* progressive,
                                                        wStream* s, UINT16 blockType,
                                                        UINT32 blockLen,
                                                        PROGRESSIVE_BLOCK_REGION* region)
{
  SSIZE_T len;

  memset(region, 0, sizeof(PROGRESSIVE_BLOCK_REGION));
  if (!Stream_CheckAndLogRequiredLength(TAG, s, 12))
    return -1011;

  region->blockType = blockType;
  region->blockLen = blockLen;
  Stream_Read_UINT8(s, region->tileSize);
  Stream_Read_UINT16(s, region->numRects);
  Stream_Read_UINT8(s, region->numQuant);
  Stream_Read_UINT8(s, region->numProgQuant);
  Stream_Read_UINT8(s, region->flags);
  Stream_Read_UINT16(s, region->numTiles);
  Stream_Read_UINT32(s, region->tileDataSize);

  if (region->tileSize != 64)
  {
    WLog_Print(progressive->log, WLOG_ERROR,
               "ProgressiveRegion tile size %" PRIu8 ", expected %" PRIuz, region->tileSize,
               64);
    return -1012;
  }

  if (region->numRects < 1)
  {
    WLog_Print(progressive->log, WLOG_ERROR, "ProgressiveRegion missing rect count %" PRIu16,
               region->numRects);
    return -1013;
  }

  if (region->numQuant > 7)
  {
    WLog_Print(progressive->log, WLOG_ERROR,
               "ProgressiveRegion quant count too high %" PRIu8 ", expected < %" PRIuz,
               region->numQuant, 7);
    return -1014;
  }

  len = Stream_GetRemainingLength(s);
  if (!Stream_CheckAndLogRequiredLengthOfSize(TAG, s, region->numRects, 8ull))
  {
    WLog_Print(progressive->log, WLOG_ERROR, "ProgressiveRegion data short for region->rects");
    return -1015;
  }
  len -= region->numRects * 8ULL;

  if (len / 5 < region->numQuant)
  {
    WLog_Print(progressive->log, WLOG_ERROR, "ProgressiveRegion data short for region->cQuant");
    return -1018;
  }
  len -= region->numQuant * 5ULL;

  if (len / 16 < region->numProgQuant)
  {
    WLog_Print(progressive->log, WLOG_ERROR,
               "ProgressiveRegion data short for region->cProgQuant");
    return -1021;
  }
  len -= region->numProgQuant * 16ULL;

  if (len < region->tileDataSize * 1ll)
  {
    WLog_Print(progressive->log, WLOG_ERROR, "ProgressiveRegion data short for region->tiles");
    return -1024;
  }
  len -= region->tileDataSize;
  if (len > 0)
    WLog_Print(progressive->log, WLOG_WARN,
               "Unused bytes detected, %" PRIuz " bytes not processed", len);
  return len;
}

static INLINE SSIZE_T progressive_wb_skip_region(PROGRESSIVE_CONTEXT* progressive, wStream* s,
                                                 UINT16 blockType, UINT32 blockLen)
{
  SSIZE_T rc;
  size_t total;
  PROGRESSIVE_BLOCK_REGION* region = &progressive->region;

  rc = progressive_wb_read_region_header(progressive, s, blockType, blockLen, region);
  if (rc < 0)
    return rc;

  total = (region->numRects * 8);
  total += (region->numQuant * 5);
  total += (region->numProgQuant * 16);
  total += region->tileDataSize;
  if (!Stream_SafeSeek(s, total))
    return -1111;

  return rc;
}

static INLINE SSIZE_T progressive_wb_region(PROGRESSIVE_CONTEXT* progressive, wStream* s,
                                            UINT16 blockType, UINT32 blockLen,
                                            PROGRESSIVE_SURFACE_CONTEXT* surface,
                                            PROGRESSIVE_BLOCK_REGION* region)
{
  SSIZE_T rc = -1;
  UINT16 boxLeft;
  UINT16 boxTop;
  UINT16 boxRight;
  UINT16 boxBottom;
  UINT16 idxLeft;
  UINT16 idxTop;
  UINT16 idxRight;
  UINT16 idxBottom;
  const PROGRESSIVE_BLOCK_CONTEXT* context = &progressive->context;

  if ((progressive->state & FLAG_WBT_FRAME_BEGIN) == 0)
  {
    WLog_WARN(TAG, "RFX_PROGRESSIVE_REGION before RFX_PROGRESSIVE_FRAME_BEGIN, ignoring");
    return progressive_wb_skip_region(progressive, s, blockType, blockLen);
  }
  if ((progressive->state & FLAG_WBT_FRAME_END) != 0)
  {
    WLog_WARN(TAG, "RFX_PROGRESSIVE_REGION after RFX_PROGRESSIVE_FRAME_END, ignoring");
    return progressive_wb_skip_region(progressive, s, blockType, blockLen);
  }

  progressive->state |= FLAG_WBT_REGION;

  rc = progressive_wb_read_region_header(progressive, s, blockType, blockLen, region);
  if (rc < 0)
    return rc;

  for (UINT16 index = 0; index < region->numRects; index++)
  {
    RFX_RECT* rect = &(region->rects[index]);
    Stream_Read_UINT16(s, rect->x);
    Stream_Read_UINT16(s, rect->y);
    Stream_Read_UINT16(s, rect->width);
    Stream_Read_UINT16(s, rect->height);
  }

  for (BYTE index = 0; index < region->numQuant; index++)
  {
    RFX_COMPONENT_CODEC_QUANT* quantVal = &(region->quantVals[index]);
    progressive_component_codec_quant_read(s, quantVal);

    if (!progressive_rfx_quant_lcmp_greater_equal(quantVal, 6))
    {
      WLog_Print(progressive->log, WLOG_ERROR,
                 "ProgressiveRegion region->cQuant[%" PRIu32 "] < 6", index);
      return -1;
    }

    if (!progressive_rfx_quant_lcmp_less_equal(quantVal, 15))
    {
      WLog_Print(progressive->log, WLOG_ERROR,
                 "ProgressiveRegion region->cQuant[%" PRIu32 "] > 15", index);
      return -1;
    }
  }

  for (BYTE index = 0; index < region->numProgQuant; index++)
  {
    RFX_PROGRESSIVE_CODEC_QUANT* quantProgVal = &(region->quantProgVals[index]);

    Stream_Read_UINT8(s, quantProgVal->quality);

    progressive_component_codec_quant_read(s, &(quantProgVal->yQuantValues));
    progressive_component_codec_quant_read(s, &(quantProgVal->cbQuantValues));
    progressive_component_codec_quant_read(s, &(quantProgVal->crQuantValues));
  }

#if defined(WITH_DEBUG_CODECS)
  WLog_Print(progressive->log, WLOG_DEBUG,
             "ProgressiveRegion: numRects: %" PRIu16 " numTiles: %" PRIu16
             " tileDataSize: %" PRIu32 " flags: 0x%02" PRIX8 " numQuant: %" PRIu8
             " numProgQuant: %" PRIu8 "",
             region->numRects, region->numTiles, region->tileDataSize, region->flags,
             region->numQuant, region->numProgQuant);
#endif

  boxLeft = surface->gridWidth;
  boxTop = surface->gridHeight;
  boxRight = 0;
  boxBottom = 0;

  for (UINT16 index = 0; index < region->numRects; index++)
  {
    RFX_RECT* rect = &(region->rects[index]);
    idxLeft = rect->x / 64;
    idxTop = rect->y / 64;
    idxRight = (rect->x + rect->width + 63) / 64;
    idxBottom = (rect->y + rect->height + 63) / 64;

    if (idxLeft < boxLeft)
      boxLeft = idxLeft;

    if (idxTop < boxTop)
      boxTop = idxTop;

    if (idxRight > boxRight)
      boxRight = idxRight;

    if (idxBottom > boxBottom)
      boxBottom = idxBottom;

#if defined(WITH_DEBUG_CODECS)
    WLog_Print(progressive->log, WLOG_DEBUG,
               "rect[%" PRIu16 "]: x: %" PRIu16 " y: %" PRIu16 " w: %" PRIu16 " h: %" PRIu16 "",
               index, rect->x, rect->y, rect->width, rect->height);
#endif
  }

  const SSIZE_T res = progressive_process_tiles(progressive, s, region, surface, context);
  if (res < 0)
    return -1;
  return (size_t)rc;
}

static SSIZE_T progressive_parse_block(PROGRESSIVE_CONTEXT* progressive, wStream* s,
                                       PROGRESSIVE_SURFACE_CONTEXT* surface,
                                       PROGRESSIVE_BLOCK_REGION* region)
{
  UINT16 blockType;
  UINT32 blockLen;
  SSIZE_T rc = -1;
  wStream sub = { 0 };

  WINPR_ASSERT(progressive);

  if (!Stream_CheckAndLogRequiredLength(TAG, s, 6))
    return -1;

  Stream_Read_UINT16(s, blockType);
  Stream_Read_UINT32(s, blockLen);

  if (blockLen < 6)
  {
    WLog_WARN(TAG, "Invalid blockLen %" PRIu32 ", expected >= 6", blockLen);
    return -1;
  }
  if (!Stream_CheckAndLogRequiredLength(TAG, s, blockLen - 6))
    return -1;
  Stream_StaticConstInit(&sub, Stream_Pointer(s), blockLen - 6);
  Stream_Seek(s, blockLen - 6);

  switch (blockType)
  {
    case PROGRESSIVE_WBT_SYNC:
      rc = progressive_wb_sync(progressive, &sub, blockType, blockLen);
      break;

    case PROGRESSIVE_WBT_FRAME_BEGIN:
      rc = progressive_wb_frame_begin(progressive, &sub, blockType, blockLen);
      break;

    case PROGRESSIVE_WBT_FRAME_END:
      rc = progressive_wb_frame_end(progressive, &sub, blockType, blockLen);
      break;

    case PROGRESSIVE_WBT_CONTEXT:
      rc = progressive_wb_context(progressive, &sub, blockType, blockLen);
      break;

    case PROGRESSIVE_WBT_REGION:
      rc = progressive_wb_region(progressive, &sub, blockType, blockLen, surface, region);
      break;

    default:
      WLog_Print(progressive->log, WLOG_ERROR, "Invalid block type %04" PRIx16, blockType);
      return -1;
  }

  if (rc < 0)
    return -1;

  if (Stream_GetRemainingLength(&sub) > 0)
  {
    WLog_Print(progressive->log, WLOG_ERROR,
               "block len %" PRIu32 " does not match read data %" PRIuz, blockLen,
               blockLen - Stream_GetRemainingLength(&sub));
    return -1;
  }

  return rc;
}

INT32 progressive_decompress(PROGRESSIVE_CONTEXT* progressive, const BYTE* pSrcData, UINT32 SrcSize,
                             BYTE* pDstData, UINT32 DstFormat, UINT32 nDstStep, UINT32 nXDst,
                             UINT32 nYDst, REGION16* invalidRegion, UINT16 surfaceId,
                             UINT32 frameId)
{
  INT32 rc = 1;

  WINPR_ASSERT(progressive);
  PROGRESSIVE_SURFACE_CONTEXT* surface = progressive_get_surface_data(progressive, surfaceId);

  if (!surface)
  {
    WLog_Print(progressive->log, WLOG_ERROR, "ProgressiveRegion no surface for %" PRIu16,
               surfaceId);
    return -1001;
  }

  PROGRESSIVE_BLOCK_REGION* region = &progressive->region;
  WINPR_ASSERT(region);

  if (surface->frameId != frameId)
  {
    surface->frameId = frameId;
    surface->numUpdatedTiles = 0;
  }

  wStream ss = { 0 };
  wStream* s = Stream_StaticConstInit(&ss, pSrcData, SrcSize);
  WINPR_ASSERT(s);

  switch (DstFormat)
  {
    case PIXEL_FORMAT_RGBA32:
    case PIXEL_FORMAT_RGBX32:
    case PIXEL_FORMAT_BGRA32:
    case PIXEL_FORMAT_BGRX32:
      progressive->format = DstFormat;
      break;
    default:
      progressive->format = PIXEL_FORMAT_XRGB32;
      break;
  }

  const size_t start = Stream_GetPosition(s);
  progressive->state = 0; /* Set state to not initialized */
  while (Stream_GetRemainingLength(s) > 0)
  {
    if (progressive_parse_block(progressive, s, surface, region) < 0)
      goto fail;
  }

  const size_t end = Stream_GetPosition(s);
  if ((end - start) != SrcSize)
  {
    WLog_Print(progressive->log, WLOG_ERROR,
               "total block len %" PRIuz " does not match read data %" PRIu32, end - start,
               SrcSize);
    rc = -1041;
    goto fail;
  }

  REGION16 clippingRects = { 0 };
  region16_init(&clippingRects);

  for (UINT32 i = 0; i < region->numRects; i++)
  {
    RECTANGLE_16 clippingRect = { 0 };
    const RFX_RECT* rect = &(region->rects[i]);

    clippingRect.left = (UINT16)nXDst + rect->x;
    clippingRect.top = (UINT16)nYDst + rect->y;
    clippingRect.right = clippingRect.left + rect->width;
    clippingRect.bottom = clippingRect.top + rect->height;
    region16_union_rect(&clippingRects, &clippingRects, &clippingRect);
  }

  for (UINT32 i = 0; i < surface->numUpdatedTiles; i++)
  {
    UINT32 nbUpdateRects = 0;
    const RECTANGLE_16* updateRects = NULL;
    RECTANGLE_16 updateRect = { 0 };

    WINPR_ASSERT(surface->updatedTileIndices);
    const UINT32 index = surface->updatedTileIndices[i];

    WINPR_ASSERT(index < surface->tilesSize);
    RFX_PROGRESSIVE_TILE* tile = surface->tiles[index];
    WINPR_ASSERT(tile);

    updateRect.left = nXDst + tile->x;
    updateRect.top = nYDst + tile->y;
    updateRect.right = updateRect.left + 64;
    updateRect.bottom = updateRect.top + 64;

    REGION16 updateRegion = { 0 };
    region16_init(&updateRegion);
    region16_intersect_rect(&updateRegion, &clippingRects, &updateRect);
    updateRects = region16_rects(&updateRegion, &nbUpdateRects);

    for (UINT32 j = 0; j < nbUpdateRects; j++)
    {
      const RECTANGLE_16* rect = &updateRects[j];
      if (rect->left < updateRect.left)
        goto fail;
      const UINT32 nXSrc = rect->left - updateRect.left;
      const UINT32 nYSrc = rect->top - updateRect.top;
      const UINT32 width = rect->right - rect->left;
      const UINT32 height = rect->bottom - rect->top;

      if (rect->left + width > surface->width)
        goto fail;
      if (rect->top + height > surface->height)
        goto fail;
      if (!freerdp_image_copy(pDstData, DstFormat, nDstStep, rect->left, rect->top, width,
                              height, tile->data, progressive->format, tile->stride, nXSrc,
                              nYSrc, NULL, FREERDP_KEEP_DST_ALPHA))
      {
        rc = -42;
        break;
      }

      if (invalidRegion)
        region16_union_rect(invalidRegion, invalidRegion, rect);
    }

    region16_uninit(&updateRegion);
    tile->dirty = FALSE;
  }

  region16_uninit(&clippingRects);
  surface->numUpdatedTiles = 0;
fail:
  return rc;
}

BOOL progressive_rfx_write_message_progressive_simple(PROGRESSIVE_CONTEXT* progressive, wStream* s,
                                                      const RFX_MESSAGE* msg)
{
  RFX_CONTEXT* context;

  WINPR_ASSERT(progressive);
  WINPR_ASSERT(s);
  WINPR_ASSERT(msg);
  context = progressive->rfx_context;
  WINPR_ASSERT(context);
  if (context->mode != RLGR1)
  {
    WLog_ERR(TAG, "error, RLGR1 mode is required!");
    return FALSE;
  }

  if (!progressive_write_wb_sync(progressive, s))
    return FALSE;

  if (!progressive_write_wb_context(progressive, s))
    return FALSE;

  if (!progressive_write_frame_begin(progressive, s, msg))
    return FALSE;

  if (!progressive_write_region(progressive, s, msg))
    return FALSE;

  if (!progressive_write_frame_end(progressive, s))
    return FALSE;

  return TRUE;
}

int progressive_compress(PROGRESSIVE_CONTEXT* progressive, const BYTE* pSrcData, UINT32 SrcSize,
                         UINT32 SrcFormat, UINT32 Width, UINT32 Height, UINT32 ScanLine,
                         const REGION16* invalidRegion, BYTE** ppDstData, UINT32* pDstSize)
{
  BOOL rc = FALSE;
  int res = -6;
  wStream* s = NULL;
  UINT32 i = 0, numRects = 0;
  UINT32 x = 0, y = 0;
  RFX_RECT* rects = NULL;
  RFX_MESSAGE* message = NULL;

  if (!progressive || !pSrcData || !ppDstData || !pDstSize)
  {
    return -1;
  }

  if (ScanLine == 0)
  {
    switch (SrcFormat)
    {
      case PIXEL_FORMAT_ABGR32:
      case PIXEL_FORMAT_ARGB32:
      case PIXEL_FORMAT_XBGR32:
      case PIXEL_FORMAT_XRGB32:
      case PIXEL_FORMAT_BGRA32:
      case PIXEL_FORMAT_BGRX32:
      case PIXEL_FORMAT_RGBA32:
      case PIXEL_FORMAT_RGBX32:
        ScanLine = Width * 4;
        break;
      default:
        return -2;
    }
  }

  if (SrcSize < Height * ScanLine)
    return -4;

  if (!invalidRegion)
  {
    numRects = (Width + 63) / 64;
    numRects *= (Height + 63) / 64;
  }
  else
    numRects = region16_n_rects(invalidRegion);

  if (numRects == 0)
    return 0;

  if (!Stream_EnsureCapacity(progressive->rects, numRects * sizeof(RFX_RECT)))
    return -5;
  rects = (RFX_RECT*)Stream_Buffer(progressive->rects);
  if (invalidRegion)
  {
    const RECTANGLE_16* region_rects = region16_rects(invalidRegion, NULL);
    for (x = 0; x < numRects; x++)
    {
      const RECTANGLE_16* r = &region_rects[x];
      RFX_RECT* rect = &rects[x];

      rect->x = r->left;
      rect->y = r->top;
      rect->width = r->right - r->left;
      rect->height = r->bottom - r->top;
    }
  }
  else
  {
    x = 0;
    y = 0;
    for (i = 0; i < numRects; i++)
    {
      RFX_RECT* r = &rects[i];
      r->x = x;
      r->y = y;
      r->width = MIN(64, Width - x);
      r->height = MIN(64, Height - y);

      if (x + 64 >= Width)
      {
        y += 64;
        x = 0;
      }
      else
        x += 64;

      WINPR_ASSERT(r->x % 64 == 0);
      WINPR_ASSERT(r->y % 64 == 0);
      WINPR_ASSERT(r->width <= 64);
      WINPR_ASSERT(r->height <= 64);
    }
  }
  s = progressive->buffer;
  Stream_SetPosition(s, 0);

  progressive->rfx_context->mode = RLGR1;
  progressive->rfx_context->width = Width;
  progressive->rfx_context->height = Height;
  rfx_context_set_pixel_format(progressive->rfx_context, SrcFormat);
  message = rfx_encode_message(progressive->rfx_context, rects, numRects, pSrcData, Width, Height,
                               ScanLine);
  if (!message)
  {
    WLog_ERR(TAG, "failed to encode rfx message");
    goto fail;
  }

  rc = progressive_rfx_write_message_progressive_simple(progressive, s, message);
  rfx_message_free(progressive->rfx_context, message);
  if (!rc)
    goto fail;

  const size_t pos = Stream_GetPosition(s);
  WINPR_ASSERT(pos <= UINT32_MAX);
  *pDstSize = (UINT32)pos;
  *ppDstData = Stream_Buffer(s);
  res = 1;
fail:
  return res;
}

BOOL progressive_context_reset(PROGRESSIVE_CONTEXT* progressive)
{
  if (!progressive)
    return FALSE;

  return TRUE;
}

PROGRESSIVE_CONTEXT* progressive_context_new(BOOL Compressor)
{
  return progressive_context_new_ex(Compressor, 0);
}

PROGRESSIVE_CONTEXT* progressive_context_new_ex(BOOL Compressor, UINT32 ThreadingFlags)
{
  PROGRESSIVE_CONTEXT* progressive =
      (PROGRESSIVE_CONTEXT*)winpr_aligned_calloc(1, sizeof(PROGRESSIVE_CONTEXT), 32);

  if (!progressive)
    return NULL;

  progressive->Compressor = Compressor;
  progressive->quantProgValFull.quality = 100;
  progressive->log = WLog_Get(TAG);
  if (!progressive->log)
    goto fail;
  progressive->rfx_context = rfx_context_new_ex(Compressor, ThreadingFlags);
  if (!progressive->rfx_context)
    goto fail;
  progressive->buffer = Stream_New(NULL, 1024);
  if (!progressive->buffer)
    goto fail;
  progressive->rects = Stream_New(NULL, 1024);
  if (!progressive->rects)
    goto fail;
  progressive->bufferPool = BufferPool_New(TRUE, (8192 + 32) * 3, 16);
  if (!progressive->bufferPool)
    goto fail;
  progressive->SurfaceContexts = HashTable_New(TRUE);
  if (!progressive->SurfaceContexts)
    goto fail;

  {
    wObject* obj = HashTable_ValueObject(progressive->SurfaceContexts);
    WINPR_ASSERT(obj);
    obj->fnObjectFree = progressive_surface_context_free;
  }
  return progressive;
fail:
  progressive_context_free(progressive);
  return NULL;
}

void progressive_context_free(PROGRESSIVE_CONTEXT* progressive)
{
  if (!progressive)
    return;

  Stream_Free(progressive->buffer, TRUE);
  Stream_Free(progressive->rects, TRUE);
  rfx_context_free(progressive->rfx_context);

  BufferPool_Free(progressive->bufferPool);
  HashTable_Free(progressive->SurfaceContexts);

  winpr_aligned_free(progressive);
}

Coverage Report

Created: 2023-09-25 06:56