Move duplicated AudioFileSource::Flags and SMFSource::Flags into Source.
[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, name, DataType::MIDI, flags)
54         , _timeline_position(0)
55         , _read_data_count(0)
56         , _write_data_count(0)
57         , _converter(s, _timeline_position)
58         , _writing (false)
59         , _last_read_end(0)
60 {
61 }
62
63 MidiSource::MidiSource (Session& s, const XMLNode& node) 
64         : Source (s, node)
65         , _timeline_position(0)
66         , _read_data_count(0)
67         , _write_data_count(0)
68         , _converter(s, _timeline_position)
69         , _writing (false)
70         , _last_read_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         Source::set_state (node);
102
103         if ((prop = node.property ("captured-for")) != 0) {
104                 _captured_for = prop->value();
105         }
106
107         return 0;
108 }
109
110 nframes_t
111 MidiSource::midi_read (MidiRingBuffer<nframes_t>& dst, nframes_t start, nframes_t cnt,
112                 nframes_t stamp_offset, nframes_t negative_stamp_offset) const
113 {
114
115         Glib::Mutex::Lock lm (_lock);
116         if (_model) {
117 #define BEATS_TO_FRAMES(t) (_converter.to(t) + stamp_offset - negative_stamp_offset)
118
119                 Evoral::Sequence<double>::const_iterator& i = _model_iter;
120                 
121                 if (_last_read_end == 0 || start != _last_read_end) {
122                         cerr << "MidiSource::midi_read seeking to frame " << start << endl;
123                         for (i = _model->begin(); i != _model->end(); ++i) {
124                                 if (BEATS_TO_FRAMES(i->time()) >= start) {
125                                         break;
126                                 }
127                         }
128                 }
129                 
130                 _last_read_end = start + cnt;
131
132                 for (; i != _model->end(); ++i) {
133                         const nframes_t time_frames = BEATS_TO_FRAMES(i->time());
134                         if (time_frames < start + cnt) {
135                                 dst.write(time_frames, i->event_type(), i->size(), i->buffer());
136                         } else {
137                                 break;
138                         }
139                 }
140                 return cnt;
141         } else {
142                 return read_unlocked (dst, start, cnt, stamp_offset, negative_stamp_offset);
143         }
144 }
145
146 nframes_t
147 MidiSource::midi_write (MidiRingBuffer<nframes_t>& dst, nframes_t cnt)
148 {
149         Glib::Mutex::Lock lm (_lock);
150         return write_unlocked (dst, cnt);
151 }
152
153 bool
154 MidiSource::file_changed (string path)
155 {
156         struct stat stat_file;
157
158         int e1 = stat (path.c_str(), &stat_file);
159         
160         return !e1;
161 }
162
163 void
164 MidiSource::set_timeline_position (nframes_t when)
165 {
166         _timeline_position = when;
167         _converter.set_origin(when);
168 }
169
170 void
171 MidiSource::mark_streaming_midi_write_started (NoteMode mode, nframes_t start_frame)
172 {
173         set_timeline_position(start_frame); // why do I have a feeling this can break somehow...
174
175         if (_model) {
176                 _model->set_note_mode(mode);
177                 _model->start_write();
178         }
179         
180         _writing = true;
181 }
182
183 void
184 MidiSource::mark_streaming_write_started ()
185 {
186         if (_model) {
187                 _model->start_write();
188         }
189
190         _writing = true;
191 }
192
193 void
194 MidiSource::mark_streaming_write_completed ()
195 {
196         if (_model) {
197                 _model->end_write(false); // FIXME: param?
198         }
199
200         _writing = false;
201 }
202
203 void
204 MidiSource::session_saved()
205 {
206         flush_midi();
207
208         if (_model && _model->edited()) {
209                 string newname;
210                 const string basename = PBD::basename_nosuffix(_name);
211                 string::size_type last_dash = basename.find_last_of("-");
212                 if (last_dash == string::npos || last_dash == basename.find_first_of("-")) {
213                         newname = basename + "-1";
214                 } else {
215                         stringstream ss(basename.substr(last_dash+1));
216                         unsigned write_count = 0;
217                         ss >> write_count;
218                         cerr << "WRITE COUNT: " << write_count << endl;
219                         ++write_count; // start at 1
220                         ss.clear();
221                         ss << basename.substr(0, last_dash) << "-" << write_count;
222                         newname = ss.str();
223                 }
224
225                 string newpath = _session.session_directory().midi_path().to_string() +"/"+ newname + ".mid";
226
227                 boost::shared_ptr<MidiSource> newsrc = boost::dynamic_pointer_cast<MidiSource>(
228                                 SourceFactory::createWritable(DataType::MIDI, _session, newpath, 1, 0, true));
229
230                 newsrc->set_timeline_position(_timeline_position);
231                 _model->write_to(newsrc);
232
233                 // cyclic dependency here, ugly :(
234                 newsrc->set_model(_model);
235                 _model->set_midi_source(newsrc.get());
236                 
237                 newsrc->flush_midi();
238
239                 Switched.emit(newsrc);
240         }
241 }
242
243 void
244 MidiSource::set_note_mode(NoteMode mode)
245 {
246         if (_model) {
247                 _model->set_note_mode(mode);
248         }
249 }
250