diff options
Diffstat (limited to 'libmpcodecs/native/roqav.c')
-rw-r--r-- | libmpcodecs/native/roqav.c | 677 |
1 files changed, 0 insertions, 677 deletions
diff --git a/libmpcodecs/native/roqav.c b/libmpcodecs/native/roqav.c deleted file mode 100644 index ffb0bb40df..0000000000 --- a/libmpcodecs/native/roqav.c +++ /dev/null @@ -1,677 +0,0 @@ -/* - RoQ A/V decoder for the MPlayer program - by Mike Melanson - based on Dr. Tim Ferguson's RoQ document and accompanying source - code found at: - http://www.csse.monash.edu.au/~timf/videocodec.html -*/ - -#include <stdio.h> -#include <stdlib.h> -#include <string.h> -#include "config.h" -#include "bswap.h" -#include "mp_msg.h" -#include "mp_image.h" - -#define LE_16(x) (le2me_16(*(unsigned short *)(x))) -#define LE_32(x) (le2me_32(*(unsigned int *)(x))) - -#define CLAMP_S16(x) if (x < -32768) x = -32768; \ - else if (x > 32767) x = 32767; -#define SE_16BIT(x) if (x & 0x8000) x -= 0x10000; - -// RoQ chunk types -#define RoQ_INFO 0x1001 -#define RoQ_QUAD_CODEBOOK 0x1002 -#define RoQ_QUAD_VQ 0x1011 -#define RoQ_SOUND_MONO 0x1020 -#define RoQ_SOUND_STEREO 0x1021 - -#define MAX_ROQ_CODEBOOK_SIZE 256 - -// codebook entry for 2x2 vector -typedef struct -{ - // upper and lower luminance value pairs of 2x2 vector: [y0 y1], [y2 y3] - unsigned short v2_y_u; - unsigned short v2_y_l; - - // chrominance components - unsigned char u, v; - - // these variables are for rendering a doublesized 8x8 block; e.g.: - // v2_y_rows12 = [y0 y0 y1 y1] - // v2_y_rows34 = [y2 y2 y3 y3] - unsigned long v2d_y_rows_12; - unsigned long v2d_y_rows_34; - // ex: v2_u_row1 = [u u] - // v2_v_row2 = [v v] - unsigned short v2d_u_rows_12; - unsigned short v2d_v_rows_12; - - // maintain separate bytes for the luminance values as well - unsigned char y0, y1, y2, y3; -} roq_v2_codebook; - -// codebook entry for 4x4 vector -typedef struct -{ - unsigned char v2_index[4]; -} roq_v4_codebook; - -typedef struct -{ - roq_v2_codebook v2[MAX_ROQ_CODEBOOK_SIZE]; - roq_v4_codebook v4[MAX_ROQ_CODEBOOK_SIZE]; - mp_image_t *prev_frame; - uint32_t numframe; -} roqvideo_info; - - -// This function fills in the missing information for a v2 vector after -// loading the Y, U and V values. -inline void prep_v2(roq_v2_codebook *v2) -{ - v2->v2_y_u = be2me_16((v2->y0 << 8) | v2->y1); - v2->v2_y_l = be2me_16((v2->y2 << 8) | v2->y3); - - v2->v2d_y_rows_12 = be2me_32((v2->y0 << 24) | (v2->y0 << 16) | - (v2->y1 << 8) | v2->y1); - v2->v2d_y_rows_34 = be2me_32((v2->y2 << 24) | (v2->y2 << 16) | - (v2->y3 << 8) | v2->y3); - - // no reason to swap these for endianness since they're the same bytes - v2->v2d_u_rows_12 = (v2->u << 8) | v2->u; - v2->v2d_v_rows_12 = (v2->v << 8) | v2->v; -} - -inline void paint_v2double_block( - unsigned char *y_plane, - unsigned char *u_plane, - unsigned char *v_plane, - roq_v2_codebook *v2, - unsigned int y_stride, - unsigned int u_stride, - unsigned int v_stride) -{ - // render the luminance components - *(unsigned int *)y_plane = v2->v2d_y_rows_12; - y_plane += y_stride; - *(unsigned int *)y_plane = v2->v2d_y_rows_12; - y_plane += y_stride; - *(unsigned int *)y_plane = v2->v2d_y_rows_34; - y_plane += y_stride; - *(unsigned int *)y_plane = v2->v2d_y_rows_34; - - // render the color planes - *(unsigned short *)u_plane = v2->v2d_u_rows_12; - u_plane += u_stride; - *(unsigned short *)u_plane = v2->v2d_u_rows_12; - - *(unsigned short *)v_plane = v2->v2d_v_rows_12; - v_plane += v_stride; - *(unsigned short *)v_plane = v2->v2d_v_rows_12; -} - -inline void paint_v4_block( - unsigned char *y_plane, - unsigned char *u_plane, - unsigned char *v_plane, - unsigned int y_stride, - unsigned int u_stride, - unsigned int v_stride, - roq_v2_codebook *v2_a, - roq_v2_codebook *v2_b, - roq_v2_codebook *v2_c, - roq_v2_codebook *v2_d) -{ - // render luminance components - ((unsigned short *)y_plane)[0] = v2_a->v2_y_u; - ((unsigned short *)y_plane)[1] = v2_b->v2_y_u; - - y_plane += y_stride; - ((unsigned short *)y_plane)[0] = v2_a->v2_y_l; - ((unsigned short *)y_plane)[1] = v2_b->v2_y_l; - - y_plane += y_stride; - ((unsigned short *)y_plane)[0] = v2_c->v2_y_u; - ((unsigned short *)y_plane)[1] = v2_d->v2_y_u; - - y_plane += y_stride; - ((unsigned short *)y_plane)[0] = v2_c->v2_y_l; - ((unsigned short *)y_plane)[1] = v2_d->v2_y_l; - - // render the color planes - u_plane[0] = v2_a->u; - u_plane[1] = v2_b->u; - u_plane += u_stride; - u_plane[0] = v2_c->u; - u_plane[1] = v2_d->u; - - v_plane[0] = v2_a->v; - v_plane[1] = v2_b->v; - v_plane += v_stride; - v_plane[0] = v2_c->v; - v_plane[1] = v2_d->v; -} - -// This function copies the 4x4 block from the prev_*_planes to the -// current *_planes. -inline void copy_4x4_block( - unsigned char *y_plane, - unsigned char *u_plane, - unsigned char *v_plane, - unsigned char *prev_y_plane, - unsigned char *prev_u_plane, - unsigned char *prev_v_plane, - unsigned int y_stride, - unsigned int u_stride, - unsigned int v_stride) -{ - int i; - - // copy over the luminance components (4 rows, 1 uint each) - for (i = 0; i < 4; i++) - { - *(unsigned int *)y_plane = *(unsigned int *)prev_y_plane; - y_plane += y_stride; - prev_y_plane += y_stride; - } - - // copy the chrominance values - for (i = 0; i < 2; i++) - { - *(unsigned short*)u_plane = *(unsigned short*)prev_u_plane; - u_plane += u_stride; - prev_u_plane += u_stride; - *(unsigned short*)v_plane = *(unsigned short*)prev_v_plane; - v_plane += v_stride; - prev_v_plane += v_stride; - } -} - -// This function copies the 8x8 block from the prev_*_planes to the -// current *_planes. -inline void copy_8x8_block( - unsigned char *y_plane, - unsigned char *u_plane, - unsigned char *v_plane, - unsigned char *prev_y_plane, - unsigned char *prev_u_plane, - unsigned char *prev_v_plane, - unsigned int y_stride, - unsigned int u_stride, - unsigned int v_stride) -{ - int i; - - // copy over the luminance components (8 rows, 2 uints each) - for (i = 0; i < 8; i++) - { - ((unsigned int *)y_plane)[0] = ((unsigned int *)prev_y_plane)[0]; - ((unsigned int *)y_plane)[1] = ((unsigned int *)prev_y_plane)[1]; - y_plane += y_stride; - prev_y_plane += y_stride; - } - - // copy the chrominance values - for (i = 0; i < 4; i++) - { - *(unsigned int*)u_plane = *(unsigned int*)prev_u_plane; - u_plane += u_stride; - prev_u_plane += u_stride; - *(unsigned int*)v_plane = *(unsigned int*)prev_v_plane; - v_plane += v_stride; - prev_v_plane += v_stride; - } -} - -// This function creates storage space for the vector codebooks. -void *roq_decode_video_init(void) -{ - roqvideo_info *info = - (roqvideo_info *)calloc(sizeof(roqvideo_info), 1); - - info->prev_frame = NULL; - info->numframe=0; - - return info; -} - -#define EMPTY_ROQ_CODEWORD 0xFFFF0000 - -#define FETCH_NEXT_CODE() \ - if (current_roq_codeword == EMPTY_ROQ_CODEWORD) \ - { \ - if (stream_ptr + 2 > encoded_size) \ - { \ - mp_msg(MSGT_DECVIDEO, MSGL_WARN, \ - "RoQ video: stream pointer just went out of bounds (1)\n"); \ - return; \ - } \ - current_roq_codeword = (0x0000FFFF) | \ - (encoded[stream_ptr + 0] << 16) | \ - (encoded[stream_ptr + 1] << 24); \ - stream_ptr += 2; \ - } \ - roq_code = ((current_roq_codeword >> 30) & 0x03); \ - current_roq_codeword <<= 2; - -#define FETCH_NEXT_ARGUMENT() \ - if (stream_ptr + 1 > encoded_size) \ - { \ - mp_msg(MSGT_DECVIDEO, MSGL_WARN, \ - "RoQ video: stream pointer just went out of bounds (2)\n"); \ - return; \ - } \ - argument = encoded[stream_ptr++]; - -#define CHECK_PREV_FRAME() \ - if (!info->prev_frame) \ - { \ - mp_msg(MSGT_DECVIDEO, MSGL_WARN, \ - "RoQ video: can't handle motion vector when there's no previous frame\n"); \ - return; \ - } - -void roq_decode_video(void *context, unsigned char *encoded, - int encoded_size, mp_image_t *mpi) -{ - roqvideo_info *info = (roqvideo_info *)context; - - int stream_ptr = 0; - int i, j; - int chunk_length; - int v2_count; - int v4_count; - - int roq_code; - unsigned int current_roq_codeword = EMPTY_ROQ_CODEWORD; - unsigned char argument = 0; - char mean_motion_x; - char mean_motion_y; - int mx, my; // for calculating the motion vector - - int mblock_x = 0; - int mblock_y = 0; - int quad8_x, quad8_y; // for pointing to 8x8 blocks in a macroblock - int quad4_x, quad4_y; // for pointing to 4x4 blocks in an 8x8 block - - unsigned char *y_plane; - unsigned char *u_plane; - unsigned char *v_plane; - unsigned char *prev_y_plane; - unsigned char *prev_u_plane; - unsigned char *prev_v_plane; - unsigned int y_stride = mpi->stride[0]; - unsigned int u_stride = mpi->stride[1]; - unsigned int v_stride = mpi->stride[2]; - - roq_v4_codebook v4; - - // make sure the encoded chunk is of minimal acceptable length - if (encoded_size < 8) - { - mp_msg(MSGT_DECVIDEO, MSGL_WARN, - "RoQ video: chunk isn't even 8 bytes long (minimum acceptable length)\n"); - return; - } - - // make sure the resolution checks out - if ((mpi->width % 16 != 0) || (mpi->height % 16 != 0)) - { - mp_msg(MSGT_DECVIDEO, MSGL_WARN, - "RoQ video resolution: %d x %d; expected dimensions divisible by 16\n", - mpi->width, mpi->height); - return; - } - - if (LE_16(&encoded[stream_ptr]) == RoQ_QUAD_CODEBOOK) - { - stream_ptr += 2; - chunk_length = LE_32(&encoded[stream_ptr]); - stream_ptr += 4; - v4_count = encoded[stream_ptr++]; - v2_count = encoded[stream_ptr++]; - if (v2_count == 0) - v2_count = 256; - if ((v4_count == 0) && (v2_count * 6 < chunk_length)) - v4_count = 256; - - // make sure the lengths agree with each other - if (((v2_count * 6) + (v4_count * 4)) != chunk_length) - { - mp_msg(MSGT_DECVIDEO, MSGL_WARN, - "RoQ video: encountered quad codebook chunk with weird lengths (1)\n"); - return; - } - if ((v2_count * 6) > (encoded_size - stream_ptr)) - { - mp_msg(MSGT_DECVIDEO, MSGL_WARN, - "RoQ video: encountered quad codebook chunk with weird lengths (2)\n"); - return; - } - - // load the 2x2 vectors - for (i = 0; i < v2_count; i++) - { - info->v2[i].y0 = encoded[stream_ptr++]; - info->v2[i].y1 = encoded[stream_ptr++]; - info->v2[i].y2 = encoded[stream_ptr++]; - info->v2[i].y3 = encoded[stream_ptr++]; - info->v2[i].u = encoded[stream_ptr++]; - info->v2[i].v = encoded[stream_ptr++]; - prep_v2(&info->v2[i]); - } - - if ((v4_count * 4) > (encoded_size - stream_ptr)) - { - mp_msg(MSGT_DECVIDEO, MSGL_WARN, - "RoQ video: encountered quad codebook chunk with weird lengths (3)\n"); - return; - } - - // load the 4x4 vectors - for (i = 0; i < v4_count; i++) - { - info->v4[i].v2_index[0] = encoded[stream_ptr++]; - info->v4[i].v2_index[1] = encoded[stream_ptr++]; - info->v4[i].v2_index[2] = encoded[stream_ptr++]; - info->v4[i].v2_index[3] = encoded[stream_ptr++]; - } - } - - if (LE_16(&encoded[stream_ptr]) == RoQ_QUAD_VQ) - { - stream_ptr += 2; - chunk_length = LE_32(&encoded[stream_ptr]); - stream_ptr += 4; - mean_motion_y = encoded[stream_ptr++]; - mean_motion_x = encoded[stream_ptr++]; - - //RoQ reuses its buffers so a transparent block keeps content - //from 2 frames ago. The only exception is 2'd frame (#1) - if(info->numframe==1) - { - CHECK_PREV_FRAME(); - memcpy(mpi->planes[0], info->prev_frame->planes[0], - mpi->width * mpi->height); - memcpy(mpi->planes[1], info->prev_frame->planes[1], - (mpi->width * mpi->height) / 4); - memcpy(mpi->planes[2], info->prev_frame->planes[2], - (mpi->width * mpi->height) / 4); - } - - // iterate through the 16x16 macroblocks - for (mblock_y = 0; mblock_y < mpi->height; mblock_y += 16) - { - for (mblock_x = 0; mblock_x < mpi->width; mblock_x += 16) - { - // iterate through the 4 quadrants of the macroblock - for (i = 0; i < 4; i++) - { - quad8_x = mblock_x; - quad8_y = mblock_y; - if (i & 0x01) quad8_x += 8; - if (i & 0x02) quad8_y += 8; - - // set up the planes - y_plane = mpi->planes[0] + quad8_y * y_stride + quad8_x; - u_plane = mpi->planes[1] + (quad8_y / 2) * u_stride + (quad8_x / 2); - v_plane = mpi->planes[2] + (quad8_y / 2) * v_stride + (quad8_x / 2); - - // decide how to handle this 8x8 quad - FETCH_NEXT_CODE(); - switch(roq_code) - { - // 8x8 block is the same as in the previous frame; - // skip it - case 0: - break; - - // 8x8 block is painted with an 8x8 block from the last frame - // (i.e., motion compensation) - case 1: - CHECK_PREV_FRAME(); - - // prepare the pointers to the planes in the previous frame - FETCH_NEXT_ARGUMENT(); // argument contains motion vectors - - // figure out the motion vectors - mx = quad8_x + 8 - (argument >> 4) - mean_motion_x; - my = quad8_y + 8 - (argument & 0x0F) - mean_motion_y; - - prev_y_plane = info->prev_frame->planes[0] + - my * y_stride + mx; - prev_u_plane = info->prev_frame->planes[1] + - (my / 2) * u_stride + (mx + 1) / 2; - prev_v_plane = info->prev_frame->planes[2] + - (my / 2) * v_stride + (mx + 1) / 2; - -// sanity check before rendering - copy_8x8_block( - y_plane, - u_plane, - v_plane, - prev_y_plane, - prev_u_plane, - prev_v_plane, - y_stride, - u_stride, - v_stride - ); - break; - - // 8x8 block is painted with a doublesized 4x4 vector - case 2: - FETCH_NEXT_ARGUMENT(); - v4 = info->v4[argument]; - -// sanity check before rendering - // iterate through 4 4x4 blocks - for (j = 0; j < 4; j++) - { - quad4_x = quad8_x; - quad4_y = quad8_y; - if (j & 0x01) quad4_x += 4; - if (j & 0x02) quad4_y += 4; - - // set up the planes - y_plane = mpi->planes[0] + quad4_y * y_stride + quad4_x; - u_plane = mpi->planes[1] + - (quad4_y / 2) * u_stride + (quad4_x / 2); - v_plane = mpi->planes[2] + - (quad4_y / 2) * v_stride + (quad4_x / 2); - - paint_v2double_block( - y_plane, - u_plane, - v_plane, - &info->v2[v4.v2_index[j]], - y_stride, - u_stride, - v_stride - ); - } - break; - - // 8x8 block is broken down into 4 4x4 blocks and painted using - // 4 different codes. - case 3: - // iterate through 4 4x4 blocks - for (j = 0; j < 4; j++) - { - quad4_x = quad8_x; - quad4_y = quad8_y; - if (j & 0x01) quad4_x += 4; - if (j & 0x02) quad4_y += 4; - - // set up the planes - y_plane = mpi->planes[0] + quad4_y * y_stride + quad4_x; - u_plane = mpi->planes[1] + - (quad4_y / 2) * u_stride + (quad4_x / 2); - v_plane = mpi->planes[2] + - (quad4_y / 2) * v_stride + (quad4_x / 2); - - // decide how to handle this 4x4 quad - FETCH_NEXT_CODE(); - switch(roq_code) - { - // 4x4 block is the same as in the previous frame; - // skip it - case 0: - break; - - // 4x4 block is motion compensated from the previous frame - case 1: - CHECK_PREV_FRAME(); - // prepare the pointers to the planes in the previous frame - FETCH_NEXT_ARGUMENT(); // argument contains motion vectors - - // figure out the motion vectors - mx = quad4_x + 8 - (argument >> 4) - mean_motion_x; - my = quad4_y + 8 - (argument & 0x0F) - mean_motion_y; - - prev_y_plane = info->prev_frame->planes[0] + - my * y_stride + mx; - prev_u_plane = info->prev_frame->planes[1] + - (my / 2) * u_stride + (mx + 1) / 2; - prev_v_plane = info->prev_frame->planes[2] + - (my / 2) * u_stride + (mx + 1) / 2; - -// sanity check before rendering - copy_4x4_block( - y_plane, - u_plane, - v_plane, - prev_y_plane, - prev_u_plane, - prev_v_plane, - y_stride, - u_stride, - v_stride - ); - break; - - // 4x4 block is copied directly from v4 vector table - case 2: - FETCH_NEXT_ARGUMENT(); - v4 = info->v4[argument]; - - paint_v4_block( - y_plane, - u_plane, - v_plane, - y_stride, - u_stride, - v_stride, - &info->v2[v4.v2_index[0]], - &info->v2[v4.v2_index[1]], - &info->v2[v4.v2_index[2]], - &info->v2[v4.v2_index[3]]); - break; - - // 4x4 block is built from 4 2x2 vectors - case 3: - if (stream_ptr + 4 > encoded_size) - { - mp_msg(MSGT_DECVIDEO, MSGL_WARN, - "RoQ video: stream pointer just went out of bounds (2)\n"); - return; - } - paint_v4_block( - y_plane, - u_plane, - v_plane, - y_stride, - u_stride, - v_stride, - &info->v2[encoded[stream_ptr + 0]], - &info->v2[encoded[stream_ptr + 1]], - &info->v2[encoded[stream_ptr + 2]], - &info->v2[encoded[stream_ptr + 3]]); - stream_ptr += 4; - break; - } - } - break; - } - } - } - } - } - - // one last sanity check on the way out - // (apparently, it's not unusual to have 2 bytes left over after decode) - if (stream_ptr < encoded_size - 2) - { - mp_msg(MSGT_DECVIDEO, MSGL_WARN, - "RoQ video: completed frame decode with bytes left over (%d < %d)\n", - stream_ptr, encoded_size); - } - - // save the current frame as the previous frame for the next iteration - info->numframe++; - info->prev_frame = mpi; -} - -// Initialize the RoQ audio decoder, which is to say, initialize the table -// of squares. -void *roq_decode_audio_init(void) -{ - short *square_array; - short square; - int i; - - square_array = (short *)malloc(256 * sizeof(short)); - if (!square_array) - return NULL; - - for (i = 0; i < 128; i++) - { - square = i * i; - square_array[i] = square; - square_array[i + 128] = -square; - } - - return square_array; -} - -int roq_decode_audio( - unsigned short *output, - unsigned char *input, - int encoded_size, - int channels, - void *context) -{ - short *square_array = (short *)context; - int i; - int predictor[2]; - int channel_number = 0; - - // prepare the initial predictors - if (channels == 1) - predictor[0] = LE_16(&input[0]); - else - { - predictor[0] = input[1] << 8; - predictor[1] = input[0] << 8; - } - SE_16BIT(predictor[0]); - SE_16BIT(predictor[1]); - - // decode the samples - for (i = 2; i < encoded_size; i++) - { - predictor[channel_number] += square_array[input[i]]; - CLAMP_S16(predictor[channel_number]); - output[i - 2] = predictor[channel_number]; - - // toggle channel - channel_number ^= channels - 1; - } - - // return the number of samples decoded - return (encoded_size - 2); -} |