mirror of https://github.com/mpv-player/mpv
436 lines
12 KiB
C
436 lines
12 KiB
C
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
|
|
#include "config.h"
|
|
#include "mp_msg.h"
|
|
|
|
#include "vd_internal.h"
|
|
|
|
static vd_info_t info = {
|
|
"Microsoft RLE decoder",
|
|
"msrle",
|
|
VFM_MSRLE,
|
|
"Mike Melanson",
|
|
"Mike Melanson",
|
|
"native codec"
|
|
};
|
|
|
|
LIBVD_EXTERN(msrle)
|
|
|
|
// to set/get/query special features/parameters
|
|
static int control(sh_video_t *sh,int cmd,void* arg,...){
|
|
return CONTROL_UNKNOWN;
|
|
}
|
|
|
|
// init driver
|
|
static int init(sh_video_t *sh){
|
|
unsigned char *palette_map = NULL;
|
|
unsigned char *orig_map = (unsigned char *)sh->bih+40;
|
|
int i;
|
|
unsigned short color;
|
|
unsigned char r, g, b;
|
|
int bits_per_pixel = sh->codec->outfmt[sh->outfmtidx] & 255;
|
|
|
|
// convert the palette for the requested output format
|
|
switch (bits_per_pixel)
|
|
{
|
|
case 15:
|
|
case 16:
|
|
if ((palette_map =
|
|
(unsigned char *)malloc(sh->bih->biClrUsed * 2)) == NULL)
|
|
return 0;
|
|
|
|
for (i = 0; i < sh->bih->biClrUsed; i++)
|
|
{
|
|
r = orig_map[i * 4 + 2];
|
|
g = orig_map[i * 4 + 1];
|
|
b = orig_map[i * 4 + 0];
|
|
if (bits_per_pixel == 15)
|
|
color = ((r>>3)<<10) | ((g>>3)<<5) | ((b>>3));
|
|
else
|
|
color = ((r>>3)<<11) | ((g>>2)<<5) | ((b>>3));
|
|
palette_map[i * 2 + 1] = color >> 8;
|
|
palette_map[i * 2 + 0] = color & 0xFF;
|
|
}
|
|
break;
|
|
|
|
case 24:
|
|
case 32:
|
|
if ((palette_map =
|
|
(unsigned char *)malloc(sh->bih->biClrUsed * 4)) == NULL)
|
|
return 0;
|
|
memcpy(palette_map, orig_map, sh->bih->biClrUsed * 4);
|
|
break;
|
|
}
|
|
|
|
sh->context = palette_map;
|
|
|
|
return mpcodecs_config_vo(sh,sh->disp_w,sh->disp_h,IMGFMT_BGR24);
|
|
}
|
|
|
|
// uninit driver
|
|
static void uninit(sh_video_t *sh){
|
|
unsigned char *palette_map = (unsigned char *)sh->context;
|
|
|
|
free(palette_map);
|
|
}
|
|
|
|
#define FETCH_NEXT_STREAM_BYTE() \
|
|
if (stream_ptr >= encoded_size) \
|
|
{ \
|
|
mp_msg(MSGT_DECVIDEO, MSGL_WARN, \
|
|
"MS RLE: stream ptr just went out of bounds (1)\n"); \
|
|
return; \
|
|
} \
|
|
stream_byte = encoded[stream_ptr++];
|
|
|
|
void decode_msrle4(
|
|
unsigned char *encoded,
|
|
int encoded_size,
|
|
unsigned char *decoded,
|
|
int width,
|
|
int height,
|
|
unsigned char *palette_map,
|
|
int bits_per_pixel)
|
|
{
|
|
int bytes_per_pixel = (bits_per_pixel + 1) / 8;
|
|
unsigned char r1, g1, b1; // for 24/32 bpp
|
|
unsigned char r2, g2, b2;
|
|
unsigned char color_hi1, color_lo1; // for 15/16 bpp
|
|
unsigned char color_hi2, color_lo2;
|
|
int stream_ptr = 0;
|
|
unsigned char rle_code;
|
|
unsigned char extra_byte;
|
|
unsigned char stream_byte;
|
|
int frame_size = width * height * bytes_per_pixel;
|
|
int pixel_ptr = 0;
|
|
int row_dec = width * bytes_per_pixel;
|
|
int row_ptr = (height - 1) * row_dec;
|
|
int i;
|
|
|
|
r1 = r2 = g1 = g2 = b1 = b2 =
|
|
color_hi1 = color_hi2 = color_lo1 = color_lo2 = 0;
|
|
while (row_ptr >= 0)
|
|
{
|
|
FETCH_NEXT_STREAM_BYTE();
|
|
rle_code = stream_byte;
|
|
if (rle_code == 0)
|
|
{
|
|
// fetch the next byte to see how to handle escape code
|
|
FETCH_NEXT_STREAM_BYTE();
|
|
if (stream_byte == 0)
|
|
{
|
|
// line is done, goto the next one
|
|
row_ptr -= row_dec;
|
|
pixel_ptr = 0;
|
|
}
|
|
else if (stream_byte == 1)
|
|
// decode is done
|
|
return;
|
|
else if (stream_byte == 2)
|
|
{
|
|
// reposition frame decode coordinates
|
|
FETCH_NEXT_STREAM_BYTE();
|
|
pixel_ptr += stream_byte * bytes_per_pixel;
|
|
FETCH_NEXT_STREAM_BYTE();
|
|
row_ptr -= stream_byte * row_dec;
|
|
}
|
|
else
|
|
{
|
|
// copy pixels from encoded stream
|
|
rle_code = stream_byte;
|
|
rle_code /= 2;
|
|
extra_byte = rle_code & 0x01;
|
|
if ((row_ptr + pixel_ptr + rle_code * bytes_per_pixel > frame_size) ||
|
|
(row_ptr < 0))
|
|
{
|
|
mp_msg(MSGT_DECVIDEO, MSGL_WARN,
|
|
"MS RLE: frame ptr just went out of bounds (1)\n");
|
|
return;
|
|
}
|
|
|
|
for (i = 0; i < rle_code; i++)
|
|
{
|
|
if (pixel_ptr >= row_dec)
|
|
break;
|
|
if (bytes_per_pixel == 2)
|
|
{
|
|
color_hi1 = palette_map[(encoded[stream_ptr + i] >> 4) * 2 + 0];
|
|
color_lo1 = palette_map[(encoded[stream_ptr + i] >> 4) * 2 + 1];
|
|
decoded[row_ptr + pixel_ptr + 0] = color_hi1;
|
|
decoded[row_ptr + pixel_ptr + 1] = color_lo1;
|
|
}
|
|
else
|
|
{
|
|
r1 = palette_map[(encoded[stream_ptr + i] >> 4) * 4 + 2];
|
|
g1 = palette_map[(encoded[stream_ptr + i] >> 4) * 4 + 1];
|
|
b1 = palette_map[(encoded[stream_ptr + i] >> 4) * 4 + 0];
|
|
decoded[row_ptr + pixel_ptr + 0] = b1;
|
|
decoded[row_ptr + pixel_ptr + 1] = g1;
|
|
decoded[row_ptr + pixel_ptr + 2] = r1;
|
|
}
|
|
pixel_ptr += bytes_per_pixel;
|
|
|
|
if (pixel_ptr >= row_dec)
|
|
break;
|
|
if (bytes_per_pixel == 2)
|
|
{
|
|
color_hi1 = palette_map[(encoded[stream_ptr + i] & 0x0F) * 2 + 0];
|
|
color_lo1 = palette_map[(encoded[stream_ptr + i] & 0x0F) * 2 + 1];
|
|
decoded[row_ptr + pixel_ptr + 0] = color_hi1;
|
|
decoded[row_ptr + pixel_ptr + 1] = color_lo1;
|
|
}
|
|
else
|
|
{
|
|
r1 = palette_map[(encoded[stream_ptr + i] & 0x0F) * 4 + 2];
|
|
g1 = palette_map[(encoded[stream_ptr + i] & 0x0F) * 4 + 1];
|
|
b1 = palette_map[(encoded[stream_ptr + i] & 0x0F) * 4 + 0];
|
|
decoded[row_ptr + pixel_ptr + 0] = b1;
|
|
decoded[row_ptr + pixel_ptr + 1] = g1;
|
|
decoded[row_ptr + pixel_ptr + 2] = r1;
|
|
}
|
|
pixel_ptr += bytes_per_pixel;
|
|
}
|
|
stream_ptr += rle_code;
|
|
|
|
// if the RLE code is odd, skip a byte in the stream
|
|
if (extra_byte)
|
|
stream_ptr++;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
// decode a run of data
|
|
FETCH_NEXT_STREAM_BYTE();
|
|
if (bytes_per_pixel == 2)
|
|
{
|
|
color_hi1 = palette_map[(stream_byte >> 4) * 2 + 0];
|
|
color_lo1 = palette_map[(stream_byte >> 4) * 2 + 1];
|
|
color_hi2 = palette_map[(stream_byte & 0x0F) * 2 + 0];
|
|
color_lo2 = palette_map[(stream_byte & 0x0F) * 2 + 1];
|
|
}
|
|
else
|
|
{
|
|
r1 = palette_map[(stream_byte >> 4) * 4 + 2];
|
|
g1 = palette_map[(stream_byte >> 4) * 4 + 1];
|
|
b1 = palette_map[(stream_byte >> 4) * 4 + 0];
|
|
r2 = palette_map[(stream_byte & 0x0F) * 4 + 2];
|
|
g2 = palette_map[(stream_byte & 0x0F) * 4 + 1];
|
|
b2 = palette_map[(stream_byte & 0x0F) * 4 + 0];
|
|
}
|
|
for (i = 0; i < rle_code; i++)
|
|
{
|
|
if (pixel_ptr >= row_dec)
|
|
break;
|
|
|
|
if ((i & 1) == 0)
|
|
{
|
|
if (bytes_per_pixel == 2)
|
|
{
|
|
decoded[row_ptr + pixel_ptr + 0] = color_hi1;
|
|
decoded[row_ptr + pixel_ptr + 1] = color_lo1;
|
|
}
|
|
else
|
|
{
|
|
decoded[row_ptr + pixel_ptr + 0] = b1;
|
|
decoded[row_ptr + pixel_ptr + 1] = g1;
|
|
decoded[row_ptr + pixel_ptr + 2] = r1;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
if (bytes_per_pixel == 2)
|
|
{
|
|
decoded[row_ptr + pixel_ptr + 0] = color_hi2;
|
|
decoded[row_ptr + pixel_ptr + 1] = color_lo2;
|
|
}
|
|
else
|
|
{
|
|
decoded[row_ptr + pixel_ptr + 0] = b2;
|
|
decoded[row_ptr + pixel_ptr + 1] = g2;
|
|
decoded[row_ptr + pixel_ptr + 2] = r2;
|
|
}
|
|
}
|
|
pixel_ptr += bytes_per_pixel;
|
|
}
|
|
}
|
|
}
|
|
|
|
// one last sanity check on the way out
|
|
if (stream_ptr < encoded_size)
|
|
mp_msg(MSGT_DECVIDEO, MSGL_WARN,
|
|
"MS RLE: ended frame decode with bytes left over (%d < %d)\n",
|
|
stream_ptr, encoded_size);
|
|
}
|
|
|
|
void decode_msrle8(
|
|
unsigned char *encoded,
|
|
int encoded_size,
|
|
unsigned char *decoded,
|
|
int width,
|
|
int height,
|
|
unsigned char *palette_map,
|
|
int bits_per_pixel)
|
|
{
|
|
int bytes_per_pixel = (bits_per_pixel + 1) / 8;
|
|
unsigned char r, g, b; // for 24/32 bpp
|
|
unsigned char color_hi, color_lo; // for 15/16 bpp
|
|
int stream_ptr = 0;
|
|
unsigned char rle_code;
|
|
unsigned char extra_byte;
|
|
unsigned char stream_byte;
|
|
int frame_size = width * height * bytes_per_pixel;
|
|
int pixel_ptr = 0;
|
|
int row_dec = width * bytes_per_pixel;
|
|
int row_ptr = (height - 1) * row_dec;
|
|
|
|
r = g = b = color_hi = color_lo = 0;
|
|
while (row_ptr >= 0)
|
|
{
|
|
FETCH_NEXT_STREAM_BYTE();
|
|
rle_code = stream_byte;
|
|
if (rle_code == 0)
|
|
{
|
|
// fetch the next byte to see how to handle escape code
|
|
FETCH_NEXT_STREAM_BYTE();
|
|
if (stream_byte == 0)
|
|
{
|
|
// line is done, goto the next one
|
|
row_ptr -= row_dec;
|
|
pixel_ptr = 0;
|
|
}
|
|
else if (stream_byte == 1)
|
|
// decode is done
|
|
return;
|
|
else if (stream_byte == 2)
|
|
{
|
|
// reposition frame decode coordinates
|
|
FETCH_NEXT_STREAM_BYTE();
|
|
pixel_ptr += stream_byte * bytes_per_pixel;
|
|
FETCH_NEXT_STREAM_BYTE();
|
|
row_ptr -= stream_byte * row_dec;
|
|
}
|
|
else
|
|
{
|
|
// copy pixels from encoded stream
|
|
if ((row_ptr + pixel_ptr + stream_byte * bytes_per_pixel > frame_size) ||
|
|
(row_ptr < 0))
|
|
{
|
|
mp_msg(MSGT_DECVIDEO, MSGL_WARN,
|
|
"MS RLE: frame ptr just went out of bounds (1)\n");
|
|
return;
|
|
}
|
|
|
|
rle_code = stream_byte;
|
|
extra_byte = stream_byte & 0x01;
|
|
if (stream_ptr + rle_code + extra_byte > encoded_size)
|
|
{
|
|
mp_msg(MSGT_DECVIDEO, MSGL_WARN,
|
|
"MS RLE: stream ptr just went out of bounds (2)\n");
|
|
return;
|
|
}
|
|
|
|
while (rle_code--)
|
|
{
|
|
FETCH_NEXT_STREAM_BYTE();
|
|
if (bytes_per_pixel == 2)
|
|
{
|
|
color_hi = palette_map[stream_byte * 2 + 0];
|
|
color_lo = palette_map[stream_byte * 2 + 1];
|
|
decoded[row_ptr + pixel_ptr + 0] = color_hi;
|
|
decoded[row_ptr + pixel_ptr + 1] = color_lo;
|
|
}
|
|
else
|
|
{
|
|
r = palette_map[stream_byte * 4 + 2];
|
|
g = palette_map[stream_byte * 4 + 1];
|
|
b = palette_map[stream_byte * 4 + 0];
|
|
decoded[row_ptr + pixel_ptr + 0] = b;
|
|
decoded[row_ptr + pixel_ptr + 1] = g;
|
|
decoded[row_ptr + pixel_ptr + 2] = r;
|
|
}
|
|
pixel_ptr += bytes_per_pixel;
|
|
}
|
|
|
|
// if the RLE code is odd, skip a byte in the stream
|
|
if (extra_byte)
|
|
stream_ptr++;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
// decode a run of data
|
|
if ((row_ptr + pixel_ptr + stream_byte * bytes_per_pixel > frame_size) ||
|
|
(row_ptr < 0))
|
|
{
|
|
mp_msg(MSGT_DECVIDEO, MSGL_WARN,
|
|
"MS RLE: frame ptr just went out of bounds (2)\n");
|
|
return;
|
|
}
|
|
|
|
FETCH_NEXT_STREAM_BYTE();
|
|
|
|
if (bytes_per_pixel == 2)
|
|
{
|
|
color_hi = palette_map[stream_byte * 2 + 0];
|
|
color_lo = palette_map[stream_byte * 2 + 1];
|
|
}
|
|
else
|
|
{
|
|
r = palette_map[stream_byte * 4 + 2];
|
|
g = palette_map[stream_byte * 4 + 1];
|
|
b = palette_map[stream_byte * 4 + 0];
|
|
}
|
|
while(rle_code--)
|
|
{
|
|
if (bytes_per_pixel == 2)
|
|
{
|
|
decoded[row_ptr + pixel_ptr + 0] = color_hi;
|
|
decoded[row_ptr + pixel_ptr + 1] = color_lo;
|
|
}
|
|
else
|
|
{
|
|
decoded[row_ptr + pixel_ptr + 0] = b;
|
|
decoded[row_ptr + pixel_ptr + 1] = g;
|
|
decoded[row_ptr + pixel_ptr + 2] = r;
|
|
}
|
|
pixel_ptr += bytes_per_pixel;
|
|
}
|
|
}
|
|
}
|
|
|
|
// one last sanity check on the way out
|
|
if (stream_ptr < encoded_size)
|
|
mp_msg(MSGT_DECVIDEO, MSGL_WARN,
|
|
"MS RLE: ended frame decode with bytes left over (%d < %d)\n",
|
|
stream_ptr, encoded_size);
|
|
}
|
|
|
|
// decode a frame
|
|
static mp_image_t* decode(sh_video_t *sh,void* data,int len,int flags){
|
|
mp_image_t* mpi;
|
|
if(len<=0) return NULL; // skipped frame
|
|
|
|
mpi=mpcodecs_get_image(sh, MP_IMGTYPE_STATIC, MP_IMGFLAG_PRESERVE,
|
|
sh->disp_w, sh->disp_h);
|
|
if(!mpi) return NULL;
|
|
|
|
if (sh->format == 1)
|
|
decode_msrle8(
|
|
data,len, mpi->planes[0],
|
|
sh->disp_w, sh->disp_h,
|
|
(unsigned char *)sh->context,
|
|
mpi->imgfmt & 255);
|
|
else if (sh->format == 2)
|
|
decode_msrle4(
|
|
data,len, mpi->planes[0],
|
|
sh->disp_w, sh->disp_h,
|
|
(unsigned char *)sh->context,
|
|
mpi->imgfmt & 255);
|
|
else
|
|
mp_msg(MSGT_DECVIDEO, MSGL_WARN,
|
|
"MS RLE: Don't know how to decode format %08X", sh->format);
|
|
|
|
return mpi;
|
|
}
|