1
0
mirror of https://github.com/FFmpeg/FFmpeg.git synced 2024-11-26 19:01:44 +02:00

Merge commit '296eff4d9dc53d441b672319524a051d04f4a8cf'

* commit '296eff4d9dc53d441b672319524a051d04f4a8cf':
  zmbvenc: get rid of a global table

Merged-by: Clément Bœsch <u@pkh.me>
This commit is contained in:
Clément Bœsch 2017-05-19 11:12:24 +02:00
commit 52a4004d8b

View File

@ -54,16 +54,18 @@ typedef struct ZmbvEncContext {
int comp_size; int comp_size;
int keyint, curfrm; int keyint, curfrm;
z_stream zstream; z_stream zstream;
int score_tab[256];
} ZmbvEncContext; } ZmbvEncContext;
static int score_tab[256];
/** Block comparing function /** Block comparing function
* XXX should be optimized and moved to DSPContext * XXX should be optimized and moved to DSPContext
* TODO handle out of edge ME * TODO handle out of edge ME
*/ */
static inline int block_cmp(uint8_t *src, int stride, uint8_t *src2, int stride2, static inline int block_cmp(ZmbvEncContext *c, uint8_t *src, int stride,
int bw, int bh, int *xored) uint8_t *src2, int stride2, int bw, int bh,
int *xored)
{ {
int sum = 0; int sum = 0;
int i, j; int i, j;
@ -81,7 +83,7 @@ static inline int block_cmp(uint8_t *src, int stride, uint8_t *src2, int stride2
} }
for(i = 1; i < 256; i++) for(i = 1; i < 256; i++)
sum += score_tab[histogram[i]]; sum += c->score_tab[histogram[i]];
return sum; return sum;
} }
@ -97,14 +99,14 @@ static int zmbv_me(ZmbvEncContext *c, uint8_t *src, int sstride, uint8_t *prev,
*mx = *my = 0; *mx = *my = 0;
bw = FFMIN(ZMBV_BLOCK, c->avctx->width - x); bw = FFMIN(ZMBV_BLOCK, c->avctx->width - x);
bh = FFMIN(ZMBV_BLOCK, c->avctx->height - y); bh = FFMIN(ZMBV_BLOCK, c->avctx->height - y);
bv = block_cmp(src, sstride, prev, pstride, bw, bh, xored); bv = block_cmp(c, src, sstride, prev, pstride, bw, bh, xored);
if(!bv) return 0; if(!bv) return 0;
for(ty = FFMAX(y - c->range, 0); ty < FFMIN(y + c->range, c->avctx->height - bh); ty++){ for(ty = FFMAX(y - c->range, 0); ty < FFMIN(y + c->range, c->avctx->height - bh); ty++){
for(tx = FFMAX(x - c->range, 0); tx < FFMIN(x + c->range, c->avctx->width - bw); tx++){ for(tx = FFMAX(x - c->range, 0); tx < FFMIN(x + c->range, c->avctx->width - bw); tx++){
if(tx == x && ty == y) continue; // we already tested this block if(tx == x && ty == y) continue; // we already tested this block
dx = tx - x; dx = tx - x;
dy = ty - y; dy = ty - y;
tv = block_cmp(src, sstride, prev + dx + dy*pstride, pstride, bw, bh, xored); tv = block_cmp(c, src, sstride, prev + dx + dy * pstride, pstride, bw, bh, xored);
if(tv < bv){ if(tv < bv){
bv = tv; bv = tv;
*mx = dx; *mx = dx;
@ -277,7 +279,7 @@ static av_cold int encode_init(AVCodecContext *avctx)
int lvl = 9; int lvl = 9;
for(i=1; i<256; i++) for(i=1; i<256; i++)
score_tab[i]= -i * log2(i/(double)(ZMBV_BLOCK*ZMBV_BLOCK)) * 256; c->score_tab[i] = -i * log2(i / (double)(ZMBV_BLOCK * ZMBV_BLOCK)) * 256;
c->avctx = avctx; c->avctx = avctx;