Skip to main content

ergo_runtime/compute/implementations/sum/
impl.rs

1use std::collections::HashMap;
2
3use crate::common::Value;
4use crate::compute::{ComputeError, ComputePrimitive, ComputePrimitiveManifest, PrimitiveState};
5
6use super::manifest::sum_manifest;
7
8pub struct Sum {
9    manifest: ComputePrimitiveManifest,
10}
11
12impl Sum {
13    pub fn new() -> Self {
14        Self {
15            manifest: sum_manifest(),
16        }
17    }
18}
19
20impl Default for Sum {
21    fn default() -> Self {
22        Self::new()
23    }
24}
25
26impl ComputePrimitive for Sum {
27    fn manifest(&self) -> &ComputePrimitiveManifest {
28        &self.manifest
29    }
30
31    fn compute(
32        &self,
33        inputs: &HashMap<String, Value>,
34        _parameters: &HashMap<String, Value>,
35        _state: Option<&mut PrimitiveState>,
36    ) -> Result<HashMap<String, Value>, ComputeError> {
37        let series = inputs
38            .get("series")
39            .and_then(|v| v.as_series())
40            .expect("missing required series input 'series'");
41
42        let sum: f64 = series.iter().copied().sum();
43        if !sum.is_finite() {
44            return Err(ComputeError::NonFiniteResult);
45        }
46
47        Ok(HashMap::from([("result".to_string(), Value::Number(sum))]))
48    }
49}