def test_basic_ks(self):
prob = Problem()
prob.model = model = Group()
model.add_subsystem('px', IndepVarComp(name="x", val=np.ones((2, ))))
model.add_subsystem('comp', DoubleArrayComp())
model.add_subsystem('ks', KSComponent(width=2))
model.connect('px.x', 'comp.x1')
model.connect('comp.y2', 'ks.g')
model.add_design_var('px.x')
model.add_objective('comp.y1')
model.add_constraint('ks.KS', upper=0.0)
prob.setup(check=False)
prob.run_driver()
assert_rel_error(self, max(prob['comp.y2']), prob['ks.KS'])
def test_lower_flag(self):
prob = Problem()
model = prob.model
model.add_subsystem('px', IndepVarComp('x', val=np.array([5.0, 4.0])))
model.add_subsystem(
'comp', ExecComp('y = 3.0*x', x=np.zeros((2, )), y=np.zeros(
(2, ))))
model.add_subsystem('ks', KSComponent(width=2))
model.connect('px.x', 'comp.x')
model.connect('comp.y', 'ks.g')
model.ks.options['lower_flag'] = True
prob.setup()
prob.run_model()
assert_rel_error(self, prob['ks.KS'], -12.0)
def test_basic(self):
import numpy as np
from openmdao.api import Problem, IndepVarComp, ExecComp
from openmdao.components.ks import KSComponent
prob = Problem()
model = prob.model
model.add_subsystem('px', IndepVarComp('x', val=np.array([5.0, 4.0])))
model.add_subsystem(
'comp', ExecComp('y = 3.0*x', x=np.zeros((2, )), y=np.zeros(
(2, ))))
model.add_subsystem('ks', KSComponent(width=2))
model.connect('px.x', 'comp.x')
model.connect('comp.y', 'ks.g')
prob.setup()
prob.run_model()
assert_rel_error(self, prob['ks.KS'], 15.0)
def setup(self):
E = self.metadata['E']
L = self.metadata['L']
b = self.metadata['b']
volume = self.metadata['volume']
max_bending = self.metadata['max_bending']
num_elements = self.metadata['num_elements']
num_nodes = num_elements + 1
num_cp = self.metadata['num_cp']
num_load_cases = self.metadata['num_load_cases']
parallel_derivs = self.metadata['parallel_derivs']
inputs_comp = IndepVarComp()
inputs_comp.add_output('h_cp', shape=num_cp)
self.add_subsystem('inputs_comp', inputs_comp)
comp = BsplinesComp(num_control_points=num_cp, num_points=num_elements, in_name='h_cp',
out_name='h')
self.add_subsystem('interp', comp)
I_comp = MomentOfInertiaComp(num_elements=num_elements, b=b)
self.add_subsystem('I_comp', I_comp)
comp = LocalStiffnessMatrixComp(num_elements=num_elements, E=E, L=L)
self.add_subsystem('local_stiffness_matrix_comp', comp)
# Parallel Subsystem for load cases.
par = self.add_subsystem('parallel', ParallelGroup())
# Determine how to split cases up over the available procs.
nprocs = self.comm.size
divide = divide_cases(num_load_cases, nprocs)
for j, this_proc in enumerate(divide):
num_rhs = len(this_proc)
name = 'sub_%d' % j
sub = par.add_subsystem(name, Group())
# Load is a sinusoidal distributed force of varying spatial frequency.
force_vector = np.zeros((2 * num_nodes, num_rhs))
for i, k in enumerate(this_proc):
end = 1.5 * np.pi
if num_load_cases > 1:
end += k * 0.5 * np.pi / (num_load_cases - 1)
x = np.linspace(0, end, num_nodes)
f = - np.sin(x)
force_vector[0:-1:2, i] = f
comp = GlobalStiffnessMatrixComp(num_elements=num_elements)
sub.add_subsystem('global_stiffness_matrix_comp', comp)
comp = MultiStatesComp(num_elements=num_elements, force_vector=force_vector,
num_rhs=num_rhs)
sub.add_subsystem('states_comp', comp)
comp = MultiDisplacementsComp(num_elements=num_elements, num_rhs=num_rhs)
sub.add_subsystem('displacements_comp', comp)
comp = MultiStressComp(num_elements=num_elements, E=E, num_rhs=num_rhs)
sub.add_subsystem('stress_comp', comp)
self.connect(
'local_stiffness_matrix_comp.K_local',
'parallel.%s.global_stiffness_matrix_comp.K_local' % name)
sub.connect(
'global_stiffness_matrix_comp.K',
'states_comp.K')
for k in range(num_rhs):
sub.connect(
'states_comp.d_%d' % k,
'displacements_comp.d_%d' % k)
sub.connect(
'displacements_comp.displacements_%d' % k,
'stress_comp.displacements_%d' % k)
comp = KSComponent(width=num_elements)
comp.options['upper'] = max_bending
sub.add_subsystem('KS_%d' % k, comp)
sub.connect(
'stress_comp.stress_%d' % k,
'KS_%d.g' % k)
if parallel_derivs:
color = 'red_%d' % k
else:
color = None
sub.add_constraint('KS_%d.KS' % k, upper=0.0,
parallel_deriv_color=color)
comp = VolumeComp(num_elements=num_elements, b=b, L=L)
self.add_subsystem('volume_comp', comp)
self.connect('inputs_comp.h_cp', 'interp.h_cp')
self.connect('interp.h', 'I_comp.h')
self.connect('I_comp.I', 'local_stiffness_matrix_comp.I')
self.connect('interp.h', 'volume_comp.h')
self.add_design_var('inputs_comp.h_cp', lower=1e-2, upper=10.)
self.add_objective('volume_comp.volume')