changed creating elements + preprocess aster med file more general for other inputs too

src/preprocess.jl

@@ -0,0 +1,75 @@
+# This file is a part of JuliaFEM.
+# License is MIT: see https://github.com/JuliaFEM/JuliaFEM.jl/blob/master/LICENSE.md
+
+importall Base
+
+using JuliaFEM
+
+type Mesh
+    nodes :: Dict{Int64, Vector{Float64}}
+    node_sets :: Dict{ASCIIString, Set{Int64}}
+    elements :: Dict{Int64, Vector{Int64}}
+    element_types :: Dict{Int64, Symbol}
+    element_sets :: Dict{ASCIIString, Set{Int64}}
+end
+
+function Mesh()
+    return Mesh(Dict(), Dict(), Dict(), Dict(), Dict())
+end
+
+function add_node!(mesh::Mesh, nid::Int, ncoords::Vector{Float64})
+    mesh.nodes[nid] = ncoords
+end
+
+function add_element!(mesh::Mesh, elid::Int, eltype::Symbol, connectivity::Vector{Int64})
+    mesh.elements[elid] = connectivity
+    mesh.element_types[elid] = eltype
+end
+
+function add_element_to_element_set!(mesh::Mesh, set_name::ASCIIString, elids...)
+    if !haskey(mesh.element_sets, set_name)
+        mesh.element_sets[set_name] = Set{Int64}()
+    end
+    push!(mesh.element_sets[set_name], elids...)
+end
+
+function copy(mesh::Mesh)
+    mesh2 = Mesh()
+    mesh2.nodes = copy(mesh.nodes)
+    mesh2.node_sets = copy(mesh.node_sets)
+    mesh2.elements = copy(mesh.elements)
+    mesh2.element_types = copy(mesh.element_types)
+    mesh2.element_sets = copy(mesh.element_sets)
+    return mesh2
+end
+
+function filter_by_element_id(mesh::Mesh, element_ids::Vector{Int64})
+    mesh2 = copy(mesh)
+    mesh2.elements = Dict()
+    for elid in element_ids
+        if haskey(mesh.elements, elid)
+            mesh2.elements[elid] = mesh.elements[elid]
+        end
+    end
+    return mesh2
+end
+
+function filter_by_element_set(mesh::Mesh, set_name::ASCIIString)
+    filter_by_element_id(mesh::Mesh, collect(mesh.element_sets[set_name]))
+end
+
+function create_elements(mesh::Mesh)
+    elements = Element[]
+    for (elid, elcon) in mesh.elements
+        eltype = mesh.element_types[elid]
+        element = Element(JuliaFEM.(eltype), elcon)
+        update!(element, "geometry", mesh.nodes)
+        push!(elements, element)
+    end
+    return elements
+end
+
+function create_elements(mesh::Mesh, element_set::ASCIIString)
+    return create_elements(filter_by_element_set(mesh, element_set))
+end
+