img2enc: ensure that the codec is rawvideo for split planes mode.

Found-by: Paul B Mahol <onemda@gmail.com>
Signed-off-by: Michael Niedermayer <michaelni@gmx.at>
This commit is contained in:
Michael Niedermayer 2012-11-06 01:13:12 +01:00
parent 7f6f8f642c
commit 6dfcc7abdd
1 changed files with 5 additions and 1 deletions

View File

@ -41,6 +41,7 @@ typedef struct {
static int write_header(AVFormatContext *s)
{
VideoMuxData *img = s->priv_data;
AVStream *st = s->streams[0];
const char *str;
av_strlcpy(img->path, s->filename, sizeof(img->path));
@ -52,7 +53,10 @@ static int write_header(AVFormatContext *s)
img->is_pipe = 1;
str = strrchr(img->path, '.');
img->split_planes = str && !av_strcasecmp(str + 1, "y");
img->split_planes = str
&& !av_strcasecmp(str + 1, "y")
&& s->nb_streams == 1
&& st->codec->codec_id == AV_CODEC_ID_RAWVIDEO;
return 0;
}