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