fe6c733a826857cf9fdb90fe462ab41a8f515e72
[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         return node;
107 }
108
109 int
110 MidiSource::set_state (const XMLNode& node, int /*version*/)
111 {
112         const XMLProperty* prop;
113
114         if ((prop = node.property ("captured-for")) != 0) {
115                 _captured_for = prop->value();
116         }
117
118         XMLNodeList children = node.children ();
119         for (XMLNodeConstIterator i = children.begin(); i != children.end(); ++i) {
120                 if ((*i)->name() == X_("InterpolationStyle")) {
121                         XMLProperty* prop;
122
123                         if ((prop = (*i)->property (X_("parameter"))) == 0) {
124                                 error << _("Missing parameter property on InterpolationStyle") << endmsg;
125                                 return -1;
126                         }
127                         
128                         Evoral::Parameter p = EventTypeMap::instance().new_parameter (prop->value());
129
130                         if ((prop = (*i)->property (X_("style"))) == 0) {
131                                 error << _("Missing style property on InterpolationStyle") << endmsg;
132                                 return -1;
133                         }
134
135                         Evoral::ControlList::InterpolationStyle s = static_cast<Evoral::ControlList::InterpolationStyle> (string_2_enum (prop->value(), s));
136                         set_interpolation_of (p, s);
137                 }
138         }
139
140         return 0;
141 }
142
143 bool
144 MidiSource::empty () const
145 {
146         return _length_beats == 0;
147 }
148
149 framecnt_t
150 MidiSource::length (framepos_t pos) const
151 {
152         if (_length_beats == 0) {
153                 return 0;
154         }
155
156         BeatsFramesConverter converter(_session.tempo_map(), pos);
157         return converter.to(_length_beats);
158 }
159
160 void
161 MidiSource::update_length (sframes_t /*pos*/, sframes_t /*cnt*/)
162 {
163         // You're not the boss of me!
164 }
165
166 void
167 MidiSource::invalidate ()
168 {
169         _model_iter_valid = false;
170         _model_iter.invalidate();
171 }
172
173 /** @param filtered A set of parameters whose MIDI messages will not be returned */
174 nframes_t
175 MidiSource::midi_read (Evoral::EventSink<nframes_t>& dst, sframes_t source_start,
176                        sframes_t start, nframes_t cnt,
177                        sframes_t stamp_offset, sframes_t negative_stamp_offset,
178                        MidiStateTracker* tracker,
179                        std::set<Evoral::Parameter> const & filtered) const
180 {
181         Glib::Mutex::Lock lm (_lock);
182
183         BeatsFramesConverter converter(_session.tempo_map(), source_start);
184
185         if (_model) {
186                 Evoral::Sequence<double>::const_iterator& i = _model_iter;
187
188                 // If the cached iterator is invalid, search for the first event past start
189                 if (_last_read_end == 0 || start != _last_read_end || !_model_iter_valid) {
190                         DEBUG_TRACE (DEBUG::MidiSourceIO, string_compose ("*** %1 search for relevant iterator for %1 / %2\n", _name, source_start, start));
191                         for (i = _model->begin(0, false, filtered); i != _model->end(); ++i) {
192                                 if (converter.to(i->time()) >= start) {
193                                         break;
194                                 }
195                         }
196                         _model_iter_valid = true;
197                 } else {
198                         DEBUG_TRACE (DEBUG::MidiSourceIO, string_compose ("*** %1 use cached iterator for %1 / %2\n", _name, source_start, start));
199                 }
200
201                 _last_read_end = start + cnt;
202
203                 // Read events up to end
204                 for (; i != _model->end(); ++i) {
205                         const sframes_t time_frames = converter.to(i->time());
206                         if (time_frames < start + cnt) {
207                                 dst.write(time_frames + stamp_offset - negative_stamp_offset,
208                                           i->event_type(), i->size(), i->buffer());
209                                 if (tracker) {
210                                         Evoral::MIDIEvent<Evoral::MusicalTime>& ev (*(Evoral::MIDIEvent<Evoral::MusicalTime>*) (&(*i)));
211                                         if (ev.is_note_on()) {
212                                                 DEBUG_TRACE (DEBUG::MidiSourceIO, string_compose ("\t%1 add note on %2 @ %3\n", _name, ev.note(), time_frames));
213                                                 tracker->add (ev.note(), ev.channel());
214                                         } else if (ev.is_note_off()) {
215                                                 DEBUG_TRACE (DEBUG::MidiSourceIO, string_compose ("\t%1 add note off %2 @ %3\n", _name, ev.note(), time_frames));
216                                                 tracker->remove (ev.note(), ev.channel());
217                                         }
218                                 }
219                         } else {
220                                 break;
221                         }
222                 }
223                 return cnt;
224         } else {
225                 return read_unlocked (dst, source_start, start, cnt, stamp_offset, negative_stamp_offset, tracker);
226         }
227 }
228
229 nframes_t
230 MidiSource::midi_write (MidiRingBuffer<nframes_t>& source, sframes_t source_start, nframes_t duration)
231 {
232         Glib::Mutex::Lock lm (_lock);
233         const nframes_t ret = write_unlocked (source, source_start, duration);
234         _last_write_end += duration;
235         return ret;
236 }
237
238 void
239 MidiSource::mark_streaming_midi_write_started (NoteMode mode, sframes_t start_frame)
240 {
241         set_timeline_position(start_frame);
242
243         if (_model) {
244                 _model->set_note_mode(mode);
245                 _model->start_write();
246         }
247
248         _last_write_end = start_frame;
249         _writing = true;
250 }
251
252 void
253 MidiSource::mark_streaming_write_started ()
254 {
255         NoteMode note_mode = _model ? _model->note_mode() : Sustained;
256         mark_streaming_midi_write_started(note_mode, _session.transport_frame());
257 }
258
259 void
260 MidiSource::mark_streaming_write_completed ()
261 {
262         if (_model) {
263                 _model->end_write(false);
264         }
265
266         _writing = false;
267 }
268
269 boost::shared_ptr<MidiSource>
270 MidiSource::clone (Evoral::MusicalTime begin, Evoral::MusicalTime end)
271 {
272         string newname = PBD::basename_nosuffix(_name.val());
273         string newpath;
274
275         /* get a new name for the MIDI file we're going to write to
276          */
277
278         do { 
279
280                 newname = bump_name_once (newname, '-');
281                 /* XXX build path safely */
282                 newpath = _session.session_directory().midi_path().to_string() +"/"+ newname + ".mid";
283
284         } while (Glib::file_test (newpath, Glib::FILE_TEST_EXISTS));
285         
286         boost::shared_ptr<MidiSource> newsrc = boost::dynamic_pointer_cast<MidiSource>(
287                 SourceFactory::createWritable(DataType::MIDI, _session,
288                                               newpath, false, _session.frame_rate()));
289         
290         newsrc->set_timeline_position(_timeline_position);
291         newsrc->copy_interpolation_from (this);
292
293         if (_model) {
294                 if (begin == Evoral::MinMusicalTime && end == Evoral::MaxMusicalTime) {
295                         _model->write_to (newsrc);
296                 } else {
297                         _model->write_section_to (newsrc, begin, end);
298                 }
299         } else {
300                 error << string_compose (_("programming error: %1"), X_("no model for MidiSource during ::clone()"));
301                 return boost::shared_ptr<MidiSource>();
302         }
303
304         newsrc->flush_midi();
305
306         /* force a reload of the model if the range is partial */
307         
308         if (begin != Evoral::MinMusicalTime || end != Evoral::MaxMusicalTime) {
309                 newsrc->load_model (true, true);
310         } else {
311                 newsrc->set_model (_model);
312         }
313         
314         return newsrc;
315 }
316
317 void
318 MidiSource::session_saved()
319 {
320         /* this writes a copy of the data to disk. 
321            XXX do we need to do this every time?
322         */
323
324         if (_model && _model->edited()) {
325
326
327
328                 // if the model is edited, write its contents into
329                 // the current source file (overwiting previous contents.
330
331                 /* temporarily drop our reference to the model so that
332                    as the model pushes its current state to us, we don't
333                    try to update it.
334                 */
335
336                 boost::shared_ptr<MidiModel> mm = _model ;
337                 _model.reset ();   
338
339                 /* flush model contents to disk
340                  */
341
342                 mm->sync_to_source ();
343
344                 /* reacquire model */
345
346                 _model = mm;
347
348         } else {
349                 flush_midi();
350         }
351 }
352
353 void
354 MidiSource::set_note_mode(NoteMode mode)
355 {
356         if (_model) {
357                 _model->set_note_mode(mode);
358         }
359 }
360
361 void
362 MidiSource::drop_model ()
363 {
364         _model.reset(); 
365         ModelChanged (); /* EMIT SIGNAL */
366 }
367
368 void
369 MidiSource::set_model (boost::shared_ptr<MidiModel> m)
370 {
371         _model = m;
372         ModelChanged (); /* EMIT SIGNAL */
373 }
374
375 /** @return Interpolation style that should be used for control parameter \a p */
376 Evoral::ControlList::InterpolationStyle
377 MidiSource::interpolation_of (Evoral::Parameter p) const
378 {
379         InterpolationStyleMap::const_iterator i = _interpolation_style.find (p);
380         if (i == _interpolation_style.end()) {
381                 return EventTypeMap::instance().interpolation_of (p);
382         }
383
384         return i->second;
385 }
386
387 /** Set interpolation style to be used for a given parameter.  This change will be
388  *  propagated to anyone who needs to know.
389  */
390 void
391 MidiSource::set_interpolation_of (Evoral::Parameter p, Evoral::ControlList::InterpolationStyle s)
392 {
393         if (interpolation_of (p) == s) {
394                 return;
395         }
396         
397         if (EventTypeMap::instance().interpolation_of (p) == s) {
398                 /* interpolation type is being set to the default, so we don't need a note in our map */
399                 _interpolation_style.erase (p);
400         } else {
401                 _interpolation_style[p] = s;
402         }
403
404         InterpolationChanged (p, s); /* EMIT SIGNAL */
405 }
406
407 void
408 MidiSource::copy_interpolation_from (boost::shared_ptr<MidiSource> s)
409 {
410         copy_interpolation_from (s.get ());
411 }
412
413 void
414 MidiSource::copy_interpolation_from (MidiSource* s)
415 {
416         _interpolation_style = s->_interpolation_style;
417
418         /* XXX: should probably emit signals here */
419 }