Use eigen functions in StaticArrays.jl (#211)

* Use eigen functions in StaticArrays.jl * Set version to 0.15.0
KeitaNakamura · Aug 31, 2024 · 7ca98fb · 7ca98fb · KeitaNakamura · Aug 31, 2024
1 parent a530364
commit 7ca98fb
Show file tree

Hide file tree

Showing 7 changed files with 18 additions and 186 deletions.
diff --git a/Project.toml b/Project.toml
@@ -1,7 +1,7 @@
 name = "Tensorial"
 uuid = "98f94333-fa9f-48a9-ad80-1c66397b2b38"
 authors = ["Keita Nakamura <[email protected]>"]
-version = "0.14.1"
+version = "0.15.0"
 
 [deps]
 ForwardDiff = "f6369f11-7733-5829-9624-2563aa707210"

diff --git a/src/Tensorial.jl b/src/Tensorial.jl
@@ -91,7 +91,6 @@ include("permute.jl")
 include("ops.jl")
 include("continuum_mechanics.jl")
 include("inv.jl")
-include("eigen.jl")
 include("voigt.jl")
 include("ad.jl")
 # include("simd.jl")

diff --git a/src/eigen.jl b/src/eigen.jl
diff --git a/src/ops.jl b/src/ops.jl
@@ -775,6 +775,16 @@ for pv in (:true, :false)
 end
 lu(A::AbstractMat; check = true) = lu(A, Val(true); check = check)
 
+# eigen
+@inline function eigvals(x::AbstractSymmetricSecondOrderTensor; permute::Bool=true, scale::Bool=true)
+    Tensor(eigvals(Symmetric(SArray(x)); permute=permute, scale = scale))
+end
+
+@inline function eigen(x::AbstractSymmetricSecondOrderTensor; permute::Bool=true, scale::Bool=true)
+    eig = eigen(Symmetric(SArray(x)); permute=permute, scale=scale)
+    Eigen(Tensor(eig.values), Tensor(eig.vectors))
+end
+
 # svd
 struct SVD{T, TU, TS, TVt} <: Factorization{T}
     U::TU

diff --git a/test/eigen.jl b/test/eigen.jl
diff --git a/test/ops.jl b/test/ops.jl
@@ -320,6 +320,13 @@ end
         @test U == Fs.U
         @test p == Fs.p
     end
+    @testset "eigen" begin
+        Random.seed!(1234)
+        x = rand(SymmetricSecondOrderTensor{3})
+        @test (@inferred eigvals(x)) ≈ eigvals(Array(x))
+        @test (@inferred eigen(x)).values ≈ eigen(Array(x)).values
+        @test (@inferred eigen(x)).vectors ≈ (@inferred eigvecs(x))
+    end
     @testset "svd" begin
         Random.seed!(1234)
         At = rand(Mat{3,3})

diff --git a/test/runtests.jl b/test/runtests.jl
@@ -19,7 +19,6 @@ include("einsum.jl")
 include("ops.jl")
 include("continuum_mechanics.jl")
 include("inv.jl")
-include("eigen.jl")
 include("voigt.jl")
 include("ad.jl")
 include("broadcast.jl")