3 * Copyright (c) 2002, 2003 Fabrice Bellard
5 * This file is part of Libav.
7 * Libav is free software; you can redistribute it and/or
8 * modify it under the terms of the GNU Lesser General Public
9 * License as published by the Free Software Foundation; either
10 * version 2.1 of the License, or (at your option) any later version.
12 * Libav is distributed in the hope that it will be useful,
13 * but WITHOUT ANY WARRANTY; without even the implied warranty of
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
15 * Lesser General Public License for more details.
17 * You should have received a copy of the GNU Lesser General Public
18 * License along with Libav; if not, write to the Free Software
19 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
23 #include "bytestream.h"
28 static int pnm_encode_frame(AVCodecContext
*avctx
, AVPacket
*pkt
,
29 const AVFrame
*pict
, int *got_packet
)
31 PNMContext
*s
= avctx
->priv_data
;
32 AVFrame
* const p
= &s
->picture
;
33 int i
, h
, h1
, c
, n
, linesize
, ret
;
34 uint8_t *ptr
, *ptr1
, *ptr2
;
36 if ((ret
= ff_alloc_packet(pkt
, avpicture_get_size(avctx
->pix_fmt
,
38 avctx
->height
) + 200)) < 0) {
39 av_log(avctx
, AV_LOG_ERROR
, "encoded frame too large\n");
44 p
->pict_type
= AV_PICTURE_TYPE_I
;
48 s
->bytestream
= pkt
->data
;
49 s
->bytestream_end
= pkt
->data
+ pkt
->size
;
53 switch (avctx
->pix_fmt
) {
54 case AV_PIX_FMT_MONOWHITE
:
56 n
= (avctx
->width
+ 7) >> 3;
58 case AV_PIX_FMT_GRAY8
:
62 case AV_PIX_FMT_GRAY16BE
:
66 case AV_PIX_FMT_RGB24
:
70 case AV_PIX_FMT_RGB48BE
:
74 case AV_PIX_FMT_YUV420P
:
82 snprintf(s
->bytestream
, s
->bytestream_end
- s
->bytestream
,
83 "P%c\n%d %d\n", c
, avctx
->width
, h1
);
84 s
->bytestream
+= strlen(s
->bytestream
);
85 if (avctx
->pix_fmt
!= AV_PIX_FMT_MONOWHITE
) {
86 snprintf(s
->bytestream
, s
->bytestream_end
- s
->bytestream
,
87 "%d\n", (avctx
->pix_fmt
!= AV_PIX_FMT_GRAY16BE
&& avctx
->pix_fmt
!= AV_PIX_FMT_RGB48BE
) ? 255 : 65535);
88 s
->bytestream
+= strlen(s
->bytestream
);
92 linesize
= p
->linesize
[0];
93 for (i
= 0; i
< h
; i
++) {
94 memcpy(s
->bytestream
, ptr
, n
);
99 if (avctx
->pix_fmt
== AV_PIX_FMT_YUV420P
) {
104 for (i
= 0; i
< h
; i
++) {
105 memcpy(s
->bytestream
, ptr1
, n
);
107 memcpy(s
->bytestream
, ptr2
, n
);
109 ptr1
+= p
->linesize
[1];
110 ptr2
+= p
->linesize
[2];
113 pkt
->size
= s
->bytestream
- s
->bytestream_start
;
114 pkt
->flags
|= AV_PKT_FLAG_KEY
;
121 #if CONFIG_PGM_ENCODER
122 AVCodec ff_pgm_encoder
= {
124 .type
= AVMEDIA_TYPE_VIDEO
,
125 .id
= AV_CODEC_ID_PGM
,
126 .priv_data_size
= sizeof(PNMContext
),
128 .encode2
= pnm_encode_frame
,
129 .pix_fmts
= (const enum AVPixelFormat
[]){
130 AV_PIX_FMT_GRAY8
, AV_PIX_FMT_GRAY16BE
, AV_PIX_FMT_NONE
132 .long_name
= NULL_IF_CONFIG_SMALL("PGM (Portable GrayMap) image"),
136 #if CONFIG_PGMYUV_ENCODER
137 AVCodec ff_pgmyuv_encoder
= {
139 .type
= AVMEDIA_TYPE_VIDEO
,
140 .id
= AV_CODEC_ID_PGMYUV
,
141 .priv_data_size
= sizeof(PNMContext
),
143 .encode2
= pnm_encode_frame
,
144 .pix_fmts
= (const enum AVPixelFormat
[]){ AV_PIX_FMT_YUV420P
, AV_PIX_FMT_NONE
},
145 .long_name
= NULL_IF_CONFIG_SMALL("PGMYUV (Portable GrayMap YUV) image"),
149 #if CONFIG_PPM_ENCODER
150 AVCodec ff_ppm_encoder
= {
152 .type
= AVMEDIA_TYPE_VIDEO
,
153 .id
= AV_CODEC_ID_PPM
,
154 .priv_data_size
= sizeof(PNMContext
),
156 .encode2
= pnm_encode_frame
,
157 .pix_fmts
= (const enum AVPixelFormat
[]){
158 AV_PIX_FMT_RGB24
, AV_PIX_FMT_RGB48BE
, AV_PIX_FMT_NONE
160 .long_name
= NULL_IF_CONFIG_SMALL("PPM (Portable PixelMap) image"),
164 #if CONFIG_PBM_ENCODER
165 AVCodec ff_pbm_encoder
= {
167 .type
= AVMEDIA_TYPE_VIDEO
,
168 .id
= AV_CODEC_ID_PBM
,
169 .priv_data_size
= sizeof(PNMContext
),
171 .encode2
= pnm_encode_frame
,
172 .pix_fmts
= (const enum AVPixelFormat
[]){ AV_PIX_FMT_MONOWHITE
,
174 .long_name
= NULL_IF_CONFIG_SMALL("PBM (Portable BitMap) image"),