*** NEW CODING POLICY ***
[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, Source::Flag flags)
53         : Source (s, DataType::MIDI, name, flags)
54         , _read_data_count(0)
55         , _write_data_count(0)
56         , _writing (false)
57         , _length_beats(0.0)
58         , _last_read_end(0)
59         , _last_write_end(0)
60 {
61 }
62
63 MidiSource::MidiSource (Session& s, const XMLNode& node) 
64         : Source (s, node)
65         , _read_data_count(0)
66         , _write_data_count(0)
67         , _writing (false)
68         , _length_beats(0.0)
69         , _last_read_end(0)
70         , _last_write_end(0)
71 {
72         _read_data_count = 0;
73         _write_data_count = 0;
74
75         if (set_state (node)) {
76                 throw failed_constructor();
77         }
78 }
79
80 MidiSource::~MidiSource ()
81 {
82 }
83
84 XMLNode&
85 MidiSource::get_state ()
86 {
87         XMLNode& node (Source::get_state());
88
89         if (_captured_for.length()) {
90                 node.add_property ("captured-for", _captured_for);
91         }
92
93         return node;
94 }
95
96 int
97 MidiSource::set_state (const XMLNode& node)
98 {
99         const XMLProperty* prop;
100
101         if ((prop = node.property ("captured-for")) != 0) {
102                 _captured_for = prop->value();
103         }
104
105         return 0;
106 }
107
108 sframes_t
109 MidiSource::length (sframes_t pos) const
110 {
111         BeatsFramesConverter converter(_session, pos);
112         return converter.to(_length_beats);
113 }
114
115 void
116 MidiSource::update_length (sframes_t pos, sframes_t cnt)
117 {
118         // You're not the boss of me!
119 }
120
121 void
122 MidiSource::invalidate ()
123 {
124         _model_iter.invalidate();
125 }
126
127 nframes_t
128 MidiSource::midi_read (MidiRingBuffer<nframes_t>& dst, sframes_t position,
129                 sframes_t start, nframes_t cnt,
130                 sframes_t stamp_offset, sframes_t negative_stamp_offset) const
131 {
132         Glib::Mutex::Lock lm (_lock);
133
134         BeatsFramesConverter converter(_session, position);
135
136         if (_model) {
137 #define BEATS_TO_FRAMES(t) (converter.to(t) + stamp_offset - negative_stamp_offset)
138
139                 Evoral::Sequence<double>::const_iterator& i = _model_iter;
140                 
141                 if (_last_read_end == 0 || start != _last_read_end || !i.valid()) {
142                         cerr << "MidiSource::midi_read seeking to frame " << start << endl;
143                         for (i = _model->begin(); i != _model->end(); ++i) {
144                                 if (BEATS_TO_FRAMES(i->time()) >= start) {
145                                         break;
146                                 }
147                         }
148                 }
149                 
150                 _last_read_end = start + cnt;
151
152                 for (; i != _model->end(); ++i) {
153                         const nframes_t time_frames = BEATS_TO_FRAMES(i->time());
154                         if (time_frames < start + cnt) {
155                                 dst.write(time_frames, i->event_type(), i->size(), i->buffer());
156                         } else {
157                                 break;
158                         }
159                 }
160                 return cnt;
161         } else {
162                 return read_unlocked (dst, position, start, cnt, stamp_offset, negative_stamp_offset);
163         }
164 }
165
166 nframes_t
167 MidiSource::midi_write (MidiRingBuffer<nframes_t>& dst, sframes_t position, nframes_t cnt)
168 {
169         Glib::Mutex::Lock lm (_lock);
170         const nframes_t ret = write_unlocked (dst, position, cnt);
171         _last_write_end = position + cnt;
172         return ret;
173 }
174
175 bool
176 MidiSource::file_changed (string path)
177 {
178         struct stat stat_file;
179
180         int e1 = stat (path.c_str(), &stat_file);
181         
182         return !e1;
183 }
184
185 void
186 MidiSource::mark_streaming_midi_write_started (NoteMode mode, sframes_t start_frame)
187 {
188         set_timeline_position(start_frame);
189
190         if (_model) {
191                 _model->set_note_mode(mode);
192                 _model->start_write();
193         }
194
195         _last_write_end = start_frame;
196         _writing = true;
197 }
198
199 void
200 MidiSource::mark_streaming_write_started ()
201 {
202         sframes_t start_frame = _session.transport_frame();
203
204         if (_model) {
205                 _model->start_write();
206         }
207
208         _last_write_end = start_frame;
209         _writing = true;
210 }
211
212 void
213 MidiSource::mark_streaming_write_completed ()
214 {
215         if (_model) {
216                 _model->end_write(false);
217         }
218
219         _writing = false;
220 }
221
222 void
223 MidiSource::session_saved()
224 {
225         flush_midi();
226
227         if (_model && _model->edited()) {
228                 string newname;
229                 const string basename = PBD::basename_nosuffix(_name);
230                 string::size_type last_dash = basename.find_last_of("-");
231                 if (last_dash == string::npos || last_dash == basename.find_first_of("-")) {
232                         newname = basename + "-1";
233                 } else {
234                         stringstream ss(basename.substr(last_dash+1));
235                         unsigned write_count = 0;
236                         ss >> write_count;
237                         cerr << "WRITE COUNT: " << write_count << endl;
238                         ++write_count; // start at 1
239                         ss.clear();
240                         ss << basename.substr(0, last_dash) << "-" << write_count;
241                         newname = ss.str();
242                 }
243
244                 string newpath = _session.session_directory().midi_path().to_string() +"/"+ newname + ".mid";
245
246                 boost::shared_ptr<MidiSource> newsrc = boost::dynamic_pointer_cast<MidiSource>(
247                                 SourceFactory::createWritable(DataType::MIDI, _session,
248                                                 newpath, true, false, _session.frame_rate()));
249
250                 newsrc->set_timeline_position(_timeline_position);
251                 _model->write_to(newsrc);
252
253                 // cyclic dependency here, ugly :(
254                 newsrc->set_model(_model);
255                 _model->set_midi_source(newsrc.get());
256                 
257                 newsrc->flush_midi();
258
259                 Switched.emit(newsrc);
260         }
261 }
262
263 void
264 MidiSource::set_note_mode(NoteMode mode)
265 {
266         if (_model) {
267                 _model->set_note_mode(mode);
268         }
269 }
270