Make geometry spline interpolation mesh independent (#435)

* Make geometry spline interpolation mesh independent

* Fix remaining interpolated values

* Fix tests

* flake8 fixes

Author: A-CGray

openaerostruct/geometry/geometry_group.py

@@ -2,6 +2,7 @@
 
 import openmdao.api as om
 from openaerostruct.utils.check_surface_dict import check_surface_dict_keys
+from openaerostruct.utils.interpolation import get_normalized_span_coords
 
 
 class Geometry(om.Group):
@@ -29,7 +30,6 @@ def setup(self):
 
         # Get the surface name and create a group to contain components
         # only for this surface
-        ny = surface["mesh"].shape[1]
 
         if self.options["DVGeo"]:
             from openaerostruct.geometry.ffd_component import GeometryMesh
@@ -39,7 +39,7 @@ def setup(self):
             if "t_over_c_cp" in surface.keys():
                 n_cp = len(surface["t_over_c_cp"])
                 # Add bspline components for active bspline geometric variables.
-                x_interp = np.linspace(0.0, 1.0, int(ny - 1))
+                x_interp = get_normalized_span_coords(surface, mid_panel=True)
                 comp = self.add_subsystem(
                     "t_over_c_bsp",
                     om.SplineComp(
@@ -67,7 +67,7 @@ def setup(self):
             if "twist_cp" in surface.keys():
                 n_cp = len(surface["twist_cp"])
                 # Add bspline components for active bspline geometric variables.
-                x_interp = np.linspace(0.0, 1.0, int(ny))
+                x_interp = get_normalized_span_coords(surface)
                 comp = self.add_subsystem(
                     "twist_bsp",
                     om.SplineComp(
@@ -86,7 +86,7 @@ def setup(self):
             if "chord_cp" in surface.keys():
                 n_cp = len(surface["chord_cp"])
                 # Add bspline components for active bspline geometric variables.
-                x_interp = np.linspace(0.0, 1.0, int(ny))
+                x_interp = get_normalized_span_coords(surface)
                 comp = self.add_subsystem(
                     "chord_bsp",
                     om.SplineComp(
@@ -103,7 +103,7 @@ def setup(self):
             if "t_over_c_cp" in surface.keys():
                 n_cp = len(surface["t_over_c_cp"])
                 # Add bspline components for active bspline geometric variables.
-                x_interp = np.linspace(0.0, 1.0, int(ny - 1))
+                x_interp = get_normalized_span_coords(surface, mid_panel=True)
                 comp = self.add_subsystem(
                     "t_over_c_bsp",
                     om.SplineComp(
@@ -119,7 +119,7 @@ def setup(self):
             if "xshear_cp" in surface.keys():
                 n_cp = len(surface["xshear_cp"])
                 # Add bspline components for active bspline geometric variables.
-                x_interp = np.linspace(0.0, 1.0, int(ny))
+                x_interp = get_normalized_span_coords(surface)
                 comp = self.add_subsystem(
                     "xshear_bsp",
                     om.SplineComp(
@@ -136,7 +136,7 @@ def setup(self):
             if "yshear_cp" in surface.keys():
                 n_cp = len(surface["yshear_cp"])
                 # Add bspline components for active bspline geometric variables.
-                x_interp = np.linspace(0.0, 1.0, int(ny))
+                x_interp = get_normalized_span_coords(surface)
                 comp = self.add_subsystem(
                     "yshear_bsp",
                     om.SplineComp(
@@ -153,7 +153,7 @@ def setup(self):
             if "zshear_cp" in surface.keys():
                 n_cp = len(surface["zshear_cp"])
                 # Add bspline components for active bspline geometric variables.
-                x_interp = np.linspace(0.0, 1.0, int(ny))
+                x_interp = get_normalized_span_coords(surface)
                 comp = self.add_subsystem(
                     "zshear_bsp",
                     om.SplineComp(

openaerostruct/structures/tube_group.py

@@ -1,7 +1,7 @@
-import numpy as np
 import openmdao.api as om
 from openaerostruct.structures.section_properties_tube import SectionPropertiesTube
 from openaerostruct.geometry.radius_comp import RadiusComp
+from openaerostruct.utils.interpolation import get_normalized_span_coords
 
 
 class TubeGroup(om.Group):
@@ -14,13 +14,11 @@ def initialize(self):
 
     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))
+            x_interp = get_normalized_span_coords(surface, mid_panel=True)
             comp = self.add_subsystem(
                 "thickness_bsp",
                 om.SplineComp(
@@ -35,7 +33,7 @@ def setup(self):
         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))
+            x_interp = get_normalized_span_coords(surface, mid_panel=True)
             comp = self.add_subsystem(
                 "radius_bsp",
                 om.SplineComp(

openaerostruct/structures/wingbox_group.py

@@ -1,7 +1,7 @@
-import numpy as np
 import openmdao.api as om
 from openaerostruct.structures.section_properties_wingbox import SectionPropertiesWingbox
 from openaerostruct.structures.wingbox_geometry import WingboxGeometry
+from openaerostruct.utils.interpolation import get_normalized_span_coords
 
 
 class WingboxGroup(om.Group):
@@ -12,12 +12,11 @@ def initialize(self):
 
     def setup(self):
         surface = self.options["surface"]
-        ny = surface["mesh"].shape[1]
 
         if "spar_thickness_cp" in surface.keys():
             n_cp = len(surface["spar_thickness_cp"])
             # Add bspline components for active bspline geometric variables.
-            x_interp = np.linspace(0.0, 1.0, int(ny - 1))
+            x_interp = get_normalized_span_coords(surface, mid_panel=True)
             comp = self.add_subsystem(
                 "spar_thickness_bsp",
                 om.SplineComp(
@@ -32,7 +31,7 @@ def setup(self):
         if "skin_thickness_cp" in surface.keys():
             n_cp = len(surface["skin_thickness_cp"])
             # Add bspline components for active bspline geometric variables.
-            x_interp = np.linspace(0.0, 1.0, int(ny - 1))
+            x_interp = get_normalized_span_coords(surface, mid_panel=True)
             comp = self.add_subsystem(
                 "skin_thickness_bsp",
                 om.SplineComp(

openaerostruct/tests/test_aero_analysis_no_symmetry_wavedrag.py

@@ -100,9 +100,9 @@ def test(self):
 
         prob.run_driver()
 
-        assert_near_equal(prob["aero_point_0.wing_perf.CL"][0], 0.464191542231, 1e-6)
-        assert_near_equal(prob["aero_point_0.wing_perf.CD"][0], 0.020417875291205534, 1e-6)
-        assert_near_equal(prob["aero_point_0.CM"][1], -1.859570880469676, 1e-6)
+        assert_near_equal(prob["aero_point_0.CL"][0], 0.46419154063077483, 1e-6)
+        assert_near_equal(prob["aero_point_0.CD"][0], 0.020863555824806052, 1e-6)
+        assert_near_equal(prob["aero_point_0.CM"][1], -1.8595708973535592, 1e-6)
 
 
 if __name__ == "__main__":

openaerostruct/tests/test_aero_opt_wavedrag.py

@@ -5,6 +5,7 @@
 from openaerostruct.geometry.utils import generate_mesh
 from openaerostruct.geometry.geometry_group import Geometry
 from openaerostruct.aerodynamics.aero_groups import AeroPoint
+from openaerostruct.utils.testing import assert_opt_successful
 
 import openmdao.api as om
 
@@ -106,11 +107,13 @@ def test(self):
         # Set up the problem
         prob.setup()
 
-        prob.run_driver()
+        optResult = prob.run_driver()
 
-        assert_near_equal(prob["aero_point_0.wing_perf.CL"][0], 0.5, 1e-6)
-        assert_near_equal(prob["aero_point_0.wing_perf.CD"][0], 0.020838936785019083, 1e-6)
-        assert_near_equal(prob["aero_point_0.CM"][1], -2.081989092575424, 1e-6)
+        assert_opt_successful(self, optResult)
+
+        assert_near_equal(prob["aero_point_0.CL"][0], 0.5, 1e-6)
+        assert_near_equal(prob["aero_point_0.CD"][0], 0.021353004050991248, 1e-6)
+        assert_near_equal(prob["aero_point_0.CM"][1], -2.0819892547514067, 1e-6)
 
 
 if __name__ == "__main__":

openaerostruct/tests/test_aerostruct.py

@@ -1,5 +1,6 @@
 from openmdao.utils.assert_utils import assert_near_equal
 import unittest
+from openaerostruct.utils.testing import assert_opt_successful
 
 
 class Test(unittest.TestCase):
@@ -144,9 +145,10 @@ def test(self):
         # group.
         assert_near_equal(prob["AS_point_0.beta"], 0.0)
 
-        prob.run_driver()
+        optResult = prob.run_driver()
+        assert_opt_successful(self, optResult)
 
-        assert_near_equal(prob["AS_point_0.fuelburn"][0], 92369.79279575823, 1e-8)
+        assert_near_equal(prob["AS_point_0.fuelburn"][0], 92523.945549167, 1e-8)
 
 
 if __name__ == "__main__":

openaerostruct/tests/test_aerostruct_analysis.py

@@ -137,8 +137,8 @@ def test(self):
 
         prob.run_model()
 
-        assert_near_equal(prob["AS_point_0.fuelburn"][0], 251362.24734663023, 1e-4)
-        assert_near_equal(prob["AS_point_0.CM"][1], -0.7033677364356814, 1e-5)
+        assert_near_equal(prob["AS_point_0.fuelburn"][0], 263398.25938918366, 1e-4)
+        assert_near_equal(prob["AS_point_0.CM"][1], -0.6462808405237332, 1e-5)
 
 
 if __name__ == "__main__":

openaerostruct/tests/test_aerostruct_analysis_Sref.py

@@ -139,8 +139,8 @@ def test(self):
 
         prob.run_model()
 
-        assert_near_equal(prob["AS_point_0.CL"][0], 1.6217443031469607, 1e-6)
-        assert_near_equal(prob["AS_point_0.CM"][1], -1.9988780195141023, 1e-5)
+        assert_near_equal(prob["AS_point_0.CL"][0], 1.6210175228727655, 1e-6)
+        assert_near_equal(prob["AS_point_0.CM"][1], -1.8365760768848112, 1e-5)
 
 
 if __name__ == "__main__":

openaerostruct/tests/test_aerostruct_analysis_compressible.py

@@ -139,8 +139,8 @@ def test(self):
 
         prob.run_model()
 
-        assert_near_equal(prob["AS_point_0.fuelburn"][0], 213840.78859689648, 1e-4)
-        assert_near_equal(prob["AS_point_0.CM"][1], -0.9866929184880228, 1e-5)
+        assert_near_equal(prob["AS_point_0.fuelburn"][0], 224121.12881258246, 1e-4)
+        assert_near_equal(prob["AS_point_0.CM"][1], -0.9083682371351329, 1e-5)
 
 
 if __name__ == "__main__":

openaerostruct/tests/test_aerostruct_engine_thrusts.py

@@ -156,8 +156,8 @@ def test(self):
         print(prob["AS_point_0.fuelburn"][0])
         print(prob["AS_point_0.CM"][1])
 
-        assert_near_equal(prob["AS_point_0.fuelburn"][0], 251929.9085951508, 1e-4)
-        assert_near_equal(prob["AS_point_0.CM"][1], -0.7008367976235399, 1e-5)
+        assert_near_equal(prob["AS_point_0.fuelburn"][0], 263992.6780138112, 1e-4)
+        assert_near_equal(prob["AS_point_0.CM"][1], -0.6438933659444002, 1e-5)
 
 
 if __name__ == "__main__":