| Line | Branch | Exec | Source |
|---|---|---|---|
| 1 | /* | ||
| 2 | * This file is part of FFmpeg. | ||
| 3 | * | ||
| 4 | * FFmpeg is free software; you can redistribute it and/or | ||
| 5 | * modify it under the terms of the GNU Lesser General Public | ||
| 6 | * License as published by the Free Software Foundation; either | ||
| 7 | * version 2.1 of the License, or (at your option) any later version. | ||
| 8 | * | ||
| 9 | * FFmpeg is distributed in the hope that it will be useful, | ||
| 10 | * but WITHOUT ANY WARRANTY; without even the implied warranty of | ||
| 11 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU | ||
| 12 | * Lesser General Public License for more details. | ||
| 13 | * | ||
| 14 | * You should have received a copy of the GNU Lesser General Public | ||
| 15 | * License along with FFmpeg; if not, write to the Free Software | ||
| 16 | * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA | ||
| 17 | */ | ||
| 18 | #include "config.h" | ||
| 19 | |||
| 20 | #include <stdlib.h> | ||
| 21 | #include <stdio.h> | ||
| 22 | #include <inttypes.h> | ||
| 23 | #include "libavutil/mem.h" | ||
| 24 | #include "libavutil/audio_fifo.c" | ||
| 25 | |||
| 26 | #define MAX_CHANNELS 32 | ||
| 27 | |||
| 28 | |||
| 29 | typedef struct TestStruct { | ||
| 30 | const enum AVSampleFormat format; | ||
| 31 | const int nb_ch; | ||
| 32 | void const *data_planes[MAX_CHANNELS]; | ||
| 33 | const int nb_samples_pch; | ||
| 34 | } TestStruct; | ||
| 35 | |||
| 36 | static const uint8_t data_U8 [] = {0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11 }; | ||
| 37 | static const int16_t data_S16[] = {0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11 }; | ||
| 38 | static const float data_FLT[] = {0.0, 1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0}; | ||
| 39 | |||
| 40 | static const TestStruct test_struct[] = { | ||
| 41 | {.format = AV_SAMPLE_FMT_U8 , .nb_ch = 1, .data_planes = {data_U8 , }, .nb_samples_pch = 12}, | ||
| 42 | {.format = AV_SAMPLE_FMT_U8P , .nb_ch = 2, .data_planes = {data_U8 , data_U8 +6, }, .nb_samples_pch = 6 }, | ||
| 43 | {.format = AV_SAMPLE_FMT_S16 , .nb_ch = 1, .data_planes = {data_S16, }, .nb_samples_pch = 12}, | ||
| 44 | {.format = AV_SAMPLE_FMT_S16P , .nb_ch = 2, .data_planes = {data_S16, data_S16+6, }, .nb_samples_pch = 6 }, | ||
| 45 | {.format = AV_SAMPLE_FMT_FLT , .nb_ch = 1, .data_planes = {data_FLT, }, .nb_samples_pch = 12}, | ||
| 46 | {.format = AV_SAMPLE_FMT_FLTP , .nb_ch = 2, .data_planes = {data_FLT, data_FLT+6, }, .nb_samples_pch = 6 } | ||
| 47 | }; | ||
| 48 | |||
| 49 | 6 | static void free_data_planes(AVAudioFifo *afifo, void **output_data) | |
| 50 | { | ||
| 51 | int i; | ||
| 52 |
2/2✓ Branch 0 taken 9 times.
✓ Branch 1 taken 6 times.
|
15 | for (i = 0; i < afifo->nb_buffers; ++i){ |
| 53 | 9 | av_freep(&output_data[i]); | |
| 54 | } | ||
| 55 | 6 | av_freep(&output_data); | |
| 56 | 6 | } | |
| 57 | |||
| 58 | ✗ | static void ERROR(const char *str) | |
| 59 | { | ||
| 60 | ✗ | fprintf(stderr, "%s\n", str); | |
| 61 | ✗ | exit(1); | |
| 62 | } | ||
| 63 | |||
| 64 | 66 | static void print_audio_bytes(const TestStruct *test_sample, void **data_planes, int nb_samples) | |
| 65 | { | ||
| 66 | int p, b, f; | ||
| 67 | 66 | int byte_offset = av_get_bytes_per_sample(test_sample->format); | |
| 68 | 66 | int buffers = av_sample_fmt_is_planar(test_sample->format) | |
| 69 |
2/2✓ Branch 0 taken 24 times.
✓ Branch 1 taken 42 times.
|
66 | ? test_sample->nb_ch : 1; |
| 70 | 66 | int line_size = (buffers > 1) ? nb_samples * byte_offset | |
| 71 |
2/2✓ Branch 0 taken 24 times.
✓ Branch 1 taken 42 times.
|
66 | : nb_samples * byte_offset * test_sample->nb_ch; |
| 72 |
2/2✓ Branch 0 taken 90 times.
✓ Branch 1 taken 66 times.
|
156 | for (p = 0; p < buffers; ++p){ |
| 73 |
2/2✓ Branch 0 taken 216 times.
✓ Branch 1 taken 90 times.
|
306 | for(b = 0; b < line_size; b+=byte_offset){ |
| 74 |
2/2✓ Branch 0 taken 504 times.
✓ Branch 1 taken 216 times.
|
720 | for (f = 0; f < byte_offset; f++){ |
| 75 | 504 | int order = !HAVE_BIGENDIAN ? (byte_offset - f - 1) : f; | |
| 76 | 504 | printf("%02x", *((uint8_t*)data_planes[p] + b + order)); | |
| 77 | } | ||
| 78 |
2/2✓ Branch 0 taken 126 times.
✓ Branch 1 taken 90 times.
|
216 | if (b + byte_offset < line_size) |
| 79 | 126 | putchar(' '); | |
| 80 | } | ||
| 81 | 90 | putchar('\n'); | |
| 82 | } | ||
| 83 | 66 | } | |
| 84 | |||
| 85 | 6 | static int read_samples_from_audio_fifo(AVAudioFifo* afifo, void ***output, int nb_samples) | |
| 86 | { | ||
| 87 | int i; | ||
| 88 | 6 | int samples = FFMIN(nb_samples, afifo->nb_samples); | |
| 89 | 6 | int tot_elements = !av_sample_fmt_is_planar(afifo->sample_fmt) | |
| 90 |
2/2✓ Branch 0 taken 3 times.
✓ Branch 1 taken 3 times.
|
6 | ? samples : afifo->channels * samples; |
| 91 | 6 | void **data_planes = av_malloc_array(afifo->nb_buffers, sizeof(void*)); | |
| 92 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 6 times.
|
6 | if (!data_planes) |
| 93 | ✗ | ERROR("failed to allocate memory!"); | |
| 94 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 6 times.
|
6 | if (*output) |
| 95 | ✗ | free_data_planes(afifo, *output); | |
| 96 | 6 | *output = data_planes; | |
| 97 | |||
| 98 |
2/2✓ Branch 0 taken 9 times.
✓ Branch 1 taken 6 times.
|
15 | for (i = 0; i < afifo->nb_buffers; ++i){ |
| 99 | 9 | data_planes[i] = av_malloc_array(tot_elements, afifo->sample_size); | |
| 100 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 9 times.
|
9 | if (!data_planes[i]) |
| 101 | ✗ | ERROR("failed to allocate memory!"); | |
| 102 | } | ||
| 103 | |||
| 104 | 6 | return av_audio_fifo_read(afifo, *output, nb_samples); | |
| 105 | } | ||
| 106 | |||
| 107 | 12 | static int write_samples_to_audio_fifo(AVAudioFifo* afifo, const TestStruct *test_sample, | |
| 108 | int nb_samples, int offset) | ||
| 109 | { | ||
| 110 | int offset_size, i; | ||
| 111 | void *data_planes[MAX_CHANNELS]; | ||
| 112 | |||
| 113 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 12 times.
|
12 | if(nb_samples > test_sample->nb_samples_pch - offset){ |
| 114 | ✗ | return 0; | |
| 115 | } | ||
| 116 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 12 times.
|
12 | if(offset >= test_sample->nb_samples_pch){ |
| 117 | ✗ | return 0; | |
| 118 | } | ||
| 119 | 12 | offset_size = offset * afifo->sample_size; | |
| 120 | |||
| 121 |
2/2✓ Branch 0 taken 18 times.
✓ Branch 1 taken 12 times.
|
30 | for (i = 0; i < afifo->nb_buffers ; ++i){ |
| 122 | 18 | data_planes[i] = (uint8_t*)test_sample->data_planes[i] + offset_size; | |
| 123 | } | ||
| 124 | |||
| 125 | 12 | return av_audio_fifo_write(afifo, data_planes, nb_samples); | |
| 126 | } | ||
| 127 | |||
| 128 | 6 | static void test_function(const TestStruct *test_sample) | |
| 129 | { | ||
| 130 | int ret, i; | ||
| 131 | 6 | void **output_data = NULL; | |
| 132 | 6 | AVAudioFifo *afifo = av_audio_fifo_alloc(test_sample->format, test_sample->nb_ch, | |
| 133 | 6 | test_sample->nb_samples_pch); | |
| 134 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 6 times.
|
6 | if (!afifo) { |
| 135 | ✗ | ERROR("ERROR: av_audio_fifo_alloc returned NULL!"); | |
| 136 | } | ||
| 137 | 6 | ret = write_samples_to_audio_fifo(afifo, test_sample, test_sample->nb_samples_pch, 0); | |
| 138 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 6 times.
|
6 | if (ret < 0){ |
| 139 | ✗ | ERROR("ERROR: av_audio_fifo_write failed!"); | |
| 140 | } | ||
| 141 | 6 | printf("written: %d\n", ret); | |
| 142 | |||
| 143 | 6 | ret = write_samples_to_audio_fifo(afifo, test_sample, test_sample->nb_samples_pch, 0); | |
| 144 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 6 times.
|
6 | if (ret < 0){ |
| 145 | ✗ | ERROR("ERROR: av_audio_fifo_write failed!"); | |
| 146 | } | ||
| 147 | 6 | printf("written: %d\n", ret); | |
| 148 | 6 | printf("remaining samples in audio_fifo: %d\n\n", av_audio_fifo_size(afifo)); | |
| 149 | |||
| 150 | 6 | ret = read_samples_from_audio_fifo(afifo, &output_data, test_sample->nb_samples_pch); | |
| 151 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 6 times.
|
6 | if (ret < 0){ |
| 152 | ✗ | ERROR("ERROR: av_audio_fifo_read failed!"); | |
| 153 | } | ||
| 154 | 6 | printf("read: %d\n", ret); | |
| 155 | 6 | print_audio_bytes(test_sample, output_data, ret); | |
| 156 | 6 | printf("remaining samples in audio_fifo: %d\n\n", av_audio_fifo_size(afifo)); | |
| 157 | |||
| 158 | /* test av_audio_fifo_peek */ | ||
| 159 | 6 | ret = av_audio_fifo_peek(afifo, output_data, afifo->nb_samples); | |
| 160 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 6 times.
|
6 | if (ret < 0){ |
| 161 | ✗ | ERROR("ERROR: av_audio_fifo_peek failed!"); | |
| 162 | } | ||
| 163 | 6 | printf("peek:\n"); | |
| 164 | 6 | print_audio_bytes(test_sample, output_data, ret); | |
| 165 | 6 | printf("\n"); | |
| 166 | |||
| 167 | /* test av_audio_fifo_peek_at */ | ||
| 168 | 6 | printf("peek_at:\n"); | |
| 169 |
2/2✓ Branch 0 taken 54 times.
✓ Branch 1 taken 6 times.
|
60 | for (i = 0; i < afifo->nb_samples; ++i){ |
| 170 | 54 | ret = av_audio_fifo_peek_at(afifo, output_data, 1, i); | |
| 171 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 54 times.
|
54 | if (ret < 0){ |
| 172 | ✗ | ERROR("ERROR: av_audio_fifo_peek_at failed!"); | |
| 173 | } | ||
| 174 | 54 | printf("%d:\n", i); | |
| 175 | 54 | print_audio_bytes(test_sample, output_data, ret); | |
| 176 | } | ||
| 177 | 6 | printf("\n"); | |
| 178 | |||
| 179 | /* test av_audio_fifo_drain */ | ||
| 180 | 6 | ret = av_audio_fifo_drain(afifo, afifo->nb_samples); | |
| 181 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 6 times.
|
6 | if (ret < 0){ |
| 182 | ✗ | ERROR("ERROR: av_audio_fifo_drain failed!"); | |
| 183 | } | ||
| 184 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 6 times.
|
6 | if (afifo->nb_samples){ |
| 185 | ✗ | ERROR("drain failed to flush all samples in audio_fifo!"); | |
| 186 | } | ||
| 187 | |||
| 188 | /* deallocate */ | ||
| 189 | 6 | free_data_planes(afifo, output_data); | |
| 190 | 6 | av_audio_fifo_free(afifo); | |
| 191 | 6 | } | |
| 192 | |||
| 193 | 1 | int main(void) | |
| 194 | { | ||
| 195 | 1 | int t, tests = sizeof(test_struct)/sizeof(test_struct[0]); | |
| 196 | |||
| 197 |
2/2✓ Branch 0 taken 6 times.
✓ Branch 1 taken 1 times.
|
7 | for (t = 0; t < tests; ++t){ |
| 198 | 6 | printf("\nTEST: %d\n\n", t+1); | |
| 199 | 6 | test_function(&test_struct[t]); | |
| 200 | } | ||
| 201 | 1 | return 0; | |
| 202 | } | ||
| 203 |