Details | Last modification | View Log | RSS feed
Rev | Author | Line No. | Line |
---|---|---|---|
6147 | serge | 1 | /* |
2 | * Copyright (c) 2010 Stefano Sabatini |
||
3 | * Copyright (c) 2008 Victor Paesa |
||
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 | /** |
||
23 | * @file |
||
24 | * movie video source |
||
25 | * |
||
26 | * @todo use direct rendering (no allocation of a new frame) |
||
27 | * @todo support a PTS correction mechanism |
||
28 | */ |
||
29 | |||
30 | #include |
||
31 | #include |
||
32 | |||
33 | #include "libavutil/attributes.h" |
||
34 | #include "libavutil/avstring.h" |
||
35 | #include "libavutil/avassert.h" |
||
36 | #include "libavutil/opt.h" |
||
37 | #include "libavutil/imgutils.h" |
||
38 | #include "libavutil/internal.h" |
||
39 | #include "libavutil/timestamp.h" |
||
40 | #include "libavformat/avformat.h" |
||
41 | #include "audio.h" |
||
42 | #include "avcodec.h" |
||
43 | #include "avfilter.h" |
||
44 | #include "formats.h" |
||
45 | #include "internal.h" |
||
46 | #include "video.h" |
||
47 | |||
48 | typedef struct MovieStream { |
||
49 | AVStream *st; |
||
50 | int done; |
||
51 | } MovieStream; |
||
52 | |||
53 | typedef struct MovieContext { |
||
54 | /* common A/V fields */ |
||
55 | const AVClass *class; |
||
56 | int64_t seek_point; ///< seekpoint in microseconds |
||
57 | double seek_point_d; |
||
58 | char *format_name; |
||
59 | char *file_name; |
||
60 | char *stream_specs; /**< user-provided list of streams, separated by + */ |
||
61 | int stream_index; /**< for compatibility */ |
||
62 | int loop_count; |
||
63 | |||
64 | AVFormatContext *format_ctx; |
||
65 | int eof; |
||
66 | AVPacket pkt, pkt0; |
||
67 | |||
68 | int max_stream_index; /**< max stream # actually used for output */ |
||
69 | MovieStream *st; /**< array of all streams, one per output */ |
||
70 | int *out_index; /**< stream number -> output number map, or -1 */ |
||
71 | } MovieContext; |
||
72 | |||
73 | #define OFFSET(x) offsetof(MovieContext, x) |
||
74 | #define FLAGS AV_OPT_FLAG_FILTERING_PARAM | AV_OPT_FLAG_AUDIO_PARAM | AV_OPT_FLAG_VIDEO_PARAM |
||
75 | |||
76 | static const AVOption movie_options[]= { |
||
77 | { "filename", NULL, OFFSET(file_name), AV_OPT_TYPE_STRING, .flags = FLAGS }, |
||
78 | { "format_name", "set format name", OFFSET(format_name), AV_OPT_TYPE_STRING, .flags = FLAGS }, |
||
79 | { "f", "set format name", OFFSET(format_name), AV_OPT_TYPE_STRING, .flags = FLAGS }, |
||
80 | { "stream_index", "set stream index", OFFSET(stream_index), AV_OPT_TYPE_INT, { .i64 = -1 }, -1, INT_MAX, FLAGS }, |
||
81 | { "si", "set stream index", OFFSET(stream_index), AV_OPT_TYPE_INT, { .i64 = -1 }, -1, INT_MAX, FLAGS }, |
||
82 | { "seek_point", "set seekpoint (seconds)", OFFSET(seek_point_d), AV_OPT_TYPE_DOUBLE, { .dbl = 0 }, 0, (INT64_MAX-1) / 1000000, FLAGS }, |
||
83 | { "sp", "set seekpoint (seconds)", OFFSET(seek_point_d), AV_OPT_TYPE_DOUBLE, { .dbl = 0 }, 0, (INT64_MAX-1) / 1000000, FLAGS }, |
||
84 | { "streams", "set streams", OFFSET(stream_specs), AV_OPT_TYPE_STRING, {.str = 0}, CHAR_MAX, CHAR_MAX, FLAGS }, |
||
85 | { "s", "set streams", OFFSET(stream_specs), AV_OPT_TYPE_STRING, {.str = 0}, CHAR_MAX, CHAR_MAX, FLAGS }, |
||
86 | { "loop", "set loop count", OFFSET(loop_count), AV_OPT_TYPE_INT, {.i64 = 1}, 0, INT_MAX, FLAGS }, |
||
87 | { NULL }, |
||
88 | }; |
||
89 | |||
90 | static int movie_config_output_props(AVFilterLink *outlink); |
||
91 | static int movie_request_frame(AVFilterLink *outlink); |
||
92 | |||
93 | static AVStream *find_stream(void *log, AVFormatContext *avf, const char *spec) |
||
94 | { |
||
95 | int i, ret, already = 0, stream_id = -1; |
||
96 | char type_char[2], dummy; |
||
97 | AVStream *found = NULL; |
||
98 | enum AVMediaType type; |
||
99 | |||
100 | ret = sscanf(spec, "d%1[av]%d%c", type_char, &stream_id, &dummy); |
||
101 | if (ret >= 1 && ret <= 2) { |
||
102 | type = type_char[0] == 'v' ? AVMEDIA_TYPE_VIDEO : AVMEDIA_TYPE_AUDIO; |
||
103 | ret = av_find_best_stream(avf, type, stream_id, -1, NULL, 0); |
||
104 | if (ret < 0) { |
||
105 | av_log(log, AV_LOG_ERROR, "No %s stream with index '%d' found\n", |
||
106 | av_get_media_type_string(type), stream_id); |
||
107 | return NULL; |
||
108 | } |
||
109 | return avf->streams[ret]; |
||
110 | } |
||
111 | for (i = 0; i < avf->nb_streams; i++) { |
||
112 | ret = avformat_match_stream_specifier(avf, avf->streams[i], spec); |
||
113 | if (ret < 0) { |
||
114 | av_log(log, AV_LOG_ERROR, |
||
115 | "Invalid stream specifier \"%s\"\n", spec); |
||
116 | return NULL; |
||
117 | } |
||
118 | if (!ret) |
||
119 | continue; |
||
120 | if (avf->streams[i]->discard != AVDISCARD_ALL) { |
||
121 | already++; |
||
122 | continue; |
||
123 | } |
||
124 | if (found) { |
||
125 | av_log(log, AV_LOG_WARNING, |
||
126 | "Ambiguous stream specifier \"%s\", using #%d\n", spec, i); |
||
127 | break; |
||
128 | } |
||
129 | found = avf->streams[i]; |
||
130 | } |
||
131 | if (!found) { |
||
132 | av_log(log, AV_LOG_WARNING, "Stream specifier \"%s\" %s\n", spec, |
||
133 | already ? "matched only already used streams" : |
||
134 | "did not match any stream"); |
||
135 | return NULL; |
||
136 | } |
||
137 | if (found->codec->codec_type != AVMEDIA_TYPE_VIDEO && |
||
138 | found->codec->codec_type != AVMEDIA_TYPE_AUDIO) { |
||
139 | av_log(log, AV_LOG_ERROR, "Stream specifier \"%s\" matched a %s stream," |
||
140 | "currently unsupported by libavfilter\n", spec, |
||
141 | av_get_media_type_string(found->codec->codec_type)); |
||
142 | return NULL; |
||
143 | } |
||
144 | return found; |
||
145 | } |
||
146 | |||
147 | static int open_stream(void *log, MovieStream *st) |
||
148 | { |
||
149 | AVCodec *codec; |
||
150 | int ret; |
||
151 | |||
152 | codec = avcodec_find_decoder(st->st->codec->codec_id); |
||
153 | if (!codec) { |
||
154 | av_log(log, AV_LOG_ERROR, "Failed to find any codec\n"); |
||
155 | return AVERROR(EINVAL); |
||
156 | } |
||
157 | |||
158 | st->st->codec->refcounted_frames = 1; |
||
159 | |||
160 | if ((ret = avcodec_open2(st->st->codec, codec, NULL)) < 0) { |
||
161 | av_log(log, AV_LOG_ERROR, "Failed to open codec\n"); |
||
162 | return ret; |
||
163 | } |
||
164 | |||
165 | return 0; |
||
166 | } |
||
167 | |||
168 | static int guess_channel_layout(MovieStream *st, int st_index, void *log_ctx) |
||
169 | { |
||
170 | AVCodecContext *dec_ctx = st->st->codec; |
||
171 | char buf[256]; |
||
172 | int64_t chl = av_get_default_channel_layout(dec_ctx->channels); |
||
173 | |||
174 | if (!chl) { |
||
175 | av_log(log_ctx, AV_LOG_ERROR, |
||
176 | "Channel layout is not set in stream %d, and could not " |
||
177 | "be guessed from the number of channels (%d)\n", |
||
178 | st_index, dec_ctx->channels); |
||
179 | return AVERROR(EINVAL); |
||
180 | } |
||
181 | |||
182 | av_get_channel_layout_string(buf, sizeof(buf), dec_ctx->channels, chl); |
||
183 | av_log(log_ctx, AV_LOG_WARNING, |
||
184 | "Channel layout is not set in output stream %d, " |
||
185 | "guessed channel layout is '%s'\n", |
||
186 | st_index, buf); |
||
187 | dec_ctx->channel_layout = chl; |
||
188 | return 0; |
||
189 | } |
||
190 | |||
191 | static av_cold int movie_common_init(AVFilterContext *ctx) |
||
192 | { |
||
193 | MovieContext *movie = ctx->priv; |
||
194 | AVInputFormat *iformat = NULL; |
||
195 | int64_t timestamp; |
||
196 | int nb_streams = 1, ret, i; |
||
197 | char default_streams[16], *stream_specs, *spec, *cursor; |
||
198 | char name[16]; |
||
199 | AVStream *st; |
||
200 | |||
201 | if (!movie->file_name) { |
||
202 | av_log(ctx, AV_LOG_ERROR, "No filename provided!\n"); |
||
203 | return AVERROR(EINVAL); |
||
204 | } |
||
205 | |||
206 | movie->seek_point = movie->seek_point_d * 1000000 + 0.5; |
||
207 | |||
208 | stream_specs = movie->stream_specs; |
||
209 | if (!stream_specs) { |
||
210 | snprintf(default_streams, sizeof(default_streams), "d%c%d", |
||
211 | !strcmp(ctx->filter->name, "amovie") ? 'a' : 'v', |
||
212 | movie->stream_index); |
||
213 | stream_specs = default_streams; |
||
214 | } |
||
215 | for (cursor = stream_specs; *cursor; cursor++) |
||
216 | if (*cursor == '+') |
||
217 | nb_streams++; |
||
218 | |||
219 | if (movie->loop_count != 1 && nb_streams != 1) { |
||
220 | av_log(ctx, AV_LOG_ERROR, |
||
221 | "Loop with several streams is currently unsupported\n"); |
||
222 | return AVERROR_PATCHWELCOME; |
||
223 | } |
||
224 | |||
225 | av_register_all(); |
||
226 | |||
227 | // Try to find the movie format (container) |
||
228 | iformat = movie->format_name ? av_find_input_format(movie->format_name) : NULL; |
||
229 | |||
230 | movie->format_ctx = NULL; |
||
231 | if ((ret = avformat_open_input(&movie->format_ctx, movie->file_name, iformat, NULL)) < 0) { |
||
232 | av_log(ctx, AV_LOG_ERROR, |
||
233 | "Failed to avformat_open_input '%s'\n", movie->file_name); |
||
234 | return ret; |
||
235 | } |
||
236 | if ((ret = avformat_find_stream_info(movie->format_ctx, NULL)) < 0) |
||
237 | av_log(ctx, AV_LOG_WARNING, "Failed to find stream info\n"); |
||
238 | |||
239 | // if seeking requested, we execute it |
||
240 | if (movie->seek_point > 0) { |
||
241 | timestamp = movie->seek_point; |
||
242 | // add the stream start time, should it exist |
||
243 | if (movie->format_ctx->start_time != AV_NOPTS_VALUE) { |
||
244 | if (timestamp > INT64_MAX - movie->format_ctx->start_time) { |
||
245 | av_log(ctx, AV_LOG_ERROR, |
||
246 | "%s: seek value overflow with start_time:%"PRId64" seek_point:%"PRId64"\n", |
||
247 | movie->file_name, movie->format_ctx->start_time, movie->seek_point); |
||
248 | return AVERROR(EINVAL); |
||
249 | } |
||
250 | timestamp += movie->format_ctx->start_time; |
||
251 | } |
||
252 | if ((ret = av_seek_frame(movie->format_ctx, -1, timestamp, AVSEEK_FLAG_BACKWARD)) < 0) { |
||
253 | av_log(ctx, AV_LOG_ERROR, "%s: could not seek to position %"PRId64"\n", |
||
254 | movie->file_name, timestamp); |
||
255 | return ret; |
||
256 | } |
||
257 | } |
||
258 | |||
259 | for (i = 0; i < movie->format_ctx->nb_streams; i++) |
||
260 | movie->format_ctx->streams[i]->discard = AVDISCARD_ALL; |
||
261 | |||
262 | movie->st = av_calloc(nb_streams, sizeof(*movie->st)); |
||
263 | if (!movie->st) |
||
264 | return AVERROR(ENOMEM); |
||
265 | |||
266 | for (i = 0; i < nb_streams; i++) { |
||
267 | spec = av_strtok(stream_specs, "+", &cursor); |
||
268 | if (!spec) |
||
269 | return AVERROR_BUG; |
||
270 | stream_specs = NULL; /* for next strtok */ |
||
271 | st = find_stream(ctx, movie->format_ctx, spec); |
||
272 | if (!st) |
||
273 | return AVERROR(EINVAL); |
||
274 | st->discard = AVDISCARD_DEFAULT; |
||
275 | movie->st[i].st = st; |
||
276 | movie->max_stream_index = FFMAX(movie->max_stream_index, st->index); |
||
277 | } |
||
278 | if (av_strtok(NULL, "+", &cursor)) |
||
279 | return AVERROR_BUG; |
||
280 | |||
281 | movie->out_index = av_calloc(movie->max_stream_index + 1, |
||
282 | sizeof(*movie->out_index)); |
||
283 | if (!movie->out_index) |
||
284 | return AVERROR(ENOMEM); |
||
285 | for (i = 0; i <= movie->max_stream_index; i++) |
||
286 | movie->out_index[i] = -1; |
||
287 | for (i = 0; i < nb_streams; i++) { |
||
288 | AVFilterPad pad = { 0 }; |
||
289 | movie->out_index[movie->st[i].st->index] = i; |
||
290 | snprintf(name, sizeof(name), "out%d", i); |
||
291 | pad.type = movie->st[i].st->codec->codec_type; |
||
292 | pad.name = av_strdup(name); |
||
293 | if (!pad.name) |
||
294 | return AVERROR(ENOMEM); |
||
295 | pad.config_props = movie_config_output_props; |
||
296 | pad.request_frame = movie_request_frame; |
||
297 | ff_insert_outpad(ctx, i, &pad); |
||
298 | ret = open_stream(ctx, &movie->st[i]); |
||
299 | if (ret < 0) |
||
300 | return ret; |
||
301 | if ( movie->st[i].st->codec->codec->type == AVMEDIA_TYPE_AUDIO && |
||
302 | !movie->st[i].st->codec->channel_layout) { |
||
303 | ret = guess_channel_layout(&movie->st[i], i, ctx); |
||
304 | if (ret < 0) |
||
305 | return ret; |
||
306 | } |
||
307 | } |
||
308 | |||
309 | av_log(ctx, AV_LOG_VERBOSE, "seek_point:%"PRIi64" format_name:%s file_name:%s stream_index:%d\n", |
||
310 | movie->seek_point, movie->format_name, movie->file_name, |
||
311 | movie->stream_index); |
||
312 | |||
313 | return 0; |
||
314 | } |
||
315 | |||
316 | static av_cold void movie_uninit(AVFilterContext *ctx) |
||
317 | { |
||
318 | MovieContext *movie = ctx->priv; |
||
319 | int i; |
||
320 | |||
321 | for (i = 0; i < ctx->nb_outputs; i++) { |
||
322 | av_freep(&ctx->output_pads[i].name); |
||
323 | if (movie->st[i].st) |
||
324 | avcodec_close(movie->st[i].st->codec); |
||
325 | } |
||
326 | av_freep(&movie->st); |
||
327 | av_freep(&movie->out_index); |
||
328 | if (movie->format_ctx) |
||
329 | avformat_close_input(&movie->format_ctx); |
||
330 | } |
||
331 | |||
332 | static int movie_query_formats(AVFilterContext *ctx) |
||
333 | { |
||
334 | MovieContext *movie = ctx->priv; |
||
335 | int list[] = { 0, -1 }; |
||
336 | int64_t list64[] = { 0, -1 }; |
||
337 | int i; |
||
338 | |||
339 | for (i = 0; i < ctx->nb_outputs; i++) { |
||
340 | MovieStream *st = &movie->st[i]; |
||
341 | AVCodecContext *c = st->st->codec; |
||
342 | AVFilterLink *outlink = ctx->outputs[i]; |
||
343 | |||
344 | switch (c->codec_type) { |
||
345 | case AVMEDIA_TYPE_VIDEO: |
||
346 | list[0] = c->pix_fmt; |
||
347 | ff_formats_ref(ff_make_format_list(list), &outlink->in_formats); |
||
348 | break; |
||
349 | case AVMEDIA_TYPE_AUDIO: |
||
350 | list[0] = c->sample_fmt; |
||
351 | ff_formats_ref(ff_make_format_list(list), &outlink->in_formats); |
||
352 | list[0] = c->sample_rate; |
||
353 | ff_formats_ref(ff_make_format_list(list), &outlink->in_samplerates); |
||
354 | list64[0] = c->channel_layout; |
||
355 | ff_channel_layouts_ref(avfilter_make_format64_list(list64), |
||
356 | &outlink->in_channel_layouts); |
||
357 | break; |
||
358 | } |
||
359 | } |
||
360 | |||
361 | return 0; |
||
362 | } |
||
363 | |||
364 | static int movie_config_output_props(AVFilterLink *outlink) |
||
365 | { |
||
366 | AVFilterContext *ctx = outlink->src; |
||
367 | MovieContext *movie = ctx->priv; |
||
368 | unsigned out_id = FF_OUTLINK_IDX(outlink); |
||
369 | MovieStream *st = &movie->st[out_id]; |
||
370 | AVCodecContext *c = st->st->codec; |
||
371 | |||
372 | outlink->time_base = st->st->time_base; |
||
373 | |||
374 | switch (c->codec_type) { |
||
375 | case AVMEDIA_TYPE_VIDEO: |
||
376 | outlink->w = c->width; |
||
377 | outlink->h = c->height; |
||
378 | outlink->frame_rate = st->st->r_frame_rate; |
||
379 | break; |
||
380 | case AVMEDIA_TYPE_AUDIO: |
||
381 | break; |
||
382 | } |
||
383 | |||
384 | return 0; |
||
385 | } |
||
386 | |||
387 | static char *describe_frame_to_str(char *dst, size_t dst_size, |
||
388 | AVFrame *frame, enum AVMediaType frame_type, |
||
389 | AVFilterLink *link) |
||
390 | { |
||
391 | switch (frame_type) { |
||
392 | case AVMEDIA_TYPE_VIDEO: |
||
393 | snprintf(dst, dst_size, |
||
394 | "video pts:%s time:%s size:%dx%d aspect:%d/%d", |
||
395 | av_ts2str(frame->pts), av_ts2timestr(frame->pts, &link->time_base), |
||
396 | frame->width, frame->height, |
||
397 | frame->sample_aspect_ratio.num, |
||
398 | frame->sample_aspect_ratio.den); |
||
399 | break; |
||
400 | case AVMEDIA_TYPE_AUDIO: |
||
401 | snprintf(dst, dst_size, |
||
402 | "audio pts:%s time:%s samples:%d", |
||
403 | av_ts2str(frame->pts), av_ts2timestr(frame->pts, &link->time_base), |
||
404 | frame->nb_samples); |
||
405 | break; |
||
406 | default: |
||
407 | snprintf(dst, dst_size, "%s BUG", av_get_media_type_string(frame_type)); |
||
408 | break; |
||
409 | } |
||
410 | return dst; |
||
411 | } |
||
412 | |||
413 | static int rewind_file(AVFilterContext *ctx) |
||
414 | { |
||
415 | MovieContext *movie = ctx->priv; |
||
416 | int64_t timestamp = movie->seek_point; |
||
417 | int ret, i; |
||
418 | |||
419 | if (movie->format_ctx->start_time != AV_NOPTS_VALUE) |
||
420 | timestamp += movie->format_ctx->start_time; |
||
421 | ret = av_seek_frame(movie->format_ctx, -1, timestamp, AVSEEK_FLAG_BACKWARD); |
||
422 | if (ret < 0) { |
||
423 | av_log(ctx, AV_LOG_ERROR, "Unable to loop: %s\n", av_err2str(ret)); |
||
424 | movie->loop_count = 1; /* do not try again */ |
||
425 | return ret; |
||
426 | } |
||
427 | |||
428 | for (i = 0; i < ctx->nb_outputs; i++) { |
||
429 | avcodec_flush_buffers(movie->st[i].st->codec); |
||
430 | movie->st[i].done = 0; |
||
431 | } |
||
432 | movie->eof = 0; |
||
433 | return 0; |
||
434 | } |
||
435 | |||
436 | /** |
||
437 | * Try to push a frame to the requested output. |
||
438 | * |
||
439 | * @param ctx filter context |
||
440 | * @param out_id number of output where a frame is wanted; |
||
441 | * if the frame is read from file, used to set the return value; |
||
442 | * if the codec is being flushed, flush the corresponding stream |
||
443 | * @return 1 if a frame was pushed on the requested output, |
||
444 | * 0 if another attempt is possible, |
||
445 | * <0 AVERROR code |
||
446 | */ |
||
447 | static int movie_push_frame(AVFilterContext *ctx, unsigned out_id) |
||
448 | { |
||
449 | MovieContext *movie = ctx->priv; |
||
450 | AVPacket *pkt = &movie->pkt; |
||
451 | enum AVMediaType frame_type; |
||
452 | MovieStream *st; |
||
453 | int ret, got_frame = 0, pkt_out_id; |
||
454 | AVFilterLink *outlink; |
||
455 | AVFrame *frame; |
||
456 | |||
457 | if (!pkt->size) { |
||
458 | if (movie->eof) { |
||
459 | if (movie->st[out_id].done) { |
||
460 | if (movie->loop_count != 1) { |
||
461 | ret = rewind_file(ctx); |
||
462 | if (ret < 0) |
||
463 | return ret; |
||
464 | movie->loop_count -= movie->loop_count > 1; |
||
465 | av_log(ctx, AV_LOG_VERBOSE, "Stream finished, looping.\n"); |
||
466 | return 0; /* retry */ |
||
467 | } |
||
468 | return AVERROR_EOF; |
||
469 | } |
||
470 | pkt->stream_index = movie->st[out_id].st->index; |
||
471 | /* packet is already ready for flushing */ |
||
472 | } else { |
||
473 | ret = av_read_frame(movie->format_ctx, &movie->pkt0); |
||
474 | if (ret < 0) { |
||
475 | av_init_packet(&movie->pkt0); /* ready for flushing */ |
||
476 | *pkt = movie->pkt0; |
||
477 | if (ret == AVERROR_EOF) { |
||
478 | movie->eof = 1; |
||
479 | return 0; /* start flushing */ |
||
480 | } |
||
481 | return ret; |
||
482 | } |
||
483 | *pkt = movie->pkt0; |
||
484 | } |
||
485 | } |
||
486 | |||
487 | pkt_out_id = pkt->stream_index > movie->max_stream_index ? -1 : |
||
488 | movie->out_index[pkt->stream_index]; |
||
489 | if (pkt_out_id < 0) { |
||
490 | av_free_packet(&movie->pkt0); |
||
491 | pkt->size = 0; /* ready for next run */ |
||
492 | pkt->data = NULL; |
||
493 | return 0; |
||
494 | } |
||
495 | st = &movie->st[pkt_out_id]; |
||
496 | outlink = ctx->outputs[pkt_out_id]; |
||
497 | |||
498 | frame = av_frame_alloc(); |
||
499 | if (!frame) |
||
500 | return AVERROR(ENOMEM); |
||
501 | |||
502 | frame_type = st->st->codec->codec_type; |
||
503 | switch (frame_type) { |
||
504 | case AVMEDIA_TYPE_VIDEO: |
||
505 | ret = avcodec_decode_video2(st->st->codec, frame, &got_frame, pkt); |
||
506 | break; |
||
507 | case AVMEDIA_TYPE_AUDIO: |
||
508 | ret = avcodec_decode_audio4(st->st->codec, frame, &got_frame, pkt); |
||
509 | break; |
||
510 | default: |
||
511 | ret = AVERROR(ENOSYS); |
||
512 | break; |
||
513 | } |
||
514 | if (ret < 0) { |
||
515 | av_log(ctx, AV_LOG_WARNING, "Decode error: %s\n", av_err2str(ret)); |
||
516 | av_frame_free(&frame); |
||
517 | av_free_packet(&movie->pkt0); |
||
518 | movie->pkt.size = 0; |
||
519 | movie->pkt.data = NULL; |
||
520 | return 0; |
||
521 | } |
||
522 | if (!ret || st->st->codec->codec_type == AVMEDIA_TYPE_VIDEO) |
||
523 | ret = pkt->size; |
||
524 | |||
525 | pkt->data += ret; |
||
526 | pkt->size -= ret; |
||
527 | if (pkt->size <= 0) { |
||
528 | av_free_packet(&movie->pkt0); |
||
529 | pkt->size = 0; /* ready for next run */ |
||
530 | pkt->data = NULL; |
||
531 | } |
||
532 | if (!got_frame) { |
||
533 | if (!ret) |
||
534 | st->done = 1; |
||
535 | av_frame_free(&frame); |
||
536 | return 0; |
||
537 | } |
||
538 | |||
539 | frame->pts = av_frame_get_best_effort_timestamp(frame); |
||
540 | ff_dlog(ctx, "movie_push_frame(): file:'%s' %s\n", movie->file_name, |
||
541 | describe_frame_to_str((char[1024]){0}, 1024, frame, frame_type, outlink)); |
||
542 | |||
543 | if (st->st->codec->codec_type == AVMEDIA_TYPE_VIDEO) { |
||
544 | if (frame->format != outlink->format) { |
||
545 | av_log(ctx, AV_LOG_ERROR, "Format changed %s -> %s, discarding frame\n", |
||
546 | av_get_pix_fmt_name(outlink->format), |
||
547 | av_get_pix_fmt_name(frame->format) |
||
548 | ); |
||
549 | av_frame_free(&frame); |
||
550 | return 0; |
||
551 | } |
||
552 | } |
||
553 | ret = ff_filter_frame(outlink, frame); |
||
554 | |||
555 | if (ret < 0) |
||
556 | return ret; |
||
557 | return pkt_out_id == out_id; |
||
558 | } |
||
559 | |||
560 | static int movie_request_frame(AVFilterLink *outlink) |
||
561 | { |
||
562 | AVFilterContext *ctx = outlink->src; |
||
563 | unsigned out_id = FF_OUTLINK_IDX(outlink); |
||
564 | int ret; |
||
565 | |||
566 | while (1) { |
||
567 | ret = movie_push_frame(ctx, out_id); |
||
568 | if (ret) |
||
569 | return FFMIN(ret, 0); |
||
570 | } |
||
571 | } |
||
572 | |||
573 | #if CONFIG_MOVIE_FILTER |
||
574 | |||
575 | AVFILTER_DEFINE_CLASS(movie); |
||
576 | |||
577 | AVFilter ff_avsrc_movie = { |
||
578 | .name = "movie", |
||
579 | .description = NULL_IF_CONFIG_SMALL("Read from a movie source."), |
||
580 | .priv_size = sizeof(MovieContext), |
||
581 | .priv_class = &movie_class, |
||
582 | .init = movie_common_init, |
||
583 | .uninit = movie_uninit, |
||
584 | .query_formats = movie_query_formats, |
||
585 | |||
586 | .inputs = NULL, |
||
587 | .outputs = NULL, |
||
588 | .flags = AVFILTER_FLAG_DYNAMIC_OUTPUTS, |
||
589 | }; |
||
590 | |||
591 | #endif /* CONFIG_MOVIE_FILTER */ |
||
592 | |||
593 | #if CONFIG_AMOVIE_FILTER |
||
594 | |||
595 | #define amovie_options movie_options |
||
596 | AVFILTER_DEFINE_CLASS(amovie); |
||
597 | |||
598 | AVFilter ff_avsrc_amovie = { |
||
599 | .name = "amovie", |
||
600 | .description = NULL_IF_CONFIG_SMALL("Read audio from a movie source."), |
||
601 | .priv_size = sizeof(MovieContext), |
||
602 | .init = movie_common_init, |
||
603 | .uninit = movie_uninit, |
||
604 | .query_formats = movie_query_formats, |
||
605 | |||
606 | .inputs = NULL, |
||
607 | .outputs = NULL, |
||
608 | .priv_class = &amovie_class, |
||
609 | .flags = AVFILTER_FLAG_DYNAMIC_OUTPUTS, |
||
610 | }; |
||
611 | |||
612 | #endif /* CONFIG_AMOVIE_FILTER */>=>>>>>0>>>>>>>>=>>>>>>>>>>=>>>>>>> |