test/upmixer_a_test.cc

   1 /*
   2     Copyright (C) 2014-2015 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 <sndfile.h>
  22 #include "lib/film.h"
  23 #include "lib/ratio.h"
  24 #include "lib/dcp_content_type.h"
  25 #include "lib/sndfile_content.h"
  26 #include "lib/player.h"
  27 #include "lib/audio_buffers.h"
  28 #include "lib/upmixer_a.h"
  29 #include "test.h"
  30
  31 using boost::shared_ptr;
  32
  33 BOOST_AUTO_TEST_CASE (upmixer_a_test)
  34 {
  35         shared_ptr<Film> film = new_test_film ("upmixer_a_test");
  36         film->set_container (Ratio::from_id ("185"));
  37         film->set_dcp_content_type (DCPContentType::from_isdcf_name ("TLR"));
  38         film->set_name ("frobozz");
  39         film->set_audio_processor (AudioProcessor::from_id ("stereo-5.1-upmix-a"));
  40         shared_ptr<SndfileContent> content (new SndfileContent (film, "test/data/white.wav"));
  41         film->examine_and_add_content (content);
  42
  43         wait_for_jobs ();
  44
  45         SF_INFO info;
  46         info.samplerate = 48000;
  47         info.channels = 1;
  48         info.format = SF_FORMAT_WAV | SF_FORMAT_PCM_16;
  49         SNDFILE* L = sf_open ("build/test/upmixer_a_test/L.wav", SFM_WRITE, &info);
  50         SNDFILE* R = sf_open ("build/test/upmixer_a_test/R.wav", SFM_WRITE, &info);
  51         SNDFILE* C = sf_open ("build/test/upmixer_a_test/C.wav", SFM_WRITE, &info);
  52         SNDFILE* Lfe = sf_open ("build/test/upmixer_a_test/Lfe.wav", SFM_WRITE, &info);
  53         SNDFILE* Ls = sf_open ("build/test/upmixer_a_test/Ls.wav", SFM_WRITE, &info);
  54         SNDFILE* Rs = sf_open ("build/test/upmixer_a_test/Rs.wav", SFM_WRITE, &info);
  55
  56         shared_ptr<Player> player = film->make_player ();
  57         for (DCPTime t; t < film->length(); t += DCPTime::from_seconds (1)) {
  58                 shared_ptr<AudioBuffers> b = player->get_audio (t, DCPTime::from_seconds (1), true);
  59                 sf_write_float (L, b->data(0), b->frames());
  60                 sf_write_float (R, b->data(1), b->frames());
  61                 sf_write_float (C, b->data(2), b->frames());
  62                 sf_write_float (Lfe, b->data(3), b->frames());
  63                 sf_write_float (Ls, b->data(4), b->frames());
  64                 sf_write_float (Rs, b->data(5), b->frames());
  65         }
  66
  67         sf_close (L);
  68         sf_close (R);
  69         sf_close (C);
  70         sf_close (Lfe);
  71         sf_close (Ls);
  72         sf_close (Rs);
  73
  74         check_audio_file ("test/data/upmixer_a_test/L.wav", "build/test/upmixer_a_test/L.wav");
  75         check_audio_file ("test/data/upmixer_a_test/R.wav", "build/test/upmixer_a_test/R.wav");
  76         check_audio_file ("test/data/upmixer_a_test/C.wav", "build/test/upmixer_a_test/C.wav");
  77         check_audio_file ("test/data/upmixer_a_test/Lfe.wav", "build/test/upmixer_a_test/Lfe.wav");
  78         check_audio_file ("test/data/upmixer_a_test/Ls.wav", "build/test/upmixer_a_test/Ls.wav");
  79         check_audio_file ("test/data/upmixer_a_test/Rs.wav", "build/test/upmixer_a_test/Rs.wav");
  80 }