test/upmixer_a_test.cc

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