]> git.donarmstrong.com Git - qmk_firmware.git/blob - quantum/process_keycode/process_music.c
Adds support for Planck Rev 6 (#2666)
[qmk_firmware.git] / quantum / process_keycode / process_music.c
1 /* Copyright 2016 Jack Humbert
2  *
3  * This program is free software: you can redistribute it and/or modify
4  * it under the terms of the GNU General Public License as published by
5  * the Free Software Foundation, either version 2 of the License, or
6  * (at your option) any later version.
7  *
8  * This program is distributed in the hope that it will be useful,
9  * but WITHOUT ANY WARRANTY; without even the implied warranty of
10  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
11  * GNU General Public License for more details.
12  *
13  * You should have received a copy of the GNU General Public License
14  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
15  */
16 #include "process_music.h"
17
18 #ifdef AUDIO_ENABLE
19 #include "process_audio.h"
20 #endif
21 #if defined(MIDI_ENABLE) && defined(MIDI_BASIC)
22 #include "process_midi.h"
23 #endif
24
25 #if defined(AUDIO_ENABLE) || (defined(MIDI_ENABLE) && defined(MIDI_BASIC))
26
27 bool music_activated = false;
28 bool midi_activated = false;
29 uint8_t music_starting_note = 0x0C;
30 int music_offset = 7;
31 uint8_t music_mode = MUSIC_MODE_MAJOR;
32
33 // music sequencer
34 static bool music_sequence_recording = false;
35 static bool music_sequence_recorded = false;
36 static bool music_sequence_playing = false;
37 static uint8_t music_sequence[16] = {0};
38 static uint8_t music_sequence_count = 0;
39 static uint8_t music_sequence_position = 0;
40
41 static uint16_t music_sequence_timer = 0;
42 static uint16_t music_sequence_interval = 100;
43
44 #ifdef AUDIO_ENABLE
45   #ifndef MUSIC_ON_SONG
46     #define MUSIC_ON_SONG SONG(MUSIC_ON_SOUND)
47   #endif
48   #ifndef MUSIC_OFF_SONG
49     #define MUSIC_OFF_SONG SONG(MUSIC_OFF_SOUND)
50   #endif
51   #ifndef MIDI_ON_SONG
52     #define MIDI_ON_SONG SONG(MUSIC_ON_SOUND)
53   #endif
54   #ifndef MIDI_OFF_SONG
55     #define MIDI_OFF_SONG SONG(MUSIC_OFF_SOUND)
56   #endif
57   #ifndef CHROMATIC_SONG
58     #define CHROMATIC_SONG SONG(CHROMATIC_SOUND)
59   #endif
60   #ifndef GUITAR_SONG
61     #define GUITAR_SONG SONG(GUITAR_SOUND)
62   #endif
63   #ifndef VIOLIN_SONG
64     #define VIOLIN_SONG SONG(VIOLIN_SOUND)
65   #endif
66   #ifndef MAJOR_SONG
67     #define MAJOR_SONG SONG(MAJOR_SOUND)
68   #endif
69   float music_mode_songs[NUMBER_OF_MODES][5][2] = {
70     CHROMATIC_SONG,
71     GUITAR_SONG,
72     VIOLIN_SONG,
73     MAJOR_SONG
74   };
75   float music_on_song[][2] = MUSIC_ON_SONG;
76   float music_off_song[][2] = MUSIC_OFF_SONG;
77   float midi_on_song[][2] = MIDI_ON_SONG;
78   float midi_off_song[][2] = MIDI_OFF_SONG;
79 #endif
80
81 static void music_noteon(uint8_t note) {
82     #ifdef AUDIO_ENABLE
83     if (music_activated)
84       process_audio_noteon(note);
85     #endif
86     #if defined(MIDI_ENABLE) && defined(MIDI_BASIC)
87     if (midi_activated)
88       process_midi_basic_noteon(note);
89     #endif
90 }
91
92 static void music_noteoff(uint8_t note) {
93     #ifdef AUDIO_ENABLE
94     if (music_activated)
95       process_audio_noteoff(note);
96     #endif
97     #if defined(MIDI_ENABLE) && defined(MIDI_BASIC)
98     if (midi_activated)
99       process_midi_basic_noteoff(note);
100     #endif
101 }
102
103 void music_all_notes_off(void) {
104     #ifdef AUDIO_ENABLE
105     if (music_activated)
106       process_audio_all_notes_off();
107     #endif
108     #if defined(MIDI_ENABLE) && defined(MIDI_BASIC)
109     if (midi_activated)
110       process_midi_all_notes_off();
111     #endif
112 }
113
114 bool process_music(uint16_t keycode, keyrecord_t *record) {
115
116     if (keycode == MU_ON && record->event.pressed) {
117         music_on();
118         return false;
119     }
120
121     if (keycode == MU_OFF && record->event.pressed) {
122         music_off();
123         return false;
124     }
125
126     if (keycode == MU_TOG && record->event.pressed) {
127         if (music_activated) {
128             music_off();
129         } else {
130             music_on();
131         }
132         return false;
133     }
134
135     if (keycode == MI_ON && record->event.pressed) {
136         midi_on();
137         return false;
138     }
139
140     if (keycode == MI_OFF && record->event.pressed) {
141         midi_off();
142         return false;
143     }
144
145     if (keycode == MI_TOG && record->event.pressed) {
146         if (midi_activated) {
147             midi_off();
148         } else {
149             midi_on();
150         }
151         return false;
152     }
153
154     if (keycode == MU_MOD && record->event.pressed) {
155       music_mode_cycle();
156       return false;
157     }
158
159     if (music_activated || midi_activated) {
160       if (record->event.pressed) {
161         if (keycode == KC_LCTL) { // Start recording
162           music_all_notes_off();
163           music_sequence_recording = true;
164           music_sequence_recorded = false;
165           music_sequence_playing = false;
166           music_sequence_count = 0;
167           return false;
168         }
169
170         if (keycode == KC_LALT) { // Stop recording/playing
171           music_all_notes_off();
172           if (music_sequence_recording) { // was recording
173             music_sequence_recorded = true;
174           }
175           music_sequence_recording = false;
176           music_sequence_playing = false;
177           return false;
178         }
179
180         if (keycode == KC_LGUI && music_sequence_recorded) { // Start playing
181           music_all_notes_off();
182           music_sequence_recording = false;
183           music_sequence_playing = true;
184           music_sequence_position = 0;
185           music_sequence_timer = 0;
186           return false;
187         }
188
189         if (keycode == KC_UP) {
190           music_sequence_interval-=10;
191           return false;
192         }
193
194         if (keycode == KC_DOWN) {
195           music_sequence_interval+=10;
196           return false;
197         }
198       }
199
200       uint8_t note = 36;
201       #ifdef MUSIC_MAP
202         if (music_mode == MUSIC_MODE_CHROMATIC) {
203           note = music_starting_note + music_offset + 36 + music_map[record->event.key.row][record->event.key.col];
204         } else {
205           uint8_t position = music_map[record->event.key.row][record->event.key.col];
206           note = music_starting_note + music_offset + 36 + SCALE[position % 12] + (position / 12)*12;
207         }
208       #else
209         if (music_mode == MUSIC_MODE_CHROMATIC)
210           note = (music_starting_note + record->event.key.col + music_offset - 3)+12*(MATRIX_ROWS - record->event.key.row);
211         else if (music_mode == MUSIC_MODE_GUITAR)
212           note = (music_starting_note + record->event.key.col + music_offset + 32)+5*(MATRIX_ROWS - record->event.key.row);
213         else if (music_mode == MUSIC_MODE_VIOLIN)
214           note = (music_starting_note + record->event.key.col + music_offset + 32)+7*(MATRIX_ROWS - record->event.key.row);
215         else if (music_mode == MUSIC_MODE_MAJOR)
216           note = (music_starting_note + SCALE[record->event.key.col + music_offset] - 3)+12*(MATRIX_ROWS - record->event.key.row);
217         else
218           note = music_starting_note;
219       #endif
220
221       if (record->event.pressed) {
222         music_noteon(note);
223         if (music_sequence_recording) {
224           music_sequence[music_sequence_count] = note;
225           music_sequence_count++;
226         }
227       } else {
228         music_noteoff(note);
229       }
230
231       if (music_mask(keycode))
232         return false;
233     }
234
235     return true;
236 }
237
238 bool music_mask(uint16_t keycode) {
239   #ifdef MUSIC_MASK
240     return MUSIC_MASK;
241   #else
242     return music_mask_kb(keycode);
243   #endif
244 }
245
246 __attribute__((weak))
247 bool music_mask_kb(uint16_t keycode) {
248   return music_mask_user(keycode);
249 }
250
251 __attribute__((weak))
252 bool music_mask_user(uint16_t keycode) {
253   return keycode < 0xFF;
254 }
255
256 bool is_music_on(void) {
257     return (music_activated != 0);
258 }
259
260 void music_toggle(void) {
261     if (!music_activated) {
262         music_on();
263     } else {
264         music_off();
265     }
266 }
267
268 void music_on(void) {
269     music_activated = 1;
270     #ifdef AUDIO_ENABLE
271       PLAY_SONG(music_on_song);
272     #endif
273     music_on_user();
274 }
275
276 void music_off(void) {
277     music_all_notes_off();
278     music_activated = 0;
279     #ifdef AUDIO_ENABLE
280       PLAY_SONG(music_off_song);
281     #endif
282 }
283
284 bool is_midi_on(void) {
285     return (midi_activated != 0);
286 }
287
288 void midi_toggle(void) {
289     if (!midi_activated) {
290         midi_on();
291     } else {
292         midi_off();
293     }
294 }
295
296 void midi_on(void) {
297     midi_activated = 1;
298     #ifdef AUDIO_ENABLE
299       PLAY_SONG(midi_on_song);
300     #endif
301     midi_on_user();
302 }
303
304 void midi_off(void) {
305     #if defined(MIDI_ENABLE) && defined(MIDI_BASIC)
306       process_midi_all_notes_off();
307     #endif
308     midi_activated = 0;
309     #ifdef AUDIO_ENABLE
310       PLAY_SONG(midi_off_song);
311     #endif
312 }
313
314 void music_mode_cycle(void) {
315   music_all_notes_off();
316   music_mode = (music_mode + 1) % NUMBER_OF_MODES;
317   #ifdef AUDIO_ENABLE
318     PLAY_SONG(music_mode_songs[music_mode]);
319   #endif
320 }
321
322 void matrix_scan_music(void) {
323   if (music_sequence_playing) {
324     if ((music_sequence_timer == 0) || (timer_elapsed(music_sequence_timer) > music_sequence_interval)) {
325       music_sequence_timer = timer_read();
326       uint8_t prev_note = music_sequence[(music_sequence_position - 1 < 0)?(music_sequence_position - 1 + music_sequence_count):(music_sequence_position - 1)];
327       uint8_t next_note = music_sequence[music_sequence_position];
328       music_noteoff(prev_note);
329       music_noteon(next_note);
330       music_sequence_position = (music_sequence_position + 1) % music_sequence_count;
331     }
332   }
333 }
334
335 __attribute__ ((weak))
336 void music_on_user() {}
337
338 __attribute__ ((weak))
339 void midi_on_user() {}
340
341 __attribute__ ((weak))
342 void music_scale_user() {}
343
344 #endif // defined(AUDIO_ENABLE) || (defined(MIDI_ENABLE) && defined(MIDI_BASIC))