cljr: split decoder and encoder
[ffmpeg.git] / libavcodec / cljrenc.c
1 /*
2  * Cirrus Logic AccuPak (CLJR) encoder
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 encoder.
25  */
26
27 #include "libavutil/common.h"
28
29 #include "avcodec.h"
30 #include "internal.h"
31 #include "put_bits.h"
32
33 static av_cold int encode_init(AVCodecContext *avctx)
34 {
35     avctx->coded_frame = av_frame_alloc();
36     if (!avctx->coded_frame)
37         return AVERROR(ENOMEM);
38
39     return 0;
40 }
41
42 static av_cold int encode_close(AVCodecContext *avctx)
43 {
44     av_frame_free(&avctx->coded_frame);
45     return 0;
46 }
47
48 static int encode_frame(AVCodecContext *avctx, AVPacket *pkt,
49                         const AVFrame *p, int *got_packet)
50 {
51     PutBitContext pb;
52     int x, y, ret;
53
54     if ((ret = ff_alloc_packet(pkt, 32*avctx->height*avctx->width/4)) < 0) {
55         av_log(avctx, AV_LOG_ERROR, "Error getting output packet.\n");
56         return ret;
57     }
58
59     avctx->coded_frame->pict_type = AV_PICTURE_TYPE_I;
60     avctx->coded_frame->key_frame = 1;
61
62     init_put_bits(&pb, pkt->data, pkt->size);
63
64     for (y = 0; y < avctx->height; y++) {
65         uint8_t *luma = &p->data[0][y * p->linesize[0]];
66         uint8_t *cb   = &p->data[1][y * p->linesize[1]];
67         uint8_t *cr   = &p->data[2][y * p->linesize[2]];
68         for (x = 0; x < avctx->width; x += 4) {
69             put_bits(&pb, 5, luma[3] >> 3);
70             put_bits(&pb, 5, luma[2] >> 3);
71             put_bits(&pb, 5, luma[1] >> 3);
72             put_bits(&pb, 5, luma[0] >> 3);
73             luma += 4;
74             put_bits(&pb, 6, *(cb++) >> 2);
75             put_bits(&pb, 6, *(cr++) >> 2);
76         }
77     }
78
79     flush_put_bits(&pb);
80
81     pkt->size   = put_bits_count(&pb) / 8;
82     pkt->flags |= AV_PKT_FLAG_KEY;
83     *got_packet = 1;
84     return 0;
85 }
86
87 AVCodec ff_cljr_encoder = {
88     .name           = "cljr",
89     .long_name      = NULL_IF_CONFIG_SMALL("Cirrus Logic AccuPak"),
90     .type           = AVMEDIA_TYPE_VIDEO,
91     .id             = AV_CODEC_ID_CLJR,
92     .init           = encode_init,
93     .encode2        = encode_frame,
94     .close          = encode_close,
95     .pix_fmts       = (const enum AVPixelFormat[]) { AV_PIX_FMT_YUV411P,
96                                                    AV_PIX_FMT_NONE },
97 };