test/ffmpeg_audio_only_test.cc

   1 /*
   2     Copyright (C) 2016 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 "lib/film.h"
  22 #include "lib/ffmpeg_content.h"
  23 #include "lib/dcp_content_type.h"
  24 #include "lib/player.h"
  25 #include "lib/job_manager.h"
  26 #include "lib/audio_buffers.h"
  27 #include "test.h"
  28 #include <sndfile.h>
  29 #include <boost/test/unit_test.hpp>
  30
  31 using boost::shared_ptr;
  32
  33 /** Test the FFmpeg code with audio-only content */
  34 static void
  35 test (boost::filesystem::path file)
  36 {
  37         shared_ptr<Film> film = new_test_film ("ffmpeg_audio_only_test");
  38         film->set_name ("test_film");
  39         film->set_dcp_content_type (DCPContentType::from_pretty_name ("Test"));
  40         shared_ptr<FFmpegContent> c (new FFmpegContent (film, file));
  41         film->examine_and_add_content (c);
  42         wait_for_jobs ();
  43         film->write_metadata ();
  44
  45         /* See if can make a DCP without any errors */
  46         film->make_dcp ();
  47         wait_for_jobs ();
  48         BOOST_CHECK (!JobManager::instance()->errors());
  49
  50         /* Compare the audio data we read with what libsndfile reads */
  51
  52         SF_INFO info;
  53         info.format = 0;
  54         SNDFILE* ref = sf_open (file.string().c_str(), SFM_READ, &info);
  55         /* We don't want to test anything that requires resampling */
  56         BOOST_REQUIRE_EQUAL (info.samplerate, 48000);
  57         float* ref_buffer = new float[info.samplerate * info.channels];
  58
  59         shared_ptr<Player> player (new Player (film, film->playlist ()));
  60
  61         for (DCPTime t; t < film->length(); t += DCPTime::from_seconds (1)) {
  62                 int const N = sf_readf_float (ref, ref_buffer, info.samplerate);
  63                 shared_ptr<AudioBuffers> b = player->get_audio (t, DCPTime::from_frames (N, info.samplerate), true);
  64                 for (int i = 0; i < N; ++i) {
  65                         switch (info.channels) {
  66                         case 1:
  67                                 BOOST_CHECK_EQUAL (ref_buffer[i], b->data(2)[i]);
  68                                 break;
  69                         case 2:
  70                                 BOOST_CHECK_EQUAL (ref_buffer[i*2 + 0], b->data(0)[i]);
  71                                 BOOST_CHECK_EQUAL (ref_buffer[i*2 + 1], b->data(1)[i]);
  72                                 break;
  73                         default:
  74                                 BOOST_REQUIRE (false);
  75                         }
  76                 }
  77         }
  78
  79         sf_close (ref);
  80 }
  81
  82 BOOST_AUTO_TEST_CASE (ffmpeg_audio_only_test1)
  83 {
  84         /* S16 */
  85         test ("test/data/staircase.wav");
  86 }
  87
  88 BOOST_AUTO_TEST_CASE (ffmpeg_audio_only_test2)
  89 {
  90         /* S32 1 channel */
  91         test ("test/data/sine_440.wav");
  92 }