path: root/src/metric/frequency.rs

use std::fmt::{Display, Formatter, Result};
use std::ops::{Add, Sub, AddAssign, SubAssign, Div, Mul, DivAssign, MulAssign};

use crate::metric::Unit;
use crate::metric::Primitive;

use crate::declare_unit_basics;

use crate::metric::time::Time;

#[derive(Copy, Clone, Debug, PartialOrd, PartialEq)]
pub enum Frequency {
    Base(f64, fn(f64) -> f64, &'static str),
    Canonical(f64),
    Hertz(f64),
    Becquerel(f64),
}

impl Unit<Frequency> for Frequency {
    fn to_si_unit(&self) -> Frequency {
        match self {
            Frequency::Base(_, _, _) => self.clone(),
            Frequency::Canonical(value) => Frequency::Base(value.clone(), |x| x, "Hz"),
            Frequency::Hertz(value) => Frequency::Base(value.clone(), |x| x, "Hz"),
            Frequency::Becquerel(value) => Frequency::Base(value.clone(), |x| x, "Bq"),
        }
    }
}

declare_unit_basics!(Frequency);

// 1/s = Hz
impl Div<Time> for f64 {
    type Output = Frequency;

    fn div(self, rhs: Time) -> Self::Output {
        Frequency::Canonical(self * rhs.to_si_primitive())
    }
}

// Hz * s = 1
impl Mul<Time> for Frequency {
    type Output = f64;

    fn mul(self, rhs: Time) -> Self::Output {
        self.to_si_primitive() * rhs.to_si_primitive()
    }
}

// s * Hz = 1
impl Mul<Frequency> for Time {
    type Output = f64;

    fn mul(self, rhs: Frequency) -> Self::Output {
        self.to_si_primitive() * rhs.to_si_primitive()
    }
}

pub type Activity = Frequency;