poisson.py

Go to the documentation of this file.

"""
 
FOR NOW, WE FIX THE NUMBER OF DOFS TO BE THE SAME AS 
THE NUMBER OF GAUSSIAN POINTS
 
 
"""
 
 
import os, sys
import argparse
import peano4.api
import petsc
from matrices import DgGenerator
 
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",     type=int, default=2, help="Dimensions")
parser.add_argument("-deg", "--degree",               dest="degree",         type=int, default=1, help="Polynomial degree")
parser.add_argument("-upcd","--unknowns_per_cell_dof",dest="unknowns_per_cell_dof", type=int, default=1, help="Unknowns per cell dof")
parser.add_argument("-meshsize",  "--meshsize",       dest="meshsize",       default=0.1, help="Mesh size")
parser.add_argument("-m",   "--mode",                 dest="mode",           choices=["release","stats","asserts"], required=True, help="Pick build type" )
args = parser.parse_args()
 
#generate matrices using file next door
matrices = DgGenerator(args.dimensions,
                       args.degree,
                       args.unknowns_per_cell_dof)
 
project = petsc.Project(project_name = "DG", 
                        namespace = [ "benchmarks", "multigrid", "petsc", "DG" ]
                        ) 
                        
                        
solver = petsc.solvers.DiscontinuousGalerkinDiscretisation("Poisson",
                                                            args.degree,
                                                            args.dimensions,
                                                            args.unknowns_per_cell_dof,
                                                            args.meshsize,
                                                            args.meshsize,
                                                            matrices.getCellCellMassMatrix(),
                                                            matrices.getCellCellSystemMatrix(),
                                                            matrices.getCellToFaceMatrix(),
                                                            matrices.getFaceFaceMatrix(),
                                                            )
 
project.add_solver( solver )
                        
#
# Configure build
#
if args.mode=="release":
  build_mode = peano4.api.output.CompileMode.RELEASE
if args.mode=="stats":
  build_mode = peano4.api.output.CompileMode.STATS
if args.mode=="asserts":
  build_mode = peano4.api.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")