2011-10-02 23:26:33 +04:00
|
|
|
/**
|
2011-10-03 05:07:07 +04:00
|
|
|
* FreeRDP: A Remote Desktop Protocol client.
|
|
|
|
* NSCodec Codec
|
|
|
|
*
|
|
|
|
* Copyright 2011 Samsung, Author Jiten Pathy
|
2012-03-06 14:52:28 +04:00
|
|
|
* Copyright 2012 Vic Lee
|
2011-10-03 05:07:07 +04:00
|
|
|
*
|
|
|
|
* 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.
|
|
|
|
*/
|
2011-10-03 04:52:17 +04:00
|
|
|
|
|
|
|
#include <stdio.h>
|
|
|
|
#include <stdlib.h>
|
|
|
|
#include <string.h>
|
|
|
|
#include <stdint.h>
|
|
|
|
#include <freerdp/codec/nsc.h>
|
|
|
|
#include <freerdp/utils/memory.h>
|
2011-10-02 23:26:33 +04:00
|
|
|
|
2012-03-06 16:06:08 +04:00
|
|
|
#include "nsc_types.h"
|
2012-03-16 08:04:38 +04:00
|
|
|
#include "nsc_encode.h"
|
2011-10-03 05:07:07 +04:00
|
|
|
|
2012-03-18 19:25:07 +04:00
|
|
|
#ifdef WITH_SSE2
|
|
|
|
#include "nsc_sse2.h"
|
|
|
|
#endif
|
|
|
|
|
2012-03-06 16:06:08 +04:00
|
|
|
#ifndef NSC_INIT_SIMD
|
|
|
|
#define NSC_INIT_SIMD(_nsc_context) do { } while (0)
|
|
|
|
#endif
|
|
|
|
|
2012-03-06 14:52:28 +04:00
|
|
|
static void nsc_decode(NSC_CONTEXT* context)
|
2011-10-02 23:26:33 +04:00
|
|
|
{
|
2012-03-06 14:52:28 +04:00
|
|
|
uint16 x;
|
|
|
|
uint16 y;
|
|
|
|
uint16 rw;
|
|
|
|
uint8 shift;
|
|
|
|
uint8* yplane;
|
|
|
|
uint8* coplane;
|
|
|
|
uint8* cgplane;
|
|
|
|
uint8* aplane;
|
|
|
|
sint16 y_val;
|
|
|
|
sint16 co_val;
|
|
|
|
sint16 cg_val;
|
|
|
|
sint16 r_val;
|
|
|
|
sint16 g_val;
|
|
|
|
sint16 b_val;
|
|
|
|
uint8* bmpdata;
|
|
|
|
|
|
|
|
bmpdata = context->bmpdata;
|
|
|
|
rw = ROUND_UP_TO(context->width, 8);
|
|
|
|
shift = context->nsc_stream.ColorLossLevel - 1; /* colorloss recovery + YCoCg shift */
|
|
|
|
|
|
|
|
for (y = 0; y < context->height; y++)
|
2011-10-02 23:26:33 +04:00
|
|
|
{
|
2012-03-06 14:52:28 +04:00
|
|
|
if (context->nsc_stream.ChromaSubSamplingLevel > 0)
|
2011-10-02 23:26:33 +04:00
|
|
|
{
|
2012-03-06 14:52:28 +04:00
|
|
|
yplane = context->priv->plane_buf[0] + y * rw; /* Y */
|
|
|
|
coplane = context->priv->plane_buf[1] + (y >> 1) * (rw >> 1); /* Co, supersampled */
|
|
|
|
cgplane = context->priv->plane_buf[2] + (y >> 1) * (rw >> 1); /* Cg, supersampled */
|
2011-10-02 23:26:33 +04:00
|
|
|
}
|
2012-03-06 14:52:28 +04:00
|
|
|
else
|
2011-10-02 23:26:33 +04:00
|
|
|
{
|
2012-03-06 14:52:28 +04:00
|
|
|
yplane = context->priv->plane_buf[0] + y * context->width; /* Y */
|
|
|
|
coplane = context->priv->plane_buf[1] + y * context->width; /* Co */
|
|
|
|
cgplane = context->priv->plane_buf[2] + y * context->width; /* Cg */
|
2011-10-02 23:26:33 +04:00
|
|
|
}
|
2012-03-06 14:52:28 +04:00
|
|
|
aplane = context->priv->plane_buf[3] + y * context->width; /* A */
|
|
|
|
for (x = 0; x < context->width; x++)
|
2011-10-02 23:26:33 +04:00
|
|
|
{
|
2012-03-06 14:52:28 +04:00
|
|
|
y_val = (sint16) *yplane;
|
|
|
|
co_val = (sint16) (sint8) (*coplane << shift);
|
|
|
|
cg_val = (sint16) (sint8) (*cgplane << shift);
|
|
|
|
r_val = y_val + co_val - cg_val;
|
|
|
|
g_val = y_val + cg_val;
|
|
|
|
b_val = y_val - co_val - cg_val;
|
|
|
|
*bmpdata++ = MINMAX(b_val, 0, 0xFF);
|
|
|
|
*bmpdata++ = MINMAX(g_val, 0, 0xFF);
|
|
|
|
*bmpdata++ = MINMAX(r_val, 0, 0xFF);
|
|
|
|
*bmpdata++ = *aplane;
|
|
|
|
yplane++;
|
|
|
|
coplane += (context->nsc_stream.ChromaSubSamplingLevel > 0 ? x % 2 : 1);
|
|
|
|
cgplane += (context->nsc_stream.ChromaSubSamplingLevel > 0 ? x % 2 : 1);
|
|
|
|
aplane++;
|
2011-10-02 23:26:33 +04:00
|
|
|
}
|
2011-10-03 05:07:07 +04:00
|
|
|
}
|
2011-10-02 23:26:33 +04:00
|
|
|
}
|
|
|
|
|
2012-03-06 14:52:28 +04:00
|
|
|
static void nsc_rle_decode(uint8* in, uint8* out, uint32 origsz)
|
2011-10-02 23:26:33 +04:00
|
|
|
{
|
2012-03-06 14:52:28 +04:00
|
|
|
uint32 len;
|
|
|
|
uint32 left;
|
2011-10-02 23:26:33 +04:00
|
|
|
uint8 value;
|
2011-10-03 05:07:07 +04:00
|
|
|
|
2012-03-06 14:52:28 +04:00
|
|
|
left = origsz;
|
|
|
|
while (left > 4)
|
2011-10-02 23:26:33 +04:00
|
|
|
{
|
2012-03-06 14:52:28 +04:00
|
|
|
value = *in++;
|
2011-10-03 05:07:07 +04:00
|
|
|
|
2012-03-06 14:52:28 +04:00
|
|
|
if (left == 5)
|
2011-10-02 23:26:33 +04:00
|
|
|
{
|
2012-03-06 14:52:28 +04:00
|
|
|
*out++ = value;
|
|
|
|
left--;
|
2011-10-02 23:26:33 +04:00
|
|
|
}
|
2012-03-06 14:52:28 +04:00
|
|
|
else if (value == *in)
|
2011-10-02 23:26:33 +04:00
|
|
|
{
|
2012-03-06 14:52:28 +04:00
|
|
|
in++;
|
|
|
|
if (*in < 0xFF)
|
2011-10-02 23:26:33 +04:00
|
|
|
{
|
2012-03-06 14:52:28 +04:00
|
|
|
len = (uint32) *in++;
|
|
|
|
len += 2;
|
2011-10-02 23:26:33 +04:00
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
2012-03-06 14:52:28 +04:00
|
|
|
in++;
|
|
|
|
len = *((uint32*) in);
|
|
|
|
in += 4;
|
2011-10-02 23:26:33 +04:00
|
|
|
}
|
2012-03-07 07:42:37 +04:00
|
|
|
memset(out, value, len);
|
|
|
|
out += len;
|
2012-03-06 14:52:28 +04:00
|
|
|
left -= len;
|
2011-10-02 23:26:33 +04:00
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
2012-03-06 14:52:28 +04:00
|
|
|
*out++ = value;
|
|
|
|
left--;
|
2011-10-02 23:26:33 +04:00
|
|
|
}
|
|
|
|
}
|
2011-10-03 05:07:07 +04:00
|
|
|
|
2012-03-06 14:52:28 +04:00
|
|
|
*((uint32*)out) = *((uint32*)in);
|
2011-10-02 23:26:33 +04:00
|
|
|
}
|
|
|
|
|
2012-03-06 14:52:28 +04:00
|
|
|
static void nsc_rle_decompress_data(NSC_CONTEXT* context)
|
2011-10-02 23:26:33 +04:00
|
|
|
{
|
2011-11-09 02:55:47 +04:00
|
|
|
uint16 i;
|
2012-03-06 14:52:28 +04:00
|
|
|
uint8* rle;
|
2011-11-09 02:55:47 +04:00
|
|
|
uint32 origsize;
|
2012-03-06 14:52:28 +04:00
|
|
|
uint32 planesize;
|
2012-03-05 14:34:29 +04:00
|
|
|
|
2012-03-06 14:52:28 +04:00
|
|
|
rle = context->nsc_stream.Planes;
|
2011-10-03 05:07:07 +04:00
|
|
|
|
|
|
|
for (i = 0; i < 4; i++)
|
2011-10-02 23:26:33 +04:00
|
|
|
{
|
|
|
|
origsize = context->OrgByteCount[i];
|
2012-03-06 14:52:28 +04:00
|
|
|
planesize = context->nsc_stream.PlaneByteCount[i];
|
2011-10-03 05:07:07 +04:00
|
|
|
|
2012-03-06 14:52:28 +04:00
|
|
|
if (planesize == 0)
|
|
|
|
memset(context->priv->plane_buf[i], 0xff, origsize);
|
|
|
|
else if (planesize < origsize)
|
|
|
|
nsc_rle_decode(rle, context->priv->plane_buf[i], origsize);
|
2011-10-02 23:26:33 +04:00
|
|
|
else
|
2012-03-06 14:52:28 +04:00
|
|
|
memcpy(context->priv->plane_buf[i], rle, origsize);
|
2011-10-03 05:07:07 +04:00
|
|
|
|
2012-03-06 14:52:28 +04:00
|
|
|
rle += planesize;
|
2011-10-02 23:26:33 +04:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2012-03-05 13:32:14 +04:00
|
|
|
static void nsc_stream_initialize(NSC_CONTEXT* context, STREAM* s)
|
2011-10-02 23:26:33 +04:00
|
|
|
{
|
|
|
|
int i;
|
2011-10-03 05:07:07 +04:00
|
|
|
|
|
|
|
for (i = 0; i < 4; i++)
|
2012-03-05 14:34:29 +04:00
|
|
|
stream_read_uint32(s, context->nsc_stream.PlaneByteCount[i]);
|
2011-10-03 05:07:07 +04:00
|
|
|
|
2012-03-05 14:34:29 +04:00
|
|
|
stream_read_uint8(s, context->nsc_stream.ColorLossLevel);
|
|
|
|
stream_read_uint8(s, context->nsc_stream.ChromaSubSamplingLevel);
|
2011-10-02 23:26:33 +04:00
|
|
|
stream_seek(s, 2);
|
2011-10-03 05:07:07 +04:00
|
|
|
|
2012-03-05 14:34:29 +04:00
|
|
|
context->nsc_stream.Planes = stream_get_tail(s);
|
2011-10-02 23:26:33 +04:00
|
|
|
}
|
|
|
|
|
2012-03-05 13:32:14 +04:00
|
|
|
static void nsc_context_initialize(NSC_CONTEXT* context, STREAM* s)
|
2011-10-02 23:26:33 +04:00
|
|
|
{
|
|
|
|
int i;
|
2012-03-05 13:32:14 +04:00
|
|
|
uint32 length;
|
2012-03-06 14:52:28 +04:00
|
|
|
uint32 tempWidth;
|
|
|
|
uint32 tempHeight;
|
2012-03-05 13:32:14 +04:00
|
|
|
|
2011-10-02 23:26:33 +04:00
|
|
|
nsc_stream_initialize(context, s);
|
2012-03-05 13:32:14 +04:00
|
|
|
length = context->width * context->height * 4;
|
|
|
|
if (context->bmpdata == NULL)
|
|
|
|
{
|
2012-03-07 07:31:10 +04:00
|
|
|
context->bmpdata = xzalloc(length + 16);
|
2012-03-05 13:32:14 +04:00
|
|
|
context->bmpdata_length = length;
|
|
|
|
}
|
|
|
|
else if (length > context->bmpdata_length)
|
|
|
|
{
|
2012-03-07 07:31:10 +04:00
|
|
|
context->bmpdata = xrealloc(context->bmpdata, length + 16);
|
2012-03-05 13:32:14 +04:00
|
|
|
context->bmpdata_length = length;
|
|
|
|
}
|
2011-10-03 05:07:07 +04:00
|
|
|
|
2012-03-06 14:52:28 +04:00
|
|
|
tempWidth = ROUND_UP_TO(context->width, 8);
|
|
|
|
tempHeight = ROUND_UP_TO(context->height, 2);
|
|
|
|
/* The maximum length a decoded plane can reach in all cases */
|
|
|
|
length = tempWidth * tempHeight;
|
|
|
|
if (length > context->priv->plane_buf_length)
|
|
|
|
{
|
|
|
|
for (i = 0; i < 4; i++)
|
|
|
|
context->priv->plane_buf[i] = (uint8*) xrealloc(context->priv->plane_buf[i], length);
|
|
|
|
context->priv->plane_buf_length = length;
|
|
|
|
}
|
|
|
|
|
2011-10-03 05:07:07 +04:00
|
|
|
for (i = 0; i < 4; i++)
|
2011-10-02 23:26:33 +04:00
|
|
|
context->OrgByteCount[i]=context->width * context->height;
|
2011-10-03 05:07:07 +04:00
|
|
|
|
2012-03-05 14:34:29 +04:00
|
|
|
if (context->nsc_stream.ChromaSubSamplingLevel > 0) /* [MS-RDPNSC] 2.2 */
|
2011-10-02 23:26:33 +04:00
|
|
|
{
|
|
|
|
context->OrgByteCount[0] = tempWidth * context->height;
|
2012-03-06 14:52:28 +04:00
|
|
|
context->OrgByteCount[1] = (tempWidth >> 1) * (tempHeight >> 1);
|
|
|
|
context->OrgByteCount[2] = context->OrgByteCount[1];
|
2011-10-02 23:26:33 +04:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2012-03-06 18:42:57 +04:00
|
|
|
static void nsc_profiler_print(NSC_CONTEXT* context)
|
|
|
|
{
|
|
|
|
PROFILER_PRINT_HEADER;
|
|
|
|
|
|
|
|
PROFILER_PRINT(context->priv->prof_nsc_rle_decompress_data);
|
|
|
|
PROFILER_PRINT(context->priv->prof_nsc_decode);
|
|
|
|
PROFILER_PRINT(context->priv->prof_nsc_rle_compress_data);
|
|
|
|
PROFILER_PRINT(context->priv->prof_nsc_encode);
|
|
|
|
|
|
|
|
PROFILER_PRINT_FOOTER;
|
|
|
|
}
|
|
|
|
|
2012-03-05 13:32:14 +04:00
|
|
|
void nsc_context_free(NSC_CONTEXT* context)
|
2011-10-02 23:26:33 +04:00
|
|
|
{
|
2012-03-06 14:52:28 +04:00
|
|
|
int i;
|
|
|
|
|
|
|
|
for (i = 0; i < 4; i++)
|
|
|
|
{
|
|
|
|
if (context->priv->plane_buf[i])
|
|
|
|
xfree(context->priv->plane_buf[i]);
|
|
|
|
}
|
2012-03-05 13:32:14 +04:00
|
|
|
if (context->bmpdata)
|
|
|
|
xfree(context->bmpdata);
|
2012-03-06 18:42:57 +04:00
|
|
|
|
|
|
|
nsc_profiler_print(context);
|
|
|
|
PROFILER_FREE(context->priv->prof_nsc_rle_decompress_data);
|
|
|
|
PROFILER_FREE(context->priv->prof_nsc_decode);
|
|
|
|
PROFILER_FREE(context->priv->prof_nsc_rle_compress_data);
|
|
|
|
PROFILER_FREE(context->priv->prof_nsc_encode);
|
|
|
|
|
2012-03-06 14:52:28 +04:00
|
|
|
xfree(context->priv);
|
2012-03-05 13:32:14 +04:00
|
|
|
xfree(context);
|
2011-10-02 23:26:33 +04:00
|
|
|
}
|
|
|
|
|
|
|
|
NSC_CONTEXT* nsc_context_new(void)
|
|
|
|
{
|
|
|
|
NSC_CONTEXT* nsc_context;
|
2012-03-05 14:34:29 +04:00
|
|
|
|
2011-10-02 23:26:33 +04:00
|
|
|
nsc_context = xnew(NSC_CONTEXT);
|
2012-03-06 14:52:28 +04:00
|
|
|
nsc_context->priv = xnew(NSC_CONTEXT_PRIV);
|
2012-03-05 14:34:29 +04:00
|
|
|
|
2012-03-06 16:06:08 +04:00
|
|
|
nsc_context->decode = nsc_decode;
|
2012-03-16 08:04:38 +04:00
|
|
|
nsc_context->encode = nsc_encode;
|
2012-03-06 16:06:08 +04:00
|
|
|
|
2012-03-06 18:42:57 +04:00
|
|
|
PROFILER_CREATE(nsc_context->priv->prof_nsc_rle_decompress_data, "nsc_rle_decompress_data");
|
|
|
|
PROFILER_CREATE(nsc_context->priv->prof_nsc_decode, "nsc_decode");
|
|
|
|
PROFILER_CREATE(nsc_context->priv->prof_nsc_rle_compress_data, "nsc_rle_compress_data");
|
|
|
|
PROFILER_CREATE(nsc_context->priv->prof_nsc_encode, "nsc_encode");
|
|
|
|
|
2012-03-16 08:04:38 +04:00
|
|
|
/* Default encoding parameters */
|
|
|
|
nsc_context->nsc_stream.ColorLossLevel = 3;
|
|
|
|
nsc_context->nsc_stream.ChromaSubSamplingLevel = 1;
|
|
|
|
|
2011-10-02 23:26:33 +04:00
|
|
|
return nsc_context;
|
|
|
|
}
|
|
|
|
|
2012-03-06 16:06:08 +04:00
|
|
|
void nsc_context_set_cpu_opt(NSC_CONTEXT* context, uint32 cpu_opt)
|
|
|
|
{
|
|
|
|
if (cpu_opt)
|
|
|
|
NSC_INIT_SIMD(context);
|
|
|
|
}
|
|
|
|
|
2012-03-15 12:55:29 +04:00
|
|
|
void nsc_context_set_pixel_format(NSC_CONTEXT* context, RDP_PIXEL_FORMAT pixel_format)
|
|
|
|
{
|
|
|
|
context->pixel_format = pixel_format;
|
|
|
|
switch (pixel_format)
|
|
|
|
{
|
|
|
|
case RDP_PIXEL_FORMAT_B8G8R8A8:
|
|
|
|
case RDP_PIXEL_FORMAT_R8G8B8A8:
|
2012-03-16 08:04:38 +04:00
|
|
|
context->bpp = 32;
|
2012-03-15 12:55:29 +04:00
|
|
|
break;
|
|
|
|
case RDP_PIXEL_FORMAT_B8G8R8:
|
|
|
|
case RDP_PIXEL_FORMAT_R8G8B8:
|
2012-03-16 08:04:38 +04:00
|
|
|
context->bpp = 24;
|
2012-03-15 12:55:29 +04:00
|
|
|
break;
|
|
|
|
case RDP_PIXEL_FORMAT_B5G6R5_LE:
|
|
|
|
case RDP_PIXEL_FORMAT_R5G6B5_LE:
|
2012-03-16 08:04:38 +04:00
|
|
|
context->bpp = 16;
|
2012-03-15 12:55:29 +04:00
|
|
|
break;
|
|
|
|
case RDP_PIXEL_FORMAT_P4_PLANER:
|
2012-03-16 08:04:38 +04:00
|
|
|
context->bpp = 4;
|
2012-03-15 12:55:29 +04:00
|
|
|
break;
|
|
|
|
case RDP_PIXEL_FORMAT_P8:
|
2012-03-16 08:04:38 +04:00
|
|
|
context->bpp = 8;
|
2012-03-15 12:55:29 +04:00
|
|
|
break;
|
|
|
|
default:
|
2012-03-16 08:04:38 +04:00
|
|
|
context->bpp = 0;
|
2012-03-15 12:55:29 +04:00
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2012-03-05 13:32:14 +04:00
|
|
|
void nsc_process_message(NSC_CONTEXT* context, uint16 bpp,
|
|
|
|
uint16 width, uint16 height, uint8* data, uint32 length)
|
2011-10-02 23:26:33 +04:00
|
|
|
{
|
|
|
|
STREAM* s;
|
2012-03-05 13:32:14 +04:00
|
|
|
|
2011-10-02 23:26:33 +04:00
|
|
|
s = stream_new(0);
|
|
|
|
stream_attach(s, data, length);
|
2012-03-05 13:32:14 +04:00
|
|
|
context->bpp = bpp;
|
|
|
|
context->width = width;
|
|
|
|
context->height = height;
|
2011-10-02 23:26:33 +04:00
|
|
|
nsc_context_initialize(context, s);
|
2012-03-06 14:52:28 +04:00
|
|
|
stream_detach(s);
|
|
|
|
stream_free(s);
|
2011-10-02 23:26:33 +04:00
|
|
|
|
|
|
|
/* RLE decode */
|
2012-03-06 18:42:57 +04:00
|
|
|
PROFILER_ENTER(context->priv->prof_nsc_rle_decompress_data);
|
2011-10-02 23:26:33 +04:00
|
|
|
nsc_rle_decompress_data(context);
|
2012-03-06 18:42:57 +04:00
|
|
|
PROFILER_EXIT(context->priv->prof_nsc_rle_decompress_data);
|
2011-10-02 23:26:33 +04:00
|
|
|
|
2012-03-06 14:52:28 +04:00
|
|
|
/* Colorloss recover, Chroma supersample and AYCoCg to ARGB Conversion in one step */
|
2012-03-06 18:42:57 +04:00
|
|
|
PROFILER_ENTER(context->priv->prof_nsc_decode);
|
2012-03-06 16:06:08 +04:00
|
|
|
context->decode(context);
|
2012-03-06 18:42:57 +04:00
|
|
|
PROFILER_EXIT(context->priv->prof_nsc_decode);
|
2011-10-02 23:26:33 +04:00
|
|
|
}
|