More attempts to fix audio / merging etc.
[dcpomatic.git] / test / audio_merger_test.cc
1 /*
2     Copyright (C) 2013 Carl Hetherington <cth@carlh.net>
3
4     This program is free software; you can redistribute it and/or modify
5     it under the terms of the GNU General Public License as published by
6     the Free Software Foundation; either version 2 of the License, or
7     (at your option) any later version.
8
9     This program 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
12     GNU General Public License for more details.
13
14     You should have received a copy of the GNU General Public License
15     along with this program; if not, write to the Free Software
16     Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
17
18 */
19
20 #include <boost/test/unit_test.hpp>
21 #include <boost/bind.hpp>
22 #include <boost/function.hpp>
23 #include <boost/signals2.hpp>
24 #include "lib/audio_merger.h"
25 #include "lib/audio_buffers.h"
26
27 using boost::shared_ptr;
28 using boost::bind;
29
30 static shared_ptr<const AudioBuffers> last_audio;
31 static int last_time;
32
33 static int
34 pass_through (int x)
35 {
36         return x;
37 }
38
39 BOOST_AUTO_TEST_CASE (audio_merger_test1)
40 {
41         AudioMerger<int, int> merger (1, bind (&pass_through, _1), boost::bind (&pass_through, _1));
42
43         /* Push 64 samples, 0 -> 63 at time 0 */
44         shared_ptr<AudioBuffers> buffers (new AudioBuffers (1, 64));
45         for (int i = 0; i < 64; ++i) {
46                 buffers->data()[0][i] = i;
47         }
48         merger.push (buffers, 0);
49
50         /* Push 64 samples, 0 -> 63 at time 22 */
51         merger.push (buffers, 22);
52
53         TimedAudioBuffers<int> tb = merger.pull (22);
54         BOOST_CHECK (tb.audio != shared_ptr<const AudioBuffers> ());
55         BOOST_CHECK_EQUAL (tb.audio->frames(), 22);
56         BOOST_CHECK_EQUAL (tb.time, 0);
57
58         /* And they should be a staircase */
59         for (int i = 0; i < 22; ++i) {
60                 BOOST_CHECK_EQUAL (tb.audio->data()[0][i], i);
61         }
62
63         tb = merger.flush ();
64
65         /* That flush should give us 64 samples at 22 */
66         BOOST_CHECK_EQUAL (tb.audio->frames(), 64);
67         BOOST_CHECK_EQUAL (tb.time, 22);
68
69         /* Check the sample values */
70         for (int i = 0; i < 64; ++i) {
71                 int correct = i;
72                 if (i < (64 - 22)) {
73                         correct += i + 22;
74                 }
75                 BOOST_CHECK_EQUAL (tb.audio->data()[0][i], correct);
76         }
77 }
78
79 BOOST_AUTO_TEST_CASE (audio_merger_test2)
80 {
81         AudioMerger<int, int> merger (1, bind (&pass_through, _1), boost::bind (&pass_through, _1));
82
83         /* Push 64 samples, 0 -> 63 at time 9 */
84         shared_ptr<AudioBuffers> buffers (new AudioBuffers (1, 64));
85         for (int i = 0; i < 64; ++i) {
86                 buffers->data()[0][i] = i;
87         }
88         merger.push (buffers, 9);
89
90         TimedAudioBuffers<int> tb = merger.pull (9);
91         BOOST_CHECK_EQUAL (tb.audio->frames(), 9);
92         BOOST_CHECK_EQUAL (tb.time, 0);
93         
94         for (int i = 0; i < 9; ++i) {
95                 BOOST_CHECK_EQUAL (tb.audio->data()[0][i], 0);
96         }
97         
98         tb = merger.flush ();
99
100         /* That flush should give us 64 samples at 9 */
101         BOOST_CHECK_EQUAL (tb.audio->frames(), 64);
102         BOOST_CHECK_EQUAL (tb.time, 9);
103         
104         /* Check the sample values */
105         for (int i = 0; i < 64; ++i) {
106                 BOOST_CHECK_EQUAL (tb.audio->data()[0][i], i);
107         }
108 }