def setup(self):
surface = self.options['surface']
ny = surface['num_y']
# Add independent variables that do not belong to a specific component
indep_var_comp = IndepVarComp()
# Add structural components to the surface-specific group
self.add_subsystem('indep_vars',
indep_var_comp,
promotes=['*'])
if 'struct' in surface['type']:
self.add_subsystem('radius_comp',
RadiusComp(surface=surface),
promotes_inputs=['mesh'],
promotes_outputs=['radius'])
if 'thickness_cp' in surface.keys():
n_cp = len(surface['thickness_cp'])
# Add bspline components for active bspline geometric variables.
self.add_subsystem('thickness_bsp', BsplinesComp(
in_name='thickness_cp', out_name='thickness',
num_control_points=n_cp, num_points=int(ny-1),
bspline_order=min(n_cp, 4), distribution='uniform'),
promotes_inputs=['thickness_cp'], promotes_outputs=['thickness'])
indep_var_comp.add_output('thickness_cp', val=surface['thickness_cp'], units='m')
self.add_subsystem('tube',
SectionPropertiesTube(surface=surface),
promotes_inputs=['thickness', 'radius'],
promotes_outputs=['A', 'Iy', 'Iz', 'J'])
def setup(self):
surface = self.options['surface']
connect_geom_DVs = self.options['connect_geom_DVs']
mesh = surface['mesh']
ny = mesh.shape[1]
if connect_geom_DVs:
# Add independent variables that do not belong to a specific component
indep_var_comp = om.IndepVarComp()
# Add structural components to the surface-specific group
self.add_subsystem('indep_vars', indep_var_comp, promotes=['*'])
if 'thickness_cp' in surface.keys():
n_cp = len(surface['thickness_cp'])
# Add bspline components for active bspline geometric variables.
x_interp = np.linspace(0., 1., int(ny - 1))
comp = self.add_subsystem(
'thickness_bsp',
om.SplineComp(method='bsplines',
x_interp_val=x_interp,
num_cp=n_cp,
interp_options={'order': min(n_cp, 4)}),
promotes_inputs=['thickness_cp'],
promotes_outputs=['thickness'])
comp.add_spline(y_cp_name='thickness_cp',
y_interp_name='thickness')
if connect_geom_DVs:
indep_var_comp.add_output('thickness_cp',
val=surface['thickness_cp'],
units='m')
if 'radius_cp' in surface.keys():
n_cp = len(surface['radius_cp'])
# Add bspline components for active bspline geometric variables.
x_interp = np.linspace(0., 1., int(ny - 1))
comp = self.add_subsystem(
'radius_bsp',
om.SplineComp(method='bsplines',
x_interp_val=x_interp,
num_cp=n_cp,
interp_options={'order': min(n_cp, 4)}),
promotes_inputs=['radius_cp'],
promotes_outputs=['radius'])
comp.add_spline(y_cp_name='radius_cp', y_interp_name='radius')
if connect_geom_DVs:
indep_var_comp.add_output('radius_cp',
val=surface['radius_cp'],
units='m')
else:
self.add_subsystem('radius_comp',
RadiusComp(surface=surface),
promotes_inputs=['mesh', 't_over_c'],
promotes_outputs=['radius'])
self.add_subsystem('tube',
SectionPropertiesTube(surface=surface),
promotes_inputs=['thickness', 'radius'],
promotes_outputs=['A', 'Iy', 'Iz', 'J'])
def test(self):
surfaces = get_default_surfaces()
group = Group()
comp = RadiusComp(surface=surfaces[0])
indep_var_comp = IndepVarComp()
indep_var_comp.add_output('mesh', val=surfaces[0]['mesh'], units='m')
group.add_subsystem('radius', comp)
group.add_subsystem('indep_var_comp', indep_var_comp)
group.connect('indep_var_comp.mesh', 'radius.mesh')
run_test(self, group)
def test(self):
surfaces = get_default_surfaces()
group = om.Group()
comp = RadiusComp(surface=surfaces[0])
ny = surfaces[0]['mesh'].shape[1]
indep_var_comp = om.IndepVarComp()
indep_var_comp.add_output('mesh', val=surfaces[0]['mesh'], units='m')
indep_var_comp.add_output('t_over_c', val=np.linspace(0.1,0.5,num=ny-1))
group.add_subsystem('radius', comp)
group.add_subsystem('indep_var_comp', indep_var_comp)
group.connect('indep_var_comp.mesh', 'radius.mesh')
group.connect('indep_var_comp.t_over_c', 'radius.t_over_c')
run_test(self, group)