annotate pnm.c @ 9440:daee921fb6bb libavcodec

VC1: add and use avg_no_rnd chroma MC functions
author conrad
date Tue, 14 Apr 2009 23:56:10 +0000
parents de31b10455cc
children a6bb56636f90
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
4978
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
1 /*
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
2 * PNM image format
8629
04423b2f6e0b cosmetics: Remove pointless period after copyright statement non-sentences.
diego
parents: 7865
diff changeset
3 * Copyright (c) 2002, 2003 Fabrice Bellard
4978
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
4 *
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
5 * This file is part of FFmpeg.
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
6 *
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
7 * FFmpeg is free software; you can redistribute it and/or
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
8 * modify it under the terms of the GNU Lesser General Public
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
9 * License as published by the Free Software Foundation; either
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
10 * version 2.1 of the License, or (at your option) any later version.
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
11 *
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
12 * FFmpeg is distributed in the hope that it will be useful,
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
13 * but WITHOUT ANY WARRANTY; without even the implied warranty of
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
15 * Lesser General Public License for more details.
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
16 *
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
17 * You should have received a copy of the GNU Lesser General Public
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
18 * License along with FFmpeg; if not, write to the Free Software
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
19 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
20 */
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
21 #include "avcodec.h"
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
22 #include "pnm.h"
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
23
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
24 static inline int pnm_space(int c)
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
25 {
6750
c93570aeb3eb Remove unnecessary parentheses from return calls.
diego
parents: 4985
diff changeset
26 return c == ' ' || c == '\n' || c == '\r' || c == '\t';
4978
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
27 }
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
28
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
29 static void pnm_get(PNMContext *sc, char *str, int buf_size)
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
30 {
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
31 char *s;
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
32 int c;
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
33
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
34 /* skip spaces and comments */
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
35 for(;;) {
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
36 c = *sc->bytestream++;
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
37 if (c == '#') {
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
38 do {
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
39 c = *sc->bytestream++;
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
40 } while (c != '\n' && sc->bytestream < sc->bytestream_end);
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
41 } else if (!pnm_space(c)) {
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
42 break;
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
43 }
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
44 }
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
45
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
46 s = str;
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
47 while (sc->bytestream < sc->bytestream_end && !pnm_space(c)) {
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
48 if ((s - str) < buf_size - 1)
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
49 *s++ = c;
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
50 c = *sc->bytestream++;
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
51 }
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
52 *s = '\0';
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
53 }
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
54
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
55 int ff_pnm_decode_header(AVCodecContext *avctx, PNMContext * const s){
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
56 char buf1[32], tuple_type[32];
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
57 int h, w, depth, maxval;
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
58
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
59 pnm_get(s, buf1, sizeof(buf1));
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
60 if (!strcmp(buf1, "P4")) {
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
61 avctx->pix_fmt = PIX_FMT_MONOWHITE;
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
62 } else if (!strcmp(buf1, "P5")) {
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
63 if (avctx->codec_id == CODEC_ID_PGMYUV)
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
64 avctx->pix_fmt = PIX_FMT_YUV420P;
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
65 else
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
66 avctx->pix_fmt = PIX_FMT_GRAY8;
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
67 } else if (!strcmp(buf1, "P6")) {
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
68 avctx->pix_fmt = PIX_FMT_RGB24;
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
69 } else if (!strcmp(buf1, "P7")) {
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
70 w = -1;
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
71 h = -1;
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
72 maxval = -1;
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
73 depth = -1;
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
74 tuple_type[0] = '\0';
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
75 for(;;) {
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
76 pnm_get(s, buf1, sizeof(buf1));
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
77 if (!strcmp(buf1, "WIDTH")) {
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
78 pnm_get(s, buf1, sizeof(buf1));
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
79 w = strtol(buf1, NULL, 10);
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
80 } else if (!strcmp(buf1, "HEIGHT")) {
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
81 pnm_get(s, buf1, sizeof(buf1));
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
82 h = strtol(buf1, NULL, 10);
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
83 } else if (!strcmp(buf1, "DEPTH")) {
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
84 pnm_get(s, buf1, sizeof(buf1));
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
85 depth = strtol(buf1, NULL, 10);
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
86 } else if (!strcmp(buf1, "MAXVAL")) {
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
87 pnm_get(s, buf1, sizeof(buf1));
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
88 maxval = strtol(buf1, NULL, 10);
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
89 } else if (!strcmp(buf1, "TUPLETYPE")) {
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
90 pnm_get(s, tuple_type, sizeof(tuple_type));
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
91 } else if (!strcmp(buf1, "ENDHDR")) {
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
92 break;
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
93 } else {
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
94 return -1;
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
95 }
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
96 }
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
97 /* check that all tags are present */
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
98 if (w <= 0 || h <= 0 || maxval <= 0 || depth <= 0 || tuple_type[0] == '\0' || avcodec_check_dimensions(avctx, w, h))
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
99 return -1;
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
100
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
101 avctx->width = w;
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
102 avctx->height = h;
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
103 if (depth == 1) {
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
104 if (maxval == 1)
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
105 avctx->pix_fmt = PIX_FMT_MONOWHITE;
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
106 else
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
107 avctx->pix_fmt = PIX_FMT_GRAY8;
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
108 } else if (depth == 3) {
7859
48d00b406a26 Return error when trying to decode non-grayscale 16-bit PNM images.
jbr
parents: 6750
diff changeset
109 if (maxval < 256) {
4978
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
110 avctx->pix_fmt = PIX_FMT_RGB24;
7859
48d00b406a26 Return error when trying to decode non-grayscale 16-bit PNM images.
jbr
parents: 6750
diff changeset
111 } else {
48d00b406a26 Return error when trying to decode non-grayscale 16-bit PNM images.
jbr
parents: 6750
diff changeset
112 av_log(avctx, AV_LOG_ERROR, "16-bit components are only supported for grayscale\n");
48d00b406a26 Return error when trying to decode non-grayscale 16-bit PNM images.
jbr
parents: 6750
diff changeset
113 avctx->pix_fmt = PIX_FMT_NONE;
48d00b406a26 Return error when trying to decode non-grayscale 16-bit PNM images.
jbr
parents: 6750
diff changeset
114 return -1;
48d00b406a26 Return error when trying to decode non-grayscale 16-bit PNM images.
jbr
parents: 6750
diff changeset
115 }
4978
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
116 } else if (depth == 4) {
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
117 avctx->pix_fmt = PIX_FMT_RGB32;
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
118 } else {
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
119 return -1;
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
120 }
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
121 return 0;
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
122 } else {
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
123 return -1;
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
124 }
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
125 pnm_get(s, buf1, sizeof(buf1));
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
126 avctx->width = atoi(buf1);
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
127 if (avctx->width <= 0)
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
128 return -1;
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
129 pnm_get(s, buf1, sizeof(buf1));
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
130 avctx->height = atoi(buf1);
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
131 if(avcodec_check_dimensions(avctx, avctx->width, avctx->height))
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
132 return -1;
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
133 if (avctx->pix_fmt != PIX_FMT_MONOWHITE) {
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
134 pnm_get(s, buf1, sizeof(buf1));
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
135 s->maxval = atoi(buf1);
7859
48d00b406a26 Return error when trying to decode non-grayscale 16-bit PNM images.
jbr
parents: 6750
diff changeset
136 if (s->maxval >= 256) {
48d00b406a26 Return error when trying to decode non-grayscale 16-bit PNM images.
jbr
parents: 6750
diff changeset
137 if (avctx->pix_fmt == PIX_FMT_GRAY8) {
7865
c7420b070961 cosmetics: fix indentation after r15321
jbr
parents: 7859
diff changeset
138 avctx->pix_fmt = PIX_FMT_GRAY16BE;
c7420b070961 cosmetics: fix indentation after r15321
jbr
parents: 7859
diff changeset
139 if (s->maxval != 65535)
c7420b070961 cosmetics: fix indentation after r15321
jbr
parents: 7859
diff changeset
140 avctx->pix_fmt = PIX_FMT_GRAY16;
9145
de31b10455cc pnm: Add missing 'else'. Fixes decoding for 16-bit pgm.
jbr
parents: 9002
diff changeset
141 } else if (avctx->pix_fmt == PIX_FMT_RGB24) {
9002
eb98d61af310 Support 48-bit RGB PPM image.
pross
parents: 8629
diff changeset
142 if (s->maxval > 255)
eb98d61af310 Support 48-bit RGB PPM image.
pross
parents: 8629
diff changeset
143 avctx->pix_fmt = PIX_FMT_RGB48BE;
7859
48d00b406a26 Return error when trying to decode non-grayscale 16-bit PNM images.
jbr
parents: 6750
diff changeset
144 } else {
9002
eb98d61af310 Support 48-bit RGB PPM image.
pross
parents: 8629
diff changeset
145 av_log(avctx, AV_LOG_ERROR, "Unsupported pixel format\n");
7859
48d00b406a26 Return error when trying to decode non-grayscale 16-bit PNM images.
jbr
parents: 6750
diff changeset
146 avctx->pix_fmt = PIX_FMT_NONE;
48d00b406a26 Return error when trying to decode non-grayscale 16-bit PNM images.
jbr
parents: 6750
diff changeset
147 return -1;
48d00b406a26 Return error when trying to decode non-grayscale 16-bit PNM images.
jbr
parents: 6750
diff changeset
148 }
4978
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
149 }
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
150 }
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
151 /* more check if YUV420 */
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
152 if (avctx->pix_fmt == PIX_FMT_YUV420P) {
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
153 if ((avctx->width & 1) != 0)
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
154 return -1;
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
155 h = (avctx->height * 2);
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
156 if ((h % 3) != 0)
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
157 return -1;
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
158 h /= 3;
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
159 avctx->height = h;
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
160 }
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
161 return 0;
95934eef9589 move pnm parser in its own file
aurel
parents:
diff changeset
162 }