SWRMeter/firmware/waveformgenerator.ino

167 lines
4.1 KiB
Arduino
Raw Normal View History

2024-02-11 12:50:31 +01:00
#include "Waveforms.h"
uint8_t wf_wave0 = 0;
uint8_t wf_pos = 0;
uint16_t wf_freq = 0;
uint8_t wf_dutyCycle = 0;
2024-02-11 21:56:13 +01:00
unsigned long wf_sample_us = 0;
unsigned long wf_prevMicros = 0;
2024-02-11 12:50:31 +01:00
bool wf_outputEnabled = true; // Variable to control waveform output state
bool wf_pwm_needs_disabling = true;
2024-02-11 12:50:31 +01:00
void initWaveformGenerator()
{
2024-02-11 21:56:13 +01:00
pinMode(PWM_PIN, OUTPUT);
2024-02-11 12:50:31 +01:00
setWaveform(WAVEFORM_SINUS);
// Call the function to set default frequency, here you might want to specify a default frequency
2024-02-11 21:56:13 +01:00
setWaveformFrequency(WAVEFORM_DEFAULT_FREQ_HZ);
setWaveformDC(0);
analogWrite(PWM_PIN, 0);
2024-02-11 12:50:31 +01:00
}
2024-02-11 21:56:13 +01:00
void setWaveform(uint8_t waveform0)
2024-02-11 12:50:31 +01:00
{
if (waveform0 >= 0 and waveform0 < WAVEFORM_MAXWAVEFORM_NUM)
{
wf_wave0 = waveform0;
2024-02-11 21:56:13 +01:00
} else {
// Set default waveforms
2024-02-11 21:56:13 +01:00
wf_wave0 = WAVEFORM_SINUS;
2024-02-11 12:50:31 +01:00
}
}
2024-02-11 21:56:13 +01:00
void setWaveformFrequency(uint16_t frequency)
2024-02-11 12:50:31 +01:00
{
if (frequency >= 1 and frequency < WF_FREQ_MAX_HZ)
2024-02-11 12:50:31 +01:00
{
2024-02-11 21:56:13 +01:00
wf_freq = frequency;
2024-02-11 12:50:31 +01:00
} else {
2024-02-11 21:56:13 +01:00
wf_freq = WAVEFORM_DEFAULT_FREQ_HZ;
2024-02-11 12:50:31 +01:00
}
2024-02-11 21:56:13 +01:00
wf_sample_us = 1000000UL / ((unsigned long)wf_freq * WAVEFORM_MAX_SAMPLES_NUM);
if (frequency >= 30 && frequency <= 1000000)
{
setupPWMFrequency(PWM_PIN, frequency);
}
2024-02-11 12:50:31 +01:00
}
void setWaveformDC(uint8_t dc)
{
wf_dutyCycle = dc;
}
2024-02-11 12:50:31 +01:00
void enableWaveformOutput()
{
wf_outputEnabled = true;
}
void disableWaveformOutput()
{
wf_outputEnabled = false;
}
bool isWaveformEnabled()
{
return wf_outputEnabled;
}
2024-02-11 12:50:31 +01:00
void pollWaveformGenerator()
{
2024-02-11 21:56:13 +01:00
2024-02-11 12:50:31 +01:00
if (wf_outputEnabled)
{
if (wf_wave0 != WAVEFORM_DUTYCYCLE)
2024-02-11 21:56:13 +01:00
{
unsigned long currentMicros = micros(); // Aktuelle Zeit abrufen
if (currentMicros - wf_prevMicros >= wf_sample_us)
{
wf_prevMicros = currentMicros;
2024-02-11 21:56:13 +01:00
uint16_t sample = map(waveformsTable[wf_wave0][wf_pos], 0, 0xfff, 0, PWM_MAX_VALUE);
sample = constrain(sample, 0, PWM_MAX_VALUE);
// TODO write the selected waveform on DAC0
analogWrite(PWM_PIN, sample);
2024-02-11 21:56:13 +01:00
wf_pos++;
if (wf_pos == WAVEFORM_MAX_SAMPLES_NUM) // Reset the counter to repeat the wave
wf_pos = 0;
2024-02-11 21:56:13 +01:00
}
} else {
// WAVEFORM_DUTYCYCLE
if (analogRead(PWM_PIN) != wf_dutyCycle)
analogWrite(PWM_PIN, wf_dutyCycle);
2024-02-11 21:56:13 +01:00
}
if (!wf_pwm_needs_disabling)
wf_pwm_needs_disabling = true;
2024-02-11 21:56:13 +01:00
} else {
if (wf_pwm_needs_disabling)
{
analogWrite(PWM_PIN, 0);
wf_pwm_needs_disabling = false;
}
2024-02-11 12:50:31 +01:00
}
}
void setupPWMFrequency(uint8_t pin, unsigned long frequency) {
byte prescalerbits = 0b001; // Voreinstellung: kein Prescaler (F_CPU)
unsigned long prescaler_value;
unsigned long ocr;
// Berechne den idealen Prescaler für die gewünschte Frequenz
prescaler_value = F_CPU / frequency;
// Finde den passenden Prescaler
if (prescaler_value < 65536UL) {
prescalerbits = 0b001; // Kein Prescaler (F_CPU)
}
else if ((prescaler_value >>= 3) < 65536UL) {
prescalerbits = 0b010; // Prescaler 8
}
else if ((prescaler_value >>= 3) < 65536UL) {
prescalerbits = 0b011; // Prescaler 64
}
else if ((prescaler_value >>= 2) < 65536UL) {
prescalerbits = 0b100; // Prescaler 256
}
else if ((prescaler_value >>= 2) < 65536UL) {
prescalerbits = 0b101; // Prescaler 1024
}
else {
prescalerbits = 0b101; // Prescaler 1024
frequency = F_CPU / (prescaler_value = 65535UL);
}
// Berechne den Wert für den Output Compare Register (OCR)
ocr = F_CPU / (prescaler_value * frequency) - 1;
// Timer konfigurieren
pinMode(pin, OUTPUT);
if (pin == 5 || pin == 6) {
TCCR0B &= ~0b111; // Prescaler löschen
TCCR0B |= prescalerbits; // Neuen Prescaler setzen
OCR0A = ocr; // Output Compare Register setzen
TCCR0A |= (1 << WGM01); // CTC-Modus aktivieren
}
else if (pin == 9 || pin == 10) {
TCCR1B &= ~0b111; // Prescaler löschen
TCCR1B |= prescalerbits; // Neuen Prescaler setzen
OCR1A = ocr; // Output Compare Register setzen
TCCR1A |= (1 << WGM11); // CTC-Modus aktivieren
}
else if (pin == 3 || pin == 11) {
TCCR2B &= ~0b111; // Prescaler löschen
TCCR2B |= prescalerbits; // Neuen Prescaler setzen
OCR2A = ocr; // Output Compare Register setzen
TCCR2A |= (1 << WGM21); // CTC-Modus aktivieren
}
}