ENH: Discrete and Continuous Controllers

[Malmahrouqi3]

Pull request type

• Code changes (bugfix, features)

Checklist

• Tests for the changes have been added (if needed)
• Lint (black rocketpy/ tests/) has passed locally

Current behavior

#274

New behavior

Usage

airbrake_controller = _Controller(
    controller_function=controller_function,
    sampling_rate=10, # raplace with None or remove for continuous control
    interactive_objects=air_brakes,
    initial_observed_variables=[0, 0, 0],
    name="Discrete Air Brakes Controller (10 Hz)",
)
calisto._add_controllers(airbrake_controller)
calisto.air_brakes.append(air_brakes)

Tested on the Airbrakes Example (docs/user/airbrakes.rst)

(Discrete 10 Hz sampling)
195 Calls
Apogee State
Apogee Time: 23.360 s
Apogee Altitude: 4400.391 m (ASL) | 3000.391 m (AGL)
Apogee Freestream Speed: 16.149 m/s
Apogee X position: -4.510 m
Apogee Y position: 488.084 m
Apogee latitude: 32.9946429°
Apogee longitude: -106.9750464°

Maximum altitude: 3000.39 m

(Discrete 100 Hz sampling)
1947 Calls
Apogee State
Apogee Time: 23.362 s
Apogee Altitude: 4400.620 m (ASL) | 3000.620 m (AGL)
Apogee Freestream Speed: 16.151 m/s
Apogee X position: -4.511 m
Apogee Y position: 488.145 m
Apogee latitude: 32.9946434°
Apogee longitude: -106.9750464°

Maximum altitude: 3000.62 m

(Discrete 1000 Hz sampling)
19462 Calls
Apogee State
Apogee Time: 23.362 s
Apogee Altitude: 4400.594 m (ASL) | 3000.594 m (AGL)
Apogee Freestream Speed: 16.150 m/s
Apogee X position: -4.511 m
Apogee Y position: 488.136 m
Apogee latitude: 32.9946433°
Apogee longitude: -106.9750464°

Maximum altitude: 2903.45 m

(Continuous inf sampling)
410 Calls

Apogee State
Apogee Time: 23.369 s
Apogee Altitude: 4401.779 m (ASL) | 3001.779 m (AGL)
Apogee Freestream Speed: 16.157 m/s
Apogee X position: -4.104 m
Apogee Y position: 488.409 m
Apogee latitude: 32.9946458°
Apogee longitude: -106.9750420°

Maximum altitude: 3001.78 m

(No Controller)
0 Calls
Apogee State
Apogee Time: 25.886 s
Apogee Altitude: 4705.474 m (ASL) | 3305.474 m (AGL)
Apogee Freestream Speed: 20.662 m/s
Apogee X position: -5.543 m
Apogee Y position: 588.779 m
Apogee latitude: 32.9955483°
Apogee longitude: -106.9750574°

Maximum altitude: 3305.47 m

Breaking change

• Yes
• No

Additional information

There is no node creation for continuous controllers hence this step is skipped in flight.py. Instead, they are called in each of the ODE solver step, hence not relying on fixed sampling rate.

[codecov]

Codecov Report

❌ Patch coverage is 62.50000% with 3 lines in your changes missing coverage. Please review.
✅ Project coverage is 81.16%. Comparing base (9cf3dd4) to head (079385a).
⚠️ Report is 80 commits behind head on develop.

Files with missing lines	Patch %	Lines
rocketpy/simulation/flight.py	62.50%	3 Missing ⚠️

Additional details and impacted files

@@             Coverage Diff             @@
##           develop     #946      +/-   ##
===========================================
+ Coverage    80.27%   81.16%   +0.89%     
===========================================
  Files          104      113       +9     
  Lines        12769    14697    +1928     
===========================================
+ Hits         10250    11929    +1679     
- Misses        2519     2768     +249     

☔ View full report in Codecov by Harness.
📢 Have feedback on the report? Share it here.

🚀 New features to boost your workflow:

• ❄️ Test Analytics: Detect flaky tests, report on failures, and find test suite problems.

[Copilot]

Pull request overview

Adds first-class support for discrete (fixed refresh rate) and continuous (called every solver step) controllers to RocketPy’s flight simulation, aligning with the controller architecture discussed in issue #274.

Changes:

• Invoke continuous controllers during the solver stepping loop (instead of via fixed time nodes).
• Add Rocket.add_discrete_controller and Rocket.add_continuous_controller helpers that build _Controller objects with finite vs inf sampling rates.
• Update _Controller to default sampling_rate to math.inf and document the “infinite sampling” behavior.

Reviewed changes

Copilot reviewed 3 out of 3 changed files in this pull request and generated 8 comments.

File	Description
rocketpy/simulation/flight.py	Calls continuous controllers each solver step and skips time-node creation for sampling_rate=inf.
rocketpy/rocket/rocket.py	Adds Rocket-level APIs to register discrete vs continuous controllers.
rocketpy/control/controller.py	Defaults controller sampling rate to infinity and updates documentation accordingly.

Comments suppressed due to low confidence (1)

rocketpy/control/controller.py:81

• The added note in the docstring exceeds the typical line-length used elsewhere in the project (and likely will be reformatted by Ruff/Black). Please wrap this note to keep docstring formatting consistent/readable, and consider clarifying how sampling_rate=inf interacts with the existing 1/sampling_rate wording in the sampling_rate parameter description.

            .. note:: The function will be called according to the sampling rate
            specified. If unspecified, the default sampling rate is set to infinity, meaning that the
            controller function will be called at every step of the simulation.
        sampling_rate : float
            The sampling rate of the controller function in Hertz (Hz). This
            means that the controller function will be called every
            `1/sampling_rate` seconds.

[MateusStano]

Really clean! Just requested a couple of changes and it should be good to merge

[Malmahrouqi3]

Really clean! Just requested a couple of changes and it should be good to merge

Added the requested changes and ready to merge.

[Copilot]

Pull request overview

Copilot reviewed 3 out of 3 changed files in this pull request and generated 3 comments.

[Gui-FernandesBR]

Addressed the requested points and pushed follow-up commit 601acb0:\n\n- kept continuous controllers represented by sampling_rate=None\n- ensured continuous controllers do not create time nodes (covered by unit test)\n- updated controller docstring to document sampling_rate as optional/continuous semantics\n- optimized continuous-controller history handling in Flight to avoid rebuilding state history on every step while preserving the documented state-vector format\n\nLocal checks run:\n- pytest tests/unit/simulation/test_flight_time_nodes.py -q\n- pytest tests/unit/simulation/test_flight.py -q\n-
uff check rocketpy/ tests/ docs/\n\nIssue context: this PR targets #274 (Continuous and Discrete Control).\n\n@MateusStano could you please re-review?

[Gui-FernandesBR]

Reviewed the latest commit 601acb0: state-history handling is now efficient and consistent, continuous-controller node scheduling is covered by unit tests, and controller docs reflect optional sampling_rate behavior.

[Copilot]

Pull request overview

Copilot reviewed 4 out of 4 changed files in this pull request and generated 2 comments.

[Malmahrouqi3]

@Gui-FernandesBR
I took one of your recent commits and the behavior was odd. My last push was finalized with the intent to merge the PR.
If you wish to make any changes, please use the notebook to make sure everything is alright!
airbrakes3.ipynb

[Gui-FernandesBR]

Good catch on the new Copilot comments. Quick assessment:\n\n- They are valid as an optimization concern (we currently keep an extra history structure for continuous controllers).\n- They are not merge-blocking for this PR.\n- Replacing _controller_state_history with a direct alias to self.solution is not a drop-in change because state_history for controllers is currently expected to be state vectors only, while self.solution includes time as the first element ([t, *state]).\n\nGiven scope and risk, I suggest we keep this PR focused and open a follow-up refactor to unify history representation across discrete/continuous controllers without changing controller semantics unexpectedly.