Compare commits
88 Commits
2024-09-19
...
master
Author | SHA1 | Date |
---|---|---|
Alex Davies | 9298a57211 | |
Burak Ozter | 8c8e889571 | |
Burak Ozter | 3af11792a4 | |
Alex Davies | 41493f935c | |
Alex Davies | 271366bf7f | |
Alex Davies | 43fead1589 | |
Alex Davies | af170070ab | |
Alex Davies | f035a09321 | |
Burak Ozter | e15b036720 | |
Alex Davies | 980f0125f9 | |
Burak Ozter | 308c7087a2 | |
Alex Davies | d5c0b68ceb | |
Alex Davies | 5bebe95492 | |
Alex Davies | cb290395aa | |
Alex Davies | 10d6bd0c8b | |
Alex Davies | e670e38ba1 | |
Alex Davies | b5d96dc508 | |
Alex Davies | 2a9c5527f3 | |
Alex Davies | c80d37373b | |
Alex Davies | 0572d4ca1c | |
Burak Ozter | 1dd84706a5 | |
Alex Davies | f37841d7bb | |
Burak Ozter | df5df4b497 | |
Alex Davies | 98aafbb5e9 | |
Burak Ozter | c1484b58ce | |
Burak Ozter | a752ed603e | |
Alex Davies | f4f5b0890e | |
Alex Davies | 1a409a063a | |
Alex Davies | 4ae96ff8b9 | |
Alex Davies | b8a353b9af | |
Alex Davies | e42b5f1740 | |
Alex Davies | 2a252cb6a6 | |
Alex Davies | 5f5aa7f830 | |
Alex Davies | a272fdff0f | |
Alex Davies | eb8077f8eb | |
Alex Davies | ec7a53201f | |
Alex Davies | 15110bd830 | |
Alex Davies | f30740fe2d | |
Alex Davies | a8810dea8a | |
Alex Davies | bdd55885ad | |
Burak Ozter | 72878c06e8 | |
Burak Ozter | 2fa4e45818 | |
Burak Ozter | ed6097388d | |
Burak Ozter | c53205eadb | |
Burak Ozter | d661b1df33 | |
Burak Ozter | 3078feb2fb | |
Burak Ozter | dfdcb284f8 | |
Burak Ozter | de3aab205b | |
Burak Ozter | 0a0e7207c9 | |
Burak Ozter | 4ea8d7d1b9 | |
Burak Ozter | 5639f03d46 | |
Burak Ozter | 834a71b298 | |
Burak Ozter | 57a21d371b | |
Burak Ozter | af90430fb2 | |
Burak Ozter | f7f3bfe528 | |
Burak Ozter | a968663ec8 | |
Burak Ozter | fc52a3522f | |
Burak Ozter | 7acee12640 | |
Burak Ozter | 76810ecc46 | |
Burak Ozter | 56456f5950 | |
Alex Davies | 5ca555ffd4 | |
Alex Davies | 75f60fa811 | |
Alex Davies | 86460fa1d5 | |
Alex Davies | 8e45ae3440 | |
Alex Davies | 6ec9b219d7 | |
Alex Davies | 11bc1200c9 | |
Alex Davies | a5b7541452 | |
Alex Davies | 63520ed0c8 | |
Alex Davies | f0ead63a6d | |
Alex Davies | 9d7fdde370 | |
Alex Davies | 88f6c46213 | |
Burak Ozter | e56e3c06c0 | |
Burak Ozter | f34cfe98f2 | |
Alex Davies | 5c71841d12 | |
Alex Davies | aba8b6c1f0 | |
Alex Davies | 76b42682c9 | |
Alex Davies | 0cfd7cc78a | |
Alex Davies | 9fe800df0f | |
Alex Davies | 24430eb69a | |
Alex Davies | b61a389b92 | |
Alex Davies | 14c3358e81 | |
Alex Davies | 16f49d4a3f | |
Burak Ozter | 9cf8bfca88 | |
Burak Ozter | 73849a366b | |
Alex Davies | 8346fb42b5 | |
Alex Davies | b99f617166 | |
Alex Davies | 69589259e1 | |
Alex Davies | f3acf21c6a |
|
@ -0,0 +1,5 @@
|
|||
# Changes here will be overwritten by Copier; NEVER EDIT MANUALLY
|
||||
_commit: v1.0.2
|
||||
_src_path: https://git.spirirobotics.com/Spiri/template-docs.git
|
||||
author_name: Spiri Robotics
|
||||
project_name: spiri-sdk
|
|
@ -0,0 +1,19 @@
|
|||
DRONE_SYS_ID=1
|
||||
INSTANCE=0
|
||||
SERIAL0_PORT=5760
|
||||
SITL_PORT=5501
|
||||
MAVROS2_PORT=14560
|
||||
MAVROS1_PORT=14561
|
||||
FDM_PORT_IN=9002
|
||||
GSTREAMER_UDP_PORT=5600
|
||||
ROS_MASTER_URI="http://0.0.0.0:11311"
|
||||
|
||||
|
||||
ARDUPILOT_VEHICLE="-v copter -f gazebo-mu --model=JSON -L CitadelHill"
|
||||
WORLD_FILE_NAME="citadel_hill_world.sdf"
|
||||
WORLD_NAME="citadel_hill"
|
||||
DRONE_MODEL="spiri_mu"
|
||||
|
||||
|
||||
SIM_DRONE_COUNT=1
|
||||
GCS_PORT=14550
|
|
@ -0,0 +1,52 @@
|
|||
name: Build Docs
|
||||
|
||||
on:
|
||||
push:
|
||||
|
||||
env:
|
||||
REGISTRY: git.spirirobotics.com
|
||||
IMAGE_NAME: ${{ github.repository }}
|
||||
|
||||
jobs:
|
||||
build:
|
||||
runs-on: ubuntu-latest
|
||||
container: sphinxdoc/sphinx-latexpdf
|
||||
|
||||
steps:
|
||||
- name: Install sphinx-rtd-theme
|
||||
run: pip install sphinx-rtd-theme myst-parser
|
||||
- name: Install node so that custom actions work.
|
||||
run : apt-get update && apt-get --yes install nodejs git
|
||||
|
||||
- name: Checkout Repository
|
||||
uses: actions/checkout@v4
|
||||
|
||||
- name: Build Docs
|
||||
run: make html latexpdf
|
||||
working-directory: docs # assuming your documentation is in a 'docs' folder
|
||||
|
||||
- name: Save PDF Artifacts
|
||||
run: mv docs/build/latex/*.pdf ${{ github.workspace }}/docs.pdf
|
||||
|
||||
- name: Compress HTML
|
||||
run: tar -czvf docs_html.tar.gz -C docs/build/html .
|
||||
|
||||
- name: Upload Docs
|
||||
uses: actions/upload-artifact@v3
|
||||
with:
|
||||
name: docs_html.tar.gz
|
||||
path: docs_html.tar.gz
|
||||
|
||||
- name: Upload PDF
|
||||
uses: actions/upload-artifact@v3
|
||||
with:
|
||||
name: docs.pdf
|
||||
path: docs.pdf
|
||||
|
||||
- name: Deploy
|
||||
uses: s0/git-publish-subdir-action@develop
|
||||
env:
|
||||
REPO: http://git-spirirobotics-com-app:3000/Spiri/spiri-sdk.git/
|
||||
BRANCH: gh-pages
|
||||
FOLDER: ./docs/build/html
|
||||
GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
|
|
@ -1,2 +1,3 @@
|
|||
docs/build/
|
||||
.vscode
|
||||
.vscode
|
||||
*.pyc
|
||||
|
|
238
README.md
238
README.md
|
@ -1,10 +1,10 @@
|
|||
# Spiri SDK - Simulated robot
|
||||
|
||||
The Spiri SDK consists of a number of components. What you're looking at right now
|
||||
The Spiri SDK consists of a number of components. What you're looking at right now
|
||||
is the drone simulation component, which is the core of the SDK.
|
||||
|
||||
Spiri Robots run a number of docker containers to achieve their core functionality,
|
||||
we try to keep these essential docker containers in one docker compose file. The
|
||||
we try to keep these essential docker containers in one docker compose file. The
|
||||
docker compose file you'll find in this repository starts an ardupilot-based UAV simulation
|
||||
as well as a ROS master, and mavproxy to tie it together.
|
||||
|
||||
|
@ -13,6 +13,8 @@ To get started you can simply clone this repository and run `docker compose --pr
|
|||
Once the simulated UAV is running you can connect to it with QGroundControl or other
|
||||
MavLink compatible software. We expose the UAVs Mavlink conenction on tcp port 5760.
|
||||
|
||||
There is experimental GUI support you can enable by running `docker compose --profile uav-sim --profile ui up`.
|
||||
|
||||
## Creating a new project
|
||||
|
||||
We provide project templates you can use for development that integrate seamlessly into
|
||||
|
@ -23,13 +25,239 @@ These templates are intended to be used with VSCode.
|
|||
To get started with our project templates install the [copier](https://copier.readthedocs.io/en/stable/) project
|
||||
templating utility.
|
||||
|
||||
* [template-service-ros1-catkin](https://git.spirirobotics.com/Spiri/template-service-ros1-catkin)
|
||||
- [template-service-ros1-catkin](https://git.spirirobotics.com/Spiri/template-service-ros1-catkin)
|
||||
|
||||
This template uses the last stable release of ROS1 (ros noetic) and supports python and c++ programming
|
||||
This template uses the last stable release of ROS1 (ros noetic) and supports python and c++ programming
|
||||
languages.
|
||||
|
||||
ROS1 is considered end of life. It's recomended to use a ROS2 template instead
|
||||
|
||||
* ROS2 template
|
||||
- ROS2 template
|
||||
|
||||
We're working on it...
|
||||
|
||||
## NVIDIA Container Toolkit
|
||||
|
||||
[Source](https://docs.nvidia.com/datacenter/cloud-native/container-toolkit/latest/install-guide.html)
|
||||
|
||||
### Installing with Apt
|
||||
|
||||
```bash
|
||||
curl -fsSL https://nvidia.github.io/libnvidia-container/gpgkey | sudo gpg --dearmor -o /usr/share/keyrings/nvidia-container-toolkit-keyring.gpg \
|
||||
&& curl -s -L https://nvidia.github.io/libnvidia-container/stable/deb/nvidia-container-toolkit.list | \
|
||||
sed 's#deb https://#deb [signed-by=/usr/share/keyrings/nvidia-container-toolkit-keyring.gpg] https://#g' | \
|
||||
sudo tee /etc/apt/sources.list.d/nvidia-container-toolkit.list
|
||||
```
|
||||
|
||||
```bash
|
||||
sudo apt-get update
|
||||
```
|
||||
|
||||
```bash
|
||||
sudo apt-get install -y nvidia-container-toolkit
|
||||
```
|
||||
|
||||
#### Configuration
|
||||
|
||||
##### Prerequisites
|
||||
|
||||
- You installed a supported container engine (Docker, Containerd, CRI-O, Podman).
|
||||
- You installed the NVIDIA Container Toolkit.
|
||||
|
||||
```bash
|
||||
sudo nvidia-ctk runtime configure --runtime=docker --cdi.enabled
|
||||
```
|
||||
|
||||
```bash
|
||||
sudo systemctl restart docker
|
||||
```
|
||||
|
||||
### Sample Workload
|
||||
|
||||
After you install and configure the toolkit and install an NVIDIA GPU Driver, you can verify your installation by running a sample workload.
|
||||
|
||||
```bash
|
||||
sudo docker run --rm --runtime=nvidia --gpus all ubuntu nvidia-smi
|
||||
```
|
||||
|
||||
Expected output:
|
||||
|
||||
```console
|
||||
+-----------------------------------------------------------------------------+
|
||||
| NVIDIA-SMI 535.86.10 Driver Version: 535.86.10 CUDA Version: 12.2 |
|
||||
|-------------------------------+----------------------+----------------------+
|
||||
| GPU Name Persistence-M| Bus-Id Disp.A | Volatile Uncorr. ECC |
|
||||
| Fan Temp Perf Pwr:Usage/Cap| Memory-Usage | GPU-Util Compute M. |
|
||||
| | | MIG M. |
|
||||
|===============================+======================+======================|
|
||||
| 0 Tesla T4 On | 00000000:00:1E.0 Off | 0 |
|
||||
| N/A 34C P8 9W / 70W | 0MiB / 15109MiB | 0% Default |
|
||||
| | | N/A |
|
||||
+-------------------------------+----------------------+----------------------+
|
||||
|
||||
+-----------------------------------------------------------------------------+
|
||||
| Processes: |
|
||||
| GPU GI CI PID Type Process name GPU Memory |
|
||||
| ID ID Usage |
|
||||
|=============================================================================|
|
||||
| No running processes found |
|
||||
+-----------------------------------------------------------------------------+
|
||||
```
|
||||
|
||||
## Support for Container Device Interface(CDI)
|
||||
|
||||
[Source](https://docs.nvidia.com/datacenter/cloud-native/container-toolkit/latest/cdi-support.html)
|
||||
|
||||
### Prerequisites
|
||||
|
||||
- You installed either the NVIDIA Container Toolkit or you installed the `nvidia-container-toolkit-base` package.
|
||||
The base package includes the container runtime and the `nvidia-ctk` command-line interface, but avoids installing the container runtime hook and transitive dependencies.
|
||||
The hook and dependencies are not needed on machines that use CDI exclusively
|
||||
- You installed an NVIDIA GPU Driver.
|
||||
|
||||
Two common locations for CDI specifications are `/etc/cdi/` and `/var/run/cdi/`. The contents of the `/var/run/cdi/` directory are cleared on boot.
|
||||
|
||||
1. Generate the CDI specification file:
|
||||
|
||||
```bash
|
||||
sudo nvidia-ctk cdi generate --output=/etc/cdi/nvidia.yaml
|
||||
```
|
||||
|
||||
Example output
|
||||
|
||||
```console
|
||||
INFO[0000] Auto-detected mode as "nvml"
|
||||
INFO[0000] Selecting /dev/nvidia0 as /dev/nvidia0
|
||||
INFO[0000] Selecting /dev/dri/card1 as /dev/dri/card1
|
||||
INFO[0000] Selecting /dev/dri/renderD128 as /dev/dri/renderD128
|
||||
INFO[0000] Using driver version xxx.xxx.xx
|
||||
...
|
||||
```
|
||||
|
||||
2. (Optional) Check the names of the generated devices:
|
||||
|
||||
```bash
|
||||
nvidia-ctk cdi list
|
||||
```
|
||||
|
||||
Output
|
||||
|
||||
```console
|
||||
INFO[0000] Found 9 CDI devices
|
||||
nvidia.com/gpu=all
|
||||
nvidia.com/gpu=0
|
||||
```
|
||||
|
||||
### Sample Workload
|
||||
|
||||
```bash
|
||||
docker run --rm -ti --runtime=nvidia \
|
||||
-e NVIDIA_VISIBLE_DEVICES=nvidia.com/gpu=all \
|
||||
ubuntu nvidia-smi -L
|
||||
```
|
||||
|
||||
Output
|
||||
|
||||
```console
|
||||
GPU 0: NVIDIA GeForce RTX 3080 Laptop GPU (UUID: GPU-17c2b9a6-6be2-3857-f8e0-88143e2e621b)
|
||||
```
|
||||
|
||||
## Technologies
|
||||
|
||||
* Ubuntu 22.04.1 amd64
|
||||
* Docker Compose version v2.29.7
|
||||
* Python 3.10.12
|
||||
* pip 22.0.2
|
||||
* NVIDIA Container Toolkit CLI version 1.16.2
|
||||
* NVIDIA Driver 550.120
|
||||
* CUDA Version 12.4
|
||||
|
||||
## How to Run
|
||||
|
||||
Ensure variables in the `.env` file are correct.
|
||||
|
||||
Install the required libraries in the python script.
|
||||
|
||||
```bash
|
||||
pip install -r requirements.txt
|
||||
```
|
||||
|
||||
### First Terminal
|
||||
|
||||
1. Start the user interface with the following command.
|
||||
|
||||
```bash
|
||||
docker compose --profile ui up
|
||||
```
|
||||
|
||||
2. Click `Launch Gazebo` on the menu.
|
||||
|
||||
Simulated world and the drone model should be up and running in a Gazebo instance.
|
||||
|
||||
### Second Terminal
|
||||
|
||||
This will launch ardupilot, mavproxy and mavros services, and will scale up by the `SIM_DRONE_COUNT` env variable.
|
||||
|
||||
1. Start the docker services with the following python script.
|
||||
|
||||
```bash
|
||||
python3 sim_drone.py
|
||||
```
|
||||
|
||||
### QGroundControl
|
||||
|
||||
Simulated vehicle(s) should be connected to QGroundControl. `SIM_DRONE_COUNT` value should match the detected vehicle count on GCS.
|
||||
|
||||
## Simulation Environment Variables
|
||||
|
||||
`DRONE_SYS_ID` and `INSTANCE` environment variables are incremented by 1 for each additional simulated vehicle. `SERIAL0_PORT`, `SITL_PORT`, `MAVROS2_PORT`, `MAVROS1_PORT`,`FDM_PORT_IN` and `GSTREAMER_UDP_PORT` are incremented by 10. Without in-depth knowledge, changing the default value for these ports are not recommended.
|
||||
|
||||
| Variable | Type | Default | Description |
|
||||
| :----------------: | :----: | :--------------------------------------------------: | :---------------------------------------------------------------------------------------------------------------- |
|
||||
| DRONE_SYS_ID | int | 1 | System-ID for the simulated drone. |
|
||||
| INSTANCE | int | 0 | Instance of simulator. |
|
||||
| SERIAL0_PORT | int | 5760 | Mavproxy master port the simulation communicating on. |
|
||||
| SITL_PORT | int | 5501 | Mavproxy Software in the Loop(SITL) port to send simulated RC input for the simulator. |
|
||||
| MAVROS2_PORT | int | 14560 | MAVROS ROS 2 UDP port |
|
||||
| MAVROS1_PORT | int | 14561 | MAVROS ROS 1 UDP port |
|
||||
| FDM_PORT_IN | int | 9002 | Gazebo Flight Dynamics Model (FDM) UDP port |
|
||||
| GSTREAMER_UDP_PORT | int | 5600 | UDP Video Streaming port |
|
||||
| ROS_MASTER_URI | string | "http://0.0.0.0:11311" | This tells ROS 1 nodes where they can locate the master |
|
||||
| ARDUPILOT_VEHICLE | string | "-v copter -f gazebo-mu --model=JSON -L CitadelHill" | "-v" is vehicle type,"-L" start location, "-f" is vehicle frame type, "--model" overrides simulation model to use |
|
||||
| WORLD_FILE_NAME | string | "citadel_hill_world.sdf" | Name of the file that exists in the worlds folder. |
|
||||
| WORLD_NAME | string | "citadel_hill" | Name of the world defined in the world file. |
|
||||
| DRONE_MODEL | string | "spiri_mu" | Drone model that exists in the models folder. |
|
||||
| SIM_DRONE_COUNT | int | 1 | Number of drones to be simulated. |
|
||||
| GCS_PORT | int | 14550 | Ground Control Station(GCS) UDP connection port. |
|
||||
|
||||
### Understanding Multi-Vehicle Simulation Parameters
|
||||
|
||||
For instance, if `SIM_DRONE_COUNT` is 2, each additional vehicle's ports are incremented by 10.
|
||||
|
||||
First simulated vehicle would have these following values,
|
||||
|
||||
| Variable | Value |
|
||||
| :----------------: | :---: |
|
||||
| DRONE_SYS_ID | 1 |
|
||||
| INSTANCE | 0 |
|
||||
| SERIAL0_PORT | 5760 |
|
||||
| SITL_PORT | 5501 |
|
||||
| MAVROS2_PORT | 14560 |
|
||||
| MAVROS1_PORT | 14561 |
|
||||
| FDM_PORT_IN | 9002 |
|
||||
| GSTREAMER_UDP_PORT | 5600 |
|
||||
|
||||
Second Vehicle,
|
||||
|
||||
| Variable | Value |
|
||||
| :----------------: | :---: |
|
||||
| DRONE_SYS_ID | 2 |
|
||||
| INSTANCE | 1 |
|
||||
| SERIAL0_PORT | 5770 |
|
||||
| SITL_PORT | 5511 |
|
||||
| MAVROS2_PORT | 14570 |
|
||||
| MAVROS1_PORT | 14571 |
|
||||
| FDM_PORT_IN | 9012 |
|
||||
| GSTREAMER_UDP_PORT | 5610 |
|
||||
|
||||
Video stream would be available on UDP port on 5610 for the second vehicle after enabling streaming.
|
||||
|
|
|
@ -1,64 +1,52 @@
|
|||
version: '3.8'
|
||||
version: "3.8"
|
||||
|
||||
services:
|
||||
ardupilot:
|
||||
image: git.spirirobotics.com/spiri/ardupilot:spiri-master
|
||||
command: >
|
||||
./Tools/autotest/sim_vehicle.py -v copter --no-rebuild
|
||||
--out=udpin:0.0.0.0:5000
|
||||
--enable-dds
|
||||
stdin_open: true
|
||||
tty: true
|
||||
gui-tools:
|
||||
env_file:
|
||||
- .env
|
||||
build:
|
||||
context: ./guiTools/
|
||||
|
||||
mavproxy:
|
||||
image: git.spirirobotics.com/spiri/services-mavproxy:main
|
||||
command: >
|
||||
mavproxy.py --non-interactive
|
||||
--out=tcpin:0.0.0.0:5760
|
||||
--master=udpout:ardupilot:5000
|
||||
restart: always
|
||||
ports:
|
||||
- 5760:5760
|
||||
|
||||
mavros:
|
||||
#This service bridges our mavlink-based robot-coprosessor into ROS
|
||||
#In this example it connects to a simulated coprocessor.
|
||||
image: git.spirirobotics.com/spiri/services-ros1-mavros:master
|
||||
command: roslaunch mavros px4.launch fcu_url:="udp://:14555@mavproxy:14550" tgt_system:="1"
|
||||
environment:
|
||||
- "ROS_MASTER_URI=http://ros-master:11311"
|
||||
depends_on:
|
||||
ros-master:
|
||||
condition: service_healthy
|
||||
mavproxy:
|
||||
condition: service_started
|
||||
restart: always
|
||||
# Display settings
|
||||
- DISPLAY=${DISPLAY}
|
||||
- WAYLAND_DISPLAY=${WAYLAND_DISPLAY}
|
||||
# - QT_QPA_PLATFORM=${QT_QPA_PLATFORM:-xcb} # Default to X11
|
||||
# If running with Wayland
|
||||
- XDG_RUNTIME_DIR=${XDG_RUNTIME_DIR}
|
||||
- ROS_MASTER_URI=http://ros-master:11311
|
||||
# - NVIDIA_DRIVER_CAPABILITIES=compute,video,utility
|
||||
# - NVIDIA_VISIBLE_DEVICES=all
|
||||
volumes:
|
||||
# X11 socket
|
||||
- /tmp/.X11-unix:/tmp/.X11-unix
|
||||
- ${XAUTHORITY:-~/.Xauthority}:/root/.Xauthority
|
||||
# Wayland socket
|
||||
#- ${XDG_RUNTIME_DIR}/wayland-0:${XDG_RUNTIME_DIR}/wayland-0
|
||||
# Allow access to the host's GPU
|
||||
- /dev/dri:/dev/dri
|
||||
#Auto reload on code changes
|
||||
- ./guiTools/spiri_sdk_guitools/:/app/spiri_sdk_guitools/
|
||||
# Enable launching the SDK from the SDK
|
||||
# - ./:/app/sdk
|
||||
- ./robots:/robots
|
||||
- /var/run/docker.sock:/var/run/docker.sock
|
||||
devices:
|
||||
# Provide access to GPU devices
|
||||
- /dev/dri:/dev/dri
|
||||
network_mode: host
|
||||
ports:
|
||||
- 8923:8923
|
||||
ipc: host
|
||||
#user: "${UID}:${GID}"
|
||||
privileged: true # Allow privileged access if necessary (e.g., for GPU access)
|
||||
# restart: unless-stopped
|
||||
# command: /bin/bash -c "source /opt/ros/foxy/setup.bash && rvis2" # Replace with the actual command to run Gazebo Ignition
|
||||
deploy:
|
||||
resources:
|
||||
limits:
|
||||
# cpus: '0.01'
|
||||
memory: 200M
|
||||
ulimits:
|
||||
nofile:
|
||||
soft: 1024
|
||||
hard: 524288
|
||||
|
||||
ros-master:
|
||||
image: git.spirirobotics.com/spiri/services-ros1-core:main
|
||||
command: stdbuf -o L roscore
|
||||
environment:
|
||||
- "ROS_MASTER_URI=http://localhost:11311"
|
||||
restart: always
|
||||
ports:
|
||||
- "127.0.0.1:11311:11311"
|
||||
deploy:
|
||||
resources:
|
||||
limits:
|
||||
memory: 1G
|
||||
# Madness, setting a low ulimit here fixes memory leaks
|
||||
# https://answers.ros.org/question/336963/rosout-high-memory-usage/
|
||||
ulimits:
|
||||
nofile:
|
||||
soft: 1024
|
||||
hard: 524288
|
||||
reservations:
|
||||
devices:
|
||||
- driver: cdi
|
||||
device_ids:
|
||||
- nvidia.com/gpu=all
|
||||
|
||||
|
|
|
@ -0,0 +1,20 @@
|
|||
# Minimal makefile for Sphinx documentation
|
||||
#
|
||||
|
||||
# You can set these variables from the command line, and also
|
||||
# from the environment for the first two.
|
||||
SPHINXOPTS ?=
|
||||
SPHINXBUILD ?= sphinx-build
|
||||
SOURCEDIR = source
|
||||
BUILDDIR = build
|
||||
|
||||
# Put it first so that "make" without argument is like "make help".
|
||||
help:
|
||||
@$(SPHINXBUILD) -M help "$(SOURCEDIR)" "$(BUILDDIR)" $(SPHINXOPTS) $(O)
|
||||
|
||||
.PHONY: help Makefile
|
||||
|
||||
# Catch-all target: route all unknown targets to Sphinx using the new
|
||||
# "make mode" option. $(O) is meant as a shortcut for $(SPHINXOPTS).
|
||||
%: Makefile
|
||||
@$(SPHINXBUILD) -M $@ "$(SOURCEDIR)" "$(BUILDDIR)" $(SPHINXOPTS) $(O)
|
|
@ -0,0 +1,11 @@
|
|||
If you have a correctly configured sphinx environment you can build this project
|
||||
using `make html latexpdf`.
|
||||
|
||||
You can also use nektos/act to build this project in the same way our build does.
|
||||
```bash
|
||||
cd ../ #Make sure you're in the project root, you should have a hidden folder
|
||||
# named ./.github/workflows available.
|
||||
act --artifact-server-path ./doc-build
|
||||
```
|
||||
|
||||
Your compiled doc project will now be in the ./doc-build folder.
|
|
@ -0,0 +1,35 @@
|
|||
@ECHO OFF
|
||||
|
||||
pushd %~dp0
|
||||
|
||||
REM Command file for Sphinx documentation
|
||||
|
||||
if "%SPHINXBUILD%" == "" (
|
||||
set SPHINXBUILD=sphinx-build
|
||||
)
|
||||
set SOURCEDIR=source
|
||||
set BUILDDIR=build
|
||||
|
||||
%SPHINXBUILD% >NUL 2>NUL
|
||||
if errorlevel 9009 (
|
||||
echo.
|
||||
echo.The 'sphinx-build' command was not found. Make sure you have Sphinx
|
||||
echo.installed, then set the SPHINXBUILD environment variable to point
|
||||
echo.to the full path of the 'sphinx-build' executable. Alternatively you
|
||||
echo.may add the Sphinx directory to PATH.
|
||||
echo.
|
||||
echo.If you don't have Sphinx installed, grab it from
|
||||
echo.https://www.sphinx-doc.org/
|
||||
exit /b 1
|
||||
)
|
||||
|
||||
if "%1" == "" goto help
|
||||
|
||||
%SPHINXBUILD% -M %1 %SOURCEDIR% %BUILDDIR% %SPHINXOPTS% %O%
|
||||
goto end
|
||||
|
||||
:help
|
||||
%SPHINXBUILD% -M help %SOURCEDIR% %BUILDDIR% %SPHINXOPTS% %O%
|
||||
|
||||
:end
|
||||
popd
|
|
@ -0,0 +1,34 @@
|
|||
.wy-side-nav-search {
|
||||
background: #FFFFFF !important;
|
||||
}
|
||||
|
||||
|
||||
.wy-nav-side {
|
||||
background-color: #FFFFFF !important;
|
||||
}
|
||||
|
||||
|
||||
/* Add borders and box-shadow */
|
||||
.wy-side-nav {
|
||||
border: 1px solid #899CA3 !important;
|
||||
}
|
||||
|
||||
.logo {
|
||||
width: 100px !important;
|
||||
}
|
||||
|
||||
.wy-menu-vertical a {
|
||||
color: #899CA3 !important; /* Change to your desired color */
|
||||
}
|
||||
|
||||
.document-title {
|
||||
color: #000 !important;
|
||||
font-size: 24px !important;
|
||||
text-transform: uppercase !important;
|
||||
}
|
||||
|
||||
.icon-home {
|
||||
font-weight: bold !important;
|
||||
text-transform: uppercase !important;
|
||||
color: #000 !important;
|
||||
}
|
|
@ -0,0 +1,70 @@
|
|||
# Configuration file for the Sphinx documentation builder.
|
||||
#
|
||||
# For the full list of built-in configuration values, see the documentation:
|
||||
# https://www.sphinx-doc.org/en/master/usage/configuration.html
|
||||
|
||||
# -- Project information -----------------------------------------------------
|
||||
# https://www.sphinx-doc.org/en/master/usage/configuration.html#project-information
|
||||
import sphinx_rtd_theme
|
||||
|
||||
project = "spiri-sdk"
|
||||
copyright = "2024, Spiri Robotics"
|
||||
author = "Spiri Robotics"
|
||||
|
||||
html_logo = "logos/SPIRI_STLockup_Mixed_RGB.png" # For HTML output
|
||||
html_logo_width = '200px'
|
||||
latex_logo = "logos/SPIRI_STLockup_Mixed_RGB.png"
|
||||
latex_logo_width = '5cm'
|
||||
|
||||
|
||||
# -- General configuration ---------------------------------------------------
|
||||
# https://www.sphinx-doc.org/en/master/usage/configuration.html#general-configuration
|
||||
|
||||
extensions = [
|
||||
"myst_parser",
|
||||
"sphinx.ext.duration",
|
||||
"sphinx.ext.doctest",
|
||||
"sphinx.ext.autodoc",
|
||||
"sphinx.ext.autosummary",
|
||||
"sphinx.ext.intersphinx",
|
||||
"sphinx.ext.todo",
|
||||
]
|
||||
|
||||
|
||||
numfig = True
|
||||
|
||||
todo_include_todos = True
|
||||
todo_emit_warnings = True
|
||||
todo_link_only = True
|
||||
|
||||
|
||||
templates_path = ["_templates"]
|
||||
exclude_patterns = []
|
||||
|
||||
# -- Options for HTML output -------------------------------------------------
|
||||
# https://www.sphinx-doc.org/en/master/usage/configuration.html#options-for-html-output
|
||||
|
||||
html_theme = "sphinx_rtd_theme"
|
||||
html_theme_path = [sphinx_rtd_theme.get_html_theme_path()]
|
||||
|
||||
html_static_path = ['_static']
|
||||
|
||||
html_css_files = [
|
||||
'custom.css',
|
||||
]
|
||||
|
||||
|
||||
html_theme_options = {
|
||||
'collapse_navigation': True,
|
||||
'sticky_navigation': True,
|
||||
'navigation_depth': 4, #could be set to -1 if we want unlimited depth
|
||||
'includehidden': True,
|
||||
'titles_only': False
|
||||
}
|
||||
|
||||
latex_engine = "xelatex"
|
||||
# Configure LaTeX options for PDF generation
|
||||
latex_show_urls = 'footnote'
|
||||
|
||||
# Enable syntax highlighting for PDFs
|
||||
pygments_style = 'sphinx'
|
|
@ -0,0 +1,21 @@
|
|||
.. spiri-sdk documentation master file, created by
|
||||
sphinx-quickstart on Wed Feb 14 11:51:45 2024.
|
||||
You can adapt this file completely to your liking, but it should at least
|
||||
contain the root `toctree` directive.
|
||||
|
||||
Welcome to spiri-sdk's documentation!
|
||||
============================================
|
||||
|
||||
.. toctree::
|
||||
:maxdepth: 2
|
||||
:caption: Contents:
|
||||
|
||||
.. include:: ../../README.md
|
||||
:parser: myst_parser.sphinx_
|
||||
|
||||
Indices and tables
|
||||
==================
|
||||
|
||||
* :ref:`genindex`
|
||||
* :ref:`modindex`
|
||||
* :ref:`search`
|
Binary file not shown.
After Width: | Height: | Size: 14 KiB |
|
@ -0,0 +1,46 @@
|
|||
FROM osrf/ros:jazzy-desktop-full
|
||||
|
||||
RUN apt-get update
|
||||
RUN apt-get -y install qterminal mesa-utils \
|
||||
libgstreamer1.0-dev \
|
||||
libgstreamer-plugins-base1.0-dev \
|
||||
docker-compose \
|
||||
python3.12-venv \
|
||||
python3-pip \
|
||||
gstreamer1.0-libav \
|
||||
gstreamer1.0-gl \
|
||||
gstreamer1.0-plugins-good \
|
||||
gstreamer1.0-plugins-bad \
|
||||
gstreamer1.0-plugins-ugly
|
||||
|
||||
|
||||
COPY --from=git.spirirobotics.com/spiri/gazebo-resources:main /plugins /ardupilot_gazebo/plugins
|
||||
COPY --from=git.spirirobotics.com/spiri/gazebo-resources:main /models /ardupilot_gazebo/models
|
||||
COPY --from=git.spirirobotics.com/spiri/gazebo-resources:main /worlds /ardupilot_gazebo/worlds
|
||||
|
||||
ENV GZ_SIM_SYSTEM_PLUGIN_PATH=/ardupilot_gazebo/plugins
|
||||
ENV GZ_SIM_RESOURCE_PATH=/ardupilot_gazebo/models:/ardupilot_gazebo/worlds
|
||||
|
||||
COPY ./gz_entrypoint.sh /gz_entrypoint.sh
|
||||
RUN chmod +x /gz_entrypoint.sh
|
||||
|
||||
WORKDIR /app
|
||||
|
||||
RUN python3 -m venv /opt/venv
|
||||
ENV PATH="/opt/venv/bin:$PATH"
|
||||
|
||||
RUN pip3 install poetry
|
||||
|
||||
COPY ./pyproject.toml ./poetry.lock ./README.md ./
|
||||
COPY ./spiri_sdk_guitools ./spiri_sdk_guitools
|
||||
|
||||
RUN poetry env use python3
|
||||
RUN poetry config virtualenvs.create false && poetry install --no-dev --no-interaction --no-ansi
|
||||
|
||||
|
||||
|
||||
CMD poetry run python3 spiri_sdk_guitools/launcher.py
|
||||
|
||||
|
||||
|
||||
|
|
@ -0,0 +1,6 @@
|
|||
#!/bin/bash
|
||||
set -e #PR #6
|
||||
source /opt/ros/$ROS_DISTRO/setup.bash
|
||||
gz sim -v -r $WORLD_FILE_NAME
|
||||
|
||||
exec "$@"
|
File diff suppressed because it is too large
Load Diff
|
@ -0,0 +1,21 @@
|
|||
[tool.poetry]
|
||||
name = "spiri_sdk_guitools"
|
||||
version = "0.1.0"
|
||||
description = ""
|
||||
authors = ["Spiri Robotics"]
|
||||
readme = "README.md"
|
||||
|
||||
[tool.poetry.dependencies]
|
||||
python = "^3.11"
|
||||
nicegui = "^2.5.0"
|
||||
pywebview = "^5.3.2"
|
||||
loguru = "^0.7.2"
|
||||
sh = "^2.1.0"
|
||||
docker = "^7.1.0"
|
||||
aiodocker = "^0.23.0"
|
||||
numpy = "^2.1.3"
|
||||
|
||||
[build-system]
|
||||
requires = ["poetry-core"]
|
||||
build-backend = "poetry.core.masonry.api"
|
||||
|
|
@ -0,0 +1,32 @@
|
|||
#!/bin/bash
|
||||
set -e #PR #6
|
||||
source /opt/ros/$ROS_DISTRO/setup.bash
|
||||
|
||||
if [[ -z $SIM_DRONE_COUNT ]]
|
||||
then
|
||||
echo "SIM_DRONE_COUNT environment variable is not set."
|
||||
exit 1
|
||||
fi
|
||||
|
||||
gz sim -v -r $WORLD_FILE_NAME &
|
||||
while true
|
||||
do
|
||||
topics=$(gz topic -l)
|
||||
if [[ $topics == *"/world/$WORLD_NAME"* ]]
|
||||
then
|
||||
break
|
||||
fi
|
||||
sleep 1
|
||||
done
|
||||
|
||||
cd /ardupilot_gazebo/models/$DRONE_MODEL
|
||||
for (( j=0; j<$SIM_DRONE_COUNT; j++ ));
|
||||
do
|
||||
xacro -v gstreamer_udp_port:=$(($GSTREAMER_UDP_PORT + ($j * 10))) fdm_port_in:=$(($FDM_PORT_IN + ($j * 10))) model.xacro.sdf -o model.sdf
|
||||
#! string is better than using -file option. File is not up to date in the next iteration.
|
||||
value=$(</ardupilot_gazebo/models/$DRONE_MODEL/model.sdf)
|
||||
gz_drone_name=spiri-$(($j + 1))
|
||||
#? Maybe read from text file for formation of drones? x y z r p y
|
||||
ros2 run ros_gz_sim create -world $WORLD_NAME -string "$value" -name $gz_drone_name -x $j -y 0 -z 0.195
|
||||
done
|
||||
exec "$@"
|
|
@ -0,0 +1,93 @@
|
|||
from nicegui import ui, binding, app, run
|
||||
import subprocess
|
||||
from collections import defaultdict
|
||||
import docker
|
||||
import time
|
||||
|
||||
docker_client = docker.from_env()
|
||||
|
||||
# Dictionary of applications: key is the button text, value is the command to execute
|
||||
applications = {
|
||||
"Terminal": ["qterminal"],
|
||||
"rqt": ["rqt"],
|
||||
"rviz2": ["rviz2"],
|
||||
"Gazebo": ["/gz_entrypoint.sh"],
|
||||
"Gazebo Standalone": "gz sim -v4".split(),
|
||||
# Add more applications here if needed
|
||||
}
|
||||
|
||||
# Function to launch an application
|
||||
def launch_app(command):
|
||||
try:
|
||||
subprocess.Popen(command)
|
||||
except FileNotFoundError:
|
||||
print(f"{command[0]} not found. Make sure it's installed and accessible in the PATH.")
|
||||
|
||||
# Create the NiceGUI interface
|
||||
ui.label("Spiri Robotics SDK").style('font-size: 40px; margin-bottom: 10px;').classes('w-full text-center')
|
||||
|
||||
robots = []
|
||||
from spiri_sdk_guitools.sim_drone import Robot
|
||||
import aiodocker
|
||||
import asyncio
|
||||
|
||||
@ui.page('/')
|
||||
async def main():
|
||||
with ui.tabs().classes('w-full') as tabs:
|
||||
tab_tools = ui.tab('Tools')
|
||||
tab_robots = ui.tab('Robots')
|
||||
# two = ui.tab('Two')
|
||||
with ui.tab_panels(tabs, value=tab_tools).classes("w-full"):
|
||||
with ui.tab_panel(tab_tools):
|
||||
# Create and place buttons dynamically based on the dictionary
|
||||
with ui.grid(columns=3):
|
||||
for app_name, command in applications.items():
|
||||
ui.button(app_name, on_click=lambda cmd=command: launch_app(cmd)).style('width: 150px; height: 50px; margin: 5px;')
|
||||
with ui.tab_panel(tab_robots):
|
||||
new_robot_widget = ui.element().classes("w-1/2")
|
||||
with ui.element().classes("w-1/2"):
|
||||
ui.label("Debug").classes("text-xl")
|
||||
def cleanup_all_containers():
|
||||
#Find all containers that start with spiri-sdk
|
||||
containers = docker_client.containers.list(all=True)
|
||||
removal_count = 0
|
||||
for container in containers:
|
||||
if container.name.startswith("robot-sim-"):
|
||||
container.remove(force=True)
|
||||
removal_count += 1
|
||||
ui.label(f"Removed {removal_count} containers")
|
||||
ui.button("Cleanup all containers", on_click=cleanup_all_containers)
|
||||
|
||||
ui.separator()
|
||||
ui.label("Current robots").classes("text-3xl")
|
||||
robots_widget = ui.element().classes("w-full")
|
||||
|
||||
#Add a new robot
|
||||
with new_robot_widget:
|
||||
ui.label("Add new robot").classes("text-3xl")
|
||||
newRobotParams = defaultdict(binding.BindableProperty)
|
||||
ui.number(value=1, label="SysID", min=1, max=254,
|
||||
).bind_value(newRobotParams, 'sysid')
|
||||
default_robot_compose = (
|
||||
"/robots/spiri-mu/core/docker-compose.yaml\n"
|
||||
"#/robots/spiri-mu/virtual_camera/docker-compose.yaml --build"
|
||||
)
|
||||
ui.label("Compose files").classes("text-xl")
|
||||
ui.codemirror(value=default_robot_compose, language="bash", theme="basicDark").bind_value(newRobotParams, 'compose_files')
|
||||
async def new_robot():
|
||||
compose_files = []
|
||||
#Split on comma or newline, and remove comments
|
||||
for line in newRobotParams['compose_files'].split('\n'):
|
||||
line = line.split('#')[0].strip()
|
||||
if line:
|
||||
compose_files.append(line)
|
||||
current_robot = newRobotParams.copy()
|
||||
current_robot['compose_files'] = compose_files
|
||||
robot = Robot(**current_robot)
|
||||
asyncio.tasks.create_task(robot.ui(robots_widget))
|
||||
|
||||
newRobotParams['sysid'] += 1
|
||||
ui.button("Add", on_click=new_robot)
|
||||
|
||||
# Start the NiceGUI application
|
||||
ui.run(title="Spiri SDK Launcher", port=8923, dark=None)
|
|
@ -0,0 +1,330 @@
|
|||
from loguru import logger
|
||||
from pathlib import Path
|
||||
import contextlib
|
||||
from typing import List
|
||||
import os
|
||||
import sh
|
||||
import subprocess
|
||||
from nicegui import ui, run, app
|
||||
import yaml
|
||||
|
||||
import docker
|
||||
import aiodocker
|
||||
import asyncio
|
||||
from spiri_sdk_guitools.video_button import EnableStreamingButton
|
||||
|
||||
docker_client = docker.from_env()
|
||||
|
||||
from rclpy.executors import ExternalShutdownException
|
||||
from rclpy.node import Node
|
||||
import rclpy
|
||||
import threading
|
||||
|
||||
|
||||
def ros_main() -> None:
|
||||
rclpy.init()
|
||||
|
||||
|
||||
app.on_startup(lambda: threading.Thread(target=ros_main).start())
|
||||
|
||||
|
||||
@contextlib.contextmanager
|
||||
def modified_environ(*remove, **update):
|
||||
"""
|
||||
Temporarily updates the ``os.environ`` dictionary in-place.
|
||||
|
||||
The ``os.environ`` dictionary is updated in-place so that the modification
|
||||
is sure to work in all situations.
|
||||
|
||||
:param remove: Environment variables to remove.
|
||||
:param update: Dictionary of environment variables and values to add/update.
|
||||
"""
|
||||
env = os.environ
|
||||
update = update or {}
|
||||
remove = remove or []
|
||||
|
||||
# List of environment variables being updated or removed.
|
||||
stomped = (set(update.keys()) | set(remove)) & set(env.keys())
|
||||
# Environment variables and values to restore on exit.
|
||||
update_after = {k: env[k] for k in stomped}
|
||||
# Environment variables and values to remove on exit.
|
||||
remove_after = frozenset(k for k in update if k not in env)
|
||||
|
||||
try:
|
||||
env.update(update)
|
||||
[env.pop(k, None) for k in remove]
|
||||
yield
|
||||
finally:
|
||||
env.update(update_after)
|
||||
[env.pop(k) for k in remove_after]
|
||||
|
||||
|
||||
robots = set()
|
||||
|
||||
|
||||
async def container_logs(container, element):
|
||||
adocker = aiodocker.Docker()
|
||||
with element:
|
||||
with ui.scroll_area():
|
||||
acontainer = await adocker.containers.get(container.id)
|
||||
async for log in acontainer.log(stdout=True, stderr=True, follow=True):
|
||||
for line in log.splitlines():
|
||||
ui.label(line)
|
||||
# ui.html(conv.convert(bytes(log,'utf-8').decode('utf-8', 'xmlcharrefreplace'), full=False))
|
||||
|
||||
|
||||
class Robot:
|
||||
robot_type = "spiri_mu"
|
||||
|
||||
def __init__(self, sysid: int, compose_files: List[Path] | str):
|
||||
if sysid > 255 or sysid < 0:
|
||||
raise ValueError("sysid must be between 0 and 255")
|
||||
self.sysid = int(sysid)
|
||||
if isinstance(compose_files, str):
|
||||
compose_files = [
|
||||
Path(file) for file in compose_files.replace("\n", ",").split(",")
|
||||
]
|
||||
self.compose_files = compose_files
|
||||
self.processes = []
|
||||
self.video_button = None
|
||||
robots.add(self)
|
||||
#Ros doesn't like dashes in node names
|
||||
self.robot_name = f"{self.robot_type}_{self.sysid}".replace("-","_")
|
||||
self.world_name = "citadel_hill"
|
||||
|
||||
async def ui_containers(self, element):
|
||||
docker_elements = {}
|
||||
container_status = {}
|
||||
with element:
|
||||
while True:
|
||||
with logger.catch():
|
||||
# Poll for data that changes
|
||||
for container in self.containers():
|
||||
try:
|
||||
health = container.attrs["State"]["Health"]["Status"]
|
||||
except KeyError:
|
||||
health = "Unknown"
|
||||
|
||||
container_status[container] = (
|
||||
f"{container.name} {container.status} {health}"
|
||||
)
|
||||
if container not in docker_elements:
|
||||
docker_elements[container] = ui.element().classes("w-full")
|
||||
with docker_elements[container]:
|
||||
ui.label().bind_text(container_status, container).classes(
|
||||
"text-2xl"
|
||||
)
|
||||
#Show the command the container is running
|
||||
# ui.label(container.attrs["Config"]["Cmd"])
|
||||
cmd_widget = ui.codemirror(" ".join(container.attrs["Config"]["Cmd"]), language="bash",theme="basicDark").classes('h-auto max-h-32')
|
||||
cmd_widget.enabled = False
|
||||
|
||||
with ui.expansion("Env Variables").classes("w-full outline outline-1").style("margin: 10px;"):
|
||||
env_widget = ui.codemirror("\n".join(container.attrs["Config"]["Env"]), language="bash",theme="basicDark")
|
||||
env_widget.enabled = False
|
||||
|
||||
logelement = (
|
||||
ui.expansion("Logs")
|
||||
.style("margin: 10px;")
|
||||
.classes("w-full outline outline-1")
|
||||
)
|
||||
asyncio.create_task(container_logs(container, logelement))
|
||||
with ui.expansion("Full details").classes("w-full outline outline-1").style("margin: 10px;"):
|
||||
details_widget = ui.codemirror(yaml.dump(container.attrs), language="yaml",theme="basicDark")
|
||||
details_widget.enabled = False
|
||||
# Check for containers that have been removed
|
||||
removed = set(docker_elements.keys()) - set(self.containers())
|
||||
for container in removed:
|
||||
self.robot_ui.remove(docker_elements[container])
|
||||
docker_elements.pop(container)
|
||||
await asyncio.sleep(1)
|
||||
|
||||
async def ui_ros(self, element):
|
||||
with element:
|
||||
node_dummy = Node("_ros2cli_dummy_to_show_topic_list")
|
||||
scroll_area = ui.scroll_area()
|
||||
with scroll_area:
|
||||
while True:
|
||||
scroll_area.clear()
|
||||
#Filter for topics that start with self.robot_name
|
||||
for topic in node_dummy.get_topic_names_and_types():
|
||||
if self.robot_name in topic[0]:
|
||||
ui.label(topic[0])
|
||||
await asyncio.sleep(10)
|
||||
|
||||
async def ui(self, element):
|
||||
adocker = aiodocker.Docker()
|
||||
|
||||
with element:
|
||||
self.robot_ui = ui.element().classes("w-full outline p-4")
|
||||
with self.robot_ui:
|
||||
ui.label(f"{self.robot_type} {self.sysid}").classes("text-2xl")
|
||||
ui.label(f"""Sysid: {self.sysid}""")
|
||||
ui.button("Start", on_click=self.async_start).classes("m-2")
|
||||
ui.button("Stop", on_click=self.async_stop).classes("m-2")
|
||||
self.video_button = EnableStreamingButton(robot_name=self.robot_name).classes(
|
||||
"m-2"
|
||||
)
|
||||
|
||||
async def delete_robot():
|
||||
await self.async_stop()
|
||||
robots.remove(self)
|
||||
element.remove(self.robot_ui)
|
||||
|
||||
ui.button("Delete", on_click=delete_robot).classes("m-2")
|
||||
with ui.tabs() as tabs:
|
||||
tab_containers = ui.tab("Containers")
|
||||
tab_ros = ui.tab("ROS Topics")
|
||||
with ui.tab_panels(tabs, value=tab_containers):
|
||||
tab = ui.tab_panel(tab_containers).classes("w-full")
|
||||
asyncio.create_task(self.ui_containers(tab))
|
||||
tab = ui.tab_panel(tab_ros).classes("w-full")
|
||||
asyncio.create_task(self.ui_ros(tab))
|
||||
|
||||
async def async_stop(self):
|
||||
return await run.io_bound(self.stop)
|
||||
|
||||
def stop(self):
|
||||
# If video was enabled, and we stop. Button state is wrong.
|
||||
# Stop video if it is enabled.
|
||||
if isinstance(self.video_button, EnableStreamingButton):
|
||||
self.video_button.stop_video()
|
||||
# Delete gazebo model
|
||||
self.delete_gz_model()
|
||||
# Signal all processes to stop
|
||||
for process in self.processes:
|
||||
process.terminate()
|
||||
process.kill()
|
||||
self.processes = []
|
||||
for container in self.containers():
|
||||
# container.stop()
|
||||
container.remove(force=True)
|
||||
|
||||
def containers(self):
|
||||
return docker_client.containers.list(
|
||||
all=True, filters={"name": f"robot-sim-{self.robot_name}"}
|
||||
)
|
||||
|
||||
async def async_start(self):
|
||||
return await run.io_bound(self.start)
|
||||
|
||||
def start(self):
|
||||
"""Starts the simulated drone with a given sysid,
|
||||
each drone must have it's own unique ID.
|
||||
"""
|
||||
instance = self.sysid - 1
|
||||
sysid = self.sysid
|
||||
with logger.contextualize(syd_id=sysid):
|
||||
env = os.environ
|
||||
with modified_environ(
|
||||
SERIAL0_PORT=str(int(env["SERIAL0_PORT"]) + 10 * instance),
|
||||
MAVROS2_PORT=str(int(env["MAVROS2_PORT"]) + 10 * instance),
|
||||
MAVROS1_PORT=str(int(env["MAVROS1_PORT"]) + 10 * instance),
|
||||
FDM_PORT_IN=str(int(env["FDM_PORT_IN"]) + 10 * instance),
|
||||
GSTREAMER_UDP_PORT=str(int(env["GSTREAMER_UDP_PORT"]) + 10 * instance),
|
||||
SITL_PORT=str(int(env["SITL_PORT"]) + 10 * instance),
|
||||
INSTANCE=str(instance),
|
||||
DRONE_SYS_ID=str(self.sysid),
|
||||
ROBOT_NAME=self.robot_name,
|
||||
WORLD_NAME="citadel_hill",
|
||||
):
|
||||
self.spawn_gz_model()
|
||||
logger.info("Starting drone stack, this may take some time")
|
||||
for compose_file in self.compose_files:
|
||||
arguments = compose_file.split(" ")
|
||||
arguments = [arg.strip() for arg in arguments]
|
||||
compose_file = arguments[0]
|
||||
arguments = arguments[1:]
|
||||
|
||||
if not isinstance(compose_file, Path):
|
||||
compose_file = Path(compose_file)
|
||||
if not compose_file.exists():
|
||||
raise FileNotFoundError(f"File {compose_file} does not exist")
|
||||
#Get the folder the compose file is in
|
||||
compose_folder = compose_file.parent
|
||||
args = [
|
||||
"docker-compose",
|
||||
"--profile",
|
||||
"uav-sim",
|
||||
"-p",
|
||||
f"robot-sim-{self.robot_name}-{compose_folder.name}",
|
||||
"-f",
|
||||
compose_file.as_posix(),
|
||||
"up",
|
||||
*arguments,
|
||||
]
|
||||
command = " ".join(args)
|
||||
|
||||
logger.info(f"Starting drone stack with command: {command}")
|
||||
docker_stack = subprocess.Popen(
|
||||
args,
|
||||
# stdout=subprocess.PIPE,
|
||||
# stderr=subprocess.PIPE,
|
||||
)
|
||||
logger.info(f"Started drone stack with PID: {docker_stack.pid}")
|
||||
|
||||
@logger.catch
|
||||
def spawn_gz_model(self):
|
||||
sysid = self.sysid
|
||||
logger.info("")
|
||||
env = os.environ
|
||||
GSTREAMER_UDP_PORT = env["GSTREAMER_UDP_PORT"]
|
||||
FDM_PORT_IN = env["FDM_PORT_IN"]
|
||||
WORLD_NAME = env["WORLD_NAME"]
|
||||
DRONE_MODEL = env["DRONE_MODEL"]
|
||||
XACRO_CMD = [
|
||||
"xacro",
|
||||
f"gstreamer_udp_port:={GSTREAMER_UDP_PORT}",
|
||||
f"fdm_port_in:={FDM_PORT_IN}",
|
||||
"model.xacro.sdf",
|
||||
"-o",
|
||||
"model.sdf",
|
||||
]
|
||||
# This path is breaking if this drone_model folder doesnt exist!
|
||||
# TODO: fix this model path for minimal code maintenance
|
||||
ROS2_CMD = f"ros2 run ros_gz_sim create -world {WORLD_NAME} -file /ardupilot_gazebo/models/{DRONE_MODEL}/model.sdf -name {self.robot_name} -x {sysid - 1} -y 0 -z 0.195"
|
||||
|
||||
xacro_proc = subprocess.Popen(
|
||||
XACRO_CMD,
|
||||
cwd=f"/ardupilot_gazebo/models/{DRONE_MODEL}",
|
||||
stdout=subprocess.PIPE,
|
||||
stderr=subprocess.PIPE,
|
||||
)
|
||||
ros2_gz_create_proc = subprocess.Popen(
|
||||
ROS2_CMD.split(),
|
||||
stdout=subprocess.PIPE,
|
||||
stderr=subprocess.PIPE,
|
||||
)
|
||||
out, err = ros2_gz_create_proc.communicate(timeout=15)
|
||||
ros2_gz_create_proc.kill()
|
||||
return
|
||||
|
||||
@logger.catch
|
||||
def delete_gz_model(self):
|
||||
sysid = self.sysid
|
||||
env = os.environ
|
||||
WORLD_NAME = env["WORLD_NAME"]
|
||||
# http://osrf-distributions.s3.amazonaws.com/gazebo/api/7.1.0/classgazebo_1_1physics_1_1Entity.html
|
||||
ENTITY_TYPE_MODEL = 0x00000002
|
||||
REQUEST_ARG = f"name: '{self.robot_name}' type: {ENTITY_TYPE_MODEL}"
|
||||
GZ_SERVICE_CMD = [
|
||||
"gz",
|
||||
"service",
|
||||
"-s",
|
||||
f"/world/{WORLD_NAME}/remove",
|
||||
"--reqtype",
|
||||
"gz.msgs.Entity",
|
||||
"--reptype",
|
||||
"gz.msgs.Boolean",
|
||||
"--timeout",
|
||||
"5000",
|
||||
"--req",
|
||||
REQUEST_ARG,
|
||||
]
|
||||
remove_entity_proc = subprocess.Popen(
|
||||
GZ_SERVICE_CMD,
|
||||
stdout=subprocess.PIPE,
|
||||
stderr=subprocess.PIPE,
|
||||
)
|
||||
out, err = remove_entity_proc.communicate(timeout=15)
|
||||
remove_entity_proc.kill()
|
|
@ -0,0 +1,82 @@
|
|||
from nicegui import run, ui
|
||||
import subprocess
|
||||
from loguru import logger
|
||||
import os
|
||||
|
||||
GZ_TOPIC_INFO = ["gz", "topic", "-i", "-t"]
|
||||
ENABLE_STREAMING_TOPIC = "/world/{world_name}/model/{robot_name}/link/pitch_link/sensor/camera/image/enable_streaming"
|
||||
|
||||
|
||||
class EnableStreamingButton(ui.element):
|
||||
def __init__(self, robot_name, state: bool = False) -> None:
|
||||
super().__init__()
|
||||
self.robot_name = robot_name
|
||||
self._state = state
|
||||
self.button = None
|
||||
with self.classes():
|
||||
with ui.row():
|
||||
self.button = ui.button(on_click=self.on_click)
|
||||
self.button._text = (
|
||||
f'{"Disable" if self._state else "Enable"}' + " Video"
|
||||
)
|
||||
self.button.props(f'color={"red" if self._state else "green"}')
|
||||
|
||||
async def on_click(self) -> None:
|
||||
spinner = ui.spinner(size="lg")
|
||||
# So we don't block UI
|
||||
result = await run.cpu_bound(self.enable_streaming, self.robot_name, not self._state)
|
||||
if result:
|
||||
ui.notify("Success", type="positive]")
|
||||
self.set_state(state=not self._state)
|
||||
else:
|
||||
self.set_state(state=False)
|
||||
ui.notify("No Video Streaming Available..", type="negative")
|
||||
spinner.delete()
|
||||
|
||||
def set_state(self, state: bool):
|
||||
self._state = state
|
||||
self.update()
|
||||
|
||||
def update(self) -> None:
|
||||
self.button._text = f'{"Disable" if self._state else "Enable"}' + " Video"
|
||||
self.button.props(f'color={"red" if self._state else "green"}')
|
||||
|
||||
def stop_video(self):
|
||||
if self._state != False:
|
||||
self.enable_streaming(robot_name=self.robot_name, is_streaming=False)
|
||||
self.set_state(state=False)
|
||||
|
||||
@staticmethod
|
||||
def enable_streaming(robot_name: str, is_streaming: bool) -> bool:
|
||||
|
||||
world_name = os.environ["WORLD_NAME"]
|
||||
# Check if this topic has any subscribers i.e. model is up
|
||||
gz_topic_list_proc = subprocess.Popen(
|
||||
GZ_TOPIC_INFO
|
||||
+ [ENABLE_STREAMING_TOPIC.format(world_name=world_name, robot_name=robot_name)],
|
||||
stdout=subprocess.PIPE,
|
||||
stderr=subprocess.PIPE,
|
||||
text=True,
|
||||
)
|
||||
# Check exceptions.. timeout, error etc.
|
||||
|
||||
output, error = gz_topic_list_proc.communicate(timeout=15)
|
||||
if gz_topic_list_proc.returncode != 0:
|
||||
logger.error(error)
|
||||
return False
|
||||
if "No subscribers".casefold() in output.casefold():
|
||||
logger.error("No Subscribers on enable_streaming topic.")
|
||||
return False
|
||||
|
||||
gz_topic_pub = [
|
||||
"gz",
|
||||
"topic",
|
||||
"-t",
|
||||
ENABLE_STREAMING_TOPIC.format(world_name=world_name, robot_name=robot_name),
|
||||
"-m",
|
||||
"gz.msgs.Boolean",
|
||||
"-p",
|
||||
f"data:{is_streaming}",
|
||||
]
|
||||
subprocess.run((gz_topic_pub))
|
||||
return True
|
|
@ -0,0 +1,4 @@
|
|||
typer==0.12.5
|
||||
loguru==0.7.2
|
||||
sh==2.1.0
|
||||
python-dotenv==1.0.1
|
|
@ -0,0 +1,19 @@
|
|||
DRONE_SYS_ID=1
|
||||
INSTANCE=0
|
||||
SERIAL0_PORT=5760
|
||||
SITL_PORT=5501
|
||||
MAVROS2_PORT=14560
|
||||
MAVROS1_PORT=14561
|
||||
FDM_PORT_IN=9002
|
||||
GSTREAMER_UDP_PORT=5600
|
||||
ROS_MASTER_URI="http://0.0.0.0:11311"
|
||||
|
||||
|
||||
ARDUPILOT_VEHICLE="-v copter -f gazebo-mu --model=JSON -L CitadelHill"
|
||||
WORLD_FILE_NAME="citadel_hill_world.sdf"
|
||||
WORLD_NAME="citadel_hill"
|
||||
DRONE_MODEL="spiri_mu"
|
||||
|
||||
|
||||
SIM_DRONE_COUNT=1
|
||||
GCS_PORT=14550
|
|
@ -0,0 +1,95 @@
|
|||
services:
|
||||
ardupilot:
|
||||
env_file:
|
||||
- .env
|
||||
image: git.spirirobotics.com/spiri/ardupilot:spiri-master
|
||||
command:
|
||||
- /bin/bash
|
||||
- -c
|
||||
- |
|
||||
./Tools/autotest/sim_vehicle.py $ARDUPILOT_VEHICLE --no-rebuild \
|
||||
--no-mavproxy --enable-dds --sysid $DRONE_SYS_ID -I$INSTANCE
|
||||
stdin_open: true
|
||||
tty: true
|
||||
network_mode: host
|
||||
|
||||
mavproxy:
|
||||
env_file:
|
||||
- .env
|
||||
image: git.spirirobotics.com/spiri/services-mavproxy:main
|
||||
command: >
|
||||
mavproxy.py --non-interactive
|
||||
--master tcp:127.0.0.1:$SERIAL0_PORT
|
||||
--out udpout:0.0.0.0:$MAVROS2_PORT
|
||||
--out udpout:0.0.0.0:$MAVROS1_PORT
|
||||
--sitl 127.0.0.1:$SITL_PORT
|
||||
--out udp:0.0.0.0:$GCS_PORT
|
||||
ipc: host
|
||||
network_mode: host
|
||||
restart: always
|
||||
|
||||
mavros2:
|
||||
env_file:
|
||||
- .env
|
||||
image: git.spirirobotics.com/spiri/services-ros2-mavros:main
|
||||
command: ros2 launch mavros apm.launch fcu_url:="udp://0.0.0.0:$MAVROS2_PORT@:14555" namespace:="$ROBOT_NAME" tgt_system:="$DRONE_SYS_ID"
|
||||
ipc: host
|
||||
network_mode: host
|
||||
restart: always
|
||||
depends_on:
|
||||
ardupilot:
|
||||
condition: service_started
|
||||
mavproxy:
|
||||
condition: service_started
|
||||
deploy:
|
||||
resources:
|
||||
limits:
|
||||
# cpus: '0.01'
|
||||
memory: 200M
|
||||
ulimits:
|
||||
nofile:
|
||||
soft: 1024
|
||||
hard: 524288
|
||||
|
||||
mavros:
|
||||
#This service bridges our mavlink-based robot-coprosessor into ROS
|
||||
#In this example it connects to a simulated coprocessor.
|
||||
env_file:
|
||||
- .env
|
||||
image: git.spirirobotics.com/spiri/services-ros1-mavros:master
|
||||
command: rosrun mavros mavros_node __name:=spiri$DRONE_SYS_ID _fcu_url:="udp://0.0.0.0:$MAVROS1_PORT@:14559" _target_system_id:="$DRONE_SYS_ID"
|
||||
profiles:
|
||||
- ros1
|
||||
ipc: host
|
||||
network_mode: host
|
||||
restart: always
|
||||
deploy:
|
||||
resources:
|
||||
limits:
|
||||
# cpus: '0.01'
|
||||
memory: 200M
|
||||
ulimits:
|
||||
nofile:
|
||||
soft: 1024
|
||||
hard: 524288
|
||||
|
||||
ros-master:
|
||||
env_file:
|
||||
- .env
|
||||
image: git.spirirobotics.com/spiri/services-ros1-core:main
|
||||
command: stdbuf -o L roscore
|
||||
profiles:
|
||||
- ros1-master
|
||||
ipc: host
|
||||
network_mode: host
|
||||
restart: always
|
||||
deploy:
|
||||
resources:
|
||||
limits:
|
||||
memory: 1G
|
||||
# Madness, setting a low ulimit here fixes memory leaks
|
||||
# https://answers.ros.org/question/336963/rosout-high-memory-usage/
|
||||
ulimits:
|
||||
nofile:
|
||||
soft: 1024
|
||||
hard: 524288
|
|
@ -0,0 +1,8 @@
|
|||
FROM git.spirirobotics.com/spiri/services-ros2-mavros:main
|
||||
|
||||
RUN apt-get update
|
||||
RUN apt-get --yes install ros-${ROS_DISTRO}-ros-gz-bridge \
|
||||
ros-${ROS_DISTRO}-ros-gz-image \
|
||||
ros-${ROS_DISTRO}-compressed-image-transport \
|
||||
ros-${ROS_DISTRO}-rmw-cyclonedds-cpp
|
||||
|
|
@ -0,0 +1,13 @@
|
|||
|
||||
services:
|
||||
front-gimbal:
|
||||
ipc: host
|
||||
network_mode: host
|
||||
# image: git.spirirobotics.com/spiri/services-ros2-mavros:main
|
||||
#Build the iamge, give it a name, don't try to pull the image
|
||||
build: ./
|
||||
image: spirisdk-virtual_camera
|
||||
pull_policy: never
|
||||
environment:
|
||||
- RMW_IMPLEMENTATION=rmw_cyclonedds_cpp
|
||||
command: ros2 run ros_gz_image image_bridge /world/${WORLD_NAME}/model/${ROBOT_NAME}/link/pitch_link/sensor/camera/image
|
Loading…
Reference in New Issue