Add weighting to rownorm²

JuliaImageRecon · Feb 19, 2024 · a0cfb72 · a0cfb72
1 parent 29aa336
commit a0cfb72
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/src/Utils.jl b/src/Utils.jl
@@ -20,7 +20,7 @@ function rownorm²(A::AbstractMatrix,row::Int)
 end
 
 rownorm²(A::AbstractLinearOperator,row) = rownorm²(Matrix(A[row, :]), 1)
-rownorm²(A::ProdOp{T, <:WeightingOp, matT}, row) where {T, matT} = rownorm²(A.B, row)
+rownorm²(A::ProdOp{T, <:WeightingOp, matT}, row) where {T, matT} = A.A.weights[row]^2*rownorm²(A.B, row)
 
 """
 This function computes the 2-norm² of a rows of S for sparse matrices.
@@ -33,9 +33,6 @@ function rownorm²(B::Transpose{T,S},row::Int) where {T,S<:SparseMatrixCSC}
 end
 
 
-solverweights(solver::S) where S <: AbstractLinearSolver = solverweights(S, solver.A) 
-solverweights(::Type{<:AbstractLinearSolver}, A) = ones(real(eltype(A)), size(A, 1))
-
 ### dot_with_matrix_row ###