Add assemble function for non-square matrices

src/assembler.jl

@@ -50,6 +50,27 @@ function assemble!(a::Assembler{T}, dofs::AbstractVector{Int}, Ke::AbstractMatri
     end
 end
 
+"""
+    assemble!(a::Assembler, rowdofs, coldofs, Ke)
+
+Assembles the matrix `Ke` into `a` according to the dofs specified by `rowdofs` and `coldofs`.
+"""
+function assemble!(a::Ferrite.Assembler{T}, rowdofs::AbstractVector{Int}, coldofs::AbstractVector{Int}, Ke::AbstractMatrix{T}) where {T}
+    nrows = length(rowdofs)
+    ncols = length(coldofs)
+
+    @assert(size(Ke,1) == nrows)
+    @assert(size(Ke,2) == ncols)
+
+    append!(a.V, Ke)
+    @inbounds for i in 1:ncols
+        for j in 1:nrows
+            push!(a.J, coldofs[i])
+            push!(a.I, rowdofs[j])
+        end
+    end
+end
+
 """
     end_assemble(a::Assembler) -> K
 

test/test_assemble.jl

@@ -18,4 +18,14 @@
     @test K[1,1] == Ke[1,1]
     @test K[1,5] == Ke[1,3]
     @test K[5,1] == Ke[3,1]
+
+    # assemble with different row and col dofs
+    rdofs = [1,4,6]
+    cdofs = [1,7]
+    a = start_assemble()
+    Ke = rand(length(rdofs), length(cdofs))
+    assemble!(a, rdofs, cdofs, Ke)
+    K = end_assemble(a)
+    @test (K[rdofs,cdofs] .== Ke) |> all
+
 end