dramatic change in logic and naming for operations related to adding a MIDI region...
[ardour.git] / libs / ardour / midi_source.cc
1 /*
2     Copyright (C) 2006 Paul Davis
3     Author: David Robillard
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 #include <sys/stat.h>
21 #include <unistd.h>
22 #include <fcntl.h>
23 #include <poll.h>
24 #include <float.h>
25 #include <cerrno>
26 #include <ctime>
27 #include <cmath>
28 #include <iomanip>
29 #include <algorithm>
30
31 #include <glibmm/fileutils.h>
32 #include <glibmm/miscutils.h>
33
34 #include "pbd/xml++.h"
35 #include "pbd/pthread_utils.h"
36 #include "pbd/basename.h"
37
38 #include "ardour/debug.h"
39 #include "ardour/midi_model.h"
40 #include "ardour/midi_state_tracker.h"
41 #include "ardour/midi_source.h"
42 #include "ardour/file_source.h"
43 #include "ardour/session.h"
44 #include "ardour/session_directory.h"
45 #include "ardour/source_factory.h"
46
47 #include "i18n.h"
48
49 namespace ARDOUR { template <typename T> class MidiRingBuffer; }
50
51 using namespace std;
52 using namespace ARDOUR;
53 using namespace PBD;
54
55 PBD::Signal1<void,MidiSource*> MidiSource::MidiSourceCreated;
56
57 MidiSource::MidiSource (Session& s, string name, Source::Flag flags)
58         : Source(s, DataType::MIDI, name, flags)
59         , _writing(false)
60         , _model_iter_valid(false)
61         , _length_beats(0.0)
62         , _last_read_end(0)
63         , _capture_length(0)
64         , _capture_loop_length(0)
65 {
66 }
67
68 MidiSource::MidiSource (Session& s, const XMLNode& node)
69         : Source(s, node)
70         , _writing(false)
71         , _model_iter_valid(false)
72         , _length_beats(0.0)
73         , _last_read_end(0)
74         , _capture_length(0)
75         , _capture_loop_length(0)
76 {
77         if (set_state (node, Stateful::loading_state_version)) {
78                 throw failed_constructor();
79         }
80 }
81
82
83 MidiSource::~MidiSource ()
84 {
85 }
86
87 XMLNode&
88 MidiSource::get_state ()
89 {
90         XMLNode& node (Source::get_state());
91
92         if (_captured_for.length()) {
93                 node.add_property ("captured-for", _captured_for);
94         }
95
96         for (InterpolationStyleMap::const_iterator i = _interpolation_style.begin(); i != _interpolation_style.end(); ++i) {
97                 XMLNode* child = node.add_child (X_("InterpolationStyle"));
98                 child->add_property (X_("parameter"), EventTypeMap::instance().to_symbol (i->first));
99                 child->add_property (X_("style"), enum_2_string (i->second));
100         }
101
102         for (AutomationStateMap::const_iterator i = _automation_state.begin(); i != _automation_state.end(); ++i) {
103                 XMLNode* child = node.add_child (X_("AutomationState"));
104                 child->add_property (X_("parameter"), EventTypeMap::instance().to_symbol (i->first));
105                 child->add_property (X_("state"), enum_2_string (i->second));
106         }
107
108         return node;
109 }
110
111 int
112 MidiSource::set_state (const XMLNode& node, int /*version*/)
113 {
114         const XMLProperty* prop;
115
116         if ((prop = node.property ("captured-for")) != 0) {
117                 _captured_for = prop->value();
118         }
119
120         XMLNodeList children = node.children ();
121         for (XMLNodeConstIterator i = children.begin(); i != children.end(); ++i) {
122                 if ((*i)->name() == X_("InterpolationStyle")) {
123                         XMLProperty* prop;
124
125                         if ((prop = (*i)->property (X_("parameter"))) == 0) {
126                                 error << _("Missing parameter property on InterpolationStyle") << endmsg;
127                                 return -1;
128                         }
129
130                         Evoral::Parameter p = EventTypeMap::instance().new_parameter (prop->value());
131
132                         if ((prop = (*i)->property (X_("style"))) == 0) {
133                                 error << _("Missing style property on InterpolationStyle") << endmsg;
134                                 return -1;
135                         }
136
137                         Evoral::ControlList::InterpolationStyle s = static_cast<Evoral::ControlList::InterpolationStyle> (string_2_enum (prop->value(), s));
138                         set_interpolation_of (p, s);
139
140                 } else if ((*i)->name() == X_("AutomationState")) {
141
142                         XMLProperty* prop;
143
144                         if ((prop = (*i)->property (X_("parameter"))) == 0) {
145                                 error << _("Missing parameter property on AutomationState") << endmsg;
146                                 return -1;
147                         }
148
149                         Evoral::Parameter p = EventTypeMap::instance().new_parameter (prop->value());
150
151                         if ((prop = (*i)->property (X_("state"))) == 0) {
152                                 error << _("Missing state property on AutomationState") << endmsg;
153                                 return -1;
154                         }
155
156                         AutoState s = static_cast<AutoState> (string_2_enum (prop->value(), s));
157                         set_automation_state_of (p, s);
158                 }
159         }
160
161         return 0;
162 }
163
164 bool
165 MidiSource::empty () const
166 {
167         return _length_beats == 0;
168 }
169
170 framecnt_t
171 MidiSource::length (framepos_t pos) const
172 {
173         if (_length_beats == 0) {
174                 return 0;
175         }
176
177         BeatsFramesConverter converter(_session.tempo_map(), pos);
178         return converter.to(_length_beats);
179 }
180
181 void
182 MidiSource::update_length (framecnt_t)
183 {
184         // You're not the boss of me!
185 }
186
187 void
188 MidiSource::invalidate ()
189 {
190         _model_iter_valid = false;
191         _model_iter.invalidate();
192 }
193
194 /** @param filtered A set of parameters whose MIDI messages will not be returned */
195 framecnt_t
196 MidiSource::midi_read (Evoral::EventSink<framepos_t>& dst, framepos_t source_start,
197                        framepos_t start, framecnt_t cnt,
198                        MidiStateTracker* tracker,
199                        std::set<Evoral::Parameter> const & filtered) const
200 {
201         Glib::Threads::Mutex::Lock lm (_lock);
202
203         BeatsFramesConverter converter(_session.tempo_map(), source_start);
204
205         DEBUG_TRACE (DEBUG::MidiSourceIO, string_compose ("MidiSource::midi-read() %5 sstart %1 start %2 cnt %3 tracker %4\n",
206                                                           source_start, start, cnt, tracker, name()));
207
208         if (_model) {
209                 Evoral::Sequence<double>::const_iterator& i = _model_iter;
210
211                 // If the cached iterator is invalid, search for the first event past start
212                 if (_last_read_end == 0 || start != _last_read_end || !_model_iter_valid) {
213                         DEBUG_TRACE (DEBUG::MidiSourceIO, string_compose ("*** %1 search for relevant iterator for %1 / %2\n", _name, source_start, start));
214                         for (i = _model->begin(0, false, filtered); i != _model->end(); ++i) {
215                                 if (converter.to(i->time()) >= start) {
216                                         DEBUG_TRACE (DEBUG::MidiSourceIO, string_compose ("***\tstop iterator search @ %1\n", i->time()));
217                                         break;
218                                 }
219                         }
220                         _model_iter_valid = true;
221                 } else {
222                         DEBUG_TRACE (DEBUG::MidiSourceIO, string_compose ("*** %1 use cachediterator for %1 / %2\n", _name, source_start, start));
223                 }
224
225                 _last_read_end = start + cnt;
226
227                 // Read events up to end
228                 for (; i != _model->end(); ++i) {
229                         const framecnt_t time_frames = converter.to(i->time());
230                         if (time_frames < start + cnt) {
231                                 /* convert event times to session frames by adding on the source start position in session frames */
232                                 dst.write (time_frames + source_start, i->event_type(), i->size(), i->buffer());
233
234                                 DEBUG_TRACE (DEBUG::MidiSourceIO, string_compose ("%1: add event @ %2 type %3 size = %4\n",
235                                                                                   _name, time_frames + source_start, i->event_type(), i->size()));
236
237                                 if (tracker) {
238                                         Evoral::MIDIEvent<Evoral::MusicalTime>& ev (*(reinterpret_cast<Evoral::MIDIEvent<Evoral::MusicalTime>*> 
239                                                                                       (const_cast<Evoral::Event<Evoral::MusicalTime>*> (&(*i)))));
240                                         if (ev.is_note_on()) {
241                                                 DEBUG_TRACE (DEBUG::MidiSourceIO, string_compose ("\t%1 track note on %2 @ %3 velocity %4\n", _name, (int) ev.note(), time_frames, (int) ev.velocity()));
242                                                 tracker->add (ev.note(), ev.channel());
243                                         } else if (ev.is_note_off()) {
244                                                 DEBUG_TRACE (DEBUG::MidiSourceIO, string_compose ("\t%1 track note off %2 @ %3\n", _name, (int) ev.note(), time_frames));
245                                                 tracker->remove (ev.note(), ev.channel());
246                                         }
247                                 }
248                         } else {
249                                 DEBUG_TRACE (DEBUG::MidiSourceIO, string_compose ("%1: reached end with event @ %2 vs. %3\n",
250                                                                                   _name, time_frames, start+cnt));
251                                 break;
252                         }
253                 }
254                 return cnt;
255         } else {
256                 return read_unlocked (dst, source_start, start, cnt, tracker);
257         }
258 }
259
260 framecnt_t
261 MidiSource::midi_write (MidiRingBuffer<framepos_t>& source,
262                         framepos_t                  source_start,
263                         framecnt_t                  cnt)
264 {
265         Glib::Threads::Mutex::Lock lm (_lock);
266
267         const framecnt_t ret = write_unlocked (source, source_start, cnt);
268
269         if (cnt == max_framecnt) {
270                 _last_read_end = 0;
271         } else {
272                 _capture_length += cnt;
273         }
274
275         return ret;
276 }
277
278 void
279 MidiSource::mark_streaming_midi_write_started (NoteMode mode)
280 {
281         if (_model) {
282                 _model->set_note_mode (mode);
283                 _model->start_write ();
284         }
285
286         _writing = true;
287 }
288
289 void
290 MidiSource::mark_write_starting_now (framecnt_t position,
291                                      framecnt_t capture_length,
292                                      framecnt_t loop_length)
293 {
294         /* I'm not sure if this is the best way to approach this, but
295            _capture_length needs to be set up with the transport frame
296            when a record actually starts, as it is used by
297            SMFSource::write_unlocked to decide whether incoming notes
298            are within the correct time range.
299            mark_streaming_midi_write_started (perhaps a more logical
300            place to do this) is not called at exactly the time when
301            record starts, and I don't think it necessarily can be
302            because it is not RT-safe.
303         */
304
305         set_timeline_position(position);
306         _capture_length      = capture_length;
307         _capture_loop_length = loop_length;
308
309         BeatsFramesConverter converter(_session.tempo_map(), position);
310         _length_beats = converter.from(capture_length);
311 }
312
313 void
314 MidiSource::mark_streaming_write_started ()
315 {
316         NoteMode note_mode = _model ? _model->note_mode() : Sustained;
317         mark_streaming_midi_write_started (note_mode);
318 }
319
320 void
321 MidiSource::mark_midi_streaming_write_completed (Evoral::Sequence<Evoral::MusicalTime>::StuckNoteOption option, Evoral::MusicalTime end)
322 {
323         if (_model) {
324                 _model->end_write (option, end);
325         }
326
327         _writing = false;
328 }
329
330 void
331 MidiSource::mark_streaming_write_completed ()
332 {
333         mark_midi_streaming_write_completed (Evoral::Sequence<Evoral::MusicalTime>::DeleteStuckNotes);
334 }
335
336 int
337 MidiSource::write_to (boost::shared_ptr<MidiSource> newsrc, Evoral::MusicalTime begin, Evoral::MusicalTime end)
338 {
339         newsrc->set_timeline_position(_timeline_position);
340         newsrc->copy_interpolation_from (this);
341         newsrc->copy_automation_state_from (this);
342
343         if (_model) {
344                 if (begin == Evoral::MinMusicalTime && end == Evoral::MaxMusicalTime) {
345                         _model->write_to (newsrc);
346                 } else {
347                         _model->write_section_to (newsrc, begin, end);
348                 }
349         } else {
350                 error << string_compose (_("programming error: %1"), X_("no model for MidiSource during ::clone()"));
351                 return -1;
352         }
353
354         newsrc->flush_midi();
355
356         /* force a reload of the model if the range is partial */
357
358         if (begin != Evoral::MinMusicalTime || end != Evoral::MaxMusicalTime) {
359                 newsrc->load_model (true, true);
360         } else {
361                 newsrc->set_model (_model);
362         }
363         
364         /* this file is not removable (but since it is MIDI, it is mutable) */
365
366         boost::dynamic_pointer_cast<FileSource> (newsrc)->prevent_deletion ();
367
368         return 0;
369 }
370
371 void
372 MidiSource::session_saved()
373 {
374         /* this writes a copy of the data to disk.
375            XXX do we need to do this every time?
376         */
377
378         if (_model && _model->edited()) {
379                 
380                 // if the model is edited, write its contents into
381                 // the current source file (overwiting previous contents.
382
383                 /* temporarily drop our reference to the model so that
384                    as the model pushes its current state to us, we don't
385                    try to update it.
386                 */
387
388                 boost::shared_ptr<MidiModel> mm = _model;
389                 _model.reset ();
390
391                 /* flush model contents to disk
392                  */
393
394                 mm->sync_to_source ();
395
396                 /* reacquire model */
397
398                 _model = mm;
399
400         } else {
401                 flush_midi();
402         }
403 }
404
405 void
406 MidiSource::set_note_mode(NoteMode mode)
407 {
408         if (_model) {
409                 _model->set_note_mode(mode);
410         }
411 }
412
413 void
414 MidiSource::drop_model ()
415 {
416         _model.reset();
417         ModelChanged (); /* EMIT SIGNAL */
418 }
419
420 void
421 MidiSource::set_model (boost::shared_ptr<MidiModel> m)
422 {
423         _model = m;
424         ModelChanged (); /* EMIT SIGNAL */
425 }
426
427 /** @return Interpolation style that should be used for control parameter \a p */
428 Evoral::ControlList::InterpolationStyle
429 MidiSource::interpolation_of (Evoral::Parameter p) const
430 {
431         InterpolationStyleMap::const_iterator i = _interpolation_style.find (p);
432         if (i == _interpolation_style.end()) {
433                 return EventTypeMap::instance().interpolation_of (p);
434         }
435
436         return i->second;
437 }
438
439 AutoState
440 MidiSource::automation_state_of (Evoral::Parameter p) const
441 {
442         AutomationStateMap::const_iterator i = _automation_state.find (p);
443         if (i == _automation_state.end()) {
444                 /* default to `play', otherwise if MIDI is recorded /
445                    imported with controllers etc. they are by default
446                    not played back, which is a little surprising.
447                 */
448                 return Play;
449         }
450
451         return i->second;
452 }
453
454 /** Set interpolation style to be used for a given parameter.  This change will be
455  *  propagated to anyone who needs to know.
456  */
457 void
458 MidiSource::set_interpolation_of (Evoral::Parameter p, Evoral::ControlList::InterpolationStyle s)
459 {
460         if (interpolation_of (p) == s) {
461                 return;
462         }
463
464         if (EventTypeMap::instance().interpolation_of (p) == s) {
465                 /* interpolation type is being set to the default, so we don't need a note in our map */
466                 _interpolation_style.erase (p);
467         } else {
468                 _interpolation_style[p] = s;
469         }
470
471         InterpolationChanged (p, s); /* EMIT SIGNAL */
472 }
473
474 void
475 MidiSource::set_automation_state_of (Evoral::Parameter p, AutoState s)
476 {
477         if (automation_state_of (p) == s) {
478                 return;
479         }
480
481         if (s == Play) {
482                 /* automation state is being set to the default, so we don't need a note in our map */
483                 _automation_state.erase (p);
484         } else {
485                 _automation_state[p] = s;
486         }
487
488         AutomationStateChanged (p, s); /* EMIT SIGNAL */
489 }
490
491 void
492 MidiSource::copy_interpolation_from (boost::shared_ptr<MidiSource> s)
493 {
494         copy_interpolation_from (s.get ());
495 }
496
497 void
498 MidiSource::copy_automation_state_from (boost::shared_ptr<MidiSource> s)
499 {
500         copy_automation_state_from (s.get ());
501 }
502
503 void
504 MidiSource::copy_interpolation_from (MidiSource* s)
505 {
506         _interpolation_style = s->_interpolation_style;
507
508         /* XXX: should probably emit signals here */
509 }
510
511 void
512 MidiSource::copy_automation_state_from (MidiSource* s)
513 {
514         _automation_state = s->_automation_state;
515
516         /* XXX: should probably emit signals here */
517 }