Browse code

tools: add uncoded_frame test program.

Nicolas George authored on 2014/02/03 17:12:27
Showing 2 changed files
... ...
@@ -31,7 +31,7 @@ OBJS-ffmpeg                   += ffmpeg_opt.o ffmpeg_filter.o
31 31
 OBJS-ffmpeg-$(HAVE_VDPAU_X11) += ffmpeg_vdpau.o
32 32
 TESTTOOLS   = audiogen videogen rotozoom tiny_psnr tiny_ssim base64
33 33
 HOSTPROGS  := $(TESTTOOLS:%=tests/%) doc/print_options
34
-TOOLS       = qt-faststart trasher
34
+TOOLS       = qt-faststart trasher uncoded_frame
35 35
 TOOLS-$(CONFIG_ZLIB) += cws2fws
36 36
 
37 37
 FFLIBS-$(CONFIG_AVDEVICE) += avdevice
... ...
@@ -61,6 +61,8 @@ $(TOOLS): %$(EXESUF): %.o $(EXEOBJS)
61 61
 	$(LD) $(LDFLAGS) $(LD_O) $^ $(ELIBS)
62 62
 
63 63
 tools/cws2fws$(EXESUF): ELIBS = $(ZLIB)
64
+tools/uncoded_frame$(EXESUF): $(FF_DEP_LIBS)
65
+tools/uncoded_frame$(EXESUF): ELIBS = $(FF_EXTRALIBS)
64 66
 
65 67
 config.h: .config
66 68
 .config: $(wildcard $(FFLIBS:%=$(SRC_PATH)/lib%/all*.c))
67 69
new file mode 100644
... ...
@@ -0,0 +1,279 @@
0
+#include <stdio.h>
1
+#include <stdlib.h>
2
+#include <string.h>
3
+#include "libavutil/avassert.h"
4
+#include "libavdevice/avdevice.h"
5
+#include "libavfilter/avfilter.h"
6
+#include "libavfilter/buffersink.h"
7
+#include "libavformat/avformat.h"
8
+
9
+typedef struct {
10
+    AVFormatContext *mux;
11
+    AVStream *stream;
12
+    AVFilterContext *sink;
13
+    AVFilterLink *link;
14
+} Stream;
15
+
16
+static int create_sink(Stream *st, AVFilterGraph *graph,
17
+                       AVFilterContext *f, int idx)
18
+{
19
+    enum AVMediaType type = avfilter_pad_get_type(f->output_pads, idx);
20
+    const char *sink_name;
21
+    int ret;
22
+
23
+    switch (type) {
24
+    case AVMEDIA_TYPE_VIDEO: sink_name =  "buffersink"; break;
25
+    case AVMEDIA_TYPE_AUDIO: sink_name = "abuffersink"; break;
26
+    default:
27
+        av_log(NULL, AV_LOG_ERROR, "Stream type not supported\n");
28
+        return AVERROR(EINVAL);
29
+    }
30
+    ret = avfilter_graph_create_filter(&st->sink,
31
+                                       avfilter_get_by_name(sink_name),
32
+                                       NULL, NULL, NULL, graph);
33
+    if (ret < 0)
34
+        return ret;
35
+    ret = avfilter_link(f, idx, st->sink, 0);
36
+    if (ret < 0)
37
+        return ret;
38
+    st->link = st->sink->inputs[0];
39
+    return 0;
40
+}
41
+
42
+int main(int argc, char **argv)
43
+{
44
+    char *in_graph_desc, **out_dev_name;
45
+    int nb_out_dev = 0, nb_streams = 0;
46
+    AVFilterGraph *in_graph = NULL;
47
+    Stream *streams = NULL, *st;
48
+    AVFrame *frame = NULL;
49
+    int i, j, run = 1, ret;
50
+
51
+    //av_log_set_level(AV_LOG_DEBUG);
52
+
53
+    if (argc < 3) {
54
+        av_log(NULL, AV_LOG_ERROR,
55
+               "Usage: %s filter_graph dev:out [dev2:out2...]\n\n"
56
+               "Examples:\n"
57
+               "%s movie=file.nut:s=v+a xv:- alsa:default\n"
58
+               "%s movie=file.nut:s=v+a uncodedframecrc:pipe:0\n",
59
+               argv[0], argv[0], argv[0]);
60
+        exit(1);
61
+    }
62
+    in_graph_desc = argv[1];
63
+    out_dev_name = argv + 2;
64
+    nb_out_dev = argc - 2;
65
+
66
+    av_register_all();
67
+    avdevice_register_all();
68
+    avfilter_register_all();
69
+
70
+    /* Create input graph */
71
+    if (!(in_graph = avfilter_graph_alloc())) {
72
+        ret = AVERROR(ENOMEM);
73
+        av_log(NULL, AV_LOG_ERROR, "Unable to alloc graph graph: %s\n",
74
+               av_err2str(ret));
75
+        goto fail;
76
+    }
77
+    ret = avfilter_graph_parse_ptr(in_graph, in_graph_desc, NULL, NULL, NULL);
78
+    if (ret < 0) {
79
+        av_log(NULL, AV_LOG_ERROR, "Unable to parse graph: %s\n",
80
+               av_err2str(ret));
81
+        goto fail;
82
+    }
83
+    nb_streams = 0;
84
+    for (i = 0; i < in_graph->nb_filters; i++) {
85
+        AVFilterContext *f = in_graph->filters[i];
86
+        for (j = 0; j < f->nb_inputs; j++) {
87
+            if (!f->inputs[j]) {
88
+                av_log(NULL, AV_LOG_ERROR, "Graph has unconnected inputs\n");
89
+                ret = AVERROR(EINVAL);
90
+                goto fail;
91
+            }
92
+        }
93
+        for (j = 0; j < f->nb_outputs; j++)
94
+            if (!f->outputs[j])
95
+                nb_streams++;
96
+    }
97
+    if (!nb_streams) {
98
+        av_log(NULL, AV_LOG_ERROR, "Graph has no output stream\n");
99
+        ret = AVERROR(EINVAL);
100
+        goto fail;
101
+    }
102
+    if (nb_out_dev != 1 && nb_out_dev != nb_streams) {
103
+        av_log(NULL, AV_LOG_ERROR,
104
+               "Graph has %d output streams, %d devices given\n",
105
+               nb_streams, nb_out_dev);
106
+        ret = AVERROR(EINVAL);
107
+        goto fail;
108
+    }
109
+
110
+    if (!(streams = av_calloc(nb_streams, sizeof(*streams)))) {
111
+        ret = AVERROR(ENOMEM);
112
+        av_log(NULL, AV_LOG_ERROR, "Could not allocate streams\n");
113
+    }
114
+    st = streams;
115
+    for (i = 0; i < in_graph->nb_filters; i++) {
116
+        AVFilterContext *f = in_graph->filters[i];
117
+        for (j = 0; j < f->nb_outputs; j++) {
118
+            if (!f->outputs[j]) {
119
+                if ((ret = create_sink(st++, in_graph, f, j)) < 0)
120
+                    goto fail;
121
+            }
122
+        }
123
+    }
124
+    av_assert0(st - streams == nb_streams);
125
+    if ((ret = avfilter_graph_config(in_graph, NULL)) < 0) {
126
+        av_log(NULL, AV_LOG_ERROR, "Failed to configure graph\n");
127
+        goto fail;
128
+    }
129
+
130
+    /* Create output devices */
131
+    for (i = 0; i < nb_out_dev; i++) {
132
+        char *fmt = NULL, *dev = out_dev_name[i];
133
+        st = &streams[i];
134
+        if ((dev = strchr(dev, ':'))) {
135
+            *(dev++) = 0;
136
+            fmt = out_dev_name[i];
137
+        }
138
+        ret = avformat_alloc_output_context2(&st->mux, NULL, fmt, dev);
139
+        if (ret < 0) {
140
+            av_log(NULL, AV_LOG_ERROR, "Failed to allocate output: %s\n",
141
+                   av_err2str(ret));
142
+            goto fail;
143
+        }
144
+        if (!(st->mux->oformat->flags & AVFMT_NOFILE)) {
145
+            ret = avio_open2(&st->mux->pb, st->mux->filename, AVIO_FLAG_WRITE,
146
+                             NULL, NULL);
147
+            if (ret < 0) {
148
+                av_log(st->mux, AV_LOG_ERROR, "Failed to init output: %s\n",
149
+                       av_err2str(ret));
150
+                goto fail;
151
+            }
152
+        }
153
+    }
154
+    for (; i < nb_streams; i++)
155
+        streams[i].mux = streams[0].mux;
156
+
157
+    /* Create output device streams */
158
+    for (i = 0; i < nb_streams; i++) {
159
+        st = &streams[i];
160
+        if (!(st->stream = avformat_new_stream(st->mux, NULL))) {
161
+            ret = AVERROR(ENOMEM);
162
+            av_log(NULL, AV_LOG_ERROR, "Failed to create output stream\n");
163
+            goto fail;
164
+        }
165
+        st->stream->codec->codec_type = st->link->type;
166
+        st->stream->time_base = st->stream->codec->time_base =
167
+            st->link->time_base;
168
+        switch (st->link->type) {
169
+        case AVMEDIA_TYPE_VIDEO:
170
+            st->stream->codec->codec_id = AV_CODEC_ID_RAWVIDEO;
171
+            st->stream->avg_frame_rate =
172
+            st->stream->  r_frame_rate = av_buffersink_get_frame_rate(st->sink);
173
+            st->stream->codec->width               = st->link->w;
174
+            st->stream->codec->height              = st->link->h;
175
+            st->stream->codec->sample_aspect_ratio = st->link->sample_aspect_ratio;
176
+            st->stream->codec->pix_fmt             = st->link->format;
177
+            break;
178
+        case AVMEDIA_TYPE_AUDIO:
179
+            st->stream->codec->codec_id =
180
+                av_get_pcm_codec(st->stream->codec->sample_fmt, -1);
181
+            st->stream->codec->channel_layout = st->link->channel_layout;
182
+            st->stream->codec->channels = avfilter_link_get_channels(st->link);
183
+            st->stream->codec->sample_rate = st->link->sample_rate;
184
+            st->stream->codec->sample_fmt = st->link->format; 
185
+            break;
186
+        default:
187
+            av_assert0(!"reached");
188
+        }
189
+    }
190
+
191
+    /* Init output devices */
192
+    for (i = 0; i < nb_out_dev; i++) {
193
+        st = &streams[i];
194
+        if ((ret = avformat_write_header(st->mux, NULL)) < 0) {
195
+            av_log(st->mux, AV_LOG_ERROR, "Failed to init output: %s\n",
196
+                   av_err2str(ret));
197
+            goto fail;
198
+        }
199
+    }
200
+
201
+    /* Check output devices */
202
+    for (i = 0; i < nb_streams; i++) {
203
+        st = &streams[i];
204
+        ret = av_write_uncoded_frame_query(st->mux, st->stream->index);
205
+        if (ret < 0) {
206
+            av_log(st->mux, AV_LOG_ERROR,
207
+                   "Uncoded frames not supported on stream #%d: %s\n",
208
+                   i, av_err2str(ret));
209
+            goto fail;
210
+        }
211
+    }
212
+
213
+    while (run) {
214
+        ret = avfilter_graph_request_oldest(in_graph);
215
+        if (ret < 0) {
216
+            if (ret == AVERROR_EOF) {
217
+                run = 0;
218
+            } else {
219
+                av_log(NULL, AV_LOG_ERROR, "Error filtering: %s\n",
220
+                       av_err2str(ret));
221
+                break;
222
+            }
223
+        }
224
+        for (i = 0; i < nb_streams; i++) {
225
+            st = &streams[i];
226
+            while (1) {
227
+                if (!frame && !(frame = av_frame_alloc())) {
228
+                    ret = AVERROR(ENOMEM);
229
+                    av_log(NULL, AV_LOG_ERROR, "Could not allocate frame\n");
230
+                    goto fail;
231
+                }
232
+                ret = av_buffersink_get_frame_flags(st->sink, frame,
233
+                                                    AV_BUFFERSINK_FLAG_NO_REQUEST);
234
+                if (ret < 0) {
235
+                    if (ret != AVERROR(EAGAIN) && ret != AVERROR_EOF)
236
+                        av_log(NULL, AV_LOG_WARNING, "Error in sink: %s\n",
237
+                               av_err2str(ret));
238
+                    break;
239
+                }
240
+                if (frame->pts != AV_NOPTS_VALUE)
241
+                    frame->pts = av_rescale_q(frame->pts,
242
+                                              st->link  ->time_base,
243
+                                              st->stream->time_base);
244
+                ret = av_interleaved_write_uncoded_frame(st->mux,
245
+                                                         st->stream->index,
246
+                                                         frame);
247
+                if (ret < 0) {
248
+                    av_log(st->stream->codec, AV_LOG_ERROR,
249
+                           "Error writing frame: %s\n", av_err2str(ret));
250
+                    goto fail;
251
+                }
252
+                frame = NULL;
253
+            }
254
+        }
255
+    }
256
+    ret = 0;
257
+
258
+    for (i = 0; i < nb_out_dev; i++) {
259
+        st = &streams[i];
260
+        av_write_trailer(st->mux);
261
+    }
262
+
263
+fail:
264
+    av_frame_free(&frame);
265
+    avfilter_graph_free(&in_graph);
266
+    if (streams) {
267
+        for (i = 0; i < nb_out_dev; i++) {
268
+            st = &streams[i];
269
+            if (st->mux) {
270
+                if (st->mux->pb)
271
+                    avio_close(st->mux->pb);
272
+                avformat_free_context(st->mux);
273
+            }
274
+        }
275
+    }
276
+    av_freep(&streams);
277
+    return ret < 0;
278
+}