Line | Branch | Exec | Source |
---|---|---|---|
1 | /* | ||
2 | * Audio Frame Queue | ||
3 | * Copyright (c) 2012 Justin Ruggles | ||
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 | #include "libavutil/attributes.h" | ||
23 | #include "libavutil/mem.h" | ||
24 | #include "audio_frame_queue.h" | ||
25 | #include "encode.h" | ||
26 | #include "libavutil/avassert.h" | ||
27 | |||
28 | 15 | av_cold void ff_af_queue_init(AVCodecContext *avctx, AudioFrameQueue *afq) | |
29 | { | ||
30 | 15 | afq->avctx = avctx; | |
31 | 15 | afq->remaining_delay = avctx->initial_padding; | |
32 | 15 | afq->remaining_samples = avctx->initial_padding; | |
33 | 15 | afq->frame_count = 0; | |
34 | 15 | } | |
35 | |||
36 | 15 | void ff_af_queue_close(AudioFrameQueue *afq) | |
37 | { | ||
38 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 15 times.
|
15 | if(afq->frame_count) |
39 | ✗ | av_log(afq->avctx, AV_LOG_WARNING, "%d frames left in the queue on closing\n", afq->frame_count); | |
40 | 15 | av_freep(&afq->frames); | |
41 | 15 | memset(afq, 0, sizeof(*afq)); | |
42 | 15 | } | |
43 | |||
44 | 11426 | int ff_af_queue_add(AudioFrameQueue *afq, const AVFrame *f) | |
45 | { | ||
46 | 11426 | AudioFrame *new = av_fast_realloc(afq->frames, &afq->frame_alloc, sizeof(*afq->frames)*(afq->frame_count+1)); | |
47 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 11426 times.
|
11426 | if(!new) |
48 | ✗ | return AVERROR(ENOMEM); | |
49 | 11426 | afq->frames = new; | |
50 | 11426 | new += afq->frame_count; | |
51 | |||
52 | /* get frame parameters */ | ||
53 | 11426 | new->duration = f->nb_samples; | |
54 | 11426 | new->duration += afq->remaining_delay; | |
55 |
1/2✓ Branch 0 taken 11426 times.
✗ Branch 1 not taken.
|
11426 | if (f->pts != AV_NOPTS_VALUE) { |
56 | 11426 | new->pts = av_rescale_q(f->pts, | |
57 | 11426 | afq->avctx->time_base, | |
58 | 11426 | (AVRational){ 1, afq->avctx->sample_rate }); | |
59 | 11426 | new->pts -= afq->remaining_delay; | |
60 |
3/4✓ Branch 0 taken 10593 times.
✓ Branch 1 taken 833 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 10593 times.
|
11426 | if(afq->frame_count && new[-1].pts >= new->pts) |
61 | ✗ | av_log(afq->avctx, AV_LOG_WARNING, "Queue input is backward in time\n"); | |
62 | } else { | ||
63 | ✗ | new->pts = AV_NOPTS_VALUE; | |
64 | } | ||
65 | 11426 | afq->remaining_delay = 0; | |
66 | |||
67 | /* add frame sample count */ | ||
68 | 11426 | afq->remaining_samples += f->nb_samples; | |
69 | |||
70 | 11426 | afq->frame_count++; | |
71 | |||
72 | 11426 | return 0; | |
73 | } | ||
74 | |||
75 | 5302 | void ff_af_queue_remove(AudioFrameQueue *afq, int nb_samples, int64_t *pts, | |
76 | int64_t *duration) | ||
77 | { | ||
78 | 5302 | int64_t out_pts = AV_NOPTS_VALUE; | |
79 | 5302 | int removed_samples = 0; | |
80 | int i; | ||
81 | |||
82 |
1/4✗ Branch 0 not taken.
✓ Branch 1 taken 5302 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
|
5302 | if (afq->frame_count || afq->frame_alloc) { |
83 |
1/2✓ Branch 0 taken 5302 times.
✗ Branch 1 not taken.
|
5302 | if (afq->frames->pts != AV_NOPTS_VALUE) |
84 | 5302 | out_pts = afq->frames->pts; | |
85 | } | ||
86 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 5302 times.
|
5302 | if(!afq->frame_count) |
87 | ✗ | av_log(afq->avctx, AV_LOG_WARNING, "Trying to remove %d samples, but the queue is empty\n", nb_samples); | |
88 |
1/2✓ Branch 0 taken 5302 times.
✗ Branch 1 not taken.
|
5302 | if (pts) |
89 | 5302 | *pts = ff_samples_to_time_base(afq->avctx, out_pts); | |
90 | |||
91 |
4/4✓ Branch 0 taken 11825 times.
✓ Branch 1 taken 5290 times.
✓ Branch 2 taken 11813 times.
✓ Branch 3 taken 12 times.
|
17115 | for(i=0; nb_samples && i<afq->frame_count; i++){ |
92 | 11813 | int n= FFMIN(afq->frames[i].duration, nb_samples); | |
93 | 11813 | afq->frames[i].duration -= n; | |
94 | 11813 | nb_samples -= n; | |
95 | 11813 | removed_samples += n; | |
96 |
1/2✓ Branch 0 taken 11813 times.
✗ Branch 1 not taken.
|
11813 | if(afq->frames[i].pts != AV_NOPTS_VALUE) |
97 | 11813 | afq->frames[i].pts += n; | |
98 | } | ||
99 | 5302 | afq->remaining_samples -= removed_samples; | |
100 |
3/4✓ Branch 0 taken 5302 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 387 times.
✓ Branch 3 taken 4915 times.
|
5302 | i -= i && afq->frames[i-1].duration; |
101 | 5302 | memmove(afq->frames, afq->frames + i, sizeof(*afq->frames) * (afq->frame_count - i)); | |
102 | 5302 | afq->frame_count -= i; | |
103 | |||
104 |
2/2✓ Branch 0 taken 12 times.
✓ Branch 1 taken 5290 times.
|
5302 | if(nb_samples){ |
105 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 12 times.
|
12 | av_assert0(!afq->frame_count); |
106 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 12 times.
|
12 | av_assert0(afq->remaining_samples == afq->remaining_delay); |
107 |
2/4✓ Branch 0 taken 12 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 12 times.
✗ Branch 3 not taken.
|
12 | if(afq->frames && afq->frames[0].pts != AV_NOPTS_VALUE) |
108 | 12 | afq->frames[0].pts += nb_samples; | |
109 | 12 | av_log(afq->avctx, AV_LOG_DEBUG, "Trying to remove %d more samples than there are in the queue\n", nb_samples); | |
110 | } | ||
111 |
1/2✓ Branch 0 taken 5302 times.
✗ Branch 1 not taken.
|
5302 | if (duration) |
112 | 5302 | *duration = ff_samples_to_time_base(afq->avctx, removed_samples); | |
113 | 5302 | } | |
114 |