logic-mesh 1.0.0

Control logic engine using event based and reactive blocks written in Rust.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103

// Copyright (c) 2022-2023, Radu Racariu.

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};

/// Calculates the median of multiple numbers from the 16 inputs
/// this block has.
/// The operation would take into account the units of those input's values,
/// if the units are not convertible, the block would be in an error state.
#[block]
#[derive(BlockProps, Debug)]
#[dis = "Median"]
#[category = "math"]
#[input(kind = "Number", count = 16)]
pub struct Median {
    #[output(kind = "Number")]
    pub out: OutputImpl,
}

impl Block for Median {
    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(mut numbers) = convert_units(&val) {
            if !matches!(self.state(), BlockState::Running) {
                self.set_state(BlockState::Running);
            }

            numbers.sort();

            let median = if numbers.len() % 2 == 0 {
                let mid = numbers.len() / 2;
                (numbers[mid - 1].value + numbers[mid].value) / 2.0
            } else {
                let mid = numbers.len() / 2;
                numbers[mid].value
            };

            let median = if let Some(Number {
                value: _,
                unit: Some(unit),
            }) = numbers.first()
            {
                Number::make_with_unit(median, unit)
            } else {
                Number::make(median)
            };

            self.out.set(median.into());
        } else {
            self.set_state(BlockState::fault("Median: 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::Median};

    #[tokio::test]
    async fn test_median_block() {
        let mut block = Median::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());

        write_block_inputs([(block._inputs.get_mut(1).unwrap(), 1)]).await;
        block.read_inputs().await;
        write_block_inputs([(block._inputs.get_mut(2).unwrap(), 2)]).await;
        block.read_inputs().await;
        write_block_inputs([(block._inputs.get_mut(3).unwrap(), 3)]).await;

        block.execute().await;
        assert_eq!(block.out.value, 2.into());
    }
}