sunspec_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).try_into().unwrap();
24        let fbytes: Vec<u8> = bytes.iter().filter(|b| **b != 0).map(|b| *b).collect();
25        return String::from_utf8(fbytes).unwrap();
26    }
27
28    fn encode(data: String) -> Vec<u16> {
29        return 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).try_into().unwrap();
36        let fbytes: Vec<u8> = bytes.iter().filter(|b| **b != 0).map(|b| *b).collect();
37        let data: Point<String> = Point { name: "", offset: 0, length: 0, write_access: false, value: String::from_utf8(fbytes).unwrap() };
38        return 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        return data[0] as i16;
53    }
54
55    fn encode(data: i16) -> Vec<u16> {
56        return 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        return i32::from_be_bytes(bytes);
64    }
65
66    fn encode(data: i32) -> Vec<u16> {
67        return 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        return i64::from_be_bytes(bytes);
75    }
76
77    fn encode(data: i64) -> Vec<u16> {
78        return to_u16_vector(&data.to_be_bytes());
79    }
80}
81
82impl PointType<u16> for u16 {
83    fn decode(data: Vec<u16>) -> u16 {
84        return data[0];
85    }
86
87    fn encode(data: u16) -> Vec<u16> {
88        return 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        return u32::from_be_bytes(bytes);
96    }
97
98    fn encode(data: u32) -> Vec<u16> {
99        return 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        return u64::from_be_bytes(bytes);
107    }
108
109    fn encode(data: u64) -> Vec<u16> {
110        return 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        return u128::from_be_bytes(bytes);
118    }
119
120    fn encode(data: u128) -> Vec<u16> {
121        return 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        return f32::from_be_bytes(bytes);
129    }
130
131    fn encode(data: f32) -> Vec<u16> {
132        return to_u16_vector(&data.to_be_bytes());
133    }
134}
135
136#[derive(Debug, Clone)]
137pub enum DataTypes {
138    SunspecString(Point<String>),
139    SunspecU16(Point<u16>),
140    SunspecU32(Point<u32>),
141    SunspecU64(Point<u64>),
142    SunspecU128(Point<u128>),
143    SunspecI16(Point<i16>),
144    SunspecI32(Point<i32>),
145    SunspecI64(Point<i64>),
146    SunspecF32(Point<f32>),
147}
148
149pub trait SunspecTypes {
150    // This new function acts as a constructor
151    fn new_string (data: &str) -> Self;
152    fn new_u16 (data: u16) -> Self;
153    fn new_u32 (data: u32) -> Self;
154    fn new_u64 (data: u64) -> Self;
155    fn new_u128 (data: u128) -> Self;
156    fn new_i16 (data: i16) -> Self;
157    fn new_i32 (data: i32) -> Self;
158    fn new_i64 (data: i64) -> Self;
159    fn new_f32 (data: f32) -> Self;
160}
161
162impl SunspecTypes for DataTypes {
163    fn new_string (data: &str) -> DataTypes {
164        DataTypes::SunspecString(Point { name: "", offset: 0, length: 0, write_access: false, value: String::from(data) } )
165    }
166    fn new_u16 (data: u16) -> DataTypes {
167        DataTypes::SunspecU16(Point { name: "", offset: 0, length: 0, write_access: false, value: data } )
168    }
169    fn new_u32 (data: u32) -> DataTypes {
170        DataTypes::SunspecU32(Point { name: "", offset: 0, length: 0, write_access: false, value: data } )
171    }
172    fn new_u64 (data: u64) -> DataTypes {
173        DataTypes::SunspecU64(Point { name: "", offset: 0, length: 0, write_access: false, value: data } )
174    }
175    fn new_u128 (data: u128) -> DataTypes {
176        DataTypes::SunspecU128(Point { name: "", offset: 0, length: 0, write_access: false, value: data } )
177    }
178    fn new_i16 (data: i16) -> DataTypes {
179        DataTypes::SunspecI16(Point { name: "", offset: 0, length: 0, write_access: false, value: data } )
180    }
181    fn new_i32 (data: i32) -> DataTypes {
182        DataTypes::SunspecI32(Point { name: "", offset: 0, length: 0, write_access: false, value: data } )
183    }
184    fn new_i64 (data: i64) -> DataTypes {
185        DataTypes::SunspecI64(Point { name: "", offset: 0, length: 0, write_access: false, value: data } )
186    }
187    fn new_f32 (data: f32) -> DataTypes {
188        DataTypes::SunspecF32(Point { name: "", offset: 0, length: 0, write_access: false, value: data } )
189    }
190}
sunspec_models/types.rs

sunspec_models/
types.rs
