GCC Code Coverage Report
Directory: ../../../ffmpeg/ Exec Total Coverage
File: src/libavfilter/fifo.c Lines: 0 123 0.0 %
Date: 2020-08-13 15:06:06 Branches: 0 74 0.0 %

Line Branch Exec Source
1
/*
2
 * Copyright (c) 2007 Bobby Bingham
3
 *
4
 * This file is part of FFmpeg.
5
 *
6
 * FFmpeg is free software; you can redistribute it and/or
7
 * modify it under the terms of the GNU Lesser General Public
8
 * License as published by the Free Software Foundation; either
9
 * version 2.1 of the License, or (at your option) any later version.
10
 *
11
 * FFmpeg is distributed in the hope that it will be useful,
12
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
14
 * Lesser General Public License for more details.
15
 *
16
 * You should have received a copy of the GNU Lesser General Public
17
 * License along with FFmpeg; if not, write to the Free Software
18
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
19
 */
20
21
/**
22
 * @file
23
 * FIFO buffering filter
24
 */
25
26
#include "libavutil/avassert.h"
27
#include "libavutil/channel_layout.h"
28
#include "libavutil/common.h"
29
#include "libavutil/mathematics.h"
30
#include "libavutil/samplefmt.h"
31
32
#include "audio.h"
33
#include "avfilter.h"
34
#include "internal.h"
35
#include "video.h"
36
37
typedef struct Buf {
38
    AVFrame *frame;
39
    struct Buf *next;
40
} Buf;
41
42
typedef struct FifoContext {
43
    Buf  root;
44
    Buf *last;   ///< last buffered frame
45
46
    /**
47
     * When a specific number of output samples is requested, the partial
48
     * buffer is stored here
49
     */
50
    AVFrame *out;
51
    int allocated_samples;      ///< number of samples out was allocated for
52
} FifoContext;
53
54
static av_cold int init(AVFilterContext *ctx)
55
{
56
    FifoContext *s = ctx->priv;
57
    s->last = &s->root;
58
59
    return 0;
60
}
61
62
static av_cold void uninit(AVFilterContext *ctx)
63
{
64
    FifoContext *s = ctx->priv;
65
    Buf *buf, *tmp;
66
67
    for (buf = s->root.next; buf; buf = tmp) {
68
        tmp = buf->next;
69
        av_frame_free(&buf->frame);
70
        av_free(buf);
71
    }
72
73
    av_frame_free(&s->out);
74
}
75
76
static int add_to_queue(AVFilterLink *inlink, AVFrame *frame)
77
{
78
    FifoContext *s = inlink->dst->priv;
79
80
    s->last->next = av_mallocz(sizeof(Buf));
81
    if (!s->last->next) {
82
        av_frame_free(&frame);
83
        return AVERROR(ENOMEM);
84
    }
85
86
    s->last = s->last->next;
87
    s->last->frame = frame;
88
89
    return 0;
90
}
91
92
static void queue_pop(FifoContext *s)
93
{
94
    Buf *tmp = s->root.next->next;
95
    if (s->last == s->root.next)
96
        s->last = &s->root;
97
    av_freep(&s->root.next);
98
    s->root.next = tmp;
99
}
100
101
/**
102
 * Move data pointers and pts offset samples forward.
103
 */
104
static void buffer_offset(AVFilterLink *link, AVFrame *frame,
105
                          int offset)
106
{
107
    int nb_channels = link->channels;
108
    int planar = av_sample_fmt_is_planar(link->format);
109
    int planes = planar ? nb_channels : 1;
110
    int block_align = av_get_bytes_per_sample(link->format) * (planar ? 1 : nb_channels);
111
    int i;
112
113
    av_assert0(frame->nb_samples > offset);
114
115
    for (i = 0; i < planes; i++)
116
        frame->extended_data[i] += block_align * offset;
117
    if (frame->data != frame->extended_data)
118
        memcpy(frame->data, frame->extended_data,
119
               FFMIN(planes, FF_ARRAY_ELEMS(frame->data)) * sizeof(*frame->data));
120
    frame->linesize[0] -= block_align*offset;
121
    frame->nb_samples -= offset;
122
123
    if (frame->pts != AV_NOPTS_VALUE) {
124
        frame->pts += av_rescale_q(offset, (AVRational){1, link->sample_rate},
125
                                   link->time_base);
126
    }
127
}
128
129
static int calc_ptr_alignment(AVFrame *frame)
130
{
131
    int planes = av_sample_fmt_is_planar(frame->format) ?
132
                 frame->channels : 1;
133
    int min_align = 128;
134
    int p;
135
136
    for (p = 0; p < planes; p++) {
137
        int cur_align = 128;
138
        while ((intptr_t)frame->extended_data[p] % cur_align)
139
            cur_align >>= 1;
140
        if (cur_align < min_align)
141
            min_align = cur_align;
142
    }
143
    return min_align;
144
}
145
146
static int return_audio_frame(AVFilterContext *ctx)
147
{
148
    AVFilterLink *link = ctx->outputs[0];
149
    FifoContext *s = ctx->priv;
150
    AVFrame *head = s->root.next ? s->root.next->frame : NULL;
151
    AVFrame *out;
152
    int ret;
153
154
    /* if head is NULL then we're flushing the remaining samples in out */
155
    if (!head && !s->out)
156
        return AVERROR_EOF;
157
158
    if (!s->out &&
159
        head->nb_samples >= link->request_samples &&
160
        calc_ptr_alignment(head) >= 32) {
161
        if (head->nb_samples == link->request_samples) {
162
            out = head;
163
            queue_pop(s);
164
        } else {
165
            out = av_frame_clone(head);
166
            if (!out)
167
                return AVERROR(ENOMEM);
168
169
            out->nb_samples = link->request_samples;
170
            buffer_offset(link, head, link->request_samples);
171
        }
172
    } else {
173
        int nb_channels = link->channels;
174
175
        if (!s->out) {
176
            s->out = ff_get_audio_buffer(link, link->request_samples);
177
            if (!s->out)
178
                return AVERROR(ENOMEM);
179
180
            s->out->nb_samples = 0;
181
            s->out->pts                   = head->pts;
182
            s->allocated_samples          = link->request_samples;
183
        } else if (link->request_samples != s->allocated_samples) {
184
            av_log(ctx, AV_LOG_ERROR, "request_samples changed before the "
185
                   "buffer was returned.\n");
186
            return AVERROR(EINVAL);
187
        }
188
189
        while (s->out->nb_samples < s->allocated_samples) {
190
            int len;
191
192
            if (!s->root.next) {
193
                ret = ff_request_frame(ctx->inputs[0]);
194
                if (ret == AVERROR_EOF) {
195
                    av_samples_set_silence(s->out->extended_data,
196
                                           s->out->nb_samples,
197
                                           s->allocated_samples -
198
                                           s->out->nb_samples,
199
                                           nb_channels, link->format);
200
                    s->out->nb_samples = s->allocated_samples;
201
                    break;
202
                } else if (ret < 0)
203
                    return ret;
204
                if (!s->root.next)
205
                    return 0;
206
            }
207
            head = s->root.next->frame;
208
209
            len = FFMIN(s->allocated_samples - s->out->nb_samples,
210
                        head->nb_samples);
211
212
            av_samples_copy(s->out->extended_data, head->extended_data,
213
                            s->out->nb_samples, 0, len, nb_channels,
214
                            link->format);
215
            s->out->nb_samples += len;
216
217
            if (len == head->nb_samples) {
218
                av_frame_free(&head);
219
                queue_pop(s);
220
            } else {
221
                buffer_offset(link, head, len);
222
            }
223
        }
224
        out = s->out;
225
        s->out = NULL;
226
    }
227
    return ff_filter_frame(link, out);
228
}
229
230
static int request_frame(AVFilterLink *outlink)
231
{
232
    FifoContext *s = outlink->src->priv;
233
    int ret = 0;
234
235
    if (!s->root.next) {
236
        if ((ret = ff_request_frame(outlink->src->inputs[0])) < 0) {
237
            if (ret == AVERROR_EOF && outlink->request_samples)
238
                return return_audio_frame(outlink->src);
239
            return ret;
240
        }
241
        if (!s->root.next)
242
            return 0;
243
    }
244
245
    if (outlink->request_samples) {
246
        return return_audio_frame(outlink->src);
247
    } else {
248
        ret = ff_filter_frame(outlink, s->root.next->frame);
249
        queue_pop(s);
250
    }
251
252
    return ret;
253
}
254
255
static const AVFilterPad avfilter_vf_fifo_inputs[] = {
256
    {
257
        .name         = "default",
258
        .type         = AVMEDIA_TYPE_VIDEO,
259
        .filter_frame = add_to_queue,
260
    },
261
    { NULL }
262
};
263
264
static const AVFilterPad avfilter_vf_fifo_outputs[] = {
265
    {
266
        .name          = "default",
267
        .type          = AVMEDIA_TYPE_VIDEO,
268
        .request_frame = request_frame,
269
    },
270
    { NULL }
271
};
272
273
AVFilter ff_vf_fifo = {
274
    .name        = "fifo",
275
    .description = NULL_IF_CONFIG_SMALL("Buffer input images and send them when they are requested."),
276
    .init        = init,
277
    .uninit      = uninit,
278
    .priv_size   = sizeof(FifoContext),
279
    .inputs      = avfilter_vf_fifo_inputs,
280
    .outputs     = avfilter_vf_fifo_outputs,
281
};
282
283
static const AVFilterPad avfilter_af_afifo_inputs[] = {
284
    {
285
        .name         = "default",
286
        .type         = AVMEDIA_TYPE_AUDIO,
287
        .filter_frame = add_to_queue,
288
    },
289
    { NULL }
290
};
291
292
static const AVFilterPad avfilter_af_afifo_outputs[] = {
293
    {
294
        .name          = "default",
295
        .type          = AVMEDIA_TYPE_AUDIO,
296
        .request_frame = request_frame,
297
    },
298
    { NULL }
299
};
300
301
AVFilter ff_af_afifo = {
302
    .name        = "afifo",
303
    .description = NULL_IF_CONFIG_SMALL("Buffer input frames and send them when they are requested."),
304
    .init        = init,
305
    .uninit      = uninit,
306
    .priv_size   = sizeof(FifoContext),
307
    .inputs      = avfilter_af_afifo_inputs,
308
    .outputs     = avfilter_af_afifo_outputs,
309
};