00001
00002
00003
00004
00005
00006
00007
00008
00009
00010
00011
00012
00013
00014
00015
00016
00017
00018
00019
00020
00021
00027 #include "avcodec.h"
00028 #include "dsputil.h"
00029
00030
00031
00032
00033
00034 #undef CONFIG_VCR1_ENCODER
00035 #define CONFIG_VCR1_ENCODER 0
00036
00037 typedef struct VCR1Context{
00038 AVCodecContext *avctx;
00039 AVFrame picture;
00040 int delta[16];
00041 int offset[4];
00042 } VCR1Context;
00043
00044 static int decode_frame(AVCodecContext *avctx,
00045 void *data, int *data_size,
00046 const uint8_t *buf, int buf_size)
00047 {
00048 VCR1Context * const a = avctx->priv_data;
00049 AVFrame *picture = data;
00050 AVFrame * const p= (AVFrame*)&a->picture;
00051 const uint8_t *bytestream= buf;
00052 int i, x, y;
00053
00054 if(p->data[0])
00055 avctx->release_buffer(avctx, p);
00056
00057 p->reference= 0;
00058 if(avctx->get_buffer(avctx, p) < 0){
00059 av_log(avctx, AV_LOG_ERROR, "get_buffer() failed\n");
00060 return -1;
00061 }
00062 p->pict_type= FF_I_TYPE;
00063 p->key_frame= 1;
00064
00065 for(i=0; i<16; i++){
00066 a->delta[i]= *(bytestream++);
00067 bytestream++;
00068 }
00069
00070 for(y=0; y<avctx->height; y++){
00071 int offset;
00072 uint8_t *luma= &a->picture.data[0][ y*a->picture.linesize[0] ];
00073
00074 if((y&3) == 0){
00075 uint8_t *cb= &a->picture.data[1][ (y>>2)*a->picture.linesize[1] ];
00076 uint8_t *cr= &a->picture.data[2][ (y>>2)*a->picture.linesize[2] ];
00077
00078 for(i=0; i<4; i++)
00079 a->offset[i]= *(bytestream++);
00080
00081 offset= a->offset[0] - a->delta[ bytestream[2]&0xF ];
00082 for(x=0; x<avctx->width; x+=4){
00083 luma[0]=( offset += a->delta[ bytestream[2]&0xF ]);
00084 luma[1]=( offset += a->delta[ bytestream[2]>>4 ]);
00085 luma[2]=( offset += a->delta[ bytestream[0]&0xF ]);
00086 luma[3]=( offset += a->delta[ bytestream[0]>>4 ]);
00087 luma += 4;
00088
00089 *(cb++) = bytestream[3];
00090 *(cr++) = bytestream[1];
00091
00092 bytestream+= 4;
00093 }
00094 }else{
00095 offset= a->offset[y&3] - a->delta[ bytestream[2]&0xF ];
00096
00097 for(x=0; x<avctx->width; x+=8){
00098 luma[0]=( offset += a->delta[ bytestream[2]&0xF ]);
00099 luma[1]=( offset += a->delta[ bytestream[2]>>4 ]);
00100 luma[2]=( offset += a->delta[ bytestream[3]&0xF ]);
00101 luma[3]=( offset += a->delta[ bytestream[3]>>4 ]);
00102 luma[4]=( offset += a->delta[ bytestream[0]&0xF ]);
00103 luma[5]=( offset += a->delta[ bytestream[0]>>4 ]);
00104 luma[6]=( offset += a->delta[ bytestream[1]&0xF ]);
00105 luma[7]=( offset += a->delta[ bytestream[1]>>4 ]);
00106 luma += 8;
00107 bytestream+= 4;
00108 }
00109 }
00110 }
00111
00112 *picture= *(AVFrame*)&a->picture;
00113 *data_size = sizeof(AVPicture);
00114
00115 emms_c();
00116
00117 return buf_size;
00118 }
00119
00120 #if CONFIG_VCR1_ENCODER
00121 static int encode_frame(AVCodecContext *avctx, unsigned char *buf, int buf_size, void *data){
00122 VCR1Context * const a = avctx->priv_data;
00123 AVFrame *pict = data;
00124 AVFrame * const p= (AVFrame*)&a->picture;
00125 int size;
00126
00127 *p = *pict;
00128 p->pict_type= FF_I_TYPE;
00129 p->key_frame= 1;
00130
00131 emms_c();
00132
00133 align_put_bits(&a->pb);
00134 while(get_bit_count(&a->pb)&31)
00135 put_bits(&a->pb, 8, 0);
00136
00137 size= get_bit_count(&a->pb)/32;
00138
00139 return size*4;
00140 }
00141 #endif
00142
00143 static av_cold void common_init(AVCodecContext *avctx){
00144 VCR1Context * const a = avctx->priv_data;
00145
00146 avctx->coded_frame= (AVFrame*)&a->picture;
00147 a->avctx= avctx;
00148 }
00149
00150 static av_cold int decode_init(AVCodecContext *avctx){
00151
00152 common_init(avctx);
00153
00154 avctx->pix_fmt= PIX_FMT_YUV410P;
00155
00156 return 0;
00157 }
00158
00159 #if CONFIG_VCR1_ENCODER
00160 static av_cold int encode_init(AVCodecContext *avctx){
00161
00162 common_init(avctx);
00163
00164 return 0;
00165 }
00166 #endif
00167
00168 AVCodec vcr1_decoder = {
00169 "vcr1",
00170 CODEC_TYPE_VIDEO,
00171 CODEC_ID_VCR1,
00172 sizeof(VCR1Context),
00173 decode_init,
00174 NULL,
00175 NULL,
00176 decode_frame,
00177 CODEC_CAP_DR1,
00178 .long_name = NULL_IF_CONFIG_SMALL("ATI VCR1"),
00179 };
00180
00181 #if CONFIG_VCR1_ENCODER
00182 AVCodec vcr1_encoder = {
00183 "vcr1",
00184 CODEC_TYPE_VIDEO,
00185 CODEC_ID_VCR1,
00186 sizeof(VCR1Context),
00187 encode_init,
00188 encode_frame,
00189
00190 .long_name = NULL_IF_CONFIG_SMALL("ATI VCR1"),
00191 };
00192 #endif