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
96
97
98

//! Direct attribute value loader.

use std::io;

use crate::{
    low::v7400::AttributeValue,
    pull_parser::{v7400::LoadAttribute, Result},
};

/// Loader for [`AttributeValue`].
///
/// [`AttributeValue`]: ../../../../low/v7400/enum.AttributeValue.html
#[derive(Default, Debug, Clone, Copy, PartialEq, Eq, PartialOrd, Ord, Hash)]
pub struct DirectLoader;

impl LoadAttribute for DirectLoader {
    type Output = AttributeValue;

    fn expecting(&self) -> String {
        "any type".into()
    }

    fn load_bool(self, v: bool) -> Result<Self::Output> {
        Ok(AttributeValue::Bool(v))
    }

    fn load_i16(self, v: i16) -> Result<Self::Output> {
        Ok(AttributeValue::I16(v))
    }

    fn load_i32(self, v: i32) -> Result<Self::Output> {
        Ok(AttributeValue::I32(v))
    }

    fn load_i64(self, v: i64) -> Result<Self::Output> {
        Ok(AttributeValue::I64(v))
    }

    fn load_f32(self, v: f32) -> Result<Self::Output> {
        Ok(AttributeValue::F32(v))
    }

    fn load_f64(self, v: f64) -> Result<Self::Output> {
        Ok(AttributeValue::F64(v))
    }

    fn load_seq_bool(
        self,
        iter: impl Iterator<Item = Result<bool>>,
        _len: usize,
    ) -> Result<Self::Output> {
        Ok(AttributeValue::ArrBool(iter.collect::<Result<_>>()?))
    }

    fn load_seq_i32(
        self,
        iter: impl Iterator<Item = Result<i32>>,
        _len: usize,
    ) -> Result<Self::Output> {
        Ok(AttributeValue::ArrI32(iter.collect::<Result<_>>()?))
    }

    fn load_seq_i64(
        self,
        iter: impl Iterator<Item = Result<i64>>,
        _len: usize,
    ) -> Result<Self::Output> {
        Ok(AttributeValue::ArrI64(iter.collect::<Result<_>>()?))
    }

    fn load_seq_f32(
        self,
        iter: impl Iterator<Item = Result<f32>>,
        _len: usize,
    ) -> Result<Self::Output> {
        Ok(AttributeValue::ArrF32(iter.collect::<Result<_>>()?))
    }

    fn load_seq_f64(
        self,
        iter: impl Iterator<Item = Result<f64>>,
        _len: usize,
    ) -> Result<Self::Output> {
        Ok(AttributeValue::ArrF64(iter.collect::<Result<_>>()?))
    }

    fn load_binary(self, mut reader: impl io::Read, len: u64) -> Result<Self::Output> {
        let mut buf = Vec::with_capacity(len as usize);
        reader.read_to_end(&mut buf)?;
        Ok(AttributeValue::Binary(buf))
    }

    fn load_string(self, mut reader: impl io::Read, len: u64) -> Result<Self::Output> {
        let mut buf = String::with_capacity(len as usize);
        reader.read_to_string(&mut buf)?;
        Ok(AttributeValue::String(buf))
    }
}