JuliaFEM.jl/test/test_io.jl

# This file is a part of JuliaFEM.
# License is MIT: see https://github.com/JuliaFEM/JuliaFEM.jl/blob/master/LICENSE.md

using JuliaFEM
using JuliaFEM.Testing
using LightXML

@testset "create new Xdmf object" begin
    r = Xdmf()
    expected = "<Xdmf xmlns:xi=\"http://www.w3.org/2001/XInclude\" Version=\"3.0\"/>"
    @test string(r.xml) == expected
end

@testset "put and get to Xdmf, low level" begin
    io = Xdmf()
    # h5
    write(io.hdf, "/Xdmf/Domain/Geometry", [1 2 3])
    @test isapprox(read(io.hdf, "/Xdmf/Domain/Geometry"), [1 2 3])
    # xml
    obj = new_child(io.xml, "Domain")
    set_attribute(obj, "Name", "Test Domain")
    obj2 = find_element(io.xml, "Domain")
    @test attribute(obj2, "Name") == "Test Domain"
end

@testset "write data to HDF, automatically generate path" begin
    xdmf = Xdmf()
    di1 = new_dataitem(xdmf, [1 2 3])
    di2 = new_dataitem(xdmf, [4 5 6])
    @test contains(content(di1), "DataItem_1")
    @test contains(content(di2), "DataItem_2")
end

@testset "Xdmf filtering" begin
    grid1 = new_element("Grid")
    add_text(grid1, "I am first grid")
    grid2 = new_element("Grid")
    add_text(grid2, "I am second grid")
    set_attribute(grid2, "Name", "Frame 2")
    grid3 = new_element("Grid")
    add_text(grid3, "I am third grid")
    grids = [grid1, grid2, grid3]

    @test content(xdmf_filter(grids, "Grid")) == "I am first grid"
    @test content(xdmf_filter(grids, "Grid[1]")) == "I am first grid"
    @test content(xdmf_filter(grids, "Grid[2]")) == "I am second grid"
    @test content(xdmf_filter(grids, "Grid[3]")) == "I am third grid"
    @test content(xdmf_filter(grids, "Grid[end]")) == "I am third grid"
    @test content(xdmf_filter(grids, "Grid[@Name=Frame 2]")) == "I am second grid"
    @test xdmf_filter(grids, "Grid[0]") == nothing
    @test xdmf_filter(grids, "Grid[4]") == nothing
    @test xdmf_filter(grids, "Grid[@Name=Frame 3]") == nothing
    @test xdmf_filter(grids, "Domain/Grid[@Name=Frame 3]") == nothing
    @test xdmf_filter(grids, "Domain") == nothing
end

@testset "XML traverse" begin
    xdmf = Xdmf()
    domain = new_child(xdmf.xml, "Domain")
    grid = new_child(domain, "Grid")
    set_attribute(grid, "CollectionType", "Temporal")
    set_attribute(grid, "GridType", "Collection")

    frame1 = new_child(grid, "Grid")
    time1 = new_child(frame1, "Time")
    set_attribute(time1, "Value", 0.0)
    X1 = new_child(frame1, "Geometry")
    set_attribute(X1, "Type", "XY")

    frame2 = new_child(grid, "Grid")
    set_attribute(frame2, "Name", "Frame 2")
    time2 = new_child(frame2, "Time")
    set_attribute(time2, "Value", 1.0)
    X2 = new_child(frame2, "Geometry")
    set_attribute(X2, "Type", "XY")

    add_child(grid, frame1)
    add_child(grid, frame2)

    dataitem = new_dataitem(xdmf, "/Domain/Grid/Grid/2/Geometry", [1.0, 2.0])
    add_child(X2, dataitem)

    println(xdmf.xml)
    @test read(xdmf, "/Domain/Grid/Grid/Time/Value") == "0.0"
    @test read(xdmf, "/Domain/Grid/Grid[2]/Time/Value") == "1.0"
    @test read(xdmf, "/Domain/Grid/Grid[end]/Time/Value") == "1.0"
    @test read(xdmf, "/Domain/Grid/Grid[@Name=Frame 2]/Time/Value") == "1.0"
    @test isapprox(read(xdmf, "/Domain/Grid/Grid[2]/Geometry/DataItem"), [1.0, 2.0])
end

@testset "write fields from different problems to Xdmf file" begin

    X = Dict(
             1 => [0.0, 0.0, 0.0],
             2 => [1.0, 0.0, 0.0],
             3 => [1.0, 1.0, 0.0],
             4 => [0.0, 1.0, 0.0],
             5 => [0.0, 0.0, 0.5],
             6 => [1.0, 0.0, 0.5],
             7 => [1.0, 1.0, 0.5],
             8 => [0.0, 1.0, 0.5],
             9 => [0.0, 0.0, 1.0],
             10 => [1.0, 0.0, 1.0],
             11 => [1.0, 1.0, 1.0],
             12 => [0.0, 1.0, 1.0])

    u = Dict()
    u[0] = Dict(
                1 => [0.0, 0.0, 0.0],
                2 => [0.0, 0.0, 0.0],
                3 => [0.0, 0.0, 0.0],
                4 => [0.0, 0.0, 0.0],
                5 => [0.0, 0.0, 0.0],
                6 => [0.0, 0.0, 0.0],
                7 => [0.0, 0.0, 0.0],
                8 => [0.0, 0.0, 0.0],
                9 => [0.0, 0.0, 0.0],
                10 => [0.0, 0.0, 0.0],
                11 => [0.0, 0.0, 0.0],
                12 => [0.0, 0.0, 0.0])
    u[1] = Dict(
                1 => [0.0, 0.0, 0.0],
                2 => [0.0, 0.0, 0.0],
                3 => [0.0, 0.0, 0.0],
                4 => [0.0, 0.0, 0.0],
                5 => [0.0, 0.0, -0.1],
                6 => [0.0, 0.0, -0.1],
                7 => [0.0, 0.0, -0.1],
                8 => [0.0, 0.0, -0.1],
                9 => [0.0, 0.0, -0.2],
                10 => [0.0, 0.0, -0.2],
                11 => [0.0, 0.0, -0.2],
                12 => [0.0, 0.0, -0.2])

    T = Dict(
             1 => 10.0,
             2 => 10.0,
             3 => 10.0,
             4 => 10.0,
             5 => 20.0,
             6 => 20.0,
             7 => 20.0,
             8 => 20.0,
             9 => 30.0,
             10 => 30.0,
             11 => 30.0,
             12 => 30.0)

    rf = Dict()
    rf[0] = Dict(
                 1 => [0.0, 0.0, 0.0],
                 2 => [0.0, 0.0, 0.0],
                 3 => [0.0, 0.0, 0.0],
                 4 => [0.0, 0.0, 0.0])
    rf[1] = Dict(
                 1 => [0.0, 0.0, 1.0],
                 2 => [0.0, 0.0, 1.0],
                 3 => [0.0, 0.0, 1.0],
                 4 => [0.0, 0.0, 1.0])

    e1 = Element(Hex8, [1, 2, 3, 4, 5, 6, 7, 8])
    e2 = Element(Hex8, [5, 6, 7, 8, 9, 10, 11, 12])
    e3 = Element(Quad4, [1, 2, 3, 4])
    update!([e1, e2, e3], "geometry", X)
    update!([e1, e2, e3], "displacement", 0.0 => u[0])
    update!([e1, e2, e3], "displacement", 1.0 => u[1])
    update!([e1, e2, e3], "temperature", T)
    update!(e3, "reaction force", 0.0 => rf[0])
    update!(e3, "reaction force", 1.0 => rf[1])

    p1 = Problem(Elasticity, "lower", 3)
    p1.elements = [e1]
    p2 = Problem(Elasticity, "upper", 3)
    p2.elements = [e2]
    p3 = Problem(Dirichlet, "bc", 3, "displacement")
    p3.elements = [e3]

    xdmf = Xdmf()
    xdmf.format = "XML"
    update_xdmf!(xdmf, p1, 0.0, ["displacement", "temperature"])
    update_xdmf!(xdmf, p2, 0.0, ["displacement"])
    update_xdmf!(xdmf, p3, 0.0, ["reaction force"])
    update_xdmf!(xdmf, p1, 1.0, ["displacement", "temperature"])
    update_xdmf!(xdmf, p2, 1.0, ["displacement"])
    update_xdmf!(xdmf, p3, 1.0, ["reaction force"])
    @test read(xdmf, "/Domain/Grid/Grid/Time/Value") == "0.0"
    @test read(xdmf, "/Domain/Grid/Grid[2]/Time/Value") == "1.0"
end