Fixed bug with gradient rotation.

JuliaHealth · Oct 21, 2021 · c370897 · c370897 · cncastillo · Oct 21, 2021
1 parent 20a1b40
commit c370897
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/src/Grad.jl b/src/Grad.jl
@@ -39,6 +39,10 @@ end
 #Gradient operations
 *(x::Grad,α::Real) = Grad(α*x.A,x.T)
 *(α::Real,x::Grad) = Grad(α*x.A,x.T)
+*(A::Matrix{Float64},GR::Matrix{Grad}) = begin
+	N, M = size(GR)
+	[sum(A[i,1:N] .* GR[:,j]) for i=1:N, j=1:M]
+end
 import Base.zero
 zero(::Grad) = Grad(0,0)
 /(x::Grad,α::Real) = Grad(x.A/α,x.T)

diff --git a/test/runtests.jl b/test/runtests.jl
@@ -40,7 +40,8 @@ end
     θ = π*t
     R = MRIsim.rotz(θ)
     s2 = R*s
-    @test s2.GR.A ≈ [A1*cos(θ)-A2*sin(θ); A1*sin(θ)+A2*cos(θ)]
+    GR2 = R*s.GR.A
+    @test s2.GR.A ≈ GR2
     # Rotation 3D case
     T, t1, t2, t3 = rand(4)
     N = 100