shared/utils/
fragment_result_impl.rs

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
use log::{debug, error};
use serde::ser::Error;

use crate::types::messages::FragmentResult;
use crate::types::pixel_data::PixelData;
use crate::types::range::Range;
use crate::types::resolution::Resolution;
use crate::types::u8data::U8Data;
use serde_json::Value;

pub trait FragmentResultOperation {
    fn new(id: U8Data, resolution: Resolution, range: Range, pixels: PixelData) -> Self;
    fn serialize(&self) -> Result<String, serde_json::Error>;

    fn deserialize(message: &str) -> Result<Self, serde_json::Error>
    where
        Self: Sized;
}
impl FragmentResultOperation for FragmentResult {
    fn new(id: U8Data, resolution: Resolution, range: Range, pixels: PixelData) -> Self {
        Self {
            id,
            resolution,
            range,
            pixels,
        }
    }

    fn serialize(&self) -> Result<String, serde_json::Error> {
        let fragment_json = serde_json::json!({"FragmentResult": &self});
        debug!(
            "Serialized message: {}",
            serde_json::to_string(&fragment_json)?
        );
        serde_json::to_string(&fragment_json)
    }

    fn deserialize(message: &str) -> Result<Self, serde_json::Error> {
        debug!("Deserializing message: {}", message);

        let v: Value = serde_json::from_str(message)?;
        if !v["FragmentResult"].is_object() {
            error!("Invalid format: FragmentResult object not found");
            return Err(serde_json::Error::custom("Invalid format"));
        }
        serde_json::from_value(v["FragmentResult"].clone())
    }
}

#[cfg(test)]
mod fragment_result_tests {
    use crate::types::point::Point;

    use super::*;

    #[test]
    fn test_serialize() {
        let fragment_result = FragmentResult::new(
            U8Data {
                offset: 0,
                count: 0,
            },
            Resolution { nx: 0, ny: 0 },
            Range {
                min: Point { x: 0.0, y: 0.0 },
                max: Point { x: 0.0, y: 0.0 },
            },
            PixelData {
                offset: 0,
                count: 0,
            },
        );

        let result = fragment_result.serialize();
        assert_eq!(result.is_ok(), true);
    }

    #[test]
    fn test_deserialize() {
        let serialized_fragment_result = "{\"FragmentResult\":{\"id\":{\"offset\":0,\"count\":0},\"resolution\":{\"nx\":0,\"ny\":0},\"range\":{\"min\":{\"x\":0.0,\"y\":0.0},\"max\":{\"x\":0.0,\"y\":0.0}},\"pixels\":{\"offset\":0,\"count\":0}}}";
        let result = FragmentResult::deserialize(serialized_fragment_result);
        assert!(result.is_ok());
        let fragment_result = result.unwrap();
        assert_eq!(fragment_result.id.offset, 0);
        assert_eq!(fragment_result.id.count, 0);
        assert_eq!(fragment_result.resolution.nx, 0);
        assert_eq!(fragment_result.resolution.ny, 0);
        assert_eq!(fragment_result.range.min.x, 0.0);
        assert_eq!(fragment_result.range.min.y, 0.0);
        assert_eq!(fragment_result.range.max.x, 0.0);
        assert_eq!(fragment_result.range.max.y, 0.0);
        assert_eq!(fragment_result.pixels.offset, 0);
        assert_eq!(fragment_result.pixels.count, 0);
    }
}