Merge pull request #8 from chahak13/benchmarks

Benchmarks

Author: kks32

.gitignore

@@ -6,3 +6,4 @@ build/**
 examples/**
 **__pycache__**
 **.egg-info
+**.npz

benchmarks/2d/uniaxial_nodal_forces/mpm-nodal-forces.toml

@@ -0,0 +1,67 @@
+# The `meta` group contains top level attributes that govern the
+# behaviour of the MPM Solver.
+#
+# Attributes:
+# title:    The title of the experiment. This is just for the user's
+#           reference.
+# type:     The type of simulation to be used. Allowed values are
+#           {"MPMExplicit"}
+# scheme:   The MPM Scheme used for simulation. Allowed values are
+#           {"usl", "usf"}
+# dt:       Timestep used in the simulation.
+# nsteps:   Number of steps to run the simulation for.
+[meta]
+title = "uniaxial-nodal-forces"
+type = "MPMExplicit"
+dimension = 2
+scheme = "usf"
+dt = 0.001
+nsteps = 1000
+velocity_update = true
+
+[output]
+format = "npz"
+folder = "results/"
+step_frequency = 10
+
+[mesh]
+# type = "file"
+# file = "mesh-1d.txt"
+# boundary_nodes = "boundary-1d.txt"
+# particle_element_ids = "particles-elements.txt"
+type = "generator"
+nelements = [3, 1]
+element_length = [0.1, 0.1]
+particle_element_ids = [0]
+element = "Quadrilateral4Node"
+
+[[mesh.constraints]]
+node_ids = [0, 4]
+dir = 0
+velocity = 0.0
+
+[[materials]]
+id = 0
+density = 1000
+poisson_ratio = 0
+youngs_modulus = 1000000
+type = "LinearElastic"
+
+[[particles]]
+file = "particles-2d-nodal-force.json"
+material_id = 0
+init_velocity = 0.0
+
+[external_loading]
+gravity = [0, 0]
+
+[[external_loading.concentrated_nodal_forces]]
+node_ids = [3, 7]
+math_function_id = 0
+dir = 0
+force = 0.05
+
+[[math_functions]]
+type = "Linear"
+xvalues = [0.0, 0.5, 1.0]
+fxvalues = [0.0, 1.0, 1.0]

benchmarks/2d/uniaxial_nodal_forces/particles-2d-nodal-force.json

@@ -0,0 +1,74 @@
+[
+  [
+    [
+      0.025,
+      0.025
+    ]
+  ],
+  [
+    [
+      0.075,
+      0.025
+    ]
+  ],
+  [
+    [
+      0.125,
+      0.025
+    ]
+  ],
+  [
+    [
+      0.175,
+      0.025
+    ]
+  ],
+  [
+    [
+      0.225,
+      0.025
+    ]
+  ],
+  [
+    [
+      0.275,
+      0.025
+    ]
+  ],
+  [
+    [
+      0.025,
+      0.075
+    ]
+  ],
+  [
+    [
+      0.075,
+      0.075
+    ]
+  ],
+  [
+    [
+      0.125,
+      0.075
+    ]
+  ],
+  [
+    [
+      0.175,
+      0.075
+    ]
+  ],
+  [
+    [
+      0.225,
+      0.075
+    ]
+  ],
+  [
+    [
+      0.275,
+      0.075
+    ]
+  ]
+]

benchmarks/2d/uniaxial_nodal_forces/test_benchmark.py

@@ -0,0 +1,30 @@
+import os
+from pathlib import Path
+import jax.numpy as jnp
+from diffmpm.solver import MPM
+
+curr_filepath = Path(__file__).absolute()
+curr_dir = curr_filepath.parent
+os.chdir(curr_dir)
+mpm = MPM("mpm-nodal-forces.toml")
+mpm.solve()
+
+result = jnp.load("results/uniaxial-nodal-forces/particles_0300.npz")
+## Step 300
+assert jnp.round(result["stress"][0, :, 0].min() - 0.5925210678182377, 6) == 0.0
+assert jnp.round(result["stress"][0, :, 0].max() - 0.5974539476363379, 6) == 0.0
+
+## Step 510
+result = jnp.load("results/uniaxial-nodal-forces/particles_0510.npz")
+assert jnp.round(result["stress"][0, :, 0].min() - 1.0026665338366039, 5) == 0.0
+assert jnp.round(result["stress"][0, :, 0].max() - 1.0112231542459431, 5) == 0.0
+
+## Step 750
+result = jnp.load("results/uniaxial-nodal-forces/particles_0750.npz")
+assert jnp.round(result["stress"][0, :, 0].min() - 1.0000053000532143, 5) == 0.0
+assert jnp.round(result["stress"][0, :, 0].max() - 1.0000225119807862, 5) == 0.0
+
+## Step 990
+result = jnp.load("results/uniaxial-nodal-forces/particles_0990.npz")
+assert jnp.round(result["stress"][0, :, 0].min() - 0.9999990078443788, 5) == 0.0
+assert jnp.round(result["stress"][0, :, 0].max() - 0.9999990292713694, 5) == 0.0

benchmarks/2d/uniaxial_stress/mpm-uniaxial-stress.toml

@@ -0,0 +1,57 @@
+# The `meta` group contains top level attributes that govern the
+# behaviour of the MPM Solver.
+#
+# Attributes:
+# title:    The title of the experiment. This is just for the user's
+#           reference.
+# type:     The type of simulation to be used. Allowed values are
+#           {"MPMExplicit"}
+# scheme:   The MPM Scheme used for simulation. Allowed values are
+#           {"usl", "usf"}
+# dt:       Timestep used in the simulation.
+# nsteps:   Number of steps to run the simulation for.
+[meta]
+title = "uniaxial-stress-usf"
+type = "MPMExplicit"
+dimension = 2
+scheme = "usf"
+dt = 0.01
+nsteps = 10
+velocity_update = false
+
+[output]
+format = "npz"
+folder = "results/"
+step_frequency = 1
+
+[mesh]
+type = "generator"
+nelements = [1, 1]
+element_length = [1, 1]
+particle_element_ids = [0]
+element = "Quadrilateral4Node"
+
+[[mesh.constraints]]
+node_ids = [0, 1]
+dir = 1
+velocity = 0.0
+
+[[mesh.constraints]]
+node_ids = [2, 3]
+dir = 1
+velocity = -0.01
+
+[[materials]]
+id = 0
+density = 1
+poisson_ratio = 0
+youngs_modulus = 1000
+type = "LinearElastic"
+
+[[particles]]
+file = "particles-2d-uniaxial-stress.json"
+material_id = 0
+init_velocity = [1.0, 0.0]
+
+[external_loading]
+gravity = [0, 0]

benchmarks/2d/uniaxial_stress/particles-2d-uniaxial-stress.json

@@ -0,0 +1,26 @@
+[
+  [
+    [
+      0.25,
+      0.25
+    ]
+  ],
+  [
+    [
+      0.75,
+      0.25
+    ]
+  ],
+  [
+    [
+      0.75,
+      0.75
+    ]
+  ],
+  [
+    [
+      0.25,
+      0.75
+    ]
+  ]
+]

benchmarks/2d/uniaxial_stress/test_benchmark.py

@@ -0,0 +1,17 @@
+import os
+from pathlib import Path
+import jax.numpy as jnp
+from diffmpm.solver import MPM
+
+curr_filepath = Path(__file__).absolute()
+curr_dir = curr_filepath.parent
+os.chdir(curr_dir)
+mpm = MPM("mpm-uniaxial-stress.toml")
+mpm.solve()
+
+result = jnp.load("results/uniaxial-stress-usf/particles_09.npz")
+true_stress_yy = -1.0
+true_stress_xx = 0.0
+
+assert jnp.round(result["stress"][0, :, 1].max() - true_stress_yy, 8) == 0.0
+assert jnp.round(result["stress"][0, :, 0].max() - true_stress_xx, 8) == 0.0