make note overlap resolution store side effects in a DiffCommand, and add its changes...
[ardour.git] / libs / ardour / midi_source.cc
index b9f7776df88722f4a775440c4e61e68add24c9ab..54b287cbdf1b9b6be031e59dbbe5ec4f77da85df 100644 (file)
 #include <iomanip>
 #include <algorithm>
 
-#include <pbd/xml++.h>
-#include <pbd/pthread_utils.h>
-#include <pbd/basename.h>
-
-#include <ardour/audioengine.h>
-#include <ardour/midi_model.h>
-#include <ardour/midi_ring_buffer.h>
-#include <ardour/midi_source.h>
-#include <ardour/session.h>
-#include <ardour/session_directory.h>
-#include <ardour/source_factory.h>
-#include <ardour/tempo.h>
+#include <glibmm/fileutils.h>
+
+#include "pbd/xml++.h"
+#include "pbd/pthread_utils.h"
+#include "pbd/basename.h"
+
+#include "ardour/audioengine.h"
+#include "ardour/debug.h"
+#include "ardour/midi_model.h"
+#include "ardour/midi_ring_buffer.h"
+#include "ardour/midi_state_tracker.h"
+#include "ardour/midi_source.h"
+#include "ardour/session.h"
+#include "ardour/session_directory.h"
+#include "ardour/source_factory.h"
+#include "ardour/tempo.h"
 
 #include "i18n.h"
 
@@ -47,32 +51,39 @@ using namespace std;
 using namespace ARDOUR;
 using namespace PBD;
 
-sigc::signal<void,MidiSource *> MidiSource::MidiSourceCreated;
+PBD::Signal1<void,MidiSource*> MidiSource::MidiSourceCreated;
 
 MidiSource::MidiSource (Session& s, string name, Source::Flag flags)
-       : Source (s, DataType::MIDI, name, flags)
+       : Source(s, DataType::MIDI, name, flags)
        , _read_data_count(0)
        , _write_data_count(0)
-       , _writing (false)
+       , _writing(false)
+       , _model_iter_valid(false)
+       , _length_beats(0.0)
        , _last_read_end(0)
+       , _last_write_end(0)
 {
 }
 
-MidiSource::MidiSource (Session& s, const XMLNode& node) 
-       : Source (s, node)
+MidiSource::MidiSource (Session& s, const XMLNode& node)
+       : Source(s, node)
        , _read_data_count(0)
        , _write_data_count(0)
-       , _writing (false)
+       , _writing(false)
+       , _model_iter_valid(false)
+       , _length_beats(0.0)
        , _last_read_end(0)
+       , _last_write_end(0)
 {
        _read_data_count = 0;
        _write_data_count = 0;
 
-       if (set_state (node)) {
+       if (set_state (node, Stateful::loading_state_version)) {
                throw failed_constructor();
        }
 }
 
+
 MidiSource::~MidiSource ()
 {
 }
@@ -90,7 +101,7 @@ MidiSource::get_state ()
 }
 
 int
-MidiSource::set_state (const XMLNode& node)
+MidiSource::set_state (const XMLNode& node, int /*version*/)
 {
        const XMLProperty* prop;
 
@@ -101,70 +112,91 @@ MidiSource::set_state (const XMLNode& node)
        return 0;
 }
 
+sframes_t
+MidiSource::length (sframes_t pos) const
+{
+       BeatsFramesConverter converter(_session.tempo_map(), pos);
+       return converter.to(_length_beats);
+}
+
+void
+MidiSource::update_length (sframes_t /*pos*/, sframes_t /*cnt*/)
+{
+       // You're not the boss of me!
+}
+
 void
 MidiSource::invalidate ()
 {
+       _model_iter_valid = false;
        _model_iter.invalidate();
 }
 
 nframes_t
-MidiSource::midi_read (MidiRingBuffer<nframes_t>& dst, nframes_t position,
-               nframes_t start, nframes_t cnt,
-               nframes_t stamp_offset, nframes_t negative_stamp_offset) const
+MidiSource::midi_read (Evoral::EventSink<nframes_t>& dst, sframes_t source_start,
+                       sframes_t start, nframes_t cnt,
+                       sframes_t stamp_offset, sframes_t negative_stamp_offset,
+                       MidiStateTracker* tracker) const
 {
        Glib::Mutex::Lock lm (_lock);
 
-       BeatsFramesConverter converter(_session, position);
+       BeatsFramesConverter converter(_session.tempo_map(), source_start);
 
        if (_model) {
-#define BEATS_TO_FRAMES(t) (converter.to(t) + stamp_offset - negative_stamp_offset)
-
                Evoral::Sequence<double>::const_iterator& i = _model_iter;
-               
-               if (_last_read_end == 0 || start != _last_read_end || !i.valid()) {
-                       cerr << "MidiSource::midi_read seeking to frame " << start << endl;
+
+               // If the cached iterator is invalid, search for the first event past start
+               if (_last_read_end == 0 || start != _last_read_end || !_model_iter_valid) {
+                       DEBUG_TRACE (DEBUG::MidiSourceIO, string_compose ("*** %1 search for relevant iterator for %1 / %2\n", _name, source_start, start));
                        for (i = _model->begin(); i != _model->end(); ++i) {
-                               if (BEATS_TO_FRAMES(i->time()) >= start) {
+                               if (converter.to(i->time()) >= start) {
                                        break;
                                }
                        }
+                       _model_iter_valid = true;
+               } else {
+                       DEBUG_TRACE (DEBUG::MidiSourceIO, string_compose ("*** %1 use cached iterator for %1 / %2\n", _name, source_start, start));
                }
-               
+
                _last_read_end = start + cnt;
 
+               // Read events up to end
                for (; i != _model->end(); ++i) {
-                       const nframes_t time_frames = BEATS_TO_FRAMES(i->time());
+                       const sframes_t time_frames = converter.to(i->time());
                        if (time_frames < start + cnt) {
-                               dst.write(time_frames, i->event_type(), i->size(), i->buffer());
+                               dst.write(time_frames + stamp_offset - negative_stamp_offset,
+                                               i->event_type(), i->size(), i->buffer());
+                               if (tracker) {
+                                       Evoral::MIDIEvent<Evoral::MusicalTime>& ev (*(Evoral::MIDIEvent<Evoral::MusicalTime>*) (&(*i)));
+                                       if (ev.is_note_on()) {
+                                               DEBUG_TRACE (DEBUG::MidiSourceIO, string_compose ("\t%1 add note on %2 @ %3\n", _name, ev.note(), time_frames));
+                                               tracker->add (ev.note(), ev.channel());
+                                       } else if (ev.is_note_off()) {
+                                               DEBUG_TRACE (DEBUG::MidiSourceIO, string_compose ("\t%1 add note off %2 @ %3\n", _name, ev.note(), time_frames));
+                                               tracker->remove (ev.note(), ev.channel());
+                                       }
+                               }
                        } else {
                                break;
                        }
                }
                return cnt;
        } else {
-               return read_unlocked (dst, position, start, cnt, stamp_offset, negative_stamp_offset);
+               return read_unlocked (dst, source_start, start, cnt, stamp_offset, negative_stamp_offset, tracker);
        }
 }
 
 nframes_t
-MidiSource::midi_write (MidiRingBuffer<nframes_t>& dst, nframes_t position, nframes_t cnt)
+MidiSource::midi_write (MidiRingBuffer<nframes_t>& source, sframes_t source_start, nframes_t duration)
 {
        Glib::Mutex::Lock lm (_lock);
-       return write_unlocked (dst, position, cnt);
-}
-
-bool
-MidiSource::file_changed (string path)
-{
-       struct stat stat_file;
-
-       int e1 = stat (path.c_str(), &stat_file);
-       
-       return !e1;
+       const nframes_t ret = write_unlocked (source, source_start, duration);
+       _last_write_end += duration;
+       return ret;
 }
 
 void
-MidiSource::mark_streaming_midi_write_started (NoteMode mode, nframes_t start_frame)
+MidiSource::mark_streaming_midi_write_started (NoteMode mode, sframes_t start_frame)
 {
        set_timeline_position(start_frame);
 
@@ -172,68 +204,87 @@ MidiSource::mark_streaming_midi_write_started (NoteMode mode, nframes_t start_fr
                _model->set_note_mode(mode);
                _model->start_write();
        }
-       
+
+       _last_write_end = start_frame;
        _writing = true;
 }
 
 void
 MidiSource::mark_streaming_write_started ()
 {
-       if (_model) {
-               _model->start_write();
-       }
-
-       _writing = true;
+       NoteMode note_mode = _model ? _model->note_mode() : Sustained;
+       mark_streaming_midi_write_started(note_mode, _session.transport_frame());
 }
 
 void
 MidiSource::mark_streaming_write_completed ()
 {
        if (_model) {
-               _model->end_write(false); // FIXME: param?
+               _model->end_write(false);
        }
 
        _writing = false;
 }
 
+boost::shared_ptr<MidiSource>
+MidiSource::clone (Evoral::MusicalTime begin, Evoral::MusicalTime end)
+{
+        string newname = PBD::basename_nosuffix(_name.val());
+        string newpath;
+
+        /* get a new name for the MIDI file we're going to write to
+         */
+
+        do { 
+
+                newname = bump_name_once (newname, '-');
+                /* XXX build path safely */
+                newpath = _session.session_directory().midi_path().to_string() +"/"+ newname + ".mid";
+
+        } while (Glib::file_test (newpath, Glib::FILE_TEST_EXISTS));
+        
+        boost::shared_ptr<MidiSource> newsrc = boost::dynamic_pointer_cast<MidiSource>(
+                SourceFactory::createWritable(DataType::MIDI, _session,
+                                              newpath, false, _session.frame_rate()));
+        
+        newsrc->set_timeline_position(_timeline_position);
+
+        if (_model) {
+                if (begin == Evoral::MinMusicalTime && end == Evoral::MaxMusicalTime) {
+                        _model->write_to (newsrc);
+                } else {
+                        _model->write_section_to (newsrc, begin, end);
+                }
+        } else {
+                error << string_compose (_("programming error: %1"), X_("no model for MidiSource during ::clone()"));
+                return boost::shared_ptr<MidiSource>();
+        }
+
+        newsrc->flush_midi();
+
+        /* force a reload of the model if the range is partial */
+        
+        if (begin != Evoral::MinMusicalTime || end != Evoral::MaxMusicalTime) {
+                newsrc->load_model (true, true);
+        } else {
+               newsrc->set_model (_model);
+       }
+        
+        return newsrc;
+}
+
 void
 MidiSource::session_saved()
 {
        flush_midi();
 
        if (_model && _model->edited()) {
-               string newname;
-               const string basename = PBD::basename_nosuffix(_name);
-               string::size_type last_dash = basename.find_last_of("-");
-               if (last_dash == string::npos || last_dash == basename.find_first_of("-")) {
-                       newname = basename + "-1";
-               } else {
-                       stringstream ss(basename.substr(last_dash+1));
-                       unsigned write_count = 0;
-                       ss >> write_count;
-                       cerr << "WRITE COUNT: " << write_count << endl;
-                       ++write_count; // start at 1
-                       ss.clear();
-                       ss << basename.substr(0, last_dash) << "-" << write_count;
-                       newname = ss.str();
-               }
-
-               string newpath = _session.session_directory().midi_path().to_string() +"/"+ newname + ".mid";
-
-               boost::shared_ptr<MidiSource> newsrc = boost::dynamic_pointer_cast<MidiSource>(
-                               SourceFactory::createWritable(DataType::MIDI, _session,
-                                               newpath, true, false, _session.frame_rate()));
-
-               newsrc->set_timeline_position(_timeline_position);
-               _model->write_to(newsrc);
+               boost::shared_ptr<MidiSource> newsrc = clone ();
 
-               // cyclic dependency here, ugly :(
-               newsrc->set_model(_model);
-               _model->set_midi_source(newsrc.get());
-               
-               newsrc->flush_midi();
-
-               Switched.emit(newsrc);
+                if (newsrc) {
+                        _model->set_midi_source (newsrc.get());
+                        Switched (newsrc); /* EMIT SIGNAL */
+                }
        }
 }
 
@@ -245,3 +296,9 @@ MidiSource::set_note_mode(NoteMode mode)
        }
 }
 
+void
+MidiSource::drop_model ()
+{
+        cerr << name() << " drop model\n";
+        _model.reset(); 
+}