huawei_models/
types.rs

1// The ideia for the PointType trait and the implementation for decode and encode functions
2// were copied from a different crate, available at:
3// https://github.com/lukaskirner/tokio-sunspec/blob/main/src/point.rs
4
5use crate::utils::*;
6
7pub trait PointType<T> {
8    fn decode(data: Vec<u16>) -> T;
9    fn encode(data: T) -> Vec<u16>;
10}
11
12#[derive(Debug, Clone, Copy)]
13pub struct Point<T: PointType<T>> {
14    pub name: &'static str,
15    pub offset: u16,
16    pub length: u16,
17    pub write_access: bool,
18    pub value: T,
19}
20
21impl PointType<String> for String {
22    fn decode(data: Vec<u16>) -> String {
23        let bytes: Vec<u8> = to_be_bytes(data);
24        let fbytes: Vec<u8> = bytes.iter().filter(|b| **b != 0).copied().collect();
25        String::from_utf8(fbytes).unwrap()
26    }
27
28    fn encode(data: String) -> Vec<u16> {
29        to_u16_vector(data.as_bytes())
30    }
31}
32
33impl PointType<Point<String>> for Point<String> {
34    fn decode(data: Vec<u16>) -> Point<String> {
35        let bytes: Vec<u8> = to_be_bytes(data);
36        let fbytes: Vec<u8> = bytes.iter().filter(|b| **b != 0).copied().collect();
37        let data: Point<String> = Point { name: "", offset: 0, length: 0, write_access: false, value: String::from_utf8(fbytes).unwrap() };
38        data
39    }
40
41    fn encode(data: Point<String>) -> Vec<u16> {
42        let mut regs = String::encode(data.value);
43        for _i in 0..(data.length-(regs.len() as u16)){
44            regs.push(0);
45        }
46        regs
47    }
48}
49
50impl PointType<i16> for i16 {
51    fn decode(data: Vec<u16>) -> i16 {
52        data[0] as i16
53    }
54
55    fn encode(data: i16) -> Vec<u16> {
56        vec![data as u16]
57    }
58}
59
60impl PointType<i32> for i32 {
61    fn decode(data: Vec<u16>) -> i32 {
62        let bytes = to_be_bytes(data).try_into().unwrap();
63        i32::from_be_bytes(bytes)
64    }
65
66    fn encode(data: i32) -> Vec<u16> {
67        to_u16_vector(&data.to_be_bytes())
68    }
69}
70
71impl PointType<i64> for i64 {
72    fn decode(data: Vec<u16>) -> i64 {
73        let bytes = to_be_bytes(data).try_into().unwrap();
74        i64::from_be_bytes(bytes)
75    }
76
77    fn encode(data: i64) -> Vec<u16> {
78        to_u16_vector(&data.to_be_bytes())
79    }
80}
81
82impl PointType<u16> for u16 {
83    fn decode(data: Vec<u16>) -> u16 {
84        data[0]
85    }
86
87    fn encode(data: u16) -> Vec<u16> {
88        vec![data]
89    }
90}
91
92impl PointType<u32> for u32 {
93    fn decode(data: Vec<u16>) -> u32 {
94        let bytes = to_be_bytes(data).try_into().unwrap();
95        u32::from_be_bytes(bytes)
96    }
97
98    fn encode(data: u32) -> Vec<u16> {
99        to_u16_vector(&data.to_be_bytes())
100    }
101}
102
103impl PointType<u64> for u64 {
104    fn decode(data: Vec<u16>) -> u64 {
105        let bytes = to_be_bytes(data).try_into().unwrap();
106        u64::from_be_bytes(bytes)
107    }
108
109    fn encode(data: u64) -> Vec<u16> {
110        to_u16_vector(&data.to_be_bytes())
111    }
112}
113
114impl PointType<u128> for u128 {
115    fn decode(data: Vec<u16>) -> u128 {
116        let bytes = to_be_bytes(data).try_into().unwrap();
117        u128::from_be_bytes(bytes)
118    }
119
120    fn encode(data: u128) -> Vec<u16> {
121        to_u16_vector(&data.to_be_bytes())
122    }
123}
124
125impl PointType<f32> for f32 {
126    fn decode(data: Vec<u16>) -> f32 {
127        let bytes = to_be_bytes(data).try_into().unwrap();
128        f32::from_be_bytes(bytes)
129    }
130
131    fn encode(data: f32) -> Vec<u16> {
132        to_u16_vector(&data.to_be_bytes())
133    }
134}
135
136#[derive(Debug, Clone)]
137pub enum HDataTypes {
138    HuaweiString(Point<String>),
139    HuaweiU16(Point<u16>),
140    HuaweiU32(Point<u32>),
141    HuaweiI16(Point<i16>),
142    HuaweiI32(Point<i32>),
143}
144
145pub trait HuaweiTypes {
146    // This new function acts as a constructor
147    fn new_string (data: &str) -> Self;
148    fn new_u16 (data: u16) -> Self;
149    fn new_u32 (data: u32) -> Self;
150    fn new_i16 (data: i16) -> Self;
151    fn new_i32 (data: i32) -> Self;
152}
153
154impl HuaweiTypes for HDataTypes {
155    fn new_string (data: &str) -> HDataTypes {
156        HDataTypes::HuaweiString(Point { name: "", offset: 0, length: 0, write_access: false, value: String::from(data) } )
157    }
158    fn new_u16 (data: u16) -> HDataTypes {
159        HDataTypes::HuaweiU16(Point { name: "", offset: 0, length: 0, write_access: false, value: data } )
160    }
161    fn new_u32 (data: u32) -> HDataTypes {
162        HDataTypes::HuaweiU32(Point { name: "", offset: 0, length: 0, write_access: false, value: data } )
163    }
164    fn new_i16 (data: i16) -> HDataTypes {
165        HDataTypes::HuaweiI16(Point { name: "", offset: 0, length: 0, write_access: false, value: data } )
166    }
167    fn new_i32 (data: i32) -> HDataTypes {
168        HDataTypes::HuaweiI32(Point { name: "", offset: 0, length: 0, write_access: false, value: data } )
169    }
170}
huawei_models/types.rs

huawei_models/
types.rs
