mirrors/gstreamer
mirrors/gstreamer
1
1
Fork 0
mirror of https://gitlab.freedesktop.org/gstreamer/gstreamer.git synced 2024-12-22 16:26:39 +00:00
Code Issues 1 Projects Releases Wiki Activity
gstreamer/gst/mve/mvevideoenc8.c
Tim-Philipp Müller 9e1b75fda3 Fix FSF address 
https://bugzilla.gnome.org/show_bug.cgi?id=687520
2012-11-04 00:09:59 +00:00

1733 lines
44 KiB
C
Raw Blame History

/*
* Interplay MVE video encoder (8 bit)
* Copyright (C) 2006 Jens Granseuer <jensgr@gmx.net>
*
* This library is free software; you can redistribute it and/or
* modify it under the terms of the GNU Library General Public
* License as published by the Free Software Foundation; either
* version 2 of the License, or (at your option) any later version.
*
* This library is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
* Library General Public License for more details.
*
* You should have received a copy of the GNU Library General Public
* License along with this library; if not, write to the
* Free Software Foundation, Inc., 51 Franklin St, Fifth Floor,
* Boston, MA 02110-1301, USA.
*/
#ifdef HAVE_CONFIG_H
# include "config.h"
#endif
#include <stdlib.h>
#include <string.h>
#include "mve.h"
#include "gstmvemux.h"
typedef struct _GstMveEncoderData GstMveEncoderData;
typedef struct _GstMveEncoding GstMveEncoding;
typedef struct _GstMveApprox GstMveApprox;
typedef struct _GstMveQuant GstMveQuant;
#define MVE_RMASK 0x00ff0000
#define MVE_GMASK 0x0000ff00
#define MVE_BMASK 0x000000ff
#define MVE_RSHIFT 16
#define MVE_GSHIFT 8
#define MVE_BSHIFT 0
#define MVE_RVAL(p) (((p) & MVE_RMASK) >> MVE_RSHIFT)
#define MVE_GVAL(p) (((p) & MVE_GMASK) >> MVE_GSHIFT)
#define MVE_BVAL(p) (((p) & MVE_BMASK) >> MVE_BSHIFT)
#define MVE_COL(r,g,b) (((r) << MVE_RSHIFT) | ((g) << MVE_GSHIFT) | ((b) << MVE_BSHIFT))
struct _GstMveEncoderData
{
GstMveMux *mve;
/* current position in frame */
guint16 x, y;
/* palette for current frame */
const guint32 *palette;
/* commonly used quantization results
(2 and 4 colors) for the current block */
guint8 q2block[64];
guint8 q2colors[2];
guint32 q2error;
gboolean q2available;
guint8 q4block[64];
guint8 q4colors[4];
guint32 q4error;
gboolean q4available;
};
struct _GstMveEncoding
{
guint8 opcode;
guint8 size;
guint32 (*approx) (GstMveEncoderData * enc, const guint8 * src,
GstMveApprox * res);
};
#define MVE_APPROX_MAX_ERROR G_MAXUINT32
struct _GstMveApprox
{
guint32 error;
guint8 type;
guint8 data[64]; /* max 64 bytes encoded per block */
guint8 block[64]; /* block in final image */
};
struct _GstMveQuant
{
guint32 col;
guint16 r_total, g_total, b_total;
guint8 r, g, b;
guint8 hits, hits_last;
guint32 max_error;
guint32 max_miss;
};
#define mve_median(mve, src) mve_median_sub ((mve), (src), 8, 8, 0)
#define mve_color_dist(c1, c2) \
mve_color_dist_rgb (MVE_RVAL (c1), MVE_GVAL (c1), MVE_BVAL (c1), \
MVE_RVAL (c2), MVE_GVAL (c2), MVE_BVAL (c2));
#define mve_color_dist2(c, r, g, b) \
mve_color_dist_rgb (MVE_RVAL (c), MVE_GVAL (c), MVE_BVAL (c), (r), (g), (b))
/* comparison function for qsort() */
static int
mve_comp_solution (const void *a, const void *b)
{
const GArray *aa = *((GArray **) a);
const GArray *bb = *((GArray **) b);
if (aa->len <= 1)
return G_MAXINT;
else if (bb->len <= 1)
return G_MININT;
else
return g_array_index (aa, GstMveApprox, aa->len - 2).error -
g_array_index (bb, GstMveApprox, bb->len - 2).error;
}
static inline guint32
mve_color_dist_rgb (guint8 r1, guint8 g1, guint8 b1,
guint8 r2, guint8 g2, guint8 b2)
{
/* euclidean distance (minus sqrt) */
gint dr = r1 - r2;
gint dg = g1 - g2;
gint db = b1 - b2;
return dr * dr + dg * dg + db * db;
}
static guint8
mve_find_pal_color (const guint32 * pal, guint32 col)
{
/* find the closest matching color in the palette */
guint i;
guint8 best = 0;
const guint8 r = MVE_RVAL (col), g = MVE_GVAL (col), b = MVE_BVAL (col);
guint32 ebest = MVE_APPROX_MAX_ERROR;
for (i = 0; i < MVE_PALETTE_COUNT; ++i, ++pal) {
guint32 e = mve_color_dist2 (*pal, r, g, b);
if (e < ebest) {
ebest = e;
best = i;
if (ebest == 0)
break;
}
}
return best;
}
static guint8
mve_find_pal_color2 (const guint32 * pal, const guint8 * subset, guint32 col,
guint size)
{
/* find the closest matching color in the partial indexed palette */
guint i;
guint8 best = 0;
const guint8 r = MVE_RVAL (col), g = MVE_GVAL (col), b = MVE_BVAL (col);
guint32 ebest = MVE_APPROX_MAX_ERROR;
for (i = 0; i < size; ++i) {
guint32 e = mve_color_dist2 (pal[subset[i]], r, g, b);
if (e < ebest) {
ebest = e;
best = subset[i];
if (ebest == 0)
break;
}
}
return best;
}
static void
mve_map_to_palette (const GstMveEncoderData * enc, const guint8 * colors,
const guint8 * data, guint8 * dest, guint w, guint h, guint ncols)
{
guint x, y;
for (y = 0; y < h; ++y) {
for (x = 0; x < w; ++x) {
dest[x] =
mve_find_pal_color2 (enc->palette, colors, enc->palette[data[x]],
ncols);
}
data += enc->mve->width;
dest += 8;
}
}
/* compute average color in a sub-block */
static guint8
mve_median_sub (const GstMveEncoderData * enc, const guint8 * src, guint w,
guint h, guint n)
{
guint x, y;
const guint max = w * h, max2 = max >> 1;
guint32 r_total = max2, g_total = max2, b_total = max2;
src += ((n * w) % 8) + (((n * (8 - h)) / (12 - w)) * h * enc->mve->width);
for (y = 0; y < h; ++y) {
for (x = 0; x < w; ++x) {
guint32 p = enc->palette[src[x]];
r_total += MVE_RVAL (p);
g_total += MVE_GVAL (p);
b_total += MVE_BVAL (p);
}
src += enc->mve->width;
}
return mve_find_pal_color (enc->palette,
MVE_COL (r_total / max, g_total / max, b_total / max));
}
static void
mve_quant_init (const GstMveEncoderData * enc, GstMveQuant * q,
guint n_clusters, const guint8 * data, guint w, guint h)
{
guint i;
guint x, y;
guint32 cols[4];
guint16 val[2];
/* init first cluster with lowest (darkest), second with highest (lightest)
color. if we need 4 clusters, fill in first and last color in the block
and hope they make for a good distribution */
cols[0] = cols[1] = cols[2] = enc->palette[data[0]];
cols[3] = enc->palette[data[(h - 1) * enc->mve->width + w - 1]];
/* favour red over green and blue */
val[0] = val[1] =
(MVE_RVAL (cols[0]) << 1) + MVE_GVAL (cols[0]) + MVE_BVAL (cols[0]);
for (y = 0; y < h; ++y) {
for (x = 0; x < w; ++x) {
guint32 c = enc->palette[data[x]];
if ((c != cols[0]) && (c != cols[1])) {
guint v = (MVE_RVAL (c) << 1) + MVE_GVAL (c) + MVE_BVAL (c);
if (v < val[0]) {
val[0] = v;
cols[0] = c;
} else if (v > val[1]) {
val[1] = v;
cols[1] = c;
}
}
}
data += enc->mve->width;
}
for (i = 0; i < n_clusters; ++i) {
q[i].col = cols[i];
q[i].r = MVE_RVAL (cols[i]);
q[i].g = MVE_GVAL (cols[i]);
q[i].b = MVE_BVAL (cols[i]);
q[i].r_total = q[i].g_total = q[i].b_total = 0;
q[i].hits = q[i].hits_last = 0;
q[i].max_error = 0;
q[i].max_miss = 0;
}
}
static gboolean
mve_quant_update_clusters (GstMveQuant * q, guint n_clusters)
{
gboolean changed = FALSE;
guint i;
for (i = 0; i < n_clusters; ++i) {
if (q[i].hits > 0) {
guint32 means = MVE_COL ((q[i].r_total + q[i].hits / 2) / q[i].hits,
(q[i].g_total + q[i].hits / 2) / q[i].hits,
(q[i].b_total + q[i].hits / 2) / q[i].hits);
if ((means != q[i].col) || (q[i].hits != q[i].hits_last))
changed = TRUE;
q[i].col = means;
q[i].r_total = q[i].g_total = q[i].b_total = 0;
} else {
guint j;
guint32 max_err = 0;
GstMveQuant *worst = NULL;
/* try to replace unused cluster with a better representative */
for (j = 0; j < n_clusters; ++j) {
if (q[j].max_error > max_err) {
worst = &q[j];
max_err = worst->max_error;
}
}
if (worst) {
q[i].col = worst->max_miss;
worst->max_error = 0;
changed = TRUE;
}
}
q[i].r = MVE_RVAL (q[i].col);
q[i].g = MVE_GVAL (q[i].col);
q[i].b = MVE_BVAL (q[i].col);
q[i].hits_last = q[i].hits;
q[i].hits = 0;
}
for (i = 0; i < n_clusters; ++i) {
q[i].max_error = 0;
}
return changed;
}
/* quantize a sub-block using a k-means algorithm */
static guint32
mve_quantize (const GstMveEncoderData * enc, const guint8 * src,
guint w, guint h, guint n, guint ncols, guint8 * dest, guint8 * cols)
{
guint x, y, i;
GstMveQuant q[4];
const guint8 *data;
guint32 error;
g_assert (n <= 4 && ncols <= 4);
src += ((n * w) % 8) + (((n * (8 - h)) / (12 - w)) * h * enc->mve->width);
dest += ((n * w) % 8) + (((n * (8 - h)) / (12 - w)) * h * 8);
mve_quant_init (enc, q, ncols, src, w, h);
do {
data = src;
error = 0;
/* for each pixel find the closest cluster */
for (y = 0; y < h; ++y) {
for (x = 0; x < w; ++x) {
guint32 c = enc->palette[data[x]];
guint8 r = MVE_RVAL (c), g = MVE_GVAL (c), b = MVE_BVAL (c);
guint32 minerr = MVE_APPROX_MAX_ERROR, err;
GstMveQuant *best = NULL;
for (i = 0; i < ncols; ++i) {
err = mve_color_dist_rgb (r, g, b, q[i].r, q[i].g, q[i].b);
if (err < minerr) {
minerr = err;
best = &q[i];
}
}
++best->hits;
best->r_total += r;
best->g_total += g;
best->b_total += b;
if (minerr > best->max_error) {
best->max_error = minerr;
best->max_miss = c;
}
error += minerr;
}
data += enc->mve->width;
}
} while (mve_quant_update_clusters (q, ncols));
/* fill cols array with result colors */
for (i = 0; i < ncols; ++i)
cols[i] = mve_find_pal_color (enc->palette, q[i].col);
/* make sure we have unique colors in slots 0/1 and 2/3 */
if (cols[0] == cols[1])
++cols[1];
if ((ncols > 2) && (cols[2] == cols[3]))
++cols[3];
/* generate the resulting quantized block */
mve_map_to_palette (enc, cols, src, dest, w, h, ncols);
return error;
}
static guint32
mve_block_error (const GstMveEncoderData * enc, const guint8 * b1,
const guint8 * b2, guint32 threshold)
{
/* compute error between two blocks in a frame */
guint32 e = 0;
guint x, y;
for (y = 0; y < 8; ++y) {
for (x = 0; x < 8; ++x) {
e += mve_color_dist (enc->palette[*b1], enc->palette[*b2]);
/* using a threshold to return early gives a huge performance bonus */
if (e >= threshold)
return MVE_APPROX_MAX_ERROR;
++b1;
++b2;
}
b1 += enc->mve->width - 8;
b2 += enc->mve->width - 8;
}
return e;
}
static guint32
mve_block_error_packed (const GstMveEncoderData * enc, const guint8 * block,
const guint8 * scratch)
{
/* compute error between a block in a frame and a (continuous) scratch pad */
guint32 e = 0;
guint x, y;
for (y = 0; y < 8; ++y) {
for (x = 0; x < 8; ++x) {
guint32 c1 = enc->palette[block[x]], c2 = enc->palette[scratch[x]];
e += mve_color_dist (c1, c2);
}
block += enc->mve->width;
scratch += 8;
}
return e;
}
static void
mve_store_block (const GstMveMux * mve, const guint8 * block, guint8 * scratch)
{
/* copy block from frame to a (continuous) scratch pad */
guint y;
for (y = 0; y < 8; ++y) {
memcpy (scratch, block, 8);
block += mve->width;
scratch += 8;
}
}
static void
mve_restore_block (const GstMveMux * mve, guint8 * block,
const guint8 * scratch)
{
/* copy block from scratch pad to frame */
guint y;
for (y = 0; y < 8; ++y) {
memcpy (block, scratch, 8);
block += mve->width;
scratch += 8;
}
}
static guint32
mve_try_vector (GstMveEncoderData * enc, const guint8 * src,
const guint8 * frame, gint pn, GstMveApprox * apx)
{
/* try to locate a similar 8x8 block in the given frame using a motion vector */
guint i;
gint dx, dy;
gint fx, fy;
guint32 err;
apx->error = MVE_APPROX_MAX_ERROR;
for (i = 0; i < 256; ++i) {
if (i < 56) {
dx = 8 + (i % 7);
dy = i / 7;
} else {
dx = -14 + ((i - 56) % 29);
dy = 8 + ((i - 56) / 29);
}
fx = enc->x + dx * pn;
fy = enc->y + dy * pn;
if ((fx >= 0) && (fy >= 0) && (fx + 8 <= enc->mve->width)
&& (fy + 8 <= enc->mve->height)) {
err =
mve_block_error (enc, src, frame + fy * enc->mve->width + fx,
apx->error);
if (err < apx->error) {
apx->data[0] = i;
mve_store_block (enc->mve, frame + fy * enc->mve->width + fx,
apx->block);
apx->error = err;
if (err == 0)
return 0;
}
}
}
return apx->error;
}
static guint32
mve_encode_0x0 (GstMveEncoderData * enc, const guint8 * src, GstMveApprox * apx)
{
/* copy a block from the last frame (0 bytes) */
if (enc->mve->last_frame == NULL)
return MVE_APPROX_MAX_ERROR;
mve_store_block (enc->mve,
GST_BUFFER_DATA (enc->mve->last_frame) +
enc->y * enc->mve->width + enc->x, apx->block);
apx->error = mve_block_error_packed (enc, src, apx->block);
return apx->error;
}
static guint32
mve_encode_0x1 (GstMveEncoderData * enc, const guint8 * src, GstMveApprox * apx)
{
/* copy a block from the second to last frame (0 bytes) */
if (enc->mve->second_last_frame == NULL)
return MVE_APPROX_MAX_ERROR;
mve_store_block (enc->mve,
GST_BUFFER_DATA (enc->mve->second_last_frame) +
enc->y * enc->mve->width + enc->x, apx->block);
apx->error = mve_block_error_packed (enc, src, apx->block);
return apx->error;
}
static guint32
mve_encode_0x2 (GstMveEncoderData * enc, const guint8 * src, GstMveApprox * apx)
{
/* copy block from 2 frames ago using a motion vector (1 byte) */
if (enc->mve->quick_encoding || enc->mve->second_last_frame == NULL)
return MVE_APPROX_MAX_ERROR;
apx->error = mve_try_vector (enc, src,
GST_BUFFER_DATA (enc->mve->second_last_frame), 1, apx);
return apx->error;
}
static guint32
mve_encode_0x3 (GstMveEncoderData * enc, const guint8 * src, GstMveApprox * apx)
{
/* copy 8x8 block from current frame from an up/left block (1 byte) */
if (enc->mve->quick_encoding)
return MVE_APPROX_MAX_ERROR;
apx->error = mve_try_vector (enc, src,
src - enc->mve->width * enc->y - enc->x, -1, apx);
return apx->error;
}
static guint32
mve_encode_0x4 (GstMveEncoderData * enc, const guint8 * src, GstMveApprox * apx)
{
/* copy a block from previous frame using a motion vector (-8/-8 to +7/+7) (1 byte) */
const GstMveMux *mve = enc->mve;
guint32 err;
const guint8 *frame;
gint x1, x2, xi, y1, y2, yi;
if (mve->last_frame == NULL)
return MVE_APPROX_MAX_ERROR;
frame = GST_BUFFER_DATA (mve->last_frame);
x1 = enc->x - 8;
x2 = enc->x + 7;
if (x1 < 0)
x1 = 0;
else if (x2 + 8 > mve->width)
x2 = mve->width - 8;
y1 = enc->y - 8;
y2 = enc->y + 7;
if (y1 < 0)
y1 = 0;
else if (y2 + 8 > mve->height)
y2 = mve->height - 8;
apx->error = MVE_APPROX_MAX_ERROR;
for (yi = y1; yi <= y2; ++yi) {
guint yoff = yi * mve->width;
for (xi = x1; xi <= x2; ++xi) {
err = mve_block_error (enc, src, frame + yoff + xi, apx->error);
if (err < apx->error) {
apx->data[0] = ((xi - enc->x + 8) & 0xF) | ((yi - enc->y + 8) << 4);
mve_store_block (mve, frame + yoff + xi, apx->block);
apx->error = err;
if (err == 0)
return 0;
}
}
}
return apx->error;
}
static guint32
mve_encode_0x5 (GstMveEncoderData * enc, const guint8 * src, GstMveApprox * apx)
{
/* copy a block from previous frame using a motion vector
(-128/-128 to +127/+127) (2 bytes) */
const GstMveMux *mve = enc->mve;
guint32 err;
const guint8 *frame;
gint x1, x2, xi, y1, y2, yi;
if (mve->quick_encoding || mve->last_frame == NULL)
return MVE_APPROX_MAX_ERROR;
frame = GST_BUFFER_DATA (mve->last_frame);
x1 = enc->x - 128;
x2 = enc->x + 127;
if (x1 < 0)
x1 = 0;
if (x2 + 8 > mve->width)
x2 = mve->width - 8;
y1 = enc->y - 128;
y2 = enc->y + 127;
if (y1 < 0)
y1 = 0;
if (y2 + 8 > mve->height)
y2 = mve->height - 8;
apx->error = MVE_APPROX_MAX_ERROR;
for (yi = y1; yi <= y2; ++yi) {
gint yoff = yi * mve->width;
for (xi = x1; xi <= x2; ++xi) {
err = mve_block_error (enc, src, frame + yoff + xi, apx->error);
if (err < apx->error) {
apx->data[0] = xi - enc->x;
apx->data[1] = yi - enc->y;
mve_store_block (mve, frame + yoff + xi, apx->block);
apx->error = err;
if (err == 0)
return 0;
}
}
}
return apx->error;
}
static guint32
mve_encode_0x7a (GstMveEncoderData * enc, const guint8 * src,
GstMveApprox * apx)
{
/* 2-color encoding for 2x2 solid blocks (4 bytes) */
guint32 pix[4];
guint8 mean;
guint32 e1, e2;
guint x, y;
guint8 r[2], g[2], b[2], rb, gb, bb;
guint8 *block = apx->block;
guint16 mask = 0x0001;
guint16 flags = 0;
/* calculate mean colors for the entire block */
if (!enc->q2available) {
enc->q2error =
mve_quantize (enc, src, 8, 8, 0, 2, enc->q2block, enc->q2colors);
enc->q2available = TRUE;
}
/* p0 > p1 */
apx->data[0] = MAX (enc->q2colors[0], enc->q2colors[1]);
apx->data[1] = MIN (enc->q2colors[0], enc->q2colors[1]);
for (x = 0; x < 2; ++x) {
r[x] = MVE_RVAL (enc->palette[apx->data[x]]);
g[x] = MVE_GVAL (enc->palette[apx->data[x]]);
b[x] = MVE_BVAL (enc->palette[apx->data[x]]);
}
/* calculate mean colors for each 2x2 block and map to global colors */
for (y = 0; y < 4; ++y) {
for (x = 0; x < 4; ++x, mask <<= 1) {
pix[0] = enc->palette[src[0]];
pix[1] = enc->palette[src[1]];
pix[2] = enc->palette[src[enc->mve->width]];
pix[3] = enc->palette[src[enc->mve->width + 1]];
rb = (MVE_RVAL (pix[0]) + MVE_RVAL (pix[1]) + MVE_RVAL (pix[2]) +
MVE_RVAL (pix[3]) + 2) / 4;
gb = (MVE_GVAL (pix[0]) + MVE_GVAL (pix[1]) + MVE_GVAL (pix[2]) +
MVE_GVAL (pix[3]) + 2) / 4;
bb = (MVE_BVAL (pix[0]) + MVE_BVAL (pix[1]) + MVE_BVAL (pix[2]) +
MVE_BVAL (pix[3]) + 2) / 4;
e1 = mve_color_dist_rgb (rb, gb, bb, r[0], g[0], b[0]);
e2 = mve_color_dist_rgb (rb, gb, bb, r[1], g[1], b[1]);
if (e1 > e2) {
mean = apx->data[1];
flags |= mask;
} else {
mean = apx->data[0];
}
block[0] = block[1] = block[8] = block[9] = mean;
src += 2;
block += 2;
}
src += (enc->mve->width * 2) - 8;
block += 8;
}
apx->data[2] = flags & 0x00FF;
apx->data[3] = (flags & 0xFF00) >> 8;
apx->error =
mve_block_error_packed (enc, src - enc->mve->width * 8, apx->block);
return apx->error;
}
static guint32
mve_encode_0x7b (GstMveEncoderData * enc, const guint8 * src,
GstMveApprox * apx)
{
/* generic 2-color encoding (10 bytes) */
guint x, y;
guint8 *data = apx->data;
guint8 *block = apx->block;
if (!enc->q2available) {
enc->q2error =
mve_quantize (enc, src, 8, 8, 0, 2, enc->q2block, enc->q2colors);
enc->q2available = TRUE;
}
memcpy (block, enc->q2block, 64);
/* p0 <= p1 */
data[0] = MIN (enc->q2colors[0], enc->q2colors[1]);
data[1] = MAX (enc->q2colors[0], enc->q2colors[1]);
data += 2;
for (y = 0; y < 8; ++y) {
guint8 flags = 0;
for (x = 0x01; x <= 0x80; x <<= 1) {
if (*block == apx->data[1])
flags |= x;
++block;
}
*data++ = flags;
}
apx->error = enc->q2error;
return apx->error;
}
static guint32
mve_encode_0x8a (GstMveEncoderData * enc, const guint8 * src,
GstMveApprox * apx)
{
/* 2-color encoding for top and bottom half (12 bytes) */
guint8 cols[2];
guint32 flags;
guint i, x, y, shifter;
guint8 *block = apx->block;
guint8 *data = apx->data;
apx->error = 0;
for (i = 0; i < 2; ++i) {
apx->error += mve_quantize (enc, src, 8, 4, i, 2, apx->block, cols);
flags = 0;
shifter = 0;
/* p0 > p1 && p2 > p3 */
data[0] = MAX (cols[0], cols[1]);
data[1] = MIN (cols[0], cols[1]);
for (y = 0; y < 4; ++y) {
for (x = 0; x < 8; ++x, ++shifter) {
if (block[x] == data[1])
flags |= 1 << shifter;
}
block += 8;
}
data[2] = flags & 0x000000FF;
data[3] = (flags & 0x0000FF00) >> 8;
data[4] = (flags & 0x00FF0000) >> 16;
data[5] = (flags & 0xFF000000) >> 24;
data += 6;
}
return apx->error;
}
static guint32
mve_encode_0x8b (GstMveEncoderData * enc, const guint8 * src,
GstMveApprox * apx)
{
/* 2-color encoding for left and right half (12 bytes) */
guint8 cols[2];
guint32 flags;
guint i, x, y, shifter;
guint8 *block = apx->block;
guint8 *data = apx->data;
apx->error = 0;
for (i = 0; i < 2; ++i) {
apx->error += mve_quantize (enc, src, 4, 8, i, 2, apx->block, cols);
flags = 0;
shifter = 0;
/* p0 > p1 && p2 <= p3 */
data[i] = MAX (cols[0], cols[1]);
data[i ^ 1] = MIN (cols[0], cols[1]);
for (y = 0; y < 8; ++y) {
for (x = 0; x < 4; ++x, ++shifter) {
if (block[x] == data[1])
flags |= 1 << shifter;
}
block += 8;
}
data[2] = flags & 0x000000FF;
data[3] = (flags & 0x0000FF00) >> 8;
data[4] = (flags & 0x00FF0000) >> 16;
data[5] = (flags & 0xFF000000) >> 24;
data += 6;
block = apx->block + 4;
}
return apx->error;
}
static guint32
mve_encode_0x8c (GstMveEncoderData * enc, const guint8 * src,
GstMveApprox * apx)
{
/* 2-color encoding for each 4x4 quadrant (16 bytes) */
guint8 cols[2];
guint16 flags;
guint i, x, y, shifter;
guint8 *block;
guint8 *data = apx->data;
apx->error = 0;
for (i = 0; i < 4; ++i) {
apx->error +=
mve_quantize (enc, src, 4, 4, ((i & 1) << 1) | ((i & 2) >> 1), 2,
apx->block, cols);
/* p0 < p1 */
if (i == 0) {
data[0] = MIN (cols[0], cols[1]);
data[1] = MAX (cols[0], cols[1]);
} else {
data[0] = cols[0];
data[1] = cols[1];
}
block = apx->block + ((i / 2) * 4) + ((i % 2) * 32);
flags = 0;
shifter = 0;
for (y = 0; y < 4; ++y) {
for (x = 0; x < 4; ++x, ++shifter) {
if (block[x] == data[1])
flags |= 1 << shifter;
}
block += 8;
}
data[2] = flags & 0x00FF;
data[3] = (flags & 0xFF00) >> 8;
data += 4;
}
return apx->error;
}
static guint32
mve_encode_0x9a (GstMveEncoderData * enc, const guint8 * src,
GstMveApprox * apx)
{
/* 4-color encoding for 2x2 solid blocks (8 bytes) */
guint32 p[4];
guint32 e, emin;
guint i, x, y, mean = 0;
guint8 r[4], g[4], b[4], rb, gb, bb;
guint8 *block = apx->block;
guint shifter = 0;
guint32 flags = 0;
/* calculate mean colors for the entire block */
if (!enc->q4available) {
enc->q4error =
mve_quantize (enc, src, 8, 8, 0, 4, enc->q4block, enc->q4colors);
enc->q4available = TRUE;
}
/* p0 <= p1 && p2 > p3 */
apx->data[0] = MIN (enc->q4colors[0], enc->q4colors[1]);
apx->data[1] = MAX (enc->q4colors[0], enc->q4colors[1]);
apx->data[2] = MAX (enc->q4colors[2], enc->q4colors[3]);
apx->data[3] = MIN (enc->q4colors[2], enc->q4colors[3]);
for (i = 0; i < 4; ++i) {
r[i] = MVE_RVAL (enc->palette[apx->data[i]]);
g[i] = MVE_GVAL (enc->palette[apx->data[i]]);
b[i] = MVE_BVAL (enc->palette[apx->data[i]]);
}
/* calculate mean colors for each 2x2 block and map to global colors */
for (y = 0; y < 4; ++y) {
for (x = 0; x < 4; ++x, shifter += 2) {
p[0] = enc->palette[src[0]];
p[1] = enc->palette[src[1]];
p[2] = enc->palette[src[enc->mve->width]];
p[3] = enc->palette[src[enc->mve->width + 1]];
rb = (MVE_RVAL (p[0]) + MVE_RVAL (p[1]) + MVE_RVAL (p[2]) +
MVE_RVAL (p[3]) + 2) / 4;
gb = (MVE_GVAL (p[0]) + MVE_GVAL (p[1]) + MVE_GVAL (p[2]) +
MVE_GVAL (p[3]) + 2) / 4;
bb = (MVE_BVAL (p[0]) + MVE_BVAL (p[1]) + MVE_BVAL (p[2]) +
MVE_BVAL (p[3]) + 2) / 4;
emin = MVE_APPROX_MAX_ERROR;
for (i = 0; i < 4; ++i) {
e = mve_color_dist_rgb (rb, gb, bb, r[i], g[i], b[i]);
if (e < emin) {
emin = e;
mean = i;
}
}
flags |= mean << shifter;
block[0] = block[1] = block[8] = block[9] = apx->data[mean];
src += 2;
block += 2;
}
src += (enc->mve->width * 2) - 8;
block += 8;
}
apx->data[4] = flags & 0x000000FF;
apx->data[5] = (flags & 0x0000FF00) >> 8;
apx->data[6] = (flags & 0x00FF0000) >> 16;
apx->data[7] = (flags & 0xFF000000) >> 24;
apx->error =
mve_block_error_packed (enc, src - 8 * enc->mve->width, apx->block);
return apx->error;
}
static guint32
mve_encode_0x9b (GstMveEncoderData * enc, const guint8 * src,
GstMveApprox * apx)
{
/* 4-color encoding for 2x1 solid blocks (12 bytes) */
guint32 p[2];
guint32 e, emin;
guint i, x, y, mean = 0;
guint8 r[4], g[4], b[4], rb, gb, bb;
guint8 *data = apx->data;
guint8 *block = apx->block;
guint shifter = 0;
guint32 flags = 0;
/* calculate mean colors for the entire block */
if (!enc->q4available) {
enc->q4error =
mve_quantize (enc, src, 8, 8, 0, 4, enc->q4block, enc->q4colors);
enc->q4available = TRUE;
}
/* p0 > p1 && p2 <= p3 */
data[0] = MAX (enc->q4colors[0], enc->q4colors[1]);
data[1] = MIN (enc->q4colors[0], enc->q4colors[1]);
data[2] = MIN (enc->q4colors[2], enc->q4colors[3]);
data[3] = MAX (enc->q4colors[2], enc->q4colors[3]);
for (i = 0; i < 4; ++i) {
r[i] = MVE_RVAL (enc->palette[data[i]]);
g[i] = MVE_GVAL (enc->palette[data[i]]);
b[i] = MVE_BVAL (enc->palette[data[i]]);
}
data += 4;
/* calculate mean colors for each 2x1 block and map to global colors */
for (y = 0; y < 8; ++y) {
for (x = 0; x < 4; ++x, shifter += 2) {
p[0] = enc->palette[src[0]];
p[1] = enc->palette[src[1]];
rb = (MVE_RVAL (p[0]) + MVE_RVAL (p[1]) + 1) / 2;
gb = (MVE_GVAL (p[0]) + MVE_GVAL (p[1]) + 1) / 2;
bb = (MVE_BVAL (p[0]) + MVE_BVAL (p[1]) + 1) / 2;
emin = MVE_APPROX_MAX_ERROR;
for (i = 0; i < 4; ++i) {
e = mve_color_dist_rgb (rb, gb, bb, r[i], g[i], b[i]);
if (e < emin) {
emin = e;
mean = i;
}
}
flags |= mean << shifter;
block[0] = block[1] = apx->data[mean];
src += 2;
block += 2;
}
if ((y == 3) || (y == 7)) {
data[0] = flags & 0x000000FF;
data[1] = (flags & 0x0000FF00) >> 8;
data[2] = (flags & 0x00FF0000) >> 16;
data[3] = (flags & 0xFF000000) >> 24;
data += 4;
flags = 0;
shifter = 0;
}
src += enc->mve->width - 8;
}
apx->error =
mve_block_error_packed (enc, src - 8 * enc->mve->width, apx->block);
return apx->error;
}
static guint32
mve_encode_0x9c (GstMveEncoderData * enc, const guint8 * src,
GstMveApprox * apx)
{
/* 4-color encoding for 1x2 solid blocks (12 bytes) */
guint32 p[2];
guint32 e, emin;
guint i, x, y, mean = 0;
guint8 r[4], g[4], b[4], rb, gb, bb;
guint8 *data = apx->data;
guint8 *block = apx->block;
guint shifter = 0;
guint32 flags = 0;
/* calculate mean colors for the entire block */
if (!enc->q4available) {
enc->q4error =
mve_quantize (enc, src, 8, 8, 0, 4, enc->q4block, enc->q4colors);
enc->q4available = TRUE;
}
/* p0 > p1 && p2 > p3 */
data[0] = MAX (enc->q4colors[0], enc->q4colors[1]);
data[1] = MIN (enc->q4colors[0], enc->q4colors[1]);
data[2] = MAX (enc->q4colors[2], enc->q4colors[3]);
data[3] = MIN (enc->q4colors[2], enc->q4colors[3]);
for (i = 0; i < 4; ++i) {
r[i] = MVE_RVAL (enc->palette[data[i]]);
g[i] = MVE_GVAL (enc->palette[data[i]]);
b[i] = MVE_BVAL (enc->palette[data[i]]);
}
data += 4;
/* calculate mean colors for each 1x2 block and map to global colors */
for (y = 0; y < 4; ++y) {
for (x = 0; x < 8; ++x, shifter += 2) {
p[0] = enc->palette[src[0]];
p[1] = enc->palette[src[enc->mve->width]];
rb = (MVE_RVAL (p[0]) + MVE_RVAL (p[1]) + 1) / 2;
gb = (MVE_GVAL (p[0]) + MVE_GVAL (p[1]) + 1) / 2;
bb = (MVE_BVAL (p[0]) + MVE_BVAL (p[1]) + 1) / 2;
emin = MVE_APPROX_MAX_ERROR;
for (i = 0; i < 4; ++i) {
e = mve_color_dist_rgb (rb, gb, bb, r[i], g[i], b[i]);
if (e < emin) {
emin = e;
mean = i;
}
}
flags |= mean << shifter;
block[0] = block[8] = apx->data[mean];
++src;
++block;
}
if ((y == 1) || (y == 3)) {
data[0] = flags & 0x000000FF;
data[1] = (flags & 0x0000FF00) >> 8;
data[2] = (flags & 0x00FF0000) >> 16;
data[3] = (flags & 0xFF000000) >> 24;
data += 4;
flags = 0;
shifter = 0;
}
src += (enc->mve->width * 2) - 8;
block += 8;
}
apx->error =
mve_block_error_packed (enc, src - 8 * enc->mve->width, apx->block);
return apx->error;
}
static guint32
mve_encode_0x9d (GstMveEncoderData * enc, const guint8 * src,
GstMveApprox * apx)
{
/* generic 4-color encoding (20 bytes) */
guint32 flags = 0;
guint shifter = 0;
guint i, x, y;
guint8 *data = apx->data;
guint8 *block = apx->block;
if (!enc->q4available) {
enc->q4error =
mve_quantize (enc, src, 8, 8, 0, 4, enc->q4block, enc->q4colors);
enc->q4available = TRUE;
}
memcpy (block, enc->q4block, 64);
/* p0 <= p1 && p2 <= p3 */
data[0] = MIN (enc->q4colors[0], enc->q4colors[1]);
data[1] = MAX (enc->q4colors[0], enc->q4colors[1]);
data[2] = MIN (enc->q4colors[2], enc->q4colors[3]);
data[3] = MAX (enc->q4colors[2], enc->q4colors[3]);
data += 4;
for (y = 0; y < 8; ++y) {
for (x = 0; x < 8; ++x, shifter += 2) {
for (i = 0; i < 3; ++i) {
if (*block == apx->data[i])
break;
}
flags |= i << shifter;
++block;
}
data[0] = flags & 0x000000FF;
data[1] = (flags & 0x0000FF00) >> 8;
data += 2;
shifter = 0;
flags = 0;
}
apx->error = enc->q4error;
return apx->error;
}
static guint32
mve_encode_0xaa (GstMveEncoderData * enc, const guint8 * src,
GstMveApprox * apx)
{
/* 4-color encoding for top and bottom half (24 bytes) */
guint8 cols[4];
guint32 flags;
guint i, j, x, y, shifter;
guint8 *block = apx->block;
guint8 *data = apx->data;
const guint8 *p;
apx->error = 0;
for (i = 0; i < 2; ++i) {
apx->error += mve_quantize (enc, src, 8, 4, i, 4, apx->block, cols);
flags = 0;
shifter = 0;
/* p0 > p1 && p4 > p5 */
data[0] = MAX (cols[0], cols[1]);
data[1] = MIN (cols[0], cols[1]);
data[2] = cols[2];
data[3] = cols[3];
p = data;
data += 4;
for (y = 0; y < 4; ++y) {
for (x = 0; x < 8; ++x, shifter += 2) {
for (j = 0; j < 3; ++j) {
if (block[x] == p[j])
break;
}
flags |= j << shifter;
}
block += 8;
if ((y == 1) || (y == 3)) {
data[0] = flags & 0x000000FF;
data[1] = (flags & 0x0000FF00) >> 8;
data[2] = (flags & 0x00FF0000) >> 16;
data[3] = (flags & 0xFF000000) >> 24;
data += 4;
flags = 0;
shifter = 0;
}
}
}
return apx->error;
}
static guint32
mve_encode_0xab (GstMveEncoderData * enc, const guint8 * src,
GstMveApprox * apx)
{
/* 4-color encoding for left and right half (24 bytes) */
guint8 cols[4];
guint32 flags;
guint i, j, x, y, shifter;
guint8 *block = apx->block;
guint8 *data = apx->data;
const guint8 *p;
apx->error = 0;
for (i = 0; i < 2; ++i) {
apx->error += mve_quantize (enc, src, 4, 8, i, 4, apx->block, cols);
flags = 0;
shifter = 0;
/* p0 > p1 && p4 <= p5 */
data[i] = MAX (cols[0], cols[1]);
data[i ^ 1] = MIN (cols[0], cols[1]);
data[2] = cols[2];
data[3] = cols[3];
p = data;
data += 4;
for (y = 0; y < 8; ++y) {
for (x = 0; x < 4; ++x, shifter += 2) {
for (j = 0; j < 3; ++j) {
if (block[x] == p[j])
break;
}
flags |= j << shifter;
}
block += 8;
if ((y == 3) || (y == 7)) {
data[0] = flags & 0x000000FF;
data[1] = (flags & 0x0000FF00) >> 8;
data[2] = (flags & 0x00FF0000) >> 16;
data[3] = (flags & 0xFF000000) >> 24;
data += 4;
flags = 0;
shifter = 0;
}
}
block = apx->block + 4;
}
return apx->error;
}
static guint32
mve_encode_0xac (GstMveEncoderData * enc, const guint8 * src,
GstMveApprox * apx)
{
/* 4-color encoding for each 4x4 quadrant (32 bytes) */
guint8 cols[4];
guint32 flags;
guint i, j, x, y, shifter;
guint8 *block;
guint8 *data = apx->data;
apx->error = 0;
for (i = 0; i < 4; ++i) {
apx->error +=
mve_quantize (enc, src, 4, 4, ((i & 1) << 1) | ((i & 2) >> 1), 4,
apx->block, cols);
/* p0 <= p1 */
data[0] = MIN (cols[0], cols[1]);
data[1] = MAX (cols[0], cols[1]);
data[2] = cols[2];
data[3] = cols[3];
block = apx->block + ((i / 2) * 4) + ((i % 2) * 32);
flags = 0;
shifter = 0;
for (y = 0; y < 4; ++y) {
for (x = 0; x < 4; ++x, shifter += 2) {
for (j = 0; j < 3; ++j) {
if (block[x] == data[j])
break;
}
flags |= j << shifter;
}
block += 8;
}
data[4] = flags & 0x000000FF;
data[5] = (flags & 0x0000FF00) >> 8;
data[6] = (flags & 0x00FF0000) >> 16;
data[7] = (flags & 0xFF000000) >> 24;
data += 8;
}
return apx->error;
}
static guint32
mve_encode_0xb (GstMveEncoderData * enc, const guint8 * src, GstMveApprox * apx)
{
/* 64-color encoding (each pixel in block is a different color) (64 bytes) */
mve_store_block (enc->mve, src, apx->block);
memcpy (apx->data, apx->block, 64);
apx->error = 0;
return 0;
}
static guint32
mve_encode_0xc (GstMveEncoderData * enc, const guint8 * src, GstMveApprox * apx)
{
/* 16-color block encoding: each 2x2 block is a different color (16 bytes) */
guint i = 0, x, y;
const guint w = enc->mve->width;
guint16 r, g, b;
/* calculate median color for each 2x2 block */
for (y = 0; y < 4; ++y) {
for (x = 0; x < 4; ++x) {
guint32 p = enc->palette[src[0]];
r = MVE_RVAL (p) + 2;
g = MVE_GVAL (p) + 2;
b = MVE_BVAL (p) + 2;
p = enc->palette[src[1]];
r += MVE_RVAL (p);
g += MVE_GVAL (p);
b += MVE_BVAL (p);
p = enc->palette[src[w]];
r += MVE_RVAL (p);
g += MVE_GVAL (p);
b += MVE_BVAL (p);
p = enc->palette[src[w + 1]];
r += MVE_RVAL (p);
g += MVE_GVAL (p);
b += MVE_BVAL (p);
apx->block[i] = apx->block[i + 1] = apx->block[i + 2] =
apx->block[i + 3] = apx->data[i >> 2] =
mve_find_pal_color (enc->palette, MVE_COL (r >> 2, g >> 2, b >> 2));
i += 4;
src += 2;
}
src += (w * 2) - 8;
}
apx->error = mve_block_error_packed (enc, src - (8 * w), apx->block);
return apx->error;
}
static guint32
mve_encode_0xd (GstMveEncoderData * enc, const guint8 * src, GstMveApprox * apx)
{
/* 4-color block encoding: each 4x4 block is a different color (4 bytes) */
guint i, y;
/* calculate median color for each 4x4 block */
for (i = 0; i < 4; ++i) {
guint8 median =
mve_median_sub (enc, src, 4, 4, ((i & 1) << 1) | ((i & 2) >> 1));
guint8 *block = apx->block + ((i / 2) * 4) + ((i % 2) * 32);
for (y = 0; y < 4; ++y) {
memset (block, median, 4);
block += 8;
}
apx->data[i] = median;
}
apx->error = mve_block_error_packed (enc, src, apx->block);
return apx->error;
}
static guint32
mve_encode_0xe (GstMveEncoderData * enc, const guint8 * src, GstMveApprox * apx)
{
/* 1-color encoding: the whole block is 1 solid color (1 bytes) */
guint8 median = mve_median (enc, src);
memset (apx->block, median, 64);
apx->data[0] = median;
apx->error = mve_block_error_packed (enc, src, apx->block);
return apx->error;
}
static guint32
mve_encode_0xf (GstMveEncoderData * enc, const guint8 * src, GstMveApprox * apx)
{
/* 2 colors dithered encoding (2 bytes) */
guint i, x, y;
guint32 r[2] = { 0 }, g[2] = {
0}, b[2] = {
0};
guint8 col[2];
/* find medians for both colors */
for (y = 0; y < 8; ++y) {
for (x = 0; x < 8; x += 2) {
guint16 p = src[x];
r[y & 1] += MVE_RVAL (p);
g[y & 1] += MVE_GVAL (p);
b[y & 1] += MVE_BVAL (p);
p = src[x + 1];
r[(y & 1) ^ 1] += MVE_RVAL (p);
g[(y & 1) ^ 1] += MVE_GVAL (p);
b[(y & 1) ^ 1] += MVE_BVAL (p);
}
src += enc->mve->width;
}
col[0] = mve_find_pal_color (enc->palette,
MVE_COL ((r[0] + 16) / 32, (g[0] + 16) / 32, (b[0] + 16) / 32));
col[1] = mve_find_pal_color (enc->palette,
MVE_COL ((r[1] + 16) / 32, (g[1] + 16) / 32, (b[1] + 16) / 32));
/* store block after encoding */
for (i = 0, y = 0; y < 8; ++y) {
for (x = 0; x < 4; ++x) {
apx->block[i++] = col[y & 1];
apx->block[i++] = col[(y & 1) ^ 1];
}
}
apx->data[0] = col[0];
apx->data[1] = col[1];
apx->error = mve_block_error_packed (enc,
src - (8 * enc->mve->width), apx->block);
return apx->error;
}
/* all available encodings in the preferred order,
i.e. in ascending encoded size */
static const GstMveEncoding mve_encodings[] = {
{0x1, 0, mve_encode_0x1},
{0x0, 0, mve_encode_0x0},
{0xe, 1, mve_encode_0xe},
{0x3, 1, mve_encode_0x3},
{0x4, 1, mve_encode_0x4},
{0x2, 1, mve_encode_0x2},
{0xf, 2, mve_encode_0xf},
{0x5, 2, mve_encode_0x5},
{0xd, 4, mve_encode_0xd},
{0x7, 4, mve_encode_0x7a},
{0x9, 8, mve_encode_0x9a},
{0x7, 10, mve_encode_0x7b},
{0x8, 12, mve_encode_0x8a},
{0x8, 12, mve_encode_0x8b},
{0x9, 12, mve_encode_0x9b},
{0x9, 12, mve_encode_0x9c},
{0xc, 16, mve_encode_0xc},
{0x8, 16, mve_encode_0x8c},
{0x9, 20, mve_encode_0x9d},
{0xa, 24, mve_encode_0xaa},
{0xa, 24, mve_encode_0xab},
{0xa, 32, mve_encode_0xac},
{0xb, 64, mve_encode_0xb}
};
static gboolean
mve_reorder_solution (GArray ** solution, guint16 n)
{
/* do a binary search to find the position to reinsert the modified element */
/* the block we need to reconsider is always at position 0 */
/* return TRUE if this block only has 1 encoding left and can be dropped */
if (mve_comp_solution (&solution[0], &solution[1]) <= 0)
return FALSE; /* already sorted */
else if (solution[0]->len <= 1)
/* drop this element from further calculations since we cannot improve here */
return TRUE;
else {
/* we know the error value can only get worse, so we can actually start at 1 */
guint lower = 1;
guint upper = n - 1;
gint cmp;
guint idx = 0;
while (upper > lower) {
idx = lower + ((upper - lower) / 2);
cmp = mve_comp_solution (&solution[0], &solution[idx]);
if (cmp < 0) {
upper = idx;
} else if (cmp > 0) {
lower = ++idx;
} else {
upper = lower = idx;
}
}
if (idx > 0) {
/* rearrange array members in new order */
GArray *a = solution[0];
memcpy (&solution[0], &solution[1], sizeof (GArray *) * idx);
solution[idx] = a;
}
}
return FALSE;
}
static guint32
gst_mve_find_solution (GArray ** approx, guint16 n, guint32 size, guint16 max)
{
/* build an array of approximations we can shuffle around */
GstMveApprox *sol_apx;
GArray **solution = g_malloc (sizeof (GArray *) * n);
GArray **current = solution;
memcpy (solution, approx, sizeof (GArray *) * n);
qsort (solution, n, sizeof (GArray *), mve_comp_solution);
do {
/* array is now sorted by error of the next to optimal approximation;
drop optimal approximation for the best block */
/* unable to reduce size further */
if (current[0]->len <= 1)
break;
sol_apx = &g_array_index (current[0], GstMveApprox, current[0]->len - 1);
size -= mve_encodings[sol_apx->type].size;
g_array_remove_index_fast (current[0], current[0]->len - 1);
sol_apx = &g_array_index (current[0], GstMveApprox, current[0]->len - 1);
size += mve_encodings[sol_apx->type].size;
if (mve_reorder_solution (current, n)) {
++current;
--n;
}
} while (size > max);
g_free (solution);
return size;
}
GstFlowReturn
mve_encode_frame8 (GstMveMux * mve, GstBuffer * frame, const guint32 * palette,
guint16 max_data)
{
guint8 *src;
GstFlowReturn ret = GST_FLOW_ERROR;
guint8 *cm = mve->chunk_code_map;
GArray **approx;
GstMveApprox apx;
GstMveEncoderData enc;
const guint16 blocks = (mve->width * mve->height) / 64;
guint32 encoded_size = 0;
guint i = 0, x, y;
src = GST_BUFFER_DATA (frame);
approx = g_malloc (sizeof (GArray *) * blocks);
enc.mve = mve;
enc.palette = palette;
for (enc.y = 0; enc.y < mve->height; enc.y += 8) {
for (enc.x = 0; enc.x < mve->width; enc.x += 8) {
guint32 err, last_err = MVE_APPROX_MAX_ERROR;
guint type = 0;
guint best = 0;
enc.q2available = enc.q4available = FALSE;
approx[i] = g_array_new (FALSE, FALSE, sizeof (GstMveApprox));
do {
err = mve_encodings[type].approx (&enc, src, &apx);
if (err < last_err) {
apx.type = best = type;
g_array_append_val (approx[i], apx);
last_err = err;
}
++type;
} while (last_err != 0);
encoded_size += mve_encodings[best].size;
++i;
src += 8;
}
src += 7 * mve->width;
}
/* find best solution with size constraints */
GST_DEBUG_OBJECT (mve, "encoded frame %u in %u bytes (lossless)",
mve->video_frames + 1, encoded_size);
#if 0
/* FIXME */
src = GST_BUFFER_DATA (frame);
for (i = 0, y = 0; y < mve->height; y += 8) {
for (x = 0; x < mve->width; x += 8, ++i) {
GstMveApprox *sol =
&g_array_index (approx[i], GstMveApprox, approx[i]->len - 1);
guint opcode = mve_encodings[sol->type].opcode;
guint j, k;
if (sol->error > 0)
GST_WARNING_OBJECT (mve, "error is %lu for %d/%d (0x%x)", sol->error, x,
y, opcode);
for (j = 0; j < 8; ++j) {
guint8 *o = src + j * mve->width;
guint8 *c = sol->block + j * 8;
if (memcmp (o, c, 8)) {
GST_WARNING_OBJECT (mve, "opcode 0x%x (type %d) at %d/%d, line %d:",
opcode, sol->type, x, y, j + 1);
for (k = 0; k < 8; ++k) {
o = src + k * mve->width;
c = sol->block + k * 8;
GST_WARNING_OBJECT (mve,
"%d should be: %4d %4d %4d %4d %4d %4d %4d %4d", k, o[0],
o[1], o[2], o[3], o[4], o[5], o[6], o[7]);
GST_WARNING_OBJECT (mve,
"%d but is : %4d %4d %4d %4d %4d %4d %4d %4d", k, c[0],
c[1], c[2], c[3], c[4], c[5], c[6], c[7]);
}
}
}
src += 8;
}
src += 7 * mve->width;
}
#endif
if (encoded_size > max_data) {
encoded_size =
gst_mve_find_solution (approx, blocks, encoded_size, max_data);
if (encoded_size > max_data) {
GST_ERROR_OBJECT (mve, "unable to compress frame to less than %d bytes",
encoded_size);
for (i = 0; i < blocks; ++i)
g_array_free (approx[i], TRUE);
goto done;
}
GST_DEBUG_OBJECT (mve, "compressed frame %u to %u bytes (lossy)",
mve->video_frames + 1, encoded_size);
}
mve->chunk_video = g_byte_array_sized_new (encoded_size);
/* encode */
src = GST_BUFFER_DATA (frame);
for (i = 0, y = 0; y < mve->height; y += 8) {
for (x = 0; x < mve->width; x += 8, ++i) {
GstMveApprox *sol =
&g_array_index (approx[i], GstMveApprox, approx[i]->len - 1);
guint opcode = mve_encodings[sol->type].opcode;
g_byte_array_append (mve->chunk_video, sol->data,
mve_encodings[sol->type].size);
if (i & 1) {
*cm |= opcode << 4;
++cm;
} else
*cm = opcode;
/* modify the frame to match the image we actually encoded */
if (sol->error > 0)
mve_restore_block (mve, src, sol->block);
src += 8;
g_array_free (approx[i], TRUE);
}
src += 7 * mve->width;
}
ret = GST_FLOW_OK;
done:
g_free (approx);
return ret;
}
Reference in a new issue View git blame Copy permalink