import openmdao.api as om
from openmdao.test_suite.components.sellar import SellarDis1withDerivatives, SellarDis2withDerivatives
from openmdao.test_suite.components.sellar import SellarImplicitDis1
# TODO -- Need to convert these over to use `setup` after we have two setup
# phases split up and have the ability to change driver settings
[docs]class SubSellar(om.Group):
[docs] def __init__(self, units=None, scaling=None, **kwargs):
super().__init__(**kwargs)
self.add_subsystem('d1', SellarDis1withDerivatives(units=units, scaling=scaling),
promotes=['x', 'z', 'y1', 'y2'])
self.add_subsystem('d2', SellarDis2withDerivatives(units=units, scaling=scaling),
promotes=['z', 'y1', 'y2'])
if units:
# auto_ivc update requires this since two 'z' inputs have different units
self.set_input_defaults('z', units='ft')
[docs]class DoubleSellar(om.Group):
[docs] def __init__(self, units=None, scaling=None, **kwargs):
super().__init__(**kwargs)
self.add_subsystem('g1', SubSellar(units=units, scaling=scaling))
self.add_subsystem('g2', SubSellar(units=units, scaling=scaling))
self.connect('g1.y2', 'g2.x')
self.connect('g2.y2', 'g1.x')
# Converge the outer loop with Gauss Seidel, with a looser tolerance.
self.nonlinear_solver = om.NewtonSolver(solve_subsystems=False)
self.linear_solver = om.DirectSolver()
[docs]class SubSellarImplicit(om.Group):
[docs] def __init__(self, units=None, scaling=None, **kwargs):
super().__init__(**kwargs)
self.add_subsystem('d1', SellarImplicitDis1(units=units, scaling=scaling),
promotes=['x', 'z', 'y1', 'y2'])
self.add_subsystem('d2', SellarImplicitDis1(units=units, scaling=scaling),
promotes=['z', 'y1', 'y2'])
[docs]class DoubleSellarImplicit(om.Group):
[docs] def __init__(self, units=None, scaling=None, **kwargs):
super().__init__(**kwargs)
self.add_subsystem('g1', SubSellar(units=units, scaling=scaling))
self.add_subsystem('g2', SubSellar(units=units, scaling=scaling))
self.connect('g1.y2', 'g2.x')
self.connect('g2.y2', 'g1.x')
# Converge the outer loop with Gauss Seidel, with a looser tolerance.
self.nonlinear_solver = om.NewtonSolver(solve_subsystems=False)
self.linear_solver = om.DirectSolver()
