From f61e47dd68582529bcf6d42d861c70a320cd1b67 Mon Sep 17 00:00:00 2001 From: Gabriel Dume Date: Mon, 1 Sep 2014 15:18:58 -0400 Subject: asv: K&R formatting cosmetics Signed-off-by: Diego Biurrun --- libavcodec/asvenc.c | 249 ++++++++++++++++++++++++++++++---------------------- 1 file changed, 143 insertions(+), 106 deletions(-) (limited to 'libavcodec/asvenc.c') diff --git a/libavcodec/asvenc.c b/libavcodec/asvenc.c index 9944ffaa7c..98baaec0f0 100644 --- a/libavcodec/asvenc.c +++ b/libavcodec/asvenc.c @@ -32,117 +32,150 @@ #include "mathops.h" #include "mpeg12data.h" -static inline void asv2_put_bits(PutBitContext *pb, int n, int v){ - put_bits(pb, n, ff_reverse[ v << (8-n) ]); +static inline void asv2_put_bits(PutBitContext *pb, int n, int v) +{ + put_bits(pb, n, ff_reverse[v << (8 - n)]); } -static inline void asv1_put_level(PutBitContext *pb, int level){ - unsigned int index= level + 3; +static inline void asv1_put_level(PutBitContext *pb, int level) +{ + unsigned int index = level + 3; - if(index <= 6) put_bits(pb, ff_asv_level_tab[index][1], ff_asv_level_tab[index][0]); - else{ + if (index <= 6) { + put_bits(pb, ff_asv_level_tab[index][1], ff_asv_level_tab[index][0]); + } else { put_bits(pb, ff_asv_level_tab[3][1], ff_asv_level_tab[3][0]); put_sbits(pb, 8, level); } } -static inline void asv2_put_level(PutBitContext *pb, int level){ - unsigned int index= level + 31; +static inline void asv2_put_level(PutBitContext *pb, int level) +{ + unsigned int index = level + 31; - if(index <= 62) put_bits(pb, ff_asv2_level_tab[index][1], ff_asv2_level_tab[index][0]); - else{ + if (index <= 62) { + put_bits(pb, ff_asv2_level_tab[index][1], ff_asv2_level_tab[index][0]); + } else { put_bits(pb, ff_asv2_level_tab[31][1], ff_asv2_level_tab[31][0]); - asv2_put_bits(pb, 8, level&0xFF); + asv2_put_bits(pb, 8, level & 0xFF); } } -static inline void asv1_encode_block(ASV1Context *a, int16_t block[64]){ +static inline void asv1_encode_block(ASV1Context *a, int16_t block[64]) +{ int i; - int nc_count=0; - - put_bits(&a->pb, 8, (block[0] + 32)>>6); - block[0]= 0; - - for(i=0; i<10; i++){ - const int index = ff_asv_scantab[4*i]; - int ccp=0; - - if( (block[index + 0] = (block[index + 0]*a->q_intra_matrix[index + 0] + (1<<15))>>16) ) ccp |= 8; - if( (block[index + 8] = (block[index + 8]*a->q_intra_matrix[index + 8] + (1<<15))>>16) ) ccp |= 4; - if( (block[index + 1] = (block[index + 1]*a->q_intra_matrix[index + 1] + (1<<15))>>16) ) ccp |= 2; - if( (block[index + 9] = (block[index + 9]*a->q_intra_matrix[index + 9] + (1<<15))>>16) ) ccp |= 1; - - if(ccp){ - for(;nc_count; nc_count--) + int nc_count = 0; + + put_bits(&a->pb, 8, (block[0] + 32) >> 6); + block[0] = 0; + + for (i = 0; i < 10; i++) { + const int index = ff_asv_scantab[4 * i]; + int ccp = 0; + + if ((block[index + 0] = (block[index + 0] * + a->q_intra_matrix[index + 0] + (1 << 15)) >> 16)) + ccp |= 8; + if ((block[index + 8] = (block[index + 8] * + a->q_intra_matrix[index + 8] + (1 << 15)) >> 16)) + ccp |= 4; + if ((block[index + 1] = (block[index + 1] * + a->q_intra_matrix[index + 1] + (1 << 15)) >> 16)) + ccp |= 2; + if ((block[index + 9] = (block[index + 9] * + a->q_intra_matrix[index + 9] + (1 << 15)) >> 16)) + ccp |= 1; + + if (ccp) { + for (; nc_count; nc_count--) put_bits(&a->pb, ff_asv_ccp_tab[0][1], ff_asv_ccp_tab[0][0]); put_bits(&a->pb, ff_asv_ccp_tab[ccp][1], ff_asv_ccp_tab[ccp][0]); - if(ccp&8) asv1_put_level(&a->pb, block[index + 0]); - if(ccp&4) asv1_put_level(&a->pb, block[index + 8]); - if(ccp&2) asv1_put_level(&a->pb, block[index + 1]); - if(ccp&1) asv1_put_level(&a->pb, block[index + 9]); - }else{ + if (ccp & 8) + asv1_put_level(&a->pb, block[index + 0]); + if (ccp & 4) + asv1_put_level(&a->pb, block[index + 8]); + if (ccp & 2) + asv1_put_level(&a->pb, block[index + 1]); + if (ccp & 1) + asv1_put_level(&a->pb, block[index + 9]); + } else { nc_count++; } } put_bits(&a->pb, ff_asv_ccp_tab[16][1], ff_asv_ccp_tab[16][0]); } -static inline void asv2_encode_block(ASV1Context *a, int16_t block[64]){ +static inline void asv2_encode_block(ASV1Context *a, int16_t block[64]) +{ int i; - int count=0; + int count = 0; - for(count=63; count>3; count--){ + for (count = 63; count > 3; count--) { const int index = ff_asv_scantab[count]; - - if( (block[index]*a->q_intra_matrix[index] + (1<<15))>>16 ) + if ((block[index] * a->q_intra_matrix[index] + (1 << 15)) >> 16) break; } count >>= 2; asv2_put_bits(&a->pb, 4, count); - asv2_put_bits(&a->pb, 8, (block[0] + 32)>>6); - block[0]= 0; - - for(i=0; i<=count; i++){ - const int index = ff_asv_scantab[4*i]; - int ccp=0; - - if( (block[index + 0] = (block[index + 0]*a->q_intra_matrix[index + 0] + (1<<15))>>16) ) ccp |= 8; - if( (block[index + 8] = (block[index + 8]*a->q_intra_matrix[index + 8] + (1<<15))>>16) ) ccp |= 4; - if( (block[index + 1] = (block[index + 1]*a->q_intra_matrix[index + 1] + (1<<15))>>16) ) ccp |= 2; - if( (block[index + 9] = (block[index + 9]*a->q_intra_matrix[index + 9] + (1<<15))>>16) ) ccp |= 1; - - assert(i || ccp<8); - if(i) put_bits(&a->pb, ff_asv_ac_ccp_tab[ccp][1], ff_asv_ac_ccp_tab[ccp][0]); - else put_bits(&a->pb, ff_asv_dc_ccp_tab[ccp][1], ff_asv_dc_ccp_tab[ccp][0]); - - if(ccp){ - if(ccp&8) asv2_put_level(&a->pb, block[index + 0]); - if(ccp&4) asv2_put_level(&a->pb, block[index + 8]); - if(ccp&2) asv2_put_level(&a->pb, block[index + 1]); - if(ccp&1) asv2_put_level(&a->pb, block[index + 9]); + asv2_put_bits(&a->pb, 8, (block[0] + 32) >> 6); + block[0] = 0; + + for (i = 0; i <= count; i++) { + const int index = ff_asv_scantab[4 * i]; + int ccp = 0; + + if ((block[index + 0] = (block[index + 0] * + a->q_intra_matrix[index + 0] + (1 << 15)) >> 16)) + ccp |= 8; + if ((block[index + 8] = (block[index + 8] * + a->q_intra_matrix[index + 8] + (1 << 15)) >> 16)) + ccp |= 4; + if ((block[index + 1] = (block[index + 1] * + a->q_intra_matrix[index + 1] + (1 << 15)) >> 16)) + ccp |= 2; + if ((block[index + 9] = (block[index + 9] * + a->q_intra_matrix[index + 9] + (1 << 15)) >> 16)) + ccp |= 1; + + assert(i || ccp < 8); + if (i) + put_bits(&a->pb, ff_asv_ac_ccp_tab[ccp][1], ff_asv_ac_ccp_tab[ccp][0]); + else + put_bits(&a->pb, ff_asv_dc_ccp_tab[ccp][1], ff_asv_dc_ccp_tab[ccp][0]); + + if (ccp) { + if (ccp & 8) + asv2_put_level(&a->pb, block[index + 0]); + if (ccp & 4) + asv2_put_level(&a->pb, block[index + 8]); + if (ccp & 2) + asv2_put_level(&a->pb, block[index + 1]); + if (ccp & 1) + asv2_put_level(&a->pb, block[index + 9]); } } } -#define MAX_MB_SIZE (30*16*16*3/2/8) +#define MAX_MB_SIZE (30 * 16 * 16 * 3 / 2 / 8) -static inline int encode_mb(ASV1Context *a, int16_t block[6][64]){ +static inline int encode_mb(ASV1Context *a, int16_t block[6][64]) +{ int i; - if (a->pb.buf_end - a->pb.buf - (put_bits_count(&a->pb)>>3) < MAX_MB_SIZE) { + if (a->pb.buf_end - a->pb.buf - (put_bits_count(&a->pb) >> 3) < MAX_MB_SIZE) { av_log(a->avctx, AV_LOG_ERROR, "encoded frame too large\n"); return -1; } - if(a->avctx->codec_id == AV_CODEC_ID_ASV1){ - for(i=0; i<6; i++) + if (a->avctx->codec_id == AV_CODEC_ID_ASV1) { + for (i = 0; i < 6; i++) asv1_encode_block(a, block[i]); - }else{ - for(i=0; i<6; i++) + } else { + for (i = 0; i < 6; i++) asv2_encode_block(a, block[i]); } return 0; @@ -151,25 +184,25 @@ static inline int encode_mb(ASV1Context *a, int16_t block[6][64]){ static inline void dct_get(ASV1Context *a, const AVFrame *frame, int mb_x, int mb_y) { - int16_t (*block)[64]= a->block; + int16_t (*block)[64] = a->block; int linesize = frame->linesize[0]; int i; - uint8_t *ptr_y = frame->data[0] + (mb_y * 16* linesize ) + mb_x * 16; - uint8_t *ptr_cb = frame->data[1] + (mb_y * 8 * frame->linesize[1]) + mb_x * 8; - uint8_t *ptr_cr = frame->data[2] + (mb_y * 8 * frame->linesize[2]) + mb_x * 8; + uint8_t *ptr_y = frame->data[0] + (mb_y * 16 * linesize) + mb_x * 16; + uint8_t *ptr_cb = frame->data[1] + (mb_y * 8 * frame->linesize[1]) + mb_x * 8; + uint8_t *ptr_cr = frame->data[2] + (mb_y * 8 * frame->linesize[2]) + mb_x * 8; a->pdsp.get_pixels(block[0], ptr_y, linesize); a->pdsp.get_pixels(block[1], ptr_y + 8, linesize); a->pdsp.get_pixels(block[2], ptr_y + 8 * linesize, linesize); a->pdsp.get_pixels(block[3], ptr_y + 8 * linesize + 8, linesize); - for(i=0; i<4; i++) + for (i = 0; i < 4; i++) a->fdsp.fdct(block[i]); - if(!(a->avctx->flags&CODEC_FLAG_GRAY)){ + if (!(a->avctx->flags & CODEC_FLAG_GRAY)) { a->pdsp.get_pixels(block[4], ptr_cb, frame->linesize[1]); a->pdsp.get_pixels(block[5], ptr_cr, frame->linesize[2]); - for(i=4; i<6; i++) + for (i = 4; i < 6; i++) a->fdsp.fdct(block[i]); } } @@ -177,37 +210,37 @@ static inline void dct_get(ASV1Context *a, const AVFrame *frame, static int encode_frame(AVCodecContext *avctx, AVPacket *pkt, const AVFrame *pict, int *got_packet) { - ASV1Context * const a = avctx->priv_data; + ASV1Context *const a = avctx->priv_data; int size, ret; int mb_x, mb_y; if (!pkt->data && - (ret = av_new_packet(pkt, a->mb_height*a->mb_width*MAX_MB_SIZE + - FF_MIN_BUFFER_SIZE)) < 0) { + (ret = av_new_packet(pkt, a->mb_height * a->mb_width * MAX_MB_SIZE + + FF_MIN_BUFFER_SIZE)) < 0) { av_log(avctx, AV_LOG_ERROR, "Error getting output packet.\n"); return ret; } init_put_bits(&a->pb, pkt->data, pkt->size); - for(mb_y=0; mb_ymb_height2; mb_y++){ - for(mb_x=0; mb_xmb_width2; mb_x++){ + for (mb_y = 0; mb_y < a->mb_height2; mb_y++) { + for (mb_x = 0; mb_x < a->mb_width2; mb_x++) { dct_get(a, pict, mb_x, mb_y); encode_mb(a, a->block); } } - if(a->mb_width2 != a->mb_width){ - mb_x= a->mb_width2; - for(mb_y=0; mb_ymb_height2; mb_y++){ + if (a->mb_width2 != a->mb_width) { + mb_x = a->mb_width2; + for (mb_y = 0; mb_y < a->mb_height2; mb_y++) { dct_get(a, pict, mb_x, mb_y); encode_mb(a, a->block); } } - if(a->mb_height2 != a->mb_height){ - mb_y= a->mb_height2; - for(mb_x=0; mb_xmb_width; mb_x++){ + if (a->mb_height2 != a->mb_height) { + mb_y = a->mb_height2; + for (mb_x = 0; mb_x < a->mb_width; mb_x++) { dct_get(a, pict, mb_x, mb_y); encode_mb(a, a->block); } @@ -215,31 +248,32 @@ static int encode_frame(AVCodecContext *avctx, AVPacket *pkt, emms_c(); avpriv_align_put_bits(&a->pb); - while(put_bits_count(&a->pb)&31) + while (put_bits_count(&a->pb) & 31) put_bits(&a->pb, 8, 0); - size= put_bits_count(&a->pb)/32; + size = put_bits_count(&a->pb) / 32; - if(avctx->codec_id == AV_CODEC_ID_ASV1) + if (avctx->codec_id == AV_CODEC_ID_ASV1) { a->bbdsp.bswap_buf((uint32_t *) pkt->data, (uint32_t *) pkt->data, size); - else{ + } else { int i; - for(i=0; i<4*size; i++) + for (i = 0; i < 4 * size; i++) pkt->data[i] = ff_reverse[pkt->data[i]]; } - pkt->size = size*4; + pkt->size = size * 4; pkt->flags |= AV_PKT_FLAG_KEY; *got_packet = 1; return 0; } -static av_cold int encode_init(AVCodecContext *avctx){ - ASV1Context * const a = avctx->priv_data; +static av_cold int encode_init(AVCodecContext *avctx) +{ + ASV1Context *const a = avctx->priv_data; int i; - const int scale= avctx->codec_id == AV_CODEC_ID_ASV1 ? 1 : 2; + const int scale = avctx->codec_id == AV_CODEC_ID_ASV1 ? 1 : 2; avctx->coded_frame = av_frame_alloc(); if (!avctx->coded_frame) @@ -251,22 +285,25 @@ static av_cold int encode_init(AVCodecContext *avctx){ ff_fdctdsp_init(&a->fdsp, avctx); ff_pixblockdsp_init(&a->pdsp, avctx); - if(avctx->global_quality == 0) avctx->global_quality= 4*FF_QUALITY_SCALE; + if (avctx->global_quality == 0) + avctx->global_quality = 4 * FF_QUALITY_SCALE; - a->inv_qscale= (32*scale*FF_QUALITY_SCALE + avctx->global_quality/2) / avctx->global_quality; + a->inv_qscale = (32 * scale * FF_QUALITY_SCALE + + avctx->global_quality / 2) / avctx->global_quality; - avctx->extradata= av_mallocz(8); - avctx->extradata_size=8; - ((uint32_t*)avctx->extradata)[0]= av_le2ne32(a->inv_qscale); - ((uint32_t*)avctx->extradata)[1]= av_le2ne32(AV_RL32("ASUS")); + avctx->extradata = av_mallocz(8); + avctx->extradata_size = 8; + ((uint32_t *) avctx->extradata)[0] = av_le2ne32(a->inv_qscale); + ((uint32_t *) avctx->extradata)[1] = av_le2ne32(AV_RL32("ASUS")); - for(i=0; i<64; i++){ - int q= 32*scale*ff_mpeg1_default_intra_matrix[i]; - a->q_intra_matrix[i]= ((a->inv_qscale<<16) + q/2) / q; + for (i = 0; i < 64; i++) { + int q = 32 * scale * ff_mpeg1_default_intra_matrix[i]; + a->q_intra_matrix[i] = ((a->inv_qscale << 16) + q / 2) / q; } return 0; } + static av_cold int asv_encode_close(AVCodecContext *avctx) { av_frame_free(&avctx->coded_frame); @@ -284,8 +321,8 @@ AVCodec ff_asv1_encoder = { .init = encode_init, .encode2 = encode_frame, .close = asv_encode_close, - .pix_fmts = (const enum AVPixelFormat[]){ AV_PIX_FMT_YUV420P, - AV_PIX_FMT_NONE }, + .pix_fmts = (const enum AVPixelFormat[]) { AV_PIX_FMT_YUV420P, + AV_PIX_FMT_NONE }, }; #endif @@ -299,7 +336,7 @@ AVCodec ff_asv2_encoder = { .init = encode_init, .encode2 = encode_frame, .close = asv_encode_close, - .pix_fmts = (const enum AVPixelFormat[]){ AV_PIX_FMT_YUV420P, - AV_PIX_FMT_NONE }, + .pix_fmts = (const enum AVPixelFormat[]) { AV_PIX_FMT_YUV420P, + AV_PIX_FMT_NONE }, }; #endif -- cgit v1.2.3