import numpy as np
import openmdao.api as om
from openaerostruct.structures.section_properties_tube import SectionPropertiesTube
from openaerostruct.geometry.radius_comp import RadiusComp
[docs]
class TubeGroup(om.Group):
"""Group that contains everything needed for a structural-only problem."""
def initialize(self):
self.options.declare("surface", types=dict)
self.options.declare("connect_geom_DVs", default=True)
# The option "connect_geom_DVs" is no longer necessary, but we still keep it to be backward compatible.
def setup(self):
surface = self.options["surface"]
mesh = surface["mesh"]
ny = mesh.shape[1]
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.0, 1.0, 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", y_units="m")
self.set_input_defaults("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.0, 1.0, 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", y_units="m")
self.set_input_defaults("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"],
)