diff -r bd13c2ae3450 -r 56c8adc32b09 src/dataterm.rs --- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/src/dataterm.rs Tue Dec 31 09:34:24 2024 -0500 @@ -0,0 +1,88 @@ +/*! +Basid definitions for data terms +*/ + +use numeric_literals::replace_float_literals; + +use alg_tools::loc::Loc; +use alg_tools::euclidean::Euclidean; +use alg_tools::linops::GEMV; +pub use alg_tools::norms::L1; +use alg_tools::norms::Norm; + +use crate::types::*; +pub use crate::types::L2Squared; +use crate::measures::DiscreteMeasure; + +/// Calculates the residual $Aμ-b$. +#[replace_float_literals(F::cast_from(literal))] +pub(crate) fn calculate_residual< + F : Float, + V : Euclidean + Clone, + A : GEMV, F>, Codomain = V>, + const N : usize +>( + μ : &DiscreteMeasure, F>, + 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, + V : Euclidean + Clone, + A : GEMV, F>, Codomain = V>, + const N : usize +>( + μ : &DiscreteMeasure, F>, + μ_delta : &DiscreteMeasure, F>, + 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, F>, Codomain = V>>( + &self, + μ : &DiscreteMeasure, F>, + opA : &A, + b : &V + ) -> F + where V : Euclidean + Clone { + 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) + } +}