X-Axis Rotation Matrix

This example generates a rotation matrix that rotates a vector about the x-axis by a specified angle.

from openmdao.api import Problem
from omtools.api import Group
import omtools.api as ot
import numpy as np


class ExampleScalarRotX(Group):
    def setup(self):
        angle_val3 = np.pi / 3

        angle_scalar = self.declare_input('scalar', val=angle_val3)

        # Rotation in the x-axis for scalar
        self.register_output('scalar_Rot_x', ot.rotmat(angle_scalar, axis='x'))


prob = Problem()
prob.model = ExampleScalarRotX()
prob.setup(force_alloc_complex=True)
prob.run_model()

print('scalar', prob['scalar'].shape)
print(prob['scalar'])
print('scalar_Rot_x', prob['scalar_Rot_x'].shape)
print(prob['scalar_Rot_x'])

scalar (1,)
[1.04719755]
scalar_Rot_x (3, 3)
[[ 1.         0.         0.       ]
 [ 0.         0.5       -0.8660254]
 [ 0.         0.8660254  0.5      ]]