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

use ndarray::prelude::*;
use ndarray::stack;

use smartnoise_validator::proto;
use smartnoise_validator::base::{Array, ReleaseNode};
use smartnoise_validator::errors::*;
use smartnoise_validator::utilities::array::slow_stack;

use crate::components::Evaluable;
use crate::NodeArguments;
use crate::utilities::to_nd;

impl Evaluable for proto::ColumnBind {
    fn evaluate(&self, _privacy_definition: &Option<proto::PrivacyDefinition>, arguments: NodeArguments) -> Result<ReleaseNode> {

        let arrays = arguments.into_iter()
            .map(|(_, v)| v.array()).collect::<Result<Vec<Array>>>()?;

        macro_rules! col_stack {
            ($func:ident, $method:ident) => {
                {
                    let inputs = arrays.into_iter()
                        .map(|v| v.$method().and_then(|v| to_nd(v, 2)))
                        .collect::<Result<Vec<ndarray::ArrayD<_>>>>()?;
                    $func(Axis(1), &inputs.iter().map(|v| v.view())
                        .collect::<Vec<ArrayViewD<_>>>())?.into()
                }
            }
        }

        Ok(ReleaseNode::new(match arrays.first().ok_or_else(|| "must have at least one argument")? {
            Array::Float(_) => col_stack!(stack, float),
            Array::Int(_) => col_stack!(stack, int),
            Array::Bool(_) => col_stack!(stack, bool),
            Array::Str(_) => col_stack!(slow_stack, string)
        }))
    }
}