2012-03-16 08:04:38 +04:00
|
|
|
/**
|
2012-10-09 07:02:04 +04:00
|
|
|
* FreeRDP: A Remote Desktop Protocol Implementation
|
2012-03-16 08:04:38 +04:00
|
|
|
* NSCodec Encoder
|
|
|
|
*
|
|
|
|
* Copyright 2012 Vic Lee
|
|
|
|
*
|
|
|
|
* 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.
|
|
|
|
*/
|
|
|
|
|
2012-08-15 01:09:01 +04:00
|
|
|
#ifdef HAVE_CONFIG_H
|
|
|
|
#include "config.h"
|
|
|
|
#endif
|
|
|
|
|
2012-03-16 08:04:38 +04:00
|
|
|
#include <stdio.h>
|
|
|
|
#include <stdlib.h>
|
|
|
|
#include <string.h>
|
2013-08-15 21:13:02 +04:00
|
|
|
|
2012-10-02 10:48:11 +04:00
|
|
|
#ifdef HAVE_STDINT_H
|
2012-03-16 08:04:38 +04:00
|
|
|
#include <stdint.h>
|
2012-10-02 10:48:11 +04:00
|
|
|
#endif
|
2012-08-15 01:09:01 +04:00
|
|
|
|
2013-08-15 04:57:16 +04:00
|
|
|
#include <winpr/crt.h>
|
|
|
|
|
2012-03-16 08:04:38 +04:00
|
|
|
#include <freerdp/codec/nsc.h>
|
|
|
|
|
|
|
|
#include "nsc_types.h"
|
|
|
|
#include "nsc_encode.h"
|
|
|
|
|
|
|
|
static void nsc_context_initialize_encode(NSC_CONTEXT* context)
|
|
|
|
{
|
|
|
|
int i;
|
2012-10-09 11:26:39 +04:00
|
|
|
UINT32 length;
|
|
|
|
UINT32 tempWidth;
|
|
|
|
UINT32 tempHeight;
|
2012-03-16 08:04:38 +04:00
|
|
|
|
|
|
|
tempWidth = ROUND_UP_TO(context->width, 8);
|
|
|
|
tempHeight = ROUND_UP_TO(context->height, 2);
|
2013-08-15 04:57:16 +04:00
|
|
|
|
2012-03-16 08:04:38 +04:00
|
|
|
/* The maximum length a decoded plane can reach in all cases */
|
2012-03-18 08:02:28 +04:00
|
|
|
length = tempWidth * tempHeight + 16;
|
2013-08-15 04:57:16 +04:00
|
|
|
|
2013-08-15 07:16:13 +04:00
|
|
|
if (length > context->priv->PlaneBuffersLength)
|
2012-03-16 08:04:38 +04:00
|
|
|
{
|
2012-03-18 08:02:28 +04:00
|
|
|
for (i = 0; i < 5; i++)
|
2013-08-15 07:16:13 +04:00
|
|
|
context->priv->PlaneBuffers[i] = (BYTE*) realloc(context->priv->PlaneBuffers[i], length);
|
2013-08-15 04:57:16 +04:00
|
|
|
|
2013-08-15 07:16:13 +04:00
|
|
|
context->priv->PlaneBuffersLength = length;
|
2012-03-16 08:04:38 +04:00
|
|
|
}
|
|
|
|
|
|
|
|
if (context->nsc_stream.ChromaSubSamplingLevel > 0)
|
|
|
|
{
|
2012-03-18 08:02:28 +04:00
|
|
|
context->OrgByteCount[0] = tempWidth * context->height;
|
|
|
|
context->OrgByteCount[1] = tempWidth * tempHeight / 4;
|
|
|
|
context->OrgByteCount[2] = tempWidth * tempHeight / 4;
|
|
|
|
context->OrgByteCount[3] = context->width * context->height;
|
2012-03-16 08:04:38 +04:00
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
2012-03-18 08:02:28 +04:00
|
|
|
context->OrgByteCount[0] = context->width * context->height;
|
|
|
|
context->OrgByteCount[1] = context->width * context->height;
|
|
|
|
context->OrgByteCount[2] = context->width * context->height;
|
|
|
|
context->OrgByteCount[3] = context->width * context->height;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2013-08-15 21:13:02 +04:00
|
|
|
static void nsc_encode_argb_to_aycocg(NSC_CONTEXT* context, BYTE* data, int scanline)
|
2012-03-18 08:02:28 +04:00
|
|
|
{
|
2012-10-09 11:01:37 +04:00
|
|
|
UINT16 x;
|
|
|
|
UINT16 y;
|
|
|
|
UINT16 rw;
|
|
|
|
BYTE ccl;
|
|
|
|
BYTE* src;
|
|
|
|
BYTE* yplane;
|
|
|
|
BYTE* coplane;
|
|
|
|
BYTE* cgplane;
|
|
|
|
BYTE* aplane;
|
|
|
|
INT16 r_val;
|
|
|
|
INT16 g_val;
|
|
|
|
INT16 b_val;
|
|
|
|
BYTE a_val;
|
2012-10-09 11:26:39 +04:00
|
|
|
UINT32 tempWidth;
|
|
|
|
UINT32 tempHeight;
|
2012-03-18 08:02:28 +04:00
|
|
|
|
|
|
|
tempWidth = ROUND_UP_TO(context->width, 8);
|
|
|
|
tempHeight = ROUND_UP_TO(context->height, 2);
|
|
|
|
rw = (context->nsc_stream.ChromaSubSamplingLevel > 0 ? tempWidth : context->width);
|
|
|
|
ccl = context->nsc_stream.ColorLossLevel;
|
2013-08-15 07:16:13 +04:00
|
|
|
yplane = context->priv->PlaneBuffers[0];
|
|
|
|
coplane = context->priv->PlaneBuffers[1];
|
|
|
|
cgplane = context->priv->PlaneBuffers[2];
|
|
|
|
aplane = context->priv->PlaneBuffers[3];
|
2012-03-18 08:02:28 +04:00
|
|
|
|
|
|
|
for (y = 0; y < context->height; y++)
|
|
|
|
{
|
2013-08-15 21:13:02 +04:00
|
|
|
src = data + (context->height - 1 - y) * scanline;
|
2013-08-15 07:16:13 +04:00
|
|
|
yplane = context->priv->PlaneBuffers[0] + y * rw;
|
|
|
|
coplane = context->priv->PlaneBuffers[1] + y * rw;
|
|
|
|
cgplane = context->priv->PlaneBuffers[2] + y * rw;
|
|
|
|
aplane = context->priv->PlaneBuffers[3] + y * context->width;
|
2013-08-15 04:57:16 +04:00
|
|
|
|
2012-03-18 08:02:28 +04:00
|
|
|
for (x = 0; x < context->width; x++)
|
|
|
|
{
|
|
|
|
switch (context->pixel_format)
|
|
|
|
{
|
|
|
|
case RDP_PIXEL_FORMAT_B8G8R8A8:
|
|
|
|
b_val = *src++;
|
|
|
|
g_val = *src++;
|
|
|
|
r_val = *src++;
|
|
|
|
a_val = *src++;
|
|
|
|
break;
|
2013-08-15 04:57:16 +04:00
|
|
|
|
2012-03-18 08:02:28 +04:00
|
|
|
case RDP_PIXEL_FORMAT_R8G8B8A8:
|
|
|
|
r_val = *src++;
|
|
|
|
g_val = *src++;
|
|
|
|
b_val = *src++;
|
|
|
|
a_val = *src++;
|
|
|
|
break;
|
2013-08-15 04:57:16 +04:00
|
|
|
|
2012-03-18 08:02:28 +04:00
|
|
|
case RDP_PIXEL_FORMAT_B8G8R8:
|
|
|
|
b_val = *src++;
|
|
|
|
g_val = *src++;
|
|
|
|
r_val = *src++;
|
|
|
|
a_val = 0xFF;
|
|
|
|
break;
|
2013-08-15 04:57:16 +04:00
|
|
|
|
2012-03-18 08:02:28 +04:00
|
|
|
case RDP_PIXEL_FORMAT_R8G8B8:
|
|
|
|
r_val = *src++;
|
|
|
|
g_val = *src++;
|
|
|
|
b_val = *src++;
|
|
|
|
a_val = 0xFF;
|
|
|
|
break;
|
2013-08-15 04:57:16 +04:00
|
|
|
|
2012-03-18 08:02:28 +04:00
|
|
|
case RDP_PIXEL_FORMAT_B5G6R5_LE:
|
2012-10-09 11:01:37 +04:00
|
|
|
b_val = (INT16) (((*(src + 1)) & 0xF8) | ((*(src + 1)) >> 5));
|
|
|
|
g_val = (INT16) ((((*(src + 1)) & 0x07) << 5) | (((*src) & 0xE0) >> 3));
|
|
|
|
r_val = (INT16) ((((*src) & 0x1F) << 3) | (((*src) >> 2) & 0x07));
|
2012-03-18 08:02:28 +04:00
|
|
|
a_val = 0xFF;
|
|
|
|
src += 2;
|
|
|
|
break;
|
2013-08-15 04:57:16 +04:00
|
|
|
|
2012-03-18 08:02:28 +04:00
|
|
|
case RDP_PIXEL_FORMAT_R5G6B5_LE:
|
2012-10-09 11:01:37 +04:00
|
|
|
r_val = (INT16) (((*(src + 1)) & 0xF8) | ((*(src + 1)) >> 5));
|
|
|
|
g_val = (INT16) ((((*(src + 1)) & 0x07) << 5) | (((*src) & 0xE0) >> 3));
|
|
|
|
b_val = (INT16) ((((*src) & 0x1F) << 3) | (((*src) >> 2) & 0x07));
|
2012-03-18 08:02:28 +04:00
|
|
|
a_val = 0xFF;
|
|
|
|
src += 2;
|
|
|
|
break;
|
2013-08-15 04:57:16 +04:00
|
|
|
|
2012-03-18 08:02:28 +04:00
|
|
|
case RDP_PIXEL_FORMAT_P4_PLANER:
|
|
|
|
{
|
|
|
|
int shift;
|
2012-10-09 11:01:37 +04:00
|
|
|
BYTE idx;
|
2012-03-18 08:02:28 +04:00
|
|
|
|
|
|
|
shift = (7 - (x % 8));
|
|
|
|
idx = ((*src) >> shift) & 1;
|
|
|
|
idx |= (((*(src + 1)) >> shift) & 1) << 1;
|
|
|
|
idx |= (((*(src + 2)) >> shift) & 1) << 2;
|
|
|
|
idx |= (((*(src + 3)) >> shift) & 1) << 3;
|
|
|
|
idx *= 3;
|
2012-10-09 11:01:37 +04:00
|
|
|
r_val = (INT16) context->palette[idx];
|
|
|
|
g_val = (INT16) context->palette[idx + 1];
|
|
|
|
b_val = (INT16) context->palette[idx + 2];
|
2012-03-18 08:02:28 +04:00
|
|
|
if (shift == 0)
|
|
|
|
src += 4;
|
|
|
|
}
|
|
|
|
a_val = 0xFF;
|
|
|
|
break;
|
2013-08-15 04:57:16 +04:00
|
|
|
|
2012-03-18 08:02:28 +04:00
|
|
|
case RDP_PIXEL_FORMAT_P8:
|
|
|
|
{
|
|
|
|
int idx = (*src) * 3;
|
|
|
|
|
2012-10-09 11:01:37 +04:00
|
|
|
r_val = (INT16) context->palette[idx];
|
|
|
|
g_val = (INT16) context->palette[idx + 1];
|
|
|
|
b_val = (INT16) context->palette[idx + 2];
|
2012-03-18 08:02:28 +04:00
|
|
|
src++;
|
|
|
|
}
|
|
|
|
a_val = 0xFF;
|
|
|
|
break;
|
2013-08-15 04:57:16 +04:00
|
|
|
|
2012-03-18 08:02:28 +04:00
|
|
|
default:
|
|
|
|
r_val = g_val = b_val = a_val = 0;
|
|
|
|
break;
|
|
|
|
}
|
2013-08-15 04:57:16 +04:00
|
|
|
|
2012-10-09 11:01:37 +04:00
|
|
|
*yplane++ = (BYTE) ((r_val >> 2) + (g_val >> 1) + (b_val >> 2));
|
2013-08-15 04:57:16 +04:00
|
|
|
|
2012-03-18 08:02:28 +04:00
|
|
|
/* Perform color loss reduction here */
|
2012-10-09 11:01:37 +04:00
|
|
|
*coplane++ = (BYTE) ((r_val - b_val) >> ccl);
|
|
|
|
*cgplane++ = (BYTE) ((-(r_val >> 1) + g_val - (b_val >> 1)) >> ccl);
|
2012-03-18 08:02:28 +04:00
|
|
|
*aplane++ = a_val;
|
|
|
|
}
|
2013-08-15 04:57:16 +04:00
|
|
|
|
2012-03-18 08:02:28 +04:00
|
|
|
if (context->nsc_stream.ChromaSubSamplingLevel > 0 && (x % 2) == 1)
|
|
|
|
{
|
|
|
|
*yplane = *(yplane - 1);
|
|
|
|
*coplane = *(coplane - 1);
|
|
|
|
*cgplane = *(cgplane - 1);
|
|
|
|
}
|
|
|
|
}
|
2013-08-15 04:57:16 +04:00
|
|
|
|
2012-03-18 08:02:28 +04:00
|
|
|
if (context->nsc_stream.ChromaSubSamplingLevel > 0 && (y % 2) == 1)
|
|
|
|
{
|
2013-08-15 04:57:16 +04:00
|
|
|
CopyMemory(yplane + rw, yplane, rw);
|
|
|
|
CopyMemory(coplane + rw, coplane, rw);
|
|
|
|
CopyMemory(cgplane + rw, cgplane, rw);
|
2012-03-18 08:02:28 +04:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
static void nsc_encode_subsampling(NSC_CONTEXT* context)
|
|
|
|
{
|
2012-10-09 11:01:37 +04:00
|
|
|
UINT16 x;
|
|
|
|
UINT16 y;
|
|
|
|
BYTE* co_dst;
|
|
|
|
BYTE* cg_dst;
|
|
|
|
INT8* co_src0;
|
|
|
|
INT8* co_src1;
|
|
|
|
INT8* cg_src0;
|
|
|
|
INT8* cg_src1;
|
2012-10-09 11:26:39 +04:00
|
|
|
UINT32 tempWidth;
|
|
|
|
UINT32 tempHeight;
|
2012-03-18 08:02:28 +04:00
|
|
|
|
|
|
|
tempWidth = ROUND_UP_TO(context->width, 8);
|
|
|
|
tempHeight = ROUND_UP_TO(context->height, 2);
|
|
|
|
|
|
|
|
for (y = 0; y < tempHeight >> 1; y++)
|
|
|
|
{
|
2013-08-15 07:16:13 +04:00
|
|
|
co_dst = context->priv->PlaneBuffers[1] + y * (tempWidth >> 1);
|
|
|
|
cg_dst = context->priv->PlaneBuffers[2] + y * (tempWidth >> 1);
|
|
|
|
co_src0 = (INT8*) context->priv->PlaneBuffers[1] + (y << 1) * tempWidth;
|
2012-03-18 08:02:28 +04:00
|
|
|
co_src1 = co_src0 + tempWidth;
|
2013-08-15 07:16:13 +04:00
|
|
|
cg_src0 = (INT8*) context->priv->PlaneBuffers[2] + (y << 1) * tempWidth;
|
2012-03-18 08:02:28 +04:00
|
|
|
cg_src1 = cg_src0 + tempWidth;
|
2013-08-15 04:57:16 +04:00
|
|
|
|
2012-03-18 08:02:28 +04:00
|
|
|
for (x = 0; x < tempWidth >> 1; x++)
|
|
|
|
{
|
2012-10-09 11:01:37 +04:00
|
|
|
*co_dst++ = (BYTE) (((INT16) *co_src0 + (INT16) *(co_src0 + 1) +
|
|
|
|
(INT16) *co_src1 + (INT16) *(co_src1 + 1)) >> 2);
|
|
|
|
*cg_dst++ = (BYTE) (((INT16) *cg_src0 + (INT16) *(cg_src0 + 1) +
|
|
|
|
(INT16) *cg_src1 + (INT16) *(cg_src1 + 1)) >> 2);
|
2012-03-18 08:02:28 +04:00
|
|
|
co_src0 += 2;
|
|
|
|
co_src1 += 2;
|
|
|
|
cg_src0 += 2;
|
|
|
|
cg_src1 += 2;
|
|
|
|
}
|
2012-03-16 08:04:38 +04:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2012-10-09 11:01:37 +04:00
|
|
|
void nsc_encode(NSC_CONTEXT* context, BYTE* bmpdata, int rowstride)
|
2012-03-16 08:04:38 +04:00
|
|
|
{
|
2012-03-18 08:02:28 +04:00
|
|
|
nsc_encode_argb_to_aycocg(context, bmpdata, rowstride);
|
2013-08-15 04:57:16 +04:00
|
|
|
|
2012-03-18 08:02:28 +04:00
|
|
|
if (context->nsc_stream.ChromaSubSamplingLevel > 0)
|
|
|
|
{
|
|
|
|
nsc_encode_subsampling(context);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2013-08-15 04:57:16 +04:00
|
|
|
static UINT32 nsc_rle_encode(BYTE* in, BYTE* out, UINT32 originalSize)
|
2012-03-18 08:02:28 +04:00
|
|
|
{
|
2012-10-09 11:26:39 +04:00
|
|
|
UINT32 left;
|
|
|
|
UINT32 runlength = 1;
|
2013-08-15 04:57:16 +04:00
|
|
|
UINT32 planeSize = 0;
|
2012-03-18 08:02:28 +04:00
|
|
|
|
2013-08-15 04:57:16 +04:00
|
|
|
left = originalSize;
|
2012-03-18 08:02:28 +04:00
|
|
|
/**
|
|
|
|
* We quit the loop if the running compressed size is larger than the original.
|
|
|
|
* In such cases data will be sent uncompressed.
|
|
|
|
*/
|
2013-08-15 04:57:16 +04:00
|
|
|
while (left > 4 && planeSize < originalSize - 4)
|
2012-03-18 08:02:28 +04:00
|
|
|
{
|
|
|
|
if (left > 5 && *in == *(in + 1))
|
|
|
|
{
|
|
|
|
runlength++;
|
|
|
|
}
|
|
|
|
else if (runlength == 1)
|
|
|
|
{
|
|
|
|
*out++ = *in;
|
2013-08-15 04:57:16 +04:00
|
|
|
planeSize++;
|
2012-03-18 08:02:28 +04:00
|
|
|
}
|
|
|
|
else if (runlength < 256)
|
|
|
|
{
|
|
|
|
*out++ = *in;
|
|
|
|
*out++ = *in;
|
|
|
|
*out++ = runlength - 2;
|
|
|
|
runlength = 1;
|
2013-08-15 04:57:16 +04:00
|
|
|
planeSize += 3;
|
2012-03-18 08:02:28 +04:00
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
*out++ = *in;
|
|
|
|
*out++ = *in;
|
|
|
|
*out++ = 0xFF;
|
|
|
|
*out++ = (runlength & 0x000000FF);
|
|
|
|
*out++ = (runlength & 0x0000FF00) >> 8;
|
|
|
|
*out++ = (runlength & 0x00FF0000) >> 16;
|
|
|
|
*out++ = (runlength & 0xFF000000) >> 24;
|
|
|
|
runlength = 1;
|
2013-08-15 04:57:16 +04:00
|
|
|
planeSize += 7;
|
2012-03-18 08:02:28 +04:00
|
|
|
}
|
|
|
|
in++;
|
|
|
|
left--;
|
|
|
|
}
|
|
|
|
|
2013-08-15 04:57:16 +04:00
|
|
|
if (planeSize < originalSize - 4)
|
|
|
|
CopyMemory(out, in, 4);
|
|
|
|
|
|
|
|
planeSize += 4;
|
|
|
|
|
|
|
|
return planeSize;
|
2012-03-18 08:02:28 +04:00
|
|
|
}
|
|
|
|
|
|
|
|
static void nsc_rle_compress_data(NSC_CONTEXT* context)
|
|
|
|
{
|
2012-10-09 11:01:37 +04:00
|
|
|
UINT16 i;
|
2013-08-15 04:57:16 +04:00
|
|
|
UINT32 planeSize;
|
|
|
|
UINT32 originalSize;
|
2012-03-18 08:02:28 +04:00
|
|
|
|
|
|
|
for (i = 0; i < 4; i++)
|
|
|
|
{
|
2013-08-15 04:57:16 +04:00
|
|
|
originalSize = context->OrgByteCount[i];
|
|
|
|
|
|
|
|
if (originalSize == 0)
|
2012-03-18 08:02:28 +04:00
|
|
|
{
|
2013-08-15 04:57:16 +04:00
|
|
|
planeSize = 0;
|
2012-03-18 08:02:28 +04:00
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
2013-08-15 07:16:13 +04:00
|
|
|
planeSize = nsc_rle_encode(context->priv->PlaneBuffers[i], context->priv->PlaneBuffers[4], originalSize);
|
2013-08-15 04:57:16 +04:00
|
|
|
|
|
|
|
if (planeSize < originalSize)
|
2013-08-15 07:16:13 +04:00
|
|
|
CopyMemory(context->priv->PlaneBuffers[i], context->priv->PlaneBuffers[4], planeSize);
|
2012-03-18 08:02:28 +04:00
|
|
|
else
|
2013-08-15 04:57:16 +04:00
|
|
|
planeSize = originalSize;
|
2012-03-18 08:02:28 +04:00
|
|
|
}
|
|
|
|
|
2013-08-15 04:57:16 +04:00
|
|
|
context->nsc_stream.PlaneByteCount[i] = planeSize;
|
2012-03-18 08:02:28 +04:00
|
|
|
}
|
2012-03-16 08:04:38 +04:00
|
|
|
}
|
|
|
|
|
2013-08-15 07:16:13 +04:00
|
|
|
UINT32 nsc_compute_byte_count(NSC_CONTEXT* context, UINT32* ByteCount, UINT32 width, UINT32 height)
|
|
|
|
{
|
|
|
|
UINT32 tempWidth;
|
|
|
|
UINT32 tempHeight;
|
|
|
|
UINT32 maxPlaneSize;
|
|
|
|
|
|
|
|
tempWidth = ROUND_UP_TO(width, 8);
|
|
|
|
tempHeight = ROUND_UP_TO(height, 2);
|
|
|
|
|
|
|
|
maxPlaneSize = tempWidth * tempHeight + 16;
|
|
|
|
|
|
|
|
if (context->nsc_stream.ChromaSubSamplingLevel > 0)
|
|
|
|
{
|
|
|
|
ByteCount[0] = tempWidth * height;
|
|
|
|
ByteCount[1] = tempWidth * tempHeight / 4;
|
|
|
|
ByteCount[2] = tempWidth * tempHeight / 4;
|
|
|
|
ByteCount[3] = width * height;
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
ByteCount[0] = width * height;
|
|
|
|
ByteCount[1] = width * height;
|
|
|
|
ByteCount[2] = width * height;
|
|
|
|
ByteCount[3] = width * height;
|
|
|
|
}
|
|
|
|
|
|
|
|
return maxPlaneSize;
|
|
|
|
}
|
|
|
|
|
|
|
|
NSC_MESSAGE* nsc_encode_messages(NSC_CONTEXT* context, BYTE* data, int x, int y,
|
|
|
|
int width, int height, int scanline, int* numMessages, int maxDataSize)
|
|
|
|
{
|
2013-08-15 21:13:02 +04:00
|
|
|
int i, j, k;
|
|
|
|
int dataOffset;
|
|
|
|
int rows, cols;
|
|
|
|
int BytesPerPixel;
|
|
|
|
int MaxRegionWidth;
|
|
|
|
int MaxRegionHeight;
|
2013-08-15 07:16:13 +04:00
|
|
|
UINT32 ByteCount[4];
|
|
|
|
UINT32 MaxPlaneSize;
|
|
|
|
UINT32 MaxMessageSize;
|
|
|
|
NSC_MESSAGE* messages;
|
2013-08-20 02:21:34 +04:00
|
|
|
UINT32 PaddedMaxPlaneSize;
|
2013-08-15 07:16:13 +04:00
|
|
|
|
2013-08-15 21:13:02 +04:00
|
|
|
k = 0;
|
|
|
|
MaxRegionWidth = 64 * 4;
|
|
|
|
MaxRegionHeight = 64 * 2;
|
|
|
|
BytesPerPixel = (context->bpp / 8);
|
2013-08-15 07:16:13 +04:00
|
|
|
|
2013-08-15 21:13:02 +04:00
|
|
|
rows = (width + (MaxRegionWidth - (width % MaxRegionWidth))) / MaxRegionWidth;
|
|
|
|
cols = (height + (MaxRegionHeight - (height % MaxRegionHeight))) / MaxRegionHeight;
|
|
|
|
*numMessages = rows * cols;
|
2013-08-15 07:16:13 +04:00
|
|
|
|
2013-08-15 21:13:02 +04:00
|
|
|
MaxPlaneSize = nsc_compute_byte_count(context, (UINT32*) ByteCount, width, height);
|
|
|
|
MaxMessageSize = ByteCount[0] + ByteCount[1] + ByteCount[2] + ByteCount[3] + 20;
|
2013-08-15 07:16:13 +04:00
|
|
|
|
2013-08-15 21:13:02 +04:00
|
|
|
maxDataSize -= 1024; /* reserve enough space for headers */
|
2013-08-15 07:16:13 +04:00
|
|
|
|
|
|
|
messages = (NSC_MESSAGE*) malloc(sizeof(NSC_MESSAGE) * (*numMessages));
|
2014-03-05 16:49:00 +04:00
|
|
|
ZeroMemory(messages, sizeof(NSC_MESSAGE) * (*numMessages));
|
2013-08-15 07:16:13 +04:00
|
|
|
|
2013-08-15 21:13:02 +04:00
|
|
|
for (i = 0; i < rows; i++)
|
2013-08-15 07:16:13 +04:00
|
|
|
{
|
2013-08-15 21:13:02 +04:00
|
|
|
for (j = 0; j < cols; j++)
|
2013-08-15 07:16:13 +04:00
|
|
|
{
|
2013-08-15 21:13:02 +04:00
|
|
|
messages[k].x = x + (i * MaxRegionWidth);
|
|
|
|
messages[k].y = y + (j * MaxRegionHeight);
|
|
|
|
messages[k].width = (i < (rows - 1)) ? MaxRegionWidth : width - (i * MaxRegionWidth);
|
|
|
|
messages[k].height = (j < (cols - 1)) ? MaxRegionHeight : height - (j * MaxRegionHeight);
|
|
|
|
messages[k].data = data;
|
|
|
|
messages[k].scanline = scanline;
|
2013-08-15 07:16:13 +04:00
|
|
|
|
2013-08-15 21:13:02 +04:00
|
|
|
messages[k].MaxPlaneSize = nsc_compute_byte_count(context,
|
|
|
|
(UINT32*) messages[k].OrgByteCount, messages[k].width, messages[k].height);
|
2013-08-15 07:16:13 +04:00
|
|
|
|
2013-08-15 21:13:02 +04:00
|
|
|
k++;
|
2013-08-15 07:16:13 +04:00
|
|
|
}
|
|
|
|
}
|
2013-08-15 21:13:02 +04:00
|
|
|
|
|
|
|
*numMessages = k;
|
|
|
|
|
|
|
|
for (i = 0; i < *numMessages; i++)
|
|
|
|
{
|
2013-08-20 02:21:34 +04:00
|
|
|
PaddedMaxPlaneSize = messages[i].MaxPlaneSize + 32;
|
|
|
|
|
|
|
|
messages[i].PlaneBuffer = (BYTE*) BufferPool_Take(context->priv->PlanePool, PaddedMaxPlaneSize * 5);
|
|
|
|
|
|
|
|
messages[i].PlaneBuffers[0] = (BYTE*) &(messages[i].PlaneBuffer[(PaddedMaxPlaneSize * 0) + 16]);
|
|
|
|
messages[i].PlaneBuffers[1] = (BYTE*) &(messages[i].PlaneBuffer[(PaddedMaxPlaneSize * 1) + 16]);
|
|
|
|
messages[i].PlaneBuffers[2] = (BYTE*) &(messages[i].PlaneBuffer[(PaddedMaxPlaneSize * 2) + 16]);
|
|
|
|
messages[i].PlaneBuffers[3] = (BYTE*) &(messages[i].PlaneBuffer[(PaddedMaxPlaneSize * 3) + 16]);
|
|
|
|
messages[i].PlaneBuffers[4] = (BYTE*) &(messages[i].PlaneBuffer[(PaddedMaxPlaneSize * 4) + 16]);
|
2013-08-15 21:13:02 +04:00
|
|
|
}
|
|
|
|
|
|
|
|
for (i = 0; i < *numMessages; i++)
|
2013-08-15 07:16:13 +04:00
|
|
|
{
|
2013-08-15 21:13:02 +04:00
|
|
|
context->width = messages[i].width;
|
|
|
|
context->height = messages[i].height;
|
|
|
|
context->OrgByteCount[0] = messages[i].OrgByteCount[0];
|
|
|
|
context->OrgByteCount[1] = messages[i].OrgByteCount[1];
|
|
|
|
context->OrgByteCount[2] = messages[i].OrgByteCount[2];
|
|
|
|
context->OrgByteCount[3] = messages[i].OrgByteCount[3];
|
|
|
|
context->priv->PlaneBuffersLength = messages[i].MaxPlaneSize;
|
|
|
|
context->priv->PlaneBuffers[0] = messages[i].PlaneBuffers[0];
|
|
|
|
context->priv->PlaneBuffers[1] = messages[i].PlaneBuffers[1];
|
|
|
|
context->priv->PlaneBuffers[2] = messages[i].PlaneBuffers[2];
|
|
|
|
context->priv->PlaneBuffers[3] = messages[i].PlaneBuffers[3];
|
|
|
|
context->priv->PlaneBuffers[4] = messages[i].PlaneBuffers[4];
|
|
|
|
|
|
|
|
dataOffset = (messages[i].y * messages[i].scanline) + (messages[i].x * BytesPerPixel);
|
|
|
|
|
|
|
|
PROFILER_ENTER(context->priv->prof_nsc_encode);
|
|
|
|
context->encode(context, &data[dataOffset], scanline);
|
|
|
|
PROFILER_EXIT(context->priv->prof_nsc_encode);
|
|
|
|
|
|
|
|
PROFILER_ENTER(context->priv->prof_nsc_rle_compress_data);
|
|
|
|
nsc_rle_compress_data(context);
|
|
|
|
PROFILER_EXIT(context->priv->prof_nsc_rle_compress_data);
|
|
|
|
|
|
|
|
messages[i].PlaneByteCount[0] = context->nsc_stream.PlaneByteCount[0];
|
|
|
|
messages[i].PlaneByteCount[1] = context->nsc_stream.PlaneByteCount[1];
|
|
|
|
messages[i].PlaneByteCount[2] = context->nsc_stream.PlaneByteCount[2];
|
|
|
|
messages[i].PlaneByteCount[3] = context->nsc_stream.PlaneByteCount[3];
|
|
|
|
}
|
2013-08-15 07:16:13 +04:00
|
|
|
|
2013-08-15 21:13:02 +04:00
|
|
|
context->priv->PlaneBuffers[0] = NULL;
|
|
|
|
context->priv->PlaneBuffers[1] = NULL;
|
|
|
|
context->priv->PlaneBuffers[2] = NULL;
|
|
|
|
context->priv->PlaneBuffers[3] = NULL;
|
|
|
|
context->priv->PlaneBuffers[4] = NULL;
|
2013-08-15 07:16:13 +04:00
|
|
|
|
2013-08-15 21:13:02 +04:00
|
|
|
return messages;
|
|
|
|
}
|
2013-08-15 07:16:13 +04:00
|
|
|
|
2013-08-15 21:13:02 +04:00
|
|
|
int nsc_write_message(NSC_CONTEXT* context, wStream* s, NSC_MESSAGE* message)
|
|
|
|
{
|
|
|
|
int i;
|
2013-08-15 07:16:13 +04:00
|
|
|
|
2013-08-15 21:13:02 +04:00
|
|
|
Stream_EnsureRemainingCapacity(s, 20);
|
|
|
|
Stream_Write_UINT32(s, message->PlaneByteCount[0]); /* LumaPlaneByteCount (4 bytes) */
|
|
|
|
Stream_Write_UINT32(s, message->PlaneByteCount[1]); /* OrangeChromaPlaneByteCount (4 bytes) */
|
|
|
|
Stream_Write_UINT32(s, message->PlaneByteCount[2]); /* GreenChromaPlaneByteCount (4 bytes) */
|
|
|
|
Stream_Write_UINT32(s, message->PlaneByteCount[3]); /* AlphaPlaneByteCount (4 bytes) */
|
|
|
|
Stream_Write_UINT8(s, context->nsc_stream.ColorLossLevel); /* ColorLossLevel (1 byte) */
|
|
|
|
Stream_Write_UINT8(s, context->nsc_stream.ChromaSubSamplingLevel); /* ChromaSubsamplingLevel (1 byte) */
|
|
|
|
Stream_Write_UINT16(s, 0); /* Reserved (2 bytes) */
|
2013-08-15 07:16:13 +04:00
|
|
|
|
2013-08-15 21:13:02 +04:00
|
|
|
for (i = 0; i < 4; i++)
|
2013-08-15 07:16:13 +04:00
|
|
|
{
|
2013-08-15 21:13:02 +04:00
|
|
|
if (message->PlaneByteCount[i] > 0)
|
2013-08-15 07:16:13 +04:00
|
|
|
{
|
2013-08-15 21:13:02 +04:00
|
|
|
Stream_EnsureRemainingCapacity(s, (int) message->PlaneByteCount[i]);
|
|
|
|
Stream_Write(s, message->PlaneBuffers[i], message->PlaneByteCount[i]);
|
2013-08-15 07:16:13 +04:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2013-08-15 21:13:02 +04:00
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
int nsc_message_free(NSC_CONTEXT* context, NSC_MESSAGE* message)
|
|
|
|
{
|
2013-08-20 02:21:34 +04:00
|
|
|
BufferPool_Return(context->priv->PlanePool, message->PlaneBuffer);
|
2013-08-15 21:13:02 +04:00
|
|
|
return 0;
|
2013-08-15 07:16:13 +04:00
|
|
|
}
|
|
|
|
|
|
|
|
void nsc_compose_message(NSC_CONTEXT* context, wStream* s, BYTE* data, int width, int height, int scanline)
|
2012-03-16 08:04:38 +04:00
|
|
|
{
|
2012-03-18 08:02:28 +04:00
|
|
|
int i;
|
|
|
|
|
2012-03-16 08:04:38 +04:00
|
|
|
context->width = width;
|
|
|
|
context->height = height;
|
|
|
|
nsc_context_initialize_encode(context);
|
2012-03-18 08:02:28 +04:00
|
|
|
|
|
|
|
/* ARGB to AYCoCg conversion, chroma subsampling and colorloss reduction */
|
|
|
|
PROFILER_ENTER(context->priv->prof_nsc_encode);
|
2013-08-15 07:16:13 +04:00
|
|
|
context->encode(context, data, scanline);
|
2012-03-18 08:02:28 +04:00
|
|
|
PROFILER_EXIT(context->priv->prof_nsc_encode);
|
|
|
|
|
|
|
|
/* RLE encode */
|
|
|
|
PROFILER_ENTER(context->priv->prof_nsc_rle_compress_data);
|
|
|
|
nsc_rle_compress_data(context);
|
|
|
|
PROFILER_EXIT(context->priv->prof_nsc_rle_compress_data);
|
|
|
|
|
|
|
|
/* Assemble the NSCodec message into stream */
|
2013-05-02 02:15:55 +04:00
|
|
|
Stream_EnsureRemainingCapacity(s, 20);
|
2013-05-09 00:09:16 +04:00
|
|
|
Stream_Write_UINT32(s, context->nsc_stream.PlaneByteCount[0]); /* LumaPlaneByteCount (4 bytes) */
|
|
|
|
Stream_Write_UINT32(s, context->nsc_stream.PlaneByteCount[1]); /* OrangeChromaPlaneByteCount (4 bytes) */
|
|
|
|
Stream_Write_UINT32(s, context->nsc_stream.PlaneByteCount[2]); /* GreenChromaPlaneByteCount (4 bytes) */
|
|
|
|
Stream_Write_UINT32(s, context->nsc_stream.PlaneByteCount[3]); /* AlphaPlaneByteCount (4 bytes) */
|
|
|
|
Stream_Write_UINT8(s, context->nsc_stream.ColorLossLevel); /* ColorLossLevel (1 byte) */
|
|
|
|
Stream_Write_UINT8(s, context->nsc_stream.ChromaSubSamplingLevel); /* ChromaSubsamplingLevel (1 byte) */
|
|
|
|
Stream_Write_UINT16(s, 0); /* Reserved (2 bytes) */
|
2012-03-24 02:26:03 +04:00
|
|
|
|
2012-03-18 08:02:28 +04:00
|
|
|
for (i = 0; i < 4; i++)
|
|
|
|
{
|
|
|
|
if (context->nsc_stream.PlaneByteCount[i] > 0)
|
|
|
|
{
|
2013-05-02 02:15:55 +04:00
|
|
|
Stream_EnsureRemainingCapacity(s, (int) context->nsc_stream.PlaneByteCount[i]);
|
2013-08-15 07:16:13 +04:00
|
|
|
Stream_Write(s, context->priv->PlaneBuffers[i], context->nsc_stream.PlaneByteCount[i]);
|
2012-03-18 08:02:28 +04:00
|
|
|
}
|
|
|
|
}
|
2012-03-16 08:04:38 +04:00
|
|
|
}
|