yading@11: /* yading@11: * Copyright (c) 2002 Fabrice Bellard yading@11: * Copyright (c) 2012 Justin Ruggles yading@11: * yading@11: * This file is part of Libav. yading@11: * yading@11: * Libav is free software; you can redistribute it and/or yading@11: * modify it under the terms of the GNU Lesser General Public yading@11: * License as published by the Free Software Foundation; either yading@11: * version 2.1 of the License, or (at your option) any later version. yading@11: * yading@11: * Libav is distributed in the hope that it will be useful, yading@11: * but WITHOUT ANY WARRANTY; without even the implied warranty of yading@11: * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU yading@11: * Lesser General Public License for more details. yading@11: * yading@11: * You should have received a copy of the GNU Lesser General Public yading@11: * License along with Libav; if not, write to the Free Software yading@11: * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA yading@11: */ yading@11: yading@11: #include yading@11: #include yading@11: yading@11: #include "libavutil/avstring.h" yading@11: #include "libavutil/common.h" yading@11: #include "libavutil/lfg.h" yading@11: #include "libavutil/libm.h" yading@11: #include "libavutil/log.h" yading@11: #include "libavutil/mem.h" yading@11: #include "libavutil/opt.h" yading@11: #include "libavutil/samplefmt.h" yading@11: #include "avresample.h" yading@11: yading@11: static double dbl_rand(AVLFG *lfg) yading@11: { yading@11: return 2.0 * (av_lfg_get(lfg) / (double)UINT_MAX) - 1.0; yading@11: } yading@11: yading@11: #define PUT_FUNC(name, fmt, type, expr) \ yading@11: static void put_sample_ ## name(void **data, enum AVSampleFormat sample_fmt,\ yading@11: int channels, int sample, int ch, \ yading@11: double v_dbl) \ yading@11: { \ yading@11: type v = expr; \ yading@11: type **out = (type **)data; \ yading@11: if (av_sample_fmt_is_planar(sample_fmt)) \ yading@11: out[ch][sample] = v; \ yading@11: else \ yading@11: out[0][sample * channels + ch] = v; \ yading@11: } yading@11: yading@11: PUT_FUNC(u8, AV_SAMPLE_FMT_U8, uint8_t, av_clip_uint8 ( lrint(v_dbl * (1 << 7)) + 128)) yading@11: PUT_FUNC(s16, AV_SAMPLE_FMT_S16, int16_t, av_clip_int16 ( lrint(v_dbl * (1 << 15)))) yading@11: PUT_FUNC(s32, AV_SAMPLE_FMT_S32, int32_t, av_clipl_int32(llrint(v_dbl * (1U << 31)))) yading@11: PUT_FUNC(flt, AV_SAMPLE_FMT_FLT, float, v_dbl) yading@11: PUT_FUNC(dbl, AV_SAMPLE_FMT_DBL, double, v_dbl) yading@11: yading@11: static void put_sample(void **data, enum AVSampleFormat sample_fmt, yading@11: int channels, int sample, int ch, double v_dbl) yading@11: { yading@11: switch (av_get_packed_sample_fmt(sample_fmt)) { yading@11: case AV_SAMPLE_FMT_U8: yading@11: put_sample_u8(data, sample_fmt, channels, sample, ch, v_dbl); yading@11: break; yading@11: case AV_SAMPLE_FMT_S16: yading@11: put_sample_s16(data, sample_fmt, channels, sample, ch, v_dbl); yading@11: break; yading@11: case AV_SAMPLE_FMT_S32: yading@11: put_sample_s32(data, sample_fmt, channels, sample, ch, v_dbl); yading@11: break; yading@11: case AV_SAMPLE_FMT_FLT: yading@11: put_sample_flt(data, sample_fmt, channels, sample, ch, v_dbl); yading@11: break; yading@11: case AV_SAMPLE_FMT_DBL: yading@11: put_sample_dbl(data, sample_fmt, channels, sample, ch, v_dbl); yading@11: break; yading@11: } yading@11: } yading@11: yading@11: static void audiogen(AVLFG *rnd, void **data, enum AVSampleFormat sample_fmt, yading@11: int channels, int sample_rate, int nb_samples) yading@11: { yading@11: int i, ch, k; yading@11: double v, f, a, ampa; yading@11: double tabf1[AVRESAMPLE_MAX_CHANNELS]; yading@11: double tabf2[AVRESAMPLE_MAX_CHANNELS]; yading@11: double taba[AVRESAMPLE_MAX_CHANNELS]; yading@11: yading@11: #define PUT_SAMPLE put_sample(data, sample_fmt, channels, k, ch, v); yading@11: yading@11: k = 0; yading@11: yading@11: /* 1 second of single freq sinus at 1000 Hz */ yading@11: a = 0; yading@11: for (i = 0; i < 1 * sample_rate && k < nb_samples; i++, k++) { yading@11: v = sin(a) * 0.30; yading@11: for (ch = 0; ch < channels; ch++) yading@11: PUT_SAMPLE yading@11: a += M_PI * 1000.0 * 2.0 / sample_rate; yading@11: } yading@11: yading@11: /* 1 second of varying frequency between 100 and 10000 Hz */ yading@11: a = 0; yading@11: for (i = 0; i < 1 * sample_rate && k < nb_samples; i++, k++) { yading@11: v = sin(a) * 0.30; yading@11: for (ch = 0; ch < channels; ch++) yading@11: PUT_SAMPLE yading@11: f = 100.0 + (((10000.0 - 100.0) * i) / sample_rate); yading@11: a += M_PI * f * 2.0 / sample_rate; yading@11: } yading@11: yading@11: /* 0.5 second of low amplitude white noise */ yading@11: for (i = 0; i < sample_rate / 2 && k < nb_samples; i++, k++) { yading@11: v = dbl_rand(rnd) * 0.30; yading@11: for (ch = 0; ch < channels; ch++) yading@11: PUT_SAMPLE yading@11: } yading@11: yading@11: /* 0.5 second of high amplitude white noise */ yading@11: for (i = 0; i < sample_rate / 2 && k < nb_samples; i++, k++) { yading@11: v = dbl_rand(rnd); yading@11: for (ch = 0; ch < channels; ch++) yading@11: PUT_SAMPLE yading@11: } yading@11: yading@11: /* 1 second of unrelated ramps for each channel */ yading@11: for (ch = 0; ch < channels; ch++) { yading@11: taba[ch] = 0; yading@11: tabf1[ch] = 100 + av_lfg_get(rnd) % 5000; yading@11: tabf2[ch] = 100 + av_lfg_get(rnd) % 5000; yading@11: } yading@11: for (i = 0; i < 1 * sample_rate && k < nb_samples; i++, k++) { yading@11: for (ch = 0; ch < channels; ch++) { yading@11: v = sin(taba[ch]) * 0.30; yading@11: PUT_SAMPLE yading@11: f = tabf1[ch] + (((tabf2[ch] - tabf1[ch]) * i) / sample_rate); yading@11: taba[ch] += M_PI * f * 2.0 / sample_rate; yading@11: } yading@11: } yading@11: yading@11: /* 2 seconds of 500 Hz with varying volume */ yading@11: a = 0; yading@11: ampa = 0; yading@11: for (i = 0; i < 2 * sample_rate && k < nb_samples; i++, k++) { yading@11: for (ch = 0; ch < channels; ch++) { yading@11: double amp = (1.0 + sin(ampa)) * 0.15; yading@11: if (ch & 1) yading@11: amp = 0.30 - amp; yading@11: v = sin(a) * amp; yading@11: PUT_SAMPLE yading@11: a += M_PI * 500.0 * 2.0 / sample_rate; yading@11: ampa += M_PI * 2.0 / sample_rate; yading@11: } yading@11: } yading@11: } yading@11: yading@11: /* formats, rates, and layouts are ordered for priority in testing. yading@11: e.g. 'avresample-test 4 2 2' will test all input/output combinations of yading@11: S16/FLTP/S16P/FLT, 48000/44100, and stereo/mono */ yading@11: yading@11: static const enum AVSampleFormat formats[] = { yading@11: AV_SAMPLE_FMT_S16, yading@11: AV_SAMPLE_FMT_FLTP, yading@11: AV_SAMPLE_FMT_S16P, yading@11: AV_SAMPLE_FMT_FLT, yading@11: AV_SAMPLE_FMT_S32P, yading@11: AV_SAMPLE_FMT_S32, yading@11: AV_SAMPLE_FMT_U8P, yading@11: AV_SAMPLE_FMT_U8, yading@11: AV_SAMPLE_FMT_DBLP, yading@11: AV_SAMPLE_FMT_DBL, yading@11: }; yading@11: yading@11: static const int rates[] = { yading@11: 48000, yading@11: 44100, yading@11: 16000 yading@11: }; yading@11: yading@11: static const uint64_t layouts[] = { yading@11: AV_CH_LAYOUT_STEREO, yading@11: AV_CH_LAYOUT_MONO, yading@11: AV_CH_LAYOUT_5POINT1, yading@11: AV_CH_LAYOUT_7POINT1, yading@11: }; yading@11: yading@11: int main(int argc, char **argv) yading@11: { yading@11: AVAudioResampleContext *s; yading@11: AVLFG rnd; yading@11: int ret = 0; yading@11: uint8_t *in_buf = NULL; yading@11: uint8_t *out_buf = NULL; yading@11: unsigned int in_buf_size; yading@11: unsigned int out_buf_size; yading@11: uint8_t *in_data[AVRESAMPLE_MAX_CHANNELS] = { 0 }; yading@11: uint8_t *out_data[AVRESAMPLE_MAX_CHANNELS] = { 0 }; yading@11: int in_linesize; yading@11: int out_linesize; yading@11: uint64_t in_ch_layout; yading@11: int in_channels; yading@11: enum AVSampleFormat in_fmt; yading@11: int in_rate; yading@11: uint64_t out_ch_layout; yading@11: int out_channels; yading@11: enum AVSampleFormat out_fmt; yading@11: int out_rate; yading@11: int num_formats, num_rates, num_layouts; yading@11: int i, j, k, l, m, n; yading@11: yading@11: num_formats = 2; yading@11: num_rates = 2; yading@11: num_layouts = 2; yading@11: if (argc > 1) { yading@11: if (!av_strncasecmp(argv[1], "-h", 3)) { yading@11: av_log(NULL, AV_LOG_INFO, "Usage: avresample-test [ " yading@11: "[ []]]\n" yading@11: "Default is 2 2 2\n"); yading@11: return 0; yading@11: } yading@11: num_formats = strtol(argv[1], NULL, 0); yading@11: num_formats = av_clip(num_formats, 1, FF_ARRAY_ELEMS(formats)); yading@11: } yading@11: if (argc > 2) { yading@11: num_rates = strtol(argv[2], NULL, 0); yading@11: num_rates = av_clip(num_rates, 1, FF_ARRAY_ELEMS(rates)); yading@11: } yading@11: if (argc > 3) { yading@11: num_layouts = strtol(argv[3], NULL, 0); yading@11: num_layouts = av_clip(num_layouts, 1, FF_ARRAY_ELEMS(layouts)); yading@11: } yading@11: yading@11: av_log_set_level(AV_LOG_DEBUG); yading@11: yading@11: av_lfg_init(&rnd, 0xC0FFEE); yading@11: yading@11: in_buf_size = av_samples_get_buffer_size(&in_linesize, 8, 48000 * 6, yading@11: AV_SAMPLE_FMT_DBLP, 0); yading@11: out_buf_size = in_buf_size; yading@11: yading@11: in_buf = av_malloc(in_buf_size); yading@11: if (!in_buf) yading@11: goto end; yading@11: out_buf = av_malloc(out_buf_size); yading@11: if (!out_buf) yading@11: goto end; yading@11: yading@11: s = avresample_alloc_context(); yading@11: if (!s) { yading@11: av_log(NULL, AV_LOG_ERROR, "Error allocating AVAudioResampleContext\n"); yading@11: ret = 1; yading@11: goto end; yading@11: } yading@11: yading@11: for (i = 0; i < num_formats; i++) { yading@11: in_fmt = formats[i]; yading@11: for (k = 0; k < num_layouts; k++) { yading@11: in_ch_layout = layouts[k]; yading@11: in_channels = av_get_channel_layout_nb_channels(in_ch_layout); yading@11: for (m = 0; m < num_rates; m++) { yading@11: in_rate = rates[m]; yading@11: yading@11: ret = av_samples_fill_arrays(in_data, &in_linesize, in_buf, yading@11: in_channels, in_rate * 6, yading@11: in_fmt, 0); yading@11: if (ret < 0) { yading@11: av_log(s, AV_LOG_ERROR, "failed in_data fill arrays\n"); yading@11: goto end; yading@11: } yading@11: audiogen(&rnd, (void **)in_data, in_fmt, in_channels, in_rate, in_rate * 6); yading@11: yading@11: for (j = 0; j < num_formats; j++) { yading@11: out_fmt = formats[j]; yading@11: for (l = 0; l < num_layouts; l++) { yading@11: out_ch_layout = layouts[l]; yading@11: out_channels = av_get_channel_layout_nb_channels(out_ch_layout); yading@11: for (n = 0; n < num_rates; n++) { yading@11: out_rate = rates[n]; yading@11: yading@11: av_log(NULL, AV_LOG_INFO, "%s to %s, %d to %d channels, %d Hz to %d Hz\n", yading@11: av_get_sample_fmt_name(in_fmt), av_get_sample_fmt_name(out_fmt), yading@11: in_channels, out_channels, in_rate, out_rate); yading@11: yading@11: ret = av_samples_fill_arrays(out_data, &out_linesize, yading@11: out_buf, out_channels, yading@11: out_rate * 6, out_fmt, 0); yading@11: if (ret < 0) { yading@11: av_log(s, AV_LOG_ERROR, "failed out_data fill arrays\n"); yading@11: goto end; yading@11: } yading@11: yading@11: av_opt_set_int(s, "in_channel_layout", in_ch_layout, 0); yading@11: av_opt_set_int(s, "in_sample_fmt", in_fmt, 0); yading@11: av_opt_set_int(s, "in_sample_rate", in_rate, 0); yading@11: av_opt_set_int(s, "out_channel_layout", out_ch_layout, 0); yading@11: av_opt_set_int(s, "out_sample_fmt", out_fmt, 0); yading@11: av_opt_set_int(s, "out_sample_rate", out_rate, 0); yading@11: yading@11: av_opt_set_int(s, "internal_sample_fmt", AV_SAMPLE_FMT_FLTP, 0); yading@11: yading@11: ret = avresample_open(s); yading@11: if (ret < 0) { yading@11: av_log(s, AV_LOG_ERROR, "Error opening context\n"); yading@11: goto end; yading@11: } yading@11: yading@11: ret = avresample_convert(s, out_data, out_linesize, out_rate * 6, yading@11: in_data, in_linesize, in_rate * 6); yading@11: if (ret < 0) { yading@11: char errbuf[256]; yading@11: av_strerror(ret, errbuf, sizeof(errbuf)); yading@11: av_log(NULL, AV_LOG_ERROR, "%s\n", errbuf); yading@11: goto end; yading@11: } yading@11: av_log(NULL, AV_LOG_INFO, "Converted %d samples to %d samples\n", yading@11: in_rate * 6, ret); yading@11: if (avresample_get_delay(s) > 0) yading@11: av_log(NULL, AV_LOG_INFO, "%d delay samples not converted\n", yading@11: avresample_get_delay(s)); yading@11: if (avresample_available(s) > 0) yading@11: av_log(NULL, AV_LOG_INFO, "%d samples available for output\n", yading@11: avresample_available(s)); yading@11: av_log(NULL, AV_LOG_INFO, "\n"); yading@11: yading@11: avresample_close(s); yading@11: } yading@11: } yading@11: } yading@11: } yading@11: } yading@11: } yading@11: yading@11: ret = 0; yading@11: yading@11: end: yading@11: av_freep(&in_buf); yading@11: av_freep(&out_buf); yading@11: avresample_free(&s); yading@11: return ret; yading@11: }