awulff-pico-playground/pico-daq/pico_daq.cpp

103 wiersze
3.0 KiB
C++

// Sample from the ADC continuously at a particular sample rate
// and then outputs base64 bytes via Serial
// much of this code is from pico-examples/adc/dma_capture/dma_capture.c
// the rest is written by Alex Wulff (www.AlexWulff.com)
#include <stdio.h>
#include <cstring>
#include "pico/stdlib.h"
#include "hardware/adc.h"
#include "hardware/dma.h"
#include "base64.h"
// set this to determine sample rate
// 0 = 500,000 Hz
// 960 = 50,000 Hz
// 9600 = 5,000 Hz
#define CLOCK_DIV 12000
// Channel 0 is GPIO26
#define CAPTURE_CHANNEL 0
#define LED_PIN 25
#define NSAMP 10000
uint16_t capture_buf[NSAMP];
float sending_buf[NSAMP];
int main() {
stdio_init_all();
gpio_init(LED_PIN);
gpio_set_dir(LED_PIN, GPIO_OUT);
adc_gpio_init(26 + CAPTURE_CHANNEL);
adc_init();
adc_select_input(CAPTURE_CHANNEL);
adc_fifo_setup(
true, // Write each completed conversion to the sample FIFO
true, // Enable DMA data request (DREQ)
1, // DREQ (and IRQ) asserted when at least 1 sample present
false, // We won't see the ERR bit because of 8 bit reads; disable.
false // Don't shift each sample to 8 bits when pushing to FIFO
);
// set sample rate
adc_set_clkdiv(CLOCK_DIV);
sleep_ms(1000);
// Set up the DMA to start transferring data as soon as it appears in FIFO
uint dma_chan = dma_claim_unused_channel(true);
dma_channel_config cfg = dma_channel_get_default_config(dma_chan);
// Reading from constant address, writing to incrementing byte addresses
channel_config_set_transfer_data_size(&cfg, DMA_SIZE_16);
channel_config_set_read_increment(&cfg, false);
channel_config_set_write_increment(&cfg, true);
// Pace transfers based on availability of ADC samples
channel_config_set_dreq(&cfg, DREQ_ADC);
while (1) {
adc_fifo_drain();
adc_run(false);
dma_channel_configure(dma_chan, &cfg,
capture_buf, // dst
&adc_hw->fifo, // src
NSAMP, // transfer count
true // start immediately
);
// if the light does not flash, then there is some data loss.
// pico can't print the data out fast enough, so the capturing
// finishes before the data is done printing. This will result
// in the ADC not collecting for portions of the sampling
// run sampling routine
gpio_put(LED_PIN, 1);
adc_run(true);
// first transmission will be garbage since we haven't filled buf
std::string encoded =
base64_encode((unsigned char const *)sending_buf, NSAMP*4);
printf("%s", encoded.c_str());
gpio_put(LED_PIN, 0);
dma_channel_wait_for_finish_blocking(dma_chan);
uint16_t min = 32768;
uint16_t max = 0;
for (uint32_t i=0; i<NSAMP; i++) {
if (capture_buf[i] > max) max = capture_buf[i];
if (capture_buf[i] < min) min = capture_buf[i];
}
for (uint32_t i=0; i<NSAMP; i++) {
features[i] = ((float)capture_buf[i]-(float)min)/((float)max-(float)min)*2-1;
}
}
}