libs/ardour/ardour/disk_writer.h

   1 /*
   2     Copyright (C) 2009-2016 Paul Davis
   3
   4     This program is free software; you can redistribute it and/or modify
   5     it under the terms of the GNU General Public License as published by
   6     the Free Software Foundation; either version 2 of the License, or
   7     (at your option) any later version.
   8
   9     This program is distributed in the hope that it will be useful,
  10     but WITHOUT ANY WARRANTY; without even the implied warranty of
  11     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12     GNU General Public License for more details.
  13
  14     You should have received a copy of the GNU General Public License
  15     along with this program; if not, write to the Free Software
  16     Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  17
  18 */
  19
  20 #ifndef __ardour_disk_writer_h__
  21 #define __ardour_disk_writer_h__
  22
  23 #include <list>
  24 #include <vector>
  25
  26 #include "pbd/i18n.h"
  27
  28 #include "ardour/disk_io.h"
  29 #include "ardour/midi_buffer.h"
  30
  31 namespace ARDOUR
  32 {
  33
  34 class AudioFileSource;
  35 class SMFSource;
  36 class MidiSource;
  37
  38 class LIBARDOUR_API DiskWriter : public DiskIOProcessor
  39 {
  40 public:
  41         DiskWriter (Session&, std::string const & name, DiskIOProcessor::Flag f = DiskIOProcessor::Flag (0));
  42         ~DiskWriter ();
  43
  44         bool set_name (std::string const & str);
  45         std::string display_name() const { return std::string (_("recorder")); }
  46
  47         bool               recordable()  const { return _flags & Recordable; }
  48
  49         static samplecnt_t chunk_samples() { return _chunk_samples; }
  50         static samplecnt_t default_chunk_samples ();
  51         static void        set_chunk_samples (samplecnt_t n) { _chunk_samples = n; }
  52
  53         void run (BufferSet& /*bufs*/, samplepos_t /*start_sample*/, samplepos_t /*end_sample*/, double speed, pframes_t /*nframes*/, bool /*result_required*/);
  54
  55         void non_realtime_locate (samplepos_t);
  56         void realtime_handle_transport_stopped ();
  57
  58         virtual XMLNode& state (bool full);
  59         int set_state (const XMLNode&, int version);
  60
  61         virtual bool set_write_source_name (const std::string& str);
  62         std::string  write_source_name () const {
  63                 if (_write_source_name.empty()) {
  64                         return name();
  65                 } else {
  66                         return _write_source_name;
  67                 }
  68         }
  69
  70         boost::shared_ptr<AudioFileSource> audio_write_source (uint32_t n=0) {
  71                 boost::shared_ptr<ChannelList> c = channels.reader();
  72                 if (n < c->size()) {
  73                         return (*c)[n]->write_source;
  74                 }
  75
  76                 return boost::shared_ptr<AudioFileSource>();
  77         }
  78
  79         boost::shared_ptr<SMFSource> midi_write_source () { return _midi_write_source; }
  80
  81         virtual std::string steal_write_source_name ();
  82         int  use_new_write_source (DataType, uint32_t n = 0);
  83         void reset_write_sources (bool, bool force = false);
  84
  85         AlignStyle alignment_style() const { return _alignment_style; }
  86         void       set_align_style (AlignStyle, bool force=false);
  87
  88         PBD::Signal0<void> AlignmentStyleChanged;
  89
  90         bool configure_io (ChanCount in, ChanCount out);
  91
  92         std::list<boost::shared_ptr<Source> >& last_capture_sources () { return _last_capture_sources; }
  93
  94         bool         record_enabled() const { return g_atomic_int_get (const_cast<gint*>(&_record_enabled)); }
  95         bool         record_safe () const { return g_atomic_int_get (const_cast<gint*>(&_record_safe)); }
  96         virtual void set_record_enabled (bool yn);
  97         virtual void set_record_safe (bool yn);
  98
  99         bool destructive() const { return _flags & Destructive; }
 100
 101         /** @return Start position of currently-running capture (in session samples) */
 102         samplepos_t current_capture_start() const { return capture_start_sample; }
 103         samplepos_t current_capture_end()   const { return capture_start_sample + capture_captured; }
 104         samplepos_t get_capture_start_sample (uint32_t n = 0) const;
 105         samplecnt_t get_captured_samples (uint32_t n = 0) const;
 106
 107         float buffer_load() const;
 108
 109         virtual void request_input_monitoring (bool) {}
 110         virtual void ensure_input_monitoring (bool) {}
 111
 112         int seek (samplepos_t sample, bool complete_refill);
 113
 114         static PBD::Signal0<void> Overrun;
 115
 116         void set_note_mode (NoteMode m);
 117
 118         /** Emitted when some MIDI data has been received for recording.
 119          *  Parameter is the source that it is destined for.
 120          *  A caller can get a copy of the data with get_gui_feed_buffer ()
 121          */
 122         PBD::Signal1<void, boost::weak_ptr<MidiSource> > DataRecorded;
 123
 124         PBD::Signal0<void> RecordEnableChanged;
 125         PBD::Signal0<void> RecordSafeChanged;
 126
 127         void transport_looped (samplepos_t transport_sample);
 128         void transport_stopped_wallclock (struct tm&, time_t, bool abort);
 129
 130         void adjust_buffering ();
 131
 132         boost::shared_ptr<MidiBuffer> get_gui_feed_buffer () const;
 133
 134 protected:
 135         friend class Track;
 136         int do_flush (RunContext context, bool force = false);
 137
 138         void get_input_sources ();
 139         void prepare_record_status (samplepos_t /*capture_start_sample*/);
 140         void set_align_style_from_io();
 141         void setup_destructive_playlist ();
 142         void use_destructive_playlist ();
 143
 144         void engage_record_enable ();
 145         void disengage_record_enable ();
 146         void engage_record_safe ();
 147         void disengage_record_safe ();
 148
 149         bool prep_record_enable ();
 150         bool prep_record_disable ();
 151
 152         void calculate_record_range (
 153                 Evoral::OverlapType ot, samplepos_t transport_sample, samplecnt_t nframes,
 154                 samplecnt_t& rec_nframes, samplecnt_t& rec_offset
 155                 );
 156
 157         mutable Glib::Threads::Mutex capture_info_lock;
 158         CaptureInfos capture_info;
 159
 160 private:
 161         gint         _record_enabled;
 162         gint         _record_safe;
 163         samplepos_t  capture_start_sample;
 164         samplecnt_t  capture_captured;
 165         bool         was_recording;
 166         samplepos_t  first_recordable_sample;
 167         samplepos_t  last_recordable_sample;
 168         int          last_possibly_recording;
 169         AlignStyle   _alignment_style;
 170         std::string  _write_source_name;
 171
 172         boost::shared_ptr<SMFSource> _midi_write_source;
 173
 174         std::list<boost::shared_ptr<Source> > _last_capture_sources;
 175         std::vector<boost::shared_ptr<AudioFileSource> > capturing_sources;
 176
 177         static samplecnt_t _chunk_samples;
 178
 179         NoteMode           _note_mode;
 180         volatile gint      _samples_pending_write;
 181         volatile gint      _num_captured_loops;
 182         samplepos_t        _accumulated_capture_offset;
 183
 184         /** A buffer that we use to put newly-arrived MIDI data in for
 185             the GUI to read (so that it can update itself).
 186         */
 187         MidiBuffer                   _gui_feed_buffer;
 188         mutable Glib::Threads::Mutex _gui_feed_buffer_mutex;
 189
 190         void check_record_status (samplepos_t transport_sample, double speed, bool can_record);
 191         void finish_capture (boost::shared_ptr<ChannelList> c);
 192 };
 193
 194 } // namespace
 195
 196 #endif /* __ardour_disk_writer_h__ */