Libav
img2enc.c
Go to the documentation of this file.
1 /*
2  * Image format
3  * Copyright (c) 2000, 2001, 2002 Fabrice Bellard
4  * Copyright (c) 2004 Michael Niedermayer
5  *
6  * This file is part of Libav.
7  *
8  * Libav is free software; you can redistribute it and/or
9  * modify it under the terms of the GNU Lesser General Public
10  * License as published by the Free Software Foundation; either
11  * version 2.1 of the License, or (at your option) any later version.
12  *
13  * Libav is distributed in the hope that it will be useful,
14  * but WITHOUT ANY WARRANTY; without even the implied warranty of
15  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
16  * Lesser General Public License for more details.
17  *
18  * You should have received a copy of the GNU Lesser General Public
19  * License along with Libav; if not, write to the Free Software
20  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
21  */
22 
23 #include "libavutil/intreadwrite.h"
24 #include "libavutil/avstring.h"
25 #include "libavutil/log.h"
26 #include "avformat.h"
27 #include "avio_internal.h"
28 #include "internal.h"
29 #include "libavutil/opt.h"
30 
31 typedef struct RenameIO {
33  char filename[1024];
34  char tmp[1024];
35 } RenameIO;
36 
37 typedef struct VideoMuxData {
38  const AVClass *class;
40  int is_pipe;
41  char path[1024];
42  int update;
43 } VideoMuxData;
44 
46 {
47  VideoMuxData *img = s->priv_data;
48 
49  av_strlcpy(img->path, s->filename, sizeof(img->path));
50 
51  /* find format */
52  if (s->oformat->flags & AVFMT_NOFILE)
53  img->is_pipe = 0;
54  else
55  img->is_pipe = 1;
56 
57  return 0;
58 }
59 
60 static int open_temporary(AVFormatContext *s, RenameIO *out, const char *filename)
61 {
62  snprintf(out->tmp, sizeof(out->tmp), "%s.tmp", filename);
63  av_strlcpy(out->filename, filename, sizeof(out->filename));
64  if (s->io_open(s, &out->pb, out->tmp, AVIO_FLAG_WRITE, NULL) < 0) {
65  av_log(s, AV_LOG_ERROR, "Could not open file : %s\n", out->tmp);
66  return AVERROR(EIO);
67  }
68 
69  return 0;
70 }
71 
73 {
74  ff_format_io_close(s, &out->pb);
75  ff_rename(out->tmp, out->filename);
76 }
77 
79 {
80  VideoMuxData *img = s->priv_data;
81  RenameIO out[3];
82  char filename[1024];
84  int i;
85 
86  if (!img->is_pipe) {
87  if (img->update) {
88  av_strlcpy(filename, img->path, sizeof(filename));
89  } else if (av_get_frame_filename(filename, sizeof(filename), img->path, img->img_number) < 0 &&
90  img->img_number > 1) {
92  "Could not get frame filename number %d from pattern '%s'\n",
93  img->img_number, img->path);
94  return AVERROR(EIO);
95  }
96  for (i = 0; i < 3; i++) {
97  int ret = open_temporary(s, &out[i], filename);
98  if (ret < 0)
99  return ret;
100  if (par->codec_id != AV_CODEC_ID_RAWVIDEO)
101  break;
102  filename[strlen(filename) - 1] = 'U' + i;
103  }
104  } else {
105  out[0].pb = s->pb;
106  }
107 
108  if (par->codec_id == AV_CODEC_ID_RAWVIDEO) {
109  int ysize = par->width * par->height;
110  avio_write(out[0].pb, pkt->data, ysize);
111  avio_write(out[1].pb, pkt->data + ysize, (pkt->size - ysize) / 2);
112  avio_write(out[2].pb, pkt->data + ysize + (pkt->size - ysize) / 2, (pkt->size - ysize) / 2);
113  close_and_rename(s, &out[1]);
114  close_and_rename(s, &out[2]);
115  } else {
117  AVStream *st = s->streams[0];
118  if (st->codecpar->extradata_size > 8 &&
119  AV_RL32(st->codecpar->extradata + 4) == MKTAG('j', 'p', '2', 'h')) {
120  if (pkt->size < 8 ||
121  AV_RL32(pkt->data + 4) != MKTAG('j', 'p', '2', 'c'))
122  goto error;
123  avio_wb32(out[0].pb, 12);
124  ffio_wfourcc(out[0].pb, "jP ");
125  avio_wb32(out[0].pb, 0x0D0A870A); // signature
126  avio_wb32(out[0].pb, 20);
127  ffio_wfourcc(out[0].pb, "ftyp");
128  ffio_wfourcc(out[0].pb, "jp2 ");
129  avio_wb32(out[0].pb, 0);
130  ffio_wfourcc(out[0].pb, "jp2 ");
131  avio_write(out[0].pb, st->codecpar->extradata, st->codecpar->extradata_size);
132  } else if (pkt->size < 8 ||
133  (!st->codecpar->extradata_size &&
134  AV_RL32(pkt->data + 4) != MKTAG('j', 'P', ' ', ' '))) { // signature
135 error:
136  av_log(s, AV_LOG_ERROR, "malformed JPEG 2000 codestream\n");
137  return -1;
138  }
139  }
140  avio_write(out[0].pb, pkt->data, pkt->size);
141  }
142  avio_flush(out[0].pb);
143  if (!img->is_pipe) {
144  close_and_rename(s, &out[0]);
145  }
146 
147  img->img_number++;
148  return 0;
149 }
150 
151 #define OFFSET(x) offsetof(VideoMuxData, x)
152 #define ENC AV_OPT_FLAG_ENCODING_PARAM
153 static const AVOption muxoptions[] = {
154  { "start_number", "first number in the sequence", OFFSET(img_number), AV_OPT_TYPE_INT, { .i64 = 1 }, 1, INT_MAX, ENC },
155  { "update", "continuously overwrite one file", OFFSET(update), AV_OPT_TYPE_INT, { .i64 = 0 }, 0, 1, ENC },
156  { NULL },
157 };
158 
159 #if CONFIG_IMAGE2_MUXER
160 static const AVClass img2mux_class = {
161  .class_name = "image2 muxer",
162  .item_name = av_default_item_name,
163  .option = muxoptions,
164  .version = LIBAVUTIL_VERSION_INT,
165 };
166 
167 AVOutputFormat ff_image2_muxer = {
168  .name = "image2",
169  .long_name = NULL_IF_CONFIG_SMALL("image2 sequence"),
170  .extensions = "bmp,dpx,jpeg,jpg,ljpg,pam,pbm,pcx,pgm,pgmyuv,png,"
171  "ppm,sgi,tga,tif,tiff,jp2,xwd,sun,ras,rs,im1,im8,im24,"
172  "sunras,webp,xbm,j2c,pix",
173  .priv_data_size = sizeof(VideoMuxData),
174  .video_codec = AV_CODEC_ID_MJPEG,
178  .priv_class = &img2mux_class,
179 };
180 #endif
181 #if CONFIG_IMAGE2PIPE_MUXER
182 AVOutputFormat ff_image2pipe_muxer = {
183  .name = "image2pipe",
184  .long_name = NULL_IF_CONFIG_SMALL("piped image2 sequence"),
185  .priv_data_size = sizeof(VideoMuxData),
186  .video_codec = AV_CODEC_ID_MJPEG,
190 };
191 #endif
int(* io_open)(struct AVFormatContext *s, AVIOContext **pb, const char *url, int flags, AVDictionary **options)
A callback for opening new IO streams.
Definition: avformat.h:1270
Bytestream IO Context.
Definition: avio.h:104
AVOption.
Definition: opt.h:234
#define OFFSET(x)
Definition: img2enc.c:151
enum AVCodecID codec_id
Specific type of the encoded data (the codec used).
Definition: avcodec.h:3483
int size
Definition: avcodec.h:1347
av_log(ac->avr, AV_LOG_TRACE, "%d samples - audio_convert: %s to %s (%s)\, len, av_get_sample_fmt_name(ac->in_fmt), av_get_sample_fmt_name(ac->out_fmt), use_generic ? ac->func_descr_generic :ac->func_descr)
#define AVIO_FLAG_WRITE
write-only
Definition: avio.h:369
This struct describes the properties of an encoded stream.
Definition: avcodec.h:3475
Format I/O context.
Definition: avformat.h:940
const char * class_name
The name of the class; usually it is the same name as the context structure type to which the AVClass...
Definition: log.h:39
int flags
can use flags: AVFMT_NOFILE, AVFMT_NEEDNUMBER, AVFMT_GLOBALHEADER, AVFMT_NOTIMESTAMPS, AVFMT_VARIABLE_FPS, AVFMT_NODIMENSIONS, AVFMT_NOSTREAMS, AVFMT_ALLOW_FLUSH, AVFMT_TS_NONSTRICT, AVFMT_TS_NEGATIVE
Definition: avformat.h:469
int width
Video only.
Definition: avcodec.h:3525
AVOptions.
void ff_format_io_close(AVFormatContext *s, AVIOContext **pb)
A wrapper around AVFormatContext.io_close that should be used instead of calling the pointer directly...
Definition: utils.c:3317
static int open_temporary(AVFormatContext *s, RenameIO *out, const char *filename)
Definition: img2enc.c:60
char path[1024]
Definition: img2enc.c:41
AVStream ** streams
A list of all streams in the file.
Definition: avformat.h:1008
static const AVOption muxoptions[]
Definition: img2enc.c:153
uint8_t * data
Definition: avcodec.h:1346
static int flags
Definition: log.c:50
static int write_header(AVFormatContext *s)
Definition: img2enc.c:45
void avio_write(AVIOContext *s, const unsigned char *buf, int size)
Definition: aviobuf.c:221
static av_always_inline void ffio_wfourcc(AVIOContext *pb, const uint8_t *s)
Definition: avio_internal.h:66
struct AVOutputFormat * oformat
The output container format.
Definition: avformat.h:959
static int ff_rename(const char *oldpath, const char *newpath)
Wrap errno on rename() error.
Definition: internal.h:431
static int write_packet(AVFormatContext *s, AVPacket *pkt)
Definition: img2enc.c:78
#define AV_LOG_ERROR
Something went wrong and cannot losslessly be recovered.
Definition: log.h:124
char filename[1024]
Definition: img2enc.c:33
#define AVERROR(e)
Definition: error.h:43
#define NULL_IF_CONFIG_SMALL(x)
Return NULL if CONFIG_SMALL is true, otherwise the argument without modification. ...
Definition: internal.h:148
size_t av_strlcpy(char *dst, const char *src, size_t size)
Copy the string src to dst, but no more than size - 1 bytes, and null-terminate dst.
Definition: avstring.c:81
int extradata_size
Size of the extradata content in bytes.
Definition: avcodec.h:3501
AVIOContext * pb
Definition: img2enc.c:32
int void avio_flush(AVIOContext *s)
Definition: aviobuf.c:236
char filename[1024]
input or output filename
Definition: avformat.h:1016
const char * name
Definition: avformat.h:450
#define AV_RL32
Definition: intreadwrite.h:146
LIBAVUTIL_VERSION_INT
Definition: eval.c:55
int av_get_frame_filename(char *buf, int buf_size, const char *path, int number)
Return in &#39;buf&#39; the path with &#39;d&#39; replaced by a number.
Definition: utils.c:2817
Stream structure.
Definition: avformat.h:705
#define AVFMT_NOTIMESTAMPS
Format does not need / have any timestamps.
Definition: avformat.h:420
NULL
Definition: eval.c:55
static void close_and_rename(AVFormatContext *s, RenameIO *out)
Definition: img2enc.c:72
int img_number
Definition: img2enc.c:39
AVIOContext * pb
I/O context.
Definition: avformat.h:982
av_default_item_name
Definition: dnxhdenc.c:55
char tmp[1024]
Definition: img2enc.c:34
Describe the class of an AVClass context structure.
Definition: log.h:34
#define ENC
Definition: img2enc.c:152
Main libavformat public API header.
int update
Definition: img2enc.c:42
#define AVFMT_NOFILE
Demuxer will use avio_open, no opened file should be provided by the caller.
Definition: avformat.h:412
enum AVCodecID ff_guess_image2_codec(const char *filename)
Definition: img2.c:101
int is_pipe
Definition: img2enc.c:40
void * priv_data
Format private data.
Definition: avformat.h:968
#define AVFMT_NODIMENSIONS
Format does not need width/height.
Definition: avformat.h:424
uint8_t * extradata
Extra binary data needed for initializing the decoder, codec-dependent.
Definition: avcodec.h:3497
void avio_wb32(AVIOContext *s, unsigned int val)
Definition: aviobuf.c:324
FILE * out
Definition: movenc.c:54
AVCodecParameters * codecpar
Definition: avformat.h:831
int stream_index
Definition: avcodec.h:1348
#define MKTAG(a, b, c, d)
Definition: common.h:256
This structure stores compressed data.
Definition: avcodec.h:1323