44 #if FF_API_PRIVATE_OPT
58 s->spatial_decomposition_type=
s->pred;
63 for(plane_index=0; plane_index<3; plane_index++){
64 s->plane[plane_index].diag_mc= 1;
65 s->plane[plane_index].htaps= 6;
66 s->plane[plane_index].hcoeff[0]= 40;
67 s->plane[plane_index].hcoeff[1]= -10;
68 s->plane[plane_index].hcoeff[2]= 2;
69 s->plane[plane_index].fast_mc= 1;
85 s->m.mb_num = (avctx->
width * avctx->
height + 255) / 256;
92 if (!
s->m.me.scratchpad || !
s->m.me.map || !
s->m.me.score_map || !
s->m.sc.obmc_scratchpad)
120 s->colorspace_type= 0;
124 s->colorspace_type = 1;
142 ff_set_cmp(&
s->mecc,
s->mecc.me_sub_cmp,
s->avctx->me_sub_cmp);
145 if (!
s->input_picture)
152 int size=
s->b_width *
s->b_height << 2*
s->block_max_depth;
153 for(
i=0;
i<
s->max_ref_frames;
i++){
156 if (!
s->ref_mvs[
i] || !
s->ref_scores[
i])
170 for (
i = 0;
i <
h;
i++) {
171 for (j = 0; j <
w; j++) {
175 pix += line_size -
w;
187 for (
i = 0;
i <
w;
i++) {
188 for (j = 0; j <
w; j ++) {
192 pix += line_size -
w;
224 #define P_TOPRIGHT P[3]
225 #define P_MEDIAN P[4]
232 uint8_t p_state[
sizeof(
s->block_state)];
233 uint8_t i_state[
sizeof(
s->block_state)];
236 uint8_t *pbbak_start=
s->c.bytestream_start;
237 int score, score2, iscore, i_len, p_len, block_s, sum, base_bits;
238 const int w=
s->b_width <<
s->block_max_depth;
239 const int h=
s->b_height <<
s->block_max_depth;
240 const int rem_depth=
s->block_max_depth -
level;
241 const int index= (x + y*
w) << rem_depth;
243 int trx= (x+1)<<rem_depth;
244 int try= (y+1)<<rem_depth;
251 int pl = left->
color[0];
252 int pcb= left->
color[1];
253 int pcr= left->
color[2];
257 const int stride=
s->current_picture->linesize[0];
258 const int uvstride=
s->current_picture->linesize[1];
259 uint8_t *current_data[3]= {
s->input_picture->data[0] + (x + y*
stride)*block_w,
260 s->input_picture->data[1] + ((x*block_w)>>
s->chroma_h_shift) + ((y*uvstride*block_w)>>
s->chroma_v_shift),
261 s->input_picture->data[2] + ((x*block_w)>>
s->chroma_h_shift) + ((y*uvstride*block_w)>>
s->chroma_v_shift)};
263 int16_t last_mv[3][2];
265 const int shift= 1+qpel;
271 int ref, best_ref, ref_score, ref_mx, ref_my;
275 set_blocks(
s,
level, x, y, pl, pcb, pcr, 0, 0, 0,
BLOCK_INTRA);
288 last_mv[0][0]=
s->block[
index].mx;
289 last_mv[0][1]=
s->block[
index].my;
290 last_mv[1][0]= right->
mx;
291 last_mv[1][1]= right->
my;
292 last_mv[2][0]= bottom->
mx;
293 last_mv[2][1]= bottom->
my;
306 c->current_mv_penalty=
c->mv_penalty[
s->m.f_code=1] +
MAX_DMV;
308 c->xmin = - x*block_w - 16+3;
309 c->ymin = - y*block_w - 16+3;
310 c->xmax = - (x+1)*block_w + (
w<<(
LOG2_MB_SIZE -
s->block_max_depth)) + 16-3;
311 c->ymax = - (y+1)*block_w + (
h<<(
LOG2_MB_SIZE -
s->block_max_depth)) + 16-3;
335 init_ref(
c, current_data,
s->last_picture[
ref]->data,
NULL, block_w*x, block_w*y, 0);
345 ref_score=
c->sub_motion_search(&
s->m, &ref_mx, &ref_my, ref_score, 0, 0,
level-
LOG2_MB_SIZE+4, block_w);
353 if(score > ref_score){
363 base_bits=
get_rac_count(&
s->c) - 8*(
s->c.bytestream -
s->c.bytestream_start);
366 pc.bytestream= p_buffer;
367 memcpy(p_state,
s->block_state,
sizeof(
s->block_state));
369 if(
level!=
s->block_max_depth)
370 put_rac(&pc, &p_state[4 + s_context], 1);
372 if(
s->ref_frames > 1)
373 put_symbol(&pc, &p_state[128 + 1024 + 32*ref_context], best_ref, 0);
374 pred_mv(
s, &pmx, &pmy, best_ref, left, top, tr);
375 put_symbol(&pc, &p_state[128 + 32*(mx_context + 16*!!best_ref)], mx - pmx, 1);
376 put_symbol(&pc, &p_state[128 + 32*(my_context + 16*!!best_ref)], my - pmy, 1);
377 p_len= pc.bytestream - pc.bytestream_start;
380 block_s= block_w*block_w;
382 l= (sum + block_s/2)/block_s;
383 iscore =
pix_norm1(current_data[0],
stride, block_w) - 2*l*sum + l*l*block_s;
385 if (
s->nb_planes > 2) {
386 block_s= block_w*block_w>>(
s->chroma_h_shift +
s->chroma_v_shift);
387 sum =
pix_sum(current_data[1], uvstride, block_w>>
s->chroma_h_shift, block_w>>
s->chroma_v_shift);
388 cb= (sum + block_s/2)/block_s;
390 sum =
pix_sum(current_data[2], uvstride, block_w>>
s->chroma_h_shift, block_w>>
s->chroma_v_shift);
391 cr= (sum + block_s/2)/block_s;
398 ic.bytestream= i_buffer;
399 memcpy(i_state,
s->block_state,
sizeof(
s->block_state));
400 if(
level!=
s->block_max_depth)
401 put_rac(&ic, &i_state[4 + s_context], 1);
404 if (
s->nb_planes > 2) {
408 i_len= ic.bytestream - ic.bytestream_start;
411 av_assert1(iscore < 255*255*256 + s->lambda2*10);
417 int varc= iscore >> 8;
418 int vard= score >> 8;
419 if (vard <= 64 || vard < varc)
422 c->scene_change_score+=
s->m.qscale;
425 if(
level!=
s->block_max_depth){
426 put_rac(&
s->c, &
s->block_state[4 + s_context], 0);
433 if(score2 < score && score2 < iscore)
438 pred_mv(
s, &pmx, &pmy, 0, left, top, tr);
439 memcpy(pbbak, i_buffer, i_len);
441 s->c.bytestream_start= pbbak_start;
442 s->c.bytestream= pbbak + i_len;
443 set_blocks(
s,
level, x, y, l,
cb,
cr, pmx, pmy, 0,
BLOCK_INTRA);
444 memcpy(
s->block_state, i_state,
sizeof(
s->block_state));
447 memcpy(pbbak, p_buffer, p_len);
449 s->c.bytestream_start= pbbak_start;
450 s->c.bytestream= pbbak + p_len;
451 set_blocks(
s,
level, x, y, pl, pcb, pcr, mx, my, best_ref, 0);
452 memcpy(
s->block_state, p_state,
sizeof(
s->block_state));
458 const int w=
s->b_width <<
s->block_max_depth;
459 const int rem_depth=
s->block_max_depth -
level;
460 const int index= (x + y*
w) << rem_depth;
461 int trx= (x+1)<<rem_depth;
467 int pl = left->
color[0];
468 int pcb= left->
color[1];
469 int pcr= left->
color[2];
477 set_blocks(
s,
level, x, y, pl, pcb, pcr, 0, 0, 0,
BLOCK_INTRA);
481 if(
level!=
s->block_max_depth){
483 put_rac(&
s->c, &
s->block_state[4 + s_context], 1);
485 put_rac(&
s->c, &
s->block_state[4 + s_context], 0);
494 pred_mv(
s, &pmx, &pmy, 0, left, top, tr);
497 if (
s->nb_planes > 2) {
501 set_blocks(
s,
level, x, y,
b->color[0],
b->color[1],
b->color[2], pmx, pmy, 0,
BLOCK_INTRA);
503 pred_mv(
s, &pmx, &pmy,
b->ref, left, top, tr);
505 if(
s->ref_frames > 1)
506 put_symbol(&
s->c, &
s->block_state[128 + 1024 + 32*ref_context],
b->ref, 0);
507 put_symbol(&
s->c, &
s->block_state[128 + 32*mx_context],
b->mx - pmx, 1);
508 put_symbol(&
s->c, &
s->block_state[128 + 32*my_context],
b->my - pmy, 1);
509 set_blocks(
s,
level, x, y, pl, pcb, pcr,
b->mx,
b->my,
b->ref, 0);
515 Plane *p= &
s->plane[plane_index];
516 const int block_size =
MB_SIZE >>
s->block_max_depth;
517 const int block_w = plane_index ? block_size>>
s->chroma_h_shift : block_size;
518 const int block_h = plane_index ? block_size>>
s->chroma_v_shift : block_size;
520 const int obmc_stride= plane_index ? (2*block_size)>>
s->chroma_h_shift : 2*block_size;
521 const int ref_stride=
s->current_picture->linesize[plane_index];
522 uint8_t *
src=
s-> input_picture->data[plane_index];
523 IDWTELEM *dst= (
IDWTELEM*)
s->m.sc.obmc_scratchpad + plane_index*block_size*block_size*4;
524 const int b_stride =
s->b_width <<
s->block_max_depth;
527 int index= mb_x + mb_y*b_stride;
536 b->color[plane_index]= 0;
537 memset(dst, 0, obmc_stride*obmc_stride*
sizeof(
IDWTELEM));
540 int mb_x2= mb_x + (
i &1) - 1;
541 int mb_y2= mb_y + (
i>>1) - 1;
542 int x= block_w*mb_x2 + block_w/2;
543 int y= block_h*mb_y2 + block_h/2;
546 x, y, block_w, block_h,
w,
h, obmc_stride, ref_stride, obmc_stride, mb_x2, mb_y2, 0, 0, plane_index);
548 for(y2=
FFMAX(y, 0); y2<
FFMIN(
h, y+block_h); y2++){
549 for(x2=
FFMAX(x, 0); x2<
FFMIN(
w, x+block_w); x2++){
550 int index= x2-(block_w*mb_x - block_w/2) + (y2-(block_h*mb_y - block_h/2))*obmc_stride;
551 int obmc_v= obmc[
index];
553 if(y<0) obmc_v += obmc[
index + block_h*obmc_stride];
554 if(x<0) obmc_v += obmc[
index + block_w];
555 if(y+block_h>
h) obmc_v += obmc[
index - block_h*obmc_stride];
556 if(x+block_w>
w) obmc_v += obmc[
index - block_w];
561 ab += (
src[x2 + y2*ref_stride] - (d>>
FRAC_BITS)) * obmc_v;
562 aa += obmc_v * obmc_v;
572 const int b_stride =
s->b_width <<
s->block_max_depth;
573 const int b_height =
s->b_height<<
s->block_max_depth;
574 int index= x + y*b_stride;
584 if(x<0 || x>=b_stride || y>=b_height)
600 pred_mv(
s, &dmx, &dmy,
b->ref, left, top, tr);
610 Plane *p= &
s->plane[plane_index];
611 const int block_size =
MB_SIZE >>
s->block_max_depth;
612 const int block_w = plane_index ? block_size>>
s->chroma_h_shift : block_size;
613 const int block_h = plane_index ? block_size>>
s->chroma_v_shift : block_size;
614 const int obmc_stride= plane_index ? (2*block_size)>>
s->chroma_h_shift : 2*block_size;
615 const int ref_stride=
s->current_picture->linesize[plane_index];
616 uint8_t *dst=
s->current_picture->data[plane_index];
617 uint8_t *
src=
s-> input_picture->data[plane_index];
621 const int b_stride =
s->b_width <<
s->block_max_depth;
622 const int b_height =
s->b_height<<
s->block_max_depth;
628 int sx= block_w*mb_x - block_w/2;
629 int sy= block_h*mb_y - block_h/2;
630 int x0=
FFMAX(0,-sx);
631 int y0=
FFMAX(0,-sy);
632 int x1=
FFMIN(block_w*2,
w-sx);
633 int y1=
FFMIN(block_h*2,
h-sy);
638 ff_snow_pred_block(
s, cur,
tmp, ref_stride, sx, sy, block_w*2, block_h*2, &
s->block[mb_x + mb_y*b_stride], plane_index,
w,
h);
640 for(y=y0; y<y1; y++){
641 const uint8_t *obmc1= obmc_edged[y];
643 uint8_t *cur1 = cur + y*ref_stride;
644 uint8_t *dst1 = dst + sx + (sy+y)*ref_stride;
645 for(x=x0; x<x1; x++){
646 #if FRAC_BITS >= LOG2_OBMC_MAX
652 if(v&(~255)) v= ~(v>>31);
659 && (mb_x == 0 || mb_x == b_stride-1)
660 && (mb_y == 0 || mb_y == b_height-1)){
670 memcpy(dst + sx+x0 + (sy+y)*ref_stride, cur + x0 + y*ref_stride, x1-x0);
681 distortion =
ff_w97_32_c(&
s->m,
src + sx + sy*ref_stride, dst + sx + sy*ref_stride, ref_stride, 32);
683 distortion =
ff_w53_32_c(&
s->m,
src + sx + sy*ref_stride, dst + sx + sy*ref_stride, ref_stride, 32);
687 int off = sx+16*(
i&1) + (sy+16*(
i>>1))*ref_stride;
688 distortion +=
s->mecc.me_cmp[0](&
s->m,
src + off, dst + off, ref_stride, 16);
693 distortion =
s->mecc.me_cmp[0](&
s->m,
src + sx + sy*ref_stride, dst + sx + sy*ref_stride, ref_stride, block_w*2);
704 if(mb_x == b_stride-2)
707 return distortion + rate*penalty_factor;
712 Plane *p= &
s->plane[plane_index];
713 const int block_size =
MB_SIZE >>
s->block_max_depth;
714 const int block_w = plane_index ? block_size>>
s->chroma_h_shift : block_size;
715 const int block_h = plane_index ? block_size>>
s->chroma_v_shift : block_size;
717 const int obmc_stride= plane_index ? (2*block_size)>>
s->chroma_h_shift : 2*block_size;
718 const int ref_stride=
s->current_picture->linesize[plane_index];
719 uint8_t *dst=
s->current_picture->data[plane_index];
720 uint8_t *
src=
s-> input_picture->data[plane_index];
724 const int b_stride =
s->b_width <<
s->block_max_depth;
734 int mb_x2= mb_x + (
i%3) - 1;
735 int mb_y2= mb_y + (
i/3) - 1;
736 int x= block_w*mb_x2 + block_w/2;
737 int y= block_h*mb_y2 + block_h/2;
740 x, y, block_w, block_h,
w,
h, 0, ref_stride, obmc_stride, mb_x2, mb_y2, 1, 1, plane_index);
743 for(y2= y; y2<0; y2++)
744 memcpy(dst + x + y2*ref_stride,
src + x + y2*ref_stride, block_w);
745 for(y2=
h; y2<y+block_h; y2++)
746 memcpy(dst + x + y2*ref_stride,
src + x + y2*ref_stride, block_w);
748 for(y2= y; y2<y+block_h; y2++)
749 memcpy(dst + x + y2*ref_stride,
src + x + y2*ref_stride, -x);
752 for(y2= y; y2<y+block_h; y2++)
753 memcpy(dst +
w + y2*ref_stride,
src +
w + y2*ref_stride, x+block_w -
w);
757 distortion +=
s->mecc.me_cmp[block_w==8](&
s->m,
src + x + y*ref_stride, dst + x + y*ref_stride, ref_stride, block_h);
771 for(
i=merged?4:0;
i<9;
i++){
772 static const int dxy[9][2] = {{0,0},{1,0},{0,1},{1,1},{2,0},{2,1},{-1,2},{0,2},{1,2}};
776 return distortion + rate*penalty_factor;
780 const int w=
b->width;
781 const int h=
b->height;
786 int *runs =
s->run_buffer;
793 int l=0, lt=0, t=0, rt=0;
815 if(px<b->parent->width && py<b->parent->height)
816 p= parent[px + py*2*
stride];
820 runs[run_index++]=
run;
828 max_index= run_index;
829 runs[run_index++]=
run;
831 run= runs[run_index++];
834 if(run_index <= max_index)
838 if(
s->c.bytestream_end -
s->c.bytestream <
w*40){
844 int l=0, lt=0, t=0, rt=0;
866 if(px<b->parent->width && py<b->parent->height)
867 p= parent[px + py*2*
stride];
872 put_rac(&
s->c, &
b->state[0][context], !!v);
875 run= runs[run_index++];
877 if(run_index <= max_index)
887 int l2= 2*
FFABS(l) + (l<0);
907 const int b_stride=
s->b_width <<
s->block_max_depth;
917 block->color[0] = p[0];
918 block->color[1] = p[1];
919 block->color[2] = p[2];
923 value=
s->me_cache_generation + (p[0]>>10) + (p[1]<<6) + (
block->ref<<12);
933 rd=
get_block_rd(
s, mb_x, mb_y, 0, obmc_edged) +
s->intra_penalty * !!intra;
949 return check_block(
s, mb_x, mb_y, p, 0, obmc_edged, best_rd);
953 const int b_stride=
s->b_width <<
s->block_max_depth;
962 backup[0] =
block[0];
963 backup[1] =
block[1];
964 backup[2] =
block[b_stride];
965 backup[3] =
block[b_stride + 1];
972 value=
s->me_cache_generation + (p0>>10) + (p1<<6) + (
block->ref<<12);
992 block[b_stride]= backup[2];
993 block[b_stride+1]= backup[3];
999 int pass, mb_x, mb_y;
1000 const int b_width =
s->b_width <<
s->block_max_depth;
1001 const int b_height=
s->b_height <<
s->block_max_depth;
1002 const int b_stride= b_width;
1008 memcpy(
state,
s->block_state,
sizeof(
s->block_state));
1009 for(mb_y= 0; mb_y<
s->b_height; mb_y++)
1010 for(mb_x= 0; mb_x<
s->b_width; mb_x++)
1013 memcpy(
s->block_state,
state,
sizeof(
s->block_state));
1019 for(mb_y= 0; mb_y<b_height; mb_y++){
1020 for(mb_x= 0; mb_x<b_width; mb_x++){
1021 int dia_change,
i, j,
ref;
1022 int best_rd= INT_MAX, ref_rd;
1024 const int index= mb_x + mb_y * b_stride;
1034 const int b_w= (
MB_SIZE >>
s->block_max_depth);
1043 if(!
s->me_cache_generation)
1044 memset(
s->me_cache, 0,
sizeof(
s->me_cache));
1045 s->me_cache_generation += 1<<22;
1050 for (y = 0; y < b_w * 2; y++)
1051 memcpy(obmc_edged[y],
ff_obmc_tab[
s->block_max_depth] + y * b_w * 2, b_w * 2);
1053 for(y=0; y<b_w*2; y++)
1054 memset(obmc_edged[y], obmc_edged[y][0] + obmc_edged[y][b_w-1], b_w);
1055 if(mb_x==b_stride-1)
1056 for(y=0; y<b_w*2; y++)
1057 memset(obmc_edged[y]+b_w, obmc_edged[y][b_w] + obmc_edged[y][b_w*2-1], b_w);
1059 for(x=0; x<b_w*2; x++)
1060 obmc_edged[0][x] += obmc_edged[b_w-1][x];
1061 for(y=1; y<b_w; y++)
1062 memcpy(obmc_edged[y], obmc_edged[0], b_w*2);
1064 if(mb_y==b_height-1){
1065 for(x=0; x<b_w*2; x++)
1066 obmc_edged[b_w*2-1][x] += obmc_edged[b_w][x];
1067 for(y=b_w; y<b_w*2-1; y++)
1068 memcpy(obmc_edged[y], obmc_edged[b_w*2-1], b_w*2);
1073 if(mb_x==0 || mb_y==0 || mb_x==b_width-1 || mb_y==b_height-1){
1075 uint8_t *dst=
s->current_picture->data[0];
1076 const int stride=
s->current_picture->linesize[0];
1077 const int block_w=
MB_SIZE >>
s->block_max_depth;
1078 const int block_h=
MB_SIZE >>
s->block_max_depth;
1079 const int sx= block_w*mb_x - block_w/2;
1080 const int sy= block_h*mb_y - block_h/2;
1081 const int w=
s->plane[0].width;
1082 const int h=
s->plane[0].height;
1087 for(y=
h; y<sy+block_h*2; y++)
1090 for(y=sy; y<sy+block_h*2; y++)
1093 if(sx+block_w*2 >
w){
1094 for(y=sy; y<sy+block_h*2; y++)
1100 for(
i=0;
i <
s->nb_planes;
i++)
1106 check_block(
s, mb_x, mb_y, color0, 1, obmc_edged, &best_rd);
1113 int16_t (*mvr)[2]= &
s->ref_mvs[
ref][
index];
1122 check_block_inter(
s, mb_x, mb_y, mvr[-b_stride][0], mvr[-b_stride][1], obmc_edged, &best_rd);
1128 check_block_inter(
s, mb_x, mb_y, mvr[b_stride][0], mvr[b_stride][1], obmc_edged, &best_rd);
1133 int newx =
block->mx;
1134 int newy =
block->my;
1135 int dia_size =
s->iterative_dia_size ?
s->iterative_dia_size :
FFMAX(
s->avctx->dia_size, 1);
1137 for(
i=0;
i < dia_size;
i++){
1139 dia_change |=
check_block_inter(
s, mb_x, mb_y, newx+4*(
i-j), newy+(4*j), obmc_edged, &best_rd);
1140 dia_change |=
check_block_inter(
s, mb_x, mb_y, newx-4*(
i-j), newy-(4*j), obmc_edged, &best_rd);
1141 dia_change |=
check_block_inter(
s, mb_x, mb_y, newx-(4*j), newy+4*(
i-j), obmc_edged, &best_rd);
1142 dia_change |=
check_block_inter(
s, mb_x, mb_y, newx+(4*j), newy-4*(
i-j), obmc_edged, &best_rd);
1148 static const int square[8][2]= {{+1, 0},{-1, 0},{ 0,+1},{ 0,-1},{+1,+1},{-1,-1},{+1,-1},{-1,+1},};
1155 mvr[0][0]=
block->mx;
1156 mvr[0][1]=
block->my;
1157 if(ref_rd > best_rd){
1184 if(
s->block_max_depth == 1){
1186 for(mb_y= 0; mb_y<b_height; mb_y+=2){
1187 for(mb_x= 0; mb_x<b_width; mb_x+=2){
1189 int best_rd, init_rd;
1190 const int index= mb_x + mb_y * b_stride;
1195 b[2]=
b[0]+b_stride;
1202 if(!
s->me_cache_generation)
1203 memset(
s->me_cache, 0,
sizeof(
s->me_cache));
1204 s->me_cache_generation += 1<<22;
1210 (
b[0]->mx +
b[1]->mx +
b[2]->mx +
b[3]->mx + 2) >> 2,
1211 (
b[0]->my +
b[1]->my +
b[2]->my +
b[3]->my + 2) >> 2, 0, &best_rd);
1217 if(init_rd != best_rd)
1248 const int w=
b->width;
1249 const int h=
b->height;
1252 int x,y, thres1, thres2;
1261 bias= bias ? 0 : (3*qmul)>>3;
1270 if((
unsigned)(
i+thres1) > thres2){
1290 if((
unsigned)(
i+thres1) > thres2){
1293 i= (
i + bias) / qmul;
1298 i= (
i + bias) / qmul;
1309 const int w=
b->width;
1310 const int h=
b->height;
1331 const int w=
b->width;
1332 const int h=
b->height;
1335 for(y=
h-1; y>=0; y--){
1336 for(x=
w-1; x>=0; x--){
1355 const int w=
b->width;
1356 const int h=
b->height;
1379 int plane_index,
level, orientation;
1381 for(plane_index=0; plane_index<
FFMIN(
s->nb_planes, 2); plane_index++){
1383 for(orientation=
level ? 1:0; orientation<4; orientation++){
1384 if(orientation==2)
continue;
1385 put_symbol(&
s->c,
s->header_state,
s->plane[plane_index].band[
level][orientation].qlog, 1);
1395 memset(kstate,
MID_STATE,
sizeof(kstate));
1398 if(
s->keyframe ||
s->always_reset){
1400 s->last_spatial_decomposition_type=
1404 s->last_block_max_depth= 0;
1405 for(plane_index=0; plane_index<2; plane_index++){
1406 Plane *p= &
s->plane[plane_index];
1414 put_rac(&
s->c,
s->header_state,
s->always_reset);
1415 put_symbol(&
s->c,
s->header_state,
s->temporal_decomposition_type, 0);
1416 put_symbol(&
s->c,
s->header_state,
s->temporal_decomposition_count, 0);
1417 put_symbol(&
s->c,
s->header_state,
s->spatial_decomposition_count, 0);
1419 if (
s->nb_planes > 2) {
1423 put_rac(&
s->c,
s->header_state,
s->spatial_scalability);
1432 for(plane_index=0; plane_index<
FFMIN(
s->nb_planes, 2); plane_index++){
1433 Plane *p= &
s->plane[plane_index];
1438 put_rac(&
s->c,
s->header_state, update_mc);
1440 for(plane_index=0; plane_index<
FFMIN(
s->nb_planes, 2); plane_index++){
1441 Plane *p= &
s->plane[plane_index];
1448 if(
s->last_spatial_decomposition_count !=
s->spatial_decomposition_count){
1450 put_symbol(&
s->c,
s->header_state,
s->spatial_decomposition_count, 0);
1456 put_symbol(&
s->c,
s->header_state,
s->spatial_decomposition_type -
s->last_spatial_decomposition_type, 1);
1458 put_symbol(&
s->c,
s->header_state,
s->mv_scale -
s->last_mv_scale, 1);
1459 put_symbol(&
s->c,
s->header_state,
s->qbias -
s->last_qbias , 1);
1460 put_symbol(&
s->c,
s->header_state,
s->block_max_depth -
s->last_block_max_depth, 1);
1468 for(plane_index=0; plane_index<2; plane_index++){
1469 Plane *p= &
s->plane[plane_index];
1476 s->last_spatial_decomposition_type =
s->spatial_decomposition_type;
1477 s->last_qlog =
s->qlog;
1478 s->last_qbias =
s->qbias;
1479 s->last_mv_scale =
s->mv_scale;
1480 s->last_block_max_depth =
s->block_max_depth;
1481 s->last_spatial_decomposition_count =
s->spatial_decomposition_count;
1494 uint32_t coef_sum= 0;
1495 int level, orientation, delta_qlog;
1498 for(orientation=
level ? 1 : 0; orientation<4; orientation++){
1501 const int w=
b->width;
1502 const int h=
b->height;
1506 const int qdiv= (1<<16)/qmul;
1516 coef_sum+=
abs(buf[x+y*
stride]) * qdiv >> 16;
1522 coef_sum = (uint64_t)coef_sum * coef_sum >> 16;
1525 s->m.current_picture.mb_var_sum= coef_sum;
1526 s->m.current_picture.mc_mb_var_sum= 0;
1528 s->m.current_picture.mc_mb_var_sum= coef_sum;
1529 s->m.current_picture.mb_var_sum= 0;
1537 s->qlog+= delta_qlog;
1544 int level, orientation, x, y;
1548 for(orientation=
level ? 1 : 0; orientation<4; orientation++){
1552 memset(
s->spatial_idwt_buffer, 0,
sizeof(*
s->spatial_idwt_buffer)*
width*
height);
1553 ibuf[
b->width/2 +
b->height/2*
b->stride]= 256*16;
1556 for(x=0; x<
width; x++){
1557 int64_t d=
s->spatial_idwt_buffer[x + y*
width]*16;
1561 if (orientation == 2)
1564 if (orientation != 1)
1572 const AVFrame *pict,
int *got_packet)
1577 const int width=
s->avctx->width;
1578 const int height=
s->avctx->height;
1579 int level, orientation, plane_index,
i, y, ret;
1580 uint8_t rc_header_bak[
sizeof(
s->header_state)];
1581 uint8_t rc_block_bak[
sizeof(
s->block_state)];
1589 for(
i=0;
i <
s->nb_planes;
i++){
1590 int hshift=
i ?
s->chroma_h_shift : 0;
1591 int vshift=
i ?
s->chroma_v_shift : 0;
1593 memcpy(&
s->input_picture->data[
i][y *
s->input_picture->linesize[
i]],
1596 s->mpvencdsp.draw_edges(
s->input_picture->data[
i],
s->input_picture->linesize[
i],
1603 pic =
s->input_picture;
1632 #if FF_API_CODED_FRAME
1638 if (
s->current_picture->data[0]) {
1639 int w =
s->avctx->width;
1640 int h =
s->avctx->height;
1642 #if FF_API_CODED_FRAME
1648 s->mpvencdsp.draw_edges(
s->current_picture->data[0],
1649 s->current_picture->linesize[0],
w ,
h ,
1651 if (
s->current_picture->data[2]) {
1652 s->mpvencdsp.draw_edges(
s->current_picture->data[1],
1653 s->current_picture->linesize[1],
w>>
s->chroma_h_shift,
h>>
s->chroma_v_shift,
1655 s->mpvencdsp.draw_edges(
s->current_picture->data[2],
1656 s->current_picture->linesize[2],
w>>
s->chroma_h_shift,
h>>
s->chroma_v_shift,
1663 #if FF_API_CODED_FRAME
1671 s->m.current_picture_ptr= &
s->m.current_picture;
1672 s->m.current_picture.f =
s->current_picture;
1673 s->m.current_picture.f->pts = pict->
pts;
1675 int block_width = (
width +15)>>4;
1676 int block_height= (
height+15)>>4;
1677 int stride=
s->current_picture->linesize[0];
1682 s->m.avctx=
s->avctx;
1684 s->m. new_picture.f =
s->input_picture;
1687 s->m.uvlinesize=
s->current_picture->linesize[1];
1690 s->m.mb_width = block_width;
1691 s->m.mb_height= block_height;
1692 s->m.mb_stride=
s->m.mb_width+1;
1693 s->m.b8_stride= 2*
s->m.mb_width+1;
1696 s->m.motion_est=
s->motion_est;
1697 s->m.me.scene_change_score=0;
1701 s->m.unrestricted_mv= 1;
1703 s->m.lambda =
s->lambda;
1709 s->m.hdsp =
s->hdsp;
1711 s->hdsp =
s->m.hdsp;
1716 memcpy(rc_header_bak,
s->header_state,
sizeof(
s->header_state));
1717 memcpy(rc_block_bak,
s->block_state,
sizeof(
s->block_state));
1722 s->spatial_decomposition_count= 5;
1724 while( !(
width >>(
s->chroma_h_shift +
s->spatial_decomposition_count))
1725 || !(
height>>(
s->chroma_v_shift +
s->spatial_decomposition_count)))
1726 s->spatial_decomposition_count--;
1728 if (
s->spatial_decomposition_count <= 0) {
1738 if(
s->last_spatial_decomposition_count !=
s->spatial_decomposition_count){
1739 for(plane_index=0; plane_index <
s->nb_planes; plane_index++){
1745 s->m.misc_bits = 8*(
s->c.bytestream -
s->c.bytestream_start);
1747 s->m.mv_bits = 8*(
s->c.bytestream -
s->c.bytestream_start) -
s->m.misc_bits;
1749 for(plane_index=0; plane_index < s->nb_planes; plane_index++){
1750 Plane *p= &
s->plane[plane_index];
1756 if (!
s->memc_only) {
1758 if(pict->
data[plane_index])
1766 #if FF_API_PRIVATE_OPT
1768 if(
s->avctx->scenechange_threshold)
1769 s->scenechange_threshold =
s->avctx->scenechange_threshold;
1776 &&
s->m.me.scene_change_score >
s->scenechange_threshold){
1781 s->current_picture->key_frame=1;
1799 ff_spatial_dwt(
s->spatial_dwt_buffer,
s->temp_dwt_buffer,
w,
h,
w,
s->spatial_decomposition_type,
s->spatial_decomposition_count);
1801 if(
s->pass1_rc && plane_index==0){
1803 if (delta_qlog <= INT_MIN)
1808 memcpy(
s->header_state, rc_header_bak,
sizeof(
s->header_state));
1809 memcpy(
s->block_state, rc_block_bak,
sizeof(
s->block_state));
1816 for(orientation=
level ? 1 : 0; orientation<4; orientation++){
1822 if (!
s->no_bitstream)
1831 for(orientation=
level ? 1 : 0; orientation<4; orientation++){
1838 ff_spatial_idwt(
s->spatial_idwt_buffer,
s->temp_idwt_buffer,
w,
h,
w,
s->spatial_decomposition_type,
s->spatial_decomposition_count);
1852 s->current_picture->data[plane_index][y*
s->current_picture->linesize[plane_index] + x]=
1853 pict->
data[plane_index][y*pict->
linesize[plane_index] + x];
1857 memset(
s->spatial_idwt_buffer, 0,
sizeof(
IDWTELEM)*
w*
h);
1864 if(pict->
data[plane_index])
1867 int d=
s->current_picture->data[plane_index][y*
s->current_picture->linesize[plane_index] + x] - pict->
data[plane_index][y*pict->
linesize[plane_index] + x];
1871 s->avctx->error[plane_index] +=
error;
1872 s->encoding_error[plane_index] =
error;
1882 s->current_picture->coded_picture_number = avctx->
frame_number;
1883 s->current_picture->pict_type = pic->
pict_type;
1884 s->current_picture->quality = pic->
quality;
1885 s->m.frame_bits = 8*(
s->c.bytestream -
s->c.bytestream_start);
1886 s->m.p_tex_bits =
s->m.frame_bits -
s->m.misc_bits -
s->m.mv_bits;
1887 s->m.current_picture.f->display_picture_number =
1888 s->m.current_picture.f->coded_picture_number = avctx->
frame_number;
1889 s->m.current_picture.f->quality = pic->
quality;
1890 s->m.total_bits += 8*(
s->c.bytestream -
s->c.bytestream_start);
1896 s->m.last_pict_type =
s->m.pict_type;
1897 #if FF_API_STAT_BITS
1911 s->current_picture->pict_type);
1913 #if FF_API_ERROR_FRAME
1915 memcpy(
s->current_picture->error,
s->encoding_error,
sizeof(
s->encoding_error));
1920 if (
s->current_picture->key_frame)
1939 #define OFFSET(x) offsetof(SnowContext, x)
1940 #define VE AV_OPT_FLAG_VIDEO_PARAM | AV_OPT_FLAG_ENCODING_PARAM
1947 {
"memc_only",
"Only do ME/MC (I frames -> ref, P frame -> ME+MC).",
OFFSET(memc_only),
AV_OPT_TYPE_BOOL, { .i64 = 0 }, 0, 1,
VE },
1948 {
"no_bitstream",
"Skip final bitstream writeout.",
OFFSET(no_bitstream),
AV_OPT_TYPE_BOOL, { .i64 = 0 }, 0, 1,
VE },
1949 {
"intra_penalty",
"Penalty for intra blocks in block decission",
OFFSET(intra_penalty),
AV_OPT_TYPE_INT, { .i64 = 0 }, 0, INT_MAX,
VE },
1950 {
"iterative_dia_size",
"Dia size for the iterative ME",
OFFSET(iterative_dia_size),
AV_OPT_TYPE_INT, { .i64 = 0 }, 0, INT_MAX,
VE },
1951 {
"sc_threshold",
"Scene change threshold",
OFFSET(scenechange_threshold),
AV_OPT_TYPE_INT, { .i64 = 0 }, INT_MIN, INT_MAX,
VE },
1955 {
"rc_eq",
"Set rate control equation. When computing the expression, besides the standard functions "
1956 "defined in the section 'Expression Evaluation', the following functions are available: "
1957 "bits2qp(bits), qp2bits(qp). Also the following constants are available: iTex pTex tex mv "
1958 "fCode iCount mcVar var isI isP isB avgQP qComp avgIITex avgPITex avgPPTex avgBPTex avgTex.",
static uint32_t inverse(uint32_t v)
find multiplicative inverse modulo 2 ^ 32
#define av_assert2(cond)
assert() equivalent, that does lie in speed critical code.
#define av_assert1(cond)
assert() equivalent, that does not lie in speed critical code.
#define av_assert0(cond)
assert() equivalent, that is always enabled.
Libavcodec external API header.
int ff_side_data_set_encoder_stats(AVPacket *pkt, int quality, int64_t *error, int error_count, int pict_type)
static av_cold int init(AVCodecContext *avctx)
#define AV_CEIL_RSHIFT(a, b)
#define ROUNDED_DIV(a, b)
#define FFABS(a)
Absolute value, Note, INT_MIN / INT64_MIN result in undefined behavior as they are not representable ...
int ff_alloc_packet2(AVCodecContext *avctx, AVPacket *avpkt, int64_t size, int64_t min_size)
Check AVPacket size and/or allocate data.
static av_noinline void put_symbol(RangeCoder *c, uint8_t *state, int v, int is_signed)
#define AV_CODEC_FLAG_QPEL
Use qpel MC.
#define AV_CODEC_FLAG_PASS2
Use internal 2pass ratecontrol in second pass mode.
#define AV_CODEC_FLAG_QSCALE
Use fixed qscale.
#define AV_CODEC_FLAG_PASS1
Use internal 2pass ratecontrol in first pass mode.
#define AV_CODEC_FLAG_PSNR
error[?] variables will be set during encoding.
#define AV_CODEC_FLAG_4MV
4 MV per MB allowed / advanced prediction for H.263.
#define AV_INPUT_BUFFER_MIN_SIZE
minimum encoding buffer size Used to avoid some checks during header writing.
#define AV_PKT_FLAG_KEY
The packet contains a keyframe.
#define FF_QP2LAMBDA
factor to convert from H.263 QP to lambda
#define AVERROR_PATCHWELCOME
Not yet implemented in FFmpeg, patches welcome.
void av_frame_unref(AVFrame *frame)
Unreference all the buffers referenced by frame and reset the frame fields.
int av_frame_ref(AVFrame *dst, const AVFrame *src)
Set up a new reference to the data described by the source frame.
void av_frame_free(AVFrame **frame)
Free the frame and any dynamically allocated objects in it, e.g.
AVFrame * av_frame_alloc(void)
Allocate an AVFrame and set its fields to default values.
int av_frame_make_writable(AVFrame *frame)
Ensure that the frame data is writable, avoiding data copy if possible.
#define AV_LOG_DEBUG
Stuff which is only useful for libav* developers.
#define AV_LOG_ERROR
Something went wrong and cannot losslessly be recovered.
const char * av_default_item_name(void *ptr)
Return the context name.
void * av_mallocz(size_t size)
Allocate a memory block with alignment suitable for all memory accesses (including vectors if availab...
void * av_mallocz_array(size_t nmemb, size_t size)
Allocate a memory block for an array with av_mallocz().
@ AV_PICTURE_TYPE_I
Intra.
@ AV_PICTURE_TYPE_P
Predicted.
#define LIBAVUTIL_VERSION_INT
void ff_h263_encode_init(MpegEncContext *s)
static void pred_mv(DiracBlock *block, int stride, int x, int y, int ref)
#define FF_CODEC_CAP_INIT_THREADSAFE
The codec does not modify any global variables in the init function, allowing to call the init functi...
#define FF_CODEC_CAP_INIT_CLEANUP
The codec allows calling the close function for deallocation even if the init function returned a fai...
common internal API header
#define NULL_IF_CONFIG_SMALL(x)
Return NULL if CONFIG_SMALL is true, otherwise the argument without modification.
#define FF_DISABLE_DEPRECATION_WARNINGS
#define FF_ENABLE_DEPRECATION_WARNINGS
static enum AVPixelFormat pix_fmts[]
Replacements for frequently missing libm functions.
const uint32_t ff_square_tab[512]
void ff_set_cmp(MECmpContext *c, me_cmp_func *cmp, int type)
int ff_init_me(MpegEncContext *s)
static void init_ref(MotionEstContext *c, uint8_t *src[3], uint8_t *ref[3], uint8_t *ref2[3], int x, int y, int ref_index)
int ff_epzs_motion_search(struct MpegEncContext *s, int *mx_ptr, int *my_ptr, int P[10][2], int src_index, int ref_index, int16_t(*last_mv)[2], int ref_mv_scale, int size, int h)
int ff_get_mb_score(struct MpegEncContext *s, int mx, int my, int src_index, int ref_index, int size, int h, int add_rate)
av_cold void ff_mpegvideoencdsp_init(MpegvideoEncDSPContext *c, AVCodecContext *avctx)
int av_pix_fmt_get_chroma_sub_sample(enum AVPixelFormat pix_fmt, int *h_shift, int *v_shift)
Utility function to access log2_chroma_w log2_chroma_h from the pixel format AVPixFmtDescriptor.
AVPixelFormat
Pixel format.
@ AV_PIX_FMT_YUV420P
planar YUV 4:2:0, 12bpp, (1 Cr & Cb sample per 2x2 Y samples)
@ AV_PIX_FMT_GRAY8
Y , 8bpp.
@ AV_PIX_FMT_YUV410P
planar YUV 4:1:0, 9bpp, (1 Cr & Cb sample per 4x4 Y samples)
@ AV_PIX_FMT_YUV444P
planar YUV 4:4:4, 24bpp, (1 Cr & Cb sample per 1x1 Y samples)
int ff_rac_terminate(RangeCoder *c, int version)
Terminates the range coder.
void ff_build_rac_states(RangeCoder *c, int factor, int max_p)
av_cold void ff_init_range_encoder(RangeCoder *c, uint8_t *buf, int buf_size)
static int get_rac_count(RangeCoder *c)
av_cold int ff_rate_control_init(MpegEncContext *s)
av_cold void ff_rate_control_uninit(MpegEncContext *s)
float ff_rate_estimate_qscale(MpegEncContext *s, int dry_run)
void ff_write_pass1_stats(MpegEncContext *s)
static const float pred[4]
av_cold int ff_snow_common_init(AVCodecContext *avctx)
void ff_snow_pred_block(SnowContext *s, uint8_t *dst, uint8_t *tmp, ptrdiff_t stride, int sx, int sy, int b_w, int b_h, const BlockNode *block, int plane_index, int w, int h)
int ff_snow_get_buffer(SnowContext *s, AVFrame *frame)
void ff_snow_release_buffer(AVCodecContext *avctx)
void ff_snow_reset_contexts(SnowContext *s)
int ff_snow_common_init_after_header(AVCodecContext *avctx)
int ff_snow_frame_start(SnowContext *s)
av_cold void ff_snow_common_end(SnowContext *s)
int ff_snow_alloc_blocks(SnowContext *s)
#define BLOCK_OPT
Block needs no checks in this round of iterative motion estiation.
static av_always_inline void predict_plane(SnowContext *s, IDWTELEM *buf, int plane_index, int add)
static void set_blocks(SnowContext *s, int level, int x, int y, int l, int cb, int cr, int mx, int my, int ref, int type)
const uint8_t *const ff_obmc_tab[4]
static void put_symbol2(RangeCoder *c, uint8_t *state, int v, int log2)
#define ENCODER_EXTRA_BITS
#define BLOCK_INTRA
Intra block, inter otherwise.
static av_always_inline int same_block(BlockNode *a, BlockNode *b)
const int8_t ff_quant3bA[256]
static av_always_inline void add_yblock(SnowContext *s, int sliced, slice_buffer *sb, IDWTELEM *dst, uint8_t *dst8, const uint8_t *obmc, int src_x, int src_y, int b_w, int b_h, int w, int h, int dst_stride, int src_stride, int obmc_stride, int b_x, int b_y, int add, int offset_dst, int plane_index)
static const BlockNode null_block
void ff_spatial_idwt(IDWTELEM *buffer, IDWTELEM *temp, int width, int height, int stride, int type, int decomposition_count)
int ff_w53_32_c(struct MpegEncContext *v, uint8_t *pix1, uint8_t *pix2, ptrdiff_t line_size, int h)
void ff_spatial_dwt(DWTELEM *buffer, DWTELEM *temp, int width, int height, int stride, int type, int decomposition_count)
int ff_w97_32_c(struct MpegEncContext *v, uint8_t *pix1, uint8_t *pix2, ptrdiff_t line_size, int h)
static int get_4block_rd(SnowContext *s, int mb_x, int mb_y, int plane_index)
static av_always_inline int check_4block_inter(SnowContext *s, int mb_x, int mb_y, int p0, int p1, int ref, int *best_rd)
static int qscale2qlog(int qscale)
static const AVClass snowenc_class
static int get_penalty_factor(int lambda, int lambda2, int type)
static const AVOption options[]
static int get_block_rd(SnowContext *s, int mb_x, int mb_y, int plane_index, uint8_t(*obmc_edged)[MB_SIZE *2])
static int encode_subband_c0run(SnowContext *s, SubBand *b, const IDWTELEM *src, const IDWTELEM *parent, int stride, int orientation)
static void encode_q_branch2(SnowContext *s, int level, int x, int y)
static void calculate_visual_weight(SnowContext *s, Plane *p)
static void correlate(SnowContext *s, SubBand *b, IDWTELEM *src, int stride, int inverse, int use_median)
static int get_block_bits(SnowContext *s, int x, int y, int w)
static av_cold int encode_init(AVCodecContext *avctx)
static av_always_inline int check_block(SnowContext *s, int mb_x, int mb_y, int p[3], int intra, uint8_t(*obmc_edged)[MB_SIZE *2], int *best_rd)
static int pix_sum(uint8_t *pix, int line_size, int w, int h)
static int ratecontrol_1pass(SnowContext *s, AVFrame *pict)
static int get_dc(SnowContext *s, int mb_x, int mb_y, int plane_index)
static int encode_frame(AVCodecContext *avctx, AVPacket *pkt, const AVFrame *pict, int *got_packet)
static void encode_qlogs(SnowContext *s)
static av_always_inline int check_block_inter(SnowContext *s, int mb_x, int mb_y, int p0, int p1, uint8_t(*obmc_edged)[MB_SIZE *2], int *best_rd)
static int encode_q_branch(SnowContext *s, int level, int x, int y)
static void decorrelate(SnowContext *s, SubBand *b, IDWTELEM *src, int stride, int inverse, int use_median)
static av_cold int encode_end(AVCodecContext *avctx)
static void encode_header(SnowContext *s)
static void dequantize(SnowContext *s, SubBand *b, IDWTELEM *src, int stride)
static void quantize(SnowContext *s, SubBand *b, IDWTELEM *dst, DWTELEM *src, int stride, int bias)
static void iterative_me(SnowContext *s)
static void update_last_header_values(SnowContext *s)
static int pix_norm1(uint8_t *pix, int line_size, int w)
static void encode_blocks(SnowContext *s, int search)
static int encode_subband(SnowContext *s, SubBand *b, const IDWTELEM *src, const IDWTELEM *parent, int stride, int orientation)
static int shift(int a, int b)
Describe the class of an AVClass context structure.
const char * class_name
The name of the class; usually it is the same name as the context structure type to which the AVClass...
main external API structure.
enum AVPixelFormat pix_fmt
Pixel format, see AV_PIX_FMT_xxx.
int width
picture width / height.
char * stats_out
pass1 encoding statistics output buffer
int global_quality
Global quality for codecs which cannot change it per frame.
attribute_deprecated int frame_bits
attribute_deprecated int mv_bits
int mb_lmax
maximum MB Lagrange multiplier
int dia_size
ME diamond size & shape.
attribute_deprecated AVFrame * coded_frame
the picture in the bitstream
attribute_deprecated int misc_bits
int64_t bit_rate
the average bitrate
int gop_size
the number of pictures in a group of pictures, or 0 for intra_only
int frame_number
Frame counter, set by libavcodec.
int refs
number of reference frames
int mb_lmin
minimum MB Lagrange multiplier
int flags
AV_CODEC_FLAG_*.
attribute_deprecated int prediction_method
attribute_deprecated int p_tex_bits
const char * name
Name of the codec implementation.
This structure describes decoded (raw) audio or video data.
int64_t pts
Presentation timestamp in time_base units (time when frame should be shown to user).
uint8_t * data[AV_NUM_DATA_POINTERS]
pointer to the picture/channel planes.
int quality
quality (between 1 (good) and FF_LAMBDA_MAX (bad))
int linesize[AV_NUM_DATA_POINTERS]
For video, size in bytes of each picture line.
enum AVPictureType pict_type
Picture type of the frame.
This structure stores compressed data.
int flags
A combination of AV_PKT_FLAG values.
uint8_t ref
Reference frame index.
int16_t mx
Motion vector component X, see mv_scale.
uint8_t color[3]
Color for intra.
int16_t my
Motion vector component Y, see mv_scale.
uint8_t type
Bitfield of BLOCK_*.
Motion estimation context.
SubBand band[DWT_LEVELS_3D][4]
int8_t last_hcoeff[HTAPS_MAX/2]
int8_t hcoeff[HTAPS_MAX/2]
int qlog
log(qscale)/log[2^(1/6)]
static void error(const char *err)
static int ref[MAX_W *MAX_W]
enum AVPictureType last_picture
static float search(FOCContext *foc, int pass, int maxpass, int xmin, int xmax, int ymin, int ymax, int *best_x, int *best_y, float best_score)
static double cr(void *priv, double x, double y)
static double cb(void *priv, double x, double y)