use the right stride

patch by Vitor: [vitor1001 gmail com]

Originally committed as revision 9443 to svn://svn.ffmpeg.org/ffmpeg/trunk
pull/126/head
Vitor Sessak 18 years ago committed by Benoit Fouet
parent 7cf0999529
commit 670e47e196
  1. 22
      libavcodec/roqvideoenc.c

@ -125,7 +125,7 @@ static inline int eval_sse(uint8_t *a, uint8_t *b, int count)
// FIXME Could use DSPContext.sse, but it is not so speed critical (used // FIXME Could use DSPContext.sse, but it is not so speed critical (used
// just for motion estimation). // just for motion estimation).
static int block_sse(uint8_t **buf1, uint8_t **buf2, int x1, int y1, int x2, static int block_sse(uint8_t **buf1, uint8_t **buf2, int x1, int y1, int x2,
int y2, int stride, int size) int y2, int *stride, int size)
{ {
int i, k; int i, k;
int sse=0; int sse=0;
@ -133,8 +133,8 @@ static int block_sse(uint8_t **buf1, uint8_t **buf2, int x1, int y1, int x2,
for (k=0; k<3; k++) { for (k=0; k<3; k++) {
int bias = (k ? CHROMA_BIAS : 4); int bias = (k ? CHROMA_BIAS : 4);
for (i=0; i<size; i++) for (i=0; i<size; i++)
sse += bias*eval_sse(buf1[k] + (y1+i)*stride + x1, sse += bias*eval_sse(buf1[k] + (y1+i)*stride[k] + x1,
buf2[k] + (y2+i)*stride + x2, size); buf2[k] + (y2+i)*stride[k] + x2, size);
} }
return sse; return sse;
@ -159,7 +159,7 @@ static int eval_motion_dist(RoqContext *enc, int x, int y, motion_vect vect,
return INT_MAX; return INT_MAX;
return block_sse(enc->frame_to_enc->data, enc->last_frame->data, x, y, return block_sse(enc->frame_to_enc->data, enc->last_frame->data, x, y,
mx, my, enc->y_stride, size); mx, my, enc->frame_to_enc->linesize, size);
} }
/** /**
@ -261,12 +261,13 @@ static void create_cel_evals(RoqContext *enc, roq_tempdata_t *tempData)
static void get_frame_mb(AVFrame *frame, int x, int y, uint8_t mb[], int dim) static void get_frame_mb(AVFrame *frame, int x, int y, uint8_t mb[], int dim)
{ {
int i, j, cp; int i, j, cp;
int stride = frame->linesize[0];
for (cp=0; cp<3; cp++) for (cp=0; cp<3; cp++) {
int stride = frame->linesize[cp];
for (i=0; i<dim; i++) for (i=0; i<dim; i++)
for (j=0; j<dim; j++) for (j=0; j<dim; j++)
*mb++ = frame->data[cp][(y+i)*stride + x + j]; *mb++ = frame->data[cp][(y+i)*stride + x + j];
}
} }
/** /**
@ -414,7 +415,9 @@ static void gather_data_for_subcel(subcel_evaluation_t *subcel, int x,
if (enc->framesSinceKeyframe >= 2) if (enc->framesSinceKeyframe >= 2)
subcel->eval_dist[RoQ_ID_MOT] = block_sse(enc->frame_to_enc->data, subcel->eval_dist[RoQ_ID_MOT] = block_sse(enc->frame_to_enc->data,
enc->current_frame->data, x, enc->current_frame->data, x,
y, x, y, enc->y_stride, 4); y, x, y,
enc->frame_to_enc->linesize,
4);
else else
subcel->eval_dist[RoQ_ID_MOT] = INT_MAX; subcel->eval_dist[RoQ_ID_MOT] = INT_MAX;
@ -476,7 +479,7 @@ static void gather_data_for_cel(cel_evaluation_t *cel, RoqContext *enc,
enc->current_frame->data, enc->current_frame->data,
cel->sourceX, cel->sourceY, cel->sourceX, cel->sourceY,
cel->sourceX, cel->sourceY, cel->sourceX, cel->sourceY,
enc->y_stride, 8); enc->frame_to_enc->linesize, 8);
else else
cel->eval_dist[RoQ_ID_MOT] = INT_MAX; cel->eval_dist[RoQ_ID_MOT] = INT_MAX;
@ -737,7 +740,7 @@ static void reconstruct_and_encode_image(RoqContext *enc, roq_tempdata_t *tempDa
av_log(enc->avctx, AV_LOG_ERROR, "Expected distortion: %i Actual: %i\n", av_log(enc->avctx, AV_LOG_ERROR, "Expected distortion: %i Actual: %i\n",
dist, dist,
block_sse(fdata, cdata, 0, 0, 0, 0, block_sse(fdata, cdata, 0, 0, 0, 0,
enc->y_stride, enc->width)); //WARNING: Square dimensions implied... enc->frame_to_enc->linesize, enc->width)); //WARNING: Square dimensions implied...
#endif #endif
} }
@ -754,6 +757,7 @@ static inline void frame_block_to_cell(uint8_t *block, uint8_t **data,
for (j=0; j<2; j++) { for (j=0; j<2; j++) {
int x = (top+i)*stride[0] + left + j; int x = (top+i)*stride[0] + left + j;
*block++ = data[0][x]; *block++ = data[0][x];
x = (top+i)*stride[1] + left + j;
u += data[1][x]; u += data[1][x];
v += data[2][x]; v += data[2][x];
} }

Loading…
Cancel
Save