Handle multiple audio streams in a single piece of content
[dcpomatic.git] / src / lib / sndfile_decoder.cc
index daa363c5e8aaef9b8ecfebac5dbe1e1b0e563b53..2c47fcd9e9ca9ddff3946590dc9008d3ff8d8815 100644 (file)
@@ -1,5 +1,5 @@
 /*
-    Copyright (C) 2012 Carl Hetherington <cth@carlh.net>
+    Copyright (C) 2012-2015 Carl Hetherington <cth@carlh.net>
 
     This program is free software; you can redistribute it and/or modify
     it under the terms of the GNU General Public License as published by
 */
 
 #include <iostream>
+#ifdef DCPOMATIC_WINDOWS
+#include <windows.h>
+#define ENABLE_SNDFILE_WINDOWS_PROTOTYPES 1
+#endif
 #include <sndfile.h>
 #include "sndfile_content.h"
 #include "sndfile_decoder.h"
-#include "film.h"
 #include "exceptions.h"
+#include "audio_buffers.h"
 
 #include "i18n.h"
 
 using std::vector;
 using std::string;
-using std::stringstream;
 using std::min;
 using std::cout;
 using boost::shared_ptr;
-using boost::optional;
 
-/* XXX */
-
-SndfileDecoder::SndfileDecoder (shared_ptr<const Film> f, shared_ptr<SndfileContent> c)
-       : Decoder (f)
-       , AudioDecoder (f, c)
+SndfileDecoder::SndfileDecoder (shared_ptr<const SndfileContent> c)
+       : Sndfile (c)
+       , AudioDecoder (c)
+       , _done (0)
+       , _remaining (_info.frames)
+       , _deinterleave_buffer (0)
 {
-       sf_count_t frames;
-       SNDFILE* sf = open_file (frames);
-       sf_close (sf);
+       
 }
 
-SNDFILE*
-SndfileDecoder::open_file (sf_count_t & frames)
+SndfileDecoder::~SndfileDecoder ()
 {
-       frames = 0;
-       
-       SF_INFO info;
-       SNDFILE* s = sf_open (_sndfile_content->file().string().c_str(), SFM_READ, &info);
-       if (!s) {
-               throw DecodeError (_("could not open external audio file for reading"));
-       }
-
-       frames = info.frames;
-       return s;
+       delete[] _deinterleave_buffer;
 }
 
 bool
-SndfileDecoder::pass ()
+SndfileDecoder::pass (PassReason)
 {
-       sf_count_t frames;
-       SNDFILE* sndfile = open_file (frames);
-
+       if (_remaining == 0) {
+               return true;
+       }
+       
        /* Do things in half second blocks as I think there may be limits
           to what FFmpeg (and in particular the resampler) can cope with.
        */
-       sf_count_t const block = _sndfile_content->audio_frame_rate() / 2;
+       sf_count_t const block = _sndfile_content->audio_stream()->frame_rate() / 2;
+       sf_count_t const this_time = min (block, _remaining);
 
-       shared_ptr<AudioBuffers> audio (new AudioBuffers (_sndfile_content->audio_channels(), block));
-       while (frames > 0) {
-               sf_count_t const this_time = min (block, frames);
-               sf_read_float (sndfile, audio->data(0), this_time);
-               audio->set_frames (this_time);
-               Audio (audio);
-               frames -= this_time;
+       int const channels = _sndfile_content->audio_stream()->channels ();
+       
+       shared_ptr<AudioBuffers> data (new AudioBuffers (channels, this_time));
+
+       if (_sndfile_content->audio_stream()->channels() == 1) {
+               /* No de-interleaving required */
+               sf_read_float (_sndfile, data->data(0), this_time);
+       } else {
+               /* Deinterleave */
+               if (!_deinterleave_buffer) {
+                       _deinterleave_buffer = new float[block * channels];
+               }
+               sf_readf_float (_sndfile, _deinterleave_buffer, this_time);
+               vector<float*> out_ptr (channels);
+               for (int i = 0; i < channels; ++i) {
+                       out_ptr[i] = data->data(i);
+               }
+               float* in_ptr = _deinterleave_buffer;
+               for (int i = 0; i < this_time; ++i) {
+                       for (int j = 0; j < channels; ++j) {
+                               *out_ptr[j]++ = *in_ptr++;
+                       }
+               }
        }
+               
+       data->set_frames (this_time);
+       audio (_sndfile_content->audio_stream (), data, ContentTime::from_frames (_done, _info.samplerate));
+       _done += this_time;
+       _remaining -= this_time;
 
-       sf_close (sndfile);
+       return _remaining == 0;
+}
+
+void
+SndfileDecoder::seek (ContentTime t, bool accurate)
+{
+       AudioDecoder::seek (t, accurate);
 
-       return true;
+       _done = t.frames (_info.samplerate);
+       _remaining = _info.frames - _done;
 }