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 "libavutil/pixdesc.h" yading@10: #include "avcodec.h" yading@10: #include "internal.h" yading@10: #include "pnm.h" yading@10: yading@10: yading@10: static int pnm_encode_frame(AVCodecContext *avctx, AVPacket *pkt, yading@10: const AVFrame *pict, int *got_packet) yading@10: { yading@10: PNMContext *s = avctx->priv_data; yading@10: AVFrame * const p = &s->picture; yading@10: int i, h, h1, c, n, linesize, ret; yading@10: uint8_t *ptr, *ptr1, *ptr2; yading@10: yading@10: if ((ret = ff_alloc_packet2(avctx, pkt, avpicture_get_size(avctx->pix_fmt, yading@10: avctx->width, yading@10: avctx->height) + 200)) < 0) yading@10: return ret; yading@10: yading@10: *p = *pict; yading@10: p->pict_type = AV_PICTURE_TYPE_I; yading@10: p->key_frame = 1; yading@10: yading@10: s->bytestream_start = yading@10: s->bytestream = pkt->data; yading@10: s->bytestream_end = pkt->data + pkt->size; yading@10: yading@10: h = avctx->height; yading@10: h1 = h; yading@10: switch (avctx->pix_fmt) { yading@10: case AV_PIX_FMT_MONOWHITE: yading@10: c = '4'; yading@10: n = (avctx->width + 7) >> 3; yading@10: break; yading@10: case AV_PIX_FMT_GRAY8: yading@10: c = '5'; yading@10: n = avctx->width; yading@10: break; yading@10: case AV_PIX_FMT_GRAY16BE: yading@10: c = '5'; yading@10: n = avctx->width * 2; yading@10: break; yading@10: case AV_PIX_FMT_RGB24: yading@10: c = '6'; yading@10: n = avctx->width * 3; yading@10: break; yading@10: case AV_PIX_FMT_RGB48BE: yading@10: c = '6'; yading@10: n = avctx->width * 6; yading@10: break; yading@10: case AV_PIX_FMT_YUV420P: yading@10: if (avctx->width & 1 || avctx->height & 1) { yading@10: av_log(avctx, AV_LOG_ERROR, "pgmyuv needs even width and height\n"); yading@10: return AVERROR(EINVAL); yading@10: } yading@10: c = '5'; yading@10: n = avctx->width; yading@10: h1 = (h * 3) / 2; yading@10: break; yading@10: case AV_PIX_FMT_YUV420P16BE: yading@10: c = '5'; yading@10: n = avctx->width * 2; yading@10: h1 = (h * 3) / 2; yading@10: break; yading@10: default: yading@10: return -1; yading@10: } yading@10: snprintf(s->bytestream, s->bytestream_end - s->bytestream, yading@10: "P%c\n%d %d\n", c, avctx->width, h1); yading@10: s->bytestream += strlen(s->bytestream); yading@10: if (avctx->pix_fmt != AV_PIX_FMT_MONOWHITE) { yading@10: int maxdepth = (1 << (av_pix_fmt_desc_get(avctx->pix_fmt)->comp[0].depth_minus1 + 1)) - 1; yading@10: snprintf(s->bytestream, s->bytestream_end - s->bytestream, yading@10: "%d\n", maxdepth); yading@10: s->bytestream += strlen(s->bytestream); yading@10: } yading@10: yading@10: ptr = p->data[0]; yading@10: linesize = p->linesize[0]; yading@10: for (i = 0; i < h; i++) { yading@10: memcpy(s->bytestream, ptr, n); yading@10: s->bytestream += n; yading@10: ptr += linesize; yading@10: } yading@10: yading@10: if (avctx->pix_fmt == AV_PIX_FMT_YUV420P || avctx->pix_fmt == AV_PIX_FMT_YUV420P16BE) { yading@10: h >>= 1; yading@10: n >>= 1; yading@10: ptr1 = p->data[1]; yading@10: ptr2 = p->data[2]; yading@10: for (i = 0; i < h; i++) { yading@10: memcpy(s->bytestream, ptr1, n); yading@10: s->bytestream += n; yading@10: memcpy(s->bytestream, ptr2, n); yading@10: s->bytestream += n; yading@10: ptr1 += p->linesize[1]; yading@10: ptr2 += p->linesize[2]; yading@10: } yading@10: } yading@10: pkt->size = s->bytestream - s->bytestream_start; yading@10: pkt->flags |= AV_PKT_FLAG_KEY; yading@10: *got_packet = 1; yading@10: yading@10: return 0; yading@10: } yading@10: yading@10: yading@10: #if CONFIG_PGM_ENCODER yading@10: AVCodec ff_pgm_encoder = { 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: .init = ff_pnm_init, yading@10: .encode2 = pnm_encode_frame, yading@10: .pix_fmts = (const enum AVPixelFormat[]){ yading@10: AV_PIX_FMT_GRAY8, AV_PIX_FMT_GRAY16BE, AV_PIX_FMT_NONE yading@10: }, yading@10: .long_name = NULL_IF_CONFIG_SMALL("PGM (Portable GrayMap) image"), yading@10: }; yading@10: #endif yading@10: yading@10: #if CONFIG_PGMYUV_ENCODER yading@10: AVCodec ff_pgmyuv_encoder = { 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: .init = ff_pnm_init, yading@10: .encode2 = pnm_encode_frame, yading@10: .pix_fmts = (const enum AVPixelFormat[]){ yading@10: AV_PIX_FMT_YUV420P, AV_PIX_FMT_YUV420P16BE, AV_PIX_FMT_NONE yading@10: }, 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_ENCODER yading@10: AVCodec ff_ppm_encoder = { 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: .init = ff_pnm_init, yading@10: .encode2 = pnm_encode_frame, yading@10: .pix_fmts = (const enum AVPixelFormat[]){ yading@10: AV_PIX_FMT_RGB24, AV_PIX_FMT_RGB48BE, AV_PIX_FMT_NONE yading@10: }, yading@10: .long_name = NULL_IF_CONFIG_SMALL("PPM (Portable PixelMap) image"), yading@10: }; yading@10: #endif yading@10: yading@10: #if CONFIG_PBM_ENCODER yading@10: AVCodec ff_pbm_encoder = { 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: .init = ff_pnm_init, yading@10: .encode2 = pnm_encode_frame, yading@10: .pix_fmts = (const enum AVPixelFormat[]){ AV_PIX_FMT_MONOWHITE, yading@10: AV_PIX_FMT_NONE }, yading@10: .long_name = NULL_IF_CONFIG_SMALL("PBM (Portable BitMap) image"), yading@10: }; yading@10: #endif