FFmpeg
pnmenc.c
Go to the documentation of this file.
1 /*
2  * PNM image format
3  * Copyright (c) 2002, 2003 Fabrice Bellard
4  *
5  * This file is part of FFmpeg.
6  *
7  * FFmpeg 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.
11  *
12  * FFmpeg 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.
16  *
17  * You should have received a copy of the GNU Lesser General Public
18  * License along with FFmpeg; if not, write to the Free Software
19  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
20  */
21 
22 #include "libavutil/intreadwrite.h"
23 #include "libavutil/imgutils.h"
24 #include "libavutil/pixdesc.h"
25 #include "avcodec.h"
26 #include "internal.h"
27 
29  const AVFrame *p, int *got_packet)
30 {
31  uint8_t *bytestream, *bytestream_start, *bytestream_end;
32  int i, h, h1, c, n, linesize, ret;
33  uint8_t *ptr, *ptr1, *ptr2;
35  avctx->width, avctx->height, 1);
36 
37  if ((ret = ff_alloc_packet2(avctx, pkt, size + 200, 0)) < 0)
38  return ret;
39 
40  bytestream_start =
41  bytestream = pkt->data;
42  bytestream_end = pkt->data + pkt->size;
43 
44  h = avctx->height;
45  h1 = h;
46  switch (avctx->pix_fmt) {
48  c = '4';
49  n = (avctx->width + 7) >> 3;
50  break;
51  case AV_PIX_FMT_GRAY8:
52  c = '5';
53  n = avctx->width;
54  break;
56  c = '5';
57  n = avctx->width * 2;
58  break;
59  case AV_PIX_FMT_RGB24:
60  c = '6';
61  n = avctx->width * 3;
62  break;
63  case AV_PIX_FMT_RGB48BE:
64  c = '6';
65  n = avctx->width * 6;
66  break;
67  case AV_PIX_FMT_YUV420P:
68  if (avctx->width & 1 || avctx->height & 1) {
69  av_log(avctx, AV_LOG_ERROR, "pgmyuv needs even width and height\n");
70  return AVERROR(EINVAL);
71  }
72  c = '5';
73  n = avctx->width;
74  h1 = (h * 3) / 2;
75  break;
77  c = '5';
78  n = avctx->width * 2;
79  h1 = (h * 3) / 2;
80  break;
81  case AV_PIX_FMT_GBRPF32:
82  c = 'F';
83  n = avctx->width * 4;
84  break;
85  default:
86  return -1;
87  }
88  snprintf(bytestream, bytestream_end - bytestream,
89  "P%c\n%d %d\n", c, avctx->width, h1);
90  bytestream += strlen(bytestream);
91  if (avctx->pix_fmt == AV_PIX_FMT_GBRPF32)
92  snprintf(bytestream, bytestream_end - bytestream,
93  "%f\n", avctx->pix_fmt == AV_PIX_FMT_GBRPF32BE ? 1.f: -1.f);
94  bytestream += strlen(bytestream);
95  if (avctx->pix_fmt != AV_PIX_FMT_MONOWHITE &&
96  avctx->pix_fmt != AV_PIX_FMT_GBRPF32) {
97  int maxdepth = (1 << av_pix_fmt_desc_get(avctx->pix_fmt)->comp[0].depth) - 1;
98  snprintf(bytestream, bytestream_end - bytestream,
99  "%d\n", maxdepth);
100  bytestream += strlen(bytestream);
101  }
102 
103  if (avctx->pix_fmt == AV_PIX_FMT_GBRPF32) {
104  float *r = (float *)p->data[2];
105  float *g = (float *)p->data[0];
106  float *b = (float *)p->data[1];
107 
108  for (int i = 0; i < avctx->height; i++) {
109  for (int j = 0; j < avctx->width; j++) {
110  AV_WN32(bytestream + 0, av_float2int(r[j]));
111  AV_WN32(bytestream + 4, av_float2int(g[j]));
112  AV_WN32(bytestream + 8, av_float2int(b[j]));
113  bytestream += 12;
114  }
115 
116  r += p->linesize[2] / 4;
117  g += p->linesize[0] / 4;
118  b += p->linesize[1] / 4;
119  }
120  } else {
121  ptr = p->data[0];
122  linesize = p->linesize[0];
123  for (i = 0; i < h; i++) {
124  memcpy(bytestream, ptr, n);
125  bytestream += n;
126  ptr += linesize;
127  }
128  }
129 
130  if (avctx->pix_fmt == AV_PIX_FMT_YUV420P || avctx->pix_fmt == AV_PIX_FMT_YUV420P16BE) {
131  h >>= 1;
132  n >>= 1;
133  ptr1 = p->data[1];
134  ptr2 = p->data[2];
135  for (i = 0; i < h; i++) {
136  memcpy(bytestream, ptr1, n);
137  bytestream += n;
138  memcpy(bytestream, ptr2, n);
139  bytestream += n;
140  ptr1 += p->linesize[1];
141  ptr2 += p->linesize[2];
142  }
143  }
144  pkt->size = bytestream - bytestream_start;
145  pkt->flags |= AV_PKT_FLAG_KEY;
146  *got_packet = 1;
147 
148  return 0;
149 }
150 
152 {
153 #if FF_API_CODED_FRAME
156  avctx->coded_frame->key_frame = 1;
158 #endif
159 
160  return 0;
161 }
162 
163 #if CONFIG_PGM_ENCODER
165  .name = "pgm",
166  .long_name = NULL_IF_CONFIG_SMALL("PGM (Portable GrayMap) image"),
167  .type = AVMEDIA_TYPE_VIDEO,
168  .id = AV_CODEC_ID_PGM,
169  .init = pnm_encode_init,
170  .encode2 = pnm_encode_frame,
171  .pix_fmts = (const enum AVPixelFormat[]){
173  },
174  .caps_internal = FF_CODEC_CAP_INIT_THREADSAFE,
175 };
176 #endif
177 
178 #if CONFIG_PGMYUV_ENCODER
180  .name = "pgmyuv",
181  .long_name = NULL_IF_CONFIG_SMALL("PGMYUV (Portable GrayMap YUV) image"),
182  .type = AVMEDIA_TYPE_VIDEO,
183  .id = AV_CODEC_ID_PGMYUV,
184  .init = pnm_encode_init,
185  .encode2 = pnm_encode_frame,
186  .pix_fmts = (const enum AVPixelFormat[]){
188  },
189  .caps_internal = FF_CODEC_CAP_INIT_THREADSAFE,
190 };
191 #endif
192 
193 #if CONFIG_PPM_ENCODER
195  .name = "ppm",
196  .long_name = NULL_IF_CONFIG_SMALL("PPM (Portable PixelMap) image"),
197  .type = AVMEDIA_TYPE_VIDEO,
198  .id = AV_CODEC_ID_PPM,
199  .init = pnm_encode_init,
200  .encode2 = pnm_encode_frame,
201  .pix_fmts = (const enum AVPixelFormat[]){
203  },
204  .caps_internal = FF_CODEC_CAP_INIT_THREADSAFE,
205 };
206 #endif
207 
208 #if CONFIG_PBM_ENCODER
210  .name = "pbm",
211  .long_name = NULL_IF_CONFIG_SMALL("PBM (Portable BitMap) image"),
212  .type = AVMEDIA_TYPE_VIDEO,
213  .id = AV_CODEC_ID_PBM,
214  .init = pnm_encode_init,
215  .encode2 = pnm_encode_frame,
216  .pix_fmts = (const enum AVPixelFormat[]){ AV_PIX_FMT_MONOWHITE,
217  AV_PIX_FMT_NONE },
218  .caps_internal = FF_CODEC_CAP_INIT_THREADSAFE,
219 };
220 #endif
221 
222 #if CONFIG_PFM_ENCODER
224  .name = "pfm",
225  .long_name = NULL_IF_CONFIG_SMALL("PFM (Portable FloatMap) image"),
226  .type = AVMEDIA_TYPE_VIDEO,
227  .id = AV_CODEC_ID_PFM,
228  .init = pnm_encode_init,
229  .encode2 = pnm_encode_frame,
230  .pix_fmts = (const enum AVPixelFormat[]){ AV_PIX_FMT_GBRPF32,
231  AV_PIX_FMT_NONE },
232  .caps_internal = FF_CODEC_CAP_INIT_THREADSAFE,
233 };
234 #endif
const AVPixFmtDescriptor * av_pix_fmt_desc_get(enum AVPixelFormat pix_fmt)
Definition: pixdesc.c:2573
This structure describes decoded (raw) audio or video data.
Definition: frame.h:314
misc image utilities
packed RGB 8:8:8, 24bpp, RGBRGB...
Definition: pixfmt.h:68
const char * g
Definition: vf_curves.c:117
AVCodec ff_pbm_encoder
int size
Definition: packet.h:364
enum AVPixelFormat pix_fmt
Pixel format, see AV_PIX_FMT_xxx.
Definition: avcodec.h:741
static AVPacket pkt
AVCodec.
Definition: codec.h:190
int ff_alloc_packet2(AVCodecContext *avctx, AVPacket *avpkt, int64_t size, int64_t min_size)
Check AVPacket size and/or allocate data.
Definition: encode.c:33
#define FF_CODEC_CAP_INIT_THREADSAFE
The codec does not modify any global variables in the init function, allowing to call the init functi...
Definition: internal.h:41
AVComponentDescriptor comp[4]
Parameters that describe how pixels are packed.
Definition: pixdesc.h:117
uint8_t
#define av_cold
Definition: attributes.h:88
Undefined Behavior In the C some operations are like signed integer dereferencing freed accessing outside allocated Undefined Behavior must not occur in a C it is not safe even if the output of undefined operations is unused The unsafety may seem nit picking but Optimizing compilers have in fact optimized code on the assumption that no undefined Behavior occurs Optimizing code based on wrong assumptions can and has in some cases lead to effects beyond the output of computations The signed integer overflow problem in speed critical code Code which is highly optimized and works with signed integers sometimes has the problem that often the output of the computation does not c
Definition: undefined.txt:32
AVCodec ff_pgm_encoder
uint8_t * data
Definition: packet.h:363
ptrdiff_t size
Definition: opengl_enc.c:100
#define av_log(a,...)
#define AV_PKT_FLAG_KEY
The packet contains a keyframe.
Definition: packet.h:401
AVCodec ff_ppm_encoder
int av_image_get_buffer_size(enum AVPixelFormat pix_fmt, int width, int height, int align)
Return the size in bytes of the amount of data required to store an image with the given parameters...
Definition: imgutils.c:466
#define AV_LOG_ERROR
Something went wrong and cannot losslessly be recovered.
Definition: log.h:194
AVCodec ff_pgmyuv_encoder
#define NULL_IF_CONFIG_SMALL(x)
Return NULL if CONFIG_SMALL is true, otherwise the argument without modification. ...
Definition: internal.h:115
static av_cold int pnm_encode_init(AVCodecContext *avctx)
Definition: pnmenc.c:151
const char * r
Definition: vf_curves.c:116
const char * name
Name of the codec implementation.
Definition: codec.h:197
int flags
A combination of AV_PKT_FLAG values.
Definition: packet.h:369
#define b
Definition: input.c:41
enum AVPictureType pict_type
Picture type of the frame.
Definition: frame.h:397
int width
picture width / height.
Definition: avcodec.h:704
static int pnm_encode_frame(AVCodecContext *avctx, AVPacket *pkt, const AVFrame *p, int *got_packet)
Definition: pnmenc.c:28
planar YUV 4:2:0, 24bpp, (1 Cr & Cb sample per 2x2 Y samples), big-endian
Definition: pixfmt.h:132
Libavcodec external API header.
int linesize[AV_NUM_DATA_POINTERS]
For video, size in bytes of each picture line.
Definition: frame.h:345
main external API structure.
Definition: avcodec.h:531
Y , 16bpp, big-endian.
Definition: pixfmt.h:97
IEEE-754 single precision planar GBR 4:4:4, 96bpp, big-endian.
Definition: pixfmt.h:318
#define snprintf
Definition: snprintf.h:34
static av_always_inline uint32_t av_float2int(float f)
Reinterpret a float as a 32-bit integer.
Definition: intfloat.h:50
uint8_t * data[AV_NUM_DATA_POINTERS]
pointer to the picture/channel planes.
Definition: frame.h:328
#define AV_PIX_FMT_GBRPF32
Definition: pixfmt.h:428
AVCodec ff_pfm_encoder
planar YUV 4:2:0, 12bpp, (1 Cr & Cb sample per 2x2 Y samples)
Definition: pixfmt.h:66
Y , 8bpp.
Definition: pixfmt.h:74
#define FF_DISABLE_DEPRECATION_WARNINGS
Definition: internal.h:83
common internal api header.
Y , 1bpp, 0 is white, 1 is black, in each byte pixels are ordered from the msb to the lsb...
Definition: pixfmt.h:75
#define AV_WN32(p, v)
Definition: intreadwrite.h:376
packed RGB 16:16:16, 48bpp, 16R, 16G, 16B, the 2-byte value for each R/G/B component is stored as big...
Definition: pixfmt.h:102
attribute_deprecated AVFrame * coded_frame
the picture in the bitstream
Definition: avcodec.h:1759
#define FF_ENABLE_DEPRECATION_WARNINGS
Definition: internal.h:84
int key_frame
1 -> keyframe, 0-> not
Definition: frame.h:392
Filter the word “frame” indicates either a video frame or a group of audio as stored in an AVFrame structure Format for each input and each output the list of supported formats For video that means pixel format For audio that means channel sample they are references to shared objects When the negotiation mechanism computes the intersection of the formats supported at each end of a all references to both lists are replaced with a reference to the intersection And when a single format is eventually chosen for a link amongst the remaining all references to the list are updated That means that if a filter requires that its input and output have the same format amongst a supported all it has to do is use a reference to the same list of formats query_formats can leave some formats unset and return AVERROR(EAGAIN) to cause the negotiation mechanism toagain later.That can be used by filters with complex requirements to use the format negotiated on one link to set the formats supported on another.Frame references ownership and permissions
int depth
Number of bits in the component.
Definition: pixdesc.h:58
AVPixelFormat
Pixel format.
Definition: pixfmt.h:64
This structure stores compressed data.
Definition: packet.h:340
for(j=16;j >0;--j)
int i
Definition: input.c:407