Merge commit '36ef5369ee9b336febc2c270f8718cec4476cb85'
[ffmpeg.git] / libavcodec / xbmdec.c
1 /*
2  * XBM image format
3  *
4  * Copyright (c) 2012 Paul B Mahol
5  *
6  * This file is part of FFmpeg.
7  *
8  * FFmpeg is free software; you can redistribute it and/or
9  * modify it under the terms of the GNU Lesser General Public
10  * License as published by the Free Software Foundation; either
11  * version 2.1 of the License, or (at your option) any later version.
12  *
13  * FFmpeg is distributed in the hope that it will be useful,
14  * but WITHOUT ANY WARRANTY; without even the implied warranty of
15  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
16  * Lesser General Public License for more details.
17  *
18  * You should have received a copy of the GNU Lesser General Public
19  * License along with FFmpeg; if not, write to the Free Software
20  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
21  */
22
23 #include "avcodec.h"
24 #include "internal.h"
25
26 static av_cold int xbm_decode_init(AVCodecContext *avctx)
27 {
28     avctx->coded_frame = avcodec_alloc_frame();
29     if (!avctx->coded_frame)
30         return AVERROR(ENOMEM);
31
32     return 0;
33 }
34
35 static int convert(uint8_t x)
36 {
37     if (x >= 'a')
38         x -= 87;
39     else if (x >= 'A')
40         x -= 55;
41     else
42         x -= '0';
43     return x;
44 }
45
46 static int xbm_decode_frame(AVCodecContext *avctx, void *data,
47                             int *data_size, AVPacket *avpkt)
48 {
49     AVFrame *p = avctx->coded_frame;
50     const uint8_t *end, *ptr = avpkt->data;
51     uint8_t *dst;
52     int ret, linesize, i, j;
53
54     end = avpkt->data + avpkt->size;
55     while (!avctx->width || !avctx->height) {
56         char name[256];
57         int number, len;
58
59         ptr += strcspn(ptr, "#");
60         if (sscanf(ptr, "#define %256s %u", name, &number) != 2) {
61             av_log(avctx, AV_LOG_ERROR, "Unexpected preprocessor directive\n");
62             return AVERROR_INVALIDDATA;
63         }
64
65         len = strlen(name);
66         if ((len > 6) && !avctx->height && !memcmp(name + len - 7, "_height", 7)) {
67                 avctx->height = number;
68         } else if ((len > 5) && !avctx->width && !memcmp(name + len - 6, "_width", 6)) {
69                 avctx->width = number;
70         } else {
71             av_log(avctx, AV_LOG_ERROR, "Unknown define '%s'\n", name);
72             return AVERROR_INVALIDDATA;
73         }
74         ptr += strcspn(ptr, "\n\r") + 1;
75     }
76
77     avctx->pix_fmt = PIX_FMT_MONOWHITE;
78
79     if (p->data[0])
80         avctx->release_buffer(avctx, p);
81
82     p->reference = 0;
83     if ((ret = avctx->get_buffer(avctx, p)) < 0)
84         return ret;
85
86     // goto start of image data
87     ptr += strcspn(ptr, "{") + 1;
88
89     linesize = (avctx->width + 7) / 8;
90     for (i = 0; i < avctx->height; i++) {
91         dst = p->data[0] + i * p->linesize[0];
92         for (j = 0; j < linesize; j++) {
93             uint8_t val;
94
95             ptr += strcspn(ptr, "x") + 1;
96             if (ptr < end && isxdigit(*ptr)) {
97                 val = convert(*ptr);
98                 ptr++;
99                 if (isxdigit(*ptr))
100                     val = (val << 4) + convert(*ptr);
101                 *dst++ = av_reverse[val];
102             } else {
103                 av_log(avctx, AV_LOG_ERROR, "Unexpected data at '%.8s'\n", ptr);
104                 return AVERROR_INVALIDDATA;
105             }
106         }
107     }
108
109     p->key_frame = 1;
110     p->pict_type = AV_PICTURE_TYPE_I;
111
112     *data_size       = sizeof(AVFrame);
113     *(AVFrame *)data = *p;
114
115     return avpkt->size;
116 }
117
118 static av_cold int xbm_decode_close(AVCodecContext *avctx)
119 {
120     if (avctx->coded_frame->data[0])
121         avctx->release_buffer(avctx, avctx->coded_frame);
122
123     av_freep(&avctx->coded_frame);
124
125     return 0;
126 }
127
128 AVCodec ff_xbm_decoder = {
129     .name         = "xbm",
130     .type         = AVMEDIA_TYPE_VIDEO,
131     .id           = AV_CODEC_ID_XBM,
132     .init         = xbm_decode_init,
133     .close        = xbm_decode_close,
134     .decode       = xbm_decode_frame,
135     .capabilities = CODEC_CAP_DR1,
136     .long_name    = NULL_IF_CONFIG_SMALL("XBM (X BitMap) image"),
137 };