|
|
|
@@ -80,13 +80,14 @@ static int infer_size(int *width_ptr, int *height_ptr, int size) |
|
|
|
{ |
|
|
|
int i; |
|
|
|
|
|
|
|
for(i=0;i<FF_ARRAY_ELEMS(sizes);i++) { |
|
|
|
for (i = 0; i < FF_ARRAY_ELEMS(sizes); i++) { |
|
|
|
if ((sizes[i][0] * sizes[i][1]) == size) { |
|
|
|
*width_ptr = sizes[i][0]; |
|
|
|
*width_ptr = sizes[i][0]; |
|
|
|
*height_ptr = sizes[i][1]; |
|
|
|
return 0; |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
return -1; |
|
|
|
} |
|
|
|
|
|
|
|
@@ -128,9 +129,9 @@ static int find_image_range(int *pfirst_index, int *plast_index, |
|
|
|
|
|
|
|
/* find the first image */ |
|
|
|
for (first_index = start_index; first_index < start_index + start_index_range; first_index++) { |
|
|
|
if (av_get_frame_filename(buf, sizeof(buf), path, first_index) < 0){ |
|
|
|
if (av_get_frame_filename(buf, sizeof(buf), path, first_index) < 0) { |
|
|
|
*pfirst_index = |
|
|
|
*plast_index = 1; |
|
|
|
*plast_index = 1; |
|
|
|
if (avio_check(buf, AVIO_FLAG_READ) > 0) |
|
|
|
return 0; |
|
|
|
return -1; |
|
|
|
@@ -143,9 +144,9 @@ static int find_image_range(int *pfirst_index, int *plast_index, |
|
|
|
|
|
|
|
/* find the last image */ |
|
|
|
last_index = first_index; |
|
|
|
for(;;) { |
|
|
|
for (;;) { |
|
|
|
range = 0; |
|
|
|
for(;;) { |
|
|
|
for (;;) { |
|
|
|
if (!range) |
|
|
|
range1 = 1; |
|
|
|
else |
|
|
|
@@ -166,14 +167,14 @@ static int find_image_range(int *pfirst_index, int *plast_index, |
|
|
|
last_index += range; |
|
|
|
} |
|
|
|
*pfirst_index = first_index; |
|
|
|
*plast_index = last_index; |
|
|
|
*plast_index = last_index; |
|
|
|
return 0; |
|
|
|
fail: |
|
|
|
|
|
|
|
fail: |
|
|
|
return -1; |
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
static int read_probe(AVProbeData *p) |
|
|
|
static int img_read_probe(AVProbeData *p) |
|
|
|
{ |
|
|
|
if (p->filename && ff_guess_image2_codec(p->filename)) { |
|
|
|
if (av_filename_number_test(p->filename)) |
|
|
|
@@ -183,12 +184,12 @@ static int read_probe(AVProbeData *p) |
|
|
|
else if(av_match_ext(p->filename, "raw")) |
|
|
|
return 5; |
|
|
|
else |
|
|
|
return AVPROBE_SCORE_MAX/2; |
|
|
|
return AVPROBE_SCORE_MAX / 2; |
|
|
|
} |
|
|
|
return 0; |
|
|
|
} |
|
|
|
|
|
|
|
static int read_header(AVFormatContext *s1) |
|
|
|
static int img_read_header(AVFormatContext *s1) |
|
|
|
{ |
|
|
|
VideoDemuxData *s = s1->priv_data; |
|
|
|
int first_index, last_index, ret = 0; |
|
|
|
@@ -204,28 +205,33 @@ static int read_header(AVFormatContext *s1) |
|
|
|
return AVERROR(ENOMEM); |
|
|
|
} |
|
|
|
|
|
|
|
if (s->pixel_format && (pix_fmt = av_get_pix_fmt(s->pixel_format)) == AV_PIX_FMT_NONE) { |
|
|
|
av_log(s1, AV_LOG_ERROR, "No such pixel format: %s.\n", s->pixel_format); |
|
|
|
if (s->pixel_format && |
|
|
|
(pix_fmt = av_get_pix_fmt(s->pixel_format)) == AV_PIX_FMT_NONE) { |
|
|
|
av_log(s1, AV_LOG_ERROR, "No such pixel format: %s.\n", |
|
|
|
s->pixel_format); |
|
|
|
return AVERROR(EINVAL); |
|
|
|
} |
|
|
|
if (s->video_size && (ret = av_parse_video_size(&width, &height, s->video_size)) < 0) { |
|
|
|
av_log(s, AV_LOG_ERROR, "Could not parse video size: %s.\n", s->video_size); |
|
|
|
if (s->video_size && |
|
|
|
(ret = av_parse_video_size(&width, &height, s->video_size)) < 0) { |
|
|
|
av_log(s, AV_LOG_ERROR, |
|
|
|
"Could not parse video size: %s.\n", s->video_size); |
|
|
|
return ret; |
|
|
|
} |
|
|
|
if ((ret = av_parse_video_rate(&framerate, s->framerate)) < 0) { |
|
|
|
av_log(s, AV_LOG_ERROR, "Could not parse framerate: %s.\n", s->framerate); |
|
|
|
av_log(s, AV_LOG_ERROR, |
|
|
|
"Could not parse framerate: %s.\n", s->framerate); |
|
|
|
return ret; |
|
|
|
} |
|
|
|
|
|
|
|
av_strlcpy(s->path, s1->filename, sizeof(s->path)); |
|
|
|
s->img_number = 0; |
|
|
|
s->img_count = 0; |
|
|
|
s->img_count = 0; |
|
|
|
|
|
|
|
/* find format */ |
|
|
|
if (s1->iformat->flags & AVFMT_NOFILE) |
|
|
|
s->is_pipe = 0; |
|
|
|
else{ |
|
|
|
s->is_pipe = 1; |
|
|
|
else { |
|
|
|
s->is_pipe = 1; |
|
|
|
st->need_parsing = AVSTREAM_PARSE_FULL; |
|
|
|
} |
|
|
|
|
|
|
|
@@ -298,43 +304,44 @@ static int read_header(AVFormatContext *s1) |
|
|
|
"Unknown value '%d' for pattern_type option\n", s->pattern_type); |
|
|
|
return AVERROR(EINVAL); |
|
|
|
} |
|
|
|
s->img_first = first_index; |
|
|
|
s->img_last = last_index; |
|
|
|
s->img_first = first_index; |
|
|
|
s->img_last = last_index; |
|
|
|
s->img_number = first_index; |
|
|
|
/* compute duration */ |
|
|
|
st->start_time = 0; |
|
|
|
st->duration = last_index - first_index + 1; |
|
|
|
st->duration = last_index - first_index + 1; |
|
|
|
} |
|
|
|
|
|
|
|
if(s1->video_codec_id){ |
|
|
|
if (s1->video_codec_id) { |
|
|
|
st->codec->codec_type = AVMEDIA_TYPE_VIDEO; |
|
|
|
st->codec->codec_id = s1->video_codec_id; |
|
|
|
}else if(s1->audio_codec_id){ |
|
|
|
st->codec->codec_id = s1->video_codec_id; |
|
|
|
} else if (s1->audio_codec_id) { |
|
|
|
st->codec->codec_type = AVMEDIA_TYPE_AUDIO; |
|
|
|
st->codec->codec_id = s1->audio_codec_id; |
|
|
|
}else{ |
|
|
|
const char *str= strrchr(s->path, '.'); |
|
|
|
s->split_planes = str && !av_strcasecmp(str + 1, "y"); |
|
|
|
st->codec->codec_id = s1->audio_codec_id; |
|
|
|
} else { |
|
|
|
const char *str = strrchr(s->path, '.'); |
|
|
|
s->split_planes = str && !av_strcasecmp(str + 1, "y"); |
|
|
|
st->codec->codec_type = AVMEDIA_TYPE_VIDEO; |
|
|
|
st->codec->codec_id = ff_guess_image2_codec(s->path); |
|
|
|
st->codec->codec_id = ff_guess_image2_codec(s->path); |
|
|
|
if (st->codec->codec_id == AV_CODEC_ID_LJPEG) |
|
|
|
st->codec->codec_id = AV_CODEC_ID_MJPEG; |
|
|
|
} |
|
|
|
if(st->codec->codec_type == AVMEDIA_TYPE_VIDEO && pix_fmt != AV_PIX_FMT_NONE) |
|
|
|
if (st->codec->codec_type == AVMEDIA_TYPE_VIDEO && |
|
|
|
pix_fmt != AV_PIX_FMT_NONE) |
|
|
|
st->codec->pix_fmt = pix_fmt; |
|
|
|
|
|
|
|
return 0; |
|
|
|
} |
|
|
|
|
|
|
|
static int read_packet(AVFormatContext *s1, AVPacket *pkt) |
|
|
|
static int img_read_packet(AVFormatContext *s1, AVPacket *pkt) |
|
|
|
{ |
|
|
|
VideoDemuxData *s = s1->priv_data; |
|
|
|
char filename_bytes[1024]; |
|
|
|
char *filename = filename_bytes; |
|
|
|
int i; |
|
|
|
int size[3]={0}, ret[3]={0}; |
|
|
|
AVIOContext *f[3] = {NULL}; |
|
|
|
AVCodecContext *codec= s1->streams[0]->codec; |
|
|
|
int size[3] = { 0 }, ret[3] = { 0 }; |
|
|
|
AVIOContext *f[3] = { NULL }; |
|
|
|
AVCodecContext *codec = s1->streams[0]->codec; |
|
|
|
|
|
|
|
if (!s->is_pipe) { |
|
|
|
/* loop over input */ |
|
|
|
@@ -349,25 +356,27 @@ static int read_packet(AVFormatContext *s1, AVPacket *pkt) |
|
|
|
#endif |
|
|
|
} else { |
|
|
|
if (av_get_frame_filename(filename_bytes, sizeof(filename_bytes), |
|
|
|
s->path, s->img_number)<0 && s->img_number > 1) |
|
|
|
s->path, |
|
|
|
s->img_number) < 0 && s->img_number > 1) |
|
|
|
return AVERROR(EIO); |
|
|
|
} |
|
|
|
for(i=0; i<3; i++){ |
|
|
|
for (i = 0; i < 3; i++) { |
|
|
|
if (avio_open2(&f[i], filename, AVIO_FLAG_READ, |
|
|
|
&s1->interrupt_callback, NULL) < 0) { |
|
|
|
if(i>=1) |
|
|
|
if (i >= 1) |
|
|
|
break; |
|
|
|
av_log(s1, AV_LOG_ERROR, "Could not open file : %s\n",filename); |
|
|
|
av_log(s1, AV_LOG_ERROR, "Could not open file : %s\n", |
|
|
|
filename); |
|
|
|
return AVERROR(EIO); |
|
|
|
} |
|
|
|
size[i]= avio_size(f[i]); |
|
|
|
size[i] = avio_size(f[i]); |
|
|
|
|
|
|
|
if(!s->split_planes) |
|
|
|
if (!s->split_planes) |
|
|
|
break; |
|
|
|
filename[ strlen(filename) - 1 ]= 'U' + i; |
|
|
|
filename[strlen(filename) - 1] = 'U' + i; |
|
|
|
} |
|
|
|
|
|
|
|
if(codec->codec_id == AV_CODEC_ID_RAWVIDEO && !codec->width) |
|
|
|
if (codec->codec_id == AV_CODEC_ID_RAWVIDEO && !codec->width) |
|
|
|
infer_size(&codec->width, &codec->height, size[0]); |
|
|
|
} else { |
|
|
|
f[0] = s1->pb; |
|
|
|
@@ -376,27 +385,27 @@ static int read_packet(AVFormatContext *s1, AVPacket *pkt) |
|
|
|
if (s->frame_size > 0) { |
|
|
|
size[0] = s->frame_size; |
|
|
|
} else { |
|
|
|
size[0]= 4096; |
|
|
|
size[0] = 4096; |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
if (av_new_packet(pkt, size[0] + size[1] + size[2]) < 0) |
|
|
|
return AVERROR(ENOMEM); |
|
|
|
pkt->stream_index = 0; |
|
|
|
pkt->flags |= AV_PKT_FLAG_KEY; |
|
|
|
pkt->flags |= AV_PKT_FLAG_KEY; |
|
|
|
|
|
|
|
pkt->size= 0; |
|
|
|
for(i=0; i<3; i++){ |
|
|
|
if(f[i]){ |
|
|
|
ret[i]= avio_read(f[i], pkt->data + pkt->size, size[i]); |
|
|
|
pkt->size = 0; |
|
|
|
for (i = 0; i < 3; i++) { |
|
|
|
if (f[i]) { |
|
|
|
ret[i] = avio_read(f[i], pkt->data + pkt->size, size[i]); |
|
|
|
if (!s->is_pipe) |
|
|
|
avio_close(f[i]); |
|
|
|
if(ret[i]>0) |
|
|
|
if (ret[i] > 0) |
|
|
|
pkt->size += ret[i]; |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
if (ret[0] <= 0 || ret[1]<0 || ret[2]<0) { |
|
|
|
if (ret[0] <= 0 || ret[1] < 0 || ret[2] < 0) { |
|
|
|
av_free_packet(pkt); |
|
|
|
return AVERROR(EIO); /* signal EOF */ |
|
|
|
} else { |
|
|
|
@@ -406,7 +415,7 @@ static int read_packet(AVFormatContext *s1, AVPacket *pkt) |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
static int read_close(struct AVFormatContext* s1) |
|
|
|
static int img_read_close(struct AVFormatContext* s1) |
|
|
|
{ |
|
|
|
VideoDemuxData *s = s1->priv_data; |
|
|
|
#if HAVE_GLOB |
|
|
|
@@ -420,19 +429,19 @@ static int read_close(struct AVFormatContext* s1) |
|
|
|
#define OFFSET(x) offsetof(VideoDemuxData, x) |
|
|
|
#define DEC AV_OPT_FLAG_DECODING_PARAM |
|
|
|
static const AVOption options[] = { |
|
|
|
{ "framerate", "set the video framerate", OFFSET(framerate), AV_OPT_TYPE_STRING, {.str = "25"}, 0, 0, DEC }, |
|
|
|
{ "loop", "force loop over input file sequence", OFFSET(loop), AV_OPT_TYPE_INT, {.i64 = 0}, 0, 1, DEC }, |
|
|
|
{ "framerate", "set the video framerate", OFFSET(framerate), AV_OPT_TYPE_STRING, {.str = "25"}, 0, 0, DEC }, |
|
|
|
{ "loop", "force loop over input file sequence", OFFSET(loop), AV_OPT_TYPE_INT, {.i64 = 0 }, 0, 1, DEC }, |
|
|
|
|
|
|
|
{ "pattern_type", "set pattern type", OFFSET(pattern_type), AV_OPT_TYPE_INT, {.i64=PT_GLOB_SEQUENCE}, 0, INT_MAX, DEC, "pattern_type"}, |
|
|
|
{ "glob_sequence","glob/sequence pattern type", 0, AV_OPT_TYPE_CONST, {.i64=PT_GLOB_SEQUENCE}, INT_MIN, INT_MAX, DEC, "pattern_type" }, |
|
|
|
{ "glob", "glob pattern type", 0, AV_OPT_TYPE_CONST, {.i64=PT_GLOB}, INT_MIN, INT_MAX, DEC, "pattern_type" }, |
|
|
|
{ "sequence", "glob pattern type", 0, AV_OPT_TYPE_CONST, {.i64=PT_SEQUENCE}, INT_MIN, INT_MAX, DEC, "pattern_type" }, |
|
|
|
{ "pattern_type", "set pattern type", OFFSET(pattern_type), AV_OPT_TYPE_INT, {.i64=PT_GLOB_SEQUENCE}, 0, INT_MAX, DEC, "pattern_type"}, |
|
|
|
{ "glob_sequence","glob/sequence pattern type", 0, AV_OPT_TYPE_CONST, {.i64=PT_GLOB_SEQUENCE}, INT_MIN, INT_MAX, DEC, "pattern_type" }, |
|
|
|
{ "glob", "glob pattern type", 0, AV_OPT_TYPE_CONST, {.i64=PT_GLOB }, INT_MIN, INT_MAX, DEC, "pattern_type" }, |
|
|
|
{ "sequence", "glob pattern type", 0, AV_OPT_TYPE_CONST, {.i64=PT_SEQUENCE }, INT_MIN, INT_MAX, DEC, "pattern_type" }, |
|
|
|
|
|
|
|
{ "pixel_format", "set video pixel format", OFFSET(pixel_format), AV_OPT_TYPE_STRING, {.str = NULL}, 0, 0, DEC }, |
|
|
|
{ "start_number", "set first number in the sequence", OFFSET(start_number), AV_OPT_TYPE_INT, {.i64 = 0}, 0, INT_MAX, DEC }, |
|
|
|
{ "pixel_format", "set video pixel format", OFFSET(pixel_format), AV_OPT_TYPE_STRING, {.str = NULL}, 0, 0, DEC }, |
|
|
|
{ "start_number", "set first number in the sequence", OFFSET(start_number), AV_OPT_TYPE_INT, {.i64 = 0 }, 0, INT_MAX, DEC }, |
|
|
|
{ "start_number_range", "set range for looking at the first sequence number", OFFSET(start_number_range), AV_OPT_TYPE_INT, {.i64 = 5}, 1, INT_MAX, DEC }, |
|
|
|
{ "video_size", "set video size", OFFSET(video_size), AV_OPT_TYPE_STRING, {.str = NULL}, 0, 0, DEC }, |
|
|
|
{ "frame_size", "force frame size in bytes", OFFSET(frame_size), AV_OPT_TYPE_INT, {.i64 = 0}, 0, INT_MAX, DEC }, |
|
|
|
{ "video_size", "set video size", OFFSET(video_size), AV_OPT_TYPE_STRING, {.str = NULL}, 0, 0, DEC }, |
|
|
|
{ "frame_size", "force frame size in bytes", OFFSET(frame_size), AV_OPT_TYPE_INT, {.i64 = 0 }, 0, INT_MAX, DEC }, |
|
|
|
{ NULL }, |
|
|
|
}; |
|
|
|
|
|
|
|
@@ -447,10 +456,10 @@ AVInputFormat ff_image2_demuxer = { |
|
|
|
.name = "image2", |
|
|
|
.long_name = NULL_IF_CONFIG_SMALL("image2 sequence"), |
|
|
|
.priv_data_size = sizeof(VideoDemuxData), |
|
|
|
.read_probe = read_probe, |
|
|
|
.read_header = read_header, |
|
|
|
.read_packet = read_packet, |
|
|
|
.read_close = read_close, |
|
|
|
.read_probe = img_read_probe, |
|
|
|
.read_header = img_read_header, |
|
|
|
.read_packet = img_read_packet, |
|
|
|
.read_close = img_read_close, |
|
|
|
.flags = AVFMT_NOFILE, |
|
|
|
.priv_class = &img2_class, |
|
|
|
}; |
|
|
|
@@ -466,8 +475,8 @@ AVInputFormat ff_image2pipe_demuxer = { |
|
|
|
.name = "image2pipe", |
|
|
|
.long_name = NULL_IF_CONFIG_SMALL("piped image2 sequence"), |
|
|
|
.priv_data_size = sizeof(VideoDemuxData), |
|
|
|
.read_header = read_header, |
|
|
|
.read_packet = read_packet, |
|
|
|
.read_header = img_read_header, |
|
|
|
.read_packet = img_read_packet, |
|
|
|
.priv_class = &img2pipe_class, |
|
|
|
}; |
|
|
|
#endif |