yading@10: /* yading@10: * PNM image format yading@10: * Copyright (c) 2002, 2003 Fabrice Bellard yading@10: * yading@10: * This file is part of FFmpeg. yading@10: * yading@10: * FFmpeg is free software; you can redistribute it and/or yading@10: * modify it under the terms of the GNU Lesser General Public yading@10: * License as published by the Free Software Foundation; either yading@10: * version 2.1 of the License, or (at your option) any later version. yading@10: * yading@10: * FFmpeg is distributed in the hope that it will be useful, yading@10: * but WITHOUT ANY WARRANTY; without even the implied warranty of yading@10: * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU yading@10: * Lesser General Public License for more details. yading@10: * yading@10: * You should have received a copy of the GNU Lesser General Public yading@10: * License along with FFmpeg; if not, write to the Free Software yading@10: * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA yading@10: */ yading@10: yading@10: #include "avcodec.h" yading@10: #include "internal.h" yading@10: #include "put_bits.h" yading@10: #include "pnm.h" yading@10: yading@10: yading@10: static int pnm_decode_frame(AVCodecContext *avctx, void *data, yading@10: int *got_frame, AVPacket *avpkt) yading@10: { yading@10: const uint8_t *buf = avpkt->data; yading@10: int buf_size = avpkt->size; yading@10: PNMContext * const s = avctx->priv_data; yading@10: AVFrame * const p = data; yading@10: int i, j, n, linesize, h, upgrade = 0, is_mono = 0; yading@10: unsigned char *ptr; yading@10: int components, sample_len, ret; yading@10: yading@10: s->bytestream_start = yading@10: s->bytestream = (uint8_t *)buf; yading@10: s->bytestream_end = (uint8_t *)buf + buf_size; yading@10: yading@10: if ((ret = ff_pnm_decode_header(avctx, s)) < 0) yading@10: return ret; yading@10: yading@10: if ((ret = ff_get_buffer(avctx, p, 0)) < 0) yading@10: return ret; yading@10: p->pict_type = AV_PICTURE_TYPE_I; yading@10: p->key_frame = 1; yading@10: yading@10: switch (avctx->pix_fmt) { yading@10: default: yading@10: return AVERROR(EINVAL); yading@10: case AV_PIX_FMT_RGBA64BE: yading@10: n = avctx->width * 8; yading@10: components=4; yading@10: sample_len=16; yading@10: goto do_read; yading@10: case AV_PIX_FMT_RGB48BE: yading@10: n = avctx->width * 6; yading@10: components=3; yading@10: sample_len=16; yading@10: goto do_read; yading@10: case AV_PIX_FMT_RGBA: yading@10: n = avctx->width * 4; yading@10: components=4; yading@10: sample_len=8; yading@10: goto do_read; yading@10: case AV_PIX_FMT_RGB24: yading@10: n = avctx->width * 3; yading@10: components=3; yading@10: sample_len=8; yading@10: goto do_read; yading@10: case AV_PIX_FMT_GRAY8: yading@10: n = avctx->width; yading@10: components=1; yading@10: sample_len=8; yading@10: if (s->maxval < 255) yading@10: upgrade = 1; yading@10: goto do_read; yading@10: case AV_PIX_FMT_GRAY8A: yading@10: n = avctx->width * 2; yading@10: components=2; yading@10: sample_len=8; yading@10: goto do_read; yading@10: case AV_PIX_FMT_GRAY16BE: yading@10: case AV_PIX_FMT_GRAY16LE: yading@10: n = avctx->width * 2; yading@10: components=1; yading@10: sample_len=16; yading@10: if (s->maxval < 65535) yading@10: upgrade = 2; yading@10: goto do_read; yading@10: case AV_PIX_FMT_MONOWHITE: yading@10: case AV_PIX_FMT_MONOBLACK: yading@10: n = (avctx->width + 7) >> 3; yading@10: components=1; yading@10: sample_len=1; yading@10: is_mono = 1; yading@10: do_read: yading@10: ptr = p->data[0]; yading@10: linesize = p->linesize[0]; yading@10: if (s->bytestream + n * avctx->height > s->bytestream_end) yading@10: return AVERROR_INVALIDDATA; yading@10: if(s->type < 4 || (is_mono && s->type==7)){ yading@10: for (i=0; iheight; i++) { yading@10: PutBitContext pb; yading@10: init_put_bits(&pb, ptr, linesize); yading@10: for(j=0; jwidth * components; j++){ yading@10: unsigned int c=0; yading@10: int v=0; yading@10: if(s->type < 4) yading@10: while(s->bytestream < s->bytestream_end && (*s->bytestream < '0' || *s->bytestream > '9' )) yading@10: s->bytestream++; yading@10: if(s->bytestream >= s->bytestream_end) yading@10: return AVERROR_INVALIDDATA; yading@10: if (is_mono) { yading@10: /* read a single digit */ yading@10: v = (*s->bytestream++)&1; yading@10: } else { yading@10: /* read a sequence of digits */ yading@10: do { yading@10: v = 10*v + c; yading@10: c = (*s->bytestream++) - '0'; yading@10: } while (c <= 9); yading@10: } yading@10: put_bits(&pb, sample_len, (((1<maxval>>1))/s->maxval); yading@10: } yading@10: flush_put_bits(&pb); yading@10: ptr+= linesize; yading@10: } yading@10: }else{ yading@10: for (i = 0; i < avctx->height; i++) { yading@10: if (!upgrade) yading@10: memcpy(ptr, s->bytestream, n); yading@10: else if (upgrade == 1) { yading@10: unsigned int j, f = (255 * 128 + s->maxval / 2) / s->maxval; yading@10: for (j = 0; j < n; j++) yading@10: ptr[j] = (s->bytestream[j] * f + 64) >> 7; yading@10: } else if (upgrade == 2) { yading@10: unsigned int j, v, f = (65535 * 32768 + s->maxval / 2) / s->maxval; yading@10: for (j = 0; j < n / 2; j++) { yading@10: v = av_be2ne16(((uint16_t *)s->bytestream)[j]); yading@10: ((uint16_t *)ptr)[j] = (v * f + 16384) >> 15; yading@10: } yading@10: } yading@10: s->bytestream += n; yading@10: ptr += linesize; yading@10: } yading@10: } yading@10: break; yading@10: case AV_PIX_FMT_YUV420P: yading@10: case AV_PIX_FMT_YUV420P9BE: yading@10: case AV_PIX_FMT_YUV420P10BE: yading@10: { yading@10: unsigned char *ptr1, *ptr2; yading@10: yading@10: n = avctx->width; yading@10: ptr = p->data[0]; yading@10: linesize = p->linesize[0]; yading@10: if (s->maxval >= 256) yading@10: n *= 2; yading@10: if (s->bytestream + n * avctx->height * 3 / 2 > s->bytestream_end) yading@10: return AVERROR_INVALIDDATA; yading@10: for (i = 0; i < avctx->height; i++) { yading@10: memcpy(ptr, s->bytestream, n); yading@10: s->bytestream += n; yading@10: ptr += linesize; yading@10: } yading@10: ptr1 = p->data[1]; yading@10: ptr2 = p->data[2]; yading@10: n >>= 1; yading@10: h = avctx->height >> 1; yading@10: for (i = 0; i < h; i++) { yading@10: memcpy(ptr1, s->bytestream, n); yading@10: s->bytestream += n; yading@10: memcpy(ptr2, s->bytestream, n); yading@10: s->bytestream += n; yading@10: ptr1 += p->linesize[1]; yading@10: ptr2 += p->linesize[2]; yading@10: } yading@10: } yading@10: break; yading@10: case AV_PIX_FMT_YUV420P16: yading@10: { yading@10: uint16_t *ptr1, *ptr2; yading@10: const int f = (65535 * 32768 + s->maxval / 2) / s->maxval; yading@10: unsigned int j, v; yading@10: yading@10: n = avctx->width * 2; yading@10: ptr = p->data[0]; yading@10: linesize = p->linesize[0]; yading@10: if (s->bytestream + n * avctx->height * 3 / 2 > s->bytestream_end) yading@10: return AVERROR_INVALIDDATA; yading@10: for (i = 0; i < avctx->height; i++) { yading@10: for (j = 0; j < n / 2; j++) { yading@10: v = av_be2ne16(((uint16_t *)s->bytestream)[j]); yading@10: ((uint16_t *)ptr)[j] = (v * f + 16384) >> 15; yading@10: } yading@10: s->bytestream += n; yading@10: ptr += linesize; yading@10: } yading@10: ptr1 = (uint16_t*)p->data[1]; yading@10: ptr2 = (uint16_t*)p->data[2]; yading@10: n >>= 1; yading@10: h = avctx->height >> 1; yading@10: for (i = 0; i < h; i++) { yading@10: for (j = 0; j < n / 2; j++) { yading@10: v = av_be2ne16(((uint16_t *)s->bytestream)[j]); yading@10: ptr1[j] = (v * f + 16384) >> 15; yading@10: } yading@10: s->bytestream += n; yading@10: yading@10: for (j = 0; j < n / 2; j++) { yading@10: v = av_be2ne16(((uint16_t *)s->bytestream)[j]); yading@10: ptr2[j] = (v * f + 16384) >> 15; yading@10: } yading@10: s->bytestream += n; yading@10: yading@10: ptr1 += p->linesize[1] / 2; yading@10: ptr2 += p->linesize[2] / 2; yading@10: } yading@10: } yading@10: break; yading@10: } yading@10: *got_frame = 1; yading@10: yading@10: return s->bytestream - s->bytestream_start; yading@10: } yading@10: yading@10: yading@10: #if CONFIG_PGM_DECODER yading@10: AVCodec ff_pgm_decoder = { yading@10: .name = "pgm", yading@10: .type = AVMEDIA_TYPE_VIDEO, yading@10: .id = AV_CODEC_ID_PGM, yading@10: .priv_data_size = sizeof(PNMContext), yading@10: .decode = pnm_decode_frame, yading@10: .capabilities = CODEC_CAP_DR1, yading@10: .long_name = NULL_IF_CONFIG_SMALL("PGM (Portable GrayMap) image"), yading@10: }; yading@10: #endif yading@10: yading@10: #if CONFIG_PGMYUV_DECODER yading@10: AVCodec ff_pgmyuv_decoder = { yading@10: .name = "pgmyuv", yading@10: .type = AVMEDIA_TYPE_VIDEO, yading@10: .id = AV_CODEC_ID_PGMYUV, yading@10: .priv_data_size = sizeof(PNMContext), yading@10: .decode = pnm_decode_frame, yading@10: .capabilities = CODEC_CAP_DR1, yading@10: .long_name = NULL_IF_CONFIG_SMALL("PGMYUV (Portable GrayMap YUV) image"), yading@10: }; yading@10: #endif yading@10: yading@10: #if CONFIG_PPM_DECODER yading@10: AVCodec ff_ppm_decoder = { yading@10: .name = "ppm", yading@10: .type = AVMEDIA_TYPE_VIDEO, yading@10: .id = AV_CODEC_ID_PPM, yading@10: .priv_data_size = sizeof(PNMContext), yading@10: .decode = pnm_decode_frame, yading@10: .capabilities = CODEC_CAP_DR1, yading@10: .long_name = NULL_IF_CONFIG_SMALL("PPM (Portable PixelMap) image"), yading@10: }; yading@10: #endif yading@10: yading@10: #if CONFIG_PBM_DECODER yading@10: AVCodec ff_pbm_decoder = { yading@10: .name = "pbm", yading@10: .type = AVMEDIA_TYPE_VIDEO, yading@10: .id = AV_CODEC_ID_PBM, yading@10: .priv_data_size = sizeof(PNMContext), yading@10: .decode = pnm_decode_frame, yading@10: .capabilities = CODEC_CAP_DR1, yading@10: .long_name = NULL_IF_CONFIG_SMALL("PBM (Portable BitMap) image"), yading@10: }; yading@10: #endif yading@10: yading@10: #if CONFIG_PAM_DECODER yading@10: AVCodec ff_pam_decoder = { yading@10: .name = "pam", yading@10: .type = AVMEDIA_TYPE_VIDEO, yading@10: .id = AV_CODEC_ID_PAM, yading@10: .priv_data_size = sizeof(PNMContext), yading@10: .decode = pnm_decode_frame, yading@10: .capabilities = CODEC_CAP_DR1, yading@10: .long_name = NULL_IF_CONFIG_SMALL("PAM (Portable AnyMap) image"), yading@10: }; yading@10: #endif