96ffd64c0c
- used `syn::parse_quote!` instead of constructing `syn` structs directly. unsurprisingly, makes a world of difference! - code is, like, 1k lines shorter now - moved tests into dedicated crate - tweaked the tests even more, to represent complete lack of dependency on external crates
99 lines
2.8 KiB
Rust
99 lines
2.8 KiB
Rust
#![allow(missing_docs, missing_debug_implementations)]
|
|
|
|
use nacfahi::models::{
|
|
FitModel, FitModelSum,
|
|
basic::{Constant, Exponent, Gaussian, Linear},
|
|
};
|
|
use static_assertions::{assert_impl_all, assert_not_impl_all};
|
|
|
|
#[cfg(feature = "num-traits")]
|
|
use num_traits::{Float, FloatConst};
|
|
|
|
#[derive(FitModelSum)]
|
|
#[scalar_type(f64)]
|
|
pub struct CustomModel {
|
|
pub cst: Constant<f64>,
|
|
pub exp: Exponent<f64>,
|
|
}
|
|
|
|
#[derive(FitModelSum)]
|
|
pub struct CustomGenericModel<Scalar> {
|
|
pub cst: Constant<Scalar>,
|
|
pub exp: Exponent<Scalar>,
|
|
}
|
|
|
|
assert_impl_all!(CustomModel: FitModel<Scalar =f64>);
|
|
assert_not_impl_all!(CustomModel: FitModel<Scalar =f32>);
|
|
|
|
#[derive(FitModelSum)]
|
|
#[scalar_type(f64)]
|
|
pub struct CustomTupleModel(Constant<f64>, pub Exponent<f64>);
|
|
|
|
assert_impl_all!(CustomTupleModel: FitModel<Scalar =f64>);
|
|
assert_not_impl_all!(CustomTupleModel: FitModel<Scalar =f32>);
|
|
|
|
#[derive(FitModelSum)]
|
|
pub struct CustomConditionedModel<Scalar: Send>
|
|
where
|
|
Scalar: Copy + Sync,
|
|
{
|
|
pub cst: Constant<Scalar>,
|
|
pub exp: Exponent<Scalar>,
|
|
}
|
|
|
|
assert_impl_all!(CustomConditionedModel<f64>: FitModel<Scalar =f64>);
|
|
assert_impl_all!(CustomConditionedModel<f32>: FitModel<Scalar =f32>);
|
|
assert_not_impl_all!(CustomConditionedModel<i32>: FitModel);
|
|
|
|
#[derive(FitModelSum)]
|
|
pub struct WithBackground<Scalar, Signal> {
|
|
signal: Signal,
|
|
bg_linear: Linear<Scalar>,
|
|
bg_exponent: Exponent<Scalar>,
|
|
}
|
|
|
|
assert_impl_all!(WithBackground<f64, Gaussian<f64>>: FitModel<Scalar =f64>);
|
|
assert_impl_all!(WithBackground<f32, Exponent<f32>>: FitModel<Scalar =f32>);
|
|
assert_not_impl_all!(WithBackground<f32, Exponent<f64>>: FitModel);
|
|
|
|
#[cfg(feature = "num-traits")]
|
|
#[derive(FitModelSum)]
|
|
#[scalar_generic(T)]
|
|
pub struct Multipeak<T: Float + FloatConst + core::iter::Sum, const N: usize, BG> {
|
|
bg: BG,
|
|
signal: [Gaussian<T>; N],
|
|
}
|
|
|
|
#[cfg(feature = "num-traits")]
|
|
assert_impl_all!(Multipeak<f64, 0, Linear<f64>>: FitModel<Scalar =f64>);
|
|
#[cfg(feature = "num-traits")]
|
|
assert_not_impl_all!(Multipeak<f64, 0, Linear<f32>>: FitModel);
|
|
|
|
#[cfg(feature = "num-traits")]
|
|
assert_impl_all!(Multipeak<f64, 10, Linear<f64>>: FitModel<Scalar =f64>);
|
|
#[cfg(feature = "num-traits")]
|
|
assert_not_impl_all!(Multipeak<f64, 10, Linear<f32>>: FitModel);
|
|
|
|
#[cfg(feature = "num-traits")]
|
|
#[derive(FitModelSum)]
|
|
pub struct BiMulti<
|
|
Scalar: Float + FloatConst + core::iter::Sum,
|
|
const EXPS: usize,
|
|
const PEAKS: usize,
|
|
> {
|
|
pub bg: [Exponent<Scalar>; EXPS],
|
|
peaks: [Gaussian<Scalar>; PEAKS],
|
|
}
|
|
|
|
#[cfg(feature = "num-traits")]
|
|
assert_impl_all!(BiMulti<f64, 0, 0>: FitModel<Scalar = f64>);
|
|
#[cfg(feature = "num-traits")]
|
|
assert_impl_all!(BiMulti<f64, 1, 5>: FitModel<Scalar = f64>);
|
|
#[cfg(feature = "num-traits")]
|
|
assert_impl_all!(BiMulti<f64, 12, 5>: FitModel<Scalar =f64>);
|
|
|
|
#[derive(FitModelSum)]
|
|
#[scalar_type(f64)]
|
|
pub struct UnitModel;
|
|
|
|
assert_impl_all!(UnitModel: FitModel<Scalar = f64>);
|