use uuid::Uuid;
use crate::{
base::{
block::{Block, BlockDesc, BlockProps, BlockState},
input::{InputProps, input_reader::InputReader},
output::Output,
},
blocks::utils::convert_units,
};
use libhaystack::val::{Number, Value, kind::HaystackKind};
use crate::{blocks::InputImpl, blocks::OutputImpl};
#[block]
#[derive(BlockProps, Debug)]
#[dis = "Average"]
#[category = "math"]
#[input(kind = "Number", count = 16)]
pub struct Average {
#[output(kind = "Number")]
pub out: OutputImpl,
}
impl Block for Average {
async fn execute(&mut self) {
self.read_inputs_until_ready().await;
let val = self
.inputs()
.into_iter()
.filter_map(|input| match input.get_value().as_ref() {
Some(Value::Number(num)) => Some(*num),
_ => None,
})
.collect::<Vec<Number>>();
if let Ok(numbers) = convert_units(&val) {
if !matches!(self.state(), BlockState::Running) {
self.set_state(BlockState::Running);
}
let avg = numbers.iter().fold(0.0, |acc, n| acc + n.value) / numbers.len() as f64;
let avg = if let Some(Number {
value: _,
unit: Some(unit),
}) = numbers.first()
{
Number::make_with_unit(avg, unit)
} else {
Number::make(avg)
};
self.out.set((avg).into())
} else {
self.set_state(BlockState::fault("Average: unit conversion failed"));
}
}
}
#[cfg(test)]
mod test {
use crate::base::block::test_utils::write_block_inputs;
use crate::{
base::block::Block, base::input::input_reader::InputReader, blocks::math::Average,
};
#[tokio::test]
async fn test_average_block() {
let mut block = Average::new();
write_block_inputs([(block._inputs.get_mut(0).unwrap(), 1)]).await;
block.read_inputs().await;
write_block_inputs([(block._inputs.get_mut(15).unwrap(), 9)]).await;
block.execute().await;
assert_eq!(block.out.value, 5.into());
}
}