comparison zmbvenc.c @ 6758:a57835341650 libavcodec

Improve motion estimation metric. Patch by Michael Niedermayer

6757:7cd794be03ae

    int comp_size;
    int keyint, curfrm;
    z_stream zstream;
} ZmbvEncContext;

/** Block comparing function
 * XXX should be optimized and moved to DSPContext
 * TODO handle out of edge ME
 */
static inline int block_cmp(uint8_t *src, int stride, uint8_t *src2, int stride2, int bw, int bh)
{
    int sum = 0;
    int i, j;

    for(j = 0; j < bh; j++){
        for(i = 0; i < bw; i++)
            sum += src[i] ^ src2[i];
        src += stride;
        src2 += stride2;
    }
    return sum;
}

/** Motion estimation function
 * TODO make better ME decisions

 */
static av_cold int encode_init(AVCodecContext *avctx)
{
    ZmbvEncContext * const c = avctx->priv_data;
    int zret; // Zlib return code
    int lvl = 9;

    c->avctx = avctx;

    c->pic.data[0] = NULL;
    c->curfrm = 0;

6758:a57835341650

    int comp_size;
    int keyint, curfrm;
    z_stream zstream;
} ZmbvEncContext;

static int score_tab[256];

/** Block comparing function
 * XXX should be optimized and moved to DSPContext
 * TODO handle out of edge ME
 */
static inline int block_cmp(uint8_t *src, int stride, uint8_t *src2, int stride2, int bw, int bh)
{
    int sum = 0;
    int i, j;
    uint8_t histogram[256]={0};

    for(j = 0; j < bh; j++){
        for(i = 0; i < bw; i++)
            histogram[src[i] ^ src2[i]]++;
        src += stride;
        src2 += stride2;
    }

    for(i=1; i<256; i++)
        sum+= score_tab[histogram[i]];

    return sum;
}

/** Motion estimation function
 * TODO make better ME decisions

 */
static av_cold int encode_init(AVCodecContext *avctx)
{
    ZmbvEncContext * const c = avctx->priv_data;
    int zret; // Zlib return code
    int i;
    int lvl = 9;

    for(i=1; i<256; i++)
        score_tab[i]= -i * log2(i/256.0) * 256;

    c->avctx = avctx;

    c->pic.data[0] = NULL;
    c->curfrm = 0;