lenna_core 0.1.6

Lenna is a library for image processing algorithms and apps.
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

use serde::{Deserialize, Serialize};
use serde_json::json;
use serde_json::Value;

#[derive(Debug, Clone, PartialEq, Serialize, Deserialize)]
pub struct Config {
    pub pipeline: Vec<ProcessorConfig>,
}

#[derive(Debug, Clone, PartialEq, Serialize, Deserialize)]
pub struct ProcessorConfig {
    pub id: String,
    #[serde(flatten)]
    pub config: Value,
}

impl Config {
    pub fn add(&mut self, id: String, config: Value) {
        self.remove(id.clone());
        self.pipeline.push(ProcessorConfig { id, config })
    }

    pub fn remove(&mut self, id: String) -> Option<ProcessorConfig> {
        let position = self.pipeline.iter().position(|c| c.id == id);
        match position {
            Some(position) => Some(self.pipeline.remove(position)),
            _ => None,
        }
    }

    pub fn find(&self, id: String) -> Option<&ProcessorConfig> {
        let config = self.pipeline.iter().find(|&c| c.id == id);
        match config {
            Some(config) => Some(&config),
            _ => None,
        }
    }
}

impl Default for Config {
    fn default() -> Self {
        let config = json!({
            "width": 100,
            "height": 100
        });

        let resize = ProcessorConfig {
            id: "resize".to_string(),
            config,
        };
        Config {
            pipeline: vec![resize],
        }
    }
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn default() {
        let config = Config::default();
        let cloned = config.clone();
        assert_eq!(config, cloned);
    }

    #[test]
    fn find() {
        let config = Config::default();
        let resize_config = config.find("resize".into());
        assert_eq!(resize_config, Some(config.pipeline.get(0).unwrap()));
    }

    #[test]
    fn add() {
        let mut config = Config::default();
        let values = json!({
            "bar": 42
        });
        config.add("foo".to_string(), values);
        assert_eq!(config.pipeline.len(), 2);
    }

    #[test]
    fn remove() {
        let mut config = Config::default();
        config.remove("resize".to_string());
        assert_eq!(config.pipeline.len(), 0);
    }
}