sndfile_writer.h

Go to the documentation of this file.

#ifndef AUDIOGRAPHER_SNDFILE_WRITER_H
#define AUDIOGRAPHER_SNDFILE_WRITER_H
 
#include <string>
 
#include "audiographer/flag_debuggable.h"
#include "audiographer/sink.h"
#include "audiographer/types.h"
#include "audiographer/sndfile/sndfile_base.h"
#include "audiographer/broadcast_info.h"
 
#include "pbd/compose.h"
#include "pbd/signals.h"
 
namespace AudioGrapher
{
 
template <typename T = DefaultSampleType>
class SndfileWriter
  : public virtual SndfileBase
  , public Sink<T>
  , public Throwing<>
  , public FlagDebuggable<>
{
  public:
        SndfileWriter (std::string const & path, int format, ChannelCount channels, samplecnt_t samplerate, std::shared_ptr<BroadcastInfo> broadcast_info)
          : SndfileHandle (path, Write, format, channels, samplerate)
          , path (path)
        {
                init();
 
                if (broadcast_info) {
                        broadcast_info->write_to_file (this);
                }
        }
 
        virtual ~SndfileWriter () {}
 
        using SndfileHandle::operator=;
 
        samplecnt_t get_samples_written() const { return samples_written; }
        void       reset_samples_written_count() { samples_written = 0; }
 
        virtual void process (ProcessContext<T> const & c)
        {
                check_flags (*this, c);
 
                if (throw_level (ThrowStrict) && c.channels() != channels()) {
                        throw Exception (*this, string_compose
                                        ("Wrong number of channels given to process(), %1 instead of %2",
                                        c.channels(), channels()));
                }
 
                samplecnt_t const written = write (c.data(), c.samples());
                samples_written += written;
 
                if (throw_level (ThrowProcess) && written != c.samples()) {
                        std::stringstream reason;
                        reason << "Could not write data to output file ("
                               << strError()
                               << ")";
 
                        throw Exception (*this, reason.str());
                }
 
                if (c.has_flag(ProcessContext<T>::EndOfInput)) {
                        writeSync();
                        FileWritten (path);
                }
        }
 
        using Sink<T>::process;
 
        PBD::Signal<void(std::string)> FileWritten;
 
  protected:
        SndfileWriter ()
        {
                init();
        }
 
        virtual void init()
        {
                if (SF_ERR_NO_ERROR != SndfileHandle::error ()) {
                        std::stringstream reason;
                        reason << "Could not write data to output file ("
                               << path
                               << ")";
 
                        throw Exception (*this, reason.str());
                }
                samples_written = 0;
                add_supported_flag (ProcessContext<T>::EndOfInput);
        }
 
  protected:
        std::string path;
        samplecnt_t samples_written;
 
        private:
        SndfileWriter (SndfileWriter const & other) {}
};
 
} // namespace
 
#endif // AUDIOGRAPHER_SNDFILE_WRITER_H