mirror of
https://github.com/JuliaFEM/JuliaFEM.jl.git
synced 2026-09-24 11:16:47 +00:00
most important tests pass now
This commit is contained in:
+1
-2
@@ -13,7 +13,6 @@ autodiffcache = ForwardDiffCache()
|
||||
# export derivative, jacobian, hessian
|
||||
|
||||
include("common.jl")
|
||||
typealias Node Vector{Float64}
|
||||
|
||||
include("fields.jl")
|
||||
export DCTI
|
||||
@@ -23,7 +22,7 @@ export DCTI
|
||||
|
||||
### ELEMENTS ###
|
||||
include("elements.jl") # common element routines
|
||||
export Element, update!
|
||||
export Node, Element, update!
|
||||
include("lagrange_macro.jl") # Continuous Galerkin (Lagrange) elements generated using macro
|
||||
export Seg2, Tri3, Quad4, Hex8, Tet4
|
||||
|
||||
|
||||
+24
-28
@@ -64,8 +64,7 @@ function assemble{El<:Union{Tri3,Tri6,Quad4}}(problem::Problem{Elasticity}, elem
|
||||
|
||||
for (w, xi) in get_integration_points(element)
|
||||
|
||||
J = element(xi, time, Val{:Jacobian})
|
||||
w = w*det(J)
|
||||
detJ = element(xi, time, Val{:detJ})
|
||||
N = element(xi, time)
|
||||
dN = element(xi, time, Val{:Grad})
|
||||
|
||||
@@ -126,16 +125,16 @@ function assemble{El<:Union{Tri3,Tri6,Quad4}}(problem::Problem{Elasticity}, elem
|
||||
S2[1,2] = S2[2,1] = S[3]
|
||||
S2[3:4,3:4] = S2[1:2,1:2]
|
||||
|
||||
Kt += w*BL'*D*BL # material stiffness
|
||||
Kt += w*BL'*D*BL*detJ # material stiffness
|
||||
if props.finite_strain # add geometric stiffness
|
||||
Kt += w*BNL'*S2*BNL # geometric stiffness
|
||||
Kt += w*BNL'*S2*BNL*detJ # geometric stiffness
|
||||
end
|
||||
f -= w*BL'*S # internal force
|
||||
f -= w*BL'*S*detJ # internal force
|
||||
|
||||
# volume load
|
||||
if haskey(element, "displacement load")
|
||||
b = element("displacement load", xi, time)
|
||||
f += vec(w*N'*b)
|
||||
f += w*vec(N'*b)*detJ
|
||||
end
|
||||
|
||||
end
|
||||
@@ -153,8 +152,7 @@ function assemble{El<:Union{Seg2,Seg3}}(problem::Problem{Elasticity}, element::E
|
||||
|
||||
for (w, xi) in get_integration_points(element)
|
||||
|
||||
J = element(xi, time, Val{:Jacobian})
|
||||
detJ = norm(J)
|
||||
detJ = element(xi, time, Val{:detJ})
|
||||
N = element(xi, time)
|
||||
|
||||
if haskey(element, "displacement traction force")
|
||||
@@ -194,16 +192,15 @@ function assemble{El<:Union{Tet4, Tet10, Hex8}}(problem::Problem{Elasticity}, el
|
||||
Kt = zeros(dim*nnodes, dim*nnodes)
|
||||
f = zeros(dim*nnodes)
|
||||
|
||||
for ip in get_integration_points(element)
|
||||
J = get_jacobian(element, ip, time)
|
||||
w = ip.weight*det(J)
|
||||
N = element(ip, time)
|
||||
dN = element(ip, time, Val{:grad})
|
||||
for (w, xi) in get_integration_points(element)
|
||||
detJ = element(xi, time, Val{:detJ})
|
||||
N = element(xi, time)
|
||||
dN = element(xi, time, Val{:Grad})
|
||||
|
||||
# kinematics; calculate deformation gradient and strain
|
||||
gradu = zeros(dim, dim)
|
||||
if haskey(element, "displacement")
|
||||
gradu += element("displacement", ip, time, Val{:grad})
|
||||
gradu += element("displacement", xi, time, Val{:Grad})
|
||||
end
|
||||
strain = zeros(dim , dim)
|
||||
strain += 1/2*(gradu' + gradu)
|
||||
@@ -213,8 +210,8 @@ function assemble{El<:Union{Tet4, Tet10, Hex8}}(problem::Problem{Elasticity}, el
|
||||
strain += 1/2*gradu'*gradu
|
||||
end
|
||||
|
||||
E = element("youngs modulus", ip, time)
|
||||
nu = element("poissons ratio", ip, time)
|
||||
E = element("youngs modulus", xi, time)
|
||||
nu = element("poissons ratio", xi, time)
|
||||
|
||||
a = 1 - nu
|
||||
b = 1 - 2*nu
|
||||
@@ -273,16 +270,16 @@ function assemble{El<:Union{Tet4, Tet10, Hex8}}(problem::Problem{Elasticity}, el
|
||||
S3[1,2] = S3[2,1] = S[6]
|
||||
S3[4:6,4:6] = S3[7:9,7:9] = S3[1:3,1:3]
|
||||
|
||||
Kt += w*BL'*D*BL
|
||||
Kt += w*BL'*D*BL*detJ
|
||||
if props.finite_strain
|
||||
Kt += w*BNL'*S3*BNL
|
||||
Kt += w*BNL'*S3*BNL*detJ
|
||||
end
|
||||
f -= w*BL'*S
|
||||
f -= w*BL'*S*detJ
|
||||
|
||||
# volume load
|
||||
if haskey(element, "displacement load")
|
||||
T = element("displacement load", ip, time)
|
||||
f += vec(w*T*N)
|
||||
f += w*vec(T*N)*detJ
|
||||
end
|
||||
end
|
||||
|
||||
@@ -298,18 +295,17 @@ function assemble{El<:Union{Tri3, Tri6, Quad4}}(problem::Problem{Elasticity}, el
|
||||
Kt = zeros(dim*nnodes, dim*nnodes)
|
||||
f = zeros(dim*nnodes)
|
||||
|
||||
for ip in get_integration_points(element)
|
||||
JT = transpose(get_jacobian(element, ip, time))
|
||||
N = element(ip, time)
|
||||
w = ip.weight*norm(cross(JT[:,1], JT[:,2]))
|
||||
for (w, xi) in get_integration_points(element)
|
||||
detJ = element(xi, time, Val{:detJ})
|
||||
N = element(xi, time)
|
||||
if haskey(element, "displacement traction force")
|
||||
T = element("displacement traction force", ip, time)
|
||||
f += vec(w*T*N)
|
||||
T = element("displacement traction force", xi, time)
|
||||
f += w*vec(T*N)*detJ
|
||||
end
|
||||
for i in 1:dim
|
||||
if haskey(element, "displacement traction force $i")
|
||||
T = element("displacement traction force $i", ip, time)
|
||||
f[i:dim:end] += vec(w*T*N)
|
||||
T = element("displacement traction force $i", xi, time)
|
||||
f[i:dim:end] += w*vec(T*N)*detJ
|
||||
end
|
||||
end
|
||||
end
|
||||
|
||||
+19
-13
@@ -1,10 +1,10 @@
|
||||
# This file is a part of JuliaFEM.
|
||||
# License is MIT: see https://github.com/JuliaFEM/JuliaFEM.jl/blob/master/LICENSE.md
|
||||
|
||||
import Base: getindex, setindex!, convert, size, length
|
||||
|
||||
abstract AbstractElement
|
||||
|
||||
typealias Node Vector{Float64}
|
||||
|
||||
type Element{E<:AbstractElement}
|
||||
connectivity :: Vector{Int}
|
||||
fields :: Dict{ASCIIString, Field}
|
||||
@@ -49,6 +49,20 @@ function call(element::Element, xi::Vector, time, ::Type{Val{:Jacobian}})
|
||||
return J
|
||||
end
|
||||
|
||||
function call(element::Element, xi::Vector, time, ::Type{Val{:detJ}})
|
||||
J = element(xi, time, Val{:Jacobian})
|
||||
n, m = size(J)
|
||||
if n == m # volume element
|
||||
return det(J)
|
||||
end
|
||||
JT = transpose(J)
|
||||
if size(JT, 2) == 1 # boundary of 2d problem, || ∂X/∂ξ ||
|
||||
return norm(JT)
|
||||
else # manifold on 3d problem, || ∂X/∂ξ₁ × ∂X/∂ξ₂ ||
|
||||
return norm(cross(JT[:,1], JT[:,2]))
|
||||
end
|
||||
end
|
||||
|
||||
function get_jacobian{E}(element::Element{E}, xi::Vector, time=0.0)
|
||||
element(xi, time, Val{:Jacobian})
|
||||
end
|
||||
@@ -128,18 +142,10 @@ function get_dualbasis(element::Element, time)
|
||||
De = zeros(nnodes, nnodes)
|
||||
Me = zeros(nnodes, nnodes)
|
||||
for (w, xi) in get_integration_points(element, Val{3})
|
||||
J = element(xi, time, Val{:Jacobian})
|
||||
JT = transpose(J)
|
||||
if size(JT, 2) == 1 # plane problem
|
||||
# || ∂X/∂ξ ||
|
||||
w *= norm(JT)
|
||||
else
|
||||
# || ∂X/∂ξ₁ × ∂X/∂ξ₂ ||
|
||||
w *= norm(cross(JT[:,1], JT[:,2]))
|
||||
end
|
||||
detJ = element(xi, time, Val{:detJ})
|
||||
N = element(xi, time)
|
||||
De += w*diagm(vec(N))
|
||||
Me += w*N'*N
|
||||
De += w*diagm(vec(N))*detJ
|
||||
Me += w*N'*N*detJ
|
||||
end
|
||||
return De, Me, De*inv(Me)
|
||||
end
|
||||
|
||||
@@ -68,10 +68,12 @@ function get_integration_points(element::LineElement, ::Type{Val{3}})
|
||||
[ (w[i], [xi[i]]) for i=1:3 ]
|
||||
end
|
||||
|
||||
#=
|
||||
function get_integration_points{E<:LineElement}(element::Element{E}, ::Type{Val{3}})
|
||||
w, xi = get_integration_points(Val{3})
|
||||
[ (w[i], [xi[i]]) for i=1:3 ]
|
||||
end
|
||||
=#
|
||||
|
||||
function get_integration_points(element::Quad4, ::Type{Val{2}})
|
||||
w, xi = get_integration_points(Val{2})
|
||||
@@ -106,6 +108,10 @@ function get_integration_points(element::Hex8)
|
||||
get_integration_points(element, Val{2})
|
||||
end
|
||||
|
||||
function get_integration_points{E}(element::Element{E}, ::Type{Val{3}})
|
||||
get_integration_points(element.properties, Val{3})
|
||||
end
|
||||
|
||||
### triangular and tetrahedral elements
|
||||
|
||||
# http://math2.uncc.edu/~shaodeng/TEACHING/math5172/Lectures/Lect_15.PDF
|
||||
|
||||
Reference in New Issue
Block a user