use core::fmt; use log::{info, warn}; use stm32f4xx_hal::hal::{ blocking::spi::Transfer, digital::v2::OutputPin, }; use super::{ regs::{self, Register, RegisterData}, checksum::{ChecksumMode, Checksum}, Mode, Input, RefSource, PostFilter, DigitalFilterOrder, }; /// AD7172-2 implementation /// /// [Manual](https://www.analog.com/media/en/technical-documentation/data-sheets/AD7172-2.pdf) pub struct Adc, NSS: OutputPin> { spi: SPI, nss: NSS, checksum_mode: ChecksumMode, } impl, NSS: OutputPin, E: fmt::Debug> Adc { pub fn new(spi: SPI, mut nss: NSS) -> Result { let _ = nss.set_high(); let mut adc = Adc { spi, nss, checksum_mode: ChecksumMode::Off, }; adc.reset()?; adc.set_checksum_mode(ChecksumMode::Crc).unwrap(); let mut retries = 0; let mut adc_id; loop { adc_id = adc.identify()?; if adc_id & 0xFFF0 == 0x00D0 { break; } else { retries += 1; } } info!("ADC id: {:04X} ({} retries)", adc_id, retries); let mut adc_mode = ::Data::empty(); adc_mode.set_ref_en(true); adc_mode.set_mode(Mode::Standby); adc.write_reg(®s::AdcMode, &mut adc_mode)?; Ok(adc) } /// `0x00DX` for AD7172-2 pub fn identify(&mut self) -> Result { self.read_reg(®s::Id) .map(|id| id.id()) } pub fn set_checksum_mode(&mut self, mode: ChecksumMode) -> Result<(), SPI::Error> { // Cannot use update_reg() here because checksum_mode is // updated between read_reg() and write_reg(). let mut ifmode = self.read_reg(®s::IfMode)?; ifmode.set_crc(mode); self.checksum_mode = mode; self.write_reg(®s::IfMode, &mut ifmode)?; Ok(()) } pub fn set_sync_enable(&mut self, enable: bool) -> Result<(), SPI::Error> { self.update_reg(®s::GpioCon, |data| { data.set_sync_en(enable); }) } pub fn setup_channel( &mut self, index: u8, in_pos: Input, in_neg: Input ) -> Result<(), SPI::Error> { self.update_reg(®s::SetupCon { index }, |data| { data.set_bipolar(false); data.set_refbuf_pos(true); data.set_refbuf_neg(true); data.set_ainbuf_pos(true); data.set_ainbuf_neg(true); data.set_ref_sel(RefSource::External); })?; self.update_reg(®s::FiltCon { index }, |data| { data.set_enh_filt_en(true); data.set_enh_filt(PostFilter::F16SPS); data.set_order(DigitalFilterOrder::Sinc5Sinc1); })?; self.update_reg(®s::Channel { index }, |data| { data.set_setup(index); data.set_enabled(true); data.set_a_in_pos(in_pos); data.set_a_in_neg(in_neg); })?; Ok(()) } pub fn disable_channel( &mut self, index: u8 ) -> Result<(), SPI::Error> { self.update_reg(®s::Channel { index }, |data| { data.set_enabled(false); })?; Ok(()) } pub fn disable_all_channels(&mut self) -> Result<(), SPI::Error> { for index in 0..4 { self.update_reg(®s::Channel { index }, |data| { data.set_enabled(false); })?; } Ok(()) } /// Calibrates offset registers pub fn calibrate(&mut self) -> Result<(), SPI::Error> { // internal offset calibration self.update_reg(®s::AdcMode, |adc_mode| { adc_mode.set_mode(Mode::InternalOffsetCalibration); })?; while ! self.read_reg(®s::Status)?.ready() {} // system offset calibration self.update_reg(®s::AdcMode, |adc_mode| { adc_mode.set_mode(Mode::SystemOffsetCalibration); })?; while ! self.read_reg(®s::Status)?.ready() {} // system gain calibration self.update_reg(®s::AdcMode, |adc_mode| { adc_mode.set_mode(Mode::SystemGainCalibration); })?; while ! self.read_reg(®s::Status)?.ready() {} Ok(()) } pub fn start_continuous_conversion(&mut self) -> Result<(), SPI::Error> { let mut adc_mode = ::Data::empty(); adc_mode.set_ref_en(true); adc_mode.set_mode(Mode::ContinuousConversion); self.write_reg(®s::AdcMode, &mut adc_mode)?; Ok(()) } pub fn get_postfilter(&mut self, index: u8) -> Result, SPI::Error> { self.read_reg(®s::FiltCon { index }) .map(|data| { if data.enh_filt_en() { Some(data.enh_filt()) } else { None } }) } pub fn set_postfilter(&mut self, index: u8, filter: Option) -> Result<(), SPI::Error> { self.update_reg(®s::FiltCon { index }, |data| { match filter { None => data.set_enh_filt_en(false), Some(filter) => { data.set_enh_filt_en(true); data.set_enh_filt(filter); } } }) } /// Returns the channel the data is from pub fn data_ready(&mut self) -> Result, SPI::Error> { self.read_reg(®s::Status) .map(|status| { if status.ready() { Some(status.channel()) } else { None } }) } /// Get data pub fn read_data(&mut self) -> Result { self.read_reg(®s::Data) .map(|data| data.data()) } fn read_reg(&mut self, reg: &R) -> Result { let mut reg_data = R::Data::empty(); let address = 0x40 | reg.address(); let mut checksum = Checksum::new(self.checksum_mode); checksum.feed(&[address]); let checksum_out = checksum.result(); loop { let checksum_in = self.transfer(address, reg_data.as_mut(), checksum_out)?; checksum.feed(®_data); let checksum_expected = checksum.result(); if checksum_expected == checksum_in { break; } // Retry warn!("read_reg {:02X}: checksum error: {:?}!={:?}, retrying", reg.address(), checksum_expected, checksum_in); } Ok(reg_data) } fn write_reg(&mut self, reg: &R, reg_data: &mut R::Data) -> Result<(), SPI::Error> { loop { let address = reg.address(); let mut checksum = Checksum::new(match self.checksum_mode { ChecksumMode::Off => ChecksumMode::Off, // write checksums are always crc ChecksumMode::Xor => ChecksumMode::Crc, ChecksumMode::Crc => ChecksumMode::Crc, }); checksum.feed(&[address]); checksum.feed(®_data); let checksum_out = checksum.result(); let mut data = reg_data.clone(); self.transfer(address, data.as_mut(), checksum_out)?; // Verification let readback_data = self.read_reg(reg)?; if *readback_data == **reg_data { return Ok(()); } warn!("write_reg {:02X}: readback error, {:?}!={:?}, retrying", address, &*readback_data, &**reg_data); } } fn update_reg(&mut self, reg: &R, f: F) -> Result where R: regs::Register, F: FnOnce(&mut R::Data) -> A, { let mut reg_data = self.read_reg(reg)?; let result = f(&mut reg_data); self.write_reg(reg, &mut reg_data)?; Ok(result) } pub fn reset(&mut self) -> Result<(), SPI::Error> { let mut buf = [0xFFu8; 8]; let _ = self.nss.set_low(); let result = self.spi.transfer(&mut buf); let _ = self.nss.set_high(); result?; Ok(()) } fn transfer<'w>(&mut self, addr: u8, reg_data: &'w mut [u8], checksum: Option) -> Result, SPI::Error> { let mut addr_buf = [addr]; let _ = self.nss.set_low(); let result = match self.spi.transfer(&mut addr_buf) { Ok(_) => self.spi.transfer(reg_data), Err(e) => Err(e), }; let result = match (result, checksum) { (Ok(_),None) => Ok(None), (Ok(_), Some(checksum_out)) => { let mut checksum_buf = [checksum_out; 1]; match self.spi.transfer(&mut checksum_buf) { Ok(_) => Ok(Some(checksum_buf[0])), Err(e) => Err(e), } } (Err(e), _) => Err(e), }; let _ = self.nss.set_high(); result } }