fb6a89c81fb9b93849841ec53b412dcfacc3d5b0
[ardour.git] / tools / bb / bb.cc
1 #include <iostream>
2 #include <cstdio>
3 #include <cmath>
4 #include <cstring>
5
6 #include <unistd.h>
7 #include <stdint.h>
8
9 #include <jack/jack.h>
10 #include <jack/midiport.h>
11
12 #include "evoral/midi_events.h"
13
14 #include "bb.h"
15 #include "gui.h"
16
17 using std::cerr;
18 using std::endl;
19
20
21 static bool
22 second_simultaneous_midi_byte_is_first (uint8_t a, uint8_t b)
23 {
24         bool b_first = false;
25
26         /* two events at identical times. we need to determine
27            the order in which they should occur.
28
29            the rule is:
30
31            Controller messages
32            Program Change
33            Note Off
34            Note On
35            Note Pressure
36            Channel Pressure
37            Pitch Bend
38         */
39
40         if ((a) >= 0xf0 || (b) >= 0xf0 || ((a & 0xf) != (b & 0xf))) {
41
42                 /* if either message is not a channel message, or if the channels are
43                  * different, we don't care about the type.
44                  */
45
46                 b_first = true;
47
48         } else {
49
50                 switch (b & 0xf0) {
51                 case MIDI_CMD_CONTROL:
52                         b_first = true;
53                         break;
54
55                 case MIDI_CMD_PGM_CHANGE:
56                         switch (a & 0xf0) {
57                         case MIDI_CMD_CONTROL:
58                                 break;
59                         case MIDI_CMD_PGM_CHANGE:
60                         case MIDI_CMD_NOTE_OFF:
61                         case MIDI_CMD_NOTE_ON:
62                         case MIDI_CMD_NOTE_PRESSURE:
63                         case MIDI_CMD_CHANNEL_PRESSURE:
64                         case MIDI_CMD_BENDER:
65                                 b_first = true;
66                         }
67                         break;
68
69                 case MIDI_CMD_NOTE_OFF:
70                         switch (a & 0xf0) {
71                         case MIDI_CMD_CONTROL:
72                         case MIDI_CMD_PGM_CHANGE:
73                                 break;
74                         case MIDI_CMD_NOTE_OFF:
75                         case MIDI_CMD_NOTE_ON:
76                         case MIDI_CMD_NOTE_PRESSURE:
77                         case MIDI_CMD_CHANNEL_PRESSURE:
78                         case MIDI_CMD_BENDER:
79                                 b_first = true;
80                         }
81                         break;
82
83                 case MIDI_CMD_NOTE_ON:
84                         switch (a & 0xf0) {
85                         case MIDI_CMD_CONTROL:
86                         case MIDI_CMD_PGM_CHANGE:
87                         case MIDI_CMD_NOTE_OFF:
88                                 break;
89                         case MIDI_CMD_NOTE_ON:
90                         case MIDI_CMD_NOTE_PRESSURE:
91                         case MIDI_CMD_CHANNEL_PRESSURE:
92                         case MIDI_CMD_BENDER:
93                                 b_first = true;
94                         }
95                         break;
96                 case MIDI_CMD_NOTE_PRESSURE:
97                         switch (a & 0xf0) {
98                         case MIDI_CMD_CONTROL:
99                         case MIDI_CMD_PGM_CHANGE:
100                         case MIDI_CMD_NOTE_OFF:
101                         case MIDI_CMD_NOTE_ON:
102                                 break;
103                         case MIDI_CMD_NOTE_PRESSURE:
104                         case MIDI_CMD_CHANNEL_PRESSURE:
105                         case MIDI_CMD_BENDER:
106                                 b_first = true;
107                         }
108                         break;
109
110                 case MIDI_CMD_CHANNEL_PRESSURE:
111                         switch (a & 0xf0) {
112                         case MIDI_CMD_CONTROL:
113                         case MIDI_CMD_PGM_CHANGE:
114                         case MIDI_CMD_NOTE_OFF:
115                         case MIDI_CMD_NOTE_ON:
116                         case MIDI_CMD_NOTE_PRESSURE:
117                                 break;
118                         case MIDI_CMD_CHANNEL_PRESSURE:
119                         case MIDI_CMD_BENDER:
120                                 b_first = true;
121                         }
122                         break;
123                 case MIDI_CMD_BENDER:
124                         switch (a & 0xf0) {
125                         case MIDI_CMD_CONTROL:
126                         case MIDI_CMD_PGM_CHANGE:
127                         case MIDI_CMD_NOTE_OFF:
128                         case MIDI_CMD_NOTE_ON:
129                         case MIDI_CMD_NOTE_PRESSURE:
130                         case MIDI_CMD_CHANNEL_PRESSURE:
131                                 break;
132                         case MIDI_CMD_BENDER:
133                                 b_first = true;
134                         }
135                         break;
136                 }
137         }
138
139         return b_first;
140 }
141
142 BeatBox::BeatBox (int sr)
143         : _start_requested (false)
144         , _running (false)
145         , _measures (2)
146         , _tempo (120)
147         , _meter_beats (4)
148         , _meter_beat_type (4)
149         , _input (0)
150         , _output (0)
151         , superclock_cnt (0)
152         , last_start (0)
153         , _sample_rate (sr)
154         , whole_note_superclocks (0)
155         , beat_superclocks (0)
156         , measure_superclocks (0)
157         , _quantize_divisor (4)
158         , clear_pending (false)
159 {
160         for (uint32_t n = 0; n < 1024; ++n) {
161                 event_pool.push_back (new Event());
162         }
163 }
164
165 BeatBox::~BeatBox ()
166 {
167 }
168
169 int
170 BeatBox::register_ports (jack_client_t* jack)
171 {
172         if ((_input = jack_port_register (jack, "midi-in", JACK_DEFAULT_MIDI_TYPE, JackPortIsInput, 0)) == 0) {
173                 cerr << "no input port\n";
174                 return -1;
175         }
176         if ((_output = jack_port_register (jack, "midi-out", JACK_DEFAULT_MIDI_TYPE, JackPortIsOutput, 0)) == 0) {
177                 cerr << "no output port\n";
178                 jack_port_unregister (jack, _input);
179                 return -1;
180         }
181
182         return 0;
183 }
184
185 void
186 BeatBox::start ()
187 {
188         /* compute tempo, beat steps etc. */
189
190         /*
191            superclocks_per_minute = superclock_ticks_per_second * 60;
192            beats_per_minute = _tempo;
193            whole_notes_per_minute = beats_per_minute / _meter_beat_type;
194         */
195
196         whole_note_superclocks = (superclock_ticks_per_second * 60) / (_tempo / _meter_beat_type);
197         cerr << "there are " << _tempo / _meter_beat_type << " whole notes per minute, which is " << superclock_ticks_per_second * 60 << " sct, so wns = " << whole_note_superclocks << endl;
198         beat_superclocks = whole_note_superclocks / _meter_beat_type;
199         measure_superclocks = beat_superclocks * _meter_beats;
200
201         /* we can start */
202
203         _start_requested = true;
204 }
205
206 void
207 BeatBox::stop ()
208 {
209         _start_requested = false;
210 }
211
212 int
213 BeatBox::process (int nsamples)
214 {
215         if (!_running) {
216                 if (_start_requested) {
217                         _running = true;
218                         last_start = superclock_cnt;
219                 }
220
221         } else {
222                 if (!_start_requested) {
223                         _running = false;
224                 }
225         }
226
227         superclock_t superclocks = samples_to_superclock (nsamples, _sample_rate);
228
229         if (!_running) {
230                 superclock_cnt += superclocks;
231                 return 0;
232         }
233
234         superclock_t process_start = superclock_cnt - last_start;
235         superclock_t process_end = process_start + superclocks;
236         const superclock_t loop_length = _measures * measure_superclocks;
237         const superclock_t orig_superclocks = superclocks;
238
239         process_start %= loop_length;
240         process_end   %= loop_length;
241
242         bool two_pass_required;
243         superclock_t offset = 0;
244
245         if (process_end < process_start) {
246                 two_pass_required = true;
247                 process_end = loop_length;
248                 superclocks = process_end - process_start;
249         } else {
250                 two_pass_required = false;
251         }
252
253         unsigned char* buffer;
254         void* out_buf;
255         void* in_buf;
256         jack_midi_event_t in_event;
257         jack_nframes_t event_index;
258         jack_nframes_t event_count;
259
260         /* do this on the first pass only */
261         out_buf = jack_port_get_buffer (_output, nsamples);
262         jack_midi_clear_buffer (out_buf);
263
264   second_pass:
265
266         /* Output */
267
268         if (clear_pending) {
269
270                 for (Events::iterator ee = _current_events.begin(); ee != _current_events.end(); ++ee) {
271                         event_pool.push_back (*ee);
272                 }
273                 _current_events.clear ();
274                 clear_pending = false;
275         }
276
277         for (Events::iterator ee = _current_events.begin(); ee != _current_events.end(); ++ee) {
278                 Event* e = (*ee);
279
280                 if (e->size && (e->time >= process_start && e->time < process_end)) {
281                         if ((buffer = jack_midi_event_reserve (out_buf, superclock_to_samples (offset + e->time - process_start, _sample_rate), e->size)) != 0) {
282                                 memcpy (buffer, e->buf, e->size);
283                         } else {
284                                 cerr << "Could not reserve space for output event @ " << e << " of size " << e->size << " @ " << offset + e->time - process_start
285                                      << " (samples: " << superclock_to_samples (offset + e->time - process_start, _sample_rate) << ") offset is " << offset
286                                      << ")\n";
287                         }
288                 }
289
290                 if (e->time >= process_end) {
291                         break;
292                 }
293         }
294
295         /* input */
296
297         in_buf = jack_port_get_buffer (_input, nsamples);
298         event_index = 0;
299
300         while (jack_midi_event_get (&in_event, in_buf, event_index++) == 0) {
301
302                 superclock_t event_time = superclock_cnt + samples_to_superclock (in_event.time, _sample_rate);
303                 superclock_t elapsed_time = event_time - last_start;
304                 superclock_t in_loop_time = elapsed_time % loop_length;
305                 superclock_t quantized_time;
306
307                 if (_quantize_divisor != 0) {
308                         const superclock_t time_per_beat = whole_note_superclocks / _quantize_divisor;
309                         quantized_time = (in_loop_time / time_per_beat) * time_per_beat;
310                 } else {
311                         quantized_time = elapsed_time;
312                 }
313
314                 if (in_event.size > 24) {
315                         cerr << "Ignored large MIDI event\n";
316                         continue;
317                 }
318
319                 if (event_pool.empty()) {
320                         cerr << "No more events, grow pool\n";
321                         continue;
322                 }
323
324                 Event* e = event_pool.back();
325                 event_pool.pop_back ();
326
327                 e->time = quantized_time;
328                 e->size = in_event.size;
329                 memcpy (e->buf, in_event.buffer, in_event.size);
330
331                 _current_events.insert (e);
332         }
333
334         superclock_cnt += superclocks;
335
336         if (two_pass_required) {
337                 offset = superclocks;
338                 superclocks = orig_superclocks - superclocks;
339                 process_start = 0;
340                 process_end = superclocks;
341                 two_pass_required = false;
342                 cerr << "2nd Pass for " << superclocks << " offset = " << offset << endl;
343                 goto second_pass;
344         }
345
346         return 0;
347 }
348
349 void
350 BeatBox::set_quantize (int divisor)
351 {
352         _quantize_divisor = divisor;
353 }
354
355 void
356 BeatBox::clear ()
357 {
358         clear_pending = true;
359 }
360
361 bool
362 BeatBox::EventComparator::operator() (Event const * a, Event const *b) const
363 {
364         if (a->time == b->time) {
365                 if (a->buf[0] == b->buf[0]) {
366                         return a < b;
367                 }
368                 return !second_simultaneous_midi_byte_is_first (a->buf[0], b->buf[0]);
369         }
370         return a->time < b->time;
371 }
372
373 static int
374 process (jack_nframes_t nsamples, void* arg)
375 {
376         BeatBox* bbox = static_cast<BeatBox*> (arg);
377         return bbox->process (nsamples);
378 }
379
380 int
381 main (int argc, char* argv[])
382 {
383         jack_client_t* jack;
384         const char *server_name = NULL;
385         jack_options_t options = JackNullOption;
386         jack_status_t status;
387
388         if ((jack = jack_client_open ("beatbox", options, &status, server_name)) == 0) {
389                 cerr << "Could not connect to JACK\n";
390                 return -1;
391         }
392
393         BeatBox* bbox = new BeatBox (jack_get_sample_rate (jack));
394         BBGUI* gui = new BBGUI (&argc, &argv, jack, bbox);
395
396         bbox->register_ports (jack);
397
398         jack_set_process_callback (jack, process, bbox);
399         jack_activate (jack);
400
401         bbox->start ();
402
403         gui->run ();
404 }