add nutils solid participant

Author: gertjanvanzwieten

perpendicular-flap/solid-nutils/clean.sh

@@ -0,0 +1,6 @@
+#!/bin/sh
+set -e -u
+
+. ../../tools/cleaning-tools.sh
+
+clean_nutils .

perpendicular-flap/solid-nutils/run.sh

@@ -0,0 +1,4 @@
+#!/bin/sh
+set -e -u
+
+python3 solid.py richoutput=no

perpendicular-flap/solid-nutils/solid.py

@@ -0,0 +1,102 @@
+#! /usr/bin/env python3
+
+from nutils import mesh, function, solver, export, cli
+from nutils.expression_v2 import Namespace
+import numpy
+import treelog
+import precice
+
+
+def main(young=4e6, density=3e3, poisson=.3, nelems=2, timestepsize=0.01, npoints_per_elem=3):
+
+    topo, geom = mesh.rectilinear([numpy.linspace(-.05, .05, nelems+1), numpy.linspace(0, 1, 10*nelems+1)])
+    wall = topo.boundary['left,top,right'].sample('uniform', npoints_per_elem)
+
+    ns = Namespace()
+    ns.X = geom
+    ns.δ = function.eye(2)
+    ns.define_for('X', gradient='∇', normal='n', jacobians=('dV', 'dS'))
+    ns.add_field('dt')
+    ns.add_field(('u', 'u0', 'testu', 'v', 'v0', 'testv'), topo.basis('std', degree=1), shape=(2,))
+    ns.add_field('F', wall.basis(), shape=(2,))
+    ns.qw = 1 / npoints_per_elem # quadrature weight
+    ns.t_i = 'F_i / qw dS'
+    ns.dudt_i = '(u_i - u0_i) / dt'
+    ns.dvdt_i = '(v_i - v0_i) / dt'
+    ns.ρ = density
+    ns.λ = young * poisson / ((1 + poisson) * (1 - 2 * poisson))
+    ns.μ = young / (2 * (1 + poisson))
+    ns.σ_ij = 'λ ∇_k(u_k) δ_ij + μ (∇_j(u_i) + ∇_i(u_j))'
+
+    # make sure we correctly scale point forces to tractions
+    testforce = numpy.random.normal(size=(wall.npoints, 2))
+    numpy.testing.assert_almost_equal(
+        actual=wall.integrate('t_i dS' @ ns, F=testforce),
+        desired=testforce.sum(0),
+        decimal=10,
+        err_msg='nutils error: failed to recover net force')
+
+    # continuum equations: ρ v' = ∇·σ + F, u' = v
+    res = topo.integral('testv_i (dudt_i - v_i) dV' @ ns, degree=2)
+    res += topo.integral('(testu_i ρ dvdt_i + ∇_j(testu_i) σ_ij) dV' @ ns, degree=2)
+    res -= wall.integral('testu_i t_i dS' @ ns)
+
+    # boundary conditions: fully constrained at y=0
+    sqr = topo.boundary['bottom'].integral('u_k u_k' @ ns, degree=2)
+    cons = solver.optimize('u,', sqr, droptol=1e-10)
+
+    # initial conditions: undeformed and unmoving
+    sqr = topo.integral('u_k u_k + v_k v_k' @ ns, degree=2)
+    arguments = solver.optimize('u,v', sqr, constrain=cons)
+
+    # preCICE setup
+    solverProcessIndex = 0
+    solverProcessSize = 1
+    interface = precice.Interface("Solid", "../precice-config.xml", solverProcessIndex, solverProcessSize)
+    meshID = interface.get_mesh_id("Solid-Mesh")
+    dataIndices = interface.set_mesh_vertices(meshID, wall.eval(ns.X))
+    writedataID = interface.get_data_id("Displacement", meshID)
+    readdataID = interface.get_data_id("Force", meshID)
+
+    # initialize preCICE
+    precice_dt = interface.initialize()
+
+    timestep = 0
+    force = numpy.zeros((wall.npoints, 2))
+
+    while interface.is_coupling_ongoing():
+      with treelog.context(f'timestep {timestep}'):
+
+        # read displacements from interface
+        if interface.is_read_data_available():
+            force = interface.read_block_vector_data(readdataID, dataIndices)
+
+        # save checkpoint
+        if interface.is_action_required(precice.action_write_iteration_checkpoint()):
+            checkpoint = timestep, arguments
+            interface.mark_action_fulfilled(precice.action_write_iteration_checkpoint())
+
+        # advance variables
+        timestep += 1
+        dt = min(precice_dt, timestepsize)
+        arguments = dict(dt=dt, u0=arguments['u'], v0=arguments['v'], F=force)
+        arguments = solver.solve_linear('u:testu,v:testv', res, arguments=arguments, constrain=cons)
+
+        # write forces to interface
+        if interface.is_write_data_required(dt):
+            writedata = wall.eval(ns.u, **arguments)
+            interface.write_block_vector_data(writedataID, dataIndices, writedata)
+
+        # do the coupling
+        precice_dt = interface.advance(dt)
+
+        # read checkpoint if required
+        if interface.is_action_required(precice.action_read_iteration_checkpoint()):
+            timestep, arguments = checkpoint
+            interface.mark_action_fulfilled(precice.action_read_iteration_checkpoint())
+
+    interface.finalize()
+
+
+if __name__ == '__main__':
+    cli.run(main)