poisson.py

Go to the documentation of this file.

"""
 
 Simple Poisson solver on the unit cube.
 
"""
 
 
import os, sys
import argparse
import peano4
import petsc
 
print( """
 
Peano 4 multigrid solver using PETSc
 
@author 2022 Sean Baccas, Dmitry Nikolaenko, Tobias Weinzierl
 
""")
 
parser = argparse.ArgumentParser(description='Multigrid - Poisson solver')
parser.add_argument("-j",   "--parallel-builds",    dest="j",              type=int, default=-1, help="Parallel builds" )
parser.add_argument("-pd",  "--peano-dir",          dest="peanodir",       default="../../../../", help="Peano4 directory" )
parser.add_argument("-v",   "--verbose",            dest="verbose",        action="store_true", default=False, help="Verbose")
parser.add_argument("-d",   "--dimensions",         dest="dimensions",     default=2, help="Dimensions")
parser.add_argument("-meshsize",  "--meshsize",     dest="meshsize",       default=0.01, help="Mesh size")
parser.add_argument("-m",   "--mode",               dest="mode",           choices=["release","stats","asserts"], required=True, help="Pick build type" )
args = parser.parse_args()
 
#add a 9 point stencil which will 
#be passed to PETSc at matrix initialisation stage
stencil=[
  [-1/3, -1/3, -1/3],
  [-1/3,  8/3, -1/3],
  [-1/3, -1/3, -1/3]
]
 
project = petsc.Project(project_name = "Poisson", 
                        namespace = [ "benchmarks", "multigrid", "petsc", "poisson" ]
                        ) 
                        
                        
solver = petsc.solvers.CollocatedLowOrderDiscretisation("Poisson", 
                                                        1,
                                                        args.meshsize,
                                                        args.meshsize,
                                                        stencil
                                                        )
 
project.add_solver( solver )
                        
#
# Configure build
#
if args.mode=="release":
  build_mode = peano4.output.CompileMode.Release
if args.mode=="stats":
  build_mode = peano4.output.CompileMode.Stats
if args.mode=="asserts":
  build_mode = peano4.output.CompileMode.Asserts
 
cube_size = 1.0
project.set_global_simulation_parameters(
  dimensions            = args.dimensions,
  offset                = [ 0.0       for _ in range(args.dimensions) ],
  domain_size           = [ cube_size for _ in range(args.dimensions) ],
)
 
 
project.set_load_balancing( "toolbox::loadbalancing::strategies::RecursiveSubdivision", "new ::exahype2::LoadBalancingConfiguration()" )
project.set_Peano4_installation( args.peanodir, build_mode )
    
    
#
# Generate code and then copy cpp file over main
#
peano4_project = project.generate_Peano4_project(args.verbose)
peano4_project.generate()
 
os.system( "make clean" )
if args.j>0:
  os.system( "make -j{}".format(args.j) )
else:
  os.system( "make -j" )
  
print("Done")