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 TrainingFile {
pub timestamp: typedef::DateTime,
pub r#type: typedef::File,
pub manufacturer: typedef::Manufacturer,
pub product: u16,
pub serial_number: u32,
pub time_created: typedef::DateTime,
pub unknown_fields: Vec<Field>,
pub developer_fields: Vec<DeveloperField>,
}
impl TrainingFile {
pub const TIMESTAMP: u8 = 253;
pub const TYPE: u8 = 0;
pub const MANUFACTURER: u8 = 1;
pub const PRODUCT: u8 = 2;
pub const SERIAL_NUMBER: u8 = 3;
pub const TIME_CREATED: u8 = 4;
pub const fn new() -> Self {
Self {
timestamp: typedef::DateTime(u32::MAX),
r#type: typedef::File(u8::MAX),
manufacturer: typedef::Manufacturer(u16::MAX),
product: u16::MAX,
serial_number: u32::MIN,
time_created: typedef::DateTime(u32::MAX),
unknown_fields: Vec::new(),
developer_fields: Vec::new(),
}
}
fn count_valid_fields(&self) -> usize {
(self.timestamp.0 != u32::MAX) as usize
+ (self.r#type.0 != u8::MAX) as usize
+ (self.manufacturer.0 != u16::MAX) as usize
+ (self.product != u16::MAX) as usize
+ (self.serial_number != u32::MIN) as usize
+ (self.time_created.0 != u32::MAX) as usize
}
}
impl Default for TrainingFile {
fn default() -> Self {
Self::new()
}
}
impl From<&Message> for TrainingFile {
fn from(mesg: &Message) -> Self {
const KNOWN_NUMS: [u64; 4] = [31, 0, 0, 2305843009213693952];
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 {
253 => v.timestamp = typedef::DateTime(field.value.as_u32()),
0 => v.r#type = typedef::File(field.value.as_u8()),
1 => v.manufacturer = typedef::Manufacturer(field.value.as_u16()),
2 => v.product = field.value.as_u16(),
3 => v.serial_number = field.value.as_u32z(),
4 => v.time_created = typedef::DateTime(field.value.as_u32()),
_ => v.unknown_fields.push(field.clone()),
};
}
v
}
}
impl From<TrainingFile> for Message {
fn from(m: TrainingFile) -> Self {
let mut fields =
Vec::<Field>::with_capacity(m.count_valid_fields() + m.unknown_fields.len());
if m.timestamp.0 != u32::MAX {
fields.push(Field {
num: 253,
base_type: FitBaseType::UINT32,
value: Value::Uint32(m.timestamp.0),
is_expanded: false,
});
};
if m.r#type.0 != u8::MAX {
fields.push(Field {
num: 0,
base_type: FitBaseType::ENUM,
value: Value::Uint8(m.r#type.0),
is_expanded: false,
});
};
if m.manufacturer.0 != u16::MAX {
fields.push(Field {
num: 1,
base_type: FitBaseType::UINT16,
value: Value::Uint16(m.manufacturer.0),
is_expanded: false,
});
};
if m.product != u16::MAX {
fields.push(Field {
num: 2,
base_type: FitBaseType::UINT16,
value: Value::Uint16(m.product),
is_expanded: false,
});
};
if m.serial_number != u32::MIN {
fields.push(Field {
num: 3,
base_type: FitBaseType::UINT32Z,
value: Value::Uint32(m.serial_number),
is_expanded: false,
});
};
if m.time_created.0 != u32::MAX {
fields.push(Field {
num: 4,
base_type: FitBaseType::UINT32,
value: Value::Uint32(m.time_created.0),
is_expanded: false,
});
};
fields.extend_from_slice(&m.unknown_fields);
Self {
header: 0,
num: typedef::MesgNum::TRAINING_FILE,
fields,
developer_fields: m.developer_fields,
}
}
}
#[cfg(feature = "serde")]
impl Serialize for TrainingFile {
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("TrainingFile", n)?;
if let Some(v) = self.timestamp.unix_timestamp() {
state.serialize_field("timestamp", &v)?;
}
if self.r#type.0 != u8::MAX {
state.serialize_field("type", &self.r#type)?;
}
if self.manufacturer.0 != u16::MAX {
state.serialize_field("manufacturer", &self.manufacturer)?;
}
if self.product != u16::MAX {
state.serialize_field("product", &self.product)?;
}
if self.serial_number != u32::MIN {
state.serialize_field("serial_number", &self.serial_number)?;
}
if let Some(v) = self.time_created.unix_timestamp() {
state.serialize_field("time_created", &v)?;
}
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 {
timestamp: Option<i64>,
r#type: typedef::File,
manufacturer: typedef::Manufacturer,
product: u16,
serial_number: u32,
time_created: Option<i64>,
unknown_fields: Vec<Field>,
developer_fields: Vec<DeveloperField>,
}
#[cfg(feature = "serde")]
impl From<De> for TrainingFile {
fn from(m: De) -> Self {
Self {
timestamp: m.timestamp.map_or_else(
|| typedef::DateTime(u32::MAX),
typedef::DateTime::from_unix_timestamp,
),
r#type: m.r#type,
manufacturer: m.manufacturer,
product: m.product,
serial_number: m.serial_number,
time_created: m.time_created.map_or_else(
|| typedef::DateTime(u32::MAX),
typedef::DateTime::from_unix_timestamp,
),
unknown_fields: m.unknown_fields,
developer_fields: m.developer_fields,
}
}
}
#[cfg(feature = "serde")]
impl Default for De {
fn default() -> Self {
Self {
timestamp: None,
r#type: typedef::File(u8::MAX),
manufacturer: typedef::Manufacturer(u16::MAX),
product: u16::MAX,
serial_number: u32::MIN,
time_created: None,
unknown_fields: Vec::new(),
developer_fields: Vec::new(),
}
}
}