use crate::{
calculate_altitude, registers::*, Address, CalibrationCoefficients, Configuration, Error,
Measurement,
};
use defmt::{debug, trace};
use embedded_hal::{delay::DelayNs, i2c::I2c};
use uom::si::f32::{Length, Pressure, ThermodynamicTemperature};
impl CalibrationCoefficients {
fn try_from_i2c_sync<I: embedded_hal::i2c::I2c>(
address: Address,
i2c: &mut I,
) -> Result<Self, Error<I::Error>> {
let mut calibration_coefficient_regs = [0; 21];
i2c.write_read(
address.into(),
&Self::write_read_write_transaction(),
&mut calibration_coefficient_regs,
)
.map_err(Error::I2c)?;
Ok(Self::from_registers(&calibration_coefficient_regs))
}
}
pub struct Bmp390<I> {
i2c: I,
address: Address,
coefficients: CalibrationCoefficients,
altitude_reference: Length,
}
impl<I, E> Bmp390<I>
where
I: I2c<Error = E>,
{
pub fn try_new<D: DelayNs>(
mut i2c: I,
address: Address,
mut delay: D,
config: &Configuration,
) -> Result<Self, Error<E>> {
delay.delay_ms(2);
let mut data = [0; 2];
i2c.write_read(
address.into(),
&[super::Register::CHIP_ID.into()],
&mut data,
)
.map_err(Error::I2c)?;
let chip_id = data[0];
let rev_id = data[1];
debug!("CHIP_ID = {=u8:#04x}; REV_ID = {=u8:#04x}", chip_id, rev_id);
if chip_id != 0x60 {
return Err(Error::WrongChip(chip_id));
}
let mut data = [0; 2];
i2c.write_read(address.into(), &[super::Register::EVENT.into()], &mut data)
.map_err(Error::I2c)?;
i2c.write(address.into(), &config.to_write_bytes())
.map_err(Error::I2c)?;
let mut err_reg = [0; 1];
i2c.write_read(
address.into(),
&[super::Register::ERR_REG.into()],
&mut err_reg,
)
.map_err(Error::I2c)
.and_then(move |_| {
let err_reg = super::ErrReg::from(err_reg[0]);
if err_reg.fatal_err {
return Err(Error::Fatal);
} else if err_reg.cmd_err {
return Err(Error::Command);
} else if err_reg.conf_err {
return Err(Error::Configuration);
} else {
Ok(())
}
})?;
let coefficients = CalibrationCoefficients::try_from_i2c_sync(address, &mut i2c)?;
Ok(Self::new_with_coefficients(i2c, address, coefficients))
}
fn new_with_coefficients(
i2c: I,
address: Address,
coefficients: CalibrationCoefficients,
) -> Self {
Self {
i2c,
address,
coefficients,
altitude_reference: Length::new::<uom::si::length::meter>(0.0),
}
}
pub fn temperature(&mut self) -> Result<ThermodynamicTemperature, Error<E>> {
let write = &[Register::DATA_3.into()];
let mut read = [0; 3];
self.i2c
.write_read(self.address.into(), write, &mut read)
.map_err(Error::I2c)?;
let temperature = u32::from(read[0]) | u32::from(read[1]) << 8 | u32::from(read[2]) << 16;
let temperature = self.coefficients.compensate_temperature(temperature);
Ok(temperature)
}
pub fn temperature_pressure(
&mut self,
) -> Result<(ThermodynamicTemperature, Pressure), Error<E>> {
let write = &[Register::DATA_0.into()];
let mut read = [0; 6];
self.i2c
.write_read(self.address.into(), write, &mut read)
.map_err(Error::I2c)?;
trace!("DATA = {=[u8]:#04x}", read);
let temperature_raw =
u32::from(read[3]) | u32::from(read[4]) << 8 | u32::from(read[5]) << 16;
let pressure_raw = u32::from(read[0]) | u32::from(read[1]) << 8 | u32::from(read[2]) << 16;
let temperature = self.coefficients.compensate_temperature(temperature_raw);
let pressure = self
.coefficients
.compensate_pressure(temperature, pressure_raw);
Ok((temperature, pressure))
}
pub fn pressure(&mut self) -> Result<Pressure, Error<E>> {
self.temperature_pressure().map(|(_, pressure)| pressure)
}
pub fn measure(&mut self) -> Result<Measurement, Error<E>> {
let (temperature, pressure) = self.temperature_pressure()?;
Ok(Measurement {
temperature,
pressure,
altitude: calculate_altitude(pressure, self.altitude_reference),
})
}
pub fn set_reference_altitude(&mut self, altitude: Length) {
self.altitude_reference = altitude;
}
pub fn altitude(&mut self) -> Result<Length, Error<E>> {
let pressure = self.pressure()?;
Ok(calculate_altitude(pressure, self.altitude_reference))
}
}
#[cfg(test)]
mod tests {
extern crate std;
use embedded_hal_mock::eh1::delay::{CheckedDelay, NoopDelay, Transaction as DelayTransaction};
use embedded_hal_mock::eh1::i2c::{Mock, Transaction as I2cTransaction};
use std::prelude::rust_2021::*;
use std::vec;
use uom::si::length::meter;
use uom::si::pressure::pascal;
use uom::si::thermodynamic_temperature::degree_celsius;
use uom::ConstZero;
use crate::{ErrReg, Event, IntStatus};
use super::*;
const PRESSURE_TEMPERATURE_BYTES: [u8; 6] = [0xcb, 0xb3, 0x6b, 0xd1, 0xba, 0x82];
fn expected_pressure() -> Pressure {
Pressure::new::<pascal>(98370.55)
}
const TEMPERATURE_BYTES: [u8; 3] = [0xd1, 0xba, 0x82];
fn expected_temperature() -> ThermodynamicTemperature {
ThermodynamicTemperature::new::<degree_celsius>(25.770_746)
}
fn expected_altitude() -> Length {
Length::new::<meter>(248.78754)
}
fn get_try_new_transactions(
addr: Address,
configuration: &Configuration,
err_reg: &ErrReg,
event: &Event,
int_status: &IntStatus,
) -> [I2cTransaction; 5] {
[
I2cTransaction::write_read(
addr.into(),
vec![Register::CHIP_ID.into()],
vec![0x60, 0x01],
),
I2cTransaction::write_read(
addr.into(),
vec![Register::EVENT.into()],
vec![u8::from(*event), u8::from(*int_status)],
),
I2cTransaction::write(addr.into(), configuration.to_write_bytes().to_vec()),
I2cTransaction::write_read(
addr.into(),
vec![Register::ERR_REG.into()],
vec![u8::from(*err_reg)],
),
I2cTransaction::write_read(
addr.into(),
CalibrationCoefficients::write_read_write_transaction().to_vec(),
vec![0; 21],
),
]
}
#[test]
fn test_try_new() {
let addr = Address::Up;
let config = Configuration::default();
let expectations = get_try_new_transactions(addr, &config, &0.into(), &0.into(), &0.into());
let mut i2c = Mock::new(&expectations);
let mut delay = CheckedDelay::new(&[
DelayTransaction::delay_ms(2), ]);
let _bmp390 = Bmp390::try_new(i2c.clone(), addr, delay.clone(), &config).unwrap();
delay.done();
i2c.done();
}
#[test]
fn test_reads_temperature_and_compensates() {
let addr = Address::Up;
let expectations = [I2cTransaction::write_read(
addr.into(),
vec![Register::DATA_3.into()],
TEMPERATURE_BYTES.to_vec(),
)];
let mut i2c = Mock::new(&expectations);
let mut bmp390 =
Bmp390::new_with_coefficients(i2c.clone(), addr, CalibrationCoefficients::default());
let temperature = bmp390.temperature().unwrap();
assert_eq!(temperature, expected_temperature());
i2c.done();
}
#[test]
fn test_reads_pressure() {
let addr = Address::Up;
let expectations = [I2cTransaction::write_read(
addr.into(),
vec![Register::DATA_0.into()],
PRESSURE_TEMPERATURE_BYTES.to_vec(),
)];
let mut i2c = Mock::new(&expectations);
let mut bmp390 =
Bmp390::new_with_coefficients(i2c.clone(), addr, CalibrationCoefficients::default());
let pressure = bmp390.pressure().unwrap();
assert_eq!(pressure, expected_pressure());
i2c.done();
}
#[test]
fn test_reads_temperature_pressure() {
let addr = Address::Up;
let expectations = [I2cTransaction::write_read(
addr.into(),
vec![Register::DATA_0.into()],
PRESSURE_TEMPERATURE_BYTES.to_vec(),
)];
let mut i2c = Mock::new(&expectations);
let mut bmp390 =
Bmp390::new_with_coefficients(i2c.clone(), addr, CalibrationCoefficients::default());
let measurement = bmp390.temperature_pressure().unwrap();
assert_eq!(measurement.0, expected_temperature());
assert_eq!(measurement.1, expected_pressure());
i2c.done();
}
#[test]
fn test_altitude() {
let addr = Address::Up;
let expectations = [I2cTransaction::write_read(
addr.into(),
vec![Register::DATA_0.into()],
PRESSURE_TEMPERATURE_BYTES.to_vec(),
)];
let mut i2c = Mock::new(&expectations);
let mut bmp390 =
Bmp390::new_with_coefficients(i2c.clone(), addr, CalibrationCoefficients::default());
let altitude = bmp390.altitude().unwrap();
assert_eq!(altitude, expected_altitude());
i2c.done();
}
#[test]
fn test_measure_reads_temperature_pressure_altitude() {
let addr = Address::Up;
let expectations = [I2cTransaction::write_read(
addr.into(),
vec![Register::DATA_0.into()],
PRESSURE_TEMPERATURE_BYTES.to_vec(),
)];
let mut i2c = Mock::new(&expectations);
let mut bmp390 =
Bmp390::new_with_coefficients(i2c.clone(), addr, CalibrationCoefficients::default());
let measurement = bmp390.measure().unwrap();
assert_eq!(measurement.temperature, expected_temperature());
assert_eq!(measurement.pressure, expected_pressure());
assert_eq!(measurement.altitude, expected_altitude());
i2c.done();
}
#[test]
fn test_altitude_custom_reference() {
let addr = Address::Up;
let expectations = [I2cTransaction::write_read(
addr.into(),
vec![Register::DATA_0.into()],
PRESSURE_TEMPERATURE_BYTES.to_vec(),
)];
let mut i2c = Mock::new(&expectations);
let mut bmp390 =
Bmp390::new_with_coefficients(i2c.clone(), addr, CalibrationCoefficients::default());
bmp390.set_reference_altitude(expected_altitude());
let altitude = bmp390.altitude().unwrap();
assert_eq!(altitude, Length::ZERO);
i2c.done();
}
#[test]
fn test_chip_id_incorrect() {
let addr = Address::Up;
let mut expectations = get_try_new_transactions(
addr,
&Configuration::default(),
&0.into(),
&0.into(),
&0.into(),
)
.into_iter()
.take(1)
.collect::<Vec<_>>();
expectations[0] = I2cTransaction::write_read(
addr.into(),
vec![Register::CHIP_ID.into()],
vec![0x42, 0x01],
);
let mut i2c = Mock::new(&expectations);
let delay = NoopDelay::new();
let result = Bmp390::try_new(i2c.clone(), addr, delay, &Configuration::default());
assert!(matches!(result, Err(Error::WrongChip(0x42))));
i2c.done();
}
#[test]
fn test_fatal_error() {
let addr = Address::Up;
let fatal_err = ErrReg {
fatal_err: true,
cmd_err: false,
conf_err: false,
};
let expectations = get_try_new_transactions(
addr,
&Configuration::default(),
&fatal_err.into(),
&0.into(),
&0.into(),
)
.into_iter()
.take(4)
.collect::<Vec<_>>();
let mut i2c = Mock::new(&expectations);
let delay = NoopDelay::new();
let result = Bmp390::try_new(i2c.clone(), addr, delay, &Configuration::default());
assert!(matches!(result, Err(Error::Fatal)));
i2c.done();
}
#[test]
fn test_command_error() {
let addr = Address::Up;
let cmd_err = ErrReg {
fatal_err: false,
cmd_err: true,
conf_err: false,
};
let expectations = get_try_new_transactions(
addr,
&Configuration::default(),
&cmd_err.into(),
&0.into(),
&0.into(),
)
.into_iter()
.take(4)
.collect::<Vec<_>>();
let mut i2c = Mock::new(&expectations);
let delay = NoopDelay::new();
let result = Bmp390::try_new(i2c.clone(), addr, delay, &Configuration::default());
assert!(matches!(result, Err(Error::Command)));
i2c.done();
}
#[test]
fn test_configuration_error() {
let addr = Address::Up;
let conf_err = ErrReg {
fatal_err: false,
cmd_err: false,
conf_err: true,
};
let expectations = get_try_new_transactions(
addr,
&Configuration::default(),
&conf_err.into(),
&0.into(),
&0.into(),
)
.into_iter()
.take(4)
.collect::<Vec<_>>();
let mut i2c = Mock::new(&expectations);
let delay = NoopDelay::new();
let result = Bmp390::try_new(i2c.clone(), addr, delay, &Configuration::default());
assert!(matches!(result, Err(Error::Configuration)));
i2c.done();
}
#[test]
fn test_any_other_error() {
let addr = Address::Up;
for err_reg_bits in 0..=7 {
let err_reg = ErrReg::from(err_reg_bits);
if err_reg.fatal_err || err_reg.cmd_err || err_reg.conf_err {
continue;
}
let mut expectations = get_try_new_transactions(
addr,
&Configuration::default(),
&0.into(),
&0.into(),
&0.into(),
);
expectations[3] = I2cTransaction::write_read(
addr.into(),
vec![Register::ERR_REG.into()],
vec![err_reg_bits],
);
let mut i2c = Mock::new(&expectations);
let delay = NoopDelay::new();
let result = Bmp390::try_new(i2c.clone(), addr, delay, &Configuration::default());
assert!(
result.is_ok(),
"Unexpected error with ERR_REG = {:#010b}",
err_reg_bits
);
i2c.done();
}
}
}