rustyfit 0.8.1

// 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(unused, clippy::manual_range_patterns)]

use crate::profile::{ProfileType, typedef};
use crate::proto::*;
use alloc::vec::Vec;

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

#[derive(Debug, Clone)]
/// AntRx is a AntRx message.
pub struct AntRx {
    /// 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 AntRx {
    /// Value's type: `u32`; Units: `s`
    pub const TIMESTAMP: u8 = 253;
    /// Value's type: `u16`; Scale: `32768`; Units: `s`
    pub const FRACTIONAL_TIMESTAMP: u8 = 0;
    /// Value's type: `u8`
    pub const MESG_ID: u8 = 1;
    /// Value's type: `Vec<u8>`
    pub const MESG_DATA: u8 = 2;
    /// Value's type: `u8`
    pub const CHANNEL_NUMBER: u8 = 3;
    /// Value's type: `Vec<u8>`
    pub const DATA: u8 = 4;

    /// Create new AntRx 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 invalid f64 when value is valid.
    pub fn fractional_timestamp_scaled(&self) -> f64 {
        if self.fractional_timestamp == u16::MAX {
            return f64::from_bits(u64::MAX);
        }
        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 AntRx {
        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 != typedef::DateTime(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 AntRx {
    fn default() -> Self {
        Self::new()
    }
}

impl From<&Message> for AntRx {
    /// from creates new AntRx 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<AntRx> for Message {
    fn from(m: AntRx) -> Self {
        let mut fields =
            Vec::<Field>::with_capacity(m.count_valid_fields() + m.unknown_fields.len());

        if m.timestamp != typedef::DateTime(u32::MAX) {
            fields.push(Field {
                num: 253,
                profile_type: ProfileType::DATE_TIME,
                value: Value::Uint32(m.timestamp.0),
                is_expanded: false,
            });
        };
        if m.fractional_timestamp != u16::MAX {
            fields.push(Field {
                num: 0,
                profile_type: ProfileType::UINT16,
                value: Value::Uint16(m.fractional_timestamp),
                is_expanded: false,
            });
        };
        if m.mesg_id != u8::MAX {
            fields.push(Field {
                num: 1,
                profile_type: ProfileType::BYTE,
                value: Value::Uint8(m.mesg_id),
                is_expanded: false,
            });
        };
        if !m.mesg_data.is_empty() {
            fields.push(Field {
                num: 2,
                profile_type: ProfileType::BYTE,
                value: Value::VecUint8(m.mesg_data),
                is_expanded: false,
            });
        };
        if m.channel_number != u8::MAX {
            fields.push(Field {
                num: 3,
                profile_type: ProfileType::UINT8,
                value: Value::Uint8(m.channel_number),
                is_expanded: is_expanded(&m.state, 3),
            });
        };
        if !m.data.is_empty() {
            fields.push(Field {
                num: 4,
                profile_type: ProfileType::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_RX,
            fields,
            developer_fields: m.developer_fields,
        }
    }
}