cljr: simplify CLJRContext
[ffmpeg.git] / libavcodec / cljr.c
1 /*
2  * Cirrus Logic AccuPak (CLJR) codec
3  * Copyright (c) 2003 Alex Beregszaszi
4  *
5  * This file is part of Libav.
6  *
7  * Libav 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  * Libav 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 Libav; 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
24  * Cirrus Logic AccuPak codec.
25  */
26
27 #include "avcodec.h"
28 #include "dsputil.h"
29 #include "get_bits.h"
30
31 /* Disable the encoder. */
32 #undef CONFIG_CLJR_ENCODER
33 #define CONFIG_CLJR_ENCODER 0
34
35 typedef struct CLJRContext{
36     AVCodecContext *avctx;
37     AVFrame picture;
38 } CLJRContext;
39
40 static int decode_frame(AVCodecContext *avctx,
41                         void *data, int *data_size,
42                         AVPacket *avpkt)
43 {
44     const uint8_t *buf = avpkt->data;
45     int buf_size = avpkt->size;
46     CLJRContext * const a = avctx->priv_data;
47     GetBitContext gb;
48     AVFrame *picture = data;
49     AVFrame * const p= (AVFrame*)&a->picture;
50     int x, y;
51
52     if(p->data[0])
53         avctx->release_buffer(avctx, p);
54
55     if(buf_size/avctx->height < avctx->width) {
56         av_log(avctx, AV_LOG_ERROR, "Resolution larger than buffer size. Invalid header?\n");
57         return -1;
58     }
59
60     p->reference= 0;
61     if(avctx->get_buffer(avctx, p) < 0){
62         av_log(avctx, AV_LOG_ERROR, "get_buffer() failed\n");
63         return -1;
64     }
65     p->pict_type= AV_PICTURE_TYPE_I;
66     p->key_frame= 1;
67
68     init_get_bits(&gb, buf, buf_size * 8);
69
70     for(y=0; y<avctx->height; y++){
71         uint8_t *luma= &a->picture.data[0][ y*a->picture.linesize[0] ];
72         uint8_t *cb= &a->picture.data[1][ y*a->picture.linesize[1] ];
73         uint8_t *cr= &a->picture.data[2][ y*a->picture.linesize[2] ];
74         for(x=0; x<avctx->width; x+=4){
75             luma[3] = get_bits(&gb, 5) << 3;
76             luma[2] = get_bits(&gb, 5) << 3;
77             luma[1] = get_bits(&gb, 5) << 3;
78             luma[0] = get_bits(&gb, 5) << 3;
79             luma+= 4;
80             *(cb++) = get_bits(&gb, 6) << 2;
81             *(cr++) = get_bits(&gb, 6) << 2;
82         }
83     }
84
85     *picture= *(AVFrame*)&a->picture;
86     *data_size = sizeof(AVPicture);
87
88     emms_c();
89
90     return buf_size;
91 }
92
93 #if CONFIG_CLJR_ENCODER
94 static int encode_frame(AVCodecContext *avctx, unsigned char *buf, int buf_size, void *data){
95     CLJRContext * const a = avctx->priv_data;
96     AVFrame *pict = data;
97     AVFrame * const p= (AVFrame*)&a->picture;
98     int size;
99
100     *p = *pict;
101     p->pict_type= AV_PICTURE_TYPE_I;
102     p->key_frame= 1;
103
104     emms_c();
105
106     avpriv_align_put_bits(&a->pb);
107     while(get_bit_count(&a->pb)&31)
108         put_bits(&a->pb, 8, 0);
109
110     size= get_bit_count(&a->pb)/32;
111
112     return size*4;
113 }
114 #endif
115
116 static av_cold void common_init(AVCodecContext *avctx){
117     CLJRContext * const a = avctx->priv_data;
118
119     avctx->coded_frame= (AVFrame*)&a->picture;
120     a->avctx= avctx;
121 }
122
123 static av_cold int decode_init(AVCodecContext *avctx){
124
125     common_init(avctx);
126
127     avctx->pix_fmt= PIX_FMT_YUV411P;
128
129     return 0;
130 }
131
132 #if CONFIG_CLJR_ENCODER
133 static av_cold int encode_init(AVCodecContext *avctx){
134
135     common_init(avctx);
136
137     return 0;
138 }
139 #endif
140
141 AVCodec ff_cljr_decoder = {
142     .name           = "cljr",
143     .type           = AVMEDIA_TYPE_VIDEO,
144     .id             = CODEC_ID_CLJR,
145     .priv_data_size = sizeof(CLJRContext),
146     .init           = decode_init,
147     .decode         = decode_frame,
148     .capabilities   = CODEC_CAP_DR1,
149     .long_name = NULL_IF_CONFIG_SMALL("Cirrus Logic AccuPak"),
150 };
151
152 #if CONFIG_CLJR_ENCODER
153 AVCodec ff_cljr_encoder = {
154     .name           = "cljr",
155     .type           = AVMEDIA_TYPE_VIDEO,
156     .id             = CODEC_ID_CLJR,
157     .priv_data_size = sizeof(CLJRContext),
158     .init           = encode_init,
159     .encode         = encode_frame,
160     .long_name = NULL_IF_CONFIG_SMALL("Cirrus Logic AccuPak"),
161 };
162 #endif