Homec4science
Diffusion RationalROMPy (master) Browse

RROMPy/examples/fenics/LagrangeSweep.py
c11471199317master

Last Change
Raw File

LagrangeSweep.py
View Options

from copy import copy
import numpy as np
from rrompy.hfengines.fenics import HelmholtzSquareBubbleProblemEngine as HSBPE
from rrompy.hfengines.fenics import HelmholtzBoxScatteringProblemEngine as HBSPE
from rrompy.hsengines.fenics import HSEngine as HS
from rrompy.reduction_methods.lagrange import ApproximantLagrangePade as Pade
from rrompy.reduction_methods.lagrange import ApproximantLagrangeRB as RB
from rrompy.utilities.parameter_sweeper import ParameterSweeper as Sweeper
from rrompy.sampling import QuadratureSampler as QS
testNo = 2
npoints = 31
if testNo == 1:
ks = [4 + .5j, 14 + .5j]
solver = HSBPE(kappa = 12 ** .5, theta = np.pi / 3, n = 10)
solver.omega = np.real(np.mean(ks))
mutars = np.linspace(0, 21, npoints)
filenamebase = '../data/output/HelmholtzBubbleLagrange'
elif testNo == 2:
ks = [10 + .25j, 14 + .25j]
solver = HBSPE(R = 5, kappa = 3, theta = - np.pi * 75 / 180, n = 10)
solver.omega = np.real(np.mean(ks))
mutars = np.linspace(9, 14, npoints)
filenamebase = '../data/output/HelmholtzBoxLagrange'
plotter = HS(solver.V)
k0 = np.mean(ks)
shift = 12
nsets = 5
stride = 3
Smax = stride * (nsets - 1) + shift + 2
paramsPade = {'S':Smax, 'POD':True, 'sampler':QS(ks, "CHEBYSHEV")}
paramsRB = copy(paramsPade)
paramsSetsPade = [None] * nsets
paramsSetsRB = [None] * nsets
for i in range(nsets):
paramsSetsPade[i] = {'N': stride * i + shift + 1, 'M': stride * i + shift,
'S': stride * i + shift + 2}
paramsSetsRB[i] = {'R': stride * i + shift + 1, 'S': stride * i + shift + 2}
appPade = Pade(solver, plotter, mu0 = k0, approxParameters = paramsPade)
appRB = RB(solver, plotter, mu0 = k0, approxParameters = paramsRB)
sweeper = Sweeper(mutars = mutars, mostExpensive = 'Approx')
sweeper.ROMEngine = appPade
sweeper.params = paramsSetsPade
filenamePade = sweeper.sweep(filenamebase + 'PadeFE.dat', outputs = 'ALL')
sweeper.ROMEngine = appRB
sweeper.params = paramsSetsRB
filenameRB = sweeper.sweep(filenamebase + 'RBFE.dat', outputs = 'ALL')
sweeper.plot(filenamePade, ['muRe'], ['HFNorm', 'AppNorm'], ['S'],
onePlot = True)
sweeper.plot(filenameRB, ['muRe'], ['HFNorm', 'AppNorm'], ['S'],
onePlot = True)
sweeper.plot(filenamePade, ['muRe'], ['ErrNorm'], ['S'])
sweeper.plot(filenameRB, ['muRe'], ['ErrNorm'], ['S'])
c4science · Help