diff -r 6105b5cd8d89 -r f0e8704d3f0e src/dataterm.rs --- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/src/dataterm.rs Mon Feb 17 13:54:53 2025 -0500 @@ -0,0 +1,94 @@ +/*! +Basid definitions for data terms +*/ + +use numeric_literals::replace_float_literals; + +use alg_tools::euclidean::Euclidean; +use alg_tools::linops::GEMV; +pub use alg_tools::norms::L1; +use alg_tools::norms::Norm; +use alg_tools::instance::{Instance, Space}; + +use crate::types::*; +pub use crate::types::L2Squared; +use crate::measures::RNDM; + +/// Calculates the residual $Aμ-b$. +#[replace_float_literals(F::cast_from(literal))] +pub(crate) fn calculate_residual< + X : Space, + I : Instance, + F : Float, + V : Euclidean + Clone, + A : GEMV, +>( + μ : I, + opA : &A, + b : &V +) -> V { + let mut r = b.clone(); + opA.gemv(&mut r, 1.0, μ, -1.0); + r +} + +/// Calculates the residual $A(μ+μ_delta)-b$. +#[replace_float_literals(F::cast_from(literal))] +pub(crate) fn calculate_residual2< + F : Float, + X : Space, + I : Instance, + J : Instance, + V : Euclidean + Clone, + A : GEMV, +>( + μ : I, + μ_delta : J, + opA : &A, + b : &V +) -> V { + let mut r = b.clone(); + opA.gemv(&mut r, 1.0, μ, -1.0); + opA.gemv(&mut r, 1.0, μ_delta, 1.0); + r +} + + +/// Trait for data terms +#[replace_float_literals(F::cast_from(literal))] +pub trait DataTerm { + /// Calculates $F(y)$, where $F$ is the data fidelity. + fn calculate_fit(&self, _residual : &V) -> F; + + /// Calculates $F(Aμ-b)$, where $F$ is the data fidelity. + fn calculate_fit_op, Codomain = V>>( + &self, + μ : I, + opA : &A, + b : &V + ) -> F + where + V : Euclidean + Clone, + I : Instance>, + { + let r = calculate_residual(μ, opA, b); + self.calculate_fit(&r) + } +} + +impl, const N : usize> +DataTerm +for L2Squared { + fn calculate_fit(&self, residual : &V) -> F { + residual.norm2_squared_div2() + } +} + + +impl + Norm, const N : usize> +DataTerm +for L1 { + fn calculate_fit(&self, residual : &V) -> F { + residual.norm(L1) + } +}