mirror of
https://github.com/JuliaFEM/JuliaFEM.jl.git
synced 2026-09-12 14:23:20 +00:00
Jukka Aho
128 lines
3.7 KiB
Julia
128 lines
3.7 KiB
Julia
# This file is a part of JuliaFEM.
|
|
# License is MIT: see https://github.com/JuliaFEM/JuliaFEM.jl/blob/master/LICENSE.md
|
|
|
|
""" Here formulation is :total or :incremental meaning that we either give
|
|
constraint for total quantity u or it's increment Δu. For elasticity we are
|
|
using incremental formulation.
|
|
"""
|
|
type Dirichlet <: BoundaryProblem
|
|
formulation :: Symbol
|
|
variational :: Bool
|
|
dual_basis :: Bool
|
|
end
|
|
|
|
function Dirichlet()
|
|
Dirichlet(:incremental, true, false)
|
|
end
|
|
|
|
function get_unknown_field_name(::Type{Dirichlet})
|
|
return "reaction force"
|
|
end
|
|
|
|
function get_formulation_type(problem::Problem{Dirichlet})
|
|
return problem.properties.formulation
|
|
end
|
|
|
|
function assemble!(assembly::Assembly, problem::Problem{Dirichlet}, element::Element, time)
|
|
|
|
# get dimension and name of PARENT field
|
|
nnodes = length(element)
|
|
field_dim = get_unknown_field_dimension(problem)
|
|
field_name = get_parent_field_name(problem)
|
|
gdofs = get_gdofs(element, field_dim)
|
|
|
|
if problem.properties.dual_basis
|
|
De, Me, Ae = get_dualbasis(element, time)
|
|
else
|
|
Ae = eye(nnodes)
|
|
De = zeros(nnodes, nnodes)
|
|
for ip in get_integration_points(element)
|
|
N = element(ip, time)
|
|
detJ = element(ip, time, Val{:detJ})
|
|
De += ip.weight*N'*N*detJ
|
|
end
|
|
end
|
|
|
|
# left hand side
|
|
for i=1:field_dim
|
|
ldofs = gdofs[i:field_dim:end]
|
|
if haskey(element, field_name*" $i")
|
|
add!(assembly.C1, ldofs, ldofs, De)
|
|
add!(assembly.C2, ldofs, ldofs, De)
|
|
end
|
|
end
|
|
|
|
# right hand side
|
|
for ip in get_integration_points(element)
|
|
detJ = element(ip, time, Val{:detJ})
|
|
w = ip.weight*detJ
|
|
N = element(ip, time)
|
|
|
|
for i=1:field_dim
|
|
ldofs = gdofs[i:field_dim:end]
|
|
if haskey(element, field_name*" $i")
|
|
g = element(field_name*" $i", ip, time)
|
|
# u = u_prev + Δu ⇒ Δu = u - u_prev
|
|
if haskey(element, field_name)
|
|
g_prev = element(field_name, ip, time)
|
|
g -= g_prev[i]
|
|
end
|
|
add!(assembly.g, ldofs, w*g*Ae*N')
|
|
end
|
|
end
|
|
|
|
end
|
|
|
|
end
|
|
|
|
#=
|
|
|
|
function assemble!(assembly::Assembly, problem::Problem{DirichletProblem}, element::Element, time::Real)
|
|
|
|
# get dimension and name of PARENT field
|
|
field_dim = problem.parent_field_dim
|
|
field_name = problem.parent_field_name
|
|
|
|
gdofs = get_gdofs(element, field_dim)
|
|
for ip in get_integration_points(element, Val{2})
|
|
w = ip.weight
|
|
J = get_jacobian(element, ip, time)
|
|
JT = transpose(J)
|
|
if size(JT, 2) == 1 # plane problem
|
|
w *= norm(JT)
|
|
else
|
|
w *= norm(cross(JT[:,1], JT[:,2]))
|
|
end
|
|
N = element(ip, time)
|
|
A = w*N'*N
|
|
|
|
if haskey(element, field_name)
|
|
# add all dimensions at once if defined
|
|
# element["blaa"] = 0.0
|
|
# or
|
|
# element["blaa"] = Vector{Float64}[[0.1, 0.2], [0.3, 0.4]]
|
|
g = element(field_name, ip, time)
|
|
if length(g) != length(N)
|
|
g = g*ones(length(N))
|
|
end
|
|
for i=1:field_dim
|
|
ldofs = gdofs[i:field_dim:end]
|
|
add!(assembly.C1, ldofs, ldofs, A)
|
|
add!(assembly.C2, ldofs, ldofs, A)
|
|
end
|
|
add!(assembly.g, gdofs, w*g*N)
|
|
end
|
|
|
|
for i=1:field_dim
|
|
if haskey(element, field_name*" $i")
|
|
g = element(field_name*" $i", ip, time)
|
|
ldofs = gdofs[i:field_dim:end]
|
|
add!(assembly.C1, ldofs, ldofs, A)
|
|
add!(assembly.C2, ldofs, ldofs, A)
|
|
add!(assembly.g, ldofs, w*g*N)
|
|
end
|
|
end
|
|
end
|
|
end
|
|
=#
|