dymos 1.13.1__py3-none-any.whl

dymos/__init__.py

@@ -0,0 +1,12 @@
+__version__ = '1.13.1'
+
+
+from .phase import Phase, AnalyticPhase
+from .transcriptions import GaussLobatto, Radau, ExplicitShooting, Analytic, \
+    Birkhoff, PicardShooting
+from .transcriptions.grid_data import GaussLobattoGrid, ChebyshevGaussLobattoGrid, \
+    RadauGrid, UniformGrid, BirkhoffGrid
+from .trajectory.trajectory import Trajectory
+from .run_problem import run_problem
+from .load_case import load_case
+from ._options import options

dymos/_options.py

@@ -0,0 +1,30 @@
+import os
+
+from openmdao.api import OptionsDictionary
+
+
+options = OptionsDictionary()
+
+_env_check_partials = os.environ.get('DYMOS_CHECK_PARTIALS', '0')
+_icp_default = _env_check_partials.lower() in ('1', 'yes', 'true')
+
+
+def _removed_option(name, value):
+    if value is not None:
+        raise ValueError(f'Option {name} has been replaced by '
+                         'Phase.timeseries_options["use_prefix"].')
+
+
+options.declare('include_check_partials', default=_icp_default, types=bool,
+                desc='If True, include dymos components when checking partials.')
+
+options.declare('plots', default='bokeh', values=['matplotlib', 'bokeh'],
+                desc='The plot library used to generate output plots for Dymos.')
+
+options.declare('notebook_mode', default=False, types=bool,
+                desc='If True, provide notebook-enhanced plots and outputs.')
+
+options.declare('use_timeseries_prefix', default=None, allow_none=True,
+                check_valid=_removed_option,
+                desc='Note: This option is no longer valid and has been '
+                     'replaced by Phase.timeseries_options["use_prefix"].')

dymos/examples/aircraft_steady_flight/aero/aero_coef_comp.py

@@ -0,0 +1,19 @@
+import numpy as np
+import openmdao.api as om
+
+from .crm_data import h_bp, alpha_bp, mach_bp, eta_bp, CL_data, CD_data, CM_data
+
+
+class AeroCoefComp(om.MetaModelStructuredComp):
+    """ Interpolates aerodynamic coefficients for the NASA Common Research Model. """
+
+    def setup(self):
+        nn = self.options['vec_size']
+        self.add_input(name='mach', val=0.2 * np.ones(nn), units=None, training_data=mach_bp)
+        self.add_input(name='alpha', val=0.0 * np.ones(nn), units='deg', training_data=alpha_bp)
+        self.add_input(name='alt', val=0.0 * np.ones(nn), units='ft', training_data=h_bp)
+        self.add_input(name='eta', val=0.0 * np.ones(nn), units='deg', training_data=eta_bp)
+
+        self.add_output(name='CL', val=np.zeros(nn), units=None, training_data=CL_data)
+        self.add_output(name='CD', val=np.zeros(nn), units=None, training_data=CD_data + 0.015)
+        self.add_output(name='CM', val=np.zeros(nn), units=None, training_data=CM_data)

dymos/examples/aircraft_steady_flight/aero/aero_forces_comp.py

@@ -0,0 +1,51 @@
+import numpy as np
+
+import openmdao.api as om
+
+
+class AeroForcesComp(om.ExplicitComponent):
+
+    def initialize(self):
+        self.options.declare('num_nodes', types=int)
+
+    def setup(self):
+        nn = self.options['num_nodes']
+
+        self.add_input('q', val=np.ones(nn), desc='dynamic pressure', units='Pa')
+        self.add_input('S', val=np.ones(nn), desc='aerodynamic reference area', units='m**2')
+        self.add_input('CL', val=np.ones(nn), desc='lift coefficient', units=None)
+        self.add_input('CD', val=np.ones(nn), desc='drag coefficient', units=None)
+
+        self.add_output('L', val=np.ones(nn), desc='lift', units='N')
+        self.add_output('D', val=np.ones(nn), desc='drag', units='N')
+
+        ar = np.arange(nn)
+
+        self.declare_partials('L', 'q', rows=ar, cols=ar)
+        self.declare_partials('L', 'S', rows=ar, cols=ar)
+        self.declare_partials('L', 'CL', rows=ar, cols=ar)
+
+        self.declare_partials('D', 'q', rows=ar, cols=ar)
+        self.declare_partials('D', 'S', rows=ar, cols=ar)
+        self.declare_partials('D', 'CD', rows=ar, cols=ar)
+
+    def compute(self, inputs, outputs):
+        qS = inputs['q'] * inputs['S']
+
+        outputs['L'] = qS * inputs['CL']
+        outputs['D'] = qS * inputs['CD']
+
+    def compute_partials(self, inputs, partials):
+        q = inputs['q']
+        S = inputs['S']
+        CL = inputs['CL']
+        CD = inputs['CD']
+        qS = inputs['q'] * inputs['S']
+
+        partials['L', 'q'] = S * CL
+        partials['L', 'S'] = q * CL
+        partials['L', 'CL'] = qS
+
+        partials['D', 'q'] = S * CD
+        partials['D', 'S'] = q * CD
+        partials['D', 'CD'] = qS

dymos/examples/aircraft_steady_flight/aero/aerodynamics_group.py

@@ -0,0 +1,29 @@
+import openmdao.api as om
+from .aero_forces_comp import AeroForcesComp
+from .interpND_aero_coef_comp import InterpNDAeroCoeffComp, setup_surrogates_all
+
+
+class AerodynamicsGroup(om.Group):
+    """
+    The purpose of the Aerodynamics is to compute the lift and
+    drag forces on the aircraft.
+    """
+    def initialize(self):
+        self.options.declare('num_nodes', types=int)
+
+    def setup(self):
+        n = self.options['num_nodes']
+
+        interpND_CL, interpND_CD, interpND_CM, interp_num = setup_surrogates_all()
+
+        self.add_subsystem(name='aero_coef_comp',
+                           subsys=InterpNDAeroCoeffComp(vec_size=n, interpND_CL=interpND_CL, interpND_CD=interpND_CD,
+                                                        interpND_CM=interpND_CM,
+                                                        interp_num=interp_num),
+                           promotes_inputs=[('M', 'mach'), 'alpha', ('h', 'alt'), 'eta'],
+                           promotes_outputs=['CL', 'CD', 'CM'])
+
+        self.add_subsystem(name='aero_forces_comp',
+                           subsys=AeroForcesComp(num_nodes=n),
+                           promotes_inputs=['q', 'S', 'CL', 'CD'],
+                           promotes_outputs=['L', 'D'])