Mercurial > libavcodec.hg
changeset 5665:834f359e7257 libavcodec
perform init after reading the values needed for init
fixes decoding of files with spatial_decomposition_count!=5
author | michael |
---|---|
date | Sun, 09 Sep 2007 13:47:25 +0000 |
parents | 1f9e8321c2fb |
children | b5c137f3f53a |
files | snow.c |
diffstat | 1 files changed, 35 insertions(+), 29 deletions(-) [+] |
line wrap: on
line diff
--- a/snow.c Sun Sep 09 12:46:25 2007 +0000 +++ b/snow.c Sun Sep 09 13:47:25 2007 +0000 @@ -1656,7 +1656,7 @@ int plane_index, level, orientation; for(plane_index=0; plane_index<3; plane_index++){ - for(level=0; level<s->spatial_decomposition_count; level++){ + for(level=0; level<MAX_DECOMPOSITIONS; level++){ for(orientation=level ? 1:0; orientation<4; orientation++){ memset(s->plane[plane_index].band[level][orientation].state, MID_STATE, sizeof(s->plane[plane_index].band[level][orientation].state)); } @@ -3697,7 +3697,6 @@ static int common_init(AVCodecContext *avctx){ SnowContext *s = avctx->priv_data; int width, height; - int level, orientation, plane_index; int i, j; s->avctx= avctx; @@ -3745,12 +3744,6 @@ if(!qexp[0]) init_qexp(); - s->spatial_decomposition_count= 5; - s->spatial_decomposition_type= avctx->prediction_method; //FIXME add decorrelator type r transform_type - - s->chroma_h_shift= 1; //FIXME XXX - s->chroma_v_shift= 1; - // dec += FFMAX(s->chroma_h_shift, s->chroma_v_shift); width= s->avctx->width; @@ -3759,8 +3752,18 @@ s->spatial_idwt_buffer= av_mallocz(width*height*sizeof(IDWTELEM)); s->spatial_dwt_buffer= av_mallocz(width*height*sizeof(DWTELEM)); //FIXME this doesnt belong here - s->mv_scale= (s->avctx->flags & CODEC_FLAG_QPEL) ? 2 : 4; - s->block_max_depth= (s->avctx->flags & CODEC_FLAG_4MV) ? 1 : 0; + for(i=0; i<MAX_REF_FRAMES; i++) + for(j=0; j<MAX_REF_FRAMES; j++) + scale_mv_ref[i][j] = 256*(i+1)/(j+1); + + s->avctx->get_buffer(s->avctx, &s->mconly_picture); + + return 0; +} + +static int common_init_after_header(AVCodecContext *avctx){ + SnowContext *s = avctx->priv_data; + int plane_index, level, orientation; for(plane_index=0; plane_index<3; plane_index++){ int w= s->avctx->width; @@ -3773,13 +3776,6 @@ s->plane[plane_index].width = w; s->plane[plane_index].height= h; - s->plane[plane_index].diag_mc= 1; - s->plane[plane_index].htaps= 6; - s->plane[plane_index].hcoeff[0]= 40; - s->plane[plane_index].hcoeff[1]= -10; - s->plane[plane_index].hcoeff[2]= 2; - s->plane[plane_index].fast_mc= 1; - //av_log(NULL, AV_LOG_DEBUG, "%d %d\n", w, h); for(level=s->spatial_decomposition_count-1; level>=0; level--){ for(orientation=level ? 1 : 0; orientation<4; orientation++){ @@ -3807,6 +3803,8 @@ if(level) b->parent= &s->plane[plane_index].band[level-1][orientation]; + //FIXME avoid this realloc + av_freep(&b->x_coeff); b->x_coeff=av_mallocz(((b->width+1) * b->height+1)*sizeof(x_and_coeff)); } w= (w+1)>>1; @@ -3814,18 +3812,6 @@ } } - for(i=0; i<MAX_REF_FRAMES; i++) - for(j=0; j<MAX_REF_FRAMES; j++) - scale_mv_ref[i][j] = 256*(i+1)/(j+1); - -/* - width= s->width= avctx->width; - height= s->height= avctx->height; - - assert(width && height); -*/ - s->avctx->get_buffer(s->avctx, &s->mconly_picture); - return 0; } @@ -3931,7 +3917,26 @@ return -1; } + s->spatial_decomposition_count= 5; + s->spatial_decomposition_type= avctx->prediction_method; //FIXME add decorrelator type r transform_type + + s->chroma_h_shift= 1; //FIXME XXX + s->chroma_v_shift= 1; + + s->mv_scale = (avctx->flags & CODEC_FLAG_QPEL) ? 2 : 4; + s->block_max_depth= (avctx->flags & CODEC_FLAG_4MV ) ? 1 : 0; + + for(plane_index=0; plane_index<3; plane_index++){ + s->plane[plane_index].diag_mc= 1; + s->plane[plane_index].htaps= 6; + s->plane[plane_index].hcoeff[0]= 40; + s->plane[plane_index].hcoeff[1]= -10; + s->plane[plane_index].hcoeff[2]= 2; + s->plane[plane_index].fast_mc= 1; + } + common_init(avctx); + common_init_after_header(avctx); alloc_blocks(s); s->version=0; @@ -4414,6 +4419,7 @@ s->current_picture.pict_type= FF_I_TYPE; //FIXME I vs. P decode_header(s); + common_init_after_header(avctx); // realloc slice buffer for the case that spatial_decomposition_count changed slice_buffer_destroy(&s->sb);