2d mortar

test/test_mortar.jl

@@ -6,78 +6,113 @@ module MortarTests
 using JuliaFEM
 using JuliaFEM.Test
 
-function test_calc_flat_2d_assembly()
-    # this is hand calculated and given example in my thesis
+using JuliaFEM: MSeg2, Seg2, MortarProblem, MortarEquation, MortarElement, Assembly, assemble!
+using JuliaFEM: get_basis, grad, project_from_slave_to_master, project_from_master_to_slave
+
+function get_test_2d_model()
+    # this is hand calculated and given as an example in my thesis
     N = Vector[
         [0.0, 2.0], [1.0, 2.0], [2.0, 2.0],
         [0.0, 0.0], [1.0, 0.0], [2.0, 0.0],
         [0.0, 1.0], [5/4, 1.0], [2.0, 1.0],
         [0.0, 1.0], [3/4, 1.0], [2.0, 1.0]]
+    rotation_matrix(phi) = [cos(phi) -sin(phi); sin(phi) cos(phi)]
+    slave1 = MSeg2([10, 11])
+    slave1["geometry"] = Vector[N[10], N[11]]
+    # should be n = [0 -1]' and t = [1 0]'
+    slave1["nodal ntsys"] = Matrix[rotation_matrix(-pi/2), rotation_matrix(-pi/2)]
+    slave2 = MSeg2([11, 12])
+    slave2["geometry"] = Vector[N[11], N[12]]
+    # should be n = [0 -1]' and t = [1 0]'
+    slave2["nodal ntsys"] = Matrix[rotation_matrix(-pi/2), rotation_matrix(-pi/2)]
+    master1 = MSeg2([7, 8])
+    master1["geometry"] = Vector[N[7], N[8]]
+    master2 = MSeg2([8, 9])
+    master2["geometry"] = Vector[N[8], N[9]]
+    push!(slave1.master_elements, master1)
+    push!(slave1.master_elements, master2)
+    push!(slave2.master_elements, master1)
+    push!(slave2.master_elements, master2)
+    return [slave1, slave2], [master1, master2]
+end
 
-    slave1 = Seg2([10, 11])
-    slave1["geometry"] = Vector[N10, N11]
 
-    slave2 = Seg2([11, 12])
-    slave2["geometry"] = Vector[N11, N12]
+function test_calc_flat_2d_projection()
+    slaves, masters = get_test_2d_model()
+    slave1, slave2 = slaves
+    master1, master2 = masters
 
-    master1 = Seg2([7, 8])
-    master1["geometry"] = Vector[N7, N8]
+    xi2a = project_from_slave_to_master(slave1, master1, [-1.0])
+    @test xi2a == [-1.0]
 
-    master2 = Seg2([8, 9])
-    master2["geometry"] = Vector[N8, N9]
+    xi2b = project_from_slave_to_master(slave1, master1, [1.0])
+    @test xi2b == [ 0.2]
+    X2 = get_basis(master1)("geometry", xi2b)
+    @test X2 == [3/4, 1.0]
 
+    xi1a = project_from_master_to_slave(slave1, master1, [-1.0])
+    @test xi1a == [-1.0]
+    xi1b = project_from_master_to_slave(slave1, master1, [1.0])
+    X1 = get_basis(slave1)("geometry", xi1b)
+    @test X1 == [5/4, 1.0]
+end
+
+function test_create_flat_2d_assembly()
+    slaves, masters = get_test_2d_model()
+    slave1, slave2 = slaves
+    master1, master2 = masters
+
+    info("creating problem")
     problem = MortarProblem()
+    info("pushing slave elements to problem")
     push!(problem, slave1)
     push!(problem, slave2)
-    push!(problem.master_elements, master1)
-    push!(problem.master_elements, master2)
-
-    rotation_matrix(phi) = [cos(phi) -sin(phi); sin(phi) cos(phi)]
-    # should be n = [0 -1]' and t = [1 0]'
-    @test isapprox(rotation_matrix(-phi/2), [[0 -1]' [1 0]'])
-
-    problem.node_csys = Dict(
-        10 => rotation_matrix(-phi/2),
-        11 => rotation_matrix(-phi/2),
-        12 => rotation_matrix(-phi/2))
-
-    # first index = master element id
-    # second index = slave element id
-    problem.element_pairs = zeros(2, 2)
-    # first slave element connects to master element 1
-    problem.element_pairs[1, 1] = true
-    # second slave element connects to master element 1
-    problem.element_pairs[1, 2] = true
-    # second slave element connects to master element 2
-    problem.element_pairs[2, 2] = true
-
-    B_expected = zeros(12, 9)
 
+    B_expected = zeros(12, 12)
     S1 = [10, 11]
     M1 = [7, 8]
     B_expected[S1,S1] += [1/4 1/8; 1/8 1/4]
-    B_expected[S1,M1] += [3/10 3/40; 9/40 3/20]
+    B_expected[S1,M1] -= [3/10 3/40; 9/40 3/20]
 
-    la = initialize_local_assembly(problem)
-    calculate_local_assembly!(la, problem.equations[1], "reaction force", 0.0, problem=problem)
-    B = full(la.lhs)
+    info("creating assembly")
+    assembly = Assembly()
+    assemble!(assembly, problem.equations[1], 0.0, problem)
+    B = round(full(assembly.lhs, 12, 12), 6)
+    info("size of B = $(size(B))")
+    info("B matrix in first slave element = \n$(B[10:11,:])")
+    info("B matrix expected = \n$(B_expected[10:11,:])")
     @test isapprox(B, B_expected) 
 
     fill!(B_expected, 0.0)
+    empty!(assembly)
 
     S2 = [11, 12]
     M2 = [7, 8]
     B_expected[S2,S2] += [49/150 11/150; 11/150 2/75]
-    B_expected[S2,M2] += [13/150 47/150; 1/75 13/150]
+    B_expected[S2,M2] -= [13/150 47/150; 1/75 13/150]
     S3 = [11, 12]
     M3 = [8, 9]
     B_expected[S3,S3] += [9/100 27/200; 27/200 39/100]
-    B_expected[S3,M3] += [3/20 3/40; 9/40 3/10]
+    B_expected[S3,M3] -= [3/20 3/40; 9/40 3/10]
+    assemble!(assembly, problem.equations[2], 0.0, problem)
+    B = full(assembly.lhs)
+    info("size of B = $(size(B))")
+    info("B matrix in second slave element = \n$(B[11:12,:])")
+    info("B matrix expected = \n$(B_expected[11:12,:])")
 
-    la = initialize_local_assembly(problem)
-    calculate_local_assembly!(la, problem.equations[1], "reaction force", 0.0, problem=problem)
-    B = full(la.lhs)
     @test isapprox(B, B_expected)
 end
 
+function test_patch_test_heat_2d()
+
+
+    
+    slaves, masters = get_test_2d_model()
+    problem2 = MortarProblem()
+    for slave in slaves:
+        push!(problem2, slave)
+    end
+
+end
+
 end