-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
0 parents
commit e0eb3cb
Showing
47 changed files
with
3,720 additions
and
0 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,200 @@ | ||
# Created by https://www.toptal.com/developers/gitignore/api/visualstudiocode,python | ||
# Edit at https://www.toptal.com/developers/gitignore?templates=visualstudiocode,python | ||
|
||
Controller/Data/* | ||
*.csv | ||
*.gz | ||
.vscode | ||
|
||
### Python ### | ||
# Byte-compiled / optimized / DLL files | ||
__pycache__/ | ||
*.py[cod] | ||
*$py.class | ||
|
||
# C extensions | ||
*.so | ||
|
||
# Distribution / packaging | ||
.Python | ||
build/ | ||
develop-eggs/ | ||
dist/ | ||
downloads/ | ||
eggs/ | ||
.eggs/ | ||
lib/ | ||
lib64/ | ||
parts/ | ||
sdist/ | ||
var/ | ||
wheels/ | ||
share/python-wheels/ | ||
*.egg-info/ | ||
.installed.cfg | ||
*.egg | ||
MANIFEST | ||
|
||
# PyInstaller | ||
# Usually these files are written by a python script from a template | ||
# before PyInstaller builds the exe, so as to inject date/other infos into it. | ||
*.manifest | ||
*.spec | ||
|
||
# Installer logs | ||
pip-log.txt | ||
pip-delete-this-directory.txt | ||
|
||
# Unit test / coverage reports | ||
htmlcov/ | ||
.tox/ | ||
.nox/ | ||
.coverage | ||
.coverage.* | ||
.cache | ||
nosetests.xml | ||
coverage.xml | ||
*.cover | ||
*.py,cover | ||
.hypothesis/ | ||
.pytest_cache/ | ||
cover/ | ||
|
||
# Translations | ||
*.mo | ||
*.pot | ||
|
||
# Django stuff: | ||
*.log | ||
local_settings.py | ||
db.sqlite3 | ||
db.sqlite3-journal | ||
|
||
# Flask stuff: | ||
instance/ | ||
.webassets-cache | ||
|
||
# Scrapy stuff: | ||
.scrapy | ||
|
||
# Sphinx documentation | ||
docs/_build/ | ||
|
||
# PyBuilder | ||
.pybuilder/ | ||
target/ | ||
|
||
# Jupyter Notebook | ||
.ipynb_checkpoints | ||
|
||
# IPython | ||
profile_default/ | ||
ipython_config.py | ||
|
||
# pyenv | ||
# For a library or package, you might want to ignore these files since the code is | ||
# intended to run in multiple environments; otherwise, check them in: | ||
# .python-version | ||
|
||
# pipenv | ||
# According to pypa/pipenv#598, it is recommended to include Pipfile.lock in version control. | ||
# However, in case of collaboration, if having platform-specific dependencies or dependencies | ||
# having no cross-platform support, pipenv may install dependencies that don't work, or not | ||
# install all needed dependencies. | ||
#Pipfile.lock | ||
|
||
# poetry | ||
# Similar to Pipfile.lock, it is generally recommended to include poetry.lock in version control. | ||
# This is especially recommended for binary packages to ensure reproducibility, and is more | ||
# commonly ignored for libraries. | ||
# https://python-poetry.org/docs/basic-usage/#commit-your-poetrylock-file-to-version-control | ||
#poetry.lock | ||
|
||
# pdm | ||
# Similar to Pipfile.lock, it is generally recommended to include pdm.lock in version control. | ||
#pdm.lock | ||
# pdm stores project-wide configurations in .pdm.toml, but it is recommended to not include it | ||
# in version control. | ||
# https://pdm.fming.dev/#use-with-ide | ||
.pdm.toml | ||
|
||
# PEP 582; used by e.g. github.com/David-OConnor/pyflow and github.com/pdm-project/pdm | ||
__pypackages__/ | ||
|
||
# Celery stuff | ||
celerybeat-schedule | ||
celerybeat.pid | ||
|
||
# SageMath parsed files | ||
*.sage.py | ||
|
||
# Environments | ||
.env | ||
.venv | ||
env/ | ||
venv/ | ||
ENV/ | ||
env.bak/ | ||
venv.bak/ | ||
|
||
# Spyder project settings | ||
.spyderproject | ||
.spyproject | ||
|
||
# Rope project settings | ||
.ropeproject | ||
|
||
# mkdocs documentation | ||
/site | ||
|
||
# mypy | ||
.mypy_cache/ | ||
.dmypy.json | ||
dmypy.json | ||
|
||
# Pyre type checker | ||
.pyre/ | ||
|
||
# pytype static type analyzer | ||
.pytype/ | ||
|
||
# Cython debug symbols | ||
cython_debug/ | ||
|
||
# PyCharm | ||
# JetBrains specific template is maintained in a separate JetBrains.gitignore that can | ||
# be found at https://github.com/github/gitignore/blob/main/Global/JetBrains.gitignore | ||
# and can be added to the global gitignore or merged into this file. For a more nuclear | ||
# option (not recommended) you can uncomment the following to ignore the entire idea folder. | ||
#.idea/ | ||
|
||
### Python Patch ### | ||
# Poetry local configuration file - https://python-poetry.org/docs/configuration/#local-configuration | ||
poetry.toml | ||
|
||
# ruff | ||
.ruff_cache/ | ||
|
||
# LSP config files | ||
pyrightconfig.json | ||
|
||
### VisualStudioCode ### | ||
.vscode/* | ||
!.vscode/settings.json | ||
!.vscode/tasks.json | ||
!.vscode/launch.json | ||
!.vscode/extensions.json | ||
!.vscode/*.code-snippets | ||
|
||
# Local History for Visual Studio Code | ||
.history/ | ||
|
||
# Built Visual Studio Code Extensions | ||
*.vsix | ||
|
||
### VisualStudioCode Patch ### | ||
# Ignore all local history of files | ||
.history | ||
.ionide | ||
|
||
# End of https://www.toptal.com/developers/gitignore/api/visualstudiocode,python |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,97 @@ | ||
import time | ||
import logging | ||
|
||
from zerolib.message import MessageType, ActionType, SensorDataMessage, EngineProgramSettingsMessage | ||
|
||
from sensor_controller import SensorController | ||
|
||
logger = logging.getLogger(__name__) | ||
|
||
class TestBenchController: | ||
""" | ||
Handle the mainloop of the controller logic | ||
""" | ||
|
||
def __init__(self, peripheral_manager, dispatcher, sens_cfg, test_program): | ||
self.peripheral_manager = peripheral_manager | ||
self.sens_cfg = sens_cfg | ||
|
||
self.dispatcher = dispatcher | ||
self.test_program = test_program | ||
|
||
self.sb_rx = SensorController(sens_cfg, peripheral_manager) | ||
self.sb_rx.register_callback(self.data_handler) | ||
|
||
self.initialization_time = time.perf_counter() | ||
|
||
def data_handler(self, timestamp, data): | ||
""" | ||
data is an array of sensor, reading pairs. | ||
""" | ||
msg = SensorDataMessage(timestamp, data) | ||
self.dispatcher.dispatch(msg) | ||
|
||
def send_engine_program_list(self, status): | ||
if status is True: | ||
logger.info("Sending engine program list to monitor...") | ||
self.dispatcher.dispatch( | ||
EngineProgramSettingsMessage( | ||
False, ','.join( self.test_program.list_programs() ) | ||
) | ||
) | ||
|
||
def handler(self, msg): | ||
if msg.get_type() == MessageType.ENGINE_PROGRAM_SETTINGS: | ||
if msg.is_assigning is False: | ||
logger.error("EngineProgramSettings message is set as non-assigning!") | ||
return | ||
|
||
if msg.payload == "": | ||
logger.error("No program is specified!") | ||
return | ||
|
||
self.test_program.load(f"../Engine Test Programs/{msg.payload}.prog") | ||
return | ||
|
||
if msg.get_type() != MessageType.ACTION: | ||
logger.error("Non-action type message received. This should not happen.") | ||
return | ||
|
||
action = msg.action | ||
logger.info(f"Received action type {action}.") | ||
match action: | ||
### GENERAL ABORT SEQUENCE | ||
# Ensure that the propellant valves and fill valves are closed. | ||
# Then, open the vent valve. | ||
case ActionType.ABORT: | ||
self.test_program.abort() | ||
self.peripheral_manager.close_propellant_valves() | ||
self.peripheral_manager.fill_valve.close() | ||
self.peripheral_manager.vent_valve.open() | ||
|
||
### BURN PHASE ABORT | ||
# Similar to the general abort sequence, but we will make sure that | ||
# the ignitor is safed and the vent valve is not opened (in case a | ||
# recycle is possible) | ||
case ActionType.ABORT_BURN_PHASE: | ||
self.test_program.abort() | ||
self.peripheral_manager.ignitor.safe() | ||
self.peripheral_manager.close_propellant_valves() | ||
|
||
### INITIATE BURN PHASE | ||
# This is used to start the testing program (burn or cold flow) | ||
case ActionType.BEGIN_BURN_PHASE: | ||
self.test_program.run_program() | ||
|
||
### All other cases are simple and handled by the hw interface. | ||
case _: | ||
self.peripheral_manager.execute_action(action) | ||
|
||
def mainloop(self): | ||
self.sb_rx.start_collection() | ||
|
||
while True: | ||
time.sleep(1) | ||
|
||
def run(self): | ||
self.mainloop() |
Oops, something went wrong.