Remove junk from MidiModel::write_to
[ardour.git] / libs / ardour / midi_model.cc
1 /*
2  Copyright (C) 2007 Paul Davis 
3  Written by Dave Robillard, 2007
4
5  This program is free software; you can redistribute it and/or modify
6  it under the terms of the GNU General Public License as published by
7  the Free Software Foundation; either version 2 of the License, or
8  (at your option) any later version.
9
10  This program is distributed in the hope that it will be useful,
11  but WITHOUT ANY WARRANTY; without even the implied warranty of
12  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
13  GNU General Public License for more details.
14
15  You should have received a copy of the GNU General Public License
16  along with this program; if not, write to the Free Software
17  Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
18
19  */
20
21 #define __STDC_LIMIT_MACROS 1
22
23 #include <iostream>
24 #include <algorithm>
25 #include <stdexcept>
26 #include <stdint.h>
27 #include <pbd/enumwriter.h>
28 #include <midi++/events.h>
29
30 #include <ardour/midi_model.h>
31 #include <ardour/midi_source.h>
32 #include <ardour/types.h>
33 #include <ardour/session.h>
34
35 using namespace std;
36 using namespace ARDOUR;
37
38 void MidiModel::write_lock() {
39         _lock.writer_lock();
40         _automation_lock.lock();
41 }
42
43 void MidiModel::write_unlock() {
44         _lock.writer_unlock();
45         _automation_lock.unlock();
46 }
47
48 void MidiModel::read_lock() const {
49         _lock.reader_lock();
50         /*_automation_lock.lock();*/
51 }
52
53 void MidiModel::read_unlock() const {
54         _lock.reader_unlock();
55         /*_automation_lock.unlock();*/
56 }
57
58 // Read iterator (const_iterator)
59
60 MidiModel::const_iterator::const_iterator(const MidiModel& model, double t)
61         : _model(&model)
62         , _is_end( (t == DBL_MAX) || model.empty())
63         , _locked( !_is_end)
64 {
65         //cerr << "Created MIDI iterator @ " << t << " (is end: " << _is_end << ")" << endl;
66
67         if (_is_end)
68                 return;
69
70         model.read_lock();
71
72         _note_iter = model.notes().end();
73         // find first note which begins after t
74         for (MidiModel::Notes::const_iterator i = model.notes().begin(); i != model.notes().end(); ++i) {
75                 if ((*i)->time() >= t) {
76                         _note_iter = i;
77                         break;
78                 }
79         }
80
81         MidiControlIterator earliest_control(boost::shared_ptr<AutomationList>(), DBL_MAX, 0.0);
82
83         _control_iters.reserve(model.controls().size());
84         for (Automatable::Controls::const_iterator i = model.controls().begin();
85                         i != model.controls().end(); ++i) {
86
87                 assert(
88                         i->first.type() == MidiCCAutomation ||
89                         i->first.type() == MidiPgmChangeAutomation ||
90                         i->first.type() == MidiPitchBenderAutomation ||
91                         i->first.type() == MidiChannelAftertouchAutomation);
92
93                 double x, y;
94                 bool ret = i->second->list()->rt_safe_earliest_event_unlocked(t, DBL_MAX, x, y);
95                 if (!ret) {
96                         //cerr << "MIDI Iterator: CC " << i->first.id() << " (size " << i->second->list()->size()
97                         //      << ") has no events past " << t << endl;
98                         continue;
99                 }
100
101                 assert(x >= 0);
102                 assert(y >= i->first.min());
103                 assert(y <= i->first.max());
104
105                 const MidiControlIterator new_iter(i->second->list(), x, y);
106
107                 //cerr << "MIDI Iterator: CC " << i->first.id() << " added (" << x << ", " << y << ")" << endl;
108                 _control_iters.push_back(new_iter);
109
110                 if (x < earliest_control.x) {
111                         earliest_control = new_iter;
112                         _control_iter = _control_iters.end();
113                         --_control_iter;
114                 }
115         }
116
117         if (_note_iter != model.notes().end()) {
118                 _event = MIDI::Event((*_note_iter)->on_event(), false);
119                 _active_notes.push(*_note_iter);
120                 //cerr << " new const iterator: size active notes: " << _active_notes.size() << " is empty: " << _active_notes.empty() << endl;
121                 ++_note_iter;
122         }
123
124         if (earliest_control.automation_list.get() && earliest_control.x < _event.time()) {
125                 model.control_to_midi_event(_event, earliest_control);
126         } else {
127                 _control_iter = _control_iters.end();
128         }
129
130         if (_event.size() == 0) {
131                 //cerr << "Created MIDI iterator @ " << t << " is at end." << endl;
132                 _is_end = true;
133
134                 // FIXME: possible race condition here....
135                 if (_locked) {
136                         _model->read_unlock();
137                         _locked = false;
138                 }
139         } else {
140                 //printf("New MIDI Iterator = %X @ %lf\n", _event.type(), _event.time());
141         }
142 }
143
144 MidiModel::const_iterator::~const_iterator()
145 {
146         if (_locked) {
147                 _model->read_unlock();
148         }
149 }
150
151 const MidiModel::const_iterator& MidiModel::const_iterator::operator++()
152 {
153         if (_is_end) {
154                 throw std::logic_error("Attempt to iterate past end of MidiModel");
155         }
156
157         /*cerr << "const_iterator::operator++: _event type:" << hex << "0x" << int(_event.type()) 
158          << "   buffer: 0x" << int(_event.buffer()[0]) << " 0x" << int(_event.buffer()[1]) 
159          << " 0x" << int(_event.buffer()[2]) << endl;*/
160
161         if (! (_event.is_note() || _event.is_cc() || _event.is_pgm_change() || _event.is_pitch_bender() || _event.is_channel_aftertouch()) ) {
162                 cerr << "FAILED event buffer: " << hex << int(_event.buffer()[0]) << int(_event.buffer()[1]) << int(_event.buffer()[2]) << endl;
163         }
164         assert((_event.is_note() || _event.is_cc() || _event.is_pgm_change() || _event.is_pitch_bender() || _event.is_channel_aftertouch()));
165
166         // Increment past current control event
167         if (!_event.is_note() && _control_iter != _control_iters.end() && _control_iter->automation_list.get()) {
168                 double x = 0.0, y = 0.0;
169                 const bool ret = _control_iter->automation_list->rt_safe_earliest_event_unlocked(
170                                 _control_iter->x, DBL_MAX, x, y, false);
171                 //cerr << "control_iter x:" << _control_iter->x << " y:" << _control_iter->y << endl;
172
173                 if (ret) {
174                         //cerr << "Incremented " << _control_iter->automation_list->parameter().id() << " to " << x << endl;
175                         _control_iter->x = x;
176                         _control_iter->y = y;
177                 } else {
178                         cerr << "Hit end of " << _control_iter->automation_list->parameter().id() << endl;
179                         _control_iter->automation_list.reset();
180                         _control_iter->x = DBL_MAX;
181                 }
182         }
183
184         // Now find and point at the earliest event
185
186         const std::vector<MidiControlIterator>::iterator old_control_iter = _control_iter;
187         _control_iter = _control_iters.begin();
188
189         for (std::vector<MidiControlIterator>::iterator i = _control_iters.begin();
190                         i != _control_iters.end(); ++i) {
191                 if (i->x < _control_iter->x) {
192                         _control_iter = i;
193                 }
194         }
195
196         enum Type {NIL, NOTE_ON, NOTE_OFF, AUTOMATION};
197
198         Type type = NIL;
199         double t = 0;
200
201         // Next earliest note on
202         if (_note_iter != _model->notes().end()) {
203                 type = NOTE_ON;
204                 t = (*_note_iter)->time();
205         }
206
207         // Use the next earliest note off iff it's earlier than the note on
208         if (_model->note_mode() == Sustained && (! _active_notes.empty())) {
209                 if (type == NIL || _active_notes.top()->end_time() <= (*_note_iter)->time()) {
210                         type = NOTE_OFF;
211                         t = _active_notes.top()->end_time();
212                 }
213         }
214
215         // Use the next earliest controller iff it's earlier than the note event
216         if (_control_iter != _control_iters.end()
217                         && _control_iter->x != DBL_MAX
218            )//&& _control_iter != old_control_iter)
219                 if (type == NIL || _control_iter->x < t)
220                         type = AUTOMATION;
221
222         if (type == NOTE_ON) {
223                 //cerr << "********** MIDI Iterator = note on" << endl;
224                 _event = MIDI::Event((*_note_iter)->on_event(), false);
225                 _active_notes.push(*_note_iter);
226                 ++_note_iter;
227         } else if (type == NOTE_OFF) {
228                 //cerr << "********** MIDI Iterator = note off" << endl;
229                 _event = MIDI::Event(_active_notes.top()->off_event(), false);
230                 _active_notes.pop();
231         } else if (type == AUTOMATION) {
232                 //cerr << "********** MIDI Iterator = Automation" << endl;
233                 _model->control_to_midi_event(_event, *_control_iter);
234         } else {
235                 //cerr << "********** MIDI Iterator = End" << endl;
236                 _is_end = true;
237         }
238
239         assert(_is_end || _event.size()> 0);
240
241         return *this;
242 }
243
244 bool MidiModel::const_iterator::operator==(const const_iterator& other) const
245 {
246         if (_is_end || other._is_end) {
247                 return (_is_end == other._is_end);
248         } else {
249                 return (_event == other._event);
250         }
251 }
252
253 MidiModel::const_iterator& MidiModel::const_iterator::operator=(const const_iterator& other)
254 {
255         if (_locked && _model != other._model)
256                 _model->read_unlock();
257
258         assert( ! other._event.owns_buffer());
259
260         _model = other._model;
261         _event = other._event;
262         _active_notes = other._active_notes;
263         _is_end = other._is_end;
264         _locked = other._locked;
265         _note_iter = other._note_iter;
266         _control_iters = other._control_iters;
267         size_t index = other._control_iter - other._control_iters.begin();
268         _control_iter = _control_iters.begin() + index;
269
270         assert( !_event.owns_buffer() );
271
272         return *this;
273 }
274
275 // MidiModel
276
277 MidiModel::MidiModel(MidiSource *s, size_t size)
278         : Automatable(s->session(), "midi model")
279         , _notes(size)
280         , _note_mode(Sustained)
281         , _writing(false)
282         , _edited(false)
283         , _end_iter(*this, DBL_MAX)
284         , _next_read(UINT32_MAX)
285         , _read_iter(*this, DBL_MAX)
286         , _midi_source(s)
287 {
288         assert(_end_iter._is_end);
289         assert( ! _end_iter._locked);
290 }
291
292 /** Read events in frame range \a start .. \a start+cnt into \a dst,
293  * adding \a stamp_offset to each event's timestamp.
294  * \return number of events written to \a dst
295  */
296 size_t MidiModel::read(MidiRingBuffer& dst, nframes_t start, nframes_t nframes,
297                 nframes_t stamp_offset, nframes_t negative_stamp_offset) const
298 {
299         //cerr << this << " MM::read @ " << start << " frames: " << nframes << " -> " << stamp_offset << endl;
300         //cerr << this << " MM # notes: " << n_notes() << endl;
301
302         size_t read_events = 0;
303
304         if (start != _next_read) {
305                 _read_iter = const_iterator(*this, (double)start);
306                 //cerr << "Repositioning iterator from " << _next_read << " to " << start << endl;
307         } else {
308                 //cerr << "Using cached iterator at " << _next_read << endl;
309         }
310
311         _next_read = start + nframes;
312
313         while (_read_iter != end() && _read_iter->time() < start + nframes) {
314                 assert(_read_iter->size()> 0);
315                 assert(_read_iter->buffer());
316                 dst.write(_read_iter->time() + stamp_offset - negative_stamp_offset,
317                                 _read_iter->size(), _read_iter->buffer());
318
319                  //cerr << this << " MidiModel::read event @ " << _read_iter->time()  
320                  //<< " type: " << hex << int(_read_iter->type()) << dec 
321                  //<< " note: " << int(_read_iter->note()) 
322                  //<< " velocity: " << int(_read_iter->velocity()) 
323                  //<< endl;
324                  
325                 ++_read_iter;
326                 ++read_events;
327         }
328
329         return read_events;
330 }
331
332 bool MidiModel::control_to_midi_event(MIDI::Event& ev,
333                 const MidiControlIterator& iter) const
334 {
335         assert(iter.automation_list.get() != 0);
336         
337         switch (iter.automation_list->parameter().type()) {
338         case MidiCCAutomation:
339                 if (ev.size() < 3) {
340                         ev.set_buffer((Byte*)malloc(3), true);
341                 }
342
343                 assert(iter.automation_list);
344                 assert(iter.automation_list->parameter().channel() < 16);
345                 assert(iter.automation_list->parameter().id() <= INT8_MAX);
346                 assert(iter.y <= INT8_MAX);
347                 ev.buffer()[0] = MIDI_CMD_CONTROL + iter.automation_list->parameter().channel();
348                 ev.buffer()[1] = (Byte)iter.automation_list->parameter().id();
349                 ev.buffer()[2] = (Byte)iter.y;
350                 ev.time() = iter.x;
351                 ev.size() = 3;
352                 return true;
353
354         case MidiPgmChangeAutomation:
355                 if (ev.size() < 2) {
356                         ev.set_buffer((Byte*)malloc(2), true);
357                 }
358
359                 assert(iter.automation_list);
360                 assert(iter.automation_list->parameter().channel() < 16);
361                 assert(iter.automation_list->parameter().id() == 0);
362                 assert(iter.y <= INT8_MAX);
363                 ev.buffer()[0] = MIDI_CMD_PGM_CHANGE + iter.automation_list->parameter().channel();
364                 ev.buffer()[1] = (Byte)iter.y;
365                 ev.time() = iter.x;
366                 ev.size() = 2;
367                 return true;
368
369         case MidiPitchBenderAutomation:
370                 if (ev.size() < 3) {
371                         ev.set_buffer((Byte*)malloc(3), true);
372                 }
373
374                 assert(iter.automation_list);
375                 assert(iter.automation_list->parameter().channel() < 16);
376                 assert(iter.automation_list->parameter().id() == 0);
377                 assert(iter.y < (1<<14));
378                 ev.buffer()[0] = MIDI_CMD_BENDER + iter.automation_list->parameter().channel();
379                 ev.buffer()[1] = ((Byte)iter.y) & 0x7F; // LSB
380                 ev.buffer()[2] = (((Byte)iter.y) >> 7) & 0x7F; // MSB
381                 ev.time() = iter.x;
382                 ev.size() = 3;
383                 return true;
384
385         case MidiChannelAftertouchAutomation:
386                 if (ev.size() < 2) {
387                         ev.set_buffer((Byte*)malloc(2), true);
388                 }
389
390                 assert(iter.automation_list);
391                 assert(iter.automation_list->parameter().channel() < 16);
392                 assert(iter.automation_list->parameter().id() == 0);
393                 assert(iter.y <= INT8_MAX);
394                 ev.buffer()[0]
395                                 = MIDI_CMD_CHANNEL_PRESSURE + iter.automation_list->parameter().channel();
396                 ev.buffer()[1] = (Byte)iter.y;
397                 ev.time() = iter.x;
398                 ev.size() = 2;
399                 return true;
400
401         default:
402                 return false;
403         }
404 }
405
406
407 /** Clear all events from the model.
408  */
409 void MidiModel::clear()
410 {
411         _lock.writer_lock();
412         _notes.clear();
413         clear_automation();
414         _next_read = 0;
415         _read_iter = end();
416         _lock.writer_unlock();
417 }
418
419
420 /** Begin a write of events to the model.
421  *
422  * If \a mode is Sustained, complete notes with duration are constructed as note
423  * on/off events are received.  Otherwise (Percussive), only note on events are
424  * stored; note off events are discarded entirely and all contained notes will
425  * have duration 0.
426  */
427 void MidiModel::start_write()
428 {
429         //cerr << "MM " << this << " START WRITE, MODE = " << enum_2_string(_note_mode) << endl;
430         write_lock();
431         _writing = true;
432         for (int i = 0; i < 16; ++i)
433                 _write_notes[i].clear();
434         
435         _dirty_automations.clear();
436         write_unlock();
437 }
438
439 /** Finish a write of events to the model.
440  *
441  * If \a delete_stuck is true and the current mode is Sustained, note on events
442  * that were never resolved with a corresonding note off will be deleted.
443  * Otherwise they will remain as notes with duration 0.
444  */
445 void MidiModel::end_write(bool delete_stuck)
446 {
447         write_lock();
448         assert(_writing);
449
450         //cerr << "MM " << this << " END WRITE: " << _notes.size() << " NOTES\n";
451
452         if (_note_mode == Sustained && delete_stuck) {
453                 for (Notes::iterator n = _notes.begin(); n != _notes.end() ;) {
454                         if ((*n)->duration() == 0) {
455                                 cerr << "WARNING: Stuck note lost: " << (*n)->note() << endl;
456                                 n = _notes.erase(n);
457                                 // we have to break here because erase invalidates the iterator
458                                 break;
459                         } else {
460                                 ++n;
461                         }
462                 }
463         }
464
465         for (int i = 0; i < 16; ++i) {
466                 if (!_write_notes[i].empty()) {
467                         cerr << "WARNING: MidiModel::end_write: Channel " << i << " has "
468                                         << _write_notes[i].size() << " stuck notes" << endl;
469                 }
470                 _write_notes[i].clear();
471         }
472
473         for (AutomationLists::const_iterator i = _dirty_automations.begin(); i != _dirty_automations.end(); ++i) {
474                 (*i)->Dirty.emit();
475                 (*i)->lookup_cache().left = -1;
476                 (*i)->search_cache().left = -1;
477         }
478         
479         _writing = false;
480         write_unlock();
481 }
482
483 /** Append \a in_event to model.  NOT realtime safe.
484  *
485  * Timestamps of events in \a buf are expected to be relative to
486  * the start of this model (t=0) and MUST be monotonically increasing
487  * and MUST be >= the latest event currently in the model.
488  */
489 void MidiModel::append(const MIDI::Event& ev)
490 {
491         write_lock();
492         _edited = true;
493
494         assert(_notes.empty() || ev.time() >= _notes.back()->time());
495         assert(_writing);
496
497         if (ev.is_note_on()) {
498                 append_note_on_unlocked(ev.channel(), ev.time(), ev.note(),
499                                 ev.velocity());
500         } else if (ev.is_note_off()) {
501                 append_note_off_unlocked(ev.channel(), ev.time(), ev.note());
502         } else if (ev.is_cc()) {
503                 append_automation_event_unlocked(MidiCCAutomation, ev.channel(),
504                                 ev.time(), ev.cc_number(), ev.cc_value());
505         } else if (ev.is_pgm_change()) {
506                 append_automation_event_unlocked(MidiPgmChangeAutomation, ev.channel(),
507                                 ev.time(), ev.pgm_number(), 0);
508         } else if (ev.is_pitch_bender()) {
509                 append_automation_event_unlocked(MidiPitchBenderAutomation,
510                                 ev.channel(), ev.time(), ev.pitch_bender_lsb(),
511                                 ev.pitch_bender_msb());
512         } else if (ev.is_channel_aftertouch()) {
513                 append_automation_event_unlocked(MidiChannelAftertouchAutomation,
514                                 ev.channel(), ev.time(), ev.channel_aftertouch(), 0);
515         } else {
516                 printf("WARNING: MidiModel: Unknown event type %X\n", ev.type());
517         }
518
519         write_unlock();
520 }
521
522 void MidiModel::append_note_on_unlocked(uint8_t chan, double time,
523                 uint8_t note_num, uint8_t velocity)
524 {
525         /*cerr << "MidiModel " << this << " chan " << (int)chan <<
526          " note " << (int)note_num << " on @ " << time << endl;*/
527
528         assert(chan < 16);
529         assert(_writing);
530         _edited = true;
531
532         boost::shared_ptr<Note> new_note(new Note(chan, time, 0, note_num, velocity));
533         _notes.push_back(new_note);
534         if (_note_mode == Sustained) {
535                 //cerr << "MM Sustained: Appending active note on " << (unsigned)(uint8_t)note_num << endl;
536                 _write_notes[chan].push_back(_notes.size() - 1);
537         }/* else {
538          cerr << "MM Percussive: NOT appending active note on" << endl;
539          }*/
540 }
541
542 void MidiModel::append_note_off_unlocked(uint8_t chan, double time,
543                 uint8_t note_num)
544 {
545         /*cerr << "MidiModel " << this << " chan " << (int)chan <<
546          " note " << (int)note_num << " off @ " << time << endl;*/
547
548         assert(chan < 16);
549         assert(_writing);
550         _edited = true;
551
552         if (_note_mode == Percussive) {
553                 cerr << "MidiModel Ignoring note off (percussive mode)" << endl;
554                 return;
555         }
556
557         /* FIXME: make _write_notes fixed size (127 noted) for speed */
558
559         /* FIXME: note off velocity for that one guy out there who actually has
560          * keys that send it */
561
562         bool resolved = false;
563
564         for (WriteNotes::iterator n = _write_notes[chan].begin(); n
565                         != _write_notes[chan].end(); ++n) {
566                 Note& note = *_notes[*n].get();
567                 //cerr << (unsigned)(uint8_t)note.note() << " ? " << (unsigned)note_num << endl;
568                 if (note.note() == note_num) {
569                         assert(time >= note.time());
570                         note.set_duration(time - note.time());
571                         _write_notes[chan].erase(n);
572                         //cerr << "MM resolved note, duration: " << note.duration() << endl;
573                         resolved = true;
574                         break;
575                 }
576         }
577
578         if (!resolved) {
579                 cerr << "MidiModel " << this << " spurious note off chan " << (int)chan
580                                 << ", note " << (int)note_num << " @ " << time << endl;
581         }
582 }
583
584 void MidiModel::append_automation_event_unlocked(AutomationType type,
585                 uint8_t chan, double time, uint8_t first_byte, uint8_t second_byte)
586 {
587         //cerr << "MidiModel " << this << " chan " << (int)chan <<
588         //              " CC " << (int)number << " = " << (int)value << " @ " << time << endl;
589
590         assert(chan < 16);
591         assert(_writing);
592         _edited = true;
593         double value;
594
595         uint32_t id = 0;
596
597         switch (type) {
598         case MidiCCAutomation:
599                 id = first_byte;
600                 value = double(second_byte);
601                 break;
602         case MidiChannelAftertouchAutomation:
603         case MidiPgmChangeAutomation:
604                 id = 0;
605                 value = double(first_byte);
606                 break;
607         case MidiPitchBenderAutomation:
608                 id = 0;
609                 value = double((0x7F & second_byte) << 7 | (0x7F & first_byte));
610                 break;
611         default:
612                 assert(false);
613         }
614
615         Parameter param(type, id, chan);
616         boost::shared_ptr<AutomationControl> control = Automatable::control(param, true);
617         control->list()->rt_add(time, value);
618 }
619
620 void MidiModel::add_note_unlocked(const boost::shared_ptr<Note> note)
621 {
622         //cerr << "MidiModel " << this << " add note " << (int)note.note() << " @ " << note.time() << endl;
623         _edited = true;
624         Notes::iterator i = upper_bound(_notes.begin(), _notes.end(), note,
625                         note_time_comparator);
626         _notes.insert(i, note);
627 }
628
629 void MidiModel::remove_note_unlocked(const boost::shared_ptr<const Note> note)
630 {
631         _edited = true;
632         //cerr << "MidiModel " << this << " remove note " << (int)note.note() << " @ " << note.time() << endl;
633         for (Notes::iterator n = _notes.begin(); n != _notes.end(); ++n) {
634                 Note& _n = *(*n);
635                 const Note& _note = *note;
636                 // TODO: There is still the issue, that after restarting ardour
637                 // persisted undo does not work, because of rounding errors in the
638                 // event times after saving/restoring to/from MIDI files
639                 cerr << "======================================= " << endl;
640                 cerr << int(_n.note()) << "@" << int(_n.time()) << "[" << int(_n.channel()) << "] --" << int(_n.duration()) << "-- #" << int(_n.velocity()) << endl;
641                 cerr << int(_note.note()) << "@" << int(_note.time()) << "[" << int(_note.channel()) << "] --" << int(_note.duration()) << "-- #" << int(_note.velocity()) << endl;
642                 cerr << "Equal: " << bool(_n == _note) << endl;
643                 cerr << endl << endl;
644                 if (_n == _note) {
645                         _notes.erase(n);
646                         // we have to break here, because erase invalidates all iterators, ie. n itself
647                         break;
648                 }
649         }
650 }
651
652 /** Slow!  for debugging only. */
653 #ifndef NDEBUG
654 bool MidiModel::is_sorted() const {
655         bool t = 0;
656         for (Notes::const_iterator n = _notes.begin(); n != _notes.end(); ++n)
657                 if ((*n)->time() < t)
658                         return false;
659                 else
660                         t = (*n)->time();
661
662         return true;
663 }
664 #endif
665
666 /** Start a new command.
667  *
668  * This has no side-effects on the model or Session, the returned command
669  * can be held on to for as long as the caller wishes, or discarded without
670  * formality, until apply_command is called and ownership is taken.
671  */
672 MidiModel::DeltaCommand* MidiModel::new_delta_command(const string name)
673 {
674         DeltaCommand* cmd = new DeltaCommand(_midi_source->model(), name);
675         return cmd;
676 }
677
678 /** Apply a command.
679  *
680  * Ownership of cmd is taken, it must not be deleted by the caller.
681  * The command will constitute one item on the undo stack.
682  */
683 void MidiModel::apply_command(Command* cmd)
684 {
685         _session.begin_reversible_command(cmd->name());
686         (*cmd)();
687         assert(is_sorted());
688         _session.commit_reversible_command(cmd);
689         _edited = true;
690 }
691
692 // MidiEditCommand
693
694 MidiModel::DeltaCommand::DeltaCommand(boost::shared_ptr<MidiModel> m,
695                 const std::string& name)
696         : Command(name)
697         , _model(m)
698         , _name(name)
699 {
700 }
701
702 MidiModel::DeltaCommand::DeltaCommand(boost::shared_ptr<MidiModel> m,
703                 const XMLNode& node)
704         : _model(m)
705 {
706         set_state(node);
707 }
708
709 void MidiModel::DeltaCommand::add(const boost::shared_ptr<Note> note)
710 {
711         //cerr << "MEC: apply" << endl;
712         _removed_notes.remove(note);
713         _added_notes.push_back(note);
714 }
715
716 void MidiModel::DeltaCommand::remove(const boost::shared_ptr<Note> note)
717 {
718         //cerr << "MEC: remove" << endl;
719         _added_notes.remove(note);
720         _removed_notes.push_back(note);
721 }
722
723 void MidiModel::DeltaCommand::operator()()
724 {
725         // This could be made much faster by using a priority_queue for added and
726         // removed notes (or sort here), and doing a single iteration over _model
727
728         // Need to reset iterator to drop the read lock it holds, or we'll deadlock
729         const bool reset_iter = (_model->_read_iter.locked());
730         const double iter_time = _model->_read_iter->time();
731
732         if (reset_iter)
733                 _model->_read_iter = _model->end(); // drop read lock
734
735         assert( ! _model->_read_iter.locked());
736
737         _model->write_lock();
738
739         for (std::list< boost::shared_ptr<Note> >::iterator i = _added_notes.begin(); i != _added_notes.end(); ++i)
740                 _model->add_note_unlocked(*i);
741
742         for (std::list< boost::shared_ptr<Note> >::iterator i = _removed_notes.begin(); i != _removed_notes.end(); ++i)
743                 _model->remove_note_unlocked(*i);
744
745         _model->write_unlock();
746
747         if (reset_iter)
748         _model->_read_iter = const_iterator(*_model.get(), iter_time);
749
750         _model->ContentsChanged(); /* EMIT SIGNAL */
751 }
752
753 void MidiModel::DeltaCommand::undo()
754 {
755         // This could be made much faster by using a priority_queue for added and
756         // removed notes (or sort here), and doing a single iteration over _model
757
758         // Need to reset iterator to drop the read lock it holds, or we'll deadlock
759         const bool reset_iter = (_model->_read_iter.locked());
760         const double iter_time = _model->_read_iter->time();
761
762         if (reset_iter)
763                 _model->_read_iter = _model->end(); // drop read lock
764
765         assert( ! _model->_read_iter.locked());
766
767         _model->write_lock();
768
769         for (std::list< boost::shared_ptr<Note> >::iterator i = _added_notes.begin(); i
770                         != _added_notes.end(); ++i)
771                 _model->remove_note_unlocked(*i);
772
773         for (std::list< boost::shared_ptr<Note> >::iterator i =
774                         _removed_notes.begin(); i != _removed_notes.end(); ++i)
775                 _model->add_note_unlocked(*i);
776
777         _model->write_unlock();
778
779         if (reset_iter)
780                 _model->_read_iter = const_iterator(*_model.get(), iter_time);
781
782         _model->ContentsChanged(); /* EMIT SIGNAL */
783 }
784
785 XMLNode & MidiModel::DeltaCommand::marshal_note(const boost::shared_ptr<Note> note)
786 {
787         XMLNode *xml_note = new XMLNode("note");
788         ostringstream note_str(ios::ate);
789         note_str << int(note->note());
790         xml_note->add_property("note", note_str.str());
791
792         ostringstream channel_str(ios::ate);
793         channel_str << int(note->channel());
794         xml_note->add_property("channel", channel_str.str());
795
796         ostringstream time_str(ios::ate);
797         time_str << int(note->time());
798         xml_note->add_property("time", time_str.str());
799
800         ostringstream duration_str(ios::ate);
801         duration_str <<(unsigned int) note->duration();
802         xml_note->add_property("duration", duration_str.str());
803
804         ostringstream velocity_str(ios::ate);
805         velocity_str << (unsigned int) note->velocity();
806         xml_note->add_property("velocity", velocity_str.str());
807
808         return *xml_note;
809 }
810
811 boost::shared_ptr<Note> MidiModel::DeltaCommand::unmarshal_note(XMLNode *xml_note)
812 {
813         unsigned int note;
814         istringstream note_str(xml_note->property("note")->value());
815         note_str >> note;
816
817         unsigned int channel;
818         istringstream channel_str(xml_note->property("channel")->value());
819         channel_str >> channel;
820
821         unsigned int time;
822         istringstream time_str(xml_note->property("time")->value());
823         time_str >> time;
824
825         unsigned int duration;
826         istringstream duration_str(xml_note->property("duration")->value());
827         duration_str >> duration;
828
829         unsigned int velocity;
830         istringstream velocity_str(xml_note->property("velocity")->value());
831         velocity_str >> velocity;
832
833         boost::shared_ptr<Note> note_ptr(new Note(channel, time, duration, note, velocity));
834         return note_ptr;
835 }
836
837 #define ADDED_NOTES_ELEMENT "added_notes"
838 #define REMOVED_NOTES_ELEMENT "removed_notes"
839 #define DELTA_COMMAND_ELEMENT "DeltaCommand"
840
841 int MidiModel::DeltaCommand::set_state(const XMLNode& delta_command)
842 {
843         if (delta_command.name() != string(DELTA_COMMAND_ELEMENT)) {
844                 return 1;
845         }
846
847         _added_notes.clear();
848         XMLNode *added_notes = delta_command.child(ADDED_NOTES_ELEMENT);
849         XMLNodeList notes = added_notes->children();
850         transform(notes.begin(), notes.end(), back_inserter(_added_notes),
851                         sigc::mem_fun(*this, &DeltaCommand::unmarshal_note));
852
853         _removed_notes.clear();
854         XMLNode *removed_notes = delta_command.child(REMOVED_NOTES_ELEMENT);
855         notes = removed_notes->children();
856         transform(notes.begin(), notes.end(), back_inserter(_removed_notes),
857                         sigc::mem_fun(*this, &DeltaCommand::unmarshal_note));
858
859         return 0;
860 }
861
862 XMLNode& MidiModel::DeltaCommand::get_state()
863 {
864         XMLNode *delta_command = new XMLNode(DELTA_COMMAND_ELEMENT);
865         delta_command->add_property("midi_source", _model->midi_source()->id().to_s());
866
867         XMLNode *added_notes = delta_command->add_child(ADDED_NOTES_ELEMENT);
868         for_each(_added_notes.begin(), _added_notes.end(), sigc::compose(
869                         sigc::mem_fun(*added_notes, &XMLNode::add_child_nocopy),
870                         sigc::mem_fun(*this, &DeltaCommand::marshal_note)));
871
872         XMLNode *removed_notes = delta_command->add_child(REMOVED_NOTES_ELEMENT);
873         for_each(_removed_notes.begin(), _removed_notes.end(), sigc::compose(
874                         sigc::mem_fun(*removed_notes, &XMLNode::add_child_nocopy),
875                         sigc::mem_fun(*this, &DeltaCommand::marshal_note)));
876
877         return *delta_command;
878 }
879
880 struct EventTimeComparator {
881         typedef const MIDI::Event* value_type;
882         inline bool operator()(const MIDI::Event& a, const MIDI::Event& b) const {
883                 return a.time() >= b.time();
884         }
885 };
886
887 /** Write the model to a MidiSource (i.e. save the model).
888  * This is different from manually using read to write to a source in that
889  * note off events are written regardless of the track mode.  This is so the
890  * user can switch a recorded track (with note durations from some instrument)
891  * to percussive, save, reload, then switch it back to sustained without
892  * destroying the original note durations.
893  */
894 bool MidiModel::write_to(boost::shared_ptr<MidiSource> source)
895 {
896         read_lock();
897
898         const NoteMode old_note_mode = _note_mode;
899         _note_mode = Sustained;
900         
901         for (const_iterator i = begin(); i != end(); ++i) {
902                 source->append_event_unlocked(Frames, *i);
903         }
904                 
905         _note_mode = old_note_mode;
906         
907         read_unlock();
908         _edited = false;
909
910         return true;
911 }
912
913 XMLNode& MidiModel::get_state()
914 {
915         XMLNode *node = new XMLNode("MidiModel");
916         return *node;
917 }
918