75 lines
3.1 KiB
Python
75 lines
3.1 KiB
Python
""" Poisson OCP problem. """
|
|
|
|
|
|
from pina import Condition
|
|
from pina.domain import CartesianDomain
|
|
from pina.equation import SystemEquation, FixedValue
|
|
from pina.problem import SpatialProblem, ParametricProblem
|
|
from pina.operators import laplacian
|
|
|
|
# ===================================================== #
|
|
# #
|
|
# This script implements the two dimensional #
|
|
# Parametric Elliptic Optimal Control problem. #
|
|
# The ParametricEllipticOptimalControl class is #
|
|
# inherited from TimeDependentProblem, SpatialProblem #
|
|
# and we denote: #
|
|
# u --> field variable #
|
|
# p --> field variable #
|
|
# y --> field variable #
|
|
# x1, x2 --> spatial variables #
|
|
# mu, alpha --> problem parameters #
|
|
# #
|
|
# More info in https://arxiv.org/pdf/2110.13530.pdf #
|
|
# Section 4.2 of the article #
|
|
# ===================================================== #
|
|
|
|
|
|
class ParametricEllipticOptimalControl(SpatialProblem, ParametricProblem):
|
|
|
|
# setting spatial variables ranges
|
|
xmin, xmax, ymin, ymax = -1, 1, -1, 1
|
|
x_range = [xmin, xmax]
|
|
y_range = [ymin, ymax]
|
|
# setting parameters range
|
|
amin, amax = 0.01, 1
|
|
mumin, mumax = 0.5, 3
|
|
mu_range = [mumin, mumax]
|
|
a_range = [amin, amax]
|
|
# setting field variables
|
|
output_variables = ['u', 'y', 'z']
|
|
# setting spatial and parameter domain
|
|
spatial_domain = CartesianDomain({'x1': x_range, 'x2': y_range})
|
|
parameter_domain = CartesianDomain({'mu': mu_range, 'alpha': a_range})
|
|
|
|
# equation terms as in https://arxiv.org/pdf/2110.13530.pdf
|
|
def term1(input_, output_):
|
|
laplace_z = laplacian(output_, input_, components=['z'], d=['x1', 'x2'])
|
|
return output_.extract(['y']) - input_.extract(['mu']) - laplace_z
|
|
|
|
def term2(input_, output_):
|
|
laplace_y = laplacian(output_, input_, components=['y'], d=['x1', 'x2'])
|
|
return - laplace_y - output_.extract(['u'])
|
|
|
|
|
|
# setting problem condition formulation
|
|
conditions = {
|
|
'gamma1': Condition(
|
|
location=CartesianDomain({'x1': x_range, 'x2': 1, 'mu': mu_range, 'alpha': a_range}),
|
|
equation=FixedValue(0, ['y',])),
|
|
'gamma2': Condition(
|
|
location=CartesianDomain({'x1': x_range, 'x2': -1, 'mu': mu_range, 'alpha': a_range}),
|
|
equation=FixedValue(0, ['y', 'z'])),
|
|
'gamma3': Condition(
|
|
location=CartesianDomain({'x1': 1, 'x2': y_range, 'mu': mu_range, 'alpha': a_range}),
|
|
equation=FixedValue(0, ['y', 'z'])),
|
|
'gamma4': Condition(
|
|
location=CartesianDomain({'x1': -1, 'x2': y_range, 'mu': mu_range, 'alpha': a_range}),
|
|
equation=FixedValue(0, ['y', 'z'])),
|
|
'D': Condition(
|
|
location=CartesianDomain(
|
|
{'x1': x_range, 'x2': y_range,
|
|
'mu': mu_range, 'alpha': a_range
|
|
}),
|
|
equation=SystemEquation([term1, term2])),
|
|
} |