mirror of
https://github.com/JuliaFEM/JuliaFEM.jl.git
synced 2026-08-19 11:43:47 +00:00
Jukka Aho
81 lines
2.4 KiB
Julia
81 lines
2.4 KiB
Julia
# This file is a part of JuliaFEM.
|
|
# License is MIT: see https://github.com/JuliaFEM/JuliaFEM.jl/blob/master/LICENSE.md
|
|
|
|
module TestAutoDiffWeakForm
|
|
|
|
using JuliaFEM.Test
|
|
using JuliaFEM
|
|
using JuliaFEM: Quad4, Equation, IntegrationPoint, assemble!, Assembly,
|
|
solve!, get_field, get_element, get_basis,
|
|
grad, get_default_integration_points
|
|
|
|
""" Plane stress formulation for 4-node bilinear element. """
|
|
type CPS4 <: Equation
|
|
element :: Quad4
|
|
integration_points :: Vector{IntegrationPoint}
|
|
end
|
|
|
|
function JuliaFEM.get_unknown_field_name(equation::CPS4)
|
|
return "displacement"
|
|
end
|
|
|
|
function CPS4(element::Quad4)
|
|
integration_points = get_default_integration_points(element)
|
|
if !haskey(element, "displacement")
|
|
element["displacement"] = 0.0 => Vector{Float64}[[0.0,0.0], [0.0,0.0], [0.0,0.0], [0.0,0.0]]
|
|
end
|
|
CPS4(element, integration_points)
|
|
end
|
|
|
|
function Base.size(eq::CPS4)
|
|
return (2, 4)
|
|
end
|
|
|
|
function JuliaFEM.get_residual_vector(equation::CPS4, ip, time; variation=nothing)
|
|
element = get_element(equation)
|
|
basis = get_basis(element)
|
|
dbasis = grad(basis)
|
|
|
|
# material parameters
|
|
E = basis("youngs modulus", ip, time)
|
|
nu = basis("poissons ratio", ip, time)
|
|
mu = E/(2*(1+nu))
|
|
la = E*nu/((1+nu)*(1-2*nu))
|
|
la = 2*la*mu/(la + 2*mu) # <- correction for 2d
|
|
|
|
# elasticity formulation
|
|
u = basis("displacement", ip, time, variation)
|
|
gradu = dbasis("displacement", ip, time, variation)
|
|
F = I + gradu
|
|
b = basis("displacement volume load", ip, time)
|
|
E = 1/2*(F'*F - I)
|
|
S = la*trace(E)*I + 2*mu*E
|
|
P = F*S
|
|
|
|
# residual vector
|
|
r_int = P*dbasis(ip,time)
|
|
r_ext = b*basis(ip,time)
|
|
r = r_int - r_ext
|
|
return vec(r)
|
|
end
|
|
|
|
function test_residual_form()
|
|
# create model -- start
|
|
element = Quad4([1, 2, 3, 4])
|
|
element["geometry"] = Vector[[0.0,0.0], [10.0,0.0], [10.0,1.0], [0.0,1.0]]
|
|
element["youngs modulus"] = 500.0
|
|
element["poissons ratio"] = 0.3
|
|
element["displacement volume load"] = Vector[[0.0,-10.0], [0.0,-10.0], [0.0,-10.0], [0.0,-10.0]]
|
|
equation = CPS4(element)
|
|
# create model -- end
|
|
|
|
free_dofs = [3, 4, 5, 6]
|
|
solve!(equation, free_dofs, 0.0) # launch a newton solver for single element
|
|
disp = get_basis(element)("displacement", [1.0, 1.0], 0.0)[2]
|
|
println("displacement at tip: $disp")
|
|
# verified using Code Aster.
|
|
@test isapprox(disp, -8.77303119819776E+00)
|
|
end
|
|
|
|
end
|