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 MixedGenerator
 
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=3, 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 = MixedGenerator(args.dimensions,
                          args.degree,
                          args.unknowns_per_cell_dof,
                          args.unknowns_per_cell_dof - 1) #for now, keep this as one less
 
project = petsc.Project(project_name = "Mixed", 
                        namespace = [ "benchmarks", "multigrid", "petsc", "mp" ]
                        ) 
                        
                        
solver = petsc.solvers.MixedDiscontinuousGalerkinDiscretisation("Poisson",
                                                                args.degree,
                                                                args.dimensions,
                                                                args.unknowns_per_cell_dof,
                                                                args.meshsize,
                                                                args.meshsize,
                                                                matrices.getCellCellMassMatrix(),     #12x12
                                                                matrices.getCellCellSystemMatrix(),   #12x12
                                                                #for now, we leave this as 32x12
                                                                matrices.getCellToFaceMatrix(),       #32x12
                                                                matrices.getFaceFaceMatrix(),         #8x8
                                                                )
 
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")