Disable encoders by undefining CONFIG_FOO_ENCODER once instead of littering
[ffmpeg.git] / libavcodec / vcr1.c
1 /*
2  * ATI VCR1 codec
3  * Copyright (c) 2003 Michael Niedermayer
4  *
5  * This file is part of FFmpeg.
6  *
7  * FFmpeg is free software; you can redistribute it and/or
8  * modify it under the terms of the GNU Lesser General Public
9  * License as published by the Free Software Foundation; either
10  * version 2.1 of the License, or (at your option) any later version.
11  *
12  * FFmpeg is distributed in the hope that it will be useful,
13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
15  * Lesser General Public License for more details.
16  *
17  * You should have received a copy of the GNU Lesser General Public
18  * License along with FFmpeg; if not, write to the Free Software
19  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
20  */
21
22 /**
23  * @file vcr1.c
24  * ati vcr1 codec.
25  */
26
27 #include "avcodec.h"
28 #include "dsputil.h"
29
30 //#undef NDEBUG
31 //#include <assert.h>
32
33 /* Disable the encoder. */
34 #undef CONFIG_VCR1_ENCODER
35
36 typedef struct VCR1Context{
37     AVCodecContext *avctx;
38     AVFrame picture;
39     int delta[16];
40     int offset[4];
41 } VCR1Context;
42
43 static int decode_frame(AVCodecContext *avctx,
44                         void *data, int *data_size,
45                         const uint8_t *buf, int buf_size)
46 {
47     VCR1Context * const a = avctx->priv_data;
48     AVFrame *picture = data;
49     AVFrame * const p= (AVFrame*)&a->picture;
50     const uint8_t *bytestream= buf;
51     int i, x, y;
52
53     if(p->data[0])
54         avctx->release_buffer(avctx, p);
55
56     p->reference= 0;
57     if(avctx->get_buffer(avctx, p) < 0){
58         av_log(avctx, AV_LOG_ERROR, "get_buffer() failed\n");
59         return -1;
60     }
61     p->pict_type= FF_I_TYPE;
62     p->key_frame= 1;
63
64     for(i=0; i<16; i++){
65         a->delta[i]= *(bytestream++);
66         bytestream++;
67     }
68
69     for(y=0; y<avctx->height; y++){
70         int offset;
71         uint8_t *luma= &a->picture.data[0][ y*a->picture.linesize[0] ];
72
73         if((y&3) == 0){
74             uint8_t *cb= &a->picture.data[1][ (y>>2)*a->picture.linesize[1] ];
75             uint8_t *cr= &a->picture.data[2][ (y>>2)*a->picture.linesize[2] ];
76
77             for(i=0; i<4; i++)
78                 a->offset[i]= *(bytestream++);
79
80             offset= a->offset[0] - a->delta[ bytestream[2]&0xF ];
81             for(x=0; x<avctx->width; x+=4){
82                 luma[0]=( offset += a->delta[ bytestream[2]&0xF ]);
83                 luma[1]=( offset += a->delta[ bytestream[2]>>4  ]);
84                 luma[2]=( offset += a->delta[ bytestream[0]&0xF ]);
85                 luma[3]=( offset += a->delta[ bytestream[0]>>4  ]);
86                 luma += 4;
87
88                 *(cb++) = bytestream[3];
89                 *(cr++) = bytestream[1];
90
91                 bytestream+= 4;
92             }
93         }else{
94             offset= a->offset[y&3] - a->delta[ bytestream[2]&0xF ];
95
96             for(x=0; x<avctx->width; x+=8){
97                 luma[0]=( offset += a->delta[ bytestream[2]&0xF ]);
98                 luma[1]=( offset += a->delta[ bytestream[2]>>4  ]);
99                 luma[2]=( offset += a->delta[ bytestream[3]&0xF ]);
100                 luma[3]=( offset += a->delta[ bytestream[3]>>4  ]);
101                 luma[4]=( offset += a->delta[ bytestream[0]&0xF ]);
102                 luma[5]=( offset += a->delta[ bytestream[0]>>4  ]);
103                 luma[6]=( offset += a->delta[ bytestream[1]&0xF ]);
104                 luma[7]=( offset += a->delta[ bytestream[1]>>4  ]);
105                 luma += 8;
106                 bytestream+= 4;
107             }
108         }
109     }
110
111     *picture= *(AVFrame*)&a->picture;
112     *data_size = sizeof(AVPicture);
113
114     emms_c();
115
116     return buf_size;
117 }
118
119 #ifdef CONFIG_VCR1_ENCODER
120 static int encode_frame(AVCodecContext *avctx, unsigned char *buf, int buf_size, void *data){
121     VCR1Context * const a = avctx->priv_data;
122     AVFrame *pict = data;
123     AVFrame * const p= (AVFrame*)&a->picture;
124     int size;
125     int mb_x, mb_y;
126
127     *p = *pict;
128     p->pict_type= FF_I_TYPE;
129     p->key_frame= 1;
130
131     emms_c();
132
133     align_put_bits(&a->pb);
134     while(get_bit_count(&a->pb)&31)
135         put_bits(&a->pb, 8, 0);
136
137     size= get_bit_count(&a->pb)/32;
138
139     return size*4;
140 }
141 #endif
142
143 static av_cold void common_init(AVCodecContext *avctx){
144     VCR1Context * const a = avctx->priv_data;
145
146     avctx->coded_frame= (AVFrame*)&a->picture;
147     a->avctx= avctx;
148 }
149
150 static av_cold int decode_init(AVCodecContext *avctx){
151
152     common_init(avctx);
153
154     avctx->pix_fmt= PIX_FMT_YUV410P;
155
156     return 0;
157 }
158
159 #ifdef CONFIG_VCR1_ENCODER
160 static av_cold int encode_init(AVCodecContext *avctx){
161
162     common_init(avctx);
163
164     return 0;
165 }
166 #endif
167
168 AVCodec vcr1_decoder = {
169     "vcr1",
170     CODEC_TYPE_VIDEO,
171     CODEC_ID_VCR1,
172     sizeof(VCR1Context),
173     decode_init,
174     NULL,
175     NULL,
176     decode_frame,
177     CODEC_CAP_DR1,
178     .long_name = NULL_IF_CONFIG_SMALL("ATI VCR1"),
179 };
180
181 #ifdef CONFIG_VCR1_ENCODER
182 AVCodec vcr1_encoder = {
183     "vcr1",
184     CODEC_TYPE_VIDEO,
185     CODEC_ID_VCR1,
186     sizeof(VCR1Context),
187     encode_init,
188     encode_frame,
189     //encode_end,
190     .long_name = NULL_IF_CONFIG_SMALL("ATI VCR1"),
191 };
192 #endif