|
353 | 353 | "source": [ |
354 | 354 | "import numpy as np\n", |
355 | 355 | "\n", |
| 356 | + "\n", |
356 | 357 | "def controller_function(\n", |
357 | 358 | " time, sampling_rate, state, state_history, observed_variables, air_brakes\n", |
358 | | - "): \n", |
359 | | - " new_deployment_level =0\n", |
360 | | - " \n", |
361 | | - " if time<=11.4:\n", |
362 | | - " new_deployment_level=1\n", |
363 | | - " else: \n", |
364 | | - " new_deployment_level=-0.002906*np.power(time,3) + 0.1497*np.power(time,2) + -2.563*time + 14.96\n", |
365 | | - " \n", |
366 | | - " if time>19.6:\n", |
367 | | - " new_deployment_level=0\n", |
368 | | - " \n", |
369 | | - " if time<3.8: \n", |
370 | | - " new_deployment_level=0\n", |
| 359 | + "):\n", |
| 360 | + " new_deployment_level = 0\n", |
| 361 | + "\n", |
| 362 | + " if time <= 11.4:\n", |
| 363 | + " new_deployment_level = 1\n", |
| 364 | + " else:\n", |
| 365 | + " new_deployment_level = (\n", |
| 366 | + " -0.002906 * np.power(time, 3)\n", |
| 367 | + " + 0.1497 * np.power(time, 2)\n", |
| 368 | + " + -2.563 * time\n", |
| 369 | + " + 14.96\n", |
| 370 | + " )\n", |
| 371 | + "\n", |
| 372 | + " if time > 19.6:\n", |
| 373 | + " new_deployment_level = 0\n", |
| 374 | + "\n", |
| 375 | + " if time < 3.8:\n", |
| 376 | + " new_deployment_level = 0\n", |
371 | 377 | "\n", |
372 | 378 | " air_brakes.deployment_level = new_deployment_level\n", |
373 | 379 | "\n", |
|
519 | 525 | " sampling_rate=10,\n", |
520 | 526 | " reference_area=None,\n", |
521 | 527 | " clamp=True,\n", |
522 | | - " initial_observed_variables=(0,0),\n", |
| 528 | + " initial_observed_variables=(0, 0),\n", |
523 | 529 | " override_rocket_drag=True,\n", |
524 | 530 | " name=\"Air Brakes\",\n", |
525 | 531 | ")" |
|
635 | 641 | "source": [ |
636 | 642 | "import matplotlib.pyplot as plt\n", |
637 | 643 | "\n", |
638 | | - "time_list, deployment_level_list= [], []\n", |
| 644 | + "time_list, deployment_level_list = [], []\n", |
639 | 645 | "\n", |
640 | 646 | "obs_vars = test_flight.get_controller_observed_variables()\n", |
641 | 647 | "\n", |
|
0 commit comments