dastgen-test.py

Go to the documentation of this file.

import os
import argparse
import numpy as np
 
import peano4.api
import exaparty
import dastgen2
from exaparty.api.particle.tests import DastgenTestDummyParticle
 
import shutil
 
 
 
 
parser = argparse.ArgumentParser(description="Dastgen2 testing script")
parser.add_argument(
    "-j", "--parallel-builds", dest="j", type=int, default=-1, help="Parallel builds"
)
parser.add_argument(
    "-v",
    "--verbose",
    dest="verbose",
    action="store_true",
    default=False,
    help="Verbose",
)
parser.add_argument(
    "-pd",
    "--peano-dir",
    dest="peanodir",
    default="../../..",
    help="Peano4 directory",
)
args = parser.parse_args()
 
 
 
 
project_namespace = ["tests", "exaparty", "dastgenTest"]
 
project = exaparty.api.Project(
    namespace=project_namespace, project_name="dastgen_test", executable="dastgen_test"
)
 
 
# Initialise the Dummy particle --------------------------------------------------
dimensions = 2  # Peano dimensions
cell_size = 0.3
 
 
particle = DastgenTestDummyParticle(
    "dummyPart",
    particles_per_cell=20,
    min_h=cell_size / 100,
    max_h=cell_size,
)
 
# Add particle to project --------------------------------------------
# Notice that any change to the particle will not take effect after adding
# the particle_set to the project
particle_set = project.add_particle_species(particle)
 
# Always build with asserts on.
build_mode = peano4.api.output.CompileMode.ASSERTS
 
# Set Peano4 simulation parameters ---------------------------------------------
offset = [0, 0, 0]
domain_size = [1, 1, 1]
periodic_boundary_conditions = [False, False, False]
 
project.set_global_simulation_parameters(
    dimensions=dimensions,
    offset=offset,
    domain_size=domain_size,
    min_end_time=1.0,
    max_end_time=1.0,
    first_plot_time_stamp=0.0,
    time_in_between_plots=1.0,
    periodic_BC=periodic_boundary_conditions,
    plotter_precision=8,
)
 
project.set_Peano4_installation(args.peanodir, build_mode)
project._project.output.makefile.add_cpp_file("DastgenTest.cpp", generated=False)
 
 
 
 
 
# Set up the coordinates for the particles
# We construct a regular (lattice) configuration.
Npart = 10
offset = 1 / Npart
 
# This should always be a 3D array, even if you are using fewer
# dimensions in your actual run
coords = np.zeros((Npart**dimensions, 3))
 
x = np.linspace(offset, 1, Npart, endpoint=False)
y = np.linspace(offset, 1, Npart, endpoint=False)
 
xp, yp = np.meshgrid(x, y)
xp = xp.reshape((np.prod(xp.shape),))
yp = yp.reshape((np.prod(yp.shape),))
 
coords[:, 0] = xp
coords[:, 1] = yp
 
initialisation_snippet = """
      /*
       * Initial Conditions: None. The dummy particles aren't even proper SPH
       * particles. Notably, they don't even contain fields like mass, density,
       * etc. Nothing to do here except read in particle positions (which is
       * done elsewhere) and give particles a search radius, which needs to
       * exist.
       */
      tarch::la::Vector<Dimensions,double> GridSize = marker.h();
      const double GridSize_x = GridSize(0);
      particle->setSearchRadius( GridSize_x / 4.0);
"""
 
#  Finally, pass this configuration to the initialisation script
project.algorithm_step_initial_conditions.add_action_set(
    exaparty.api.input.InsertParticlesByCoordinates(
        particle_set=particle_set,
        initialisation_call=initialisation_snippet,
        coordinates=coords,
        additional_includes="",
    )
)
 
 
 
 
peano4_project = project.generate_Peano4_project(verbose=args.verbose)
 
 
 
 
# Always build project from scratch
dirs = ["vertexdata/", "repositories/", "globaldata/", "observers/"]
for dir in dirs:
    if os.path.exists(dir):
        shutil.rmtree(dir)
 
# It is safe (and recommended) to overwrite any existing main file, since this
# is generated automatically by the ExaParty graph compiler.
peano4_project.generate(
    overwrite=peano4.api.output.Overwrite.Always, throw_away_data_after_generation=True
)
peano4_project.build(make_clean_first=True, number_of_parallel_builds=args.j)
 
# remove output files from previous runs
output_files = [
    f
    for f in os.listdir(".")
    if f.endswith(".peano-patch-file") or f.endswith(".vtu") or f.endswith(".pvd")
]
for f in output_files:
    os.remove(f)