Add sparsity pattern generator for CSR.

Ferrite-FEM · Dec 28, 2023 · d227ef7 · d227ef7
1 parent 4a47a3f
commit d227ef7
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git a/ext/FerriteSparseMatrixCSR.jl b/ext/FerriteSparseMatrixCSR.jl
@@ -63,4 +63,11 @@ function Ferrite.zero_out_columns!(K::SparseMatrixCSR, ch::ConstraintHandler)
     end
 end
 
+function Ferrite.create_sparsity_pattern(::Type{<:SparseMatrixCSR}, dh, ch; kwargs...)
+    # create SparseMatrixCSC
+    K = create_sparsity_pattern(dh, ch; kwargs...)
+    # transform to SparseMatrixCSR
+    return SparseMatrixCSR(transpose(sparse(transpose(K)))) 
+end
+
 end
diff --git a/src/assembler.jl b/src/assembler.jl
@@ -89,7 +89,7 @@ Assembles the element residual `ge` into the global residual vector `g`.
     end
 end
 
-abstract type AbstractSparseAssembler{MT <: AbstractSparseMatrix} end
+abstract type AbstractSparseAssembler end
 
 """
     matrix_handle(a::AbstractSparseAssembler)
@@ -99,7 +99,7 @@ Return a reference to the underlying matrix/vector of the assembler.
 """
 matrix_handle, vector_handle
 
-struct AssemblerSparsityPattern{Tv, MT <: Union{AbstractSparseMatrix{Tv}, Symmetric{Tv,<:AbstractSparseMatrix{Tv}}}} <: AbstractSparseAssembler{MT}
+struct AssemblerSparsityPattern{Tv, MT <: Union{AbstractSparseMatrix{Tv}, Symmetric{Tv,<:AbstractSparseMatrix{Tv}}}} <: AbstractSparseAssembler
     K::MT
     f::Vector{Tv}
     permutation::Vector{Int}