70abb99e567f3dd6f72b312aed7f97d8189871be
[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 <pbd/xml++.h>
32 #include <pbd/pthread_utils.h>
33 #include <pbd/basename.h>
34
35 #include <ardour/audioengine.h>
36 #include <ardour/midi_model.h>
37 #include <ardour/midi_ring_buffer.h>
38 #include <ardour/midi_source.h>
39 #include <ardour/session.h>
40 #include <ardour/session_directory.h>
41 #include <ardour/source_factory.h>
42 #include <ardour/tempo.h>
43
44 #include "i18n.h"
45
46 using namespace std;
47 using namespace ARDOUR;
48 using namespace PBD;
49
50 sigc::signal<void,MidiSource *> MidiSource::MidiSourceCreated;
51
52 MidiSource::MidiSource (Session& s, string name)
53         : Source (s, name, DataType::MIDI)
54         , _timeline_position(0)
55         , _writing (false)
56         , _last_read_end(0)
57 {
58         _read_data_count = 0;
59         _write_data_count = 0;
60 }
61
62 MidiSource::MidiSource (Session& s, const XMLNode& node) 
63         : Source (s, node)
64         , _timeline_position(0)
65         , _writing (false)
66         , _last_read_end(0)
67 {
68         _read_data_count = 0;
69         _write_data_count = 0;
70
71         if (set_state (node)) {
72                 throw failed_constructor();
73         }
74 }
75
76 MidiSource::~MidiSource ()
77 {
78 }
79
80 XMLNode&
81 MidiSource::get_state ()
82 {
83         XMLNode& node (Source::get_state());
84
85         if (_captured_for.length()) {
86                 node.add_property ("captured-for", _captured_for);
87         }
88
89         return node;
90 }
91
92 int
93 MidiSource::set_state (const XMLNode& node)
94 {
95         const XMLProperty* prop;
96
97         Source::set_state (node);
98
99         if ((prop = node.property ("captured-for")) != 0) {
100                 _captured_for = prop->value();
101         }
102
103         return 0;
104 }
105
106 nframes_t
107 MidiSource::midi_read (MidiRingBuffer<nframes_t>& dst, nframes_t start, nframes_t cnt,
108                 nframes_t stamp_offset, nframes_t negative_stamp_offset) const
109 {
110
111         Glib::Mutex::Lock lm (_lock);
112         if (_model) {
113                 // FIXME: assumes tempo never changes after start
114                 const Tempo& tempo = _session.tempo_map().tempo_at(_timeline_position);
115                 const double frames_per_beat = tempo.frames_per_beat(
116                                 _session.engine().frame_rate(),
117                                 _session.tempo_map().meter_at(_timeline_position));
118 #define BEATS_TO_FRAMES(t) (((t) * frames_per_beat) + stamp_offset - negative_stamp_offset)
119
120                 Evoral::Sequence<double>::const_iterator& i = _model_iter;
121                 
122                 if (_last_read_end == 0 || start != _last_read_end) {
123                         cerr << "MidiSource::midi_read seeking to frame " << start << endl;
124                         for (i = _model->begin(); i != _model->end(); ++i) {
125                                 if (BEATS_TO_FRAMES(i->time()) >= start) {
126                                         break;
127                                 }
128                         }
129                 }
130                 
131                 _last_read_end = start + cnt;
132
133                 for (; i != _model->end(); ++i) {
134                         const nframes_t time_frames = BEATS_TO_FRAMES(i->time());
135                         if (time_frames < start + cnt) {
136                                 dst.write(time_frames, i->event_type(), i->size(), i->buffer());
137                         } else {
138                                 break;
139                         }
140                 }
141                 return cnt;
142         } else {
143                 return read_unlocked (dst, start, cnt, stamp_offset, negative_stamp_offset);
144         }
145 }
146
147 nframes_t
148 MidiSource::midi_write (MidiRingBuffer<nframes_t>& dst, nframes_t cnt)
149 {
150         Glib::Mutex::Lock lm (_lock);
151         return write_unlocked (dst, cnt);
152 }
153
154 bool
155 MidiSource::file_changed (string path)
156 {
157         struct stat stat_file;
158
159         int e1 = stat (path.c_str(), &stat_file);
160         
161         return !e1;
162 }
163
164 void
165 MidiSource::mark_streaming_midi_write_started (NoteMode mode, nframes_t start_frame)
166 {
167         set_timeline_position(start_frame); // why do I have a feeling this can break somehow...
168
169         if (_model) {
170                 _model->set_note_mode(mode);
171                 _model->start_write();
172         }
173         
174         _writing = true;
175 }
176
177 void
178 MidiSource::mark_streaming_write_started ()
179 {
180         if (_model) {
181                 _model->start_write();
182         }
183
184         _writing = true;
185 }
186
187 void
188 MidiSource::mark_streaming_write_completed ()
189 {
190         if (_model) {
191                 _model->end_write(false); // FIXME: param?
192         }
193
194         _writing = false;
195 }
196
197 void
198 MidiSource::session_saved()
199 {
200         flush_midi();
201
202         if (_model && _model->edited()) {
203                 string newname;
204                 const string basename = PBD::basename_nosuffix(_name);
205                 string::size_type last_dash = basename.find_last_of("-");
206                 if (last_dash == string::npos || last_dash == basename.find_first_of("-")) {
207                         newname = basename + "-1";
208                 } else {
209                         stringstream ss(basename.substr(last_dash+1));
210                         unsigned write_count = 0;
211                         ss >> write_count;
212                         cerr << "WRITE COUNT: " << write_count << endl;
213                         ++write_count; // start at 1
214                         ss.clear();
215                         ss << basename.substr(0, last_dash) << "-" << write_count;
216                         newname = ss.str();
217                 }
218
219                 string newpath = _session.session_directory().midi_path().to_string() +"/"+ newname + ".mid";
220
221                 boost::shared_ptr<MidiSource> newsrc = boost::dynamic_pointer_cast<MidiSource>(
222                                 SourceFactory::createWritable(DataType::MIDI, _session, newpath, 1, 0, true));
223
224                 newsrc->set_timeline_position(_timeline_position);
225                 _model->write_to(newsrc);
226
227                 // cyclic dependency here, ugly :(
228                 newsrc->set_model(_model);
229                 _model->set_midi_source(newsrc.get());
230                 
231                 newsrc->flush_midi();
232
233                 Switched.emit(newsrc);
234         }
235 }
236
237 void
238 MidiSource::set_note_mode(NoteMode mode)
239 {
240         if (_model) {
241                 _model->set_note_mode(mode);
242         }
243 }
244