Skip to content

Commit e6e1843

Browse files
authored
Add instance variables to physics node (#403)
1 parent f729327 commit e6e1843

File tree

1 file changed

+13
-0
lines changed

1 file changed

+13
-0
lines changed

src/boat_simulator/boat_simulator/nodes/physics_engine/physics_engine_node.py

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,7 @@
55
import sys
66
from typing import Optional
77

8+
import numpy as np
89
import rclpy
910
import rclpy.utilities
1011
from custom_interfaces.action import SimRudderActuation, SimSailTrimTabActuation
@@ -30,7 +31,10 @@
3031
from rclpy.subscription import Subscription
3132

3233
import boat_simulator.common.constants as Constants
34+
from boat_simulator.common.generators import MVGaussianGenerator
3335
from boat_simulator.common.types import Scalar
36+
from boat_simulator.nodes.physics_engine.fluid_generation import FluidGenerator
37+
from boat_simulator.nodes.physics_engine.model import BoatState
3438

3539
from .decorators import require_all_subs_active
3640

@@ -106,6 +110,15 @@ def __init_private_attributes(self):
106110
self.__rudder_angle = 0
107111
self.__sail_trim_tab_angle = 0
108112
self.__desired_heading = None
113+
self.__boat_state = BoatState(
114+
0.5, 1, np.array([[0.5, 0.5, 0.5], [0.0, 0.5, 0.5], [0.0, 0.0, 0.5]], dtype=np.float32)
115+
)
116+
self.__wind_generator = FluidGenerator(
117+
generator=MVGaussianGenerator(np.array([5, 5]), np.array([[2, 1], [1, 2]]))
118+
)
119+
self.__current_generator = FluidGenerator(
120+
generator=MVGaussianGenerator(np.array([1, 1]), np.array([[2, 1], [1, 2]]))
121+
)
109122

110123
def __declare_ros_parameters(self):
111124
"""Declares ROS parameters from the global configuration file that will be used in this

0 commit comments

Comments
 (0)