use crate::profile::typedef::{self, FitBaseType};
use crate::proto::*;
use alloc::vec::Vec;
#[cfg(feature = "serde")]
use serde::{Deserialize, Serialize, Serializer, ser::SerializeStruct};
#[cfg_attr(feature = "serde", derive(Deserialize), serde(from = "De"))]
#[derive(Debug, Clone)]
pub struct MesgCapabilities {
pub message_index: typedef::MessageIndex,
pub file: typedef::File,
pub mesg_num: typedef::MesgNum,
pub count_type: typedef::MesgCount,
pub count: u16,
pub unknown_fields: Vec<Field>,
pub developer_fields: Vec<DeveloperField>,
}
impl MesgCapabilities {
pub const MESSAGE_INDEX: u8 = 254;
pub const FILE: u8 = 0;
pub const MESG_NUM: u8 = 1;
pub const COUNT_TYPE: u8 = 2;
pub const COUNT: u8 = 3;
pub const fn new() -> Self {
Self {
message_index: typedef::MessageIndex(u16::MAX),
file: typedef::File(u8::MAX),
mesg_num: typedef::MesgNum(u16::MAX),
count_type: typedef::MesgCount(u8::MAX),
count: u16::MAX,
unknown_fields: Vec::new(),
developer_fields: Vec::new(),
}
}
fn count_valid_fields(&self) -> usize {
(self.message_index.0 != u16::MAX) as usize
+ (self.file.0 != u8::MAX) as usize
+ (self.mesg_num.0 != u16::MAX) as usize
+ (self.count_type.0 != u8::MAX) as usize
+ (self.count != u16::MAX) as usize
}
}
impl Default for MesgCapabilities {
fn default() -> Self {
Self::new()
}
}
impl From<&Message> for MesgCapabilities {
fn from(mesg: &Message) -> Self {
const KNOWN_NUMS: [u64; 4] = [15, 0, 0, 4611686018427387904];
let mut n = 0u64;
for field in &mesg.fields {
n += (KNOWN_NUMS[field.num as usize >> 6] >> (field.num & 63)) & 1 ^ 1
}
let mut v = Self::new();
v.unknown_fields = Vec::<Field>::with_capacity(n as usize);
v.developer_fields = mesg.developer_fields.clone();
for field in &mesg.fields {
match field.num {
254 => v.message_index = typedef::MessageIndex(field.value.as_u16()),
0 => v.file = typedef::File(field.value.as_u8()),
1 => v.mesg_num = typedef::MesgNum(field.value.as_u16()),
2 => v.count_type = typedef::MesgCount(field.value.as_u8()),
3 => v.count = field.value.as_u16(),
_ => v.unknown_fields.push(field.clone()),
};
}
v
}
}
impl From<MesgCapabilities> for Message {
fn from(m: MesgCapabilities) -> Self {
let mut fields =
Vec::<Field>::with_capacity(m.count_valid_fields() + m.unknown_fields.len());
if m.message_index.0 != u16::MAX {
fields.push(Field {
num: 254,
base_type: FitBaseType::UINT16,
value: Value::Uint16(m.message_index.0),
is_expanded: false,
});
};
if m.file.0 != u8::MAX {
fields.push(Field {
num: 0,
base_type: FitBaseType::ENUM,
value: Value::Uint8(m.file.0),
is_expanded: false,
});
};
if m.mesg_num.0 != u16::MAX {
fields.push(Field {
num: 1,
base_type: FitBaseType::UINT16,
value: Value::Uint16(m.mesg_num.0),
is_expanded: false,
});
};
if m.count_type.0 != u8::MAX {
fields.push(Field {
num: 2,
base_type: FitBaseType::ENUM,
value: Value::Uint8(m.count_type.0),
is_expanded: false,
});
};
if m.count != u16::MAX {
fields.push(Field {
num: 3,
base_type: FitBaseType::UINT16,
value: Value::Uint16(m.count),
is_expanded: false,
});
};
fields.extend_from_slice(&m.unknown_fields);
Self {
header: 0,
num: typedef::MesgNum::MESG_CAPABILITIES,
fields,
developer_fields: m.developer_fields,
}
}
}
#[cfg(feature = "serde")]
impl Serialize for MesgCapabilities {
fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
where
S: Serializer,
{
let n = self.count_valid_fields() + 2;
let mut state = serializer.serialize_struct("MesgCapabilities", n)?;
if self.message_index.0 != u16::MAX {
state.serialize_field("message_index", &self.message_index)?;
}
if self.file.0 != u8::MAX {
state.serialize_field("file", &self.file)?;
}
if self.mesg_num.0 != u16::MAX {
state.serialize_field("mesg_num", &self.mesg_num)?;
}
if self.count_type.0 != u8::MAX {
state.serialize_field("count_type", &self.count_type)?;
}
if self.count != u16::MAX {
state.serialize_field("count", &self.count)?;
}
if !self.unknown_fields.is_empty() {
state.serialize_field("unknown_fields", &self.unknown_fields)?;
}
if !self.developer_fields.is_empty() {
state.serialize_field("developer_fields", &self.developer_fields)?;
}
state.end()
}
}
#[cfg(feature = "serde")]
#[cfg_attr(feature = "serde", derive(Deserialize), serde(default))]
struct De {
message_index: typedef::MessageIndex,
file: typedef::File,
mesg_num: typedef::MesgNum,
count_type: typedef::MesgCount,
count: u16,
unknown_fields: Vec<Field>,
developer_fields: Vec<DeveloperField>,
}
#[cfg(feature = "serde")]
impl From<De> for MesgCapabilities {
fn from(m: De) -> Self {
Self {
message_index: m.message_index,
file: m.file,
mesg_num: m.mesg_num,
count_type: m.count_type,
count: m.count,
unknown_fields: m.unknown_fields,
developer_fields: m.developer_fields,
}
}
}
#[cfg(feature = "serde")]
impl Default for De {
fn default() -> Self {
Self {
message_index: typedef::MessageIndex(u16::MAX),
file: typedef::File(u8::MAX),
mesg_num: typedef::MesgNum(u16::MAX),
count_type: typedef::MesgCount(u8::MAX),
count: u16::MAX,
unknown_fields: Vec::new(),
developer_fields: Vec::new(),
}
}
}