Disable encoders by undefining CONFIG_FOO_ENCODER once instead of littering
[libav.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