Function execute_map

src/python/py_kernel.rs:300–338

Execute a map spec: resolve named args to positional via KernelSpec.args[idx].name.

(map_spec: &PyMapSpec)

/// Execute a map spec: resolve named args to positional via KernelSpec.args[idx].name.
pub fn execute_map(map_spec: &PyMapSpec) -> crate::error::ParsecResult<Buffer> {
    let spec = &map_spec.spec.inner;
    let expr = &spec.expr;
    let mut missing: Vec<String> = spec
        .args
        .iter()
        .filter(|arg| !map_spec.args.contains_key(&arg.name))
        .map(|arg| arg.name.clone())
        .collect();
    missing.sort();
    if !missing.is_empty() {
        let message = if missing.len() == 1 {
            format!("missing arg: {}", missing[0])
        } else {
            format!("missing args: {}", missing.join(", "))
        };
        return Err(crate::error::ParsecError::ArgError(message));
    }

    let expected_names: HashSet<&str> = spec.args.iter().map(|arg| arg.name.as_str()).collect();
    let mut extra: Vec<String> = map_spec
        .args
        .keys()
        .filter(|name| !expected_names.contains(name.as_str()))
        .cloned()
        .collect();
    extra.sort();
    if !extra.is_empty() {
        return Err(crate::error::ParsecError::ArgError(format!(
            "unexpected args: {}",
            extra.join(", ")
        )));
    }

    let mut positional: Vec<ArgValue> = Vec::with_capacity(spec.args.len());
    positional.extend(spec.args.iter().map(|arg| map_spec.args[&arg.name].clone()));

    eval_elementwise(expr, &positional)
}

/// Execute a reduce spec.
pub fn execute_reduce(spec: &PyReduceSpec) -> crate::error::ParsecResult<f64> {