EHN: add prints and plots to linear generic surface files

RocketPy-Team · Sep 15, 2024 · e9a8274 · e9a8274
1 parent 1ed8a9e
commit e9a8274
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/rocketpy/rocket/aero_surface/linear_generic_surface.py b/rocketpy/rocket/aero_surface/linear_generic_surface.py
@@ -1,5 +1,7 @@
 from rocketpy.mathutils import Function
 from rocketpy.rocket.aero_surface.generic_surface import GenericSurface
+from rocketpy.prints.aero_surface_prints import _LinearGenericSurfacePrints
+from rocketpy.plots.aero_surface_plots import _LinearGenericSurfacePlots
 
 
 class LinearGenericSurface(GenericSurface):
@@ -166,6 +168,9 @@ def __init__(
 
         self.compute_all_coefficients()
 
+        self.prints = _LinearGenericSurfacePrints(self)
+        self.plots = _LinearGenericSurfacePlots(self)
+
     def _get_default_coefficients(self):
         """Returns default coefficients