From 11dad1097c3d64f20b88a9f62d2affdfe706e555 Mon Sep 17 00:00:00 2001
From: Jukka Aho <ahojukka5@gmail.com>
Date: Thu, 25 Jun 2015 19:58:26 +0300
Subject: [PATCH] added basic abaqus inp parser

---
 src/JuliaFEM.jl            |   5 +-
 src/abaqus_reader.jl       | 132 +++++++++++++++++++++++++++++++++++++
 src/xdmf.jl                |   5 +-
 test/test_abaqus_reader.jl |  17 +++++
 4 files changed, 155 insertions(+), 4 deletions(-)
 create mode 100644 src/abaqus_reader.jl
 create mode 100644 test/test_abaqus_reader.jl

diff --git a/src/JuliaFEM.jl b/src/JuliaFEM.jl
index ae541f8..96f249a 100644
--- a/src/JuliaFEM.jl
+++ b/src/JuliaFEM.jl
@@ -1,13 +1,12 @@
-# This file is a part of JuliaFEM. 
+# This file is a part of JuliaFEM.
 # License is MIT: see https://github.com/JuliaFEM/JuliaFEM.jl/blob/master/LICENSE.md
 module JuliaFEM
 
 VERSION < v"0.4-" && using Docile
 
-# import solvers
 include("elasticity_solver.jl")
 include("xdmf.jl")
-#using elasticity_solver
+include("abaqus_reader.jl")
 
 export Model, new_model, new_field, get_field, add_nodes, get_nodes, add_elements, get_elements
 
diff --git a/src/abaqus_reader.jl b/src/abaqus_reader.jl
new file mode 100644
index 0000000..686b07f
--- /dev/null
+++ b/src/abaqus_reader.jl
@@ -0,0 +1,132 @@
+# This file is a part of JuliaFEM. License is MIT: https://github.com/ovainola/JuliaFEM/blob/master/README.md
+module abaqus_reader
+
+using Logging
+@Logging.configure(level=DEBUG)
+
+VERSION < v"0.4-" && using Docile
+
+eldims = Dict("C3D10" => 10)
+global handlers = Dict()
+
+"""
+Register new handler for parser
+"""
+function add_handler(section, function_name)
+  handlers[section] = function_name
+end
+
+function create_or_get(model, key)
+    if !(key in keys(model))
+        model[key] = Dict()
+    end
+    return model[key]
+end
+
+function parse_header(header_line)
+    args = map(s -> strip(s), split(header_line, ","))
+    args[1] = strip(args[1], '*')
+    d = Dict({"section" => args[1]})
+    options = Dict()
+    for k in args[2:end]
+        args2 = split(k, "=")
+        options[args2[1]] = args2[2]
+    end
+    d["options"] = options
+    return d
+end
+
+function parse_node_section(model, header, data)
+    nodes = create_or_get(model, "nodes")
+    for line in split(data, "\n")
+        m = matchall(r"[-0-9.]+", line)
+        id = parse(Int, m[1])
+        coords = float(m[2:end])
+        nodes[id] = coords
+    end
+end
+
+function parse_element_section(model, header, data)
+    eltype = header["options"]["TYPE"]
+    if !(eltype in keys(eldims))
+        throw("Element $eltype dimension information missing")
+    end
+    eldim = eldims[eltype]
+    m = matchall(r"[0-9]+", data)
+    m = map(integer, m)
+    elements = create_or_get(model, "elements")
+    m = reshape(m, eldim+1, round(Int, length(m)/(eldim+1)))
+    nel = size(m)[2]
+    Logging.debug("$nel elements found")
+    for i=1:nel
+        elements[m[1,i]] = m[2:end,i]
+    end
+    if "ELSET" in keys(header["options"])
+        elsets = create_or_get(model, "elsets")
+        elset_name = header["options"]["ELSET"]
+        Logging.info("Creating ELSET $elset_name")
+        elsets[elset_name] = Int64[]
+        for i=1:nel
+            push!(elsets[elset_name], m[1,i])
+        end
+    end
+end
+
+
+function parse_nodeset_section(model, header, data)
+    nset_name = header["options"]["NSET"]
+    Logging.debug("Creating node set $nset_name")
+    m = matchall(r"[0-9]+", data)
+    node_ids = map(integer, m)
+    nsets = create_or_get(model, "nsets")
+    nsets[nset_name] = Int64[]
+    for j in node_ids
+        push!(nsets[nset_name], j)
+    end
+end
+
+
+function parse_abaqus(fid)
+    model = Dict()
+    section = None
+    header = None
+    data = ""
+    Logging.info("Registered handlers: $(keys(handlers))")
+
+    function process_section(section)
+        if section == None
+            return
+        end
+        if !(section in keys(handlers))
+            Logging.info("Don't know what to do with data in section $section")
+            Logging.info("Skipping $(length(data)) bytes of unknown data")
+            return
+        end
+        handlers[section](model, header, strip(data))
+        data = ""
+    end
+
+    for line in eachline(fid)
+        if beginswith(line, "**")
+            continue
+        end
+        if beginswith(line, "*")
+            process_section(section)
+            header = parse_header(line)
+            Logging.debug("Found ", header["section"], " section")
+            section = header["section"]
+            continue
+        end
+        data *= line
+    end
+    process_section(section)
+    return model
+end
+
+# add handlers
+add_handler("NODE", parse_node_section)
+add_handler("ELEMENT", parse_element_section)
+add_handler("NSET", parse_nodeset_section)
+
+
+end
\ No newline at end of file
diff --git a/src/xdmf.jl b/src/xdmf.jl
index 9126b2e..50d8584 100644
--- a/src/xdmf.jl
+++ b/src/xdmf.jl
@@ -102,5 +102,8 @@ function xdmf_new_field(grid, name, source, data)
     add_text(dataitem, join(data, " "))
 end
 
+function xdmf_save_model(xdoc, filename)
+  save_file(xdoc, filename)
+end
 
-end
\ No newline at end of file
+end
diff --git a/test/test_abaqus_reader.jl b/test/test_abaqus_reader.jl
new file mode 100644
index 0000000..cf0f91e
--- /dev/null
+++ b/test/test_abaqus_reader.jl
@@ -0,0 +1,17 @@
+using FactCheck
+using Logging
+@Logging.configure(level=INFO)
+
+using JuliaFEM.abaqus_reader: parse_abaqus
+
+facts("test import abaqus model") do
+    fid = open("../geometry/3d_beam/palkki.inp")
+    model = parse_abaqus(fid)
+    close(fid)
+    @fact length(model["nodes"]) => 298
+    @fact length(model["elements"]) => 120
+    @fact length(model["elsets"]["Body1"]) => 120
+    @fact length(model["nsets"]["SUPPORT"]) => 9
+    @fact length(model["nsets"]["LOAD"]) => 9
+    @fact length(model["nsets"]["TOP"]) => 83
+end
\ No newline at end of file