use crate::prelude::*;
#[derive(Copy, Clone, Debug)]
pub enum Sample {
F16(f16),
F32(f32),
U32(u32)
}
impl Sample {
pub fn f32(f32: f32) -> Self { Sample::F32(f32) }
pub fn f16(f16: f16) -> Self { Sample::F16(f16) }
pub fn u32(u32: u32) -> Self { Sample::U32(u32) }
#[inline]
pub fn to_f16(self) -> f16 {
match self {
Sample::F16(sample) => sample,
Sample::F32(sample) => f16::from_f32(sample),
Sample::U32(sample) => f16::from_f32(sample as f32),
}
}
#[inline]
pub fn to_f32(self) -> f32 {
match self {
Sample::F32(sample) => sample,
Sample::F16(sample) => sample.to_f32(),
Sample::U32(sample) => sample as f32,
}
}
#[inline]
pub fn to_u32(self) -> u32 {
match self {
Sample::F16(sample) => sample.to_f32() as u32,
Sample::F32(sample) => sample as u32,
Sample::U32(sample) => sample,
}
}
#[inline]
pub fn is_nan(self) -> bool {
match self {
Sample::F16(value) => value.is_nan(),
Sample::F32(value) => value.is_nan(),
Sample::U32(_) => false,
}
}
#[inline]
pub fn is_zero(&self) -> bool {
match *self {
Sample::F16(value) => value == f16::ZERO || value == f16::NEG_ZERO,
Sample::F32(value) => value == 0.0,
Sample::U32(value) => value == 0,
}
}
}
impl PartialEq for Sample {
fn eq(&self, other: &Self) -> bool {
match *self {
Sample::F16(num) => num == other.to_f16(),
Sample::F32(num) => num == other.to_f32(),
Sample::U32(num) => num == other.to_u32(),
}
}
}
impl Default for Sample {
fn default() -> Self { Sample::F32(0.0) }
}
impl From<f16> for Sample { #[inline] fn from(f: f16) -> Self { Sample::F16(f) } }
impl From<f32> for Sample { #[inline] fn from(f: f32) -> Self { Sample::F32(f) } }
impl From<u32> for Sample { #[inline] fn from(f: u32) -> Self { Sample::U32(f) } }
impl<T> From<Option<T>> for Sample where T: Into<Sample> + Default {
#[inline] fn from(num: Option<T>) -> Self { num.unwrap_or_default().into() }
}
impl From<Sample> for f16 { #[inline] fn from(s: Sample) -> Self { s.to_f16() } }
impl From<Sample> for f32 { #[inline] fn from(s: Sample) -> Self { s.to_f32() } }
impl From<Sample> for u32 { #[inline] fn from(s: Sample) -> Self { s.to_u32() } }