yading@10
|
1 /*
|
yading@10
|
2 * PNM image parser
|
yading@10
|
3 * Copyright (c) 2002, 2003 Fabrice Bellard
|
yading@10
|
4 *
|
yading@10
|
5 * This file is part of FFmpeg.
|
yading@10
|
6 *
|
yading@10
|
7 * FFmpeg is free software; you can redistribute it and/or
|
yading@10
|
8 * modify it under the terms of the GNU Lesser General Public
|
yading@10
|
9 * License as published by the Free Software Foundation; either
|
yading@10
|
10 * version 2.1 of the License, or (at your option) any later version.
|
yading@10
|
11 *
|
yading@10
|
12 * FFmpeg is distributed in the hope that it will be useful,
|
yading@10
|
13 * but WITHOUT ANY WARRANTY; without even the implied warranty of
|
yading@10
|
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
yading@10
|
15 * Lesser General Public License for more details.
|
yading@10
|
16 *
|
yading@10
|
17 * You should have received a copy of the GNU Lesser General Public
|
yading@10
|
18 * License along with FFmpeg; if not, write to the Free Software
|
yading@10
|
19 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
|
yading@10
|
20 */
|
yading@10
|
21
|
yading@10
|
22 #include "parser.h" //for ParseContext
|
yading@10
|
23 #include "pnm.h"
|
yading@10
|
24
|
yading@10
|
25
|
yading@10
|
26 static int pnm_parse(AVCodecParserContext *s, AVCodecContext *avctx,
|
yading@10
|
27 const uint8_t **poutbuf, int *poutbuf_size,
|
yading@10
|
28 const uint8_t *buf, int buf_size)
|
yading@10
|
29 {
|
yading@10
|
30 ParseContext *pc = s->priv_data;
|
yading@10
|
31 PNMContext pnmctx;
|
yading@10
|
32 int next;
|
yading@10
|
33
|
yading@10
|
34 for (; pc->overread > 0; pc->overread--) {
|
yading@10
|
35 pc->buffer[pc->index++]= pc->buffer[pc->overread_index++];
|
yading@10
|
36 }
|
yading@10
|
37 retry:
|
yading@10
|
38 if (pc->index) {
|
yading@10
|
39 pnmctx.bytestream_start =
|
yading@10
|
40 pnmctx.bytestream = pc->buffer;
|
yading@10
|
41 pnmctx.bytestream_end = pc->buffer + pc->index;
|
yading@10
|
42 } else {
|
yading@10
|
43 pnmctx.bytestream_start =
|
yading@10
|
44 pnmctx.bytestream = (uint8_t *) buf; /* casts avoid warnings */
|
yading@10
|
45 pnmctx.bytestream_end = (uint8_t *) buf + buf_size;
|
yading@10
|
46 }
|
yading@10
|
47 if (ff_pnm_decode_header(avctx, &pnmctx) < 0) {
|
yading@10
|
48 if (pnmctx.bytestream < pnmctx.bytestream_end) {
|
yading@10
|
49 if (pc->index) {
|
yading@10
|
50 pc->index = 0;
|
yading@10
|
51 } else {
|
yading@10
|
52 buf++;
|
yading@10
|
53 buf_size--;
|
yading@10
|
54 }
|
yading@10
|
55 goto retry;
|
yading@10
|
56 }
|
yading@10
|
57 #if 0
|
yading@10
|
58 if (pc->index && pc->index * 2 + FF_INPUT_BUFFER_PADDING_SIZE < pc->buffer_size && buf_size > pc->index) {
|
yading@10
|
59 memcpy(pc->buffer + pc->index, buf, pc->index);
|
yading@10
|
60 pc->index += pc->index;
|
yading@10
|
61 buf += pc->index;
|
yading@10
|
62 buf_size -= pc->index;
|
yading@10
|
63 goto retry;
|
yading@10
|
64 }
|
yading@10
|
65 #endif
|
yading@10
|
66 next = END_NOT_FOUND;
|
yading@10
|
67 } else {
|
yading@10
|
68 next = pnmctx.bytestream - pnmctx.bytestream_start
|
yading@10
|
69 + avpicture_get_size(avctx->pix_fmt, avctx->width, avctx->height);
|
yading@10
|
70 if (pnmctx.bytestream_start != buf)
|
yading@10
|
71 next -= pc->index;
|
yading@10
|
72 if (next > buf_size)
|
yading@10
|
73 next = END_NOT_FOUND;
|
yading@10
|
74 }
|
yading@10
|
75
|
yading@10
|
76 if (ff_combine_frame(pc, next, &buf, &buf_size) < 0) {
|
yading@10
|
77 *poutbuf = NULL;
|
yading@10
|
78 *poutbuf_size = 0;
|
yading@10
|
79 return buf_size;
|
yading@10
|
80 }
|
yading@10
|
81 *poutbuf = buf;
|
yading@10
|
82 *poutbuf_size = buf_size;
|
yading@10
|
83 return next;
|
yading@10
|
84 }
|
yading@10
|
85
|
yading@10
|
86 AVCodecParser ff_pnm_parser = {
|
yading@10
|
87 .codec_ids = { AV_CODEC_ID_PGM, AV_CODEC_ID_PGMYUV, AV_CODEC_ID_PPM,
|
yading@10
|
88 AV_CODEC_ID_PBM, AV_CODEC_ID_PAM },
|
yading@10
|
89 .priv_data_size = sizeof(ParseContext),
|
yading@10
|
90 .parser_parse = pnm_parse,
|
yading@10
|
91 .parser_close = ff_parse_close,
|
yading@10
|
92 };
|