Glassvision/JuliaFEM.jl
1
0
Fork 0
mirror of https://github.com/JuliaFEM/JuliaFEM.jl.git synced 2026-09-24 03:07:53 +00:00
Code Issues Packages Projects Releases Wiki Activity

update all kind of fields using update! function

Browse Source
This commit is contained in:
Jukka Aho
2016-01-01 15:00:18 +02:00
parent 91173dc78e
commit 52fd3f5258
1 changed files with 24 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/elements.jl
+24 -3
View File
@@ -282,12 +282,33 @@ function calculate_normal_tangential_coordinates!{E}(elements::Vector{Element{E}
end
end
""" Pick values from nodes and set to element according to connectivity. """
""" Update element field based on a vector or dictionary of nodal data and connectivity information.
Examples
--------
julia> data = Dict(1 => [0.0, 0.0], 2 => [1.0, 2.0])
julia> element = Seg2([1, 2])
julia> update!(element, "geometry", data)
As a result element now have time invariant (variable) vector field "geometry" with data ([0.0, 0.0], [1.0, 2.0]).
"""
function update!(element::Element, field_name::ASCIIString, data::Union{Vector, Dict})
element[field_name] = [data[i] for i in get_connectivity(element)]
end
function update!{T}(elements::Vector{Element{T}}, field_name::ASCIIString, data::Union{Vector, Dict})
# info("update $field_name for $(length(elements)) elements.")
function update!(element::Element, field_name::ASCIIString, data::Union{Real, Pair})
element[field_name] = data
end
""" Update values for several elements at once. """
# FIXME: with or without {T} ?
function update!{T}(elements::Vector{Element{T}}, field_name::ASCIIString, data)
for element in elements
update!(element, field_name, data)
end
end
function update!(elements::Vector{Element}, field_name::ASCIIString, data)
for element in elements
update!(element, field_name, data)
end