mpv/libmpcodecs/vd_mtga.c

249 lines
5.7 KiB
C

/* author: Tilman Sauerbeck <tsauerbeck@users.sourceforge.net>
* based on: XreaL's x_r_img_tga.* (http://www.sourceforge.net/projects/xreal/)
* libtarga.*
* xli's tga.*
*/
#include <stdio.h>
#include <stdlib.h>
#include "config.h"
#include "mp_msg.h"
#include "bswap.h"
#include "postproc/rgb2rgb.h"
#include "libvo/fastmemcpy.h"
#include "vd_internal.h"
static vd_info_t info =
{
"TGA Images decoder",
"mtga",
"Tilman Sauerbeck",
"Tilman Sauerbeck",
"only 24bpp and 32bpp RGB targa files support so far"
};
LIBVD_EXTERN(mtga)
typedef enum
{
TGA_NO_DATA,
TGA_UNCOMP_PALETTED,
TGA_UNCOMP_TRUECOLOR,
TGA_UNCOMP_GRAYSCALE,
TGA_RLE_PALETTED = 9,
TGA_RLE_TRUECOLOR,
TGA_RLE_GRAYSCALE
} TGAImageType;
typedef struct
{
unsigned char id_len;
unsigned short img_type;
unsigned short width;
unsigned short height;
unsigned char bpp;
unsigned char origin; /* 0 = lower left, 1 = upper left */
unsigned short start_row;
unsigned short increment;
} TGAInfo;
static unsigned int out_fmt = 0;
static int last_w = -1;
static int last_h = -1;
static int last_c = -1;
/* to set/get/query special features/parameters */
static int control(sh_video_t *sh, int cmd, void *arg, ...)
{
TGAInfo *info = (TGAInfo *) sh->context;
switch (cmd)
{
case VDCTRL_QUERY_FORMAT:
if ((*((int *) arg) == IMGFMT_BGR32 && info->bpp == 32) || (*((int *) arg) == IMGFMT_BGR24 && info->bpp == 24))
return CONTROL_TRUE;
else
return CONTROL_FALSE;
break;
default:
return CONTROL_UNKNOWN;
break;
}
}
/* init driver */
static int init(sh_video_t *sh)
{
sh->context = (TGAInfo *) calloc(1, sizeof(TGAInfo));
last_w = -1;
return 1;
}
/* uninit driver */
static void uninit(sh_video_t *sh)
{
TGAInfo *info = sh->context;
free(info);
return;
}
/* decode a runlength-encoded tga */
static void decode_rle_tga(TGAInfo *info, unsigned char *data, mp_image_t **mpi)
{
short i, num_bytes = info->bpp / 8;
unsigned char repetitions, packet_header, *final;
unsigned short row, col;
/* see line 207 to see why this loop is set up like this */
for (row = info->start_row; (!info->origin && row) || (info->origin && row < info->height); row += info->increment)
{
final = (*mpi)->planes[0] + (*mpi)->stride[0] * row;
for (col = 0; col < info->width; col += repetitions)
{
packet_header = *data++;
repetitions = 1 + (packet_header & 0x7f);
if (packet_header & 0x80) /* runlength encoded packet */
{
memcpy(final, data, num_bytes);
for (i = num_bytes; i < repetitions; i *= 2)
memcpy(&final[i], final, i);
for (; i < repetitions * num_bytes; i += 3)
memcpy(&final[i], final, num_bytes);
}
else /* raw packet */
memcpy(final, data, repetitions * num_bytes);
data += (packet_header & 0x80) ? num_bytes : repetitions * num_bytes;
final += repetitions * num_bytes;
}
}
return;
}
static void decode_uncompressed_tga(TGAInfo *info, unsigned char *data, mp_image_t **mpi)
{
unsigned short row;
unsigned char *final;
short num_bytes = info->bpp / 8;
/* see line 207 to see why this loop is set up like this */
for (row = info->start_row; (!info->origin && row) || (info->origin && row < info->height); row += info->increment)
{
final = (*mpi)->planes[0] + (*mpi)->stride[0] * row;
memcpy(final, data, info->width * num_bytes);
data += info->width * num_bytes;
}
return;
}
static short read_tga_header(unsigned char *buf, TGAInfo **info)
{
(*info)->id_len = buf[0];
(*info)->img_type = buf[2];
/* targa data is always stored in little endian byte order */
(*info)->width = le2me_16(*(unsigned short *) &buf[12]);
(*info)->height = le2me_16(*(unsigned short *) &buf[14]);
(*info)->bpp = buf[16];
(*info)->origin = (buf[17] & 0x20) >> 5;
/* FIXME check for valid targa data */
return 0;
}
/* decode a frame */
static mp_image_t *decode(sh_video_t *sh, void *raw, int len, int flags)
{
TGAInfo *info = sh->context;
unsigned char *data = raw;
mp_image_t *mpi;
if (len <= 0)
return NULL; /* skip frame */
read_tga_header(data, &info); /* read information about the file */
if (info->bpp == 24)
out_fmt = IMGFMT_BGR24;
else if (info->bpp == 32)
out_fmt = IMGFMT_BGR32;
else
{
mp_msg(MSGT_DECVIDEO, MSGL_INFO, "Unsupported TGA type!\n");
return NULL;
}
if (info->img_type != TGA_UNCOMP_TRUECOLOR && info->img_type != TGA_RLE_TRUECOLOR) /* not a true color image */
{
mp_msg(MSGT_DECVIDEO, MSGL_INFO, "Unsupported TGA type: %i!\n", info->img_type);
return NULL;
}
/* if img.origin is 0, we decode from bottom to top. if it's 1, we decode from top to bottom */
info->start_row = (info->origin) ? 0 : info->height - 1;
info->increment = (info->origin) ? 1 : -1;
/* set data to the beginning of the image data */
data += 18 + info->id_len;
/* (re)init libvo if image parameters changed (width/height/colorspace) */
if (last_w != info->width || last_h != info->height || last_c != out_fmt)
{
last_w = info->width;
last_h = info->height;
last_c = out_fmt;
if (!out_fmt || !mpcodecs_config_vo(sh, info->width, info->height, out_fmt))
return NULL;
}
if (!(mpi = mpcodecs_get_image(sh, MP_IMGTYPE_TEMP, MP_IMGFLAG_ACCEPT_STRIDE, info->width, info->height)))
return NULL;
/* finally decode the image */
if (info->img_type == TGA_UNCOMP_TRUECOLOR)
decode_uncompressed_tga(info, data, &mpi);
else if (info->img_type == TGA_RLE_TRUECOLOR)
decode_rle_tga(info, data, &mpi);
else
mpi = NULL;
return mpi;
}