/*

  Test des Free run mode im ADC
  ATmega328P, aka Arduino Uno/Micro/Nano

*/

#define F_CPU 16000000UL

#include <stdbool.h>
#include <avr/io.h>
#include <avr/interrupt.h>
#include <avr/sleep.h>
#include <util/delay.h>

#define ADC_CHANNELS 3
#define ADC_AVERAGE_CNT  4
#define FREERUN

volatile uint16_t adc_avg[ADC_CHANNELS];
volatile bool adc_run;

// ADC initialisieren

void adc_init(void) {
    ADMUX  = (1 << REFS0);   // AVcc als Referenz, ADC0 als Eingang
    ADCSRA = (1 << ADEN)   | // ADC einschalten
             (1 << ADIE)   | // Interrupt einschalten
#ifdef FREERUN
             (1 << ADATE)  | // Free Running Modus aktivieren
#endif
             (1 << ADPS2);   // Prescaler auf 16

    sei(); // globale Interrupts aktivieren
}

// ISR für den ADC-Interrupt

ISR(ADC_vect) {
    static uint8_t channel;
    static uint16_t adc_accu[ADC_CHANNELS];
    static uint8_t avg_cnt;
    uint8_t i;

    PINB = (1<<PB0);         // Test, PB0 umschalten
#ifndef FREERUN
    ADCSRA |= (1 << ADSC);   // nächste Wandlung anuell starten
#endif

    // Mittelwertbildung
    adc_accu[channel] += ADCW;
    if (++channel >= ADC_CHANNELS) {
        channel = 0;        
        if (++avg_cnt == ADC_AVERAGE_CNT) {
            avg_cnt = 0;
            for (i=0; i<ADC_CHANNELS; i++) {
                adc_avg[i]  = adc_accu[i] / ADC_AVERAGE_CNT;
                adc_accu[i] = 0;
            }
            adc_run = 0;
            ADCSRA &= ~(1 << ADIE);     // ISR aus
        }
    }
    ADMUX = (channel & 0x03) | (ADMUX & ~0x3);
}

void start_adc_avg(void) {
    ADCSRA |= (1<<ADIF);        // aktiven Interrupt löschen
    ADCSRA |= (1<<ADIE);        // Interrupt einschalten
    ADMUX  &= ~0x03;            // MUX rücksetzen
    adc_run = 1;                // Software Flag
    PORTB  &= ~(1<<PB0);        // Testpin
    ADCSRA |= (1<<ADSC);        // ADC starten
}

int main(void) {

    DDRB = (1<<PB0);
    adc_init();                 // ADC initialisieren

    start_adc_avg();            // Mittelwertbildung starten     
    while (adc_run) {
        // Hauptprogramm läuft weiter, ohne zu blockieren
        // Hier kannst du andere Aufgaben erledigen
        set_sleep_mode(SLEEP_MODE_IDLE);
        sleep_mode();
    }

    // hier ist die Mittelwertbildung zu Ende
    // Werte stehen in adc_avg

    _delay_us(50);

    // nochmal
    start_adc_avg();            // Mittelwertbildung starten     
    while (adc_run) {
        // Hauptprogramm läuft weiter, ohne zu blockieren
        // Hier kannst du andere Aufgaben erledigen
        set_sleep_mode(SLEEP_MODE_IDLE);
        sleep_mode();
    }

    // hier ist die Mittelwertbildung zu Ende
    // Werte stehen wieder in adc_avg
}