test/audio_filter_test.cc

   1 /*
   2     Copyright (C) 2014-2021 Carl Hetherington <cth@carlh.net>
   3
   4     This file is part of DCP-o-matic.
   5
   6     DCP-o-matic is free software; you can redistribute it and/or modify
   7     it under the terms of the GNU General Public License as published by
   8     the Free Software Foundation; either version 2 of the License, or
   9     (at your option) any later version.
  10
  11     DCP-o-matic 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
  14     GNU General Public License for more details.
  15
  16     You should have received a copy of the GNU General Public License
  17     along with DCP-o-matic.  If not, see <http://www.gnu.org/licenses/>.
  18
  19 */
  20
  21
  22 /** @file  test/audio_filter_test.cc
  23  *  @brief Test AudioFilter, LowPassAudioFilter, HighPassAudioFilter classes.
  24  *  @ingroup selfcontained
  25  */
  26
  27
  28 #include <boost/test/unit_test.hpp>
  29 #include "lib/audio_filter.h"
  30 #include "lib/audio_buffers.h"
  31
  32
  33 using std::make_shared;
  34 using std::shared_ptr;
  35
  36
  37 static void
  38 audio_filter_impulse_test_one (AudioFilter& f, int block_size, int num_blocks)
  39 {
  40         int c = 0;
  41
  42         for (int i = 0; i < num_blocks; ++i) {
  43
  44                 auto in = make_shared<AudioBuffers>(1, block_size);
  45                 for (int j = 0; j < block_size; ++j) {
  46                         in->data()[0][j] = c + j;
  47                 }
  48
  49                 auto out = f.run (in);
  50
  51                 for (int j = 0; j < out->frames(); ++j) {
  52                         BOOST_CHECK_EQUAL (out->data()[0][j], c + j);
  53                 }
  54
  55                 c += block_size;
  56         }
  57 }
  58
  59
  60 /** Create a filter with an impulse as a kernel and check that it
  61  *  passes data through unaltered.
  62  */
  63 BOOST_AUTO_TEST_CASE (audio_filter_impulse_kernel_test)
  64 {
  65         AudioFilter f (0.02);
  66
  67         f._ir.reserve(f._M + 1);
  68         f._ir[0] = 1;
  69         for (int i = 1; i <= f._M; ++i) {
  70                 f._ir[i] = 0;
  71         }
  72
  73         audio_filter_impulse_test_one (f, 32, 1);
  74         audio_filter_impulse_test_one (f, 256, 1);
  75         audio_filter_impulse_test_one (f, 2048, 1);
  76 }
  77
  78
  79 /** Create filters and pass them impulses as input and check that
  80  *  the filter kernels comes back.
  81  */
  82 BOOST_AUTO_TEST_CASE (audio_filter_impulse_input_test)
  83 {
  84         LowPassAudioFilter lpf (0.02, 0.3);
  85
  86         auto in = make_shared<AudioBuffers>(1, 1751);
  87         in->make_silent ();
  88         in->data(0)[0] = 1;
  89
  90         auto out = lpf.run (in);
  91         for (int j = 0; j < out->frames(); ++j) {
  92                 if (j <= lpf._M) {
  93                         BOOST_CHECK_EQUAL (out->data(0)[j], lpf._ir[j]);
  94                 } else {
  95                         BOOST_CHECK_EQUAL (out->data(0)[j], 0);
  96                 }
  97         }
  98
  99         HighPassAudioFilter hpf (0.02, 0.3);
 100
 101         in = make_shared<AudioBuffers>(1, 9133);
 102         in->make_silent ();
 103         in->data(0)[0] = 1;
 104
 105         out = hpf.run (in);
 106         for (int j = 0; j < out->frames(); ++j) {
 107                 if (j <= hpf._M) {
 108                         BOOST_CHECK_EQUAL (out->data(0)[j], hpf._ir[j]);
 109                 } else {
 110                         BOOST_CHECK_EQUAL (out->data(0)[j], 0);
 111                 }
 112         }
 113 }