mirror of
https://github.com/JuliaFEM/JuliaFEM.jl.git
synced 2026-09-06 19:26:23 +00:00
Jukka Aho
126 lines
3.6 KiB
Julia
126 lines
3.6 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
|
|
dual_basis :: Bool
|
|
end
|
|
|
|
function Dirichlet()
|
|
Dirichlet(:total, true)
|
|
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::Real)
|
|
|
|
@assert problem.properties.dual_basis
|
|
|
|
# get dimension and name of PARENT field
|
|
field_dim = get_unknown_field_dimension(problem)
|
|
field_name = get_parent_field_name(problem)
|
|
gdofs = get_gdofs(element, field_dim)
|
|
|
|
De, Me, Ae = get_dualbasis(element, time)
|
|
|
|
# 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, Val{3})
|
|
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)
|
|
Phi = (Ae*N')'
|
|
|
|
g_prev = element(field_name, ip, time)
|
|
#info("g_prev = $g_prev")
|
|
for i=1:field_dim
|
|
ldofs = gdofs[i:field_dim:end]
|
|
if haskey(element, field_name*" $i")
|
|
g = element(field_name*" $i", ip, time)
|
|
if get_formulation_type(problem) == :incremental
|
|
g = g - g_prev[i]
|
|
end
|
|
#info("g_new = $g")
|
|
add!(assembly.g, ldofs, w*g*Phi')
|
|
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
|
|
=#
|