qmk_firmware/quantum/audio.c

553 lines
14 KiB
C
Raw Normal View History

2016-01-20 05:06:52 +00:00
#include <stdio.h>
#include <string.h>
#include <math.h>
#include <avr/pgmspace.h>
#include <avr/interrupt.h>
#include <avr/io.h>
2016-04-17 07:52:38 +00:00
#include "print.h"
2016-01-20 05:06:52 +00:00
#include "audio.h"
#include "keymap_common.h"
2016-04-16 03:38:21 +00:00
#include "eeconfig.h"
2016-04-19 21:00:45 +00:00
#include "vibrato_lut.h"
2016-01-20 05:06:52 +00:00
#define PI 3.14159265
#define CPU_PRESCALER 8
2016-04-19 21:00:45 +00:00
// Largely untested PWM audio mode (doesn't sound as good)
2016-01-20 05:06:52 +00:00
// #define PWM_AUDIO
#ifdef PWM_AUDIO
#include "wave.h"
#define SAMPLE_DIVIDER 39
#define SAMPLE_RATE (2000000.0/SAMPLE_DIVIDER/2048)
// Resistor value of 1/ (2 * PI * 10nF * (2000000 hertz / SAMPLE_DIVIDER / 10)) for 10nF cap
#endif
void delay_us(int count) {
while(count--) {
_delay_us(1);
}
}
int voices = 0;
int voice_place = 0;
double frequency = 0;
int volume = 0;
long position = 0;
double frequencies[8] = {0, 0, 0, 0, 0, 0, 0, 0};
int volumes[8] = {0, 0, 0, 0, 0, 0, 0, 0};
bool sliding = false;
int max = 0xFF;
float sum = 0;
int value = 128;
float place = 0;
float places[8] = {0, 0, 0, 0, 0, 0, 0, 0};
uint16_t place_int = 0;
bool repeat = true;
uint8_t * sample;
uint16_t sample_length = 0;
2016-04-19 16:58:13 +00:00
double freq = 0;
2016-01-20 05:06:52 +00:00
bool notes = false;
bool note = false;
float note_frequency = 0;
float note_length = 0;
2016-04-17 07:52:38 +00:00
float note_tempo = TEMPO_DEFAULT;
float note_timbre = TIMBRE_DEFAULT;
2016-01-20 05:06:52 +00:00
uint16_t note_position = 0;
float (* notes_pointer)[][2];
2016-04-17 07:52:38 +00:00
uint8_t notes_count;
2016-01-20 05:06:52 +00:00
bool notes_repeat;
2016-04-17 01:31:40 +00:00
float notes_rest;
bool note_resting = false;
2016-01-20 05:06:52 +00:00
uint8_t current_note = 0;
2016-04-17 01:31:40 +00:00
uint8_t rest_counter = 0;
2016-01-20 05:06:52 +00:00
2016-04-19 21:00:45 +00:00
float vibrato_counter = 0;
float vibrato_strength = .5;
float vibrato_rate = 0.125;
2016-04-19 16:58:13 +00:00
2016-04-19 21:00:45 +00:00
float polyphony_rate = .5;
2016-04-19 16:58:13 +00:00
2016-04-16 03:38:21 +00:00
audio_config_t audio_config;
void audio_toggle(void) {
audio_config.enable ^= 1;
eeconfig_write_audio(audio_config.raw);
}
void audio_on(void) {
audio_config.enable = 1;
eeconfig_write_audio(audio_config.raw);
}
void audio_off(void) {
audio_config.enable = 0;
eeconfig_write_audio(audio_config.raw);
}
2016-04-19 21:00:45 +00:00
// Vibrato rate functions
void set_vibrato_rate(float rate) {
vibrato_rate = rate;
}
void increase_vibrato_rate(float change) {
vibrato_rate *= change;
}
void decrease_vibrato_rate(float change) {
vibrato_rate /= change;
}
#ifdef VIBRATO_STRENGTH_ENABLE
void set_vibrato_strength(float strength) {
vibrato_strength = strength;
}
void increase_vibrato_strength(float change) {
vibrato_strength *= change;
}
void decrease_vibrato_strength(float change) {
vibrato_strength /= change;
}
#endif
// Polyphony functions
void set_polyphony_rate(float rate) {
polyphony_rate = rate;
}
void enable_polyphony() {
polyphony_rate = 5;
}
void disable_polyphony() {
polyphony_rate = 0;
}
void increase_polyphony_rate(float change) {
polyphony_rate *= change;
}
void decrease_polyphony_rate(float change) {
polyphony_rate /= change;
}
// Timbre function
void set_timbre(float timbre) {
note_timbre = timbre;
}
// Tempo functions
void set_tempo(float tempo) {
note_tempo = tempo;
}
void decrease_tempo(uint8_t tempo_change) {
note_tempo += (float) tempo_change;
}
void increase_tempo(uint8_t tempo_change) {
if (note_tempo - (float) tempo_change < 10) {
note_tempo = 10;
} else {
note_tempo -= (float) tempo_change;
}
}
2016-04-16 03:38:21 +00:00
2016-01-20 05:06:52 +00:00
void stop_all_notes() {
voices = 0;
#ifdef PWM_AUDIO
TIMSK3 &= ~_BV(OCIE3A);
#else
TIMSK3 &= ~_BV(OCIE3A);
TCCR3A &= ~_BV(COM3A1);
#endif
notes = false;
note = false;
frequency = 0;
volume = 0;
for (int i = 0; i < 8; i++) {
frequencies[i] = 0;
volumes[i] = 0;
}
}
void stop_note(double freq) {
if (note) {
2016-04-19 21:00:45 +00:00
cli();
2016-01-20 05:06:52 +00:00
#ifdef PWM_AUDIO
freq = freq / SAMPLE_RATE;
2016-01-20 05:06:52 +00:00
#endif
for (int i = 7; i >= 0; i--) {
if (frequencies[i] == freq) {
frequencies[i] = 0;
volumes[i] = 0;
for (int j = i; (j < 7); j++) {
frequencies[j] = frequencies[j+1];
frequencies[j+1] = 0;
volumes[j] = volumes[j+1];
volumes[j+1] = 0;
}
2016-04-19 21:00:45 +00:00
break;
2016-01-20 05:06:52 +00:00
}
}
voices--;
if (voices < 0)
voices = 0;
2016-04-19 01:01:48 +00:00
if (voice_place >= voices) {
voice_place = 0;
}
if (voices == 0) {
#ifdef PWM_AUDIO
TIMSK3 &= ~_BV(OCIE3A);
#else
TIMSK3 &= ~_BV(OCIE3A);
TCCR3A &= ~_BV(COM3A1);
#endif
frequency = 0;
volume = 0;
note = false;
2016-01-20 05:06:52 +00:00
}
2016-04-19 21:00:45 +00:00
sei();
2016-01-20 05:06:52 +00:00
}
}
void init_notes() {
2016-04-16 03:38:21 +00:00
/* check signature */
if (!eeconfig_is_enabled()) {
eeconfig_init();
}
audio_config.raw = eeconfig_read_audio();
2016-01-20 05:06:52 +00:00
#ifdef PWM_AUDIO
PLLFRQ = _BV(PDIV2);
PLLCSR = _BV(PLLE);
while(!(PLLCSR & _BV(PLOCK)));
PLLFRQ |= _BV(PLLTM0); /* PCK 48MHz */
/* Init a fast PWM on Timer4 */
TCCR4A = _BV(COM4A0) | _BV(PWM4A); /* Clear OC4A on Compare Match */
TCCR4B = _BV(CS40); /* No prescaling => f = PCK/256 = 187500Hz */
OCR4A = 0;
/* Enable the OC4A output */
DDRC |= _BV(PORTC6);
TIMSK3 &= ~_BV(OCIE3A); // Turn off 3A interputs
2016-04-17 03:10:18 +00:00
2016-01-20 05:06:52 +00:00
TCCR3A = 0x0; // Options not needed
TCCR3B = _BV(CS31) | _BV(CS30) | _BV(WGM32); // 64th prescaling and CTC
OCR3A = SAMPLE_DIVIDER - 1; // Correct count/compare, related to sample playback
#else
DDRC |= _BV(PORTC6);
TIMSK3 &= ~_BV(OCIE3A); // Turn off 3A interputs
TCCR3A = (0 << COM3A1) | (0 << COM3A0) | (1 << WGM31) | (0 << WGM30);
TCCR3B = (1 << WGM33) | (1 << WGM32) | (0 << CS32) | (1 << CS31) | (0 << CS30);
#endif
}
2016-04-19 16:58:13 +00:00
float mod(float a, int b)
{
float r = fmod(a, b);
return r < 0 ? r + b : r;
}
2016-01-20 05:06:52 +00:00
2016-04-19 21:00:45 +00:00
float vibrato(float average_freq) {
#ifdef VIBRATO_STRENGTH_ENABLE
float vibrated_freq = average_freq * pow(VIBRATO_LUT[(int)vibrato_counter], vibrato_strength);
#else
float vibrated_freq = average_freq * VIBRATO_LUT[(int)vibrato_counter];
#endif
vibrato_counter = mod((vibrato_counter + vibrato_rate * (1.0 + 440.0/average_freq)), VIBRATO_LUT_LENGTH);
return vibrated_freq;
}
2016-01-20 05:06:52 +00:00
ISR(TIMER3_COMPA_vect) {
if (note) {
#ifdef PWM_AUDIO
if (voices == 1) {
// SINE
OCR4A = pgm_read_byte(&sinewave[(uint16_t)place]) >> 2;
2016-04-17 03:10:18 +00:00
2016-01-20 05:06:52 +00:00
// SQUARE
// if (((int)place) >= 1024){
// OCR4A = 0xFF >> 2;
// } else {
// OCR4A = 0x00;
// }
2016-04-17 03:10:18 +00:00
2016-01-20 05:06:52 +00:00
// SAWTOOTH
// OCR4A = (int)place / 4;
// TRIANGLE
// if (((int)place) >= 1024) {
// OCR4A = (int)place / 2;
// } else {
// OCR4A = 2048 - (int)place / 2;
// }
place += frequency;
if (place >= SINE_LENGTH)
place -= SINE_LENGTH;
} else {
int sum = 0;
for (int i = 0; i < voices; i++) {
// SINE
sum += pgm_read_byte(&sinewave[(uint16_t)places[i]]) >> 2;
// SQUARE
// if (((int)places[i]) >= 1024){
// sum += 0xFF >> 2;
// } else {
// sum += 0x00;
// }
places[i] += frequencies[i];
if (places[i] >= SINE_LENGTH)
places[i] -= SINE_LENGTH;
}
OCR4A = sum;
}
#else
2016-04-19 16:58:13 +00:00
if (voices > 0) {
2016-04-19 21:00:45 +00:00
if (polyphony_rate > 0) {
2016-04-19 16:58:13 +00:00
if (voices > 1) {
voice_place %= voices;
2016-04-19 21:00:45 +00:00
if (place++ > (frequencies[voice_place] / polyphony_rate / CPU_PRESCALER)) {
2016-04-19 16:58:13 +00:00
voice_place = (voice_place + 1) % voices;
place = 0.0;
}
2016-04-19 01:01:48 +00:00
}
2016-04-19 16:58:13 +00:00
if (vibrato_strength > 0) {
2016-04-19 21:00:45 +00:00
freq = vibrato(frequencies[voice_place]);
2016-04-19 16:58:13 +00:00
} else {
freq = frequencies[voice_place];
}
2016-04-19 01:01:48 +00:00
} else {
2016-04-19 21:00:45 +00:00
if (frequency != 0 && frequency < frequencies[voices - 1] && frequency < frequencies[voices - 1] * pow(2, -440/frequencies[voices - 1]/12/2)) {
frequency = frequency * pow(2, 440/frequency/12/2);
} else if (frequency != 0 && frequency > frequencies[voices - 1] && frequency > frequencies[voices - 1] * pow(2, 440/frequencies[voices - 1]/12/2)) {
frequency = frequency * pow(2, -440/frequency/12/2);
2016-04-19 01:01:48 +00:00
} else {
frequency = frequencies[voices - 1];
}
2016-04-19 16:58:13 +00:00
2016-04-19 21:00:45 +00:00
if (vibrato_strength > 0) {
freq = vibrato(frequency);
2016-04-19 16:58:13 +00:00
} else {
freq = frequency;
}
2016-01-20 05:06:52 +00:00
}
2016-04-19 01:01:48 +00:00
ICR3 = (int)(((double)F_CPU) / (freq * CPU_PRESCALER)); // Set max to the period
OCR3A = (int)((((double)F_CPU) / (freq * CPU_PRESCALER)) * note_timbre); // Set compare to half the period
2016-01-20 05:06:52 +00:00
}
#endif
}
// SAMPLE
// OCR4A = pgm_read_byte(&sample[(uint16_t)place_int]);
// place_int++;
// if (place_int >= sample_length)
// if (repeat)
// place_int -= sample_length;
// else
// TIMSK3 &= ~_BV(OCIE3A);
if (notes) {
#ifdef PWM_AUDIO
OCR4A = pgm_read_byte(&sinewave[(uint16_t)place]) >> 0;
place += note_frequency;
if (place >= SINE_LENGTH)
place -= SINE_LENGTH;
#else
if (note_frequency > 0) {
2016-04-19 16:58:13 +00:00
float freq;
2016-04-19 21:00:45 +00:00
if (vibrato_strength > 0) {
freq = vibrato(note_frequency);
2016-04-19 16:58:13 +00:00
} else {
freq = note_frequency;
}
ICR3 = (int)(((double)F_CPU) / (freq * CPU_PRESCALER)); // Set max to the period
OCR3A = (int)((((double)F_CPU) / (freq * CPU_PRESCALER)) * note_timbre); // Set compare to half the period
2016-04-05 13:21:44 +00:00
} else {
ICR3 = 0;
OCR3A = 0;
2016-01-20 05:06:52 +00:00
}
#endif
note_position++;
2016-04-15 17:44:07 +00:00
bool end_of_note = false;
2016-04-17 03:10:18 +00:00
if (ICR3 > 0)
2016-04-15 17:44:07 +00:00
end_of_note = (note_position >= (note_length / ICR3 * 0xFFFF));
2016-04-17 03:10:18 +00:00
else
2016-04-15 17:44:07 +00:00
end_of_note = (note_position >= (note_length * 0x7FF));
if (end_of_note) {
2016-01-20 05:06:52 +00:00
current_note++;
2016-04-17 07:52:38 +00:00
if (current_note >= notes_count) {
2016-01-20 05:06:52 +00:00
if (notes_repeat) {
current_note = 0;
} else {
#ifdef PWM_AUDIO
TIMSK3 &= ~_BV(OCIE3A);
#else
TIMSK3 &= ~_BV(OCIE3A);
TCCR3A &= ~_BV(COM3A1);
#endif
notes = false;
return;
}
}
2016-04-17 03:10:18 +00:00
if (!note_resting && (notes_rest > 0)) {
2016-04-17 01:31:40 +00:00
note_resting = true;
note_frequency = 0;
note_length = notes_rest;
current_note--;
} else {
note_resting = false;
#ifdef PWM_AUDIO
note_frequency = (*notes_pointer)[current_note][0] / SAMPLE_RATE;
2016-04-17 07:52:38 +00:00
note_length = (*notes_pointer)[current_note][1] * (note_tempo / 100);
2016-04-17 01:31:40 +00:00
#else
note_frequency = (*notes_pointer)[current_note][0];
2016-04-17 07:52:38 +00:00
note_length = ((*notes_pointer)[current_note][1] / 4) * (note_tempo / 100);
2016-04-17 01:31:40 +00:00
#endif
}
2016-01-20 05:06:52 +00:00
note_position = 0;
}
}
2016-04-16 03:38:21 +00:00
if (!audio_config.enable) {
notes = false;
note = false;
}
2016-01-20 05:06:52 +00:00
}
2016-04-17 07:52:38 +00:00
void play_notes(float (*np)[][2], uint8_t n_count, bool n_repeat, float n_rest) {
2016-04-16 03:38:21 +00:00
if (audio_config.enable) {
2016-04-19 21:00:45 +00:00
cli();
2016-04-18 03:08:05 +00:00
// Cancel note if a note is playing
2016-04-18 01:26:06 +00:00
if (note)
2016-01-20 05:06:52 +00:00
stop_all_notes();
2016-04-17 22:14:42 +00:00
notes = true;
2016-01-20 05:06:52 +00:00
notes_pointer = np;
2016-04-17 07:52:38 +00:00
notes_count = n_count;
2016-01-20 05:06:52 +00:00
notes_repeat = n_repeat;
2016-04-17 01:31:40 +00:00
notes_rest = n_rest;
2016-01-20 05:06:52 +00:00
place = 0;
current_note = 0;
#ifdef PWM_AUDIO
note_frequency = (*notes_pointer)[current_note][0] / SAMPLE_RATE;
2016-04-17 07:52:38 +00:00
note_length = (*notes_pointer)[current_note][1] * (note_tempo / 100);
2016-01-20 05:06:52 +00:00
#else
note_frequency = (*notes_pointer)[current_note][0];
2016-04-17 07:52:38 +00:00
note_length = ((*notes_pointer)[current_note][1] / 4) * (note_tempo / 100);
2016-01-20 05:06:52 +00:00
#endif
note_position = 0;
#ifdef PWM_AUDIO
TIMSK3 |= _BV(OCIE3A);
#else
TIMSK3 |= _BV(OCIE3A);
TCCR3A |= _BV(COM3A1);
#endif
2016-04-19 21:00:45 +00:00
sei();
2016-01-20 05:06:52 +00:00
}
2016-04-16 03:38:21 +00:00
}
2016-01-20 05:06:52 +00:00
void play_sample(uint8_t * s, uint16_t l, bool r) {
2016-04-16 03:38:21 +00:00
if (audio_config.enable) {
2016-01-20 05:06:52 +00:00
stop_all_notes();
place_int = 0;
sample = s;
sample_length = l;
repeat = r;
#ifdef PWM_AUDIO
TIMSK3 |= _BV(OCIE3A);
#else
#endif
2016-04-16 03:38:21 +00:00
}
2016-01-20 05:06:52 +00:00
}
void play_note(double freq, int vol) {
2016-04-16 03:38:21 +00:00
if (audio_config.enable && voices < 8) {
2016-04-19 21:00:45 +00:00
cli();
2016-04-18 03:08:05 +00:00
// Cancel notes if notes are playing
2016-04-18 01:26:06 +00:00
if (notes)
2016-01-20 05:06:52 +00:00
stop_all_notes();
2016-04-17 22:14:42 +00:00
note = true;
2016-01-20 05:06:52 +00:00
#ifdef PWM_AUDIO
freq = freq / SAMPLE_RATE;
#endif
if (freq > 0) {
2016-04-19 01:01:48 +00:00
frequencies[voices] = freq;
volumes[voices] = vol;
2016-01-20 05:06:52 +00:00
voices++;
}
#ifdef PWM_AUDIO
TIMSK3 |= _BV(OCIE3A);
#else
TIMSK3 |= _BV(OCIE3A);
TCCR3A |= _BV(COM3A1);
#endif
2016-04-19 21:00:45 +00:00
sei();
2016-04-17 07:52:38 +00:00
}
}
2016-04-18 01:14:37 +00:00
//------------------------------------------------------------------------------
// Override these functions in your keymap file to play different tunes on
// startup and bootloader jump
__attribute__ ((weak))
void play_startup_tone()
{
}
__attribute__ ((weak))
void play_goodbye_tone()
{
}
//------------------------------------------------------------------------------