rustyfit 0.10.0

// Code generated by fitgen/main.go. DO NOT EDIT.

// Copyright 2025 The RustyFIT Authors. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.

#![allow(clippy::manual_range_patterns)]

use crate::profile::typedef::{self, FitBaseType};
use crate::proto::*;
use alloc::vec::Vec;
#[cfg(feature = "serde")]
use serde::{Deserialize, Serialize, Serializer, ser::SerializeStruct};

fn is_expanded(state: &[u8], num: u8) -> bool {
    match num {
        3 | 4 => (state[num as usize >> 3] >> (num & 7)) & 1 == 1,
        _ => false,
    }
}

/// Ant Tx message.
#[cfg_attr(feature = "serde", derive(Deserialize), serde(from = "De"))]
#[derive(Debug, Clone)]
pub struct AntTx {
    /// Units: s
    pub timestamp: typedef::DateTime,
    /// Scale: 32768; Units: s
    pub fractional_timestamp: u16,
    pub mesg_id: u8,
    pub mesg_data: Vec<u8>,
    pub channel_number: u8,
    pub data: Vec<u8>,
    state: [u8; 1], // Used for tracking expanded fields.
    /// unknown_fields are fields that are exist but they are not defined in Profile.xlsx
    pub unknown_fields: Vec<Field>,
    /// developer_fields are custom data fields (Added since protocol version 2.0)
    pub developer_fields: Vec<DeveloperField>,
}

impl AntTx {
    /// Value's type: `u32`; FitBaseType::UINT32; ProfileType::DateTime; Units: `s`
    pub const TIMESTAMP: u8 = 253;
    /// Value's type: `u16`; FitBaseType::UINT16; ProfileType::Uint16; Scale: `32768`; Units: `s`
    pub const FRACTIONAL_TIMESTAMP: u8 = 0;
    /// Value's type: `u8`; FitBaseType::BYTE; ProfileType::Byte
    pub const MESG_ID: u8 = 1;
    /// Value's type: `Vec<u8>`; FitBaseType::BYTE; ProfileType::Byte
    pub const MESG_DATA: u8 = 2;
    /// Value's type: `u8`; FitBaseType::UINT8; ProfileType::Uint8
    pub const CHANNEL_NUMBER: u8 = 3;
    /// Value's type: `Vec<u8>`; FitBaseType::BYTE; ProfileType::Byte
    pub const DATA: u8 = 4;

    /// Create new AntTx with all fields being set to its corresponding invalid value.
    pub const fn new() -> Self {
        Self {
            timestamp: typedef::DateTime(u32::MAX),
            fractional_timestamp: u16::MAX,
            mesg_id: u8::MAX,
            mesg_data: Vec::new(),
            channel_number: u8::MAX,
            data: Vec::new(),
            state: [0u8; 1],
            unknown_fields: Vec::new(),
            developer_fields: Vec::new(),
        }
    }

    /// Returns `fractional_timestamp` in its scaled value. It returns `None` when value is invalid.
    ///
    /// Units: s
    pub fn fractional_timestamp_scaled(&self) -> Option<f64> {
        if self.fractional_timestamp == u16::MAX {
            return None;
        }
        Some(self.fractional_timestamp as f64 / 32768.0 - 0.0)
    }

    /// Set `fractional_timestamp` with scaled value, it will automatically be converted to its corresponding integer value.
    pub fn set_fractional_timestamp_scaled(&mut self, v: f64) -> &mut Self {
        let unscaled = (v + 0.0) * 32768.0;
        if unscaled.is_nan() || unscaled.is_infinite() || unscaled > u16::MAX as f64 {
            self.fractional_timestamp = u16::MAX;
            return self;
        }
        self.fractional_timestamp = unscaled as u16;
        self
    }

    /// Marks whether given field's num is an expanded field (field that being generated through a component expansion).
    pub fn mark_as_expanded(&mut self, num: u8, flag: bool) -> bool {
        match num {
            3 | 4 => {
                if flag {
                    self.state[num as usize >> 3] |= 1 << (num & 7)
                } else {
                    self.state[num as usize >> 3] &= !(1 << (num & 7))
                }
                true
            }
            _ => false,
        }
    }

    /// checks whether given field's num is a field generated through a component expansion.
    pub fn is_expanded(&self, num: u8) -> bool {
        is_expanded(&self.state, num)
    }

    fn count_valid_fields(&self) -> usize {
        (self.timestamp.0 != u32::MAX) as usize
            + (self.fractional_timestamp != u16::MAX) as usize
            + (self.mesg_id != u8::MAX) as usize
            + (!self.mesg_data.is_empty()) as usize
            + (self.channel_number != u8::MAX) as usize
            + (!self.data.is_empty()) as usize
    }
}

impl Default for AntTx {
    fn default() -> Self {
        Self::new()
    }
}

impl From<&Message> for AntTx {
    /// from creates new AntTx struct based on given mesg.
    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.fractional_timestamp = field.value.as_u16(),
                1 => v.mesg_id = field.value.as_u8(),
                2 => v.mesg_data = field.value.to_vec_u8(),
                3 => v.channel_number = field.value.as_u8(),
                4 => v.data = field.value.to_vec_u8(),
                _ => {
                    v.unknown_fields.push(field.clone());
                    continue;
                }
            };
            if field.is_expanded && field.num < 5 {
                v.state[field.num as usize >> 3] |= 1 << (field.num & 7)
            }
        }

        v
    }
}

impl From<AntTx> for Message {
    fn from(m: AntTx) -> 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.fractional_timestamp != u16::MAX {
            fields.push(Field {
                num: 0,
                base_type: FitBaseType::UINT16,
                value: Value::Uint16(m.fractional_timestamp),
                is_expanded: false,
            });
        };
        if m.mesg_id != u8::MAX {
            fields.push(Field {
                num: 1,
                base_type: FitBaseType::BYTE,
                value: Value::Uint8(m.mesg_id),
                is_expanded: false,
            });
        };
        if !m.mesg_data.is_empty() {
            fields.push(Field {
                num: 2,
                base_type: FitBaseType::BYTE,
                value: Value::VecUint8(m.mesg_data),
                is_expanded: false,
            });
        };
        if m.channel_number != u8::MAX {
            fields.push(Field {
                num: 3,
                base_type: FitBaseType::UINT8,
                value: Value::Uint8(m.channel_number),
                is_expanded: is_expanded(&m.state, 3),
            });
        };
        if !m.data.is_empty() {
            fields.push(Field {
                num: 4,
                base_type: FitBaseType::BYTE,
                value: Value::VecUint8(m.data),
                is_expanded: is_expanded(&m.state, 4),
            });
        };

        fields.extend_from_slice(&m.unknown_fields);

        Self {
            header: 0,
            num: typedef::MesgNum::ANT_TX,
            fields,
            developer_fields: m.developer_fields,
        }
    }
}

#[cfg(feature = "serde")]
impl Serialize for AntTx {
    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("AntTx", n)?;
        if let Some(v) = self.timestamp.unix_timestamp() {
            state.serialize_field("timestamp", &v)?;
        }
        if let Some(v) = self.fractional_timestamp_scaled() {
            state.serialize_field("fractional_timestamp", &v)?;
        }
        if self.mesg_id != u8::MAX {
            state.serialize_field("mesg_id", &self.mesg_id)?;
        }
        if !self.mesg_data.is_empty() {
            state.serialize_field("mesg_data", &self.mesg_data)?;
        }
        if self.channel_number != u8::MAX {
            state.serialize_field("channel_number", &self.channel_number)?;
        }
        if !self.data.is_empty() {
            state.serialize_field("data", &self.data)?;
        }
        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>,
    fractional_timestamp: f64,
    mesg_id: u8,
    mesg_data: Vec<u8>,
    channel_number: u8,
    data: Vec<u8>,
    unknown_fields: Vec<Field>,
    developer_fields: Vec<DeveloperField>,
}

#[cfg(feature = "serde")]
impl From<De> for AntTx {
    fn from(m: De) -> Self {
        Self {
            timestamp: m.timestamp.map_or_else(
                || typedef::DateTime(u32::MAX),
                typedef::DateTime::from_unix_timestamp,
            ),
            fractional_timestamp: {
                let unscaled = (m.fractional_timestamp + 0.0) * 32768.0;
                if unscaled.is_nan() || unscaled.is_infinite() || unscaled > u16::MAX as f64 {
                    u16::MAX
                } else {
                    unscaled as u16
                }
            },
            mesg_id: m.mesg_id,
            mesg_data: m.mesg_data,
            channel_number: m.channel_number,
            data: m.data,
            state: [0u8; 1],
            unknown_fields: m.unknown_fields,
            developer_fields: m.developer_fields,
        }
    }
}

#[cfg(feature = "serde")]
impl Default for De {
    fn default() -> Self {
        Self {
            timestamp: None,
            fractional_timestamp: f64::from_bits(u64::MAX),
            mesg_id: u8::MAX,
            mesg_data: Vec::new(),
            channel_number: u8::MAX,
            data: Vec::new(),
            unknown_fields: Vec::new(),
            developer_fields: Vec::new(),
        }
    }
}