tmp_file_rt.h

Go to the documentation of this file.

#ifndef AUDIOGRAPHER_TMP_FILE_RT_H
#define AUDIOGRAPHER_TMP_FILE_RT_H
 
#include <cstdio>
#include <string>
 
#include <glib.h>
 
#include "pbd/gstdio_compat.h"
#include "pbd/pthread_utils.h"
#include "pbd/ringbuffer.h"
 
#include "audiographer/flag_debuggable.h"
#include "audiographer/sink.h"
#include "sndfile_writer.h"
#include "sndfile_reader.h"
 
#include "tmp_file.h"
 
namespace AudioGrapher
{
 
        static const samplecnt_t rb_chunksize = 8192; // samples
 
template<typename T = DefaultSampleType>
class TmpFileRt
        : public TmpFile<T>
{
  public:
 
        TmpFileRt (char * filename_template, int format, ChannelCount channels, samplecnt_t samplerate)
                : SndfileHandle (g_mkstemp(filename_template), true, SndfileBase::ReadWrite, format, channels, samplerate)
                , filename (filename_template)
  , _chunksize (rb_chunksize * channels)
  , _rb (std::max (_chunksize * 16, 5 * samplerate * channels))
        {
                init ();
        }
 
        using SndfileHandle::operator=;
 
        ~TmpFileRt()
        {
                end_write ();
                /* explicitly close first, some OS (yes I'm looking at you windows)
                 * cannot delete files that are still open
                 */
                if (!filename.empty()) {
                        SndfileBase::close();
                        std::remove(filename.c_str());
                }
                pthread_mutex_destroy (&_disk_thread_lock);
                pthread_cond_destroy  (&_data_ready);
        }
 
        void process (ProcessContext<T> const & c)
        {
                SndfileWriter<T>::check_flags (*this, c);
 
                if (SndfileWriter<T>::throw_level (ThrowStrict) && c.channels() != SndfileHandle::channels()) {
                        throw Exception (*this, boost::str (boost::format
                                ("Wrong number of channels given to process(), %1% instead of %2%")
                                % c.channels() % SndfileHandle::channels()));
                }
 
                if (SndfileWriter<T>::throw_level (ThrowProcess) && _rb.write_space() < (size_t) c.samples()) {
                        throw Exception (*this, boost::str (boost::format
                                ("Could not write data to ringbuffer/output file (%1%)")
                                % SndfileHandle::strError()));
                }
 
                _rb.write (c.data(), c.samples());
 
                if (c.has_flag(ProcessContext<T>::EndOfInput)) {
                        _capture = false;
                        SndfileWriter<T>::FileWritten (filename);
                }
 
                if (pthread_mutex_trylock (&_disk_thread_lock) == 0) {
                        pthread_cond_signal (&_data_ready);
                        pthread_mutex_unlock (&_disk_thread_lock);
                }
        }
 
        using Sink<T>::process;
 
        void disk_thread ()
        {
                T *framebuf = (T*) malloc (_chunksize * sizeof (T));
 
                pthread_mutex_lock (&_disk_thread_lock);
 
                while (_capture) {
                        if ((samplecnt_t)_rb.read_space () >= _chunksize) {
                                _rb.read (framebuf, _chunksize);
                                samplecnt_t const written = SndfileBase::write (framebuf, _chunksize);
                                assert (written == _chunksize);
                                SndfileWriter<T>::samples_written += written;
                        }
                        if (!_capture) {
                                break;
                        }
                        pthread_cond_wait (&_data_ready, &_disk_thread_lock);
                }
 
                // flush ringbuffer
                while (_rb.read_space () > 0) {
                        size_t remain = std::min ((samplecnt_t)_rb.read_space (), _chunksize);
                        _rb.read (framebuf, remain);
                        samplecnt_t const written = SndfileBase::write (framebuf, remain);
                        SndfileWriter<T>::samples_written += written;
                }
 
                SndfileWriter<T>::writeSync();
                pthread_mutex_unlock (&_disk_thread_lock);
                free (framebuf);
                TmpFile<T>::FileFlushed ();
        }
 
  protected:
        std::string filename;
 
        bool _capture;
        samplecnt_t _chunksize;
        PBD::RingBuffer<T> _rb;
 
        pthread_mutex_t _disk_thread_lock;
        pthread_cond_t  _data_ready;
        pthread_t _thread_id;
 
        static void * _disk_thread (void *arg)
        {
                TmpFileRt *d = static_cast<TmpFileRt *>(arg);
                pthread_set_name ("ExportDiskIO");
                d->disk_thread ();
                pthread_exit (0);
                return 0;
        }
 
        void end_write () {
                pthread_mutex_lock (&_disk_thread_lock);
                _capture = false;
                pthread_cond_signal (&_data_ready);
                pthread_mutex_unlock (&_disk_thread_lock);
                pthread_join (_thread_id, NULL);
        }
 
        void init()
        {
                SndfileWriter<T>::samples_written = 0;
                _capture = true;
                SndfileWriter<T>::add_supported_flag (ProcessContext<T>::EndOfInput);
                pthread_mutex_init (&_disk_thread_lock, 0);
                pthread_cond_init  (&_data_ready, 0);
 
                if (pthread_create (&_thread_id, NULL, _disk_thread, this)) {
                        _capture = false;
                        if (SndfileWriter<T>::throw_level (ThrowStrict)) {
                                throw Exception (*this, "Cannot create export disk writer");
                        }
                }
        }
 
        private:
        TmpFileRt (TmpFileRt const & other) : SndfileHandle (other) {}
};
 
} // namespace
 
#endif // AUDIOGRAPHER_TMP_FILE_RT_H