libs/audiographer/audiographer/sndfile/sndfile_writer.h

   1 #ifndef AUDIOGRAPHER_SNDFILE_WRITER_H
   2 #define AUDIOGRAPHER_SNDFILE_WRITER_H
   3
   4 #include <string>
   5
   6 #include <boost/format.hpp>
   7
   8 #include "audiographer/flag_debuggable.h"
   9 #include "audiographer/sink.h"
  10 #include "audiographer/types.h"
  11 #include "audiographer/sndfile/sndfile_base.h"
  12 #include "audiographer/broadcast_info.h"
  13
  14 #include "pbd/signals.h"
  15
  16 namespace AudioGrapher
  17 {
  18
  19 /** Writer for audio files using libsndfile.
  20   * Only short, int and float are valid template parameters
  21   */
  22 template <typename T = DefaultSampleType>
  23 class SndfileWriter
  24   : public virtual SndfileBase
  25   , public Sink<T>
  26   , public Throwing<>
  27   , public FlagDebuggable<>
  28 {
  29   public:
  30         SndfileWriter (std::string const & path, int format, ChannelCount channels, samplecnt_t samplerate, boost::shared_ptr<BroadcastInfo> broadcast_info)
  31           : SndfileHandle (path, Write, format, channels, samplerate)
  32           , path (path)
  33         {
  34                 init();
  35
  36                 if (broadcast_info) {
  37                         broadcast_info->write_to_file (this);
  38                 }
  39         }
  40
  41         virtual ~SndfileWriter () {}
  42
  43         using SndfileHandle::operator=;
  44
  45         samplecnt_t get_samples_written() const { return samples_written; }
  46         void       reset_samples_written_count() { samples_written = 0; }
  47
  48         /// Writes data to file
  49         virtual void process (ProcessContext<T> const & c)
  50         {
  51                 check_flags (*this, c);
  52
  53                 if (throw_level (ThrowStrict) && c.channels() != channels()) {
  54                         throw Exception (*this, boost::str (boost::format
  55                                 ("Wrong number of channels given to process(), %1% instead of %2%")
  56                                 % c.channels() % channels()));
  57                 }
  58
  59                 samplecnt_t const written = write (c.data(), c.samples());
  60                 samples_written += written;
  61
  62                 if (throw_level (ThrowProcess) && written != c.samples()) {
  63                         throw Exception (*this, boost::str (boost::format
  64                                 ("Could not write data to output file (%1%)")
  65                                 % strError()));
  66                 }
  67
  68                 if (c.has_flag(ProcessContext<T>::EndOfInput)) {
  69                         writeSync();
  70                         FileWritten (path);
  71                 }
  72         }
  73
  74         using Sink<T>::process;
  75
  76         PBD::Signal1<void, std::string> FileWritten;
  77
  78   protected:
  79         /// SndfileHandle has to be constructed directly by deriving classes
  80         SndfileWriter ()
  81         {
  82                 init();
  83         }
  84
  85         virtual void init()
  86         {
  87                 if (SF_ERR_NO_ERROR != SndfileHandle::error ()) {
  88                         throw Exception (*this, boost::str (boost::format
  89                                                 ("Could create output file (%1%)") % path));
  90                 }
  91                 samples_written = 0;
  92                 add_supported_flag (ProcessContext<T>::EndOfInput);
  93         }
  94
  95   protected:
  96         std::string path;
  97         samplecnt_t samples_written;
  98
  99         private:
 100         SndfileWriter (SndfileWriter const & other) {}
 101 };
 102
 103 } // namespace
 104
 105 #endif // AUDIOGRAPHER_SNDFILE_WRITER_H