Commit 9ec2b9fc authored by Michael Niedermayer's avatar Michael Niedermayer

avformat/img2enc: Disable rename&atomic writing for non file protocol and split planes

For protocols other than local files ff_rename() is not implemented
For split planes support the implementation is simply wrong
Signed-off-by: 's avatarMichael Niedermayer <michael@niedermayer.cc>
parent 6255bf3d
...@@ -42,6 +42,7 @@ typedef struct VideoMuxData { ...@@ -42,6 +42,7 @@ typedef struct VideoMuxData {
int update; int update;
int use_strftime; int use_strftime;
const char *muxer; const char *muxer;
int use_rename;
} VideoMuxData; } VideoMuxData;
static int write_header(AVFormatContext *s) static int write_header(AVFormatContext *s)
...@@ -49,6 +50,7 @@ static int write_header(AVFormatContext *s) ...@@ -49,6 +50,7 @@ static int write_header(AVFormatContext *s)
VideoMuxData *img = s->priv_data; VideoMuxData *img = s->priv_data;
AVStream *st = s->streams[0]; AVStream *st = s->streams[0];
const AVPixFmtDescriptor *desc = av_pix_fmt_desc_get(st->codec->pix_fmt); const AVPixFmtDescriptor *desc = av_pix_fmt_desc_get(st->codec->pix_fmt);
const char *proto = avio_find_protocol_name(s->filename);
av_strlcpy(img->path, s->filename, sizeof(img->path)); av_strlcpy(img->path, s->filename, sizeof(img->path));
snprintf(img->tmp, sizeof(img->tmp), "%s.tmp", s->filename); snprintf(img->tmp, sizeof(img->tmp), "%s.tmp", s->filename);
...@@ -70,6 +72,13 @@ static int write_header(AVFormatContext *s) ...@@ -70,6 +72,13 @@ static int write_header(AVFormatContext *s)
&&(desc->flags & AV_PIX_FMT_FLAG_PLANAR) &&(desc->flags & AV_PIX_FMT_FLAG_PLANAR)
&& desc->nb_components >= 3; && desc->nb_components >= 3;
} }
img->use_rename = proto && !strcmp(proto, "file");
//The current atomic rename implementation is not compatible with split planes
if (img->split_planes)
img->use_rename = 0;
return 0; return 0;
} }
...@@ -102,9 +111,9 @@ static int write_packet(AVFormatContext *s, AVPacket *pkt) ...@@ -102,9 +111,9 @@ static int write_packet(AVFormatContext *s, AVPacket *pkt)
return AVERROR(EINVAL); return AVERROR(EINVAL);
} }
for (i = 0; i < 4; i++) { for (i = 0; i < 4; i++) {
if (avio_open2(&pb[i], img->tmp, AVIO_FLAG_WRITE, if (avio_open2(&pb[i], img->use_rename ? img->tmp : filename, AVIO_FLAG_WRITE,
&s->interrupt_callback, NULL) < 0) { &s->interrupt_callback, NULL) < 0) {
av_log(s, AV_LOG_ERROR, "Could not open file : %s\n", img->tmp); av_log(s, AV_LOG_ERROR, "Could not open file : %s\n", img->use_rename ? img->tmp : filename);
return AVERROR(EIO); return AVERROR(EIO);
} }
...@@ -169,7 +178,8 @@ static int write_packet(AVFormatContext *s, AVPacket *pkt) ...@@ -169,7 +178,8 @@ static int write_packet(AVFormatContext *s, AVPacket *pkt)
avio_flush(pb[0]); avio_flush(pb[0]);
if (!img->is_pipe) { if (!img->is_pipe) {
avio_closep(&pb[0]); avio_closep(&pb[0]);
ff_rename(img->tmp, filename, s); if (img->use_rename)
ff_rename(img->tmp, filename, s);
} }
img->img_number++; img->img_number++;
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment