mirror of
https://github.com/JuliaFEM/JuliaFEM.jl.git
synced 2026-09-25 11:35:30 +00:00
Jukka Aho
97 lines
2.9 KiB
Julia
97 lines
2.9 KiB
Julia
# This file is a part of JuliaFEM.
|
|
# License is MIT: see https://github.com/JuliaFEM/JuliaFEM.jl/blob/master/LICENSE.md
|
|
|
|
""" Calculate nodal vector from set of elements.
|
|
|
|
For example element 1 with dofs [1, 2, 3, 4] has [1, 1, 1, 1] and
|
|
element 2 with dofs [3, 4, 5, 6] has [2, 2, 2, 2] the result will
|
|
be sparse matrix with values [1, 1, 3, 3, 2, 2].
|
|
|
|
Parameters
|
|
----------
|
|
field_name
|
|
name of field, e.g. "geometry"
|
|
field_dim
|
|
degrees of freedom / node
|
|
elements
|
|
elements used to calculate vector
|
|
time
|
|
"""
|
|
function calculate_nodal_vector(field_name::ASCIIString, field_dim::Int,
|
|
elements::Vector{Element}, time::Real)
|
|
A = SparseMatrixCOO()
|
|
b = SparseMatrixCOO()
|
|
for element in elements
|
|
haskey(element, field_name) || continue
|
|
gdofs = get_gdofs(element, 1)
|
|
for ip in get_integration_points(element, Val{2})
|
|
J = get_jacobian(element, ip, time)
|
|
w = ip.weight*norm(J)
|
|
f = element(field_name, ip, time)
|
|
N = element(ip, time)
|
|
add!(A, gdofs, gdofs, w*kron(N', N))
|
|
for dim=1:field_dim
|
|
add!(b, gdofs, w*f[dim]*N, dim)
|
|
end
|
|
end
|
|
end
|
|
A = sparse(A)
|
|
b = sparse(b)
|
|
nz = sort(unique(rowvals(A)))
|
|
x = zeros(size(b)...)
|
|
x[nz, :] = A[nz,nz] \ b[nz, :]
|
|
return vec(transpose(x))
|
|
end
|
|
|
|
function calculate_rotated_nodal_vector(field_name::ASCIIString, field_dim::Int,
|
|
elements::Vector{Element}, time::Real)
|
|
A = SparseMatrixCOO()
|
|
b = SparseMatrixCOO()
|
|
for element in elements
|
|
haskey(element, field_name) || continue
|
|
gdofs = get_gdofs(element, 1)
|
|
for ip in get_integration_points(element, Val{5})
|
|
J = get_jacobian(element, ip, time)
|
|
w = ip.weight*norm(J)
|
|
Q = element("normal-tangential coordinates", ip, time)
|
|
f = element(field_name, ip, time)
|
|
f = Q'*f
|
|
N = element(ip, time)
|
|
add!(A, gdofs, gdofs, w*kron(N', N))
|
|
for dim=1:field_dim
|
|
add!(b, gdofs, w*f[dim]*N, dim)
|
|
end
|
|
end
|
|
end
|
|
A = sparse(A)
|
|
b = sparse(b)
|
|
nz = sort(unique(rowvals(A)))
|
|
x = zeros(size(b)...)
|
|
x[nz, :] = A[nz,nz] \ b[nz, :]
|
|
return vec(transpose(x))
|
|
end
|
|
|
|
""" Collect normal-tangential coordinates to rotation matrix Q.
|
|
"""
|
|
function get_rotation_matrix(elements, time)
|
|
Q = Dict{Int64, Matrix{Float64}}()
|
|
for element in elements
|
|
node_ids = get_connectivity(element)
|
|
q = element("normal-tangential coordinates", time).data
|
|
for (qi, node_id) in zip(q, node_ids)
|
|
if haskey(Q, node_id)
|
|
@assert isapprox(Q[node_id], qi)
|
|
else
|
|
Q[node_id] = qi
|
|
end
|
|
end
|
|
end
|
|
R = SparseMatrixCOO()
|
|
for (k, q) in Q
|
|
dofs = [(k-1)*2+1, (k-1)*2+2]
|
|
add!(R, dofs, dofs, q)
|
|
end
|
|
return R
|
|
end
|
|
|