mirror of
https://github.com/CopterExpress/clover.git
synced 2026-06-01 15:39:32 +00:00
Compare commits
19 Commits
simple-off
...
fix-builde
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
23c745c17b | ||
|
|
c646aa3b43 | ||
|
|
dbeef76ec0 | ||
|
|
91543a9c0e | ||
|
|
f25816f2d5 | ||
|
|
62e5a20a79 | ||
|
|
857e41516a | ||
|
|
48549de07f | ||
|
|
1c70499ebd | ||
|
|
6190605312 | ||
|
|
27ae453136 | ||
|
|
9d544a139a | ||
|
|
bf84a98cb6 | ||
|
|
9270b6a278 | ||
|
|
8a95789d77 | ||
|
|
db328ba25e | ||
|
|
afa50add8f | ||
|
|
735b88d498 | ||
|
|
8aaaae929e |
1
.gitattributes
vendored
1
.gitattributes
vendored
@@ -3,7 +3,6 @@ roslib.js linguist-vendored
|
|||||||
eventemitter2.js linguist-vendored
|
eventemitter2.js linguist-vendored
|
||||||
ros3d.js linguist-vendored
|
ros3d.js linguist-vendored
|
||||||
three.min.js linguist-vendored
|
three.min.js linguist-vendored
|
||||||
json-to-pretty-yaml.js linguist-vendored
|
|
||||||
aruco_pose/vendor/* linguist-vendored
|
aruco_pose/vendor/* linguist-vendored
|
||||||
blockly/* linguist-vendored
|
blockly/* linguist-vendored
|
||||||
highlight/* linguist-vendored
|
highlight/* linguist-vendored
|
||||||
|
|||||||
29
.github/workflows/build-image.yaml
vendored
29
.github/workflows/build-image.yaml
vendored
@@ -1,29 +0,0 @@
|
|||||||
name: RPi image
|
|
||||||
|
|
||||||
on:
|
|
||||||
push:
|
|
||||||
branches: [ '*' ]
|
|
||||||
pull_request:
|
|
||||||
branches: [ master ]
|
|
||||||
release:
|
|
||||||
types: [ created ]
|
|
||||||
|
|
||||||
jobs:
|
|
||||||
build:
|
|
||||||
runs-on: ubuntu-latest
|
|
||||||
steps:
|
|
||||||
- uses: actions/checkout@v2
|
|
||||||
- name: Build image
|
|
||||||
run: |
|
|
||||||
docker run --privileged --rm -v /dev:/dev -v $(pwd):/builder/repo -e TRAVIS_TAG="${{ github.event.release.tag_name }}" sfalexrog/img-tool:qemu-update
|
|
||||||
- name: Compress image
|
|
||||||
run: |
|
|
||||||
cd images && sudo chmod -R 777 . && zip -9 $(echo clover_*).zip clover_* && ls -l . && unzip -l clover_*.zip
|
|
||||||
- name: Upload image
|
|
||||||
uses: softprops/action-gh-release@v1
|
|
||||||
if: ${{ github.event_name == 'release' }}
|
|
||||||
with:
|
|
||||||
files: images/clover_*.zip
|
|
||||||
prerelease: true
|
|
||||||
env:
|
|
||||||
GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
|
|
||||||
51
.github/workflows/build.yml
vendored
51
.github/workflows/build.yml
vendored
@@ -1,51 +0,0 @@
|
|||||||
name: Build
|
|
||||||
|
|
||||||
on:
|
|
||||||
push:
|
|
||||||
branches: [ '*' ]
|
|
||||||
pull_request:
|
|
||||||
branches: [ master ]
|
|
||||||
|
|
||||||
jobs:
|
|
||||||
# melodic:
|
|
||||||
# runs-on: ubuntu-latest
|
|
||||||
# steps:
|
|
||||||
# - uses: actions/checkout@v2
|
|
||||||
# - name: Native Melodic build
|
|
||||||
# run: |
|
|
||||||
# docker run --rm -v $(pwd):/root/catkin_ws/src/clover ros:melodic-ros-base /root/catkin_ws/src/clover/builder/standalone-install.sh
|
|
||||||
noetic:
|
|
||||||
runs-on: ubuntu-latest
|
|
||||||
container: ros:noetic-ros-base
|
|
||||||
defaults:
|
|
||||||
run:
|
|
||||||
working-directory: catkin_ws
|
|
||||||
shell: bash
|
|
||||||
steps:
|
|
||||||
- uses: actions/checkout@v2
|
|
||||||
with:
|
|
||||||
path: catkin_ws/src/clover
|
|
||||||
- name: Install requirements
|
|
||||||
run: apt-get update && apt-get -y install python3-pip fakeroot python3-bloom debhelper dpkg-dev
|
|
||||||
- name: Install dependencies
|
|
||||||
run: rosdep update && rosdep install --from-paths src --ignore-src -y
|
|
||||||
- name: Install GeographicLib datasets
|
|
||||||
run: wget -qO- https://raw.githubusercontent.com/mavlink/mavros/master/mavros/scripts/install_geographiclib_datasets.sh | bash
|
|
||||||
- name: catkin_make
|
|
||||||
run: source /opt/ros/$ROS_DISTRO/setup.bash && catkin_make
|
|
||||||
- name: Run tests
|
|
||||||
run: source devel/setup.bash && catkin_make run_tests && catkin_test_results
|
|
||||||
- name: Build Debian packages
|
|
||||||
run: |
|
|
||||||
source devel/setup.bash
|
|
||||||
for file in `find . -name "package.xml"`; do
|
|
||||||
cd $(dirname ${file})
|
|
||||||
bloom-generate rosdebian --os-name ubuntu --os-version $(lsb_release -cs) --ros-distro $ROS_DISTRO
|
|
||||||
fakeroot debian/rules binary
|
|
||||||
cd -
|
|
||||||
done
|
|
||||||
- uses: actions/upload-artifact@v3
|
|
||||||
with:
|
|
||||||
name: debian-packages
|
|
||||||
path: catkin_ws/src/clover/*.deb
|
|
||||||
retention-days: 1
|
|
||||||
85
.github/workflows/docs.yml
vendored
85
.github/workflows/docs.yml
vendored
@@ -1,85 +0,0 @@
|
|||||||
name: Documentation
|
|
||||||
|
|
||||||
on:
|
|
||||||
push:
|
|
||||||
branches: [ '*' ]
|
|
||||||
pull_request:
|
|
||||||
branches: [ '*' ]
|
|
||||||
|
|
||||||
permissions:
|
|
||||||
contents: read
|
|
||||||
pages: write
|
|
||||||
id-token: write
|
|
||||||
|
|
||||||
defaults:
|
|
||||||
run:
|
|
||||||
shell: bash
|
|
||||||
|
|
||||||
jobs:
|
|
||||||
docs:
|
|
||||||
runs-on: ubuntu-22.04
|
|
||||||
steps:
|
|
||||||
- uses: actions/checkout@v2
|
|
||||||
- name: Use Node.js
|
|
||||||
uses: actions/setup-node@v1
|
|
||||||
with: { node-version: '10' }
|
|
||||||
- name: Setup tools
|
|
||||||
run: |
|
|
||||||
sudo sh -c "echo ttf-mscorefonts-installer msttcorefonts/accepted-mscorefonts-eula select true | debconf-set-selections"
|
|
||||||
sudo apt-get update && sudo apt-get install -y calibre msttcorefonts
|
|
||||||
builder/assets/install_gitbook.sh
|
|
||||||
npm install markdownlint-cli@0.28.1 -g # FIXME: https://github.com/DavidAnson/markdownlint/issues/435
|
|
||||||
npm install svgexport -g
|
|
||||||
gitbook -V
|
|
||||||
markdownlint -V
|
|
||||||
- name: Run markdownlint
|
|
||||||
run: markdownlint docs
|
|
||||||
- name: Check Assets
|
|
||||||
run: |
|
|
||||||
./check_assets_size.py
|
|
||||||
./check_unused_assets.py
|
|
||||||
- name: Build GitBook
|
|
||||||
run: |
|
|
||||||
gitbook install
|
|
||||||
gitbook build
|
|
||||||
- name: Generate PDF
|
|
||||||
id: generate-pdf
|
|
||||||
env:
|
|
||||||
GITBOOK_SKIP_PDF: ${{ secrets.GITBOOK_SKIP_PDF }}
|
|
||||||
continue-on-error: ${{ env.GITBOOK_SKIP_PDF != '' }}
|
|
||||||
if: ${{ github.event_name == 'push' }}
|
|
||||||
run: |
|
|
||||||
for i in 1 2 3 4; do gitbook pdf ./ _book/clover.pdf && break || sleep 1; done
|
|
||||||
sudo apt-get -q install ghostscript
|
|
||||||
gs -sDEVICE=pdfwrite -dCompatibilityLevel=1.4 -dPDFSETTINGS=/default -dNOPAUSE -dQUIET -dBATCH -dDetectDuplicateImages -dCompressFonts=true -r150 -sOutputFile=_book/clover_ru_compressed.pdf _book/clover_ru.pdf
|
|
||||||
gs -sDEVICE=pdfwrite -dCompatibilityLevel=1.4 -dPDFSETTINGS=/default -dNOPAUSE -dQUIET -dBATCH -dDetectDuplicateImages -dCompressFonts=true -r150 -sOutputFile=_book/clover_en_compressed.pdf _book/clover_en.pdf
|
|
||||||
rm _book/clover_ru.pdf && mv _book/clover_ru_compressed.pdf _book/clover_ru.pdf
|
|
||||||
rm _book/clover_en.pdf && mv _book/clover_en_compressed.pdf _book/clover_en.pdf
|
|
||||||
ls -lah _book/clover*.pdf
|
|
||||||
echo '::set-output name=GITBOOK_PDF_OK::1'
|
|
||||||
- name: Download older PDFs
|
|
||||||
if: ${{ !steps.generate-pdf.outputs.GITBOOK_PDF_OK }}
|
|
||||||
run: |
|
|
||||||
rm -f _book/clover*.pdf
|
|
||||||
wget --no-verbose https://clover.coex.tech/clover_ru.pdf -P _book/
|
|
||||||
wget --no-verbose https://clover.coex.tech/clover_en.pdf -P _book/
|
|
||||||
- name: Upload artifact
|
|
||||||
# if: ${{ github.event_name == 'push' && github.ref == 'refs/heads/master' }}
|
|
||||||
uses: actions/upload-pages-artifact@v1
|
|
||||||
with:
|
|
||||||
path: _book
|
|
||||||
|
|
||||||
deploy-docs:
|
|
||||||
if: ${{ github.event_name == 'push' && github.ref == 'refs/heads/master' }}
|
|
||||||
concurrency:
|
|
||||||
group: "pages"
|
|
||||||
cancel-in-progress: true
|
|
||||||
environment:
|
|
||||||
name: github-pages
|
|
||||||
url: ${{ steps.deployment.outputs.page_url }}
|
|
||||||
runs-on: ubuntu-latest
|
|
||||||
needs: docs
|
|
||||||
steps:
|
|
||||||
- name: Deploy to GitHub Pages
|
|
||||||
id: deployment
|
|
||||||
uses: actions/deploy-pages@v1
|
|
||||||
18
.github/workflows/editorconfig.yaml
vendored
18
.github/workflows/editorconfig.yaml
vendored
@@ -1,18 +0,0 @@
|
|||||||
name: Editorconfig
|
|
||||||
|
|
||||||
on:
|
|
||||||
push:
|
|
||||||
branches: [ '*' ]
|
|
||||||
pull_request:
|
|
||||||
branches: [ master ]
|
|
||||||
|
|
||||||
jobs:
|
|
||||||
editorconfig:
|
|
||||||
runs-on: ubuntu-latest
|
|
||||||
steps:
|
|
||||||
- uses: actions/checkout@v2
|
|
||||||
- name: .editorconfig Linter
|
|
||||||
run: |
|
|
||||||
wget --no-verbose https://github.com/okalachev/editorconfig-checker/releases/download/1.2.1-disable-spaces-amount/ec-linux-amd64
|
|
||||||
chmod +x ec-linux-amd64
|
|
||||||
./ec-linux-amd64 -spaces-after-tabs -e "roslib.js|ros3d.js|eventemitter2.js|json-to-pretty-yaml.js|draw.cpp|BinUtils.swift|\.idea|apps/android/app|blockly/|clover_blocks/programs/|highlight/|python.js|Assets.xcassets|test_parser_pass.txt|test_node_failure.txt|aruco_pose/vendor|\.stl|\.dxf|\.dae|\.material"
|
|
||||||
1
.gitignore
vendored
1
.gitignore
vendored
@@ -6,4 +6,3 @@ _book/
|
|||||||
package-lock.json
|
package-lock.json
|
||||||
clover_blocks/programs/*.*
|
clover_blocks/programs/*.*
|
||||||
!clover_blocks/programs/examples/*
|
!clover_blocks/programs/examples/*
|
||||||
/.vscode/
|
|
||||||
|
|||||||
@@ -21,7 +21,6 @@
|
|||||||
"ROS",
|
"ROS",
|
||||||
"ROS Kinetic",
|
"ROS Kinetic",
|
||||||
"ROS Melodic",
|
"ROS Melodic",
|
||||||
"ROS Noetic",
|
|
||||||
"OpenCV",
|
"OpenCV",
|
||||||
"OpenVPN",
|
"OpenVPN",
|
||||||
"Gazebo",
|
"Gazebo",
|
||||||
@@ -110,8 +109,7 @@
|
|||||||
"Li-ion",
|
"Li-ion",
|
||||||
"Nvidia",
|
"Nvidia",
|
||||||
"VirtualBox",
|
"VirtualBox",
|
||||||
"VMware",
|
"VMware"
|
||||||
"DuoCam"
|
|
||||||
],
|
],
|
||||||
"code_blocks": false
|
"code_blocks": false
|
||||||
},
|
},
|
||||||
|
|||||||
117
.travis.yml
Normal file
117
.travis.yml
Normal file
@@ -0,0 +1,117 @@
|
|||||||
|
os: linux
|
||||||
|
dist: xenial
|
||||||
|
language: generic
|
||||||
|
services:
|
||||||
|
- docker
|
||||||
|
env:
|
||||||
|
global:
|
||||||
|
- DOCKER="sfalexrog/img-tool:qemu-update"
|
||||||
|
- TARGET_REPO="https://github.com/${TRAVIS_REPO_SLUG}.git"
|
||||||
|
- IMAGE_VERSION=${TRAVIS_TAG:-${TRAVIS_COMMIT:0:7}}
|
||||||
|
- IMAGE_NAME="$(basename -s '.git' ${TARGET_REPO})_${IMAGE_VERSION}.img"
|
||||||
|
git:
|
||||||
|
depth: 1
|
||||||
|
jobs:
|
||||||
|
fast_finish: true
|
||||||
|
include:
|
||||||
|
- stage: Build
|
||||||
|
name: "Raspberry Pi Image Build"
|
||||||
|
cache:
|
||||||
|
directories:
|
||||||
|
- imgcache
|
||||||
|
before_script:
|
||||||
|
- docker pull ${DOCKER}
|
||||||
|
# Check if there are any cached images, copy them to our "images" directory
|
||||||
|
- if [ -n "$(ls -A imgcache/*.zip)" ]; then mkdir -p images && cp imgcache/*.zip images; fi
|
||||||
|
script:
|
||||||
|
- if [[ -z ${TRAVIS_TAG} && "${TRAVIS_PULL_REQUEST}" != "false" ]]; then
|
||||||
|
echo "Commit range is ${TRAVIS_COMMIT_RANGE}" &&
|
||||||
|
if [ $(git diff --name-only ${TRAVIS_COMMIT_RANGE} | grep -v ^"docs/" | wc -l) -eq 0 ]; then
|
||||||
|
echo " === Docs-only change; skipping build ===" &&
|
||||||
|
export SKIP_BUILD=true;
|
||||||
|
fi;
|
||||||
|
fi
|
||||||
|
- if [ -z ${SKIP_BUILD} ]; then
|
||||||
|
docker run --privileged --rm -v /dev:/dev -v $(pwd):/builder/repo -e TRAVIS_TAG="${TRAVIS_TAG}" ${DOCKER};
|
||||||
|
fi
|
||||||
|
before_cache:
|
||||||
|
- cp images/*.zip imgcache
|
||||||
|
after_success:
|
||||||
|
- sudo chmod -R 777 *
|
||||||
|
- cd images && zip -9 ${IMAGE_NAME}.zip ${IMAGE_NAME} && stat --printf="Compressed image size:%s\n" ${IMAGE_NAME}.zip
|
||||||
|
before_deploy:
|
||||||
|
# Set up git user name and tag this commit
|
||||||
|
- git config --local user.name "goldarte"
|
||||||
|
- git config --local user.email "goldartt@gmail.com"
|
||||||
|
deploy:
|
||||||
|
provider: releases
|
||||||
|
token: ${GITHUB_OAUTH_TOKEN}
|
||||||
|
file: ${IMAGE_NAME}.zip
|
||||||
|
skip_cleanup: true
|
||||||
|
on:
|
||||||
|
tags: true
|
||||||
|
draft: true
|
||||||
|
name: ${TRAVIS_TAG}
|
||||||
|
- stage: Build
|
||||||
|
name: "Native Kinetic build"
|
||||||
|
env:
|
||||||
|
- NATIVE_DOCKER=ros:kinetic-ros-base
|
||||||
|
before_script:
|
||||||
|
- docker pull ${NATIVE_DOCKER}
|
||||||
|
script:
|
||||||
|
- docker run --rm -v $(pwd):/root/catkin_ws/src/clover ${NATIVE_DOCKER} /root/catkin_ws/src/clover/builder/standalone-install.sh
|
||||||
|
- stage: Build
|
||||||
|
name: "Native Melodic build"
|
||||||
|
env:
|
||||||
|
- NATIVE_DOCKER=ros:melodic-ros-base
|
||||||
|
before_script:
|
||||||
|
- docker pull ${NATIVE_DOCKER}
|
||||||
|
script:
|
||||||
|
- docker run --rm -v $(pwd):/root/catkin_ws/src/clover ${NATIVE_DOCKER} /root/catkin_ws/src/clover/builder/standalone-install.sh
|
||||||
|
- stage: Build
|
||||||
|
name: "Documentation"
|
||||||
|
language: node_js
|
||||||
|
node_js:
|
||||||
|
- "10"
|
||||||
|
before_script:
|
||||||
|
- sudo sh -c "echo ttf-mscorefonts-installer msttcorefonts/accepted-mscorefonts-eula select true | debconf-set-selections"
|
||||||
|
- sudo apt update && sudo apt install -y calibre msttcorefonts
|
||||||
|
- npm install gitbook-cli -g
|
||||||
|
- npm install markdownlint-cli -g
|
||||||
|
- npm install svgexport -g
|
||||||
|
- gitbook -V
|
||||||
|
- markdownlint -V
|
||||||
|
script:
|
||||||
|
- markdownlint docs
|
||||||
|
- ./check_assets_size.py
|
||||||
|
- ./check_unused_assets.py
|
||||||
|
- gitbook install
|
||||||
|
- gitbook build
|
||||||
|
- gitbook pdf ./ _book/clover.pdf
|
||||||
|
deploy:
|
||||||
|
provider: pages
|
||||||
|
local_dir: _book
|
||||||
|
skip_cleanup: true
|
||||||
|
token: ${GITHUB_OAUTH_TOKEN}
|
||||||
|
keep_history: true
|
||||||
|
target_branch: master
|
||||||
|
repo: CopterExpress/clover.coex.tech
|
||||||
|
fqdn: clover.coex.tech
|
||||||
|
verbose: true
|
||||||
|
on:
|
||||||
|
branch: master
|
||||||
|
- stage: Build
|
||||||
|
name: Editorconfig-lint
|
||||||
|
language: generic
|
||||||
|
before_script:
|
||||||
|
- wget https://github.com/okalachev/editorconfig-checker/releases/download/1.2.1-disable-spaces-amount/ec-linux-amd64
|
||||||
|
- chmod +x ec-linux-amd64
|
||||||
|
script:
|
||||||
|
- ./ec-linux-amd64 -spaces-after-tabs -e "roslib.js|ros3d.js|eventemitter2.js|draw.cpp|BinUtils.swift|\.idea|apps/android/app|blockly/|clover_blocks/programs/|highlight/|python.js|Assets.xcassets|test_parser_pass.txt|test_node_failure.txt|aruco_pose/vendor|\.stl|\.dxf|\.dae"
|
||||||
|
stages:
|
||||||
|
- Build
|
||||||
|
# More info there
|
||||||
|
# https://github.com/travis-ci/travis-ci/issues/6893
|
||||||
|
# https://docs.travis-ci.com/user/customizing-the-build/
|
||||||
|
# https://docs.travis-ci.com/user/deployment/releases
|
||||||
|
# https://docs.travis-ci.com/user/environment-variables/
|
||||||
11
README.md
11
README.md
@@ -1,6 +1,6 @@
|
|||||||
# clover🍀: create autonomous drones easily
|
# clover🍀: create autonomous drones easily
|
||||||
|
|
||||||
<img src="docs/assets/clover42-main-margin.png" align="right" width="400px" alt="COEX Clover Drone">
|
<img src="docs/assets/clover42-main.png" align="right" width="400px" alt="COEX Clover Drone">
|
||||||
|
|
||||||
Clover is an open source [ROS](https://www.ros.org)-based framework, providing user-friendly tools to control [PX4](https://px4.io)-powered drones. Clover is available as a ROS package, but is shipped mainly as a preconfigured image for Raspberry Pi. Once you've installed Raspberry Pi on your drone and flashed the image to its microSD card, taking the drone up in the air is a matter of minutes.
|
Clover is an open source [ROS](https://www.ros.org)-based framework, providing user-friendly tools to control [PX4](https://px4.io)-powered drones. Clover is available as a ROS package, but is shipped mainly as a preconfigured image for Raspberry Pi. Once you've installed Raspberry Pi on your drone and flashed the image to its microSD card, taking the drone up in the air is a matter of minutes.
|
||||||
|
|
||||||
@@ -20,13 +20,12 @@ Clover drone is used on a wide range of educational events, including [Copter Ha
|
|||||||
|
|
||||||
Preconfigured image for Raspberry Pi with installed and configured software, ready to fly, is available [in the Releases section](https://github.com/CopterExpress/clover/releases).
|
Preconfigured image for Raspberry Pi with installed and configured software, ready to fly, is available [in the Releases section](https://github.com/CopterExpress/clover/releases).
|
||||||
|
|
||||||

|
[](https://travis-ci.org/CopterExpress/clover)
|
||||||

|
|
||||||
|
|
||||||
Image features:
|
Image features:
|
||||||
|
|
||||||
* Raspbian Buster
|
* Raspbian Buster
|
||||||
* [ROS Noetic](http://wiki.ros.org/noetic)
|
* [ROS Melodic](http://wiki.ros.org/melodic)
|
||||||
* Configured networking
|
* Configured networking
|
||||||
* OpenCV
|
* OpenCV
|
||||||
* [`mavros`](http://wiki.ros.org/mavros)
|
* [`mavros`](http://wiki.ros.org/mavros)
|
||||||
@@ -38,10 +37,6 @@ API description for autonomous flights is available [on GitBook](https://clover.
|
|||||||
|
|
||||||
For manual package installation and running see [`clover` package documentation](clover/README.md).
|
For manual package installation and running see [`clover` package documentation](clover/README.md).
|
||||||
|
|
||||||
## Support
|
|
||||||
|
|
||||||
[](https://t.me/COEXHelpdesk)
|
|
||||||
|
|
||||||
## License
|
## License
|
||||||
|
|
||||||
While the Clover platform source code is available under the MIT License, note, that the [documentation](docs/) is licensed under the Creative Commons Attribution-NonCommercial-ShareAlike 4.0 International License.
|
While the Clover platform source code is available under the MIT License, note, that the [documentation](docs/) is licensed under the Creative Commons Attribution-NonCommercial-ShareAlike 4.0 International License.
|
||||||
|
|||||||
@@ -1,4 +1,5 @@
|
|||||||
# iOS-приложение для управления Клевером
|
iOS-приложение для управления Клевером
|
||||||
|
--------------------------------------
|
||||||
|
|
||||||
Для установки зависимостей необходим [CocoaPods](https://cocoapods.org):
|
Для установки зависимостей необходим [CocoaPods](https://cocoapods.org):
|
||||||
|
|
||||||
|
|||||||
@@ -22,21 +22,13 @@ find_package(catkin REQUIRED COMPONENTS
|
|||||||
dynamic_reconfigure
|
dynamic_reconfigure
|
||||||
)
|
)
|
||||||
|
|
||||||
# Workaround for OpenCV 3/4 support
|
find_package(OpenCV 3 REQUIRED COMPONENTS core imgproc calib3d)
|
||||||
set(_opencv_version 4)
|
|
||||||
find_package(OpenCV ${_opencv_version} QUIET COMPONENTS core imgproc calib3d)
|
|
||||||
if (NOT OpenCV_FOUND)
|
|
||||||
message(STATUS "Did not find OpenCV 4, searching for OpenCV 3")
|
|
||||||
set(_opencv_version 3)
|
|
||||||
endif()
|
|
||||||
|
|
||||||
find_package(OpenCV ${_opencv_version} REQUIRED COMPONENTS core imgproc calib3d)
|
|
||||||
if ("${OpenCV_VERSION_MINOR}" LESS "9")
|
if ("${OpenCV_VERSION_MINOR}" LESS "9")
|
||||||
message(STATUS "OpenCV version too low, using vendored ArUco package")
|
message(STATUS "OpenCV version too low, using vendored ArUco package")
|
||||||
include(vendor/VendorOpenCV.cmake)
|
include(vendor/VendorOpenCV.cmake)
|
||||||
else()
|
else()
|
||||||
message(STATUS "Using system OpenCV ArUco package")
|
message(STATUS "Using system OpenCV ArUco package")
|
||||||
find_package(OpenCV ${_opencv_version} REQUIRED COMPONENTS aruco)
|
find_package(OpenCV 3 REQUIRED COMPONENTS aruco)
|
||||||
endif()
|
endif()
|
||||||
message(STATUS "OpenCV include dirs: ${OpenCV_INCLUDE_DIRS}")
|
message(STATUS "OpenCV include dirs: ${OpenCV_INCLUDE_DIRS}")
|
||||||
message(STATUS "OpenCV libraries: ${OpenCV_LIBRARIES}")
|
message(STATUS "OpenCV libraries: ${OpenCV_LIBRARIES}")
|
||||||
@@ -83,10 +75,11 @@ add_message_files(
|
|||||||
)
|
)
|
||||||
|
|
||||||
## Generate services in the 'srv' folder
|
## Generate services in the 'srv' folder
|
||||||
add_service_files(
|
# add_service_files(
|
||||||
FILES
|
# FILES
|
||||||
SetMarkers.srv
|
# Service1.srv
|
||||||
)
|
# Service2.srv
|
||||||
|
# )
|
||||||
|
|
||||||
## Generate actions in the 'action' folder
|
## Generate actions in the 'action' folder
|
||||||
# add_action_files(
|
# add_action_files(
|
||||||
@@ -118,8 +111,7 @@ generate_messages(
|
|||||||
|
|
||||||
## Generate dynamic reconfigure parameters in the 'cfg' folder
|
## Generate dynamic reconfigure parameters in the 'cfg' folder
|
||||||
generate_dynamic_reconfigure_options(
|
generate_dynamic_reconfigure_options(
|
||||||
cfg/Detector.cfg
|
cfg/DetectorParams.cfg
|
||||||
cfg/Map.cfg
|
|
||||||
)
|
)
|
||||||
|
|
||||||
###################################
|
###################################
|
||||||
@@ -180,13 +172,6 @@ target_link_libraries(aruco_pose
|
|||||||
${OpenCV_LIBRARIES}
|
${OpenCV_LIBRARIES}
|
||||||
)
|
)
|
||||||
|
|
||||||
# Prevent aruco_pose from having undefined symbols
|
|
||||||
set_property(TARGET aruco_pose
|
|
||||||
APPEND
|
|
||||||
PROPERTY LINK_FLAGS
|
|
||||||
-Wl,--no-undefined
|
|
||||||
)
|
|
||||||
|
|
||||||
#############
|
#############
|
||||||
## Install ##
|
## Install ##
|
||||||
#############
|
#############
|
||||||
@@ -202,11 +187,11 @@ set_property(TARGET aruco_pose
|
|||||||
# )
|
# )
|
||||||
|
|
||||||
## Mark executables and/or libraries for installation
|
## Mark executables and/or libraries for installation
|
||||||
install(TARGETS ${PROJECT_NAME}
|
# install(TARGETS ${PROJECT_NAME} ${PROJECT_NAME}_node
|
||||||
ARCHIVE DESTINATION ${CATKIN_PACKAGE_LIB_DESTINATION}
|
# ARCHIVE DESTINATION ${CATKIN_PACKAGE_LIB_DESTINATION}
|
||||||
LIBRARY DESTINATION ${CATKIN_PACKAGE_LIB_DESTINATION}
|
# LIBRARY DESTINATION ${CATKIN_PACKAGE_LIB_DESTINATION}
|
||||||
RUNTIME DESTINATION ${CATKIN_PACKAGE_BIN_DESTINATION}
|
# RUNTIME DESTINATION ${CATKIN_PACKAGE_BIN_DESTINATION}
|
||||||
)
|
# )
|
||||||
|
|
||||||
## Mark cpp header files for installation
|
## Mark cpp header files for installation
|
||||||
# install(DIRECTORY include/${PROJECT_NAME}/
|
# install(DIRECTORY include/${PROJECT_NAME}/
|
||||||
@@ -222,14 +207,6 @@ install(TARGETS ${PROJECT_NAME}
|
|||||||
# DESTINATION ${CATKIN_PACKAGE_SHARE_DESTINATION}
|
# DESTINATION ${CATKIN_PACKAGE_SHARE_DESTINATION}
|
||||||
# )
|
# )
|
||||||
|
|
||||||
catkin_install_python(PROGRAMS src/genmap.py
|
|
||||||
DESTINATION ${CATKIN_PACKAGE_BIN_DESTINATION}
|
|
||||||
)
|
|
||||||
|
|
||||||
install(DIRECTORY launch DESTINATION ${CATKIN_PACKAGE_SHARE_DESTINATION})
|
|
||||||
|
|
||||||
install(DIRECTORY map DESTINATION ${CATKIN_PACKAGE_SHARE_DESTINATION})
|
|
||||||
|
|
||||||
#############
|
#############
|
||||||
## Testing ##
|
## Testing ##
|
||||||
#############
|
#############
|
||||||
@@ -251,5 +228,4 @@ if (CATKIN_ENABLE_TESTING)
|
|||||||
add_rostest(test/test_node_failure.test)
|
add_rostest(test/test_node_failure.test)
|
||||||
add_rostest(test/largemap.test)
|
add_rostest(test/largemap.test)
|
||||||
add_rostest(test/crash_opencv.test)
|
add_rostest(test/crash_opencv.test)
|
||||||
add_rostest(test/duplicate.test)
|
|
||||||
endif()
|
endif()
|
||||||
|
|||||||
@@ -43,8 +43,7 @@ It's recommended to run it within the same nodelet manager with the camera nodel
|
|||||||
* `~frame_id_prefix` (*string*) – prefix for TF transforms names, marker's ID is appended (default: `aruco_`)
|
* `~frame_id_prefix` (*string*) – prefix for TF transforms names, marker's ID is appended (default: `aruco_`)
|
||||||
* `~length` (*double*) – markers' sides length
|
* `~length` (*double*) – markers' sides length
|
||||||
* `~length_override` (*map*) – lengths of markers with specified ids
|
* `~length_override` (*map*) – lengths of markers with specified ids
|
||||||
* `~known_vertical` (*string*) – known vertical (Z axis) of all the markers as a frame
|
* `~known_tilt` (*string*) – known tilt (pitch and roll) of all the markers as a frame
|
||||||
* `~flip_vertical` – flip vertical vector
|
|
||||||
|
|
||||||
### Topics
|
### Topics
|
||||||
|
|
||||||
@@ -52,7 +51,6 @@ It's recommended to run it within the same nodelet manager with the camera nodel
|
|||||||
|
|
||||||
* `image_raw` (*sensor_msgs/Image*) – camera image
|
* `image_raw` (*sensor_msgs/Image*) – camera image
|
||||||
* `camera_info` (*sensor_msgs/CameraInfo*) – camera calibration info
|
* `camera_info` (*sensor_msgs/CameraInfo*) – camera calibration info
|
||||||
* `map_markers` (*aruco_pose/MarkerArray*) – list of markers to disable TF transform publishing
|
|
||||||
|
|
||||||
#### Published
|
#### Published
|
||||||
|
|
||||||
@@ -72,12 +70,10 @@ It's recommended to run it within the same nodelet manager with the camera nodel
|
|||||||
|
|
||||||
* `~map` – path to text file with markers list
|
* `~map` – path to text file with markers list
|
||||||
* `~frame_id` – published frame id (default: `aruco_map`)
|
* `~frame_id` – published frame id (default: `aruco_map`)
|
||||||
* `~known_vertical` – known vertical (Z axis) of markers map as a frame
|
* `~known_tilt` – debug image width
|
||||||
* `~flip_vertical` – flip vertical vector
|
|
||||||
* `~image_width` – debug image width (default: 2000)
|
* `~image_width` – debug image width (default: 2000)
|
||||||
* `~image_height` – debug image height (default: 2000)
|
* `~image_height` – debug image height (default: 2000)
|
||||||
* `~image_margin` – debug image margin (default: 200)
|
* `~image_margin` – debug image margin (default: 200)
|
||||||
* `~image_axis` – whether debug image should contain axis (default: true)
|
|
||||||
* `~dictionary` (*int*) – ArUco dictionary (default: 2) - should be the same as `dictionary` parameter of `aruco_detect` nodelet
|
* `~dictionary` (*int*) – ArUco dictionary (default: 2) - should be the same as `dictionary` parameter of `aruco_detect` nodelet
|
||||||
|
|
||||||
Map file has one marker per line with the following line format:
|
Map file has one marker per line with the following line format:
|
||||||
@@ -101,7 +97,6 @@ See examples in [`map`](map/) directory.
|
|||||||
#### Published
|
#### Published
|
||||||
|
|
||||||
* `~pose` (*geometry_msgs/PoseWithCovarianceStamped*) – estimated map pose
|
* `~pose` (*geometry_msgs/PoseWithCovarianceStamped*) – estimated map pose
|
||||||
* `~map` (*aruco_pose/MarkerArray*) – list of markers in the loaded map
|
|
||||||
* `~image` (*sensor_msgs/Image*) – planarized map image
|
* `~image` (*sensor_msgs/Image*) – planarized map image
|
||||||
* `~visualization` (*visualization_msgs/MarkerArray*) – markers map visualization for rviz
|
* `~visualization` (*visualization_msgs/MarkerArray*) – markers map visualization for rviz
|
||||||
* `~debug` (*sensor_msgs/Image*) – debug image with detected markers and map axis
|
* `~debug` (*sensor_msgs/Image*) – debug image with detected markers and map axis
|
||||||
|
|||||||
@@ -10,8 +10,6 @@ gen = ParameterGenerator()
|
|||||||
|
|
||||||
gen.add("enabled", bool_t, 0, "if detection enabled", True)
|
gen.add("enabled", bool_t, 0, "if detection enabled", True)
|
||||||
|
|
||||||
gen.add("length", double_t, 0, "markers' side length", min=0, max=10)
|
|
||||||
|
|
||||||
gen.add("adaptiveThreshConstant", double_t, 0,
|
gen.add("adaptiveThreshConstant", double_t, 0,
|
||||||
"Constant for adaptive thresholding before finding contours",
|
"Constant for adaptive thresholding before finding contours",
|
||||||
p.adaptiveThreshConstant, 0, 100)
|
p.adaptiveThreshConstant, 0, 100)
|
||||||
@@ -1,14 +0,0 @@
|
|||||||
#!/usr/bin/env python
|
|
||||||
PACKAGE = "aruco_pose"
|
|
||||||
|
|
||||||
from dynamic_reconfigure.parameter_generator_catkin import *
|
|
||||||
|
|
||||||
gen = ParameterGenerator()
|
|
||||||
|
|
||||||
gen.add("enabled", bool_t, 0, "if map detection enabled", True)
|
|
||||||
|
|
||||||
gen.add("map", str_t, 0, "full path for the map file")
|
|
||||||
|
|
||||||
gen.add("image_axis", bool_t, 0, "debug image axis", default=True)
|
|
||||||
|
|
||||||
exit(gen.generate(PACKAGE, "aruco_pose", "Map"))
|
|
||||||
@@ -1,4 +1,3 @@
|
|||||||
# id length x y z rot_z rot_y rot_x
|
|
||||||
0 0.33 0.0 9.0 0 0 0 0
|
0 0.33 0.0 9.0 0 0 0 0
|
||||||
1 0.33 1.0 9.0 0 0 0 0
|
1 0.33 1.0 9.0 0 0 0 0
|
||||||
2 0.33 2.0 9.0 0 0 0 0
|
2 0.33 2.0 9.0 0 0 0 0
|
||||||
|
|||||||
@@ -1,4 +1,3 @@
|
|||||||
# id length x y z rot_z rot_y rot_x
|
|
||||||
107 0.33 0 0 0 0 0 0
|
107 0.33 0 0 0 0 0 0
|
||||||
106 0.33 0.77 0 0 0 0 0
|
106 0.33 0.77 0 0 0 0 0
|
||||||
105 0.33 0 0.77 0 0 0 0
|
105 0.33 0 0.77 0 0 0 0
|
||||||
|
|||||||
@@ -1,4 +1,3 @@
|
|||||||
# id length x y z rot_z rot_y rot_x
|
|
||||||
14 0.365 0.000 0.0 0 0 0 0
|
14 0.365 0.000 0.0 0 0 0 0
|
||||||
15 0.365 1.335 0.0 0 0 0 0
|
15 0.365 1.335 0.0 0 0 0 0
|
||||||
30 0.365 2.865 0.0 0 0 0 0
|
30 0.365 2.865 0.0 0 0 0 0
|
||||||
|
|||||||
@@ -1,7 +1,7 @@
|
|||||||
<?xml version="1.0"?>
|
<?xml version="1.0"?>
|
||||||
<package format="3">
|
<package format="2">
|
||||||
<name>aruco_pose</name>
|
<name>aruco_pose</name>
|
||||||
<version>0.23.0</version>
|
<version>0.21.1</version>
|
||||||
<description>Positioning with ArUco markers</description>
|
<description>Positioning with ArUco markers</description>
|
||||||
|
|
||||||
<maintainer email="okalachev@gmail.com">Oleg Kalachev</maintainer>
|
<maintainer email="okalachev@gmail.com">Oleg Kalachev</maintainer>
|
||||||
@@ -28,8 +28,6 @@
|
|||||||
<depend>sensor_msgs</depend>
|
<depend>sensor_msgs</depend>
|
||||||
<depend>rostest</depend>
|
<depend>rostest</depend>
|
||||||
<depend>dynamic_reconfigure</depend>
|
<depend>dynamic_reconfigure</depend>
|
||||||
<depend condition="$ROS_PYTHON_VERSION == 2">python-docopt</depend>
|
|
||||||
<depend condition="$ROS_PYTHON_VERSION == 3">python3-docopt</depend>
|
|
||||||
|
|
||||||
<test_depend>image_publisher</test_depend>
|
<test_depend>image_publisher</test_depend>
|
||||||
<test_depend>ros_pytest</test_depend>
|
<test_depend>ros_pytest</test_depend>
|
||||||
|
|||||||
@@ -48,7 +48,6 @@
|
|||||||
#include <aruco_pose/Marker.h>
|
#include <aruco_pose/Marker.h>
|
||||||
#include <aruco_pose/MarkerArray.h>
|
#include <aruco_pose/MarkerArray.h>
|
||||||
#include <aruco_pose/DetectorConfig.h>
|
#include <aruco_pose/DetectorConfig.h>
|
||||||
#include <aruco_pose/SetMarkers.h>
|
|
||||||
|
|
||||||
#include "utils.h"
|
#include "utils.h"
|
||||||
#include <memory>
|
#include <memory>
|
||||||
@@ -70,13 +69,10 @@ private:
|
|||||||
image_transport::CameraSubscriber img_sub_;
|
image_transport::CameraSubscriber img_sub_;
|
||||||
ros::Publisher markers_pub_, vis_markers_pub_;
|
ros::Publisher markers_pub_, vis_markers_pub_;
|
||||||
ros::Subscriber map_markers_sub_;
|
ros::Subscriber map_markers_sub_;
|
||||||
ros::ServiceServer set_markers_srv_;
|
bool estimate_poses_, send_tf_, auto_flip_;
|
||||||
bool estimate_poses_, send_tf_, flip_vertical_, auto_flip_, use_map_markers_;
|
|
||||||
bool waiting_for_map_;
|
|
||||||
double length_;
|
double length_;
|
||||||
ros::Duration transform_timeout_;
|
|
||||||
std::unordered_map<int, double> length_override_;
|
std::unordered_map<int, double> length_override_;
|
||||||
std::string frame_id_prefix_, known_vertical_;
|
std::string frame_id_prefix_, known_tilt_;
|
||||||
Mat camera_matrix_, dist_coeffs_;
|
Mat camera_matrix_, dist_coeffs_;
|
||||||
aruco_pose::MarkerArray array_;
|
aruco_pose::MarkerArray array_;
|
||||||
std::unordered_set<int> map_markers_ids_;
|
std::unordered_set<int> map_markers_ids_;
|
||||||
@@ -96,17 +92,13 @@ public:
|
|||||||
dictionary = nh_priv_.param("dictionary", 2);
|
dictionary = nh_priv_.param("dictionary", 2);
|
||||||
estimate_poses_ = nh_priv_.param("estimate_poses", true);
|
estimate_poses_ = nh_priv_.param("estimate_poses", true);
|
||||||
send_tf_ = nh_priv_.param("send_tf", true);
|
send_tf_ = nh_priv_.param("send_tf", true);
|
||||||
use_map_markers_ = nh_priv_.param("use_map_markers", false);
|
|
||||||
waiting_for_map_ = use_map_markers_;
|
|
||||||
if (estimate_poses_ && !nh_priv_.getParam("length", length_)) {
|
if (estimate_poses_ && !nh_priv_.getParam("length", length_)) {
|
||||||
NODELET_FATAL("can't estimate marker's poses as ~length parameter is not defined");
|
NODELET_FATAL("can't estimate marker's poses as ~length parameter is not defined");
|
||||||
ros::shutdown();
|
ros::shutdown();
|
||||||
}
|
}
|
||||||
readLengthOverride(nh_priv_);
|
readLengthOverride(nh_priv_);
|
||||||
transform_timeout_ = ros::Duration(nh_priv_.param("transform_timeout", 0.02));
|
|
||||||
|
|
||||||
known_vertical_ = nh_priv_.param("known_vertical", nh_priv_.param("known_tilt", std::string(""))); // known_tilt is an old name
|
known_tilt_ = nh_priv_.param<std::string>("known_tilt", "");
|
||||||
flip_vertical_ = nh_priv_.param<bool>("flip_vertical", false);
|
|
||||||
auto_flip_ = nh_priv_.param("auto_flip", false);
|
auto_flip_ = nh_priv_.param("auto_flip", false);
|
||||||
|
|
||||||
frame_id_prefix_ = nh_priv_.param<std::string>("frame_id_prefix", "aruco_");
|
frame_id_prefix_ = nh_priv_.param<std::string>("frame_id_prefix", "aruco_");
|
||||||
@@ -119,16 +111,17 @@ public:
|
|||||||
image_transport::ImageTransport it(nh_);
|
image_transport::ImageTransport it(nh_);
|
||||||
image_transport::ImageTransport it_priv(nh_priv_);
|
image_transport::ImageTransport it_priv(nh_priv_);
|
||||||
|
|
||||||
dyn_srv_ = std::make_shared<dynamic_reconfigure::Server<aruco_pose::DetectorConfig>>(nh_priv_);
|
map_markers_sub_ = nh_.subscribe("map_markers", 1, &ArucoDetect::mapMarkersCallback, this);
|
||||||
dyn_srv_->setCallback(std::bind(&ArucoDetect::paramCallback, this, std::placeholders::_1, std::placeholders::_2));
|
|
||||||
|
|
||||||
set_markers_srv_ = nh_priv_.advertiseService("set_length_override", &ArucoDetect::setMarkers, this);
|
|
||||||
|
|
||||||
debug_pub_ = it_priv.advertise("debug", 1);
|
debug_pub_ = it_priv.advertise("debug", 1);
|
||||||
markers_pub_ = nh_priv_.advertise<aruco_pose::MarkerArray>("markers", 1);
|
markers_pub_ = nh_priv_.advertise<aruco_pose::MarkerArray>("markers", 1);
|
||||||
vis_markers_pub_ = nh_priv_.advertise<visualization_msgs::MarkerArray>("visualization", 1);
|
vis_markers_pub_ = nh_priv_.advertise<visualization_msgs::MarkerArray>("visualization", 1);
|
||||||
img_sub_ = it.subscribeCamera("image_raw", 1, &ArucoDetect::imageCallback, this);
|
img_sub_ = it.subscribeCamera("image_raw", 1, &ArucoDetect::imageCallback, this);
|
||||||
map_markers_sub_ = nh_.subscribe("map_markers", 1, &ArucoDetect::mapMarkersCallback, this);
|
|
||||||
|
dyn_srv_ = std::make_shared<dynamic_reconfigure::Server<aruco_pose::DetectorConfig>>(nh_priv_);
|
||||||
|
dynamic_reconfigure::Server<aruco_pose::DetectorConfig>::CallbackType cb;
|
||||||
|
|
||||||
|
cb = std::bind(&ArucoDetect::paramCallback, this, std::placeholders::_1, std::placeholders::_2);
|
||||||
|
dyn_srv_->setCallback(cb);
|
||||||
|
|
||||||
NODELET_INFO("ready");
|
NODELET_INFO("ready");
|
||||||
}
|
}
|
||||||
@@ -137,7 +130,6 @@ private:
|
|||||||
void imageCallback(const sensor_msgs::ImageConstPtr& msg, const sensor_msgs::CameraInfoConstPtr &cinfo)
|
void imageCallback(const sensor_msgs::ImageConstPtr& msg, const sensor_msgs::CameraInfoConstPtr &cinfo)
|
||||||
{
|
{
|
||||||
if (!enabled_) return;
|
if (!enabled_) return;
|
||||||
if (waiting_for_map_) return;
|
|
||||||
|
|
||||||
Mat image = cv_bridge::toCvShare(msg, "bgr8")->image;
|
Mat image = cv_bridge::toCvShare(msg, "bgr8")->image;
|
||||||
|
|
||||||
@@ -145,7 +137,7 @@ private:
|
|||||||
vector<vector<cv::Point2f>> corners, rejected;
|
vector<vector<cv::Point2f>> corners, rejected;
|
||||||
vector<cv::Vec3d> rvecs, tvecs;
|
vector<cv::Vec3d> rvecs, tvecs;
|
||||||
vector<cv::Point3f> obj_points;
|
vector<cv::Point3f> obj_points;
|
||||||
geometry_msgs::TransformStamped vertical;
|
geometry_msgs::TransformStamped snap_to;
|
||||||
|
|
||||||
// Detect markers
|
// Detect markers
|
||||||
cv::aruco::detectMarkers(image, dictionary_, corners, ids, parameters_, rejected);
|
cv::aruco::detectMarkers(image, dictionary_, corners, ids, parameters_, rejected);
|
||||||
@@ -180,20 +172,18 @@ private:
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
if (!known_vertical_.empty()) {
|
if (!known_tilt_.empty()) {
|
||||||
try {
|
try {
|
||||||
vertical = tf_buffer_->lookupTransform(msg->header.frame_id, known_vertical_,
|
snap_to = tf_buffer_->lookupTransform(msg->header.frame_id, known_tilt_,
|
||||||
msg->header.stamp, transform_timeout_);
|
msg->header.stamp, ros::Duration(0.02));
|
||||||
} catch (const tf2::TransformException& e) {
|
} catch (const tf2::TransformException& e) {
|
||||||
NODELET_WARN_THROTTLE(5, "can't retrieve known vertical: %s", e.what());
|
NODELET_WARN_THROTTLE(5, "can't snap: %s", e.what());
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
array_.markers.reserve(ids.size());
|
array_.markers.reserve(ids.size());
|
||||||
aruco_pose::Marker marker;
|
aruco_pose::Marker marker;
|
||||||
vector<geometry_msgs::TransformStamped> transforms;
|
|
||||||
transforms.reserve(ids.size());
|
|
||||||
geometry_msgs::TransformStamped transform;
|
geometry_msgs::TransformStamped transform;
|
||||||
transform.header.stamp = msg->header.stamp;
|
transform.header.stamp = msg->header.stamp;
|
||||||
transform.header.frame_id = msg->header.frame_id;
|
transform.header.frame_id = msg->header.frame_id;
|
||||||
@@ -206,38 +196,25 @@ private:
|
|||||||
if (estimate_poses_) {
|
if (estimate_poses_) {
|
||||||
fillPose(marker.pose, rvecs[i], tvecs[i]);
|
fillPose(marker.pose, rvecs[i], tvecs[i]);
|
||||||
|
|
||||||
// apply known vertical (if enabled and vertical frame available)
|
// snap orientation (if enabled and snap frame available)
|
||||||
if (!known_vertical_.empty() && !vertical.header.frame_id.empty()) {
|
if (!known_tilt_.empty() && !snap_to.header.frame_id.empty()) {
|
||||||
applyVertical(marker.pose.orientation, vertical.transform.rotation, false, auto_flip_);
|
snapOrientation(marker.pose.orientation, snap_to.transform.rotation, auto_flip_);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// TODO: check IDs are unique
|
||||||
if (send_tf_) {
|
if (send_tf_) {
|
||||||
transform.child_frame_id = getChildFrameId(ids[i]);
|
transform.child_frame_id = getChildFrameId(ids[i]);
|
||||||
|
|
||||||
// check if such static transform is in the map
|
// check if such static transform is in the map
|
||||||
if (map_markers_ids_.find(ids[i]) == map_markers_ids_.end()) {
|
if (map_markers_ids_.find(ids[i]) == map_markers_ids_.end()) {
|
||||||
// check if a markers with that id is already added
|
transform.transform.rotation = marker.pose.orientation;
|
||||||
bool send = true;
|
fillTranslation(transform.transform.translation, tvecs[i]);
|
||||||
for (auto &t : transforms) {
|
br_->sendTransform(transform);
|
||||||
if (t.child_frame_id == transform.child_frame_id) {
|
|
||||||
send = false;
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if (send) {
|
|
||||||
transform.transform.rotation = marker.pose.orientation;
|
|
||||||
fillTranslation(transform.transform.translation, tvecs[i]);
|
|
||||||
transforms.push_back(transform);
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
array_.markers.push_back(marker);
|
array_.markers.push_back(marker);
|
||||||
}
|
}
|
||||||
|
|
||||||
if (send_tf_) {
|
|
||||||
br_->sendTransform(transforms);
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
markers_pub_.publish(array_);
|
markers_pub_.publish(array_);
|
||||||
@@ -372,47 +349,17 @@ private:
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
bool setMarkers(aruco_pose::SetMarkers::Request& req, aruco_pose::SetMarkers::Response& res)
|
|
||||||
{
|
|
||||||
for (auto const& marker : req.markers) {
|
|
||||||
if (marker.id > 999) {
|
|
||||||
res.message = "Invalid marker id: " + std::to_string(marker.id);
|
|
||||||
ROS_ERROR("%s", res.message.c_str());
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
if (!std::isfinite(marker.length) || marker.length <= 0) {
|
|
||||||
res.message = "Invalid marker " + std::to_string(marker.id) + " length: " + std::to_string(marker.length);
|
|
||||||
ROS_ERROR("%s", res.message.c_str());
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
for (auto const& marker : req.markers) {
|
|
||||||
length_override_[marker.id] = marker.length;
|
|
||||||
}
|
|
||||||
|
|
||||||
res.success = true;
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
|
|
||||||
void mapMarkersCallback(const aruco_pose::MarkerArray& msg)
|
void mapMarkersCallback(const aruco_pose::MarkerArray& msg)
|
||||||
{
|
{
|
||||||
map_markers_ids_.clear();
|
map_markers_ids_.clear();
|
||||||
for (auto const& marker : msg.markers) {
|
for (auto const& marker : msg.markers) {
|
||||||
map_markers_ids_.insert(marker.id);
|
map_markers_ids_.insert(marker.id);
|
||||||
if (use_map_markers_) {
|
|
||||||
if (length_override_.find(marker.id) == length_override_.end()) {
|
|
||||||
length_override_[marker.id] = marker.length;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
waiting_for_map_ = false;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
void paramCallback(aruco_pose::DetectorConfig &config, uint32_t level)
|
void paramCallback(aruco_pose::DetectorConfig &config, uint32_t level)
|
||||||
{
|
{
|
||||||
enabled_ = config.enabled && config.length > 0;
|
enabled_ = config.enabled;
|
||||||
length_ = config.length;
|
|
||||||
parameters_->adaptiveThreshConstant = config.adaptiveThreshConstant;
|
parameters_->adaptiveThreshConstant = config.adaptiveThreshConstant;
|
||||||
parameters_->adaptiveThreshWinSizeMin = config.adaptiveThreshWinSizeMin;
|
parameters_->adaptiveThreshWinSizeMin = config.adaptiveThreshWinSizeMin;
|
||||||
parameters_->adaptiveThreshWinSizeMax = config.adaptiveThreshWinSizeMax;
|
parameters_->adaptiveThreshWinSizeMax = config.adaptiveThreshWinSizeMax;
|
||||||
|
|||||||
@@ -19,13 +19,11 @@
|
|||||||
#include <vector>
|
#include <vector>
|
||||||
#include <fstream>
|
#include <fstream>
|
||||||
#include <algorithm>
|
#include <algorithm>
|
||||||
#include <memory>
|
|
||||||
#include <ros/ros.h>
|
#include <ros/ros.h>
|
||||||
#include <nodelet/nodelet.h>
|
#include <nodelet/nodelet.h>
|
||||||
#include <pluginlib/class_list_macros.h>
|
#include <pluginlib/class_list_macros.h>
|
||||||
#include <image_transport/image_transport.h>
|
#include <image_transport/image_transport.h>
|
||||||
#include <cv_bridge/cv_bridge.h>
|
#include <cv_bridge/cv_bridge.h>
|
||||||
#include <dynamic_reconfigure/server.h>
|
|
||||||
#include <tf/transform_datatypes.h>
|
#include <tf/transform_datatypes.h>
|
||||||
#include <tf2_ros/buffer.h>
|
#include <tf2_ros/buffer.h>
|
||||||
#include <tf2_ros/transform_listener.h>
|
#include <tf2_ros/transform_listener.h>
|
||||||
@@ -43,7 +41,6 @@
|
|||||||
|
|
||||||
#include <aruco_pose/MarkerArray.h>
|
#include <aruco_pose/MarkerArray.h>
|
||||||
#include <aruco_pose/Marker.h>
|
#include <aruco_pose/Marker.h>
|
||||||
#include <aruco_pose/MapConfig.h>
|
|
||||||
|
|
||||||
#include <opencv2/opencv.hpp>
|
#include <opencv2/opencv.hpp>
|
||||||
#include <opencv2/aruco.hpp>
|
#include <opencv2/aruco.hpp>
|
||||||
@@ -77,13 +74,10 @@ private:
|
|||||||
tf2_ros::StaticTransformBroadcaster static_br_;
|
tf2_ros::StaticTransformBroadcaster static_br_;
|
||||||
tf2_ros::Buffer tf_buffer_;
|
tf2_ros::Buffer tf_buffer_;
|
||||||
tf2_ros::TransformListener tf_listener_{tf_buffer_};
|
tf2_ros::TransformListener tf_listener_{tf_buffer_};
|
||||||
std::shared_ptr<dynamic_reconfigure::Server<aruco_pose::MapConfig>> dyn_srv_;
|
|
||||||
bool enabled_ = true;
|
|
||||||
std::string type_;
|
|
||||||
visualization_msgs::MarkerArray vis_array_;
|
visualization_msgs::MarkerArray vis_array_;
|
||||||
std::string known_vertical_, map_, markers_frame_, markers_parent_frame_;
|
std::string known_tilt_, map_, markers_frame_, markers_parent_frame_;
|
||||||
int image_width_, image_height_, image_margin_;
|
int image_width_, image_height_, image_margin_;
|
||||||
bool flip_vertical_, auto_flip_, image_axis_;
|
bool auto_flip_, image_axis_;
|
||||||
|
|
||||||
public:
|
public:
|
||||||
virtual void onInit()
|
virtual void onInit()
|
||||||
@@ -95,17 +89,17 @@ public:
|
|||||||
|
|
||||||
// TODO: why image_transport doesn't work here?
|
// TODO: why image_transport doesn't work here?
|
||||||
img_pub_ = nh_priv_.advertise<sensor_msgs::Image>("image", 1, true);
|
img_pub_ = nh_priv_.advertise<sensor_msgs::Image>("image", 1, true);
|
||||||
markers_pub_ = nh_priv_.advertise<aruco_pose::MarkerArray>("map", 1, true);
|
markers_pub_ = nh_priv_.advertise<aruco_pose::MarkerArray>("markers", 1, true);
|
||||||
|
|
||||||
board_ = cv::makePtr<cv::aruco::Board>();
|
board_ = cv::makePtr<cv::aruco::Board>();
|
||||||
board_->dictionary = cv::aruco::getPredefinedDictionary(
|
board_->dictionary = cv::aruco::getPredefinedDictionary(
|
||||||
static_cast<cv::aruco::PREDEFINED_DICTIONARY_NAME>(nh_priv_.param("dictionary", 2)));
|
static_cast<cv::aruco::PREDEFINED_DICTIONARY_NAME>(nh_priv_.param("dictionary", 2)));
|
||||||
camera_matrix_ = cv::Mat::zeros(3, 3, CV_64F);
|
camera_matrix_ = cv::Mat::zeros(3, 3, CV_64F);
|
||||||
|
|
||||||
type_ = nh_priv_.param<std::string>("type", "map");
|
std::string type, map;
|
||||||
|
type = nh_priv_.param<std::string>("type", "map");
|
||||||
transform_.child_frame_id = nh_priv_.param<std::string>("frame_id", "aruco_map");
|
transform_.child_frame_id = nh_priv_.param<std::string>("frame_id", "aruco_map");
|
||||||
known_vertical_ = nh_priv_.param("known_vertical", nh_priv_.param("known_tilt", std::string(""))); // known_tilt is an old name
|
known_tilt_ = nh_priv_.param<std::string>("known_tilt", "");
|
||||||
flip_vertical_ = nh_priv_.param<bool>("flip_vertical", false);
|
|
||||||
auto_flip_ = nh_priv_.param("auto_flip", false);
|
auto_flip_ = nh_priv_.param("auto_flip", false);
|
||||||
image_width_ = nh_priv_.param("image_width" , 2000);
|
image_width_ = nh_priv_.param("image_width" , 2000);
|
||||||
image_height_ = nh_priv_.param("image_height", 2000);
|
image_height_ = nh_priv_.param("image_height", 2000);
|
||||||
@@ -116,13 +110,13 @@ public:
|
|||||||
|
|
||||||
// createStripLine();
|
// createStripLine();
|
||||||
|
|
||||||
if (type_ == "map") {
|
if (type == "map") {
|
||||||
map_ = nh_priv_.param<std::string>("map" , "");
|
param(nh_priv_, "map", map);
|
||||||
loadMap(map_);
|
loadMap(map);
|
||||||
} else if (type_ == "gridboard") {
|
} else if (type == "gridboard") {
|
||||||
createGridBoard(nh_priv_);
|
createGridBoard(nh_priv_);
|
||||||
} else {
|
} else {
|
||||||
NODELET_FATAL("unknown type: %s", type_.c_str());
|
NODELET_FATAL("unknown type: %s", type.c_str());
|
||||||
ros::shutdown();
|
ros::shutdown();
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -130,8 +124,6 @@ public:
|
|||||||
vis_markers_pub_ = nh_priv_.advertise<visualization_msgs::MarkerArray>("visualization", 1, true);
|
vis_markers_pub_ = nh_priv_.advertise<visualization_msgs::MarkerArray>("visualization", 1, true);
|
||||||
debug_pub_ = it_priv.advertise("debug", 1);
|
debug_pub_ = it_priv.advertise("debug", 1);
|
||||||
|
|
||||||
publishMap();
|
|
||||||
|
|
||||||
image_sub_.subscribe(nh_, "image_raw", 1);
|
image_sub_.subscribe(nh_, "image_raw", 1);
|
||||||
info_sub_.subscribe(nh_, "camera_info", 1);
|
info_sub_.subscribe(nh_, "camera_info", 1);
|
||||||
markers_sub_.subscribe(nh_, "markers", 1);
|
markers_sub_.subscribe(nh_, "markers", 1);
|
||||||
@@ -139,11 +131,10 @@ public:
|
|||||||
sync_.reset(new message_filters::Synchronizer<SyncPolicy>(SyncPolicy(10), image_sub_, info_sub_, markers_sub_));
|
sync_.reset(new message_filters::Synchronizer<SyncPolicy>(SyncPolicy(10), image_sub_, info_sub_, markers_sub_));
|
||||||
sync_->registerCallback(boost::bind(&ArucoMap::callback, this, _1, _2, _3));
|
sync_->registerCallback(boost::bind(&ArucoMap::callback, this, _1, _2, _3));
|
||||||
|
|
||||||
dyn_srv_ = std::make_shared<dynamic_reconfigure::Server<aruco_pose::MapConfig>>(nh_priv_);
|
publishMarkersFrames();
|
||||||
dynamic_reconfigure::Server<aruco_pose::MapConfig>::CallbackType cb;
|
publishMarkers();
|
||||||
|
publishMapImage();
|
||||||
cb = std::bind(&ArucoMap::paramCallback, this, std::placeholders::_1, std::placeholders::_2);
|
vis_markers_pub_.publish(vis_array_);
|
||||||
dyn_srv_->setCallback(cb);
|
|
||||||
|
|
||||||
NODELET_INFO("ready");
|
NODELET_INFO("ready");
|
||||||
}
|
}
|
||||||
@@ -152,9 +143,6 @@ public:
|
|||||||
const sensor_msgs::CameraInfoConstPtr& cinfo,
|
const sensor_msgs::CameraInfoConstPtr& cinfo,
|
||||||
const aruco_pose::MarkerArrayConstPtr& markers)
|
const aruco_pose::MarkerArrayConstPtr& markers)
|
||||||
{
|
{
|
||||||
if (!enabled_) return;
|
|
||||||
if (markers->markers.empty()) return; // map not loaded
|
|
||||||
|
|
||||||
int valid = 0;
|
int valid = 0;
|
||||||
int count = markers->markers.size();
|
int count = markers->markers.size();
|
||||||
std::vector<int> ids;
|
std::vector<int> ids;
|
||||||
@@ -178,7 +166,7 @@ public:
|
|||||||
corners.push_back(marker_corners);
|
corners.push_back(marker_corners);
|
||||||
}
|
}
|
||||||
|
|
||||||
if (known_vertical_.empty()) {
|
if (known_tilt_.empty()) {
|
||||||
// simple estimation
|
// simple estimation
|
||||||
valid = cv::aruco::estimatePoseBoard(corners, ids, board_, camera_matrix_, dist_coeffs_,
|
valid = cv::aruco::estimatePoseBoard(corners, ids, board_, camera_matrix_, dist_coeffs_,
|
||||||
rvec, tvec, false);
|
rvec, tvec, false);
|
||||||
@@ -192,7 +180,7 @@ public:
|
|||||||
|
|
||||||
} else {
|
} else {
|
||||||
Mat obj_points, img_points;
|
Mat obj_points, img_points;
|
||||||
// estimation with known vertical
|
// estimation with "snapping"
|
||||||
cv::aruco::getBoardObjectAndImagePoints(board_, corners, ids, obj_points, img_points);
|
cv::aruco::getBoardObjectAndImagePoints(board_, corners, ids, obj_points, img_points);
|
||||||
if (obj_points.empty()) goto publish_debug;
|
if (obj_points.empty()) goto publish_debug;
|
||||||
|
|
||||||
@@ -204,11 +192,11 @@ public:
|
|||||||
|
|
||||||
fillTransform(transform_.transform, rvec, tvec);
|
fillTransform(transform_.transform, rvec, tvec);
|
||||||
try {
|
try {
|
||||||
geometry_msgs::TransformStamped vertical = tf_buffer_.lookupTransform(markers->header.frame_id,
|
geometry_msgs::TransformStamped snap_to = tf_buffer_.lookupTransform(markers->header.frame_id,
|
||||||
known_vertical_, markers->header.stamp, ros::Duration(0.02));
|
known_tilt_, markers->header.stamp, ros::Duration(0.02));
|
||||||
applyVertical(transform_.transform.rotation, vertical.transform.rotation, flip_vertical_, auto_flip_);
|
snapOrientation(transform_.transform.rotation, snap_to.transform.rotation, auto_flip_);
|
||||||
} catch (const tf2::TransformException& e) {
|
} catch (const tf2::TransformException& e) {
|
||||||
NODELET_WARN_THROTTLE(1, "can't retrieve known vertical: %s", e.what());
|
NODELET_WARN_THROTTLE(1, "can't snap: %s", e.what());
|
||||||
}
|
}
|
||||||
|
|
||||||
geometry_msgs::TransformStamped shift;
|
geometry_msgs::TransformStamped shift;
|
||||||
@@ -280,17 +268,9 @@ publish_debug:
|
|||||||
std::ifstream f(filename);
|
std::ifstream f(filename);
|
||||||
std::string line;
|
std::string line;
|
||||||
|
|
||||||
clearMarkers();
|
|
||||||
|
|
||||||
if (map_ == "") {
|
|
||||||
NODELET_INFO("No map loaded");
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (!f.good()) {
|
if (!f.good()) {
|
||||||
NODELET_ERROR("%s - %s", strerror(errno), filename.c_str());
|
NODELET_FATAL("%s - %s", strerror(errno), filename.c_str());
|
||||||
map_ = "";
|
ros::shutdown();
|
||||||
return;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
while (std::getline(f, line)) {
|
while (std::getline(f, line)) {
|
||||||
@@ -316,10 +296,9 @@ publish_debug:
|
|||||||
s.putback(first);
|
s.putback(first);
|
||||||
} else {
|
} else {
|
||||||
// Probably garbage data; inform user and throw an exception, possibly killing nodelet
|
// Probably garbage data; inform user and throw an exception, possibly killing nodelet
|
||||||
NODELET_ERROR("Malformed input: %s", line.c_str());
|
NODELET_FATAL("Malformed input: %s", line.c_str());
|
||||||
map_ = "";
|
ros::shutdown();
|
||||||
clearMarkers();
|
throw std::runtime_error("Malformed input");
|
||||||
return;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
if (!(s >> id >> length >> x >> y)) {
|
if (!(s >> id >> length >> x >> y)) {
|
||||||
@@ -350,14 +329,6 @@ publish_debug:
|
|||||||
NODELET_INFO("loading %s complete (%d markers)", filename.c_str(), static_cast<int>(board_->ids.size()));
|
NODELET_INFO("loading %s complete (%d markers)", filename.c_str(), static_cast<int>(board_->ids.size()));
|
||||||
}
|
}
|
||||||
|
|
||||||
void publishMap()
|
|
||||||
{
|
|
||||||
publishMarkersFrames();
|
|
||||||
publishMarkers();
|
|
||||||
publishMapImage();
|
|
||||||
vis_markers_pub_.publish(vis_array_);
|
|
||||||
}
|
|
||||||
|
|
||||||
void createGridBoard(ros::NodeHandle& nh)
|
void createGridBoard(ros::NodeHandle& nh)
|
||||||
{
|
{
|
||||||
NODELET_INFO("generate gridboard");
|
NODELET_INFO("generate gridboard");
|
||||||
@@ -399,15 +370,6 @@ publish_debug:
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
void clearMarkers()
|
|
||||||
{
|
|
||||||
board_->ids.clear();
|
|
||||||
board_->objPoints.clear();
|
|
||||||
markers_.markers.clear();
|
|
||||||
vis_array_.markers.clear();
|
|
||||||
markers_transforms_.clear();
|
|
||||||
}
|
|
||||||
|
|
||||||
// void createStripLine()
|
// void createStripLine()
|
||||||
// {
|
// {
|
||||||
// visualization_msgs::Marker marker;
|
// visualization_msgs::Marker marker;
|
||||||
@@ -504,7 +466,7 @@ publish_debug:
|
|||||||
vis_marker.pose.position.x = x;
|
vis_marker.pose.position.x = x;
|
||||||
vis_marker.pose.position.y = y;
|
vis_marker.pose.position.y = y;
|
||||||
vis_marker.pose.position.z = z;
|
vis_marker.pose.position.z = z;
|
||||||
tf::quaternionTFToMsg(q, vis_marker.pose.orientation);
|
tf::quaternionTFToMsg(q, marker.pose.orientation);
|
||||||
vis_marker.frame_locked = true;
|
vis_marker.frame_locked = true;
|
||||||
vis_array_.markers.push_back(vis_marker);
|
vis_array_.markers.push_back(vis_marker);
|
||||||
|
|
||||||
@@ -547,22 +509,6 @@ publish_debug:
|
|||||||
msg.image = image;
|
msg.image = image;
|
||||||
img_pub_.publish(msg.toImageMsg());
|
img_pub_.publish(msg.toImageMsg());
|
||||||
}
|
}
|
||||||
|
|
||||||
void paramCallback(aruco_pose::MapConfig &config, uint32_t level)
|
|
||||||
{
|
|
||||||
// https://github.com/CopterExpress/clover/commit/2cd334c474e3ed04ef65ca1ba7f08ab535a3dc6d#diff-942723f9452c398ae93f1a91427f9a7b614be5e5871f8a3e590f324d804f0d58R356
|
|
||||||
enabled_ = config.enabled;
|
|
||||||
if (type_ == "map" && config.map != map_) {
|
|
||||||
map_ = config.map;
|
|
||||||
loadMap(map_);
|
|
||||||
publishMap();
|
|
||||||
}
|
|
||||||
|
|
||||||
if (config.image_axis != image_axis_) {
|
|
||||||
image_axis_ = config.image_axis;
|
|
||||||
publishMapImage();
|
|
||||||
}
|
|
||||||
}
|
|
||||||
};
|
};
|
||||||
|
|
||||||
PLUGINLIB_EXPORT_CLASS(ArucoMap, nodelet::Nodelet)
|
PLUGINLIB_EXPORT_CLASS(ArucoMap, nodelet::Nodelet)
|
||||||
|
|||||||
@@ -3,11 +3,26 @@
|
|||||||
|
|
||||||
#include "draw.h"
|
#include "draw.h"
|
||||||
#include <math.h>
|
#include <math.h>
|
||||||
#include <vector>
|
|
||||||
|
|
||||||
using namespace cv;
|
using namespace cv;
|
||||||
using namespace cv::aruco;
|
using namespace cv::aruco;
|
||||||
|
|
||||||
|
static void _cvProjectPoints2( const CvMat* object_points, const CvMat* rotation_vector,
|
||||||
|
const CvMat* translation_vector, const CvMat* camera_matrix,
|
||||||
|
const CvMat* distortion_coeffs, CvMat* image_points,
|
||||||
|
CvMat* dpdrot CV_DEFAULT(NULL), CvMat* dpdt CV_DEFAULT(NULL),
|
||||||
|
CvMat* dpdf CV_DEFAULT(NULL), CvMat* dpdc CV_DEFAULT(NULL),
|
||||||
|
CvMat* dpddist CV_DEFAULT(NULL),
|
||||||
|
double aspect_ratio CV_DEFAULT(0));
|
||||||
|
|
||||||
|
static void _projectPoints( InputArray objectPoints,
|
||||||
|
InputArray rvec, InputArray tvec,
|
||||||
|
InputArray cameraMatrix, InputArray distCoeffs,
|
||||||
|
OutputArray imagePoints,
|
||||||
|
OutputArray jacobian = noArray(),
|
||||||
|
double aspectRatio = 0 );
|
||||||
|
|
||||||
|
|
||||||
void _drawPlanarBoard(Board *_board, Size outSize, OutputArray _img, int marginSize,
|
void _drawPlanarBoard(Board *_board, Size outSize, OutputArray _img, int marginSize,
|
||||||
int borderBits, bool drawAxis) {
|
int borderBits, bool drawAxis) {
|
||||||
|
|
||||||
@@ -127,194 +142,35 @@ void _drawPlanarBoard(Board *_board, Size outSize, OutputArray _img, int marginS
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/* Draw a (potentially partially visible) line. */
|
||||||
* @brief Convert point coordinates from world space to camera space.
|
static void linePartial(InputOutputArray image, Point3f p1, Point3f p2, const Scalar& color,
|
||||||
*
|
int thickness = 1, int lineType = LINE_8, int shift = 0)
|
||||||
* @param points A vector of points in world space.
|
|
||||||
* @param rvec Rotation matrix or Rodrigues rotation vector.
|
|
||||||
* @param tvec Translation vector from world to camera space.
|
|
||||||
*
|
|
||||||
* @return A vector of points in camera space.
|
|
||||||
*/
|
|
||||||
template<typename CvPointType>
|
|
||||||
static std::vector<CvPointType> worldToCamera(const std::vector<CvPointType>& points,
|
|
||||||
const cv::Mat& rvec, const cv::Mat& tvec)
|
|
||||||
{
|
{
|
||||||
// We operate with CV_64F matrices internally to avoid precision loss
|
// If both points are behind the screen, don't draw anything
|
||||||
cv::Mat rvec_64f;
|
if (p1.z <= 0 && p2.z <= 0) {
|
||||||
cv::Mat tvec_64f;
|
return;
|
||||||
rvec.convertTo(rvec_64f, CV_64F);
|
|
||||||
tvec.convertTo(tvec_64f, CV_64F);
|
|
||||||
|
|
||||||
// Convert Rodrigues vector to rotation matrix
|
|
||||||
cv::Mat rmat;
|
|
||||||
if ((rvec_64f.cols == 3 && rvec_64f.rows == 1) ||
|
|
||||||
(rvec_64f.cols == 1 && rvec_64f.rows == 3))
|
|
||||||
{
|
|
||||||
Rodrigues(rvec_64f, rmat);
|
|
||||||
}
|
}
|
||||||
else
|
Point2f p1p{p1.x, p1.y};
|
||||||
{
|
Point2f p2p{p2.x, p2.y};
|
||||||
rmat = rvec_64f.clone();
|
// If points are on the different sides of the plane, compute intersection point
|
||||||
|
if (p1.z * p2.z < 0) {
|
||||||
|
// Compute intersection point with the screen
|
||||||
|
// We denote alpha as such:
|
||||||
|
// xi = (1 - alpha) * x1 + alpha * x2
|
||||||
|
// yi = (1 - alpha) * y1 + alpha * y2
|
||||||
|
// zi = (1 - alpha) * z1 + alpha * z2 = 0
|
||||||
|
// Thus, alpha can be expressed as
|
||||||
|
// alpha = z1 / (z1 - z2)
|
||||||
|
float alpha = p1.z / (p1.z - p2.z);
|
||||||
|
Point2f pi{(1 - alpha) * p1.x + alpha * p2.x, (1 - alpha) * p1.y + alpha * p2.y};
|
||||||
|
// Now, if z1 is negative, we draw the line from (xi, yi) to (x2, y2), else we draw from (x1, y1) to (xi, yi)
|
||||||
|
if (p1.z < 0) {
|
||||||
|
p1p = pi;
|
||||||
|
} else {
|
||||||
|
p2p = pi;
|
||||||
|
}
|
||||||
}
|
}
|
||||||
// Make sure tvec has a size of (3, 1)
|
line(image, p1p, p2p, color, thickness, lineType, shift);
|
||||||
if (tvec_64f.rows == 1)
|
|
||||||
{
|
|
||||||
tvec_64f = tvec_64f.t();
|
|
||||||
}
|
|
||||||
std::vector<CvPointType> result;
|
|
||||||
result.reserve(points.size());
|
|
||||||
for(const auto& point : points)
|
|
||||||
{
|
|
||||||
// Calculate point coordinates in camera frame
|
|
||||||
// static_casts are here to silence potential narrowing conversion warnings
|
|
||||||
CvPointType camPoint{
|
|
||||||
static_cast<decltype(CvPointType::x)>(point.x * rmat.at<double>(0,0) + point.y * rmat.at<double>(0,1) + point.z * rmat.at<double>(0,2) + tvec_64f.at<double>(0)),
|
|
||||||
static_cast<decltype(CvPointType::y)>(point.x * rmat.at<double>(1,0) + point.y * rmat.at<double>(1,1) + point.z * rmat.at<double>(1,2) + tvec_64f.at<double>(1)),
|
|
||||||
static_cast<decltype(CvPointType::z)>(point.x * rmat.at<double>(2,0) + point.y * rmat.at<double>(2,1) + point.z * rmat.at<double>(2,2) + tvec_64f.at<double>(2))
|
|
||||||
};
|
|
||||||
result.push_back(camPoint);
|
|
||||||
}
|
|
||||||
return result;
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* @brief Project points from camera space to screen space, applying distortion in the process.
|
|
||||||
*
|
|
||||||
* @param points A vector of points in camera space.
|
|
||||||
* @param cameraMatrix OpenCV intrinsic camera matrix.
|
|
||||||
* @param distCoeffs OpenCV distortion model coefficients.
|
|
||||||
*
|
|
||||||
* @return A vector of points in screen space.
|
|
||||||
*/
|
|
||||||
template<typename CvPointType>
|
|
||||||
static std::vector<CvPointType> cameraToScreen(const std::vector<CvPointType>& points,
|
|
||||||
const cv::Mat& cameraMatrix,
|
|
||||||
const cv::Mat& distCoeffs)
|
|
||||||
{
|
|
||||||
// We operate with CV_64F matrices internally to avoid precision loss
|
|
||||||
cv::Mat cm_64f; // camera matrix, CV_64F
|
|
||||||
cv::Mat dc_64f; // distortion coefficients, CV_64F
|
|
||||||
cameraMatrix.convertTo(cm_64f, CV_64F);
|
|
||||||
distCoeffs.convertTo(dc_64f, CV_64F);
|
|
||||||
|
|
||||||
// Make sure distortion vector has a size of (N, 1)
|
|
||||||
if (dc_64f.rows == 1)
|
|
||||||
{
|
|
||||||
dc_64f = dc_64f.t();
|
|
||||||
}
|
|
||||||
|
|
||||||
// We will always use 12 distortion coefficients,
|
|
||||||
// and we can safely pad missing ones with zeroes
|
|
||||||
dc_64f.resize(12, 0.0);
|
|
||||||
|
|
||||||
std::vector<CvPointType> result;
|
|
||||||
result.reserve(points.size());
|
|
||||||
|
|
||||||
for(const auto& point : points)
|
|
||||||
{
|
|
||||||
// Apply perspective projection, preserving initial Z coordinate
|
|
||||||
// Always use double-precision
|
|
||||||
cv::Point3d camPoint{
|
|
||||||
point.x / point.z,
|
|
||||||
point.y / point.z,
|
|
||||||
point.z
|
|
||||||
};
|
|
||||||
|
|
||||||
// Apply distortion
|
|
||||||
// Note that we do not consider tilted sensor distortion
|
|
||||||
// r^2 - distance from the image center squared
|
|
||||||
double r2 = camPoint.x * camPoint.x + camPoint.y * camPoint.y;
|
|
||||||
// r^4 - same, but to the 4th power
|
|
||||||
double r4 = r2 * r2;
|
|
||||||
// r^6 - same, but to the 6th power
|
|
||||||
double r6 = r4 * r2;
|
|
||||||
// tg1 - first tangential shift factor (2 * x * y)
|
|
||||||
double tg1 = 2 * camPoint.x * camPoint.y;
|
|
||||||
// tg2 - second tangential shift factor (r^2 + 2 * x^2)
|
|
||||||
double tg2 = r2 + 2 * camPoint.x * camPoint.x;
|
|
||||||
// tg3 - third tangential shift factor (r^2 + 2 * y^2)
|
|
||||||
double tg3 = r2 + 2 * camPoint.y * camPoint.y;
|
|
||||||
// polynomial distortion factor (numerator)
|
|
||||||
double pndist = 1 + dc_64f.at<double>(0) * r2 + dc_64f.at<double>(1) * r4 + dc_64f.at<double>(4) * r6;
|
|
||||||
// polynomial distortion factror (denominator)
|
|
||||||
double pddist = 1.0 / (1 + dc_64f.at<double>(5) * r2 + dc_64f.at<double>(6) * r4 + dc_64f.at<double>(7) * r6);
|
|
||||||
// Distorted point coordinates (always double-precision)
|
|
||||||
cv::Point3d distortedPoint{
|
|
||||||
camPoint.x * pndist * pddist + dc_64f.at<double>(2) * tg1 + dc_64f.at<double>(3) * tg2 + dc_64f.at<double>(8) * r2 + dc_64f.at<double>(9) * r4,
|
|
||||||
camPoint.y * pndist * pddist + dc_64f.at<double>(2) * tg3 + dc_64f.at<double>(3) * tg1 + dc_64f.at<double>(10) * r2 + dc_64f.at<double>(11) * r4,
|
|
||||||
camPoint.z
|
|
||||||
};
|
|
||||||
|
|
||||||
// Convert to screen space
|
|
||||||
// We use static_cast here to silence potential warnings about narrowing conversions
|
|
||||||
// (we expect that to be the case)
|
|
||||||
CvPointType screenPoint{
|
|
||||||
static_cast<decltype(CvPointType::x)>(distortedPoint.x * cm_64f.at<double>(0, 0) + cm_64f.at<double>(0, 2)),
|
|
||||||
static_cast<decltype(CvPointType::y)>(distortedPoint.y * cm_64f.at<double>(1, 1) + cm_64f.at<double>(1, 2)),
|
|
||||||
static_cast<decltype(CvPointType::z)>(distortedPoint.z)
|
|
||||||
};
|
|
||||||
|
|
||||||
result.push_back(screenPoint);
|
|
||||||
}
|
|
||||||
return result;
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* @brief Clip a line against a clip plane.
|
|
||||||
*
|
|
||||||
* This function "clips" a line (described by two points in *camera space*)
|
|
||||||
* against a clip plane that is `clipPlaneDistance` meters away from the
|
|
||||||
* camera focal point. If both points are further away from the focal point
|
|
||||||
* than `clipPlaneDistance`, they will be returned unmodified. If one of the
|
|
||||||
* points is behind the clipping plane, a point *on* the clipping plane will
|
|
||||||
* be computed and returned as one of the points.
|
|
||||||
*
|
|
||||||
* If none of the points are visible, an empty vector will be returned.
|
|
||||||
*
|
|
||||||
* @param p1 First point on the line, in camera space.
|
|
||||||
* @param p2 Second point on the line, in camera space.
|
|
||||||
* @param clipPlaneDistance Distance from the focal point to the clipping plane.
|
|
||||||
* @return A vector of zero or two points on the clipped line, in camera space.
|
|
||||||
*/
|
|
||||||
static std::vector<Point3f> lineClip(Point3f p1, Point3f p2, float clipPlaneDistance = 0.1f)
|
|
||||||
{
|
|
||||||
// We don't need to compute an intersection if both points are
|
|
||||||
// behind us
|
|
||||||
if (p1.z < clipPlaneDistance && p2.z < clipPlaneDistance)
|
|
||||||
{
|
|
||||||
return {};
|
|
||||||
}
|
|
||||||
// We don't need to compute an intersection if both points are
|
|
||||||
// in front of us
|
|
||||||
if (p1.z > clipPlaneDistance && p2.z > clipPlaneDistance)
|
|
||||||
{
|
|
||||||
return {p1, p2};
|
|
||||||
}
|
|
||||||
// We don't really want to compute an intersection if both Z coordinates
|
|
||||||
// are sufficiently close to each other
|
|
||||||
if (std::abs(p1.z - p2.z) < 0.0001) // The number here is chosen arbitrarily
|
|
||||||
{
|
|
||||||
return {p1, p2};
|
|
||||||
}
|
|
||||||
// We compute the intersection as such:
|
|
||||||
// zi = (1 - alpha) * p1.z + alpha * p2.z = clipPlaneDistance
|
|
||||||
// alpha = (p1.z - clipPlaneDistance) / (p1.z - p2.z)
|
|
||||||
double alpha = (p1.z - clipPlaneDistance) / (p1.z - p2.z);
|
|
||||||
Point3f clipPlanePoint{
|
|
||||||
static_cast<float>((1 - alpha) * p1.x + alpha * p2.x),
|
|
||||||
static_cast<float>((1 - alpha) * p1.y + alpha * p2.y),
|
|
||||||
clipPlaneDistance
|
|
||||||
};
|
|
||||||
if (p1.z < clipPlaneDistance)
|
|
||||||
{
|
|
||||||
return {clipPlanePoint, p2};
|
|
||||||
}
|
|
||||||
else
|
|
||||||
{
|
|
||||||
return {p1, clipPlanePoint};
|
|
||||||
}
|
|
||||||
// Unreachable?
|
|
||||||
}
|
}
|
||||||
|
|
||||||
void _drawAxis(InputOutputArray _image, InputArray _cameraMatrix, InputArray _distCoeffs,
|
void _drawAxis(InputOutputArray _image, InputArray _cameraMatrix, InputArray _distCoeffs,
|
||||||
@@ -330,23 +186,647 @@ void _drawAxis(InputOutputArray _image, InputArray _cameraMatrix, InputArray _di
|
|||||||
axisPoints.push_back(Point3f(length, 0, 0));
|
axisPoints.push_back(Point3f(length, 0, 0));
|
||||||
axisPoints.push_back(Point3f(0, length, 0));
|
axisPoints.push_back(Point3f(0, length, 0));
|
||||||
axisPoints.push_back(Point3f(0, 0, length));
|
axisPoints.push_back(Point3f(0, 0, length));
|
||||||
auto camAxisPoints = worldToCamera(axisPoints, _rvec.getMat(), _tvec.getMat());
|
std::vector<Point3f> imagePointsZ;
|
||||||
auto axisX = cameraToScreen(lineClip(camAxisPoints[0], camAxisPoints[1]), _cameraMatrix.getMat(), _distCoeffs.getMat());
|
_projectPoints(axisPoints, _rvec, _tvec, _cameraMatrix, _distCoeffs, imagePointsZ);
|
||||||
auto axisY = cameraToScreen(lineClip(camAxisPoints[0], camAxisPoints[2]), _cameraMatrix.getMat(), _distCoeffs.getMat());
|
|
||||||
auto axisZ = cameraToScreen(lineClip(camAxisPoints[0], camAxisPoints[3]), _cameraMatrix.getMat(), _distCoeffs.getMat());
|
// draw axis lines
|
||||||
if (axisX.size() > 0)
|
linePartial(_image, imagePointsZ[0], imagePointsZ[1], Scalar(0, 0, 255), 3);
|
||||||
{
|
linePartial(_image, imagePointsZ[0], imagePointsZ[2], Scalar(0, 255, 0), 3);
|
||||||
line(_image, Point2f{axisX[0].x, axisX[0].y}, Point2f{axisX[1].x, axisX[1].y},
|
linePartial(_image, imagePointsZ[0], imagePointsZ[3], Scalar(255, 0, 0), 3);
|
||||||
Scalar(0, 0, 255), 3);
|
}
|
||||||
}
|
|
||||||
if (axisY.size() > 0)
|
static CvMat _cvMat(const cv::Mat& m)
|
||||||
{
|
{
|
||||||
line(_image, Point2f{axisY[0].x, axisY[0].y}, Point2f{axisY[1].x, axisY[1].y},
|
CvMat self;
|
||||||
Scalar(0, 255, 0), 3);
|
CV_DbgAssert(m.dims <= 2);
|
||||||
}
|
self = cvMat(m.rows, m.dims == 1 ? 1 : m.cols, m.type(), m.data);
|
||||||
if (axisZ.size() > 0)
|
self.step = (int)m.step[0];
|
||||||
{
|
self.type = (self.type & ~cv::Mat::CONTINUOUS_FLAG) | (m.flags & cv::Mat::CONTINUOUS_FLAG);
|
||||||
line(_image, Point2f{axisZ[0].x, axisZ[0].y}, Point2f{axisZ[1].x, axisZ[1].y},
|
return self;
|
||||||
Scalar(255, 0, 0), 3);
|
}
|
||||||
}
|
|
||||||
|
static void _projectPoints( InputArray _opoints,
|
||||||
|
InputArray _rvec,
|
||||||
|
InputArray _tvec,
|
||||||
|
InputArray _cameraMatrix,
|
||||||
|
InputArray _distCoeffs,
|
||||||
|
OutputArray _ipoints,
|
||||||
|
OutputArray _jacobian,
|
||||||
|
double aspectRatio )
|
||||||
|
{
|
||||||
|
Mat opoints = _opoints.getMat();
|
||||||
|
int npoints = opoints.checkVector(3), depth = opoints.depth();
|
||||||
|
CV_Assert(npoints >= 0 && (depth == CV_32F || depth == CV_64F));
|
||||||
|
|
||||||
|
CvMat dpdrot, dpdt, dpdf, dpdc, dpddist;
|
||||||
|
CvMat *pdpdrot = 0, *pdpdt = 0, *pdpdf = 0, *pdpdc = 0, *pdpddist = 0;
|
||||||
|
|
||||||
|
CV_Assert(_ipoints.needed());
|
||||||
|
|
||||||
|
_ipoints.create(npoints, 1, CV_MAKETYPE(depth, 3), -1, true);
|
||||||
|
Mat imagePoints = _ipoints.getMat();
|
||||||
|
CvMat c_imagePoints = _cvMat(imagePoints);
|
||||||
|
CvMat c_objectPoints = _cvMat(opoints);
|
||||||
|
Mat cameraMatrix = _cameraMatrix.getMat();
|
||||||
|
|
||||||
|
Mat rvec = _rvec.getMat(), tvec = _tvec.getMat();
|
||||||
|
CvMat c_cameraMatrix = _cvMat(cameraMatrix);
|
||||||
|
CvMat c_rvec = _cvMat(rvec), c_tvec = _cvMat(tvec);
|
||||||
|
|
||||||
|
double dc0buf[5] = {0};
|
||||||
|
Mat dc0(5, 1, CV_64F, dc0buf);
|
||||||
|
Mat distCoeffs = _distCoeffs.getMat();
|
||||||
|
if (distCoeffs.empty())
|
||||||
|
distCoeffs = dc0;
|
||||||
|
CvMat c_distCoeffs = _cvMat(distCoeffs);
|
||||||
|
int ndistCoeffs = distCoeffs.rows + distCoeffs.cols - 1;
|
||||||
|
|
||||||
|
Mat jacobian;
|
||||||
|
if (_jacobian.needed())
|
||||||
|
{
|
||||||
|
_jacobian.create(npoints * 2, 3 + 3 + 2 + 2 + ndistCoeffs, CV_64F);
|
||||||
|
jacobian = _jacobian.getMat();
|
||||||
|
pdpdrot = &(dpdrot = _cvMat(jacobian.colRange(0, 3)));
|
||||||
|
pdpdt = &(dpdt = _cvMat(jacobian.colRange(3, 6)));
|
||||||
|
pdpdf = &(dpdf = _cvMat(jacobian.colRange(6, 8)));
|
||||||
|
pdpdc = &(dpdc = _cvMat(jacobian.colRange(8, 10)));
|
||||||
|
pdpddist = &(dpddist = _cvMat(jacobian.colRange(10, 10 + ndistCoeffs)));
|
||||||
|
}
|
||||||
|
|
||||||
|
_cvProjectPoints2(&c_objectPoints, &c_rvec, &c_tvec, &c_cameraMatrix, &c_distCoeffs,
|
||||||
|
&c_imagePoints, pdpdrot, pdpdt, pdpdf, pdpdc, pdpddist, aspectRatio);
|
||||||
|
}
|
||||||
|
|
||||||
|
namespace _detail
|
||||||
|
{
|
||||||
|
template <typename FLOAT>
|
||||||
|
void computeTiltProjectionMatrix(FLOAT tauX,
|
||||||
|
FLOAT tauY,
|
||||||
|
Matx<FLOAT, 3, 3>* matTilt = 0,
|
||||||
|
Matx<FLOAT, 3, 3>* dMatTiltdTauX = 0,
|
||||||
|
Matx<FLOAT, 3, 3>* dMatTiltdTauY = 0,
|
||||||
|
Matx<FLOAT, 3, 3>* invMatTilt = 0)
|
||||||
|
{
|
||||||
|
FLOAT cTauX = cos(tauX);
|
||||||
|
FLOAT sTauX = sin(tauX);
|
||||||
|
FLOAT cTauY = cos(tauY);
|
||||||
|
FLOAT sTauY = sin(tauY);
|
||||||
|
Matx<FLOAT, 3, 3> matRotX = Matx<FLOAT, 3, 3>(1,0,0,0,cTauX,sTauX,0,-sTauX,cTauX);
|
||||||
|
Matx<FLOAT, 3, 3> matRotY = Matx<FLOAT, 3, 3>(cTauY,0,-sTauY,0,1,0,sTauY,0,cTauY);
|
||||||
|
Matx<FLOAT, 3, 3> matRotXY = matRotY * matRotX;
|
||||||
|
Matx<FLOAT, 3, 3> matProjZ = Matx<FLOAT, 3, 3>(matRotXY(2,2),0,-matRotXY(0,2),0,matRotXY(2,2),-matRotXY(1,2),0,0,1);
|
||||||
|
if (matTilt)
|
||||||
|
{
|
||||||
|
// Matrix for trapezoidal distortion of tilted image sensor
|
||||||
|
*matTilt = matProjZ * matRotXY;
|
||||||
|
}
|
||||||
|
if (dMatTiltdTauX)
|
||||||
|
{
|
||||||
|
// Derivative with respect to tauX
|
||||||
|
Matx<FLOAT, 3, 3> dMatRotXYdTauX = matRotY * Matx<FLOAT, 3, 3>(0,0,0,0,-sTauX,cTauX,0,-cTauX,-sTauX);
|
||||||
|
Matx<FLOAT, 3, 3> dMatProjZdTauX = Matx<FLOAT, 3, 3>(dMatRotXYdTauX(2,2),0,-dMatRotXYdTauX(0,2),
|
||||||
|
0,dMatRotXYdTauX(2,2),-dMatRotXYdTauX(1,2),0,0,0);
|
||||||
|
*dMatTiltdTauX = (matProjZ * dMatRotXYdTauX) + (dMatProjZdTauX * matRotXY);
|
||||||
|
}
|
||||||
|
if (dMatTiltdTauY)
|
||||||
|
{
|
||||||
|
// Derivative with respect to tauY
|
||||||
|
Matx<FLOAT, 3, 3> dMatRotXYdTauY = Matx<FLOAT, 3, 3>(-sTauY,0,-cTauY,0,0,0,cTauY,0,-sTauY) * matRotX;
|
||||||
|
Matx<FLOAT, 3, 3> dMatProjZdTauY = Matx<FLOAT, 3, 3>(dMatRotXYdTauY(2,2),0,-dMatRotXYdTauY(0,2),
|
||||||
|
0,dMatRotXYdTauY(2,2),-dMatRotXYdTauY(1,2),0,0,0);
|
||||||
|
*dMatTiltdTauY = (matProjZ * dMatRotXYdTauY) + (dMatProjZdTauY * matRotXY);
|
||||||
|
}
|
||||||
|
if (invMatTilt)
|
||||||
|
{
|
||||||
|
FLOAT inv = 1./matRotXY(2,2);
|
||||||
|
Matx<FLOAT, 3, 3> invMatProjZ = Matx<FLOAT, 3, 3>(inv,0,inv*matRotXY(0,2),0,inv,inv*matRotXY(1,2),0,0,1);
|
||||||
|
*invMatTilt = matRotXY.t()*invMatProjZ;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
static const char* cvDistCoeffErr = "Distortion coefficients must be 1x4, 4x1, 1x5, 5x1, 1x8, 8x1, 1x12, 12x1, 1x14 or 14x1 floating-point vector";
|
||||||
|
|
||||||
|
static void _cvProjectPoints2Internal( const CvMat* objectPoints,
|
||||||
|
const CvMat* r_vec,
|
||||||
|
const CvMat* t_vec,
|
||||||
|
const CvMat* A,
|
||||||
|
const CvMat* distCoeffs,
|
||||||
|
CvMat* imagePoints, CvMat* dpdr CV_DEFAULT(NULL),
|
||||||
|
CvMat* dpdt CV_DEFAULT(NULL), CvMat* dpdf CV_DEFAULT(NULL),
|
||||||
|
CvMat* dpdc CV_DEFAULT(NULL), CvMat* dpdk CV_DEFAULT(NULL),
|
||||||
|
CvMat* dpdo CV_DEFAULT(NULL),
|
||||||
|
double aspectRatio CV_DEFAULT(0) )
|
||||||
|
{
|
||||||
|
Ptr<CvMat> matM, _m;
|
||||||
|
Ptr<CvMat> _dpdr, _dpdt, _dpdc, _dpdf, _dpdk;
|
||||||
|
Ptr<CvMat> _dpdo;
|
||||||
|
|
||||||
|
int i, j, count;
|
||||||
|
int calc_derivatives;
|
||||||
|
const CvPoint3D64f* M;
|
||||||
|
CvPoint3D64f* m;
|
||||||
|
double r[3], R[9], dRdr[27], t[3], a[9], k[14] = {0,0,0,0,0,0,0,0,0,0,0,0,0,0}, fx, fy, cx, cy;
|
||||||
|
Matx33d matTilt = Matx33d::eye();
|
||||||
|
Matx33d dMatTiltdTauX(0,0,0,0,0,0,0,-1,0);
|
||||||
|
Matx33d dMatTiltdTauY(0,0,0,0,0,0,1,0,0);
|
||||||
|
CvMat _r, _t, _a = cvMat( 3, 3, CV_64F, a ), _k;
|
||||||
|
CvMat matR = cvMat( 3, 3, CV_64F, R ), _dRdr = cvMat( 3, 9, CV_64F, dRdr );
|
||||||
|
double *dpdr_p = 0, *dpdt_p = 0, *dpdk_p = 0, *dpdf_p = 0, *dpdc_p = 0;
|
||||||
|
double* dpdo_p = 0;
|
||||||
|
int dpdr_step = 0, dpdt_step = 0, dpdk_step = 0, dpdf_step = 0, dpdc_step = 0;
|
||||||
|
int dpdo_step = 0;
|
||||||
|
bool fixedAspectRatio = aspectRatio > FLT_EPSILON;
|
||||||
|
|
||||||
|
if( !CV_IS_MAT(objectPoints) || !CV_IS_MAT(r_vec) ||
|
||||||
|
!CV_IS_MAT(t_vec) || !CV_IS_MAT(A) ||
|
||||||
|
/*!CV_IS_MAT(distCoeffs) ||*/ !CV_IS_MAT(imagePoints) )
|
||||||
|
CV_Error( CV_StsBadArg, "One of required arguments is not a valid matrix" );
|
||||||
|
|
||||||
|
int total = objectPoints->rows * objectPoints->cols * CV_MAT_CN(objectPoints->type);
|
||||||
|
if(total % 3 != 0)
|
||||||
|
{
|
||||||
|
//we have stopped support of homogeneous coordinates because it cause ambiguity in interpretation of the input data
|
||||||
|
CV_Error( CV_StsBadArg, "Homogeneous coordinates are not supported" );
|
||||||
|
}
|
||||||
|
count = total / 3;
|
||||||
|
|
||||||
|
if( CV_IS_CONT_MAT(objectPoints->type) &&
|
||||||
|
(CV_MAT_DEPTH(objectPoints->type) == CV_32F || CV_MAT_DEPTH(objectPoints->type) == CV_64F)&&
|
||||||
|
((objectPoints->rows == 1 && CV_MAT_CN(objectPoints->type) == 3) ||
|
||||||
|
(objectPoints->rows == count && CV_MAT_CN(objectPoints->type)*objectPoints->cols == 3) ||
|
||||||
|
(objectPoints->rows == 3 && CV_MAT_CN(objectPoints->type) == 1 && objectPoints->cols == count)))
|
||||||
|
{
|
||||||
|
matM.reset(cvCreateMat( objectPoints->rows, objectPoints->cols, CV_MAKETYPE(CV_64F,CV_MAT_CN(objectPoints->type)) ));
|
||||||
|
cvConvert(objectPoints, matM);
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
// matM = cvCreateMat( 1, count, CV_64FC3 );
|
||||||
|
// cvConvertPointsHomogeneous( objectPoints, matM );
|
||||||
|
CV_Error( CV_StsBadArg, "Homogeneous coordinates are not supported" );
|
||||||
|
}
|
||||||
|
|
||||||
|
if( CV_IS_CONT_MAT(imagePoints->type) &&
|
||||||
|
(CV_MAT_DEPTH(imagePoints->type) == CV_32F || CV_MAT_DEPTH(imagePoints->type) == CV_64F) &&
|
||||||
|
((imagePoints->rows == 1 && CV_MAT_CN(imagePoints->type) == 3) ||
|
||||||
|
(imagePoints->rows == count && CV_MAT_CN(imagePoints->type)*imagePoints->cols == 3) ||
|
||||||
|
(imagePoints->rows == 3 && CV_MAT_CN(imagePoints->type) == 1 && imagePoints->cols == count)))
|
||||||
|
{
|
||||||
|
_m.reset(cvCreateMat( imagePoints->rows, imagePoints->cols, CV_MAKETYPE(CV_64F,CV_MAT_CN(imagePoints->type)) ));
|
||||||
|
cvConvert(imagePoints, _m);
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
// _m = cvCreateMat( 1, count, CV_64FC2 );
|
||||||
|
CV_Error( CV_StsBadArg, "Homogeneous coordinates are not supported" );
|
||||||
|
}
|
||||||
|
|
||||||
|
M = (CvPoint3D64f*)matM->data.db;
|
||||||
|
m = (CvPoint3D64f*)_m->data.db;
|
||||||
|
|
||||||
|
if( (CV_MAT_DEPTH(r_vec->type) != CV_64F && CV_MAT_DEPTH(r_vec->type) != CV_32F) ||
|
||||||
|
(((r_vec->rows != 1 && r_vec->cols != 1) ||
|
||||||
|
r_vec->rows*r_vec->cols*CV_MAT_CN(r_vec->type) != 3) &&
|
||||||
|
((r_vec->rows != 3 && r_vec->cols != 3) || CV_MAT_CN(r_vec->type) != 1)))
|
||||||
|
CV_Error( CV_StsBadArg, "Rotation must be represented by 1x3 or 3x1 "
|
||||||
|
"floating-point rotation vector, or 3x3 rotation matrix" );
|
||||||
|
|
||||||
|
if( r_vec->rows == 3 && r_vec->cols == 3 )
|
||||||
|
{
|
||||||
|
_r = cvMat( 3, 1, CV_64FC1, r );
|
||||||
|
cvRodrigues2( r_vec, &_r );
|
||||||
|
cvRodrigues2( &_r, &matR, &_dRdr );
|
||||||
|
cvCopy( r_vec, &matR );
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
_r = cvMat( r_vec->rows, r_vec->cols, CV_MAKETYPE(CV_64F,CV_MAT_CN(r_vec->type)), r );
|
||||||
|
cvConvert( r_vec, &_r );
|
||||||
|
cvRodrigues2( &_r, &matR, &_dRdr );
|
||||||
|
}
|
||||||
|
|
||||||
|
if( (CV_MAT_DEPTH(t_vec->type) != CV_64F && CV_MAT_DEPTH(t_vec->type) != CV_32F) ||
|
||||||
|
(t_vec->rows != 1 && t_vec->cols != 1) ||
|
||||||
|
t_vec->rows*t_vec->cols*CV_MAT_CN(t_vec->type) != 3 )
|
||||||
|
CV_Error( CV_StsBadArg,
|
||||||
|
"Translation vector must be 1x3 or 3x1 floating-point vector" );
|
||||||
|
|
||||||
|
_t = cvMat( t_vec->rows, t_vec->cols, CV_MAKETYPE(CV_64F,CV_MAT_CN(t_vec->type)), t );
|
||||||
|
cvConvert( t_vec, &_t );
|
||||||
|
|
||||||
|
if( (CV_MAT_TYPE(A->type) != CV_64FC1 && CV_MAT_TYPE(A->type) != CV_32FC1) ||
|
||||||
|
A->rows != 3 || A->cols != 3 )
|
||||||
|
CV_Error( CV_StsBadArg, "Instrinsic parameters must be 3x3 floating-point matrix" );
|
||||||
|
|
||||||
|
cvConvert( A, &_a );
|
||||||
|
fx = a[0]; fy = a[4];
|
||||||
|
cx = a[2]; cy = a[5];
|
||||||
|
|
||||||
|
if( fixedAspectRatio )
|
||||||
|
fx = fy*aspectRatio;
|
||||||
|
|
||||||
|
if( distCoeffs )
|
||||||
|
{
|
||||||
|
if( !CV_IS_MAT(distCoeffs) ||
|
||||||
|
(CV_MAT_DEPTH(distCoeffs->type) != CV_64F &&
|
||||||
|
CV_MAT_DEPTH(distCoeffs->type) != CV_32F) ||
|
||||||
|
(distCoeffs->rows != 1 && distCoeffs->cols != 1) ||
|
||||||
|
(distCoeffs->rows*distCoeffs->cols*CV_MAT_CN(distCoeffs->type) != 4 &&
|
||||||
|
distCoeffs->rows*distCoeffs->cols*CV_MAT_CN(distCoeffs->type) != 5 &&
|
||||||
|
distCoeffs->rows*distCoeffs->cols*CV_MAT_CN(distCoeffs->type) != 8 &&
|
||||||
|
distCoeffs->rows*distCoeffs->cols*CV_MAT_CN(distCoeffs->type) != 12 &&
|
||||||
|
distCoeffs->rows*distCoeffs->cols*CV_MAT_CN(distCoeffs->type) != 14) )
|
||||||
|
CV_Error( CV_StsBadArg, cvDistCoeffErr );
|
||||||
|
|
||||||
|
_k = cvMat( distCoeffs->rows, distCoeffs->cols,
|
||||||
|
CV_MAKETYPE(CV_64F,CV_MAT_CN(distCoeffs->type)), k );
|
||||||
|
cvConvert( distCoeffs, &_k );
|
||||||
|
if(k[12] != 0 || k[13] != 0)
|
||||||
|
{
|
||||||
|
_detail::computeTiltProjectionMatrix(k[12], k[13],
|
||||||
|
&matTilt, &dMatTiltdTauX, &dMatTiltdTauY);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if( dpdr )
|
||||||
|
{
|
||||||
|
if( !CV_IS_MAT(dpdr) ||
|
||||||
|
(CV_MAT_TYPE(dpdr->type) != CV_32FC1 &&
|
||||||
|
CV_MAT_TYPE(dpdr->type) != CV_64FC1) ||
|
||||||
|
dpdr->rows != count*2 || dpdr->cols != 3 )
|
||||||
|
CV_Error( CV_StsBadArg, "dp/drot must be 2Nx3 floating-point matrix" );
|
||||||
|
|
||||||
|
if( CV_MAT_TYPE(dpdr->type) == CV_64FC1 )
|
||||||
|
{
|
||||||
|
_dpdr.reset(cvCloneMat(dpdr));
|
||||||
|
}
|
||||||
|
else
|
||||||
|
_dpdr.reset(cvCreateMat( 2*count, 3, CV_64FC1 ));
|
||||||
|
dpdr_p = _dpdr->data.db;
|
||||||
|
dpdr_step = _dpdr->step/sizeof(dpdr_p[0]);
|
||||||
|
}
|
||||||
|
|
||||||
|
if( dpdt )
|
||||||
|
{
|
||||||
|
if( !CV_IS_MAT(dpdt) ||
|
||||||
|
(CV_MAT_TYPE(dpdt->type) != CV_32FC1 &&
|
||||||
|
CV_MAT_TYPE(dpdt->type) != CV_64FC1) ||
|
||||||
|
dpdt->rows != count*2 || dpdt->cols != 3 )
|
||||||
|
CV_Error( CV_StsBadArg, "dp/dT must be 2Nx3 floating-point matrix" );
|
||||||
|
|
||||||
|
if( CV_MAT_TYPE(dpdt->type) == CV_64FC1 )
|
||||||
|
{
|
||||||
|
_dpdt.reset(cvCloneMat(dpdt));
|
||||||
|
}
|
||||||
|
else
|
||||||
|
_dpdt.reset(cvCreateMat( 2*count, 3, CV_64FC1 ));
|
||||||
|
dpdt_p = _dpdt->data.db;
|
||||||
|
dpdt_step = _dpdt->step/sizeof(dpdt_p[0]);
|
||||||
|
}
|
||||||
|
|
||||||
|
if( dpdf )
|
||||||
|
{
|
||||||
|
if( !CV_IS_MAT(dpdf) ||
|
||||||
|
(CV_MAT_TYPE(dpdf->type) != CV_32FC1 && CV_MAT_TYPE(dpdf->type) != CV_64FC1) ||
|
||||||
|
dpdf->rows != count*2 || dpdf->cols != 2 )
|
||||||
|
CV_Error( CV_StsBadArg, "dp/df must be 2Nx2 floating-point matrix" );
|
||||||
|
|
||||||
|
if( CV_MAT_TYPE(dpdf->type) == CV_64FC1 )
|
||||||
|
{
|
||||||
|
_dpdf.reset(cvCloneMat(dpdf));
|
||||||
|
}
|
||||||
|
else
|
||||||
|
_dpdf.reset(cvCreateMat( 2*count, 2, CV_64FC1 ));
|
||||||
|
dpdf_p = _dpdf->data.db;
|
||||||
|
dpdf_step = _dpdf->step/sizeof(dpdf_p[0]);
|
||||||
|
}
|
||||||
|
|
||||||
|
if( dpdc )
|
||||||
|
{
|
||||||
|
if( !CV_IS_MAT(dpdc) ||
|
||||||
|
(CV_MAT_TYPE(dpdc->type) != CV_32FC1 && CV_MAT_TYPE(dpdc->type) != CV_64FC1) ||
|
||||||
|
dpdc->rows != count*2 || dpdc->cols != 2 )
|
||||||
|
CV_Error( CV_StsBadArg, "dp/dc must be 2Nx2 floating-point matrix" );
|
||||||
|
|
||||||
|
if( CV_MAT_TYPE(dpdc->type) == CV_64FC1 )
|
||||||
|
{
|
||||||
|
_dpdc.reset(cvCloneMat(dpdc));
|
||||||
|
}
|
||||||
|
else
|
||||||
|
_dpdc.reset(cvCreateMat( 2*count, 2, CV_64FC1 ));
|
||||||
|
dpdc_p = _dpdc->data.db;
|
||||||
|
dpdc_step = _dpdc->step/sizeof(dpdc_p[0]);
|
||||||
|
}
|
||||||
|
|
||||||
|
if( dpdk )
|
||||||
|
{
|
||||||
|
if( !CV_IS_MAT(dpdk) ||
|
||||||
|
(CV_MAT_TYPE(dpdk->type) != CV_32FC1 && CV_MAT_TYPE(dpdk->type) != CV_64FC1) ||
|
||||||
|
dpdk->rows != count*2 || (dpdk->cols != 14 && dpdk->cols != 12 && dpdk->cols != 8 && dpdk->cols != 5 && dpdk->cols != 4 && dpdk->cols != 2) )
|
||||||
|
CV_Error( CV_StsBadArg, "dp/df must be 2Nx14, 2Nx12, 2Nx8, 2Nx5, 2Nx4 or 2Nx2 floating-point matrix" );
|
||||||
|
|
||||||
|
if( !distCoeffs )
|
||||||
|
CV_Error( CV_StsNullPtr, "distCoeffs is NULL while dpdk is not" );
|
||||||
|
|
||||||
|
if( CV_MAT_TYPE(dpdk->type) == CV_64FC1 )
|
||||||
|
{
|
||||||
|
_dpdk.reset(cvCloneMat(dpdk));
|
||||||
|
}
|
||||||
|
else
|
||||||
|
_dpdk.reset(cvCreateMat( dpdk->rows, dpdk->cols, CV_64FC1 ));
|
||||||
|
dpdk_p = _dpdk->data.db;
|
||||||
|
dpdk_step = _dpdk->step/sizeof(dpdk_p[0]);
|
||||||
|
}
|
||||||
|
|
||||||
|
if( dpdo )
|
||||||
|
{
|
||||||
|
if( !CV_IS_MAT( dpdo ) || ( CV_MAT_TYPE( dpdo->type ) != CV_32FC1
|
||||||
|
&& CV_MAT_TYPE( dpdo->type ) != CV_64FC1 )
|
||||||
|
|| dpdo->rows != count * 2 || dpdo->cols != count * 3 )
|
||||||
|
CV_Error( CV_StsBadArg, "dp/do must be 2Nx3N floating-point matrix" );
|
||||||
|
|
||||||
|
if( CV_MAT_TYPE( dpdo->type ) == CV_64FC1 )
|
||||||
|
{
|
||||||
|
_dpdo.reset( cvCloneMat( dpdo ) );
|
||||||
|
}
|
||||||
|
else
|
||||||
|
_dpdo.reset( cvCreateMat( 2 * count, 3 * count, CV_64FC1 ) );
|
||||||
|
cvZero(_dpdo);
|
||||||
|
dpdo_p = _dpdo->data.db;
|
||||||
|
dpdo_step = _dpdo->step / sizeof( dpdo_p[0] );
|
||||||
|
}
|
||||||
|
|
||||||
|
calc_derivatives = dpdr || dpdt || dpdf || dpdc || dpdk || dpdo;
|
||||||
|
|
||||||
|
for( i = 0; i < count; i++ )
|
||||||
|
{
|
||||||
|
double X = M[i].x, Y = M[i].y, Z = M[i].z;
|
||||||
|
double x = R[0]*X + R[1]*Y + R[2]*Z + t[0];
|
||||||
|
double y = R[3]*X + R[4]*Y + R[5]*Z + t[1];
|
||||||
|
double z = R[6]*X + R[7]*Y + R[8]*Z + t[2];
|
||||||
|
double r2, r4, r6, a1, a2, a3, cdist, icdist2;
|
||||||
|
double xd, yd, xd0, yd0, invProj;
|
||||||
|
Vec3d vecTilt;
|
||||||
|
Vec3d dVecTilt;
|
||||||
|
Matx22d dMatTilt;
|
||||||
|
Vec2d dXdYd;
|
||||||
|
|
||||||
|
double z0 = z;
|
||||||
|
z = z ? 1./z : 1;
|
||||||
|
x *= z; y *= z;
|
||||||
|
|
||||||
|
r2 = x*x + y*y;
|
||||||
|
r4 = r2*r2;
|
||||||
|
r6 = r4*r2;
|
||||||
|
a1 = 2*x*y;
|
||||||
|
a2 = r2 + 2*x*x;
|
||||||
|
a3 = r2 + 2*y*y;
|
||||||
|
cdist = 1 + k[0]*r2 + k[1]*r4 + k[4]*r6;
|
||||||
|
icdist2 = 1./(1 + k[5]*r2 + k[6]*r4 + k[7]*r6);
|
||||||
|
xd0 = x*cdist*icdist2 + k[2]*a1 + k[3]*a2 + k[8]*r2+k[9]*r4;
|
||||||
|
yd0 = y*cdist*icdist2 + k[2]*a3 + k[3]*a1 + k[10]*r2+k[11]*r4;
|
||||||
|
|
||||||
|
// additional distortion by projecting onto a tilt plane
|
||||||
|
vecTilt = matTilt*Vec3d(xd0, yd0, 1);
|
||||||
|
invProj = vecTilt(2) ? 1./vecTilt(2) : 1;
|
||||||
|
xd = invProj * vecTilt(0);
|
||||||
|
yd = invProj * vecTilt(1);
|
||||||
|
|
||||||
|
m[i].x = xd*fx + cx;
|
||||||
|
m[i].y = yd*fy + cy;
|
||||||
|
m[i].z = z; // Just put the projected Z coordinate here, we mainly care about the sign
|
||||||
|
|
||||||
|
if( calc_derivatives )
|
||||||
|
{
|
||||||
|
if( dpdc_p )
|
||||||
|
{
|
||||||
|
dpdc_p[0] = 1; dpdc_p[1] = 0; // dp_xdc_x; dp_xdc_y
|
||||||
|
dpdc_p[dpdc_step] = 0;
|
||||||
|
dpdc_p[dpdc_step+1] = 1;
|
||||||
|
dpdc_p += dpdc_step*2;
|
||||||
|
}
|
||||||
|
|
||||||
|
if( dpdf_p )
|
||||||
|
{
|
||||||
|
if( fixedAspectRatio )
|
||||||
|
{
|
||||||
|
dpdf_p[0] = 0; dpdf_p[1] = xd*aspectRatio; // dp_xdf_x; dp_xdf_y
|
||||||
|
dpdf_p[dpdf_step] = 0;
|
||||||
|
dpdf_p[dpdf_step+1] = yd;
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
dpdf_p[0] = xd; dpdf_p[1] = 0;
|
||||||
|
dpdf_p[dpdf_step] = 0;
|
||||||
|
dpdf_p[dpdf_step+1] = yd;
|
||||||
|
}
|
||||||
|
dpdf_p += dpdf_step*2;
|
||||||
|
}
|
||||||
|
for (int row = 0; row < 2; ++row)
|
||||||
|
for (int col = 0; col < 2; ++col)
|
||||||
|
dMatTilt(row,col) = matTilt(row,col)*vecTilt(2)
|
||||||
|
- matTilt(2,col)*vecTilt(row);
|
||||||
|
double invProjSquare = (invProj*invProj);
|
||||||
|
dMatTilt *= invProjSquare;
|
||||||
|
if( dpdk_p )
|
||||||
|
{
|
||||||
|
dXdYd = dMatTilt*Vec2d(x*icdist2*r2, y*icdist2*r2);
|
||||||
|
dpdk_p[0] = fx*dXdYd(0);
|
||||||
|
dpdk_p[dpdk_step] = fy*dXdYd(1);
|
||||||
|
dXdYd = dMatTilt*Vec2d(x*icdist2*r4, y*icdist2*r4);
|
||||||
|
dpdk_p[1] = fx*dXdYd(0);
|
||||||
|
dpdk_p[dpdk_step+1] = fy*dXdYd(1);
|
||||||
|
if( _dpdk->cols > 2 )
|
||||||
|
{
|
||||||
|
dXdYd = dMatTilt*Vec2d(a1, a3);
|
||||||
|
dpdk_p[2] = fx*dXdYd(0);
|
||||||
|
dpdk_p[dpdk_step+2] = fy*dXdYd(1);
|
||||||
|
dXdYd = dMatTilt*Vec2d(a2, a1);
|
||||||
|
dpdk_p[3] = fx*dXdYd(0);
|
||||||
|
dpdk_p[dpdk_step+3] = fy*dXdYd(1);
|
||||||
|
if( _dpdk->cols > 4 )
|
||||||
|
{
|
||||||
|
dXdYd = dMatTilt*Vec2d(x*icdist2*r6, y*icdist2*r6);
|
||||||
|
dpdk_p[4] = fx*dXdYd(0);
|
||||||
|
dpdk_p[dpdk_step+4] = fy*dXdYd(1);
|
||||||
|
|
||||||
|
if( _dpdk->cols > 5 )
|
||||||
|
{
|
||||||
|
dXdYd = dMatTilt*Vec2d(
|
||||||
|
x*cdist*(-icdist2)*icdist2*r2, y*cdist*(-icdist2)*icdist2*r2);
|
||||||
|
dpdk_p[5] = fx*dXdYd(0);
|
||||||
|
dpdk_p[dpdk_step+5] = fy*dXdYd(1);
|
||||||
|
dXdYd = dMatTilt*Vec2d(
|
||||||
|
x*cdist*(-icdist2)*icdist2*r4, y*cdist*(-icdist2)*icdist2*r4);
|
||||||
|
dpdk_p[6] = fx*dXdYd(0);
|
||||||
|
dpdk_p[dpdk_step+6] = fy*dXdYd(1);
|
||||||
|
dXdYd = dMatTilt*Vec2d(
|
||||||
|
x*cdist*(-icdist2)*icdist2*r6, y*cdist*(-icdist2)*icdist2*r6);
|
||||||
|
dpdk_p[7] = fx*dXdYd(0);
|
||||||
|
dpdk_p[dpdk_step+7] = fy*dXdYd(1);
|
||||||
|
if( _dpdk->cols > 8 )
|
||||||
|
{
|
||||||
|
dXdYd = dMatTilt*Vec2d(r2, 0);
|
||||||
|
dpdk_p[8] = fx*dXdYd(0); //s1
|
||||||
|
dpdk_p[dpdk_step+8] = fy*dXdYd(1); //s1
|
||||||
|
dXdYd = dMatTilt*Vec2d(r4, 0);
|
||||||
|
dpdk_p[9] = fx*dXdYd(0); //s2
|
||||||
|
dpdk_p[dpdk_step+9] = fy*dXdYd(1); //s2
|
||||||
|
dXdYd = dMatTilt*Vec2d(0, r2);
|
||||||
|
dpdk_p[10] = fx*dXdYd(0);//s3
|
||||||
|
dpdk_p[dpdk_step+10] = fy*dXdYd(1); //s3
|
||||||
|
dXdYd = dMatTilt*Vec2d(0, r4);
|
||||||
|
dpdk_p[11] = fx*dXdYd(0);//s4
|
||||||
|
dpdk_p[dpdk_step+11] = fy*dXdYd(1); //s4
|
||||||
|
if( _dpdk->cols > 12 )
|
||||||
|
{
|
||||||
|
dVecTilt = dMatTiltdTauX * Vec3d(xd0, yd0, 1);
|
||||||
|
dpdk_p[12] = fx * invProjSquare * (
|
||||||
|
dVecTilt(0) * vecTilt(2) - dVecTilt(2) * vecTilt(0));
|
||||||
|
dpdk_p[dpdk_step+12] = fy*invProjSquare * (
|
||||||
|
dVecTilt(1) * vecTilt(2) - dVecTilt(2) * vecTilt(1));
|
||||||
|
dVecTilt = dMatTiltdTauY * Vec3d(xd0, yd0, 1);
|
||||||
|
dpdk_p[13] = fx * invProjSquare * (
|
||||||
|
dVecTilt(0) * vecTilt(2) - dVecTilt(2) * vecTilt(0));
|
||||||
|
dpdk_p[dpdk_step+13] = fy * invProjSquare * (
|
||||||
|
dVecTilt(1) * vecTilt(2) - dVecTilt(2) * vecTilt(1));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
dpdk_p += dpdk_step*2;
|
||||||
|
}
|
||||||
|
|
||||||
|
if( dpdt_p )
|
||||||
|
{
|
||||||
|
double dxdt[] = { z, 0, -x*z }, dydt[] = { 0, z, -y*z };
|
||||||
|
for( j = 0; j < 3; j++ )
|
||||||
|
{
|
||||||
|
double dr2dt = 2*x*dxdt[j] + 2*y*dydt[j];
|
||||||
|
double dcdist_dt = k[0]*dr2dt + 2*k[1]*r2*dr2dt + 3*k[4]*r4*dr2dt;
|
||||||
|
double dicdist2_dt = -icdist2*icdist2*(k[5]*dr2dt + 2*k[6]*r2*dr2dt + 3*k[7]*r4*dr2dt);
|
||||||
|
double da1dt = 2*(x*dydt[j] + y*dxdt[j]);
|
||||||
|
double dmxdt = (dxdt[j]*cdist*icdist2 + x*dcdist_dt*icdist2 + x*cdist*dicdist2_dt +
|
||||||
|
k[2]*da1dt + k[3]*(dr2dt + 4*x*dxdt[j]) + k[8]*dr2dt + 2*r2*k[9]*dr2dt);
|
||||||
|
double dmydt = (dydt[j]*cdist*icdist2 + y*dcdist_dt*icdist2 + y*cdist*dicdist2_dt +
|
||||||
|
k[2]*(dr2dt + 4*y*dydt[j]) + k[3]*da1dt + k[10]*dr2dt + 2*r2*k[11]*dr2dt);
|
||||||
|
dXdYd = dMatTilt*Vec2d(dmxdt, dmydt);
|
||||||
|
dpdt_p[j] = fx*dXdYd(0);
|
||||||
|
dpdt_p[dpdt_step+j] = fy*dXdYd(1);
|
||||||
|
}
|
||||||
|
dpdt_p += dpdt_step*2;
|
||||||
|
}
|
||||||
|
|
||||||
|
if( dpdr_p )
|
||||||
|
{
|
||||||
|
double dx0dr[] =
|
||||||
|
{
|
||||||
|
X*dRdr[0] + Y*dRdr[1] + Z*dRdr[2],
|
||||||
|
X*dRdr[9] + Y*dRdr[10] + Z*dRdr[11],
|
||||||
|
X*dRdr[18] + Y*dRdr[19] + Z*dRdr[20]
|
||||||
|
};
|
||||||
|
double dy0dr[] =
|
||||||
|
{
|
||||||
|
X*dRdr[3] + Y*dRdr[4] + Z*dRdr[5],
|
||||||
|
X*dRdr[12] + Y*dRdr[13] + Z*dRdr[14],
|
||||||
|
X*dRdr[21] + Y*dRdr[22] + Z*dRdr[23]
|
||||||
|
};
|
||||||
|
double dz0dr[] =
|
||||||
|
{
|
||||||
|
X*dRdr[6] + Y*dRdr[7] + Z*dRdr[8],
|
||||||
|
X*dRdr[15] + Y*dRdr[16] + Z*dRdr[17],
|
||||||
|
X*dRdr[24] + Y*dRdr[25] + Z*dRdr[26]
|
||||||
|
};
|
||||||
|
for( j = 0; j < 3; j++ )
|
||||||
|
{
|
||||||
|
double dxdr = z*(dx0dr[j] - x*dz0dr[j]);
|
||||||
|
double dydr = z*(dy0dr[j] - y*dz0dr[j]);
|
||||||
|
double dr2dr = 2*x*dxdr + 2*y*dydr;
|
||||||
|
double dcdist_dr = (k[0] + 2*k[1]*r2 + 3*k[4]*r4)*dr2dr;
|
||||||
|
double dicdist2_dr = -icdist2*icdist2*(k[5] + 2*k[6]*r2 + 3*k[7]*r4)*dr2dr;
|
||||||
|
double da1dr = 2*(x*dydr + y*dxdr);
|
||||||
|
double dmxdr = (dxdr*cdist*icdist2 + x*dcdist_dr*icdist2 + x*cdist*dicdist2_dr +
|
||||||
|
k[2]*da1dr + k[3]*(dr2dr + 4*x*dxdr) + (k[8] + 2*r2*k[9])*dr2dr);
|
||||||
|
double dmydr = (dydr*cdist*icdist2 + y*dcdist_dr*icdist2 + y*cdist*dicdist2_dr +
|
||||||
|
k[2]*(dr2dr + 4*y*dydr) + k[3]*da1dr + (k[10] + 2*r2*k[11])*dr2dr);
|
||||||
|
dXdYd = dMatTilt*Vec2d(dmxdr, dmydr);
|
||||||
|
dpdr_p[j] = fx*dXdYd(0);
|
||||||
|
dpdr_p[dpdr_step+j] = fy*dXdYd(1);
|
||||||
|
}
|
||||||
|
dpdr_p += dpdr_step*2;
|
||||||
|
}
|
||||||
|
|
||||||
|
if( dpdo_p )
|
||||||
|
{
|
||||||
|
double dxdo[] = { z * ( R[0] - x * z * z0 * R[6] ),
|
||||||
|
z * ( R[1] - x * z * z0 * R[7] ),
|
||||||
|
z * ( R[2] - x * z * z0 * R[8] ) };
|
||||||
|
double dydo[] = { z * ( R[3] - y * z * z0 * R[6] ),
|
||||||
|
z * ( R[4] - y * z * z0 * R[7] ),
|
||||||
|
z * ( R[5] - y * z * z0 * R[8] ) };
|
||||||
|
for( j = 0; j < 3; j++ )
|
||||||
|
{
|
||||||
|
double dr2do = 2 * x * dxdo[j] + 2 * y * dydo[j];
|
||||||
|
double dr4do = 2 * r2 * dr2do;
|
||||||
|
double dr6do = 3 * r4 * dr2do;
|
||||||
|
double da1do = 2 * y * dxdo[j] + 2 * x * dydo[j];
|
||||||
|
double da2do = dr2do + 4 * x * dxdo[j];
|
||||||
|
double da3do = dr2do + 4 * y * dydo[j];
|
||||||
|
double dcdist_do
|
||||||
|
= k[0] * dr2do + k[1] * dr4do + k[4] * dr6do;
|
||||||
|
double dicdist2_do = -icdist2 * icdist2
|
||||||
|
* ( k[5] * dr2do + k[6] * dr4do + k[7] * dr6do );
|
||||||
|
double dxd0_do = cdist * icdist2 * dxdo[j]
|
||||||
|
+ x * icdist2 * dcdist_do + x * cdist * dicdist2_do
|
||||||
|
+ k[2] * da1do + k[3] * da2do + k[8] * dr2do
|
||||||
|
+ k[9] * dr4do;
|
||||||
|
double dyd0_do = cdist * icdist2 * dydo[j]
|
||||||
|
+ y * icdist2 * dcdist_do + y * cdist * dicdist2_do
|
||||||
|
+ k[2] * da3do + k[3] * da1do + k[10] * dr2do
|
||||||
|
+ k[11] * dr4do;
|
||||||
|
dXdYd = dMatTilt * Vec2d( dxd0_do, dyd0_do );
|
||||||
|
dpdo_p[i * 3 + j] = fx * dXdYd( 0 );
|
||||||
|
dpdo_p[dpdo_step + i * 3 + j] = fy * dXdYd( 1 );
|
||||||
|
}
|
||||||
|
dpdo_p += dpdo_step * 2;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if( _m != imagePoints )
|
||||||
|
cvConvert( _m, imagePoints );
|
||||||
|
|
||||||
|
if( _dpdr != dpdr )
|
||||||
|
cvConvert( _dpdr, dpdr );
|
||||||
|
|
||||||
|
if( _dpdt != dpdt )
|
||||||
|
cvConvert( _dpdt, dpdt );
|
||||||
|
|
||||||
|
if( _dpdf != dpdf )
|
||||||
|
cvConvert( _dpdf, dpdf );
|
||||||
|
|
||||||
|
if( _dpdc != dpdc )
|
||||||
|
cvConvert( _dpdc, dpdc );
|
||||||
|
|
||||||
|
if( _dpdk != dpdk )
|
||||||
|
cvConvert( _dpdk, dpdk );
|
||||||
|
|
||||||
|
if( _dpdo != dpdo )
|
||||||
|
cvConvert( _dpdo, dpdo );
|
||||||
|
}
|
||||||
|
|
||||||
|
static void _cvProjectPoints2( const CvMat* objectPoints,
|
||||||
|
const CvMat* r_vec,
|
||||||
|
const CvMat* t_vec,
|
||||||
|
const CvMat* A,
|
||||||
|
const CvMat* distCoeffs,
|
||||||
|
CvMat* imagePoints, CvMat* dpdr,
|
||||||
|
CvMat* dpdt, CvMat* dpdf,
|
||||||
|
CvMat* dpdc, CvMat* dpdk,
|
||||||
|
double aspectRatio )
|
||||||
|
{
|
||||||
|
_cvProjectPoints2Internal( objectPoints, r_vec, t_vec, A, distCoeffs, imagePoints, dpdr, dpdt,
|
||||||
|
dpdf, dpdc, dpdk, NULL, aspectRatio );
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -13,7 +13,7 @@
|
|||||||
Generate map file for aruco_map nodelet.
|
Generate map file for aruco_map nodelet.
|
||||||
|
|
||||||
Usage:
|
Usage:
|
||||||
genmap.py <length> <x> <y> <dist_x> <dist_y> [<first>] [<x0>] [<y0>] [--top-left | --bottom-left] [-o <filename>]
|
genmap.py <length> <x> <y> <dist_x> <dist_y> [<first>] [<x0>] [<y0>] [--top-left | --bottom-left]
|
||||||
genmap.py (-h | --help)
|
genmap.py (-h | --help)
|
||||||
|
|
||||||
Options:
|
Options:
|
||||||
@@ -27,16 +27,13 @@ Options:
|
|||||||
<y0> Y coordinate for the first marker [default: 0]
|
<y0> Y coordinate for the first marker [default: 0]
|
||||||
--top-left First marker is on top-left (default)
|
--top-left First marker is on top-left (default)
|
||||||
--bottom-left First marker is on bottom-left
|
--bottom-left First marker is on bottom-left
|
||||||
-o <filename> Output map file name in the 'map' subdirectory of aruco_pose package
|
|
||||||
|
|
||||||
Example:
|
Example:
|
||||||
rosrun aruco_pose genmap.py 0.33 2 4 1 1 0 -o test_map.txt
|
rosrun aruco_pose genmap.py 0.33 2 4 1 1 0 > $(catkin_find aruco_pose map)/test_map.txt
|
||||||
"""
|
"""
|
||||||
|
|
||||||
from __future__ import print_function
|
from __future__ import print_function
|
||||||
|
|
||||||
import sys
|
|
||||||
from os import path
|
|
||||||
from docopt import docopt
|
from docopt import docopt
|
||||||
|
|
||||||
|
|
||||||
@@ -52,19 +49,14 @@ dist_x = float(arguments['<dist_x>'])
|
|||||||
dist_y = float(arguments['<dist_y>'])
|
dist_y = float(arguments['<dist_y>'])
|
||||||
bottom_left = arguments['--bottom-left']
|
bottom_left = arguments['--bottom-left']
|
||||||
|
|
||||||
if arguments['-o'] is None:
|
|
||||||
output = sys.stdout
|
|
||||||
else:
|
|
||||||
output = open(path.join(path.dirname(__file__), '..', 'map', arguments['-o']), 'w')
|
|
||||||
|
|
||||||
max_y = y0 + (markers_y - 1) * dist_y
|
max_y = y0 + (markers_y - 1) * dist_y
|
||||||
|
|
||||||
output.write('# id\tlength\tx\ty\tz\trot_z\trot_y\trot_x\n')
|
print('# id\tlength\tx\ty\tz\trot_z\trot_y\trot_x')
|
||||||
for y in range(markers_y):
|
for y in range(markers_y):
|
||||||
for x in range(markers_x):
|
for x in range(markers_x):
|
||||||
pos_x = x0 + x * dist_x
|
pos_x = x0 + x * dist_x
|
||||||
pos_y = y0 + y * dist_y
|
pos_y = y0 + y * dist_y
|
||||||
if not bottom_left:
|
if not bottom_left:
|
||||||
pos_y = max_y - pos_y
|
pos_y = max_y - pos_y
|
||||||
output.write('{}\t{}\t{}\t{}\t{}\t{}\t{}\t{}\n'.format(first, length, pos_x, pos_y, 0, 0, 0, 0))
|
print('{}\t{}\t{}\t{}\t{}\t{}\t{}\t{}'.format(first, length, pos_x, pos_y, 0, 0, 0, 0))
|
||||||
first += 1
|
first += 1
|
||||||
|
|||||||
@@ -106,25 +106,26 @@ inline bool isFlipped(tf::Quaternion& q)
|
|||||||
return (abs(pitch) > M_PI / 2) || (abs(roll) > M_PI / 2);
|
return (abs(pitch) > M_PI / 2) || (abs(roll) > M_PI / 2);
|
||||||
}
|
}
|
||||||
|
|
||||||
/* Apply a vertical to an orientation */
|
/* Set roll and pitch from "from" to "to", keeping yaw */
|
||||||
inline void applyVertical(geometry_msgs::Quaternion& orientation, const geometry_msgs::Quaternion& vertical,
|
inline void snapOrientation(geometry_msgs::Quaternion& to, const geometry_msgs::Quaternion& from, bool auto_flip = false)
|
||||||
bool flip_vertical = false, bool auto_flip = false) // editorconfig-checker-disable-line
|
|
||||||
{
|
{
|
||||||
tf::Quaternion _vertical, _orientation;
|
tf::Quaternion _from, _to;
|
||||||
tf::quaternionMsgToTF(vertical, _vertical);
|
tf::quaternionMsgToTF(from, _from);
|
||||||
tf::quaternionMsgToTF(orientation, _orientation);
|
tf::quaternionMsgToTF(to, _to);
|
||||||
|
|
||||||
if (flip_vertical || (auto_flip && !isFlipped(_orientation))) {
|
if (auto_flip) {
|
||||||
static const tf::Quaternion flip = tf::createQuaternionFromRPY(M_PI, 0, 0);
|
if (!isFlipped(_from)) {
|
||||||
_vertical *= flip; // flip vertical
|
static const tf::Quaternion flip = tf::createQuaternionFromRPY(M_PI, 0, 0);
|
||||||
|
_from *= flip; // flip "from"
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
auto diff = tf::Matrix3x3(_orientation).transposeTimes(tf::Matrix3x3(_vertical));
|
auto diff = tf::Matrix3x3(_to).transposeTimes(tf::Matrix3x3(_from));
|
||||||
double _, yaw;
|
double _, yaw;
|
||||||
diff.getRPY(_, _, yaw);
|
diff.getRPY(_, _, yaw);
|
||||||
auto q = tf::createQuaternionFromRPY(0, 0, -yaw);
|
auto q = tf::createQuaternionFromRPY(0, 0, -yaw);
|
||||||
_vertical = _vertical * q; // set yaw from orientation to vertical
|
_from = _from * q; // set yaw from "to" to "from"
|
||||||
tf::quaternionTFToMsg(_vertical, orientation); // set vertical to orientation
|
tf::quaternionTFToMsg(_from, to); // set "from" to "to"
|
||||||
}
|
}
|
||||||
|
|
||||||
inline void transformToPose(const geometry_msgs::Transform& transform, geometry_msgs::Pose& pose)
|
inline void transformToPose(const geometry_msgs::Transform& transform, geometry_msgs::Pose& pose)
|
||||||
|
|||||||
@@ -1,7 +0,0 @@
|
|||||||
# * Add or change markers in the map
|
|
||||||
# * Change markers' properties, e. g. lengths
|
|
||||||
|
|
||||||
Marker[] markers # if length or pose is nan - remove from map
|
|
||||||
---
|
|
||||||
bool success
|
|
||||||
string message
|
|
||||||
@@ -6,7 +6,7 @@ import tf2_geometry_msgs
|
|||||||
from geometry_msgs.msg import PoseWithCovarianceStamped
|
from geometry_msgs.msg import PoseWithCovarianceStamped
|
||||||
from sensor_msgs.msg import Image
|
from sensor_msgs.msg import Image
|
||||||
from aruco_pose.msg import MarkerArray
|
from aruco_pose.msg import MarkerArray
|
||||||
from visualization_msgs.msg import MarkerArray as VisMarkerArray, Marker as VisMarker
|
from visualization_msgs.msg import MarkerArray as VisMarkerArray
|
||||||
|
|
||||||
|
|
||||||
@pytest.fixture
|
@pytest.fixture
|
||||||
@@ -143,7 +143,7 @@ def test_map_image(node):
|
|||||||
assert img.encoding in ('mono8', 'rgb8')
|
assert img.encoding in ('mono8', 'rgb8')
|
||||||
|
|
||||||
def test_map_markers(node):
|
def test_map_markers(node):
|
||||||
markers = rospy.wait_for_message('aruco_map/map', MarkerArray, timeout=5)
|
markers = rospy.wait_for_message('aruco_map/markers', MarkerArray, timeout=5)
|
||||||
assert markers.markers[0].id == 1
|
assert markers.markers[0].id == 1
|
||||||
assert markers.markers[1].id == 2
|
assert markers.markers[1].id == 2
|
||||||
assert markers.markers[2].id == 3
|
assert markers.markers[2].id == 3
|
||||||
@@ -199,36 +199,6 @@ def test_map_markers(node):
|
|||||||
|
|
||||||
def test_map_visualization(node):
|
def test_map_visualization(node):
|
||||||
vis = rospy.wait_for_message('aruco_map/visualization', VisMarkerArray, timeout=5)
|
vis = rospy.wait_for_message('aruco_map/visualization', VisMarkerArray, timeout=5)
|
||||||
assert len(vis.markers) == 7
|
|
||||||
assert vis.markers[0].header.frame_id == 'aruco_map'
|
|
||||||
assert vis.markers[0].type == VisMarker.CUBE
|
|
||||||
assert vis.markers[0].action == VisMarker.ADD
|
|
||||||
assert vis.markers[0].pose.position.x == 0
|
|
||||||
assert vis.markers[0].pose.position.y == 0
|
|
||||||
assert vis.markers[0].pose.position.z == 0
|
|
||||||
assert vis.markers[0].pose.orientation.x == 0
|
|
||||||
assert vis.markers[0].pose.orientation.y == 0
|
|
||||||
assert vis.markers[0].pose.orientation.z == 0
|
|
||||||
assert vis.markers[0].pose.orientation.w == 1
|
|
||||||
assert vis.markers[0].scale.x == approx(0.33)
|
|
||||||
assert vis.markers[0].scale.y == approx(0.33)
|
|
||||||
assert vis.markers[0].scale.z == approx(0.001)
|
|
||||||
assert vis.markers[1].pose.position.x == 1
|
|
||||||
assert vis.markers[1].pose.position.y == 0
|
|
||||||
assert vis.markers[1].pose.position.z == 0
|
|
||||||
assert vis.markers[1].pose.orientation.x == 0
|
|
||||||
assert vis.markers[1].pose.orientation.y == 0
|
|
||||||
assert vis.markers[1].pose.orientation.z == 0
|
|
||||||
assert vis.markers[1].pose.orientation.w == 1
|
|
||||||
# non-zero yaw marker:
|
|
||||||
assert vis.markers[4].scale.x == approx(0.5)
|
|
||||||
assert vis.markers[4].pose.position.x == approx(0.5)
|
|
||||||
assert vis.markers[4].pose.position.y == 2
|
|
||||||
assert vis.markers[4].pose.position.z == 0
|
|
||||||
assert vis.markers[4].pose.orientation.x == 0
|
|
||||||
assert vis.markers[4].pose.orientation.y == 0
|
|
||||||
assert vis.markers[4].pose.orientation.z == approx(0.5646424733950354)
|
|
||||||
assert vis.markers[4].pose.orientation.w == approx(0.8253356149096783)
|
|
||||||
|
|
||||||
def test_map_debug(node):
|
def test_map_debug(node):
|
||||||
img = rospy.wait_for_message('aruco_map/debug', Image, timeout=5)
|
img = rospy.wait_for_message('aruco_map/debug', Image, timeout=5)
|
||||||
|
|||||||
Binary file not shown.
|
Before Width: | Height: | Size: 62 KiB |
@@ -1,8 +0,0 @@
|
|||||||
import pytest
|
|
||||||
import subprocess
|
|
||||||
|
|
||||||
def test_no_tf_repeated_data():
|
|
||||||
# `/rosout` acts weirdly inside rostest, so using a subprocess
|
|
||||||
cmd = """python -c 'import rospy, tf; rospy.init_node("foo"); listener = tf.TransformListener(); rospy.sleep(2)'"""
|
|
||||||
output = str(subprocess.check_output(cmd, shell=True, stderr=subprocess.STDOUT))
|
|
||||||
assert 'TF_REPEATED_DATA' not in output, 'TF_REPEATED_DATA was logged on duplicate markers'
|
|
||||||
@@ -1,21 +0,0 @@
|
|||||||
<launch>
|
|
||||||
<node pkg="image_publisher" type="image_publisher" name="main_camera" args="$(find aruco_pose)/test/duplicate.png">
|
|
||||||
<param name="frame_id" value="main_camera_optical"/>
|
|
||||||
<param name="publish_rate" value="10"/>
|
|
||||||
<param name="camera_info_url" value="file://$(find aruco_pose)/test/camera_info.yaml" />
|
|
||||||
</node>
|
|
||||||
|
|
||||||
<node pkg="nodelet" type="nodelet" name="nodelet_manager" args="manager" required="true"/>
|
|
||||||
|
|
||||||
<node pkg="nodelet" clear_params="true" type="nodelet" name="aruco_detect" args="load aruco_pose/aruco_detect nodelet_manager" required="true">
|
|
||||||
<remap from="image_raw" to="main_camera/image_raw"/>
|
|
||||||
<remap from="camera_info" to="main_camera/camera_info"/>
|
|
||||||
<param name="length" value="0.33"/>
|
|
||||||
<param name="estimate_poses" value="true"/>
|
|
||||||
<param name="send_tf" value="true"/>
|
|
||||||
<param name="cornerRefinementMethod" value="1"/>
|
|
||||||
</node>
|
|
||||||
|
|
||||||
<param name="test_module" value="$(find aruco_pose)/test/duplicate.py"/>
|
|
||||||
<test test-name="aruco_pose_test" pkg="ros_pytest" type="ros_pytest_runner"/>
|
|
||||||
</launch>
|
|
||||||
@@ -2,7 +2,6 @@ import rospy
|
|||||||
import pytest
|
import pytest
|
||||||
|
|
||||||
from visualization_msgs.msg import MarkerArray as VisMarkerArray
|
from visualization_msgs.msg import MarkerArray as VisMarkerArray
|
||||||
from aruco_pose.msg import MarkerArray
|
|
||||||
|
|
||||||
|
|
||||||
@pytest.fixture
|
@pytest.fixture
|
||||||
@@ -10,5 +9,5 @@ def node():
|
|||||||
return rospy.init_node('aruco_pose_test', anonymous=True)
|
return rospy.init_node('aruco_pose_test', anonymous=True)
|
||||||
|
|
||||||
def test_node_failure(node):
|
def test_node_failure(node):
|
||||||
assert rospy.wait_for_message('aruco_map/visualization', VisMarkerArray, timeout=5).markers == []
|
with pytest.raises(rospy.exceptions.ROSException):
|
||||||
assert rospy.wait_for_message('aruco_map/map', MarkerArray, timeout=5).markers == []
|
rospy.wait_for_message('aruco_map/visualization', VisMarkerArray, timeout=5)
|
||||||
|
|||||||
3
aruco_pose/vendor/VendorOpenCV.cmake
vendored
3
aruco_pose/vendor/VendorOpenCV.cmake
vendored
@@ -7,7 +7,6 @@ endif()
|
|||||||
|
|
||||||
message(STATUS "Adding vendored aruco_pose OpenCV module")
|
message(STATUS "Adding vendored aruco_pose OpenCV module")
|
||||||
add_library(_opencv_aruco STATIC
|
add_library(_opencv_aruco STATIC
|
||||||
vendor/aruco/src/apriltag_quad_thresh.cpp
|
|
||||||
vendor/aruco/src/aruco.cpp
|
vendor/aruco/src/aruco.cpp
|
||||||
vendor/aruco/src/charuco.cpp
|
vendor/aruco/src/charuco.cpp
|
||||||
vendor/aruco/src/dictionary.cpp
|
vendor/aruco/src/dictionary.cpp
|
||||||
@@ -24,7 +23,7 @@ target_compile_definitions(_opencv_aruco PRIVATE
|
|||||||
CV_OVERRIDE=override
|
CV_OVERRIDE=override
|
||||||
)
|
)
|
||||||
target_compile_options(_opencv_aruco PRIVATE
|
target_compile_options(_opencv_aruco PRIVATE
|
||||||
-fpic -fPIC -fvisibility=hidden
|
-fpic -fPIC
|
||||||
)
|
)
|
||||||
|
|
||||||
target_include_directories(_opencv_aruco PUBLIC
|
target_include_directories(_opencv_aruco PUBLIC
|
||||||
|
|||||||
@@ -94,7 +94,7 @@ void ptsort_(struct pt *pts, int sz)
|
|||||||
|
|
||||||
// Use stack storage if it's not too big.
|
// Use stack storage if it's not too big.
|
||||||
cv::AutoBuffer<struct pt, 1024> _tmp_stack(sz);
|
cv::AutoBuffer<struct pt, 1024> _tmp_stack(sz);
|
||||||
memcpy(_tmp_stack, pts, sizeof(struct pt) * sz);
|
memcpy(_tmp_stack.data(), pts, sizeof(struct pt) * sz);
|
||||||
|
|
||||||
int asz = sz/2;
|
int asz = sz/2;
|
||||||
int bsz = sz - asz;
|
int bsz = sz - asz;
|
||||||
@@ -470,11 +470,11 @@ int quad_segment_agg(int sz, struct line_fit_pt *lfps, int indices[4]){
|
|||||||
int rvalloc_pos = 0;
|
int rvalloc_pos = 0;
|
||||||
int rvalloc_size = 3*sz;
|
int rvalloc_size = 3*sz;
|
||||||
cv::AutoBuffer<struct remove_vertex, 0> rvalloc_(std::max(1, rvalloc_size));
|
cv::AutoBuffer<struct remove_vertex, 0> rvalloc_(std::max(1, rvalloc_size));
|
||||||
memset(rvalloc_, 0, sizeof(rvalloc_[0]) * rvalloc_.size()); // TODO Add AutoBuffer zero fill
|
memset(rvalloc_.data(), 0, sizeof(rvalloc_[0]) * rvalloc_.size()); // TODO Add AutoBuffer zero fill
|
||||||
struct remove_vertex *rvalloc = rvalloc_;
|
struct remove_vertex *rvalloc = rvalloc_.data();
|
||||||
cv::AutoBuffer<struct segment, 0> segs_(std::max(1, sz)); // TODO Add AutoBuffer zero fill
|
cv::AutoBuffer<struct segment, 0> segs_(std::max(1, sz)); // TODO Add AutoBuffer zero fill
|
||||||
memset(segs_, 0, sizeof(segs_[0]) * segs_.size());
|
memset(segs_.data(), 0, sizeof(segs_[0]) * segs_.size());
|
||||||
struct segment *segs = segs_;
|
struct segment *segs = segs_.data();
|
||||||
|
|
||||||
// populate with initial entries
|
// populate with initial entries
|
||||||
for (int i = 0; i < sz; i++) {
|
for (int i = 0; i < sz; i++) {
|
||||||
@@ -753,8 +753,8 @@ int fit_quad(const Ptr<DetectorParameters> &_params, const Mat im, zarray_t *clu
|
|||||||
// efficiently computed for any contiguous range of indices.
|
// efficiently computed for any contiguous range of indices.
|
||||||
|
|
||||||
cv::AutoBuffer<struct line_fit_pt, 64> lfps_(sz);
|
cv::AutoBuffer<struct line_fit_pt, 64> lfps_(sz);
|
||||||
memset(lfps_, 0, sizeof(lfps_[0]) * lfps_.size()); // TODO Add AutoBuffer zero fill
|
memset(lfps_.data(), 0, sizeof(lfps_[0]) * lfps_.size()); // TODO Add AutoBuffer zero fill
|
||||||
struct line_fit_pt *lfps = lfps_;
|
struct line_fit_pt *lfps = lfps_.data();
|
||||||
|
|
||||||
for (int i = 0; i < sz; i++) {
|
for (int i = 0; i < sz; i++) {
|
||||||
struct pt *p;
|
struct pt *p;
|
||||||
|
|||||||
@@ -10,7 +10,7 @@
|
|||||||
"yametrika",
|
"yametrika",
|
||||||
"anchors",
|
"anchors",
|
||||||
"collapsible-menu",
|
"collapsible-menu",
|
||||||
"validate-links@https://github.com/okalachev/gitbook-plugin-validate-links.git",
|
"validate-links",
|
||||||
"bulk-redirect@https://github.com/okalachev/gitbook-plugin-bulk-redirect.git",
|
"bulk-redirect@https://github.com/okalachev/gitbook-plugin-bulk-redirect.git",
|
||||||
"sitemap@https://github.com/okalachev/plugin-sitemap.git",
|
"sitemap@https://github.com/okalachev/plugin-sitemap.git",
|
||||||
"toolbar@https://github.com/hamishwillee/gitbook-plugin-toolbar.git",
|
"toolbar@https://github.com/hamishwillee/gitbook-plugin-toolbar.git",
|
||||||
|
|||||||
@@ -1,4 +1,4 @@
|
|||||||
#!/usr/bin/env python3
|
#!/usr/bin/env python
|
||||||
|
|
||||||
from distutils.core import setup
|
from distutils.core import setup
|
||||||
|
|
||||||
|
|||||||
@@ -8,9 +8,5 @@ ExecStart=/bin/bash -c ". /home/pi/catkin_ws/devel/setup.sh; \
|
|||||||
ROS_HOSTNAME=`hostname`.local exec stdbuf -o L roslaunch clover clover.launch --wait --screen --skip-log-check \
|
ROS_HOSTNAME=`hostname`.local exec stdbuf -o L roslaunch clover clover.launch --wait --screen --skip-log-check \
|
||||||
2> >(tee /tmp/clover.err)"
|
2> >(tee /tmp/clover.err)"
|
||||||
|
|
||||||
ExecStartPre=+rm /var/log/clover.log
|
|
||||||
StandardOutput=file:/var/log/clover.log
|
|
||||||
StandardError=file:/var/log/clover.log
|
|
||||||
|
|
||||||
[Install]
|
[Install]
|
||||||
WantedBy=multi-user.target
|
WantedBy=multi-user.target
|
||||||
|
|||||||
@@ -15,17 +15,17 @@ set_attitude = rospy.ServiceProxy('set_attitude', srv.SetAttitude)
|
|||||||
set_rates = rospy.ServiceProxy('set_rates', srv.SetRates)
|
set_rates = rospy.ServiceProxy('set_rates', srv.SetRates)
|
||||||
land = rospy.ServiceProxy('land', Trigger)
|
land = rospy.ServiceProxy('land', Trigger)
|
||||||
|
|
||||||
print('Take off and hover 1 m above the ground')
|
# Take off and hover 1 m above the ground
|
||||||
navigate(x=0, y=0, z=1, frame_id='body', auto_arm=True)
|
navigate(x=0, y=0, z=1, frame_id='body', auto_arm=True)
|
||||||
|
|
||||||
# Wait for 5 seconds
|
# Wait for 3 seconds
|
||||||
rospy.sleep(5)
|
rospy.sleep(3)
|
||||||
|
|
||||||
print('Fly forward 1 m')
|
# Fly forward 1 m
|
||||||
navigate(x=1, y=0, z=0, frame_id='body')
|
navigate(x=1, y=0, z=0, frame_id='body')
|
||||||
|
|
||||||
# Wait for 5 seconds
|
# Wait for 3 seconds
|
||||||
rospy.sleep(5)
|
rospy.sleep(3)
|
||||||
|
|
||||||
print('Perform landing')
|
# Perform landing
|
||||||
land()
|
land()
|
||||||
@@ -15,23 +15,23 @@ set_attitude = rospy.ServiceProxy('set_attitude', srv.SetAttitude)
|
|||||||
set_rates = rospy.ServiceProxy('set_rates', srv.SetRates)
|
set_rates = rospy.ServiceProxy('set_rates', srv.SetRates)
|
||||||
land = rospy.ServiceProxy('land', Trigger)
|
land = rospy.ServiceProxy('land', Trigger)
|
||||||
|
|
||||||
print('Take off and hover 1 m above the ground')
|
# Take off and hover 1 m above the ground
|
||||||
navigate(x=0, y=0, z=1, frame_id='body', auto_arm=True)
|
navigate(x=0, y=0, z=1, frame_id='body', auto_arm=True)
|
||||||
|
|
||||||
# Wait for 5 seconds
|
# Wait for 3 seconds
|
||||||
rospy.sleep(5)
|
rospy.sleep(3)
|
||||||
|
|
||||||
print('Fly 1 meter above ArUco marker 0')
|
# Fly 1 meter above ArUco marker 0
|
||||||
navigate(x=0, y=0, z=1, frame_id='aruco_0')
|
navigate(x=0, y=0, z=1, frame_id='aruco_0')
|
||||||
|
|
||||||
# Wait for 5 seconds
|
# Wait for 3 seconds
|
||||||
rospy.sleep(5)
|
rospy.sleep(3)
|
||||||
|
|
||||||
print('Fly to x=1 y=1 z=1 relative to ArUco markers map')
|
# Fly to x=1 y=1 z=1 relative to ArUco markers map
|
||||||
navigate(x=1, y=1, z=1, frame_id='aruco_map')
|
navigate(x=1, y=1, z=1, frame_id='aruco_map')
|
||||||
|
|
||||||
# Wait for 5 seconds
|
# Wait for 3 seconds
|
||||||
rospy.sleep(5)
|
rospy.sleep(3)
|
||||||
|
|
||||||
print('Perform landing')
|
# Perform landing
|
||||||
land()
|
land()
|
||||||
@@ -16,8 +16,11 @@ set_attitude = rospy.ServiceProxy('set_attitude', srv.SetAttitude)
|
|||||||
set_rates = rospy.ServiceProxy('set_rates', srv.SetRates)
|
set_rates = rospy.ServiceProxy('set_rates', srv.SetRates)
|
||||||
land = rospy.ServiceProxy('land', Trigger)
|
land = rospy.ServiceProxy('land', Trigger)
|
||||||
|
|
||||||
def navigate_wait(x=0, y=0, z=0, yaw=math.nan, speed=0.5, frame_id='body', tolerance=0.2, auto_arm=False):
|
def navigate_wait(x=0, y=0, z=0, yaw=float('nan'), yaw_rate=0, speed=0.5, \
|
||||||
res = navigate(x=x, y=y, z=z, yaw=yaw, speed=speed, frame_id=frame_id, auto_arm=auto_arm)
|
frame_id='body', tolerance=0.2, auto_arm=False):
|
||||||
|
|
||||||
|
res = navigate(x=x, y=y, z=z, yaw=yaw, yaw_rate=yaw_rate, speed=speed, \
|
||||||
|
frame_id=frame_id, auto_arm=auto_arm)
|
||||||
|
|
||||||
if not res.success:
|
if not res.success:
|
||||||
return res
|
return res
|
||||||
@@ -28,11 +31,11 @@ def navigate_wait(x=0, y=0, z=0, yaw=math.nan, speed=0.5, frame_id='body', toler
|
|||||||
return res
|
return res
|
||||||
rospy.sleep(0.2)
|
rospy.sleep(0.2)
|
||||||
|
|
||||||
print('Take off 1 meter')
|
# Take off 1 meter
|
||||||
navigate_wait(z=1, frame_id='body', auto_arm=True)
|
navigate_wait(z=1, frame_id='body', auto_arm=True)
|
||||||
|
|
||||||
print('Fly forward 1 m')
|
# Fly forward 1 m
|
||||||
navigate_wait(x=1, frame_id='body')
|
navigate_wait(x=1, frame_id='body')
|
||||||
|
|
||||||
print('Land')
|
# Land
|
||||||
land()
|
land()
|
||||||
@@ -65,7 +65,7 @@ echo_stamp "#6 Turn on UART"
|
|||||||
# https://github.com/RPi-Distro/raspi-config/pull/75
|
# https://github.com/RPi-Distro/raspi-config/pull/75
|
||||||
/usr/bin/raspi-config nonint do_serial 1
|
/usr/bin/raspi-config nonint do_serial 1
|
||||||
/usr/bin/raspi-config nonint set_config_var enable_uart 1 /boot/config.txt
|
/usr/bin/raspi-config nonint set_config_var enable_uart 1 /boot/config.txt
|
||||||
echo dtoverlay=pi3-disable-bt >> /boot/config.txt
|
/usr/bin/raspi-config nonint set_config_var dtoverlay pi3-disable-bt /boot/config.txt
|
||||||
systemctl disable hciuart.service
|
systemctl disable hciuart.service
|
||||||
|
|
||||||
# After adding to Raspbian OS
|
# After adding to Raspbian OS
|
||||||
|
|||||||
@@ -1,9 +0,0 @@
|
|||||||
#!/usr/bin/env bash
|
|
||||||
|
|
||||||
# GitBook CLI is deprecated, its installation is broken.
|
|
||||||
# This script fixes it until we stop using GitBook.
|
|
||||||
|
|
||||||
export NPM_CONFIG_UNSAFE_PERM=1
|
|
||||||
|
|
||||||
npm install gitbook-cli -g
|
|
||||||
gitbook fetch 3.2.3 && npm i npm@3.10.10 --prefix=~/.gitbook/versions/3.2.3/ # fixing https://travis-ci.org/github/CopterExpress/clover/jobs/766541125#L932
|
|
||||||
@@ -1,18 +0,0 @@
|
|||||||
async_web_server_cpp:
|
|
||||||
debian:
|
|
||||||
buster: [ros-noetic-async-web-server-cpp]
|
|
||||||
led_msgs:
|
|
||||||
debian:
|
|
||||||
buster: [ros-noetic-led-msgs]
|
|
||||||
ros_pytest:
|
|
||||||
debian:
|
|
||||||
buster: [ros-noetic-ros-pytest]
|
|
||||||
tf2_web_republisher:
|
|
||||||
debian:
|
|
||||||
buster: [ros-noetic-tf2-web-republisher]
|
|
||||||
web_video_server:
|
|
||||||
debian:
|
|
||||||
buster: [ros-noetic-web-video-server]
|
|
||||||
ws281x:
|
|
||||||
debian:
|
|
||||||
buster: [ros-noetic-ws281x]
|
|
||||||
@@ -3,7 +3,7 @@ Description=Launcher for the ROS master, parameter server and rosout logging nod
|
|||||||
|
|
||||||
[Service]
|
[Service]
|
||||||
User=pi
|
User=pi
|
||||||
ExecStart=/bin/sh -c ". /opt/ros/noetic/setup.sh; ROS_HOSTNAME=`hostname`.local exec roscore"
|
ExecStart=/bin/sh -c ". /opt/ros/melodic/setup.sh; ROS_HOSTNAME=`hostname`.local exec roscore"
|
||||||
Restart=on-failure
|
Restart=on-failure
|
||||||
RestartSec=3
|
RestartSec=3
|
||||||
|
|
||||||
|
|||||||
@@ -15,8 +15,7 @@
|
|||||||
|
|
||||||
set -e # Exit immidiately on non-zero result
|
set -e # Exit immidiately on non-zero result
|
||||||
|
|
||||||
# https://www.raspberrypi.org/software/operating-systems/#raspberry-pi-os-32-bit
|
SOURCE_IMAGE="https://downloads.raspberrypi.org/raspios_lite_armhf/images/raspios_lite_armhf-2021-01-12/2021-01-11-raspios-buster-armhf-lite.zip"
|
||||||
SOURCE_IMAGE="https://downloads.raspberrypi.org/raspios_lite_armhf/images/raspios_lite_armhf-2021-05-28/2021-05-07-raspios-buster-armhf-lite.zip"
|
|
||||||
|
|
||||||
export DEBIAN_FRONTEND=${DEBIAN_FRONTEND:='noninteractive'}
|
export DEBIAN_FRONTEND=${DEBIAN_FRONTEND:='noninteractive'}
|
||||||
export LANG=${LANG:='C.UTF-8'}
|
export LANG=${LANG:='C.UTF-8'}
|
||||||
@@ -105,6 +104,8 @@ ${BUILDER_DIR}/image-chroot.sh ${IMAGE_PATH} copy ${SCRIPTS_DIR}'/assets/butterf
|
|||||||
${BUILDER_DIR}/image-chroot.sh ${IMAGE_PATH} copy ${SCRIPTS_DIR}'/assets/monkey.service' '/lib/systemd/system/'
|
${BUILDER_DIR}/image-chroot.sh ${IMAGE_PATH} copy ${SCRIPTS_DIR}'/assets/monkey.service' '/lib/systemd/system/'
|
||||||
# software install
|
# software install
|
||||||
${BUILDER_DIR}/image-chroot.sh ${IMAGE_PATH} exec ${SCRIPTS_DIR}'/image-software.sh'
|
${BUILDER_DIR}/image-chroot.sh ${IMAGE_PATH} exec ${SCRIPTS_DIR}'/image-software.sh'
|
||||||
|
# examples
|
||||||
|
${BUILDER_DIR}/image-chroot.sh ${IMAGE_PATH} copy ${SCRIPTS_DIR}'/assets/examples' '/home/pi/' # TODO: symlink?
|
||||||
# network setup
|
# network setup
|
||||||
${BUILDER_DIR}/image-chroot.sh ${IMAGE_PATH} exec ${SCRIPTS_DIR}'/image-network.sh'
|
${BUILDER_DIR}/image-chroot.sh ${IMAGE_PATH} exec ${SCRIPTS_DIR}'/image-network.sh'
|
||||||
# avahi setup
|
# avahi setup
|
||||||
@@ -113,11 +114,15 @@ ${BUILDER_DIR}/image-chroot.sh ${IMAGE_PATH} copy ${SCRIPTS_DIR}'/assets/avahi-s
|
|||||||
# If RPi then use a one thread to build a ROS package on RPi, else use all
|
# If RPi then use a one thread to build a ROS package on RPi, else use all
|
||||||
[[ $(arch) == 'armv7l' ]] && NUMBER_THREADS=1 || NUMBER_THREADS=$(nproc --all)
|
[[ $(arch) == 'armv7l' ]] && NUMBER_THREADS=1 || NUMBER_THREADS=$(nproc --all)
|
||||||
# Clover
|
# Clover
|
||||||
${BUILDER_DIR}/image-chroot.sh ${IMAGE_PATH} copy ${SCRIPTS_DIR}'/assets/noetic-rosdep-clover.yaml' '/etc/ros/rosdep/'
|
${BUILDER_DIR}/image-chroot.sh ${IMAGE_PATH} copy ${SCRIPTS_DIR}'/assets/clover.service' '/lib/systemd/system/'
|
||||||
|
${BUILDER_DIR}/image-chroot.sh ${IMAGE_PATH} copy ${SCRIPTS_DIR}'/assets/roscore.service' '/lib/systemd/system/'
|
||||||
|
${BUILDER_DIR}/image-chroot.sh ${IMAGE_PATH} copy ${SCRIPTS_DIR}'/assets/melodic-rosdep-clover.yaml' '/etc/ros/rosdep/'
|
||||||
${BUILDER_DIR}/image-chroot.sh ${IMAGE_PATH} copy ${SCRIPTS_DIR}'/assets/ros_python_paths' '/etc/sudoers.d/'
|
${BUILDER_DIR}/image-chroot.sh ${IMAGE_PATH} copy ${SCRIPTS_DIR}'/assets/ros_python_paths' '/etc/sudoers.d/'
|
||||||
${BUILDER_DIR}/image-chroot.sh ${IMAGE_PATH} copy ${SCRIPTS_DIR}'/assets/pigpiod.service' '/lib/systemd/system/'
|
${BUILDER_DIR}/image-chroot.sh ${IMAGE_PATH} copy ${SCRIPTS_DIR}'/assets/pigpiod.service' '/lib/systemd/system/'
|
||||||
${BUILDER_DIR}/image-chroot.sh ${IMAGE_PATH} copy ${SCRIPTS_DIR}'/assets/launch.nanorc' '/usr/share/nano/'
|
${BUILDER_DIR}/image-chroot.sh ${IMAGE_PATH} copy ${SCRIPTS_DIR}'/assets/launch.nanorc' '/usr/share/nano/'
|
||||||
# ${BUILDER_DIR}/image-chroot.sh ${IMAGE_PATH} copy ${SCRIPTS_DIR}'/assets/kinetic-ros-clover.rosinstall' '/home/pi/ros_catkin_ws/'
|
# ${BUILDER_DIR}/image-chroot.sh ${IMAGE_PATH} copy ${SCRIPTS_DIR}'/assets/kinetic-ros-clover.rosinstall' '/home/pi/ros_catkin_ws/'
|
||||||
|
# Add PX4 udev rules
|
||||||
|
${BUILDER_DIR}/image-chroot.sh ${IMAGE_PATH} copy ${REPO_DIR}'/clover/config/99-px4fmu.rules' '/lib/udev/rules.d/'
|
||||||
# Add rename script
|
# Add rename script
|
||||||
${BUILDER_DIR}/image-chroot.sh ${IMAGE_PATH} exec ${SCRIPTS_DIR}'/image-ros.sh' ${REPO_URL} ${IMAGE_VERSION} false false ${NUMBER_THREADS}
|
${BUILDER_DIR}/image-chroot.sh ${IMAGE_PATH} exec ${SCRIPTS_DIR}'/image-ros.sh' ${REPO_URL} ${IMAGE_VERSION} false false ${NUMBER_THREADS}
|
||||||
${BUILDER_DIR}/image-chroot.sh ${IMAGE_PATH} exec ${SCRIPTS_DIR}'/image-validate.sh'
|
${BUILDER_DIR}/image-chroot.sh ${IMAGE_PATH} exec ${SCRIPTS_DIR}'/image-validate.sh'
|
||||||
|
|||||||
@@ -13,7 +13,7 @@
|
|||||||
# copies or substantial portions of the Software.
|
# copies or substantial portions of the Software.
|
||||||
#
|
#
|
||||||
|
|
||||||
set -ex # exit on error, echo commands
|
set -ex # Exit immidiately on non-zero result
|
||||||
|
|
||||||
REPO=$1
|
REPO=$1
|
||||||
REF=$2
|
REF=$2
|
||||||
@@ -21,9 +21,6 @@ INSTALL_ROS_PACK_SOURCES=$3
|
|||||||
DISCOVER_ROS_PACK=$4
|
DISCOVER_ROS_PACK=$4
|
||||||
NUMBER_THREADS=$5
|
NUMBER_THREADS=$5
|
||||||
|
|
||||||
# Current ROS distribution
|
|
||||||
ROS_DISTRO=noetic
|
|
||||||
|
|
||||||
echo_stamp() {
|
echo_stamp() {
|
||||||
# TEMPLATE: echo_stamp <TEXT> <TYPE>
|
# TEMPLATE: echo_stamp <TEXT> <TYPE>
|
||||||
# TYPE: SUCCESS, ERROR, INFO
|
# TYPE: SUCCESS, ERROR, INFO
|
||||||
@@ -71,8 +68,7 @@ my_travis_retry() {
|
|||||||
# TODO: 'kinetic-rosdep-clover.yaml' should add only if we use our repo?
|
# TODO: 'kinetic-rosdep-clover.yaml' should add only if we use our repo?
|
||||||
echo_stamp "Init rosdep"
|
echo_stamp "Init rosdep"
|
||||||
my_travis_retry rosdep init
|
my_travis_retry rosdep init
|
||||||
# FIXME: Re-add this after missing packages are built
|
echo "yaml file:///etc/ros/rosdep/melodic-rosdep-clover.yaml" >> /etc/ros/rosdep/sources.list.d/20-default.list
|
||||||
echo "yaml file:///etc/ros/rosdep/${ROS_DISTRO}-rosdep-clover.yaml" >> /etc/ros/rosdep/sources.list.d/20-default.list
|
|
||||||
my_travis_retry rosdep update
|
my_travis_retry rosdep update
|
||||||
|
|
||||||
echo_stamp "Populate rosdep for ROS user"
|
echo_stamp "Populate rosdep for ROS user"
|
||||||
@@ -80,40 +76,20 @@ my_travis_retry sudo -u pi rosdep update
|
|||||||
|
|
||||||
export ROS_IP='127.0.0.1' # needed for running tests
|
export ROS_IP='127.0.0.1' # needed for running tests
|
||||||
|
|
||||||
# echo_stamp "Reconfiguring Clover repository for simplier unshallowing"
|
echo_stamp "Reconfiguring Clover repository for simplier unshallowing"
|
||||||
cd /home/pi/catkin_ws/src/clover
|
cd /home/pi/catkin_ws/src/clover
|
||||||
git config remote.origin.fetch "+refs/heads/*:refs/remotes/origin/*"
|
git config remote.origin.fetch "+refs/heads/*:refs/remotes/origin/*"
|
||||||
|
|
||||||
# This is sort of a hack to force "custom" packages to be installed - the ones built by COEX, linked against OpenCV 4.2
|
|
||||||
# I **wish** OpenCV would not be such a mess, but, well, here we are.
|
|
||||||
echo_stamp "Installing OpenCV 4.2-compatible ROS packages"
|
|
||||||
apt install -y --no-install-recommends \
|
|
||||||
ros-${ROS_DISTRO}-compressed-image-transport=1.14.0-0buster \
|
|
||||||
ros-${ROS_DISTRO}-cv-bridge=1.15.0-0buster \
|
|
||||||
ros-${ROS_DISTRO}-cv-camera=0.5.1-0buster \
|
|
||||||
ros-${ROS_DISTRO}-image-publisher=1.15.3-0buster \
|
|
||||||
ros-${ROS_DISTRO}-web-video-server=0.2.1-0buster
|
|
||||||
apt-mark hold \
|
|
||||||
ros-${ROS_DISTRO}-compressed-image-transport \
|
|
||||||
ros-${ROS_DISTRO}-cv-bridge \
|
|
||||||
ros-${ROS_DISTRO}-cv-camera \
|
|
||||||
ros-${ROS_DISTRO}-image-publisher \
|
|
||||||
ros-${ROS_DISTRO}-web-video-server
|
|
||||||
|
|
||||||
echo_stamp "Installing libboost-dev" # https://travis-ci.org/github/CopterExpress/clover/jobs/766318908#L6536
|
|
||||||
my_travis_retry apt-get install -y --no-install-recommends libboost-dev libboost-all-dev
|
|
||||||
|
|
||||||
echo_stamp "Build and install Clover"
|
echo_stamp "Build and install Clover"
|
||||||
cd /home/pi/catkin_ws
|
cd /home/pi/catkin_ws
|
||||||
# Don't try to install gazebo_ros
|
# Don't try to install gazebo_ros
|
||||||
my_travis_retry rosdep install -y --from-paths src --ignore-src --rosdistro ${ROS_DISTRO} --os=debian:buster \
|
my_travis_retry rosdep install -y --from-paths src --ignore-src --rosdistro melodic --os=debian:buster \
|
||||||
--skip-keys=gazebo_ros --skip-keys=gazebo_plugins
|
--skip-keys=gazebo_ros --skip-keys=gazebo_plugins
|
||||||
my_travis_retry pip3 install wheel
|
my_travis_retry pip install wheel
|
||||||
my_travis_retry pip3 install -r /home/pi/catkin_ws/src/clover/clover/requirements.txt
|
my_travis_retry pip install -r /home/pi/catkin_ws/src/clover/clover/requirements.txt
|
||||||
source /opt/ros/${ROS_DISTRO}/setup.bash
|
source /opt/ros/melodic/setup.bash
|
||||||
# Don't build simulation plugins for actual drone
|
# Don't build simulation plugins for actual drone
|
||||||
catkin_make -j2 -DCMAKE_BUILD_TYPE=RelWithDebInfo
|
catkin_make -j2 -DCMAKE_BUILD_TYPE=Release -DCATKIN_BLACKLIST_PACKAGES=clover_gazebo_plugins
|
||||||
source devel/setup.bash
|
|
||||||
|
|
||||||
echo_stamp "Install clever package (for backwards compatibility)"
|
echo_stamp "Install clever package (for backwards compatibility)"
|
||||||
cd /home/pi/catkin_ws/src/clover/builder/assets/clever
|
cd /home/pi/catkin_ws/src/clover/builder/assets/clever
|
||||||
@@ -122,27 +98,30 @@ rm -rf build # remove build artifacts
|
|||||||
|
|
||||||
echo_stamp "Build Clover documentation"
|
echo_stamp "Build Clover documentation"
|
||||||
cd /home/pi/catkin_ws/src/clover
|
cd /home/pi/catkin_ws/src/clover
|
||||||
builder/assets/install_gitbook.sh
|
NPM_CONFIG_UNSAFE_PERM=true npm install gitbook-cli -g
|
||||||
gitbook install
|
NPM_CONFIG_UNSAFE_PERM=true gitbook install
|
||||||
gitbook build
|
gitbook build
|
||||||
touch node_modules/CATKIN_IGNORE docs/CATKIN_IGNORE _book/CATKIN_IGNORE clover/www/CATKIN_IGNORE apps/CATKIN_IGNORE # ignore documentation files by catkin
|
touch node_modules/CATKIN_IGNORE docs/CATKIN_IGNORE _book/CATKIN_IGNORE clover/www/CATKIN_IGNORE apps/CATKIN_IGNORE # ignore documentation files by catkin
|
||||||
|
|
||||||
echo_stamp "Installing additional ROS packages"
|
echo_stamp "Installing additional ROS packages"
|
||||||
my_travis_retry apt-get install -y --no-install-recommends \
|
my_travis_retry apt-get install -y --no-install-recommends \
|
||||||
ros-${ROS_DISTRO}-dynamic-reconfigure \
|
ros-melodic-dynamic-reconfigure \
|
||||||
ros-${ROS_DISTRO}-rosbridge-suite \
|
ros-melodic-compressed-image-transport \
|
||||||
ros-${ROS_DISTRO}-rosserial \
|
ros-melodic-rosbridge-suite \
|
||||||
ros-${ROS_DISTRO}-usb-cam \
|
ros-melodic-rosserial \
|
||||||
ros-${ROS_DISTRO}-vl53l1x \
|
ros-melodic-usb-cam \
|
||||||
ros-${ROS_DISTRO}-ws281x \
|
ros-melodic-vl53l1x \
|
||||||
ros-${ROS_DISTRO}-rosshow \
|
ros-melodic-ws281x \
|
||||||
ros-${ROS_DISTRO}-cmake-modules \
|
ros-melodic-rosshow
|
||||||
ros-${ROS_DISTRO}-image-view
|
|
||||||
|
|
||||||
# TODO move GeographicLib datasets to Mavros debian package
|
# TODO move GeographicLib datasets to Mavros debian package
|
||||||
echo_stamp "Install GeographicLib datasets (needed for mavros)" \
|
echo_stamp "Install GeographicLib datasets (needed for mavros)" \
|
||||||
&& wget -qO- https://raw.githubusercontent.com/mavlink/mavros/master/mavros/scripts/install_geographiclib_datasets.sh | bash
|
&& wget -qO- https://raw.githubusercontent.com/mavlink/mavros/master/mavros/scripts/install_geographiclib_datasets.sh | bash
|
||||||
|
|
||||||
|
# FIXME: Buster comes with tornado==5.1.1 but we need tornado==4.2.1 for rosbridge_suite
|
||||||
|
# (note that Python 3 will still have a more recent version)
|
||||||
|
pip install tornado==4.2.1
|
||||||
|
|
||||||
echo_stamp "Running tests"
|
echo_stamp "Running tests"
|
||||||
cd /home/pi/catkin_ws
|
cd /home/pi/catkin_ws
|
||||||
# FIXME: Investigate failing tests
|
# FIXME: Investigate failing tests
|
||||||
@@ -151,29 +130,15 @@ catkin_make run_tests #&& catkin_test_results
|
|||||||
echo_stamp "Change permissions for catkin_ws"
|
echo_stamp "Change permissions for catkin_ws"
|
||||||
chown -Rf pi:pi /home/pi/catkin_ws
|
chown -Rf pi:pi /home/pi/catkin_ws
|
||||||
|
|
||||||
echo_stamp "Update www"
|
echo_stamp "Change permissions for examples"
|
||||||
sudo -u pi sh -c ". devel/setup.sh && rosrun clover www"
|
|
||||||
|
|
||||||
echo_stamp "Make \$HOME/examples symlink"
|
|
||||||
ln -s "$(catkin_find clover examples --first-only)" /home/pi
|
|
||||||
chown -Rf pi:pi /home/pi/examples
|
chown -Rf pi:pi /home/pi/examples
|
||||||
|
|
||||||
echo_stamp "Make systemd services symlinks"
|
|
||||||
ln -s /home/pi/catkin_ws/src/clover/builder/assets/clover.service /lib/systemd/system/
|
|
||||||
ln -s /home/pi/catkin_ws/src/clover/builder/assets/roscore.service /lib/systemd/system/
|
|
||||||
# validate
|
|
||||||
[ -f /lib/systemd/system/clover.service ]
|
|
||||||
[ -f /lib/systemd/system/roscore.service ]
|
|
||||||
|
|
||||||
echo_stamp "Make udev rules symlink"
|
|
||||||
ln -s "$(catkin_find clover udev --first-only)"/* /lib/udev/rules.d/
|
|
||||||
|
|
||||||
echo_stamp "Setup ROS environment"
|
echo_stamp "Setup ROS environment"
|
||||||
cat << EOF >> /home/pi/.bashrc
|
cat << EOF >> /home/pi/.bashrc
|
||||||
LANG='C.UTF-8'
|
LANG='C.UTF-8'
|
||||||
LC_ALL='C.UTF-8'
|
LC_ALL='C.UTF-8'
|
||||||
export ROS_HOSTNAME=\`hostname\`.local
|
export ROS_HOSTNAME=\`hostname\`.local
|
||||||
source /opt/ros/${ROS_DISTRO}/setup.bash
|
source /opt/ros/melodic/setup.bash
|
||||||
source /home/pi/catkin_ws/devel/setup.bash
|
source /home/pi/catkin_ws/devel/setup.bash
|
||||||
EOF
|
EOF
|
||||||
|
|
||||||
|
|||||||
@@ -64,14 +64,15 @@ echo "APT::Acquire::Retries \"3\";" > /etc/apt/apt.conf.d/80-retries
|
|||||||
echo_stamp "Install apt keys & repos"
|
echo_stamp "Install apt keys & repos"
|
||||||
|
|
||||||
# TODO: This STDOUT consist 'OK'
|
# TODO: This STDOUT consist 'OK'
|
||||||
|
curl http://deb.coex.tech/aptly_repo_signing.key 2> /dev/null | apt-key add -
|
||||||
apt-get update \
|
apt-get update \
|
||||||
&& apt-get install --no-install-recommends -y dirmngr > /dev/null \
|
&& apt-get install --no-install-recommends -y dirmngr > /dev/null \
|
||||||
&& apt-key adv --keyserver hkp://keyserver.ubuntu.com:80 --recv-key C1CF6E31E6BADE8868B172B4F42ED6FBAB17C654
|
&& apt-key adv --keyserver hkp://keyserver.ubuntu.com:80 --recv-key C1CF6E31E6BADE8868B172B4F42ED6FBAB17C654
|
||||||
|
|
||||||
echo "deb http://packages.ros.org/ros/ubuntu buster main" > /etc/apt/sources.list.d/ros-latest.list
|
echo "deb http://packages.ros.org/ros/ubuntu buster main" > /etc/apt/sources.list.d/ros-latest.list
|
||||||
|
echo "deb http://deb.coex.tech/opencv3 buster main" > /etc/apt/sources.list.d/opencv3.list
|
||||||
wget -O - 'http://packages.coex.tech/key.asc' | apt-key add -
|
echo "deb http://deb.coex.tech/rpi-ros-melodic buster main" > /etc/apt/sources.list.d/rpi-ros-melodic.list
|
||||||
echo 'deb http://packages.coex.tech buster main' >> /etc/apt/sources.list
|
echo "deb http://deb.coex.tech/clover buster main" > /etc/apt/sources.list.d/clover.list
|
||||||
|
|
||||||
echo_stamp "Update apt cache"
|
echo_stamp "Update apt cache"
|
||||||
|
|
||||||
@@ -81,7 +82,9 @@ apt-get update
|
|||||||
|
|
||||||
# Let's retry fetching those packages several times, just in case
|
# Let's retry fetching those packages several times, just in case
|
||||||
echo_stamp "Software installing"
|
echo_stamp "Software installing"
|
||||||
my_travis_retry apt-get install --no-install-recommends -y cmake-data=3.13.4-1 cmake=3.13.4-1 # FIXME: using older CMake due to https://travis-ci.org/github/CopterExpress/clover/jobs/764367665#L6984
|
apt-cache policy cmake
|
||||||
|
apt-cache policy cmake-data
|
||||||
|
my_travis_retry apt-get install --no-install-recommends -y cmake-data=3.13.4-1 cmake=3.13.4-1
|
||||||
my_travis_retry apt-get install --no-install-recommends -y \
|
my_travis_retry apt-get install --no-install-recommends -y \
|
||||||
unzip \
|
unzip \
|
||||||
zip \
|
zip \
|
||||||
@@ -94,22 +97,21 @@ lsof \
|
|||||||
git \
|
git \
|
||||||
dnsmasq \
|
dnsmasq \
|
||||||
tmux \
|
tmux \
|
||||||
tree \
|
|
||||||
vim \
|
vim \
|
||||||
libjpeg8 \
|
libjpeg8 \
|
||||||
tcpdump \
|
tcpdump \
|
||||||
|
ltrace \
|
||||||
libpoco-dev \
|
libpoco-dev \
|
||||||
libzbar0 \
|
libzbar0 \
|
||||||
python3-rosdep \
|
python-rosdep \
|
||||||
python3-rosinstall-generator \
|
python-rosinstall-generator \
|
||||||
python3-wstool \
|
python-wstool \
|
||||||
python3-rosinstall \
|
python-rosinstall \
|
||||||
build-essential \
|
build-essential \
|
||||||
libffi-dev \
|
libffi-dev \
|
||||||
monkey \
|
monkey \
|
||||||
pigpio python-pigpio python3-pigpio \
|
pigpio python-pigpio python3-pigpio \
|
||||||
i2c-tools \
|
i2c-tools \
|
||||||
espeak espeak-data python-espeak python3-espeak \
|
|
||||||
ntpdate \
|
ntpdate \
|
||||||
python-dev \
|
python-dev \
|
||||||
python3-dev \
|
python3-dev \
|
||||||
@@ -127,7 +129,7 @@ curl https://bootstrap.pypa.io/pip/2.7/get-pip.py -o get-pip2.py
|
|||||||
python3 get-pip.py
|
python3 get-pip.py
|
||||||
python get-pip2.py
|
python get-pip2.py
|
||||||
rm get-pip.py get-pip2.py
|
rm get-pip.py get-pip2.py
|
||||||
#my_travis_retry pip install --upgrade pip
|
my_travis_retry pip install --upgrade pip
|
||||||
#my_travis_retry pip3 install --upgrade pip
|
#my_travis_retry pip3 install --upgrade pip
|
||||||
|
|
||||||
echo_stamp "Make sure both pip and pip3 are installed"
|
echo_stamp "Make sure both pip and pip3 are installed"
|
||||||
@@ -137,25 +139,22 @@ pip3 --version
|
|||||||
echo_stamp "Install and enable Butterfly (web terminal)"
|
echo_stamp "Install and enable Butterfly (web terminal)"
|
||||||
echo_stamp "Workaround for tornado >= 6.0 breaking butterfly"
|
echo_stamp "Workaround for tornado >= 6.0 breaking butterfly"
|
||||||
export CRYPTOGRAPHY_DONT_BUILD_RUST=1
|
export CRYPTOGRAPHY_DONT_BUILD_RUST=1
|
||||||
my_travis_retry pip3 install cryptography==3.4.6 # https://stackoverflow.com/a/68472128/6850197
|
|
||||||
my_travis_retry pip3 install pyOpenSSL==20.0.1
|
|
||||||
my_travis_retry pip3 install tornado==5.1.1
|
my_travis_retry pip3 install tornado==5.1.1
|
||||||
my_travis_retry pip3 install butterfly
|
my_travis_retry pip3 install butterfly
|
||||||
my_travis_retry pip3 install butterfly[systemd]
|
my_travis_retry pip3 install butterfly[systemd]
|
||||||
systemctl enable butterfly.socket
|
systemctl enable butterfly.socket
|
||||||
|
|
||||||
echo_stamp "Install ws281x library"
|
echo_stamp "Install ws281x library"
|
||||||
my_travis_retry pip3 install --prefer-binary rpi_ws281x
|
my_travis_retry pip install --prefer-binary rpi_ws281x
|
||||||
|
|
||||||
echo_stamp "Setup Monkey"
|
echo_stamp "Setup Monkey"
|
||||||
mv /etc/monkey/sites/default /etc/monkey/sites/default.orig
|
mv /etc/monkey/sites/default /etc/monkey/sites/default.orig
|
||||||
mv /root/monkey /etc/monkey/sites/default
|
mv /root/monkey /etc/monkey/sites/default
|
||||||
sed -i 's/SymLink Off/SymLink On/' /etc/monkey/monkey.conf
|
|
||||||
systemctl enable monkey.service
|
systemctl enable monkey.service
|
||||||
|
|
||||||
echo_stamp "Install Node.js"
|
echo_stamp "Install Node.js"
|
||||||
cd /home/pi
|
cd /home/pi
|
||||||
wget --no-verbose https://nodejs.org/dist/v10.15.0/node-v10.15.0-linux-armv6l.tar.gz
|
wget https://nodejs.org/dist/v10.15.0/node-v10.15.0-linux-armv6l.tar.gz
|
||||||
tar -xzf node-v10.15.0-linux-armv6l.tar.gz
|
tar -xzf node-v10.15.0-linux-armv6l.tar.gz
|
||||||
cp -R node-v10.15.0-linux-armv6l/* /usr/local/
|
cp -R node-v10.15.0-linux-armv6l/* /usr/local/
|
||||||
rm -rf node-v10.15.0-linux-armv6l/
|
rm -rf node-v10.15.0-linux-armv6l/
|
||||||
|
|||||||
@@ -16,24 +16,16 @@ set -ex
|
|||||||
|
|
||||||
echo "Run image tests"
|
echo "Run image tests"
|
||||||
|
|
||||||
export ROS_DISTRO='noetic'
|
export ROS_DISTRO='melodic'
|
||||||
export ROS_IP='127.0.0.1'
|
export ROS_IP='127.0.0.1'
|
||||||
source /opt/ros/${ROS_DISTRO}/setup.bash
|
source /opt/ros/melodic/setup.bash
|
||||||
source /home/pi/catkin_ws/devel/setup.bash
|
source /home/pi/catkin_ws/devel/setup.bash
|
||||||
systemctl start roscore
|
|
||||||
|
|
||||||
cd /home/pi/catkin_ws/src/clover/builder/test/
|
cd /home/pi/catkin_ws/src/clover/builder/test/
|
||||||
./tests.sh
|
./tests.sh
|
||||||
./tests.py
|
./tests.py
|
||||||
./tests_py3.py
|
./tests_py3.py
|
||||||
[[ $(./test_qr.py) == "Found QRCODE with data Проверка Unicode with center at x=66.0, y=66.0" ]]
|
|
||||||
[[ $(./tests_clever.py) == "Warning: clever package is renamed to clover" ]] # test backwards compatibility
|
[[ $(./tests_clever.py) == "Warning: clever package is renamed to clover" ]] # test backwards compatibility
|
||||||
|
|
||||||
systemctl stop roscore
|
|
||||||
|
|
||||||
# check documented packages available
|
|
||||||
apt-cache show gst-rtsp-launch
|
|
||||||
apt-cache show openvpn
|
|
||||||
|
|
||||||
echo "Move /etc/ld.so.preload back to its original position"
|
echo "Move /etc/ld.so.preload back to its original position"
|
||||||
mv /etc/ld.so.preload.disabled-for-build /etc/ld.so.preload
|
mv /etc/ld.so.preload.disabled-for-build /etc/ld.so.preload
|
||||||
|
|||||||
@@ -3,45 +3,25 @@
|
|||||||
# Perform a "standalone install" in a Docker container
|
# Perform a "standalone install" in a Docker container
|
||||||
set -e
|
set -e
|
||||||
# Step 1: Install pip
|
# Step 1: Install pip
|
||||||
apt-key adv --keyserver 'hkp://keyserver.ubuntu.com:80' --recv-key C1CF6E31E6BADE8868B172B4F42ED6FBAB17C654 # https://github.com/osrf/docker_images/issues/535
|
apt update
|
||||||
apt-get update
|
apt install -y curl
|
||||||
apt-get install -y curl
|
curl https://bootstrap.pypa.io/pip/2.7/get-pip.py -o get-pip.py
|
||||||
if [ "x${ROS_PYTHON_VERSION}" = "x3" ]; then
|
python ./get-pip.py
|
||||||
PYTHON=python3
|
|
||||||
curl https://bootstrap.pypa.io/get-pip.py -o get-pip.py
|
|
||||||
else
|
|
||||||
PYTHON=python
|
|
||||||
curl https://bootstrap.pypa.io/pip/2.7/get-pip.py -o get-pip.py
|
|
||||||
fi
|
|
||||||
${PYTHON} ./get-pip.py
|
|
||||||
|
|
||||||
# Step 1.5: Add deb.coex.tech to apt
|
# Step 1.5: Add deb.coex.tech to apt
|
||||||
curl http://deb.coex.tech/aptly_repo_signing.key 2> /dev/null | apt-key add -
|
curl http://deb.coex.tech/aptly_repo_signing.key 2> /dev/null | apt-key add -
|
||||||
echo "deb http://deb.coex.tech/ros xenial main" > /etc/apt/sources.list.d/coex.tech.list
|
echo "deb http://deb.coex.tech/ros xenial main" > /etc/apt/sources.list.d/coex.tech.list
|
||||||
echo "yaml file:///etc/ros/rosdep/coex.yaml" > /etc/ros/rosdep/sources.list.d/99-coex.list
|
echo "yaml file:///etc/ros/rosdep/coex.yaml" > /etc/ros/rosdep/sources.list.d/99-coex.list
|
||||||
CODENAME=$(lsb_release -sc)
|
|
||||||
|
|
||||||
cat <<EOF > /etc/ros/rosdep/coex.yaml
|
cat <<EOF > /etc/ros/rosdep/coex.yaml
|
||||||
led_msgs:
|
led_msgs:
|
||||||
ubuntu:
|
ubuntu:
|
||||||
${CODENAME}: [ros-${ROS_DISTRO}-led-msgs]
|
xenial: ros-kinetic-led-msgs
|
||||||
async_web_server_cpp:
|
bionic: ros-melodic-led-msgs
|
||||||
ubuntu:
|
debian:
|
||||||
${CODENAME}: [ros-${ROS_DISTRO}-async-web-server-cpp]
|
stretch: ros-kinetic-led-msgs
|
||||||
ros_pytest:
|
buster: ros-melodic-led-msgs
|
||||||
ubuntu:
|
|
||||||
${CODENAME}: [ros-${ROS_DISTRO}-ros-pytest]
|
|
||||||
tf2_web_republisher:
|
|
||||||
ubuntu:
|
|
||||||
${CODENAME}: [ros-${ROS_DISTRO}-tf2-web-republisher]
|
|
||||||
web_video_server:
|
|
||||||
ubuntu:
|
|
||||||
${CODENAME}: [ros-${ROS_DISTRO}-web-video-server]
|
|
||||||
ws281x:
|
|
||||||
ubuntu:
|
|
||||||
${CODENAME}: [ros-${ROS_DISTRO}-ws281x]
|
|
||||||
EOF
|
EOF
|
||||||
apt-get update
|
apt update
|
||||||
rosdep update
|
rosdep update
|
||||||
|
|
||||||
# Step 2: Run rosdep to install all dependencies
|
# Step 2: Run rosdep to install all dependencies
|
||||||
@@ -57,10 +37,7 @@ cd /root/catkin_ws
|
|||||||
catkin_make
|
catkin_make
|
||||||
|
|
||||||
# Step 4: Run tests
|
# Step 4: Run tests
|
||||||
${PYTHON} -m pip install --upgrade pytest
|
pip install --upgrade pytest
|
||||||
cd /root/catkin_ws
|
cd /root/catkin_ws
|
||||||
source devel/setup.bash
|
source devel/setup.bash
|
||||||
catkin_make run_tests && catkin_test_results
|
catkin_make run_tests && catkin_test_results
|
||||||
|
|
||||||
# Step 5: Install packages
|
|
||||||
catkin_make install
|
|
||||||
|
|||||||
Binary file not shown.
|
Before Width: | Height: | Size: 1.8 KiB |
@@ -1,42 +0,0 @@
|
|||||||
#!/usr/bin/env python3
|
|
||||||
|
|
||||||
# Test QG recognition example
|
|
||||||
# Should be synced with the documentation: /docs/en/camera.md, /docs/ru/camera.md
|
|
||||||
# TODO: use real ROS topics
|
|
||||||
|
|
||||||
import rospy
|
|
||||||
from pyzbar import pyzbar
|
|
||||||
from cv_bridge import CvBridge
|
|
||||||
from sensor_msgs.msg import Image
|
|
||||||
|
|
||||||
bridge = CvBridge()
|
|
||||||
|
|
||||||
# rospy.init_node('barcode_test')
|
|
||||||
|
|
||||||
# Image subscriber callback function
|
|
||||||
def image_callback(data):
|
|
||||||
cv_image = bridge.imgmsg_to_cv2(data, 'bgr8') # OpenCV image
|
|
||||||
barcodes = pyzbar.decode(cv_image)
|
|
||||||
for barcode in barcodes:
|
|
||||||
b_data = barcode.data.decode("utf-8")
|
|
||||||
b_type = barcode.type
|
|
||||||
(x, y, w, h) = barcode.rect
|
|
||||||
xc = x + w/2
|
|
||||||
yc = y + h/2
|
|
||||||
print("Found {} with data {} with center at x={}, y={}".format(b_type, b_data, xc, yc))
|
|
||||||
# rospy.signal_shutdown('done')
|
|
||||||
|
|
||||||
# image_sub = rospy.Subscriber('main_camera/image_raw', Image, image_callback, queue_size=1)
|
|
||||||
|
|
||||||
# ==============================================================================
|
|
||||||
# Publish test image
|
|
||||||
# rospy.sleep(2)
|
|
||||||
|
|
||||||
import cv2
|
|
||||||
img = cv2.imread('qr.png')
|
|
||||||
image_callback(bridge.cv2_to_imgmsg(img, 'bgr8'))
|
|
||||||
|
|
||||||
# image_pub = rospy.Publisher('/main_camera/image_raw', Image, queue_size=1, latch=True)
|
|
||||||
# image_pub.publish(bridge.cv2_to_imgmsg(img, 'bgr8'))
|
|
||||||
|
|
||||||
# rospy.spin()
|
|
||||||
@@ -1,31 +1,21 @@
|
|||||||
#!/usr/bin/env python3
|
#!/usr/bin/env python
|
||||||
|
|
||||||
# validate all required modules installed
|
# validate all required modules installed
|
||||||
|
|
||||||
import os
|
|
||||||
import rospy
|
import rospy
|
||||||
from geometry_msgs.msg import PoseStamped
|
from geometry_msgs.msg import PoseStamped
|
||||||
from sensor_msgs.msg import Range, BatteryState
|
|
||||||
|
|
||||||
import cv2
|
import cv2
|
||||||
import cv2.aruco
|
import cv2.aruco
|
||||||
from sensor_msgs.msg import Image
|
|
||||||
from cv_bridge import CvBridge
|
|
||||||
|
|
||||||
import numpy
|
import numpy
|
||||||
import mavros
|
import mavros
|
||||||
from mavros_msgs.msg import State, StatusText, ExtendedState, RCIn, Mavlink
|
from mavros_msgs.msg import State, StatusText, ExtendedState
|
||||||
from mavros_msgs.srv import CommandBool, CommandLong, SetMode
|
from mavros_msgs.srv import CommandBool, CommandLong, SetMode
|
||||||
|
|
||||||
from std_srvs.srv import Trigger
|
from std_srvs.srv import Trigger
|
||||||
from clover.srv import GetTelemetry, Navigate, NavigateGlobal, SetPosition, SetVelocity, \
|
from clover.srv import GetTelemetry, Navigate, NavigateGlobal, SetPosition, SetVelocity, \
|
||||||
SetAttitude, SetRates, SetLEDEffect
|
SetAttitude, SetRates, SetLEDEffect
|
||||||
from led_msgs.srv import SetLEDs
|
|
||||||
from led_msgs.msg import LEDStateArray, LEDState
|
|
||||||
from aruco_pose.msg import Marker, MarkerArray, Point2D
|
|
||||||
from clover import long_callback
|
|
||||||
|
|
||||||
import dynamic_reconfigure.client
|
|
||||||
|
|
||||||
import tf2_ros
|
import tf2_ros
|
||||||
import tf2_geometry_msgs
|
import tf2_geometry_msgs
|
||||||
@@ -33,12 +23,9 @@ import tf2_geometry_msgs
|
|||||||
import VL53L1X
|
import VL53L1X
|
||||||
import pymavlink
|
import pymavlink
|
||||||
from pymavlink import mavutil
|
from pymavlink import mavutil
|
||||||
|
import rpi_ws281x
|
||||||
|
import pigpio
|
||||||
# from espeak import espeak
|
# from espeak import espeak
|
||||||
from pyzbar import pyzbar
|
from pyzbar import pyzbar
|
||||||
import docopt
|
|
||||||
|
|
||||||
print(cv2.getBuildInformation())
|
print cv2.getBuildInformation()
|
||||||
|
|
||||||
if not os.environ.get('VM'):
|
|
||||||
import rpi_ws281x
|
|
||||||
import pigpio
|
|
||||||
|
|||||||
@@ -6,10 +6,16 @@ set -ex
|
|||||||
|
|
||||||
# validate required software is installed
|
# validate required software is installed
|
||||||
|
|
||||||
|
python --version
|
||||||
python2 --version
|
python2 --version
|
||||||
python3 --version
|
python3 --version
|
||||||
|
ipython --version
|
||||||
ipython3 --version
|
ipython3 --version
|
||||||
|
|
||||||
|
# ptvsd does not have a stand-alone binary
|
||||||
|
python -m ptvsd --version
|
||||||
|
python3 -m ptvsd --version
|
||||||
|
|
||||||
node -v
|
node -v
|
||||||
npm -v
|
npm -v
|
||||||
|
|
||||||
@@ -19,77 +25,33 @@ lsof -v
|
|||||||
git --version
|
git --version
|
||||||
vim --version
|
vim --version
|
||||||
pip --version
|
pip --version
|
||||||
|
pip2 --version
|
||||||
pip3 --version
|
pip3 --version
|
||||||
tcpdump --version
|
tcpdump --version
|
||||||
monkey --version
|
monkey --version
|
||||||
|
pigpiod -v
|
||||||
|
i2cdetect -V
|
||||||
|
butterfly -h
|
||||||
# espeak --version
|
# espeak --version
|
||||||
systemctl --version
|
mjpg_streamer --version
|
||||||
|
|
||||||
if [ -z $VM ]; then
|
|
||||||
# rpi only software
|
|
||||||
python --version
|
|
||||||
ipython --version
|
|
||||||
pip2 --version
|
|
||||||
# `python` is python2 for now
|
|
||||||
[[ $(python -c 'import sys;print(sys.version_info.major)') == "2" ]]
|
|
||||||
|
|
||||||
# ptvsd does not have a stand-alone binary
|
|
||||||
python -m ptvsd --version
|
|
||||||
python3 -m ptvsd --version
|
|
||||||
|
|
||||||
pigpiod -v
|
|
||||||
i2cdetect -V
|
|
||||||
butterfly -h
|
|
||||||
mjpg_streamer --version
|
|
||||||
fi
|
|
||||||
|
|
||||||
# ros stuff
|
# ros stuff
|
||||||
|
|
||||||
roscore -h
|
roscore -h
|
||||||
rosversion clover
|
rosversion clover
|
||||||
rosversion aruco_pose
|
rosversion aruco_pose
|
||||||
|
rosversion vl53l1x
|
||||||
rosversion mavros
|
rosversion mavros
|
||||||
rosversion mavros_extras
|
rosversion mavros_extras
|
||||||
rosversion ws281x
|
|
||||||
rosversion led_msgs
|
|
||||||
rosversion dynamic_reconfigure
|
rosversion dynamic_reconfigure
|
||||||
rosversion tf2_web_republisher
|
rosversion tf2_web_republisher
|
||||||
rosversion rosbridge_server
|
rosversion compressed_image_transport
|
||||||
|
rosversion rosbridge_suite
|
||||||
|
rosversion rosserial
|
||||||
rosversion usb_cam
|
rosversion usb_cam
|
||||||
rosversion cv_camera
|
rosversion cv_camera
|
||||||
rosversion web_video_server
|
rosversion web_video_server
|
||||||
rosversion nodelet
|
rosversion rosshow
|
||||||
rosversion image_view
|
|
||||||
|
|
||||||
[[ $(rosversion ws281x) == "0.0.13" ]]
|
|
||||||
|
|
||||||
if [ -z $VM ]; then
|
|
||||||
rosversion compressed_image_transport
|
|
||||||
rosversion rosshow
|
|
||||||
rosversion vl53l1x
|
|
||||||
rosversion rosserial
|
|
||||||
[[ $(rosversion cv_camera) == "0.5.1" ]] # patched version with init fix
|
|
||||||
fi
|
|
||||||
|
|
||||||
# determine user home directory
|
|
||||||
[ $VM ] && H="/home/clover" || H="/home/pi"
|
|
||||||
|
|
||||||
# test basic ros tool work
|
|
||||||
source $H/catkin_ws/devel/setup.bash
|
|
||||||
roscd
|
|
||||||
rosrun
|
|
||||||
rosmsg
|
|
||||||
rossrv
|
|
||||||
rosnode || [ $? -eq 64 ] # usage output code is 64
|
|
||||||
rostopic || [ $? -eq 64 ]
|
|
||||||
rosservice || [ $? -eq 64 ]
|
|
||||||
rosparam
|
|
||||||
roslaunch -h
|
|
||||||
|
|
||||||
# validate examples are present
|
# validate examples are present
|
||||||
[[ $(ls $H/examples/*) ]]
|
[[ $(ls /home/pi/examples/*) ]]
|
||||||
|
|
||||||
# validate web tools present
|
|
||||||
[ -d $H/.ros/www ]
|
|
||||||
[ "$(readlink $H/.ros/www/clover)" = "$H/catkin_ws/src/clover/clover/www" ]
|
|
||||||
[ "$(readlink $H/.ros/www/clover_blocks)" = "$H/catkin_ws/src/clover/clover_blocks/www" ]
|
|
||||||
|
|||||||
@@ -1,4 +1,4 @@
|
|||||||
#!/usr/bin/env python3
|
#!/usr/bin/env python
|
||||||
|
|
||||||
# test backwards compatibility
|
# test backwards compatibility
|
||||||
|
|
||||||
|
|||||||
@@ -5,7 +5,7 @@ import sys
|
|||||||
import subprocess
|
import subprocess
|
||||||
|
|
||||||
EXCLUDE = ('clever4-front-white.png', 'clever4-front-white-large.png', '.DS_Store',
|
EXCLUDE = ('clever4-front-white.png', 'clever4-front-white-large.png', '.DS_Store',
|
||||||
'clever4-front-black-large.png','clover42-black.png', 'clover42-main-margin.png')
|
'clever4-front-black-large.png','clover42-black.png')
|
||||||
|
|
||||||
code = 0
|
code = 0
|
||||||
|
|
||||||
|
|||||||
@@ -1,4 +1,4 @@
|
|||||||
cmake_minimum_required(VERSION 3.0)
|
cmake_minimum_required(VERSION 2.8.3)
|
||||||
project(clover)
|
project(clover)
|
||||||
|
|
||||||
## Compile as C++11, supported in ROS Kinetic and newer
|
## Compile as C++11, supported in ROS Kinetic and newer
|
||||||
@@ -17,7 +17,6 @@ find_package(catkin REQUIRED COMPONENTS
|
|||||||
message_generation
|
message_generation
|
||||||
geometry_msgs
|
geometry_msgs
|
||||||
sensor_msgs
|
sensor_msgs
|
||||||
led_msgs
|
|
||||||
geographic_msgs
|
geographic_msgs
|
||||||
tf
|
tf
|
||||||
tf2
|
tf2
|
||||||
@@ -25,22 +24,13 @@ find_package(catkin REQUIRED COMPONENTS
|
|||||||
tf2_ros
|
tf2_ros
|
||||||
image_transport
|
image_transport
|
||||||
cv_bridge
|
cv_bridge
|
||||||
dynamic_reconfigure
|
|
||||||
)
|
)
|
||||||
|
|
||||||
list(APPEND CMAKE_MODULE_PATH "${CMAKE_CURRENT_LIST_DIR}/cmake")
|
list(APPEND CMAKE_MODULE_PATH "${CMAKE_CURRENT_LIST_DIR}/cmake")
|
||||||
|
|
||||||
find_package(GeographicLib REQUIRED)
|
find_package(GeographicLib REQUIRED)
|
||||||
|
|
||||||
# Workaround for OpenCV 3/4 support
|
find_package(OpenCV 3 REQUIRED
|
||||||
set(_opencv_version 4)
|
|
||||||
find_package(OpenCV ${_opencv_version} QUIET COMPONENTS calib3d imgproc)
|
|
||||||
if (NOT OpenCV_FOUND)
|
|
||||||
message(STATUS "Did not find OpenCV 4, searching for OpenCV 3")
|
|
||||||
set(_opencv_version 3)
|
|
||||||
endif()
|
|
||||||
|
|
||||||
find_package(OpenCV ${_opencv_version} REQUIRED
|
|
||||||
COMPONENTS
|
COMPONENTS
|
||||||
calib3d
|
calib3d
|
||||||
imgproc
|
imgproc
|
||||||
@@ -53,7 +43,7 @@ find_package(OpenCV ${_opencv_version} REQUIRED
|
|||||||
## Uncomment this if the package has a setup.py. This macro ensures
|
## Uncomment this if the package has a setup.py. This macro ensures
|
||||||
## modules and global scripts declared therein get installed
|
## modules and global scripts declared therein get installed
|
||||||
## See http://ros.org/doc/api/catkin/html/user_guide/setup_dot_py.html
|
## See http://ros.org/doc/api/catkin/html/user_guide/setup_dot_py.html
|
||||||
catkin_python_setup()
|
# catkin_python_setup()
|
||||||
|
|
||||||
################################################
|
################################################
|
||||||
## Declare ROS messages, services and actions ##
|
## Declare ROS messages, services and actions ##
|
||||||
@@ -80,10 +70,11 @@ catkin_python_setup()
|
|||||||
## * add every package in MSG_DEP_SET to generate_messages(DEPENDENCIES ...)
|
## * add every package in MSG_DEP_SET to generate_messages(DEPENDENCIES ...)
|
||||||
|
|
||||||
## Generate messages in the 'msg' folder
|
## Generate messages in the 'msg' folder
|
||||||
add_message_files(
|
# add_message_files(
|
||||||
FILES
|
# FILES
|
||||||
State.msg
|
# Message1.msg
|
||||||
)
|
# Message2.msg
|
||||||
|
# )
|
||||||
|
|
||||||
## Generate services in the 'srv' folder
|
## Generate services in the 'srv' folder
|
||||||
add_service_files(
|
add_service_files(
|
||||||
@@ -91,9 +82,6 @@ add_service_files(
|
|||||||
GetTelemetry.srv
|
GetTelemetry.srv
|
||||||
Navigate.srv
|
Navigate.srv
|
||||||
NavigateGlobal.srv
|
NavigateGlobal.srv
|
||||||
SetAltitude.srv
|
|
||||||
SetYaw.srv
|
|
||||||
SetYawRate.srv
|
|
||||||
SetPosition.srv
|
SetPosition.srv
|
||||||
SetVelocity.srv
|
SetVelocity.srv
|
||||||
SetAttitude.srv
|
SetAttitude.srv
|
||||||
@@ -130,9 +118,10 @@ generate_messages(
|
|||||||
## and list every .cfg file to be processed
|
## and list every .cfg file to be processed
|
||||||
|
|
||||||
## Generate dynamic reconfigure parameters in the 'cfg' folder
|
## Generate dynamic reconfigure parameters in the 'cfg' folder
|
||||||
generate_dynamic_reconfigure_options(
|
# generate_dynamic_reconfigure_options(
|
||||||
cfg/Flow.cfg
|
# cfg/DynReconf1.cfg
|
||||||
)
|
# cfg/DynReconf2.cfg
|
||||||
|
# )
|
||||||
|
|
||||||
###################################
|
###################################
|
||||||
## catkin specific configuration ##
|
## catkin specific configuration ##
|
||||||
@@ -214,8 +203,6 @@ add_dependencies(clover_led ${PROJECT_NAME}_generate_messages_cpp)
|
|||||||
|
|
||||||
add_dependencies(shell ${PROJECT_NAME}_generate_messages_cpp)
|
add_dependencies(shell ${PROJECT_NAME}_generate_messages_cpp)
|
||||||
|
|
||||||
add_dependencies(${PROJECT_NAME} ${PROJECT_NAME}_gencfg)
|
|
||||||
|
|
||||||
## Rename C++ executable without prefix
|
## Rename C++ executable without prefix
|
||||||
## The above recommended prefix causes long target names, the following renames the
|
## The above recommended prefix causes long target names, the following renames the
|
||||||
## target back to the shorter version for ease of user use
|
## target back to the shorter version for ease of user use
|
||||||
@@ -246,12 +233,12 @@ target_link_libraries(${PROJECT_NAME}
|
|||||||
# DESTINATION ${CATKIN_PACKAGE_BIN_DESTINATION}
|
# DESTINATION ${CATKIN_PACKAGE_BIN_DESTINATION}
|
||||||
# )
|
# )
|
||||||
|
|
||||||
# Mark executables and/or libraries for installation
|
## Mark executables and/or libraries for installation
|
||||||
install(TARGETS simple_offboard clover_rc camera_markers vpe_publisher clover_led shell clover
|
# install(TARGETS ${PROJECT_NAME} ${PROJECT_NAME}_node
|
||||||
ARCHIVE DESTINATION ${CATKIN_PACKAGE_LIB_DESTINATION}
|
# ARCHIVE DESTINATION ${CATKIN_PACKAGE_LIB_DESTINATION}
|
||||||
LIBRARY DESTINATION ${CATKIN_PACKAGE_LIB_DESTINATION}
|
# LIBRARY DESTINATION ${CATKIN_PACKAGE_LIB_DESTINATION}
|
||||||
RUNTIME DESTINATION ${CATKIN_PACKAGE_BIN_DESTINATION}
|
# RUNTIME DESTINATION ${CATKIN_PACKAGE_BIN_DESTINATION}
|
||||||
)
|
# )
|
||||||
|
|
||||||
## Mark cpp header files for installation
|
## Mark cpp header files for installation
|
||||||
# install(DIRECTORY include/${PROJECT_NAME}/
|
# install(DIRECTORY include/${PROJECT_NAME}/
|
||||||
@@ -267,25 +254,13 @@ install(TARGETS simple_offboard clover_rc camera_markers vpe_publisher clover_le
|
|||||||
# DESTINATION ${CATKIN_PACKAGE_SHARE_DESTINATION}
|
# DESTINATION ${CATKIN_PACKAGE_SHARE_DESTINATION}
|
||||||
# )
|
# )
|
||||||
|
|
||||||
catkin_install_python(PROGRAMS src/selfcheck.py
|
|
||||||
DESTINATION ${CATKIN_PACKAGE_BIN_DESTINATION}
|
|
||||||
)
|
|
||||||
|
|
||||||
install(DIRECTORY launch DESTINATION ${CATKIN_PACKAGE_SHARE_DESTINATION})
|
|
||||||
|
|
||||||
install(DIRECTORY examples DESTINATION ${CATKIN_PACKAGE_SHARE_DESTINATION})
|
|
||||||
|
|
||||||
install(DIRECTORY www DESTINATION ${CATKIN_PACKAGE_SHARE_DESTINATION})
|
|
||||||
|
|
||||||
# TODO: install www
|
|
||||||
|
|
||||||
# Only install udev rules when building a Debian package
|
# Only install udev rules when building a Debian package
|
||||||
# FIXME: Other operating systems may have other prefixes
|
# FIXME: Other operating systems may have other prefixes
|
||||||
string(FIND ${CMAKE_INSTALL_PREFIX} "/opt/ros" _PREFIX_INDEX)
|
string(FIND ${CMAKE_INSTALL_PREFIX} "/opt/ros" _PREFIX_INDEX)
|
||||||
if (${_PREFIX_INDEX} EQUAL 0)
|
if (${_PREFIX_INDEX} EQUAL 0)
|
||||||
message(STATUS "Building as a Debian package - adding udev rules as installable files")
|
message(STATUS "Building as a Debian package - adding udev rules as installable files")
|
||||||
install(FILES
|
install(FILES
|
||||||
udev/99-px4fmu.rules
|
config/99-px4fmu.rules
|
||||||
DESTINATION /lib/udev/rules.d
|
DESTINATION /lib/udev/rules.d
|
||||||
)
|
)
|
||||||
else()
|
else()
|
||||||
@@ -308,5 +283,4 @@ endif()
|
|||||||
if (CATKIN_ENABLE_TESTING)
|
if (CATKIN_ENABLE_TESTING)
|
||||||
find_package(rostest REQUIRED)
|
find_package(rostest REQUIRED)
|
||||||
add_rostest(test/basic.test)
|
add_rostest(test/basic.test)
|
||||||
add_rostest(test/offboard.test)
|
|
||||||
endif()
|
endif()
|
||||||
|
|||||||
@@ -4,7 +4,7 @@ A bundle for autonomous navigation and drone control.
|
|||||||
|
|
||||||
## Manual installation
|
## Manual installation
|
||||||
|
|
||||||
Install ROS Noetic according to the [documentation](http://wiki.ros.org/noetic/Installation), then [create a Catkin workspace](http://wiki.ros.org/catkin/Tutorials/create_a_workspace).
|
Install ROS Melodic according to the [documentation](http://wiki.ros.org/melodic/Installation), then [create a Catkin workspace](http://wiki.ros.org/catkin/Tutorials/create_a_workspace).
|
||||||
|
|
||||||
Clone this repo to directory `~/catkin_ws/src/clover`:
|
Clone this repo to directory `~/catkin_ws/src/clover`:
|
||||||
|
|
||||||
@@ -36,7 +36,7 @@ curl https://raw.githubusercontent.com/mavlink/mavros/master/mavros/scripts/inst
|
|||||||
You may optionally install udev rules to provide `/dev/px4fmu` symlink to your PX4-based flight controller connected over USB. Copy `99-px4fmu.rules` to your `/lib/udev/rules.d` folder:
|
You may optionally install udev rules to provide `/dev/px4fmu` symlink to your PX4-based flight controller connected over USB. Copy `99-px4fmu.rules` to your `/lib/udev/rules.d` folder:
|
||||||
|
|
||||||
```bash
|
```bash
|
||||||
cd ~/catkin_ws/src/clover/clover/udev
|
cd ~/catkin_ws/src/clover/clover/config
|
||||||
sudo cp 99-px4fmu.rules /lib/udev/rules.d
|
sudo cp 99-px4fmu.rules /lib/udev/rules.d
|
||||||
```
|
```
|
||||||
|
|
||||||
|
|||||||
@@ -1,10 +0,0 @@
|
|||||||
#!/usr/bin/env python
|
|
||||||
PACKAGE = "clover"
|
|
||||||
|
|
||||||
from dynamic_reconfigure.parameter_generator_catkin import *
|
|
||||||
|
|
||||||
gen = ParameterGenerator()
|
|
||||||
|
|
||||||
gen.add("enabled", bool_t, 0, "if optical flow enabled", True)
|
|
||||||
|
|
||||||
exit(gen.generate(PACKAGE, "clover", "Flow"))
|
|
||||||
@@ -12,6 +12,4 @@ SUBSYSTEM=="tty", ATTRS{idVendor}=="26ac", ATTRS{idProduct}=="0016", ATTRS{produ
|
|||||||
SUBSYSTEM=="tty", ATTRS{idVendor}=="26ac", ATTRS{idProduct}=="0013", ATTRS{product}=="PX4 FMU v4.x PRO", SYMLINK+="px4fmu"
|
SUBSYSTEM=="tty", ATTRS{idVendor}=="26ac", ATTRS{idProduct}=="0013", ATTRS{product}=="PX4 FMU v4.x PRO", SYMLINK+="px4fmu"
|
||||||
# Omnibus
|
# Omnibus
|
||||||
SUBSYSTEM=="tty", ATTRS{idVendor}=="26ac", ATTRS{idProduct}=="0001", ATTRS{product}=="PX4 OmnibusF4SD", SYMLINK+="px4fmu"
|
SUBSYSTEM=="tty", ATTRS{idVendor}=="26ac", ATTRS{idProduct}=="0001", ATTRS{product}=="PX4 OmnibusF4SD", SYMLINK+="px4fmu"
|
||||||
# CUAV X7 Pro
|
|
||||||
SUBSYSTEM=="tty", ATTRS{idVendor}=="3163", ATTRS{idProduct}=="004c", ATTRS{product}=="PX4 CUAV X7Pro", SYMLINK+="px4fmu"
|
|
||||||
|
|
||||||
@@ -1,64 +0,0 @@
|
|||||||
# Information: https://clover.coex.tech/camera
|
|
||||||
|
|
||||||
# Example on basic working with the camera and image processing:
|
|
||||||
|
|
||||||
# - cuts out a central square from the camera image;
|
|
||||||
# - publishes this cropped image to the topic `/cv/center`;
|
|
||||||
# - computes the average color of it;
|
|
||||||
# - prints its name to the console.
|
|
||||||
|
|
||||||
import rospy
|
|
||||||
import cv2
|
|
||||||
from sensor_msgs.msg import Image
|
|
||||||
from cv_bridge import CvBridge
|
|
||||||
from clover import long_callback
|
|
||||||
|
|
||||||
rospy.init_node('cv')
|
|
||||||
bridge = CvBridge()
|
|
||||||
|
|
||||||
printed_color = None
|
|
||||||
center_pub = rospy.Publisher('~center', Image, queue_size=1)
|
|
||||||
|
|
||||||
def get_color_name(h):
|
|
||||||
if h < 15: return 'red'
|
|
||||||
elif h < 30: return 'orange'
|
|
||||||
elif h < 60: return 'yellow'
|
|
||||||
elif h < 90: return 'green'
|
|
||||||
elif h < 120: return 'cyan'
|
|
||||||
elif h < 150: return 'blue'
|
|
||||||
elif h < 170: return 'magenta'
|
|
||||||
else: return 'red'
|
|
||||||
|
|
||||||
|
|
||||||
@long_callback
|
|
||||||
def image_callback(msg):
|
|
||||||
img = bridge.imgmsg_to_cv2(msg, 'bgr8')
|
|
||||||
|
|
||||||
# convert to HSV to work with color hue
|
|
||||||
img_hsv = cv2.cvtColor(img, cv2.COLOR_BGR2HSV)
|
|
||||||
|
|
||||||
# cut out a central square
|
|
||||||
w = img.shape[1]
|
|
||||||
h = img.shape[0]
|
|
||||||
r = 20
|
|
||||||
center = img_hsv[h // 2 - r:h // 2 + r, w // 2 - r:w // 2 + r]
|
|
||||||
|
|
||||||
# compute and print the average hue
|
|
||||||
mean_hue = center[:, :, 0].mean()
|
|
||||||
color = get_color_name(mean_hue)
|
|
||||||
global printed_color
|
|
||||||
if color != printed_color:
|
|
||||||
print(color)
|
|
||||||
printed_color = color
|
|
||||||
|
|
||||||
# publish the cropped image
|
|
||||||
center = cv2.cvtColor(center, cv2.COLOR_HSV2BGR)
|
|
||||||
center_pub.publish(bridge.cv2_to_imgmsg(center, 'bgr8'))
|
|
||||||
|
|
||||||
# process every frame:
|
|
||||||
image_sub = rospy.Subscriber('main_camera/image_raw', Image, image_callback, queue_size=1)
|
|
||||||
|
|
||||||
# process 5 frames per second:
|
|
||||||
# image_sub = rospy.Subscriber('main_camera/image_raw_throttled', Image, image_callback, queue_size=1)
|
|
||||||
|
|
||||||
rospy.spin()
|
|
||||||
@@ -1,11 +0,0 @@
|
|||||||
# Information: https://clover.coex.tech/en/simple_offboard.html#gettelemetry
|
|
||||||
|
|
||||||
import rospy
|
|
||||||
from clover import srv
|
|
||||||
|
|
||||||
rospy.init_node('flight')
|
|
||||||
|
|
||||||
get_telemetry = rospy.ServiceProxy('get_telemetry', srv.GetTelemetry)
|
|
||||||
|
|
||||||
# Print drone's state
|
|
||||||
print(get_telemetry())
|
|
||||||
@@ -1,47 +0,0 @@
|
|||||||
# Information: https://clover.coex.tech/en/simple_offboard.html#navigateglobal
|
|
||||||
|
|
||||||
import rospy
|
|
||||||
from clover import srv
|
|
||||||
from std_srvs.srv import Trigger
|
|
||||||
import math
|
|
||||||
|
|
||||||
rospy.init_node('flight')
|
|
||||||
|
|
||||||
get_telemetry = rospy.ServiceProxy('get_telemetry', srv.GetTelemetry)
|
|
||||||
navigate = rospy.ServiceProxy('navigate', srv.Navigate)
|
|
||||||
navigate_global = rospy.ServiceProxy('navigate_global', srv.NavigateGlobal)
|
|
||||||
set_position = rospy.ServiceProxy('set_position', srv.SetPosition)
|
|
||||||
set_velocity = rospy.ServiceProxy('set_velocity', srv.SetVelocity)
|
|
||||||
set_attitude = rospy.ServiceProxy('set_attitude', srv.SetAttitude)
|
|
||||||
set_rates = rospy.ServiceProxy('set_rates', srv.SetRates)
|
|
||||||
land = rospy.ServiceProxy('land', Trigger)
|
|
||||||
|
|
||||||
# https://clover.coex.tech/en/snippets.html#wait_arrival
|
|
||||||
def wait_arrival(tolerance=0.2):
|
|
||||||
while not rospy.is_shutdown():
|
|
||||||
telem = get_telemetry(frame_id='navigate_target')
|
|
||||||
if math.sqrt(telem.x ** 2 + telem.y ** 2 + telem.z ** 2) < tolerance:
|
|
||||||
break
|
|
||||||
rospy.sleep(0.2)
|
|
||||||
|
|
||||||
start = get_telemetry()
|
|
||||||
|
|
||||||
if math.isnan(start.lat):
|
|
||||||
raise Exception('No global position, install and configure GPS sensor: https://clover.coex.tech/gps')
|
|
||||||
|
|
||||||
print('Start point global position: lat={}, lon={}'.format(start.lat, start.lon))
|
|
||||||
|
|
||||||
print('Take off 3 meters')
|
|
||||||
navigate(x=0, y=0, z=3, frame_id='body', auto_arm=True)
|
|
||||||
wait_arrival()
|
|
||||||
|
|
||||||
print('Fly 1 arcsecond to the North (approx. 30 meters)')
|
|
||||||
navigate_global(lat=start.lat+1.0/60/60, lon=start.lon, z=start.z+3, yaw=math.inf, speed=5)
|
|
||||||
wait_arrival()
|
|
||||||
|
|
||||||
print('Fly to home position')
|
|
||||||
navigate_global(lat=start.lat, lon=start.lon, z=start.z+3, yaw=math.inf, speed=5)
|
|
||||||
wait_arrival()
|
|
||||||
|
|
||||||
print('Land')
|
|
||||||
land()
|
|
||||||
@@ -1,15 +0,0 @@
|
|||||||
# Information: https://clover.coex.tech/en/laser.html
|
|
||||||
|
|
||||||
import rospy
|
|
||||||
from sensor_msgs.msg import Range
|
|
||||||
|
|
||||||
rospy.init_node('process_rangefinder')
|
|
||||||
|
|
||||||
def range_callback(msg):
|
|
||||||
# Process data from the rangefinder
|
|
||||||
print('Rangefinder distance:', msg.range)
|
|
||||||
|
|
||||||
# Subscribe to laser rangefinder data
|
|
||||||
rospy.Subscriber('rangefinder/range', Range, range_callback)
|
|
||||||
|
|
||||||
rospy.spin()
|
|
||||||
@@ -2,42 +2,30 @@
|
|||||||
<arg name="aruco_detect" default="true"/>
|
<arg name="aruco_detect" default="true"/>
|
||||||
<arg name="aruco_map" default="false"/>
|
<arg name="aruco_map" default="false"/>
|
||||||
<arg name="aruco_vpe" default="false"/>
|
<arg name="aruco_vpe" default="false"/>
|
||||||
<arg name="placement" default="floor"/> <!-- markers placement: floor, ceiling, unknown -->
|
|
||||||
<arg name="length" default="0.22"/> <!-- not-in-map markers length, m -->
|
|
||||||
<arg name="map" default="map.txt"/> <!-- markers map file name -->
|
|
||||||
|
|
||||||
<!-- For additional help go to https://clover.coex.tech/aruco -->
|
<!-- For additional help go to https://clover.coex.tech/aruco -->
|
||||||
|
|
||||||
<arg name="force_init" default="false"/>
|
|
||||||
<arg name="disable" default="false"/> <!-- only force init -->
|
|
||||||
|
|
||||||
<!-- aruco_detect: detect aruco markers, estimate poses -->
|
<!-- aruco_detect: detect aruco markers, estimate poses -->
|
||||||
<node name="aruco_detect" pkg="nodelet" if="$(eval aruco_detect and not disable)" type="nodelet" args="load aruco_pose/aruco_detect main_camera_nodelet_manager" output="screen" clear_params="true" respawn="true">
|
<node name="aruco_detect" pkg="nodelet" if="$(arg aruco_detect)" type="nodelet" args="load aruco_pose/aruco_detect nodelet_manager" output="screen" clear_params="true">
|
||||||
<remap from="image_raw" to="main_camera/image_raw"/>
|
<remap from="image_raw" to="main_camera/image_raw"/>
|
||||||
<remap from="camera_info" to="main_camera/camera_info"/>
|
<remap from="camera_info" to="main_camera/camera_info"/>
|
||||||
<remap from="map_markers" to="aruco_map/map"/>
|
<remap from="map_markers" to="aruco_map/markers" if="$(arg aruco_map)"/>
|
||||||
<param name="estimate_poses" value="true"/>
|
<param name="estimate_poses" value="true"/>
|
||||||
<param name="send_tf" value="true"/>
|
<param name="send_tf" value="true"/>
|
||||||
<param name="use_map_markers" value="true"/>
|
<param name="known_tilt" value="map"/>
|
||||||
<param name="known_vertical" value="map" if="$(eval placement == 'floor' or placement == 'ceiling')"/>
|
<param name="length" value="0.33"/>
|
||||||
<param name="flip_vertical" value="true" if="$(eval placement == 'ceiling')"/>
|
|
||||||
<param name="length" value="$(arg length)"/>
|
|
||||||
<param name="transform_timeout" value="0.1"/>
|
|
||||||
<!-- aruco detector parameters -->
|
<!-- aruco detector parameters -->
|
||||||
<param name="cornerRefinementMethod" value="2"/> <!-- contour refinement -->
|
<param name="cornerRefinementMethod" value="2"/> <!-- contour refinement -->
|
||||||
<param name="minMarkerPerimeterRate" value="0.075"/> <!-- 0.075 for 320x240, 0.0375 for 640x480 -->
|
<param name="minMarkerPerimeterRate" value="0.075"/> <!-- 0.075 for 320x240, 0.0375 for 640x480 -->
|
||||||
<!-- length override example: -->
|
|
||||||
<!-- <param name="length_override/3" value="0.1"/> -->
|
|
||||||
</node>
|
</node>
|
||||||
|
|
||||||
<!-- aruco_map: estimate aruco map pose -->
|
<!-- aruco_map: estimate aruco map pose -->
|
||||||
<node name="aruco_map" pkg="nodelet" type="nodelet" if="$(eval aruco_map and not disable)" args="load aruco_pose/aruco_map main_camera_nodelet_manager" output="screen" clear_params="true" respawn="true">
|
<node name="aruco_map" pkg="nodelet" type="nodelet" if="$(arg aruco_map)" args="load aruco_pose/aruco_map nodelet_manager" output="screen" clear_params="true">
|
||||||
<remap from="image_raw" to="main_camera/image_raw"/>
|
<remap from="image_raw" to="main_camera/image_raw"/>
|
||||||
<remap from="camera_info" to="main_camera/camera_info"/>
|
<remap from="camera_info" to="main_camera/camera_info"/>
|
||||||
<remap from="markers" to="aruco_detect/markers"/>
|
<remap from="markers" to="aruco_detect/markers"/>
|
||||||
<param name="map" value="$(find aruco_pose)/map/$(arg map)"/>
|
<param name="map" value="$(find aruco_pose)/map/map.txt"/>
|
||||||
<param name="known_vertical" value="map" if="$(eval placement == 'floor' or placement == 'ceiling')"/>
|
<param name="known_tilt" value="map"/>
|
||||||
<param name="flip_vertical" value="true" if="$(eval placement == 'ceiling')"/>
|
|
||||||
<param name="image_axis" value="true"/>
|
<param name="image_axis" value="true"/>
|
||||||
<param name="frame_id" value="aruco_map_detected" if="$(arg aruco_vpe)"/>
|
<param name="frame_id" value="aruco_map_detected" if="$(arg aruco_vpe)"/>
|
||||||
<param name="frame_id" value="aruco_map" unless="$(arg aruco_vpe)"/>
|
<param name="frame_id" value="aruco_map" unless="$(arg aruco_vpe)"/>
|
||||||
@@ -46,11 +34,11 @@
|
|||||||
</node>
|
</node>
|
||||||
|
|
||||||
<!-- vpe publisher from aruco markers -->
|
<!-- vpe publisher from aruco markers -->
|
||||||
<node name="vpe_publisher" pkg="clover" type="vpe_publisher" if="$(eval aruco_vpe or force_init)" output="screen" clear_params="true">
|
<node name="vpe_publisher" pkg="clover" type="vpe_publisher" if="$(arg aruco_vpe)" output="screen" clear_params="true">
|
||||||
<remap from="~pose_cov" to="aruco_map/pose" if="$(arg aruco_vpe)"/>
|
<remap from="~pose_cov" to="aruco_map/pose"/>
|
||||||
<remap from="~vpe" to="mavros/vision_pose/pose"/>
|
<remap from="~vpe" to="mavros/vision_pose/pose"/>
|
||||||
<param name="frame_id" value="aruco_map_detected" if="$(arg aruco_vpe)"/>
|
<param name="frame_id" value="aruco_map_detected"/>
|
||||||
<param name="force_init" value="$(arg force_init)"/>
|
<param name="publish_zero" value="true"/>
|
||||||
<param name="offset_frame_id" value="aruco_map"/>
|
<param name="offset_frame_id" value="aruco_map"/>
|
||||||
</node>
|
</node>
|
||||||
</launch>
|
</launch>
|
||||||
|
|||||||
@@ -11,8 +11,8 @@
|
|||||||
<arg name="rangefinder_vl53l1x" default="true"/>
|
<arg name="rangefinder_vl53l1x" default="true"/>
|
||||||
<arg name="led" default="true"/>
|
<arg name="led" default="true"/>
|
||||||
<arg name="blocks" default="false"/>
|
<arg name="blocks" default="false"/>
|
||||||
<arg name="rc" default="false"/>
|
<arg name="rc" default="true"/>
|
||||||
<arg name="force_init" default="true"/> <!-- force estimator to init by publishing zero pose -->
|
<arg name="shell" default="true"/>
|
||||||
|
|
||||||
<arg name="simulator" default="false"/> <!-- flag that we are operating on a simulated drone -->
|
<arg name="simulator" default="false"/> <!-- flag that we are operating on a simulated drone -->
|
||||||
|
|
||||||
@@ -34,22 +34,28 @@
|
|||||||
</node>
|
</node>
|
||||||
|
|
||||||
<!-- aruco markers -->
|
<!-- aruco markers -->
|
||||||
<include file="$(find clover)/launch/aruco.launch" if="$(eval aruco or force_init)">
|
<include file="$(find clover)/launch/aruco.launch" if="$(arg aruco)"/>
|
||||||
<arg name="force_init" value="$(arg force_init)"/>
|
|
||||||
<arg name="disable" value="$(eval not aruco)"/>
|
|
||||||
</include>
|
|
||||||
|
|
||||||
<!-- optical flow -->
|
<!-- optical flow -->
|
||||||
<node pkg="nodelet" type="nodelet" name="optical_flow" args="load clover/optical_flow main_camera_nodelet_manager" if="$(arg optical_flow)" clear_params="true" output="screen" respawn="true">
|
<node pkg="nodelet" type="nodelet" name="optical_flow" args="load clover/optical_flow nodelet_manager" if="$(arg optical_flow)" clear_params="true" output="screen">
|
||||||
<remap from="image_raw" to="main_camera/image_raw"/>
|
<remap from="image_raw" to="main_camera/image_raw"/>
|
||||||
<remap from="camera_info" to="main_camera/camera_info"/>
|
<remap from="camera_info" to="main_camera/camera_info"/>
|
||||||
<param name="calc_flow_gyro" value="true"/>
|
<param name="calc_flow_gyro" value="true"/>
|
||||||
<param name="roi_rad" value="0.8"/>
|
<param name="roi_rad" value="0.8"/>
|
||||||
<param name="disable_on_vpe" value="true"/>
|
|
||||||
</node>
|
</node>
|
||||||
|
|
||||||
|
<!-- main nodelet manager -->
|
||||||
|
<node pkg="nodelet" type="nodelet" name="nodelet_manager" args="manager" output="screen" clear_params="true">
|
||||||
|
<param name="num_worker_threads" value="2"/>
|
||||||
|
</node>
|
||||||
|
|
||||||
|
<node pkg="tf2_ros" type="static_transform_publisher" name="map_flipped_frame" args="0 0 0 3.1415926 3.1415926 0 map map_flipped"/>
|
||||||
|
|
||||||
<!-- simplified offboard control -->
|
<!-- simplified offboard control -->
|
||||||
<node name="simple_offboard" pkg="clover" type="simple_offboard" output="screen" clear_params="true">
|
<node name="simple_offboard" pkg="clover" type="simple_offboard" output="screen" clear_params="true">
|
||||||
|
<param name="reference_frames/body" value="map"/>
|
||||||
|
<param name="reference_frames/base_link" value="map"/>
|
||||||
|
<param name="reference_frames/navigate_target" value="map"/>
|
||||||
<param name="reference_frames/main_camera_optical" value="map"/>
|
<param name="reference_frames/main_camera_optical" value="map"/>
|
||||||
</node>
|
</node>
|
||||||
|
|
||||||
@@ -85,4 +91,11 @@
|
|||||||
<param name="use_fake_gcs" value="false"/>
|
<param name="use_fake_gcs" value="false"/>
|
||||||
</node>
|
</node>
|
||||||
|
|
||||||
|
<!-- Shell access through ROS service -->
|
||||||
|
<node name="shell" pkg="clover" type="shell" output="screen" if="$(arg shell)"/>
|
||||||
|
|
||||||
|
<!-- Update static directory -->
|
||||||
|
<node pkg="roswww_static" name="roswww_static" type="main.py" clear_params="true">
|
||||||
|
<param name="default_package" value="clover"/>
|
||||||
|
</node>
|
||||||
</launch>
|
</launch>
|
||||||
|
|||||||
@@ -36,7 +36,7 @@
|
|||||||
posctl: { r: 50, g: 100, b: 220 }
|
posctl: { r: 50, g: 100, b: 220 }
|
||||||
offboard: { r: 220, g: 20, b: 250 }
|
offboard: { r: 220, g: 20, b: 250 }
|
||||||
low_battery: { threshold: 3.6, effect: blink_fast, r: 255, g: 0, b: 0 }
|
low_battery: { threshold: 3.6, effect: blink_fast, r: 255, g: 0, b: 0 }
|
||||||
error: { effect: flash, r: 255, g: 0, b: 0, ignore: [ "[lpe] vision position timeout" ]}
|
error: { effect: flash, r: 255, g: 0, b: 0 }
|
||||||
</rosparam>
|
</rosparam>
|
||||||
</node>
|
</node>
|
||||||
</launch>
|
</launch>
|
||||||
|
|||||||
@@ -3,9 +3,6 @@
|
|||||||
|
|
||||||
<arg name="direction_z" default="down"/> <!-- direction the camera points: down, up -->
|
<arg name="direction_z" default="down"/> <!-- direction the camera points: down, up -->
|
||||||
<arg name="direction_y" default="backward"/> <!-- direction the camera cable points: backward, forward -->
|
<arg name="direction_y" default="backward"/> <!-- direction the camera cable points: backward, forward -->
|
||||||
<arg name="device" default="/dev/video0"/> <!-- v4l2 device -->
|
|
||||||
<arg name="throttled_topic" default="true"/> <!-- enable throttled image topic -->
|
|
||||||
<arg name="throttled_topic_rate" default="5.0"/> <!-- throttled image topic rate -->
|
|
||||||
<arg name="simulator" default="false"/>
|
<arg name="simulator" default="false"/>
|
||||||
|
|
||||||
<node if="$(eval direction_z == 'down' and direction_y == 'backward')" pkg="tf2_ros" type="static_transform_publisher" name="main_camera_frame" args="0.05 0 -0.07 -1.5707963 0 3.1415926 base_link main_camera_optical"/>
|
<node if="$(eval direction_z == 'down' and direction_y == 'backward')" pkg="tf2_ros" type="static_transform_publisher" name="main_camera_frame" args="0.05 0 -0.07 -1.5707963 0 3.1415926 base_link main_camera_optical"/>
|
||||||
@@ -20,14 +17,9 @@
|
|||||||
<!-- static_transform_publisher arguments: x y z yaw pitch roll frame_id child_frame_id -->
|
<!-- static_transform_publisher arguments: x y z yaw pitch roll frame_id child_frame_id -->
|
||||||
<!-- <node pkg="tf2_ros" type="static_transform_publisher" name="main_camera_frame" args="0.05 0 -0.07 -1.5707963 0 3.1415926 base_link main_camera_optical"/> -->
|
<!-- <node pkg="tf2_ros" type="static_transform_publisher" name="main_camera_frame" args="0.05 0 -0.07 -1.5707963 0 3.1415926 base_link main_camera_optical"/> -->
|
||||||
|
|
||||||
<!-- camera nodelet manager -->
|
|
||||||
<node pkg="nodelet" type="nodelet" name="main_camera_nodelet_manager" args="manager" output="screen" clear_params="true" respawn="true">
|
|
||||||
<param name="num_worker_threads" value="2"/>
|
|
||||||
</node>
|
|
||||||
|
|
||||||
<!-- camera node -->
|
<!-- camera node -->
|
||||||
<node pkg="nodelet" type="nodelet" name="main_camera" args="load cv_camera/CvCameraNodelet main_camera_nodelet_manager" launch-prefix="rosrun clover waitfile $(arg device)" clear_params="true" unless="$(arg simulator)" respawn="true">
|
<node pkg="nodelet" type="nodelet" name="main_camera" args="load cv_camera/CvCameraNodelet nodelet_manager" clear_params="true" unless="$(arg simulator)">
|
||||||
<param name="device_path" value="$(arg device)"/>
|
<param name="device_path" value="/dev/video0"/> <!-- v4l2 device -->
|
||||||
<param name="frame_id" value="main_camera_optical"/>
|
<param name="frame_id" value="main_camera_optical"/>
|
||||||
<param name="camera_info_url" value="file://$(find clover)/camera_info/fisheye_cam.yaml"/>
|
<param name="camera_info_url" value="file://$(find clover)/camera_info/fisheye_cam.yaml"/>
|
||||||
|
|
||||||
@@ -45,8 +37,4 @@
|
|||||||
<node pkg="clover" type="camera_markers" ns="main_camera" name="main_camera_markers">
|
<node pkg="clover" type="camera_markers" ns="main_camera" name="main_camera_markers">
|
||||||
<param name="scale" value="3.0"/>
|
<param name="scale" value="3.0"/>
|
||||||
</node>
|
</node>
|
||||||
|
|
||||||
<!-- image topic throttled -->
|
|
||||||
<node pkg="topic_tools" name="main_camera_throttle" type="throttle" ns="main_camera"
|
|
||||||
args="messages image_raw $(arg throttled_topic_rate) image_raw_throttled" if="$(arg throttled_topic)"/>
|
|
||||||
</launch>
|
</launch>
|
||||||
|
|||||||
@@ -1,21 +1,18 @@
|
|||||||
<launch>
|
<launch>
|
||||||
<arg name="fcu_conn" default="usb"/> <!-- options: usb, uart, tcp, udp, sitl, hitl -->
|
<arg name="fcu_conn" default="usb"/> <!-- options: usb, uart, tcp, udp, sitl -->
|
||||||
<arg name="fcu_ip" default="127.0.0.1"/>
|
<arg name="fcu_ip" default="127.0.0.1"/>
|
||||||
<arg name="fcu_sys_id" default="1"/>
|
<arg name="fcu_sys_id" default="1"/>
|
||||||
<arg name="gcs_bridge" default="tcp"/>
|
<arg name="gcs_bridge" default="tcp"/>
|
||||||
<arg name="viz" default="true"/>
|
<arg name="viz" default="true"/>
|
||||||
<arg name="respawn" default="true"/>
|
<arg name="respawn" default="true"/>
|
||||||
<arg name="distance_sensor_remap" default="rangefinder/range"/>
|
<arg name="distance_sensor_remap" default="rangefinder/range"/>
|
||||||
<arg name="usb_device" default="/dev/px4fmu"/>
|
|
||||||
<arg name="prefix" default="" unless="$(eval fcu_conn == 'usb')"/>
|
|
||||||
<arg name="prefix" default="rosrun clover waitfile $(arg usb_device)" if="$(eval fcu_conn == 'usb')"/>
|
|
||||||
|
|
||||||
<node pkg="mavros" type="mavros_node" name="mavros" launch-prefix="$(arg prefix)" required="false" clear_params="true" respawn="$(arg respawn)" unless="$(eval fcu_conn == 'none')" respawn_delay="1" output="screen">
|
<node pkg="mavros" type="mavros_node" name="mavros" required="false" clear_params="true" respawn="$(arg respawn)" unless="$(eval fcu_conn == 'none')" respawn_delay="1" output="screen">
|
||||||
<!-- UART connection -->
|
<!-- UART connection -->
|
||||||
<param name="fcu_url" value="/dev/ttyAMA0:921600" if="$(eval fcu_conn is None or fcu_conn == 'uart')"/>
|
<param name="fcu_url" value="/dev/ttyAMA0:921600" if="$(eval fcu_conn is None or fcu_conn == 'uart')"/>
|
||||||
|
|
||||||
<!-- USB connection -->
|
<!-- USB connection -->
|
||||||
<param name="fcu_url" value="$(arg usb_device)" if="$(eval fcu_conn == 'usb')"/>
|
<param name="fcu_url" value="/dev/px4fmu" if="$(eval fcu_conn == 'usb')"/>
|
||||||
|
|
||||||
<!-- sitl before PX4 1.9.0 -->
|
<!-- sitl before PX4 1.9.0 -->
|
||||||
<param name="fcu_url" value="udp://@$(arg fcu_ip):14557" if="$(eval fcu_conn == 'udp')"/>
|
<param name="fcu_url" value="udp://@$(arg fcu_ip):14557" if="$(eval fcu_conn == 'udp')"/>
|
||||||
@@ -23,9 +20,6 @@
|
|||||||
<!-- sitl since PX4 1.9.0 -->
|
<!-- sitl since PX4 1.9.0 -->
|
||||||
<param name="fcu_url" value="udp://@$(arg fcu_ip):14580" if="$(eval fcu_conn == 'sitl')"/>
|
<param name="fcu_url" value="udp://@$(arg fcu_ip):14580" if="$(eval fcu_conn == 'sitl')"/>
|
||||||
|
|
||||||
<!-- hitl connection (to gazebo_mavlink_interface plugin) -->
|
|
||||||
<param name="fcu_url" value="udp://$(arg fcu_ip):14540@" if="$(eval fcu_conn == 'hitl')"/>
|
|
||||||
|
|
||||||
<!-- set target_system_id -->
|
<!-- set target_system_id -->
|
||||||
<param name="target_system_id" value="$(arg fcu_sys_id)" />
|
<param name="target_system_id" value="$(arg fcu_sys_id)" />
|
||||||
|
|
||||||
@@ -42,7 +36,7 @@
|
|||||||
<rosparam command="load" file="$(find clover)/launch/mavros_config.yaml"/>
|
<rosparam command="load" file="$(find clover)/launch/mavros_config.yaml"/>
|
||||||
|
|
||||||
<!-- remap rangefinder -->
|
<!-- remap rangefinder -->
|
||||||
<remap from="mavros/distance_sensor/rangefinder_sub" to="$(arg distance_sensor_remap)" if="$(eval bool(distance_sensor_remap))"/>
|
<remap from="mavros/distance_sensor/rangefinder_sub" to="rangefinder/range"/>
|
||||||
|
|
||||||
<rosparam param="plugin_whitelist">
|
<rosparam param="plugin_whitelist">
|
||||||
- altitude
|
- altitude
|
||||||
|
|||||||
@@ -1,11 +1,11 @@
|
|||||||
# Config file for mavros
|
# Config file for mavros
|
||||||
# Based on https://raw.githubusercontent.com/mavlink/mavros/master/mavros/launch/px4_config.yaml
|
# Based on https://raw.githubusercontent.com/mavlink/mavros/master/mavros/launch/px4_config.yaml
|
||||||
|
|
||||||
startup_px4_usb_quirk: false
|
startup_px4_usb_quirk: true
|
||||||
|
|
||||||
conn:
|
conn:
|
||||||
heartbeat_rate: 1.0 # send heartbeat rate in Hertz
|
heartbeat_rate: 1.0 # send hertbeat rate in Hertz
|
||||||
timeout: 10.0 # heartbeat timeout in seconds
|
timeout: 10.0 # hertbeat timeout in seconds
|
||||||
timesync_rate: 10.0 # TIMESYNC rate in Hertz (feature disabled if 0.0)
|
timesync_rate: 10.0 # TIMESYNC rate in Hertz (feature disabled if 0.0)
|
||||||
system_time_rate: 1.0 # send system time to FCU rate in Hertz (disabled if 0.0)
|
system_time_rate: 1.0 # send system time to FCU rate in Hertz (disabled if 0.0)
|
||||||
|
|
||||||
@@ -13,7 +13,6 @@ time:
|
|||||||
time_ref_source: "fcu" # time_reference source
|
time_ref_source: "fcu" # time_reference source
|
||||||
timesync_mode: MAVLINK
|
timesync_mode: MAVLINK
|
||||||
timesync_avg_alpha: 0.6 # timesync averaging factor
|
timesync_avg_alpha: 0.6 # timesync averaging factor
|
||||||
publish_sim_time: false # don't publish /clock
|
|
||||||
|
|
||||||
global_position:
|
global_position:
|
||||||
frame_id: "map" # origin frame
|
frame_id: "map" # origin frame
|
||||||
@@ -78,9 +77,6 @@ distance_sensor:
|
|||||||
field_of_view: 0.5
|
field_of_view: 0.5
|
||||||
rangefinder_sub:
|
rangefinder_sub:
|
||||||
subscriber: true
|
subscriber: true
|
||||||
id: 1
|
|
||||||
orientation: PITCH_270
|
|
||||||
covariance: 1 # cm
|
|
||||||
|
|
||||||
# fake_gps
|
# fake_gps
|
||||||
fake_gps:
|
fake_gps:
|
||||||
|
|||||||
@@ -1,4 +0,0 @@
|
|||||||
<launch>
|
|
||||||
<!-- shortcut for running the simulation (`roslaunch clover simulator.launch`) -->
|
|
||||||
<include file="$(find clover_simulation)/launch/simulator.launch"/>
|
|
||||||
</launch>
|
|
||||||
@@ -1,38 +0,0 @@
|
|||||||
uint8 MODE_NONE = 0
|
|
||||||
uint8 MODE_NAVIGATE = 1
|
|
||||||
uint8 MODE_NAVIGATE_GLOBAL = 2
|
|
||||||
uint8 MODE_POSITION = 3
|
|
||||||
uint8 MODE_VELOCITY = 4
|
|
||||||
uint8 MODE_ATTITUDE = 5
|
|
||||||
uint8 MODE_RATES = 6
|
|
||||||
|
|
||||||
uint8 YAW_MODE_YAW = 0
|
|
||||||
uint8 YAW_MODE_YAW_RATE = 1
|
|
||||||
uint8 YAW_MODE_YAW_TOWARDS = 2
|
|
||||||
|
|
||||||
# type of offboard control
|
|
||||||
uint8 mode
|
|
||||||
uint8 yaw_mode
|
|
||||||
|
|
||||||
# targets
|
|
||||||
float32 x
|
|
||||||
float32 y
|
|
||||||
float32 z
|
|
||||||
float32 speed
|
|
||||||
float32 lat
|
|
||||||
float32 lon
|
|
||||||
float32 vx
|
|
||||||
float32 vy
|
|
||||||
float32 vz
|
|
||||||
float32 roll
|
|
||||||
float32 pitch
|
|
||||||
float32 yaw
|
|
||||||
float32 roll_rate
|
|
||||||
float32 pitch_rate
|
|
||||||
float32 yaw_rate
|
|
||||||
float32 thrust
|
|
||||||
|
|
||||||
# frames of reference
|
|
||||||
string xy_frame_id
|
|
||||||
string z_frame_id
|
|
||||||
string yaw_frame_id
|
|
||||||
@@ -1,7 +1,7 @@
|
|||||||
<?xml version="1.0"?>
|
<?xml version="1.0"?>
|
||||||
<package format="3">
|
<package format="2">
|
||||||
<name>clover</name>
|
<name>clover</name>
|
||||||
<version>0.23.0</version>
|
<version>0.21.1</version>
|
||||||
<description>The Clover package</description>
|
<description>The Clover package</description>
|
||||||
|
|
||||||
<maintainer email="okalachev@gmail.com">Oleg Kalachev</maintainer>
|
<maintainer email="okalachev@gmail.com">Oleg Kalachev</maintainer>
|
||||||
@@ -37,13 +37,10 @@
|
|||||||
<depend>rosbridge_server</depend>
|
<depend>rosbridge_server</depend>
|
||||||
<depend>web_video_server</depend>
|
<depend>web_video_server</depend>
|
||||||
<depend>tf2_web_republisher</depend>
|
<depend>tf2_web_republisher</depend>
|
||||||
<depend>libxml2</depend>
|
<depend>python-lxml</depend>
|
||||||
<depend>libxslt</depend>
|
|
||||||
<depend condition="$ROS_PYTHON_VERSION == 2">python-lxml</depend>
|
|
||||||
<depend condition="$ROS_PYTHON_VERSION == 3">python3-lxml</depend>
|
|
||||||
<depend>dynamic_reconfigure</depend>
|
|
||||||
<exec_depend>python-pymavlink</exec_depend>
|
<exec_depend>python-pymavlink</exec_depend>
|
||||||
<test_depend>ros_pytest</test_depend>
|
<!-- Use test_depend for packages you need only for testing: -->
|
||||||
|
<!-- <test_depend>gtest</test_depend> -->
|
||||||
|
|
||||||
<!-- The export tag contains other, unspecified, tags -->
|
<!-- The export tag contains other, unspecified, tags -->
|
||||||
<export>
|
<export>
|
||||||
|
|||||||
@@ -1,4 +1,5 @@
|
|||||||
flask==1.1.1
|
flask==1.1.1
|
||||||
|
docopt==0.6.2
|
||||||
geopy==1.11.0
|
geopy==1.11.0
|
||||||
smbus2==0.3.0
|
smbus2==0.3.0
|
||||||
VL53L1X==0.0.5
|
VL53L1X==0.0.5
|
||||||
|
|||||||
@@ -1,11 +0,0 @@
|
|||||||
## ! DO NOT MANUALLY INVOKE THIS setup.py, USE CATKIN INSTEAD
|
|
||||||
|
|
||||||
from distutils.core import setup
|
|
||||||
from catkin_pkg.python_setup import generate_distutils_setup
|
|
||||||
|
|
||||||
# fetch values from package.xml
|
|
||||||
setup_args = generate_distutils_setup(
|
|
||||||
packages=['clover'],
|
|
||||||
package_dir={'': 'src'})
|
|
||||||
|
|
||||||
setup(**setup_args)
|
|
||||||
@@ -1,90 +0,0 @@
|
|||||||
#!/usr/bin/env python3
|
|
||||||
|
|
||||||
import rospy
|
|
||||||
import math
|
|
||||||
import signal
|
|
||||||
import sys
|
|
||||||
import dynamic_reconfigure.client
|
|
||||||
from clover import srv
|
|
||||||
from std_srvs.srv import Trigger
|
|
||||||
from sensor_msgs.msg import Range
|
|
||||||
from aruco_pose.msg import MarkerArray
|
|
||||||
from util import handle_response
|
|
||||||
|
|
||||||
rospy.init_node('autotest_aruco', disable_signals=True) # disable signals to allow interrupting with ctrl+c
|
|
||||||
|
|
||||||
try:
|
|
||||||
flow_client = dynamic_reconfigure.client.Client('optical_flow', timeout=2)
|
|
||||||
except rospy.ROSException:
|
|
||||||
flow_client = None
|
|
||||||
print('Cannot configure optical flow, skip')
|
|
||||||
|
|
||||||
get_telemetry = rospy.ServiceProxy('get_telemetry', srv.GetTelemetry)
|
|
||||||
navigate = handle_response(rospy.ServiceProxy('navigate', srv.Navigate))
|
|
||||||
land = handle_response(rospy.ServiceProxy('land', Trigger))
|
|
||||||
|
|
||||||
def interrupt(sig, frame):
|
|
||||||
print('\nInterrupted, landing...')
|
|
||||||
land()
|
|
||||||
sys.exit(0)
|
|
||||||
|
|
||||||
signal.signal(signal.SIGINT, interrupt)
|
|
||||||
|
|
||||||
def print_current_map_position():
|
|
||||||
telem = get_telemetry()
|
|
||||||
dist = rospy.wait_for_message('rangefinder/range', Range).range
|
|
||||||
print('Map position:\tx={:.1f}\ty={:.1f}\tz={:.1f}\tyaw={:.1f}\tdist={:.2f}'.format(telem.x, telem.y, telem.z, telem.yaw, dist))
|
|
||||||
|
|
||||||
def navigate_wait(x=0, y=0, z=0, yaw=math.nan, speed=0.5, frame_id='body', tolerance=0.2, auto_arm=False):
|
|
||||||
res = navigate(x=x, y=y, z=z, yaw=yaw, speed=speed, frame_id=frame_id, auto_arm=auto_arm)
|
|
||||||
|
|
||||||
if not res.success:
|
|
||||||
return res
|
|
||||||
|
|
||||||
while not rospy.is_shutdown():
|
|
||||||
telem = get_telemetry(frame_id='navigate_target')
|
|
||||||
if math.sqrt(telem.x ** 2 + telem.y ** 2 + telem.z ** 2) < tolerance:
|
|
||||||
return res
|
|
||||||
rospy.sleep(0.2)
|
|
||||||
|
|
||||||
markers = rospy.wait_for_message('aruco_map/map', MarkerArray, timeout=3)
|
|
||||||
left = min(marker.pose.position.x for marker in markers.markers)
|
|
||||||
bottom = min(marker.pose.position.y for marker in markers.markers)
|
|
||||||
width = max(marker.pose.position.x for marker in markers.markers)
|
|
||||||
height = max(marker.pose.position.y for marker in markers.markers)
|
|
||||||
center_x = left + width / 2
|
|
||||||
center_y = bottom + height / 2
|
|
||||||
|
|
||||||
print('Map rect: %g %g - %g %g' % (left, bottom, width, height))
|
|
||||||
|
|
||||||
input('Take off and hover 1 m [enter] ')
|
|
||||||
navigate_wait(x=0, y=0, z=1, frame_id='body', auto_arm=True)
|
|
||||||
print_current_map_position()
|
|
||||||
|
|
||||||
input('Go to corner %g %g 1.5 speed 1 [enter] ' % (width, height))
|
|
||||||
navigate_wait(x=width, y=height, z=1.5, speed=1, frame_id='aruco_map')
|
|
||||||
print_current_map_position()
|
|
||||||
|
|
||||||
input('Go to center %g %g 1.5 speed 5 [enter] ' % (center_x, center_y))
|
|
||||||
navigate_wait(x=center_x, y=center_y, z=1.5, speed=5, frame_id='aruco_map')
|
|
||||||
print_current_map_position()
|
|
||||||
|
|
||||||
if flow_client:
|
|
||||||
input('Disable optical flow and keep hovering [enter] ')
|
|
||||||
flow_client.update_configuration({'enabled': False})
|
|
||||||
rospy.sleep(5)
|
|
||||||
|
|
||||||
input('Enable optical flow back [enter] ')
|
|
||||||
flow_client.update_configuration({'enabled': True})
|
|
||||||
|
|
||||||
input('Go to side 1 %g 2 heading top [enter] ' % (center_y))
|
|
||||||
navigate_wait(x=1, y=center_y, z=2, yaw=1.57, frame_id='aruco_map')
|
|
||||||
print_current_map_position()
|
|
||||||
|
|
||||||
marker_id = markers.markers[0].id
|
|
||||||
input('Go to marker %d z=1.5 [enter] ' % marker_id)
|
|
||||||
navigate_wait(x=0, y=0, z=1.5, yaw=0, frame_id='aruco_%d' % marker_id)
|
|
||||||
print_current_map_position()
|
|
||||||
|
|
||||||
input('Perform landing [enter] ')
|
|
||||||
land()
|
|
||||||
@@ -1,99 +0,0 @@
|
|||||||
#!/usr/bin/env python3
|
|
||||||
|
|
||||||
import rospy
|
|
||||||
import math
|
|
||||||
from math import nan, inf
|
|
||||||
import signal
|
|
||||||
import sys
|
|
||||||
from clover import srv
|
|
||||||
from std_srvs.srv import Trigger
|
|
||||||
from sensor_msgs.msg import Range
|
|
||||||
from util import handle_response
|
|
||||||
|
|
||||||
rospy.init_node('autotest_flight', disable_signals=True) # disable signals to allow interrupting with ctrl+c
|
|
||||||
|
|
||||||
get_telemetry = rospy.ServiceProxy('get_telemetry', srv.GetTelemetry)
|
|
||||||
navigate = handle_response(rospy.ServiceProxy('navigate', srv.Navigate))
|
|
||||||
navigate_global = handle_response(rospy.ServiceProxy('navigate_global', srv.NavigateGlobal))
|
|
||||||
set_yaw = handle_response(rospy.ServiceProxy('set_yaw', srv.SetYaw))
|
|
||||||
set_yaw_rate = handle_response(rospy.ServiceProxy('set_yaw_rate', srv.SetYawRate))
|
|
||||||
set_position = handle_response(rospy.ServiceProxy('set_position', srv.SetPosition))
|
|
||||||
set_velocity = handle_response(rospy.ServiceProxy('set_velocity', srv.SetVelocity))
|
|
||||||
set_attitude = handle_response(rospy.ServiceProxy('set_attitude', srv.SetAttitude))
|
|
||||||
set_rates = handle_response(rospy.ServiceProxy('set_rates', srv.SetRates))
|
|
||||||
land = handle_response(rospy.ServiceProxy('land', Trigger))
|
|
||||||
|
|
||||||
def interrupt(sig, frame):
|
|
||||||
print('\nInterrupted, landing...')
|
|
||||||
land()
|
|
||||||
sys.exit(0)
|
|
||||||
|
|
||||||
signal.signal(signal.SIGINT, interrupt)
|
|
||||||
|
|
||||||
def navigate_wait(x=0, y=0, z=0, yaw=nan, speed=0.5, frame_id='body', tolerance=0.2, auto_arm=False):
|
|
||||||
res = navigate(x=x, y=y, z=z, yaw=yaw, speed=speed, frame_id=frame_id, auto_arm=auto_arm)
|
|
||||||
|
|
||||||
if not res.success:
|
|
||||||
return res
|
|
||||||
|
|
||||||
while not rospy.is_shutdown():
|
|
||||||
telem = get_telemetry(frame_id='navigate_target')
|
|
||||||
if math.sqrt(telem.x ** 2 + telem.y ** 2 + telem.z ** 2) < tolerance:
|
|
||||||
return res
|
|
||||||
rospy.sleep(0.2)
|
|
||||||
|
|
||||||
def print_distance():
|
|
||||||
dist = rospy.wait_for_message('rangefinder/range', Range).range
|
|
||||||
print('Distance: {:.2f}'.format(dist))
|
|
||||||
|
|
||||||
input('Take off and hover 1 m [enter] ')
|
|
||||||
navigate_wait(z=1, frame_id='body', auto_arm=True)
|
|
||||||
print_distance()
|
|
||||||
start = get_telemetry()
|
|
||||||
|
|
||||||
input('Fly forward 2 m [enter] ')
|
|
||||||
navigate_wait(x=2, frame_id='navigate_target')
|
|
||||||
print_distance()
|
|
||||||
|
|
||||||
input('Climb 0.5 m [enter] ')
|
|
||||||
navigate_wait(z=0.5, frame_id='navigate_target')
|
|
||||||
print_distance()
|
|
||||||
|
|
||||||
input('Rotate left 90° [enter] ')
|
|
||||||
navigate(yaw=math.pi / 2, frame_id='navigate_target')
|
|
||||||
rospy.sleep(3)
|
|
||||||
|
|
||||||
input('Use set_velocity to fly forward 2 m speed 1 [enter]')
|
|
||||||
set_velocity(vx=1, vy=0.0, vz=0, frame_id='body')
|
|
||||||
rospy.sleep(2)
|
|
||||||
set_position(frame_id='body')
|
|
||||||
|
|
||||||
input('Rotate right 90° using set_yaw [enter] ')
|
|
||||||
set_yaw(yaw=-math.pi / 2, frame_id='navigate_target')
|
|
||||||
rospy.sleep(3)
|
|
||||||
|
|
||||||
input('Use set_attitude to fly backwards [enter]')
|
|
||||||
set_attitude(roll=0, pitch=-0.3, yaw=0, thrust=0.5, frame_id='body')
|
|
||||||
rospy.sleep(0.3)
|
|
||||||
set_position(frame_id='body')
|
|
||||||
|
|
||||||
input('Use set_attitude to fly right [enter]')
|
|
||||||
set_attitude(roll=0.3, pitch=0, yaw=0, thrust=0.5, frame_id='body')
|
|
||||||
rospy.sleep(0.5)
|
|
||||||
set_position(frame_id='body')
|
|
||||||
|
|
||||||
input('Use set_rates to fly right [enter]')
|
|
||||||
set_rates(roll_rate=1.2, thrust=0.5)
|
|
||||||
rospy.sleep(0.4)
|
|
||||||
set_position(frame_id='body')
|
|
||||||
|
|
||||||
input('Rotate 360° to the right using set_yaw_rate [enter]')
|
|
||||||
set_yaw_rate(yaw_rate=-1)
|
|
||||||
rospy.sleep(2 * math.pi)
|
|
||||||
set_position(frame_id='body')
|
|
||||||
|
|
||||||
input('Return to start point heading forward [enter]')
|
|
||||||
navigate_wait(x=start.x, y=start.y, z=start.z, yaw=inf, speed=1, frame_id='map')
|
|
||||||
|
|
||||||
input('Land [enter]')
|
|
||||||
land()
|
|
||||||
@@ -1,72 +0,0 @@
|
|||||||
#!/usr/bin/env python3
|
|
||||||
|
|
||||||
import rospy
|
|
||||||
import functools
|
|
||||||
from clover.srv import SetLEDEffect
|
|
||||||
from led_msgs.srv import SetLEDs
|
|
||||||
from led_msgs.msg import LEDStateArray, LEDState
|
|
||||||
from util import handle_response
|
|
||||||
|
|
||||||
rospy.init_node('autotest_led', disable_signals=True)
|
|
||||||
|
|
||||||
set_leds = handle_response(rospy.ServiceProxy('led/set_leds', SetLEDs))
|
|
||||||
set_effect = handle_response(rospy.ServiceProxy('led/set_effect', SetLEDEffect))
|
|
||||||
|
|
||||||
led_count = len(rospy.wait_for_message('led/state', LEDStateArray, timeout=10).leds)
|
|
||||||
print('LED count =', led_count)
|
|
||||||
|
|
||||||
print('== Testing effects ==')
|
|
||||||
|
|
||||||
input('Fill red [enter] ')
|
|
||||||
set_effect(r=255, g=0, b=0)
|
|
||||||
|
|
||||||
input('Fill green [enter] ')
|
|
||||||
set_effect(r=0, g=100, b=0)
|
|
||||||
|
|
||||||
input('Blink white [enter] ')
|
|
||||||
set_effect(effect='blink', r=255, g=255, b=255)
|
|
||||||
rospy.sleep(3)
|
|
||||||
|
|
||||||
input('Blink fast violet [enter] ')
|
|
||||||
set_effect(effect='blink_fast', r=220, g=20, b=250)
|
|
||||||
rospy.sleep(3)
|
|
||||||
|
|
||||||
input('Fade to blue [enter] ')
|
|
||||||
set_effect(effect='fade', r=0, g=0, b=255)
|
|
||||||
|
|
||||||
input('Wipe to yellow [enter] ')
|
|
||||||
set_effect(effect='wipe', r=255, g=255, b=40)
|
|
||||||
|
|
||||||
input('Flash red [enter] ')
|
|
||||||
set_effect(effect='flash', r=255, g=0, b=0)
|
|
||||||
rospy.sleep(1)
|
|
||||||
|
|
||||||
input('Rainbow [enter] ')
|
|
||||||
set_effect(effect='rainbow')
|
|
||||||
rospy.sleep(4)
|
|
||||||
|
|
||||||
input('Rainbow fill [enter] ')
|
|
||||||
set_effect(effect='rainbow_fill')
|
|
||||||
rospy.sleep(4)
|
|
||||||
|
|
||||||
input('Turn off [enter] ')
|
|
||||||
set_effect()
|
|
||||||
|
|
||||||
print('== Testing low-level control ==')
|
|
||||||
|
|
||||||
input('Fill orange [enter] ')
|
|
||||||
set_leds(leds=[LEDState(index=i, r=245, g=155, b=0) for i in range(led_count)])
|
|
||||||
|
|
||||||
input('Fill blue gradient [enter] ')
|
|
||||||
set_leds(leds=[LEDState(index=i, r=0, g=20, b=int(255 * i / led_count)) for i in range(led_count)])
|
|
||||||
|
|
||||||
input('Animate green dot [enter] ')
|
|
||||||
set_effect()
|
|
||||||
for i in range(led_count):
|
|
||||||
if i > 0:
|
|
||||||
set_leds(leds=[LEDState(index=i - 1, r=0, g=0, b=0)])
|
|
||||||
set_leds(leds=[LEDState(index=i, r=0, g=255, b=0)])
|
|
||||||
rospy.sleep(0.05)
|
|
||||||
|
|
||||||
input('Turn off [enter] ')
|
|
||||||
set_effect()
|
|
||||||
@@ -1,11 +0,0 @@
|
|||||||
import functools
|
|
||||||
|
|
||||||
# decorator to handle response and print error message
|
|
||||||
def handle_response(fn):
|
|
||||||
@functools.wraps(fn)
|
|
||||||
def wrapper(*args, **kwargs):
|
|
||||||
res = fn(*args, **kwargs)
|
|
||||||
if not res.success:
|
|
||||||
print('\033[91mError:\033[0m {}'.format(res.message))
|
|
||||||
return res
|
|
||||||
return wrapper
|
|
||||||
@@ -1,35 +0,0 @@
|
|||||||
import rospy
|
|
||||||
from threading import Thread, Event
|
|
||||||
|
|
||||||
def long_callback(fn):
|
|
||||||
"""
|
|
||||||
Decorator fixing a rospy issue for long-running topic callbacks, primarily
|
|
||||||
for image processing.
|
|
||||||
|
|
||||||
See: https://github.com/ros/ros_comm/issues/1901.
|
|
||||||
|
|
||||||
Usage example:
|
|
||||||
|
|
||||||
@long_callback
|
|
||||||
def image_callback(msg):
|
|
||||||
# perform image processing
|
|
||||||
# ...
|
|
||||||
|
|
||||||
rospy.Subscriber('main_camera/image_raw', Image, image_callback, queue_size=1)
|
|
||||||
"""
|
|
||||||
e = Event()
|
|
||||||
|
|
||||||
def thread():
|
|
||||||
while not rospy.is_shutdown():
|
|
||||||
e.wait()
|
|
||||||
e.clear()
|
|
||||||
fn(thread.current_msg)
|
|
||||||
|
|
||||||
thread.current_msg = None
|
|
||||||
Thread(target=thread, daemon=True).start()
|
|
||||||
|
|
||||||
def wrapper(msg):
|
|
||||||
thread.current_msg = msg
|
|
||||||
e.set()
|
|
||||||
|
|
||||||
return wrapper
|
|
||||||
@@ -12,7 +12,6 @@
|
|||||||
|
|
||||||
#include <ros/ros.h>
|
#include <ros/ros.h>
|
||||||
#include <string>
|
#include <string>
|
||||||
#include <vector>
|
|
||||||
#include <boost/algorithm/string.hpp>
|
#include <boost/algorithm/string.hpp>
|
||||||
|
|
||||||
#include <clover/SetLEDEffect.h>
|
#include <clover/SetLEDEffect.h>
|
||||||
@@ -30,7 +29,7 @@ ros::Timer timer;
|
|||||||
ros::Time start_time;
|
ros::Time start_time;
|
||||||
double blink_rate, blink_fast_rate, flash_delay, fade_period, wipe_period, rainbow_period;
|
double blink_rate, blink_fast_rate, flash_delay, fade_period, wipe_period, rainbow_period;
|
||||||
double low_battery_threshold;
|
double low_battery_threshold;
|
||||||
std::vector<std::string> error_ignore;
|
bool blink_state;
|
||||||
led_msgs::SetLEDs set_leds;
|
led_msgs::SetLEDs set_leds;
|
||||||
led_msgs::LEDStateArray state, start_state;
|
led_msgs::LEDStateArray state, start_state;
|
||||||
ros::ServiceClient set_leds_srv;
|
ros::ServiceClient set_leds_srv;
|
||||||
@@ -86,8 +85,9 @@ void proceed(const ros::TimerEvent& event)
|
|||||||
set_leds.request.leds.resize(led_count);
|
set_leds.request.leds.resize(led_count);
|
||||||
|
|
||||||
if (current_effect.effect == "blink" || current_effect.effect == "blink_fast") {
|
if (current_effect.effect == "blink" || current_effect.effect == "blink_fast") {
|
||||||
// enable on odd counter
|
blink_state = !blink_state;
|
||||||
if (counter % 2 != 0) {
|
// toggle all leds
|
||||||
|
if (blink_state) {
|
||||||
fill(current_effect.r, current_effect.g, current_effect.b);
|
fill(current_effect.r, current_effect.g, current_effect.b);
|
||||||
} else {
|
} else {
|
||||||
fill(0, 0, 0);
|
fill(0, 0, 0);
|
||||||
@@ -220,7 +220,6 @@ bool setEffect(clover::SetLEDEffect::Request& req, clover::SetLEDEffect::Respons
|
|||||||
counter = 0;
|
counter = 0;
|
||||||
start_state = state;
|
start_state = state;
|
||||||
start_time = ros::Time::now();
|
start_time = ros::Time::now();
|
||||||
proceed({ .current_real = start_time });
|
|
||||||
|
|
||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
@@ -275,10 +274,6 @@ void handleMavrosState(const mavros_msgs::State& msg)
|
|||||||
void handleLog(const rosgraph_msgs::Log& log)
|
void handleLog(const rosgraph_msgs::Log& log)
|
||||||
{
|
{
|
||||||
if (log.level >= rosgraph_msgs::Log::ERROR) {
|
if (log.level >= rosgraph_msgs::Log::ERROR) {
|
||||||
// check if ignored
|
|
||||||
for (auto const& str : error_ignore) {
|
|
||||||
if (log.msg.find(str) != std::string::npos) return;
|
|
||||||
}
|
|
||||||
notify("error");
|
notify("error");
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -307,7 +302,6 @@ int main(int argc, char **argv)
|
|||||||
nh_priv.param("rainbow_period", rainbow_period, 5.0);
|
nh_priv.param("rainbow_period", rainbow_period, 5.0);
|
||||||
|
|
||||||
nh_priv.param("notify/low_battery/threshold", low_battery_threshold, 3.7);
|
nh_priv.param("notify/low_battery/threshold", low_battery_threshold, 3.7);
|
||||||
nh_priv.param("notify/error/ignore", error_ignore, {});
|
|
||||||
|
|
||||||
ros::service::waitForService("set_leds"); // cannot work without set_leds service
|
ros::service::waitForService("set_leds"); // cannot work without set_leds service
|
||||||
set_leds_srv = nh.serviceClient<led_msgs::SetLEDs>("set_leds", true);
|
set_leds_srv = nh.serviceClient<led_msgs::SetLEDs>("set_leds", true);
|
||||||
@@ -319,8 +313,8 @@ int main(int argc, char **argv)
|
|||||||
|
|
||||||
auto set_effect = nh.advertiseService("set_effect", &setEffect);
|
auto set_effect = nh.advertiseService("set_effect", &setEffect);
|
||||||
|
|
||||||
auto mavros_state_sub = nh.subscribe("mavros/state", 1, &handleMavrosState);
|
auto mavros_state_sub = nh.subscribe("/mavros/state", 1, &handleMavrosState);
|
||||||
auto battery_sub = nh.subscribe("mavros/battery", 1, &handleBattery);
|
auto battery_sub = nh.subscribe("/mavros/battery", 1, &handleBattery);
|
||||||
auto rosout_sub = nh.subscribe("/rosout_agg", 1, &handleLog);
|
auto rosout_sub = nh.subscribe("/rosout_agg", 1, &handleLog);
|
||||||
|
|
||||||
timer = nh.createTimer(ros::Duration(0), &proceed, false, false);
|
timer = nh.createTimer(ros::Duration(0), &proceed, false, false);
|
||||||
|
|||||||
@@ -22,14 +22,11 @@
|
|||||||
#include <tf2/utils.h>
|
#include <tf2/utils.h>
|
||||||
#include <tf2_ros/transform_listener.h>
|
#include <tf2_ros/transform_listener.h>
|
||||||
#include <tf2_geometry_msgs/tf2_geometry_msgs.h>
|
#include <tf2_geometry_msgs/tf2_geometry_msgs.h>
|
||||||
#include <dynamic_reconfigure/server.h>
|
|
||||||
#include <mavros_msgs/OpticalFlowRad.h>
|
#include <mavros_msgs/OpticalFlowRad.h>
|
||||||
#include <sensor_msgs/Imu.h>
|
#include <sensor_msgs/Imu.h>
|
||||||
#include <geometry_msgs/PoseStamped.h>
|
|
||||||
#include <geometry_msgs/Vector3Stamped.h>
|
#include <geometry_msgs/Vector3Stamped.h>
|
||||||
#include <geometry_msgs/PointStamped.h>
|
#include <geometry_msgs/PointStamped.h>
|
||||||
#include <geometry_msgs/TwistStamped.h>
|
#include <geometry_msgs/TwistStamped.h>
|
||||||
#include <clover/FlowConfig.h>
|
|
||||||
|
|
||||||
using cv::Mat;
|
using cv::Mat;
|
||||||
|
|
||||||
@@ -41,7 +38,6 @@ public:
|
|||||||
{}
|
{}
|
||||||
|
|
||||||
private:
|
private:
|
||||||
bool enabled_;
|
|
||||||
ros::Publisher flow_pub_, velo_pub_, shift_pub_;
|
ros::Publisher flow_pub_, velo_pub_, shift_pub_;
|
||||||
ros::Time prev_stamp_;
|
ros::Time prev_stamp_;
|
||||||
std::string fcu_frame_id_, local_frame_id_;
|
std::string fcu_frame_id_, local_frame_id_;
|
||||||
@@ -57,11 +53,6 @@ private:
|
|||||||
std::unique_ptr<tf2_ros::Buffer> tf_buffer_;
|
std::unique_ptr<tf2_ros::Buffer> tf_buffer_;
|
||||||
std::unique_ptr<tf2_ros::TransformListener> tf_listener_;
|
std::unique_ptr<tf2_ros::TransformListener> tf_listener_;
|
||||||
bool calc_flow_gyro_;
|
bool calc_flow_gyro_;
|
||||||
float flow_gyro_default_;
|
|
||||||
bool disable_on_vpe_;
|
|
||||||
ros::Subscriber vpe_sub_;
|
|
||||||
ros::Time last_vpe_time_;
|
|
||||||
std::shared_ptr<dynamic_reconfigure::Server<clover::FlowConfig>> dyn_srv_;
|
|
||||||
|
|
||||||
void onInit()
|
void onInit()
|
||||||
{
|
{
|
||||||
@@ -78,30 +69,20 @@ private:
|
|||||||
roi_px_ = nh_priv.param("roi", 128);
|
roi_px_ = nh_priv.param("roi", 128);
|
||||||
roi_rad_ = nh_priv.param("roi_rad", 0.0);
|
roi_rad_ = nh_priv.param("roi_rad", 0.0);
|
||||||
calc_flow_gyro_ = nh_priv.param("calc_flow_gyro", false);
|
calc_flow_gyro_ = nh_priv.param("calc_flow_gyro", false);
|
||||||
flow_gyro_default_ = nh_priv.param("flow_gyro_default", NAN);
|
|
||||||
|
|
||||||
|
img_sub_ = it.subscribeCamera("image_raw", 1, &OpticalFlow::flow, this);
|
||||||
img_pub_ = it_priv.advertise("debug", 1);
|
img_pub_ = it_priv.advertise("debug", 1);
|
||||||
flow_pub_ = nh.advertise<mavros_msgs::OpticalFlowRad>("mavros/px4flow/raw/send", 1);
|
flow_pub_ = nh.advertise<mavros_msgs::OpticalFlowRad>("mavros/px4flow/raw/send", 1);
|
||||||
velo_pub_ = nh_priv.advertise<geometry_msgs::TwistStamped>("angular_velocity", 1);
|
velo_pub_ = nh_priv.advertise<geometry_msgs::TwistStamped>("angular_velocity", 1);
|
||||||
shift_pub_ = nh_priv.advertise<geometry_msgs::Vector3Stamped>("shift", 1);
|
shift_pub_ = nh_priv.advertise<geometry_msgs::Vector3Stamped>("shift", 1);
|
||||||
|
|
||||||
|
flow_.integrated_xgyro = NAN; // no IMU available
|
||||||
|
flow_.integrated_ygyro = NAN;
|
||||||
|
flow_.integrated_zgyro = NAN;
|
||||||
flow_.time_delta_distance_us = 0;
|
flow_.time_delta_distance_us = 0;
|
||||||
flow_.distance = -1; // no distance sensor available
|
flow_.distance = -1; // no distance sensor available
|
||||||
flow_.temperature = 0;
|
flow_.temperature = 0;
|
||||||
|
|
||||||
img_sub_ = it.subscribeCamera("image_raw", 1, &OpticalFlow::flow, this);
|
|
||||||
|
|
||||||
disable_on_vpe_ = nh_priv.param("disable_on_vpe", false);
|
|
||||||
if (disable_on_vpe_) {
|
|
||||||
vpe_sub_ = nh.subscribe("mavros/vision_pose/pose", 1, &OpticalFlow::vpeCallback, this);
|
|
||||||
}
|
|
||||||
|
|
||||||
dyn_srv_ = std::make_shared<dynamic_reconfigure::Server<clover::FlowConfig>>(nh_priv);
|
|
||||||
dynamic_reconfigure::Server<clover::FlowConfig>::CallbackType cb;
|
|
||||||
|
|
||||||
cb = std::bind(&OpticalFlow::paramCallback, this, std::placeholders::_1, std::placeholders::_2);
|
|
||||||
dyn_srv_->setCallback(cb);
|
|
||||||
|
|
||||||
NODELET_INFO("Optical Flow initialized");
|
NODELET_INFO("Optical Flow initialized");
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -128,14 +109,6 @@ private:
|
|||||||
|
|
||||||
void flow(const sensor_msgs::ImageConstPtr& msg, const sensor_msgs::CameraInfoConstPtr& cinfo)
|
void flow(const sensor_msgs::ImageConstPtr& msg, const sensor_msgs::CameraInfoConstPtr& cinfo)
|
||||||
{
|
{
|
||||||
if (!enabled_) return;
|
|
||||||
|
|
||||||
if (disable_on_vpe_ &&
|
|
||||||
!last_vpe_time_.isZero() &&
|
|
||||||
(msg->header.stamp - last_vpe_time_).toSec() < 0.1) {
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
parseCameraInfo(cinfo);
|
parseCameraInfo(cinfo);
|
||||||
|
|
||||||
auto img = cv_bridge::toCvShare(msg, "mono8")->image;
|
auto img = cv_bridge::toCvShare(msg, "mono8")->image;
|
||||||
@@ -169,7 +142,7 @@ private:
|
|||||||
|
|
||||||
img.convertTo(curr_, CV_32F);
|
img.convertTo(curr_, CV_32F);
|
||||||
|
|
||||||
if (prev_.empty() || (msg->header.stamp - prev_stamp_).toSec() > 0.1) { // outdated previous frame
|
if (prev_.empty()) {
|
||||||
prev_ = curr_.clone();
|
prev_ = curr_.clone();
|
||||||
prev_stamp_ = msg->header.stamp;
|
prev_stamp_ = msg->header.stamp;
|
||||||
cv::createHanningWindow(hann_, curr_.size(), CV_32F);
|
cv::createHanningWindow(hann_, curr_.size(), CV_32F);
|
||||||
@@ -179,7 +152,7 @@ private:
|
|||||||
cv::Point2d shift = cv::phaseCorrelate(prev_, curr_, hann_, &response);
|
cv::Point2d shift = cv::phaseCorrelate(prev_, curr_, hann_, &response);
|
||||||
|
|
||||||
// Publish raw shift in pixels
|
// Publish raw shift in pixels
|
||||||
geometry_msgs::Vector3Stamped shift_vec;
|
static geometry_msgs::Vector3Stamped shift_vec;
|
||||||
shift_vec.header.stamp = msg->header.stamp;
|
shift_vec.header.stamp = msg->header.stamp;
|
||||||
shift_vec.header.frame_id = msg->header.frame_id;
|
shift_vec.header.frame_id = msg->header.frame_id;
|
||||||
shift_vec.vector.x = shift.x;
|
shift_vec.vector.x = shift.x;
|
||||||
@@ -205,8 +178,8 @@ private:
|
|||||||
double flow_x = atan2(points_undist[0].x, focal_length_x);
|
double flow_x = atan2(points_undist[0].x, focal_length_x);
|
||||||
double flow_y = atan2(points_undist[0].y, focal_length_y);
|
double flow_y = atan2(points_undist[0].y, focal_length_y);
|
||||||
|
|
||||||
// Convert to FCU frame
|
// // Convert to FCU frame
|
||||||
geometry_msgs::Vector3Stamped flow_camera, flow_fcu;
|
static geometry_msgs::Vector3Stamped flow_camera, flow_fcu;
|
||||||
flow_camera.header.frame_id = msg->header.frame_id;
|
flow_camera.header.frame_id = msg->header.frame_id;
|
||||||
flow_camera.header.stamp = msg->header.stamp;
|
flow_camera.header.stamp = msg->header.stamp;
|
||||||
flow_camera.vector.x = flow_y; // +y means counter-clockwise rotation around Y axis
|
flow_camera.vector.x = flow_y; // +y means counter-clockwise rotation around Y axis
|
||||||
@@ -222,21 +195,18 @@ private:
|
|||||||
ros::Duration integration_time = msg->header.stamp - prev_stamp_;
|
ros::Duration integration_time = msg->header.stamp - prev_stamp_;
|
||||||
uint32_t integration_time_us = integration_time.toSec() * 1.0e6;
|
uint32_t integration_time_us = integration_time.toSec() * 1.0e6;
|
||||||
|
|
||||||
// Calculate flow gyro
|
|
||||||
flow_.integrated_xgyro = flow_gyro_default_;
|
|
||||||
flow_.integrated_ygyro = flow_gyro_default_;
|
|
||||||
flow_.integrated_zgyro = flow_gyro_default_;
|
|
||||||
|
|
||||||
if (calc_flow_gyro_) {
|
if (calc_flow_gyro_) {
|
||||||
try {
|
try {
|
||||||
auto flow_gyro_camera = calcFlowGyro(msg->header.frame_id, prev_stamp_, msg->header.stamp);
|
auto flow_gyro_camera = calcFlowGyro(msg->header.frame_id, prev_stamp_, msg->header.stamp);
|
||||||
geometry_msgs::Vector3Stamped flow_gyro_fcu;
|
static geometry_msgs::Vector3Stamped flow_gyro_fcu;
|
||||||
tf_buffer_->transform(flow_gyro_camera, flow_gyro_fcu, fcu_frame_id_);
|
tf_buffer_->transform(flow_gyro_camera, flow_gyro_fcu, fcu_frame_id_);
|
||||||
flow_.integrated_xgyro = flow_gyro_fcu.vector.x;
|
flow_.integrated_xgyro = flow_gyro_fcu.vector.x;
|
||||||
flow_.integrated_ygyro = flow_gyro_fcu.vector.y;
|
flow_.integrated_ygyro = flow_gyro_fcu.vector.y;
|
||||||
flow_.integrated_zgyro = flow_gyro_fcu.vector.z;
|
flow_.integrated_zgyro = flow_gyro_fcu.vector.z;
|
||||||
} catch (const tf2::TransformException& e) {
|
} catch (const tf2::TransformException& e) {
|
||||||
// Transform not available, keep NANs in flow gyro
|
// Invalidate previous frame
|
||||||
|
prev_.release();
|
||||||
|
return;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -248,18 +218,6 @@ private:
|
|||||||
flow_.quality = (uint8_t)(response * 255);
|
flow_.quality = (uint8_t)(response * 255);
|
||||||
flow_pub_.publish(flow_);
|
flow_pub_.publish(flow_);
|
||||||
|
|
||||||
prev_ = curr_.clone();
|
|
||||||
prev_stamp_ = msg->header.stamp;
|
|
||||||
|
|
||||||
// Publish estimated angular velocity
|
|
||||||
geometry_msgs::TwistStamped velo;
|
|
||||||
velo.header.stamp = msg->header.stamp;
|
|
||||||
velo.header.frame_id = fcu_frame_id_;
|
|
||||||
velo.twist.angular.x = flow_fcu.vector.x / integration_time.toSec();
|
|
||||||
velo.twist.angular.y = flow_fcu.vector.y / integration_time.toSec();
|
|
||||||
velo_pub_.publish(velo);
|
|
||||||
|
|
||||||
publish_debug:
|
|
||||||
// Publish debug image
|
// Publish debug image
|
||||||
if (img_pub_.getNumSubscribers() > 0) {
|
if (img_pub_.getNumSubscribers() > 0) {
|
||||||
// publish debug image
|
// publish debug image
|
||||||
@@ -271,6 +229,17 @@ publish_debug:
|
|||||||
out_msg.image = img;
|
out_msg.image = img;
|
||||||
img_pub_.publish(out_msg.toImageMsg());
|
img_pub_.publish(out_msg.toImageMsg());
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Publish estimated angular velocity
|
||||||
|
static geometry_msgs::TwistStamped velo;
|
||||||
|
velo.header.stamp = msg->header.stamp;
|
||||||
|
velo.header.frame_id = fcu_frame_id_;
|
||||||
|
velo.twist.angular.x = flow_.integrated_x / integration_time.toSec();
|
||||||
|
velo.twist.angular.y = flow_.integrated_y / integration_time.toSec();
|
||||||
|
velo_pub_.publish(velo);
|
||||||
|
|
||||||
|
prev_ = curr_.clone();
|
||||||
|
prev_stamp_ = msg->header.stamp;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -291,18 +260,6 @@ publish_debug:
|
|||||||
|
|
||||||
return flow;
|
return flow;
|
||||||
}
|
}
|
||||||
|
|
||||||
void paramCallback(clover::FlowConfig &config, uint32_t level)
|
|
||||||
{
|
|
||||||
enabled_ = config.enabled;
|
|
||||||
if (!enabled_) {
|
|
||||||
prev_ = Mat(); // clear previous frame
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
void vpeCallback(const geometry_msgs::PoseStamped& vpe) {
|
|
||||||
last_vpe_time_ = vpe.header.stamp;
|
|
||||||
}
|
|
||||||
};
|
};
|
||||||
|
|
||||||
PLUGINLIB_EXPORT_CLASS(OpticalFlow, nodelet::Nodelet)
|
PLUGINLIB_EXPORT_CLASS(OpticalFlow, nodelet::Nodelet)
|
||||||
|
|||||||
@@ -91,7 +91,7 @@ private:
|
|||||||
void fakeGCSThread()
|
void fakeGCSThread()
|
||||||
{
|
{
|
||||||
// Awful workaround for fixing PX4 not sending STATUSTEXTs
|
// Awful workaround for fixing PX4 not sending STATUSTEXTs
|
||||||
// if there is no GCS heartbeats.
|
// if there is no GCS hearbeats.
|
||||||
// TODO: use timer
|
// TODO: use timer
|
||||||
// TODO: remove, when PX4 get this fixed.
|
// TODO: remove, when PX4 get this fixed.
|
||||||
ros::Publisher mavlink_pub = nh.advertise<mavros_msgs::Mavlink>("mavlink/to", 1);
|
ros::Publisher mavlink_pub = nh.advertise<mavros_msgs::Mavlink>("mavlink/to", 1);
|
||||||
|
|||||||
@@ -9,14 +9,13 @@
|
|||||||
# The above copyright notice and this permission notice shall be included in all
|
# The above copyright notice and this permission notice shall be included in all
|
||||||
# copies or substantial portions of the Software.
|
# copies or substantial portions of the Software.
|
||||||
|
|
||||||
import os, sys
|
import os
|
||||||
import math
|
import math
|
||||||
import subprocess
|
import subprocess
|
||||||
import re
|
import re
|
||||||
from collections import OrderedDict
|
from collections import OrderedDict
|
||||||
import traceback
|
import traceback
|
||||||
import threading
|
from threading import Event
|
||||||
from threading import Event, Thread, Lock
|
|
||||||
import numpy
|
import numpy
|
||||||
import rospy
|
import rospy
|
||||||
import tf2_ros
|
import tf2_ros
|
||||||
@@ -28,86 +27,67 @@ from mavros_msgs.msg import State, OpticalFlowRad, Mavlink
|
|||||||
from mavros_msgs.srv import ParamGet
|
from mavros_msgs.srv import ParamGet
|
||||||
from geometry_msgs.msg import PoseStamped, TwistStamped, PoseWithCovarianceStamped, Vector3Stamped
|
from geometry_msgs.msg import PoseStamped, TwistStamped, PoseWithCovarianceStamped, Vector3Stamped
|
||||||
from visualization_msgs.msg import MarkerArray as VisualizationMarkerArray
|
from visualization_msgs.msg import MarkerArray as VisualizationMarkerArray
|
||||||
from diagnostic_msgs.msg import DiagnosticArray
|
|
||||||
import tf.transformations as t
|
import tf.transformations as t
|
||||||
from aruco_pose.msg import MarkerArray
|
from aruco_pose.msg import MarkerArray
|
||||||
from mavros import mavlink
|
from mavros import mavlink
|
||||||
import locale
|
|
||||||
|
|
||||||
|
# TODO: check attitude is present
|
||||||
|
# TODO: disk free space
|
||||||
|
# TODO: map, base_link, body
|
||||||
|
# TODO: rc service
|
||||||
|
# TODO: perform commander check, ekf2 status on PX4
|
||||||
|
# TODO: check if FCU params setter succeed
|
||||||
|
# TODO: selfcheck ROS service (with blacklists for checks)
|
||||||
|
|
||||||
|
|
||||||
rospy.init_node('selfcheck')
|
rospy.init_node('selfcheck')
|
||||||
|
|
||||||
os.environ['ROSCONSOLE_FORMAT']='${message}'
|
|
||||||
|
|
||||||
# use user's locale to convert numbers, etc
|
|
||||||
locale.setlocale(locale.LC_ALL, '')
|
|
||||||
|
|
||||||
tf_buffer = tf2_ros.Buffer()
|
tf_buffer = tf2_ros.Buffer()
|
||||||
tf_listener = tf2_ros.TransformListener(tf_buffer)
|
tf_listener = tf2_ros.TransformListener(tf_buffer)
|
||||||
|
|
||||||
|
|
||||||
thread_local = threading.local()
|
failures = []
|
||||||
reports_lock = Lock()
|
infos = []
|
||||||
|
current_check = None
|
||||||
|
|
||||||
# formatting colors
|
|
||||||
if sys.stdout.isatty():
|
|
||||||
GREY = '\033[90m'
|
|
||||||
GREEN = '\033[92m'
|
|
||||||
RED = '\033[31m'
|
|
||||||
END = '\033[0m'
|
|
||||||
else:
|
|
||||||
GREY = GREEN = RED = END = ''
|
|
||||||
|
|
||||||
|
|
||||||
def failure(text, *args):
|
def failure(text, *args):
|
||||||
msg = text % args
|
msg = text % args
|
||||||
thread_local.reports += [{'failure': msg}]
|
rospy.logwarn('%s: %s', current_check, msg)
|
||||||
|
failures.append(msg)
|
||||||
|
|
||||||
|
|
||||||
def info(text, *args):
|
def info(text, *args):
|
||||||
msg = text % args
|
msg = text % args
|
||||||
thread_local.reports += [{'info': msg}]
|
rospy.loginfo('%s: %s', current_check, msg)
|
||||||
|
infos.append(msg)
|
||||||
|
|
||||||
|
|
||||||
def check(name):
|
def check(name):
|
||||||
def inner(fn):
|
def inner(fn):
|
||||||
def wrapper(*args, **kwargs):
|
def wrapper(*args, **kwargs):
|
||||||
start = rospy.get_time()
|
failures[:] = []
|
||||||
thread_local.reports = []
|
infos[:] = []
|
||||||
|
global current_check
|
||||||
|
current_check = name
|
||||||
try:
|
try:
|
||||||
fn(*args, **kwargs)
|
fn(*args, **kwargs)
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
traceback.print_exc()
|
traceback.print_exc()
|
||||||
rospy.logerr('%s: exception occurred', name)
|
rospy.logerr('%s: exception occurred', name)
|
||||||
with reports_lock:
|
return
|
||||||
for report in thread_local.reports:
|
if not failures and not infos:
|
||||||
if 'failure' in report:
|
rospy.loginfo('%s: OK', name)
|
||||||
rospy.logerr('%s: %s', name, report['failure'])
|
|
||||||
elif 'info' in report:
|
|
||||||
rospy.loginfo(GREY + name + END + ': ' + report['info'])
|
|
||||||
if not thread_local.reports:
|
|
||||||
rospy.loginfo(GREY + name + END + ': ' + GREEN + 'OK' + END)
|
|
||||||
if rospy.get_param('~time', False):
|
|
||||||
rospy.loginfo('%s: %.1f sec', name, rospy.get_time() - start)
|
|
||||||
return wrapper
|
return wrapper
|
||||||
return inner
|
return inner
|
||||||
|
|
||||||
|
|
||||||
def ff(value, precision=2):
|
|
||||||
# safely format float or int
|
|
||||||
if value is None:
|
|
||||||
return RED + '???' + END
|
|
||||||
if isinstance(value, float):
|
|
||||||
return ('{:.' + str(precision + 1) + '}').format(value)
|
|
||||||
elif isinstance(value, int):
|
|
||||||
return str(value)
|
|
||||||
|
|
||||||
|
|
||||||
param_get = rospy.ServiceProxy('mavros/param/get', ParamGet)
|
param_get = rospy.ServiceProxy('mavros/param/get', ParamGet)
|
||||||
|
|
||||||
|
|
||||||
def get_param(name, default=None):
|
def get_param(name):
|
||||||
try:
|
try:
|
||||||
res = param_get(param_id=name)
|
res = param_get(param_id=name)
|
||||||
except rospy.ServiceException as e:
|
except rospy.ServiceException as e:
|
||||||
@@ -116,17 +96,12 @@ def get_param(name, default=None):
|
|||||||
|
|
||||||
if not res.success:
|
if not res.success:
|
||||||
failure('unable to retrieve PX4 parameter %s', name)
|
failure('unable to retrieve PX4 parameter %s', name)
|
||||||
return default
|
|
||||||
else:
|
else:
|
||||||
if res.value.integer != 0:
|
if res.value.integer != 0:
|
||||||
return res.value.integer
|
return res.value.integer
|
||||||
return res.value.real
|
return res.value.real
|
||||||
|
|
||||||
|
|
||||||
def get_paramf(name, precision=2):
|
|
||||||
return ff(get_param(name), precision)
|
|
||||||
|
|
||||||
|
|
||||||
recv_event = Event()
|
recv_event = Event()
|
||||||
link = mavutil.mavlink.MAVLink('', 255, 1)
|
link = mavutil.mavlink.MAVLink('', 255, 1)
|
||||||
mavlink_pub = rospy.Publisher('mavlink/to', Mavlink, queue_size=1)
|
mavlink_pub = rospy.Publisher('mavlink/to', Mavlink, queue_size=1)
|
||||||
@@ -163,7 +138,7 @@ def mavlink_exec(cmd, timeout=3.0):
|
|||||||
timeout=3,
|
timeout=3,
|
||||||
baudrate=0,
|
baudrate=0,
|
||||||
count=len(cmd),
|
count=len(cmd),
|
||||||
data=[ord(c) for c in cmd.ljust(70, '\0')])
|
data=map(ord, cmd.ljust(70, '\0')))
|
||||||
msg.pack(link)
|
msg.pack(link)
|
||||||
ros_msg = mavlink.convert_to_rosmsg(msg)
|
ros_msg = mavlink.convert_to_rosmsg(msg)
|
||||||
mavlink_pub.publish(ros_msg)
|
mavlink_pub.publish(ros_msg)
|
||||||
@@ -171,24 +146,6 @@ def mavlink_exec(cmd, timeout=3.0):
|
|||||||
return mavlink_recv
|
return mavlink_recv
|
||||||
|
|
||||||
|
|
||||||
def read_diagnostics(name, key):
|
|
||||||
e = Event()
|
|
||||||
def cb(msg):
|
|
||||||
for status in msg.status:
|
|
||||||
if status.name.lower() == name.lower():
|
|
||||||
for value in status.values:
|
|
||||||
if value.key.lower() == key.lower():
|
|
||||||
cb.value = value.value
|
|
||||||
e.set()
|
|
||||||
return
|
|
||||||
|
|
||||||
cb.value = None
|
|
||||||
sub = rospy.Subscriber('/diagnostics', DiagnosticArray, cb)
|
|
||||||
e.wait(1.0) # wait to read all the diagnostics from nodes publishing them
|
|
||||||
sub.unregister()
|
|
||||||
return cb.value
|
|
||||||
|
|
||||||
|
|
||||||
BOARD_ROTATIONS = {
|
BOARD_ROTATIONS = {
|
||||||
0: 'no rotation',
|
0: 'no rotation',
|
||||||
1: 'yaw 45°',
|
1: 'yaw 45°',
|
||||||
@@ -234,31 +191,26 @@ def check_fcu():
|
|||||||
state = rospy.wait_for_message('mavros/state', State, timeout=3)
|
state = rospy.wait_for_message('mavros/state', State, timeout=3)
|
||||||
if not state.connected:
|
if not state.connected:
|
||||||
failure('no connection to the FCU (check wiring)')
|
failure('no connection to the FCU (check wiring)')
|
||||||
info('fcu_url = %s', rospy.get_param('mavros/fcu_url', '?'))
|
|
||||||
return
|
return
|
||||||
|
|
||||||
if not is_process_running('px4', exact=True): # can't use px4 console in SITL
|
# Make sure the console is available to us
|
||||||
clover_tag = re.compile(r'-cl[oe]ver\.\d+$')
|
mavlink_exec('\n')
|
||||||
clover_fw = False
|
version_str = mavlink_exec('ver all')
|
||||||
|
if version_str == '':
|
||||||
|
info('no version data available from SITL')
|
||||||
|
|
||||||
# Make sure the console is available to us
|
r = re.compile(r'^FW (git tag|version): (v?\d\.\d\.\d.*)$')
|
||||||
mavlink_exec('\n')
|
is_clover_firmware = False
|
||||||
version_str = mavlink_exec('ver all')
|
for ver_line in version_str.split('\n'):
|
||||||
if version_str == '':
|
match = r.search(ver_line)
|
||||||
info('no version data available from SITL')
|
if match is not None:
|
||||||
|
field, version = match.groups()
|
||||||
|
info('firmware %s: %s' % (field, version))
|
||||||
|
if 'clover' in version or 'clever' in version:
|
||||||
|
is_clover_firmware = True
|
||||||
|
|
||||||
for line in version_str.split('\n'):
|
if not is_clover_firmware:
|
||||||
if line.startswith('FW version: '):
|
failure('not running Clover PX4 firmware, https://clover.coex.tech/firmware')
|
||||||
info(line[len('FW version: '):])
|
|
||||||
elif line.startswith('FW git tag: '): # only Clover's firmware
|
|
||||||
tag = line[len('FW git tag: '):]
|
|
||||||
clover_fw = clover_tag.search(tag)
|
|
||||||
info(tag)
|
|
||||||
elif line.startswith('HW arch: '):
|
|
||||||
info(line[len('HW arch: '):])
|
|
||||||
|
|
||||||
if not clover_fw:
|
|
||||||
info('not Clover PX4 firmware, check https://clover.coex.tech/firmware')
|
|
||||||
|
|
||||||
est = get_param('SYS_MC_EST_GROUP')
|
est = get_param('SYS_MC_EST_GROUP')
|
||||||
if est == 1:
|
if est == 1:
|
||||||
@@ -295,29 +247,21 @@ def check_fcu():
|
|||||||
if cbrk_usb_chk != 197848:
|
if cbrk_usb_chk != 197848:
|
||||||
failure('set parameter CBRK_USB_CHK to 197848 for flying with USB connected')
|
failure('set parameter CBRK_USB_CHK to 197848 for flying with USB connected')
|
||||||
|
|
||||||
if not is_process_running('px4', exact=True): # skip battery check in SITL
|
|
||||||
try:
|
|
||||||
battery = rospy.wait_for_message('mavros/battery', BatteryState, timeout=3)
|
|
||||||
if not battery.cell_voltage:
|
|
||||||
failure('cell voltage is not available, https://clover.coex.tech/power')
|
|
||||||
else:
|
|
||||||
cell = battery.cell_voltage[0]
|
|
||||||
if cell > 4.3 or cell < 3.0:
|
|
||||||
failure('incorrect cell voltage: %.2f V, https://clover.coex.tech/power', cell)
|
|
||||||
elif cell < 3.7:
|
|
||||||
failure('critically low cell voltage: %.2f V, recharge battery', cell)
|
|
||||||
except rospy.ROSException:
|
|
||||||
failure('no battery state')
|
|
||||||
|
|
||||||
# time sync check
|
|
||||||
try:
|
try:
|
||||||
info('time sync offset: %.2f s', float(read_diagnostics('mavros: Time Sync', 'Estimated time offset (s)')))
|
battery = rospy.wait_for_message('mavros/battery', BatteryState, timeout=3)
|
||||||
except:
|
if not battery.cell_voltage:
|
||||||
failure('cannot read time sync offset')
|
failure('cell voltage is not available, https://clover.coex.tech/power')
|
||||||
|
else:
|
||||||
|
cell = battery.cell_voltage[0]
|
||||||
|
if cell > 4.3 or cell < 3.0:
|
||||||
|
failure('incorrect cell voltage: %.2f V, https://clover.coex.tech/power', cell)
|
||||||
|
elif cell < 3.7:
|
||||||
|
failure('critically low cell voltage: %.2f V, recharge battery', cell)
|
||||||
|
except rospy.ROSException:
|
||||||
|
failure('no battery state')
|
||||||
|
|
||||||
except rospy.ROSException:
|
except rospy.ROSException:
|
||||||
failure('no MAVROS state (check wiring)')
|
failure('no MAVROS state (check wiring)')
|
||||||
info('fcu_url = %s', rospy.get_param('mavros/fcu_url', '?'))
|
|
||||||
|
|
||||||
|
|
||||||
def describe_direction(v):
|
def describe_direction(v):
|
||||||
@@ -384,7 +328,7 @@ def is_process_running(binary, exact=False, full=False):
|
|||||||
if exact:
|
if exact:
|
||||||
args.append('-x') # match exactly with the command name
|
args.append('-x') # match exactly with the command name
|
||||||
if full:
|
if full:
|
||||||
args.append('-f') # use full command line (including arguments) to match
|
args.append('-f') # use full process name to match
|
||||||
args.append(binary)
|
args.append(binary)
|
||||||
subprocess.check_output(args)
|
subprocess.check_output(args)
|
||||||
return True
|
return True
|
||||||
@@ -394,24 +338,19 @@ def is_process_running(binary, exact=False, full=False):
|
|||||||
|
|
||||||
@check('ArUco markers')
|
@check('ArUco markers')
|
||||||
def check_aruco():
|
def check_aruco():
|
||||||
markers = None
|
|
||||||
|
|
||||||
if is_process_running('aruco_detect', full=True):
|
if is_process_running('aruco_detect', full=True):
|
||||||
try:
|
try:
|
||||||
info('aruco_detect/length = %g m', rospy.get_param('aruco_detect/length', '?'))
|
info('aruco_detect/length = %g m', rospy.get_param('aruco_detect/length'))
|
||||||
except KeyError:
|
except KeyError:
|
||||||
failure('aruco_detect/length parameter is not set')
|
failure('aruco_detect/length parameter is not set')
|
||||||
known_vertical = rospy.get_param('aruco_detect/known_vertical', '')
|
known_tilt = rospy.get_param('aruco_detect/known_tilt', '')
|
||||||
flip_vertical = rospy.get_param('aruco_detect/flip_vertical', False)
|
if known_tilt == 'map':
|
||||||
description = ''
|
known_tilt += ' (ALL markers are on the floor)'
|
||||||
if known_vertical == 'map' and not flip_vertical:
|
elif known_tilt == 'map_flipped':
|
||||||
description = ' (all markers are on the floor)'
|
known_tilt += ' (ALL markers are on the ceiling)'
|
||||||
elif known_vertical == 'map' and flip_vertical:
|
info('aruco_detector/known_tilt = %s', known_tilt)
|
||||||
description = ' (all markers are on the ceiling)'
|
|
||||||
info('aruco_detect/known_vertical = %s', known_vertical)
|
|
||||||
info('aruco_detect/flip_vertical = %s%s', flip_vertical, description)
|
|
||||||
try:
|
try:
|
||||||
markers = rospy.wait_for_message('aruco_detect/markers', MarkerArray, timeout=0.8)
|
rospy.wait_for_message('aruco_detect/markers', MarkerArray, timeout=1)
|
||||||
except rospy.ROSException:
|
except rospy.ROSException:
|
||||||
failure('no markers detection')
|
failure('no markers detection')
|
||||||
return
|
return
|
||||||
@@ -420,61 +359,42 @@ def check_aruco():
|
|||||||
return
|
return
|
||||||
|
|
||||||
if is_process_running('aruco_map', full=True):
|
if is_process_running('aruco_map', full=True):
|
||||||
known_vertical = rospy.get_param('aruco_map/known_vertical', '')
|
known_tilt = rospy.get_param('aruco_map/known_tilt', '')
|
||||||
flip_vertical = rospy.get_param('aruco_map/flip_vertical', False)
|
if known_tilt == 'map':
|
||||||
description = ''
|
known_tilt += ' (marker\'s map is on the floor)'
|
||||||
if known_vertical == 'map' and not flip_vertical:
|
elif known_tilt == 'map_flipped':
|
||||||
description += ' (markers map is on the floor)'
|
known_tilt += ' (marker\'s map is on the ceiling)'
|
||||||
elif known_vertical == 'map' and flip_vertical:
|
info('aruco_map/known_tilt = %s', known_tilt)
|
||||||
description += ' (markers map is on the ceiling)'
|
|
||||||
info('aruco_map/known_vertical = %s', known_vertical)
|
|
||||||
info('aruco_map/flip_vertical = %s%s', flip_vertical, description)
|
|
||||||
|
|
||||||
try:
|
try:
|
||||||
visualization = rospy.wait_for_message('aruco_map/visualization', VisualizationMarkerArray, timeout=0.8)
|
visualization = rospy.wait_for_message('aruco_map/visualization', VisualizationMarkerArray, timeout=1)
|
||||||
info('map has %s markers', len(visualization.markers))
|
info('map has %s markers', len(visualization.markers))
|
||||||
except:
|
except:
|
||||||
failure('cannot read aruco_map/visualization topic')
|
failure('cannot read aruco_map/visualization topic')
|
||||||
|
|
||||||
try:
|
try:
|
||||||
rospy.wait_for_message('aruco_map/pose', PoseWithCovarianceStamped, timeout=0.8)
|
rospy.wait_for_message('aruco_map/pose', PoseWithCovarianceStamped, timeout=1)
|
||||||
except rospy.ROSException:
|
except rospy.ROSException:
|
||||||
if not markers:
|
failure('no map detection')
|
||||||
info('no map detection as no markers detection')
|
|
||||||
elif not markers.markers:
|
|
||||||
info('no map detection as no markers detected')
|
|
||||||
else:
|
|
||||||
failure('no map detection')
|
|
||||||
else:
|
else:
|
||||||
info('aruco_map is not running')
|
info('aruco_map is not running')
|
||||||
|
|
||||||
|
|
||||||
def is_on_the_floor():
|
|
||||||
try:
|
|
||||||
dist = rospy.wait_for_message('rangefinder/range', Range, timeout=1)
|
|
||||||
return dist.range < 0.3
|
|
||||||
except rospy.ROSException:
|
|
||||||
return False
|
|
||||||
|
|
||||||
|
|
||||||
@check('Vision position estimate')
|
@check('Vision position estimate')
|
||||||
def check_vpe():
|
def check_vpe():
|
||||||
vis = None
|
vis = None
|
||||||
try:
|
try:
|
||||||
vis = rospy.wait_for_message('mavros/vision_pose/pose', PoseStamped, timeout=0.8)
|
vis = rospy.wait_for_message('mavros/vision_pose/pose', PoseStamped, timeout=1)
|
||||||
except rospy.ROSException:
|
except rospy.ROSException:
|
||||||
try:
|
try:
|
||||||
vis = rospy.wait_for_message('mavros/mocap/pose', PoseStamped, timeout=0.8)
|
vis = rospy.wait_for_message('mavros/mocap/pose', PoseStamped, timeout=1)
|
||||||
except rospy.ROSException:
|
except rospy.ROSException:
|
||||||
if not is_process_running('vpe_publisher', full=True):
|
failure('no VPE or MoCap messages')
|
||||||
info('no vision position estimate, vpe_publisher is not running')
|
# check if vpe_publisher is running
|
||||||
elif rospy.get_param('aruco_map/known_vertical', '') == 'map' \
|
try:
|
||||||
and rospy.get_param('aruco_map/flip_vertical', False):
|
subprocess.check_output(['pgrep', '-x', 'vpe_publisher'])
|
||||||
failure('no vision position estimate, markers are on the ceiling')
|
except subprocess.CalledProcessError:
|
||||||
elif is_on_the_floor():
|
return # it's not running, skip following checks
|
||||||
info('no vision position estimate, the drone is on the floor')
|
|
||||||
else:
|
|
||||||
failure('no vision position estimate')
|
|
||||||
|
|
||||||
# check PX4 settings
|
# check PX4 settings
|
||||||
est = get_param('SYS_MC_EST_GROUP')
|
est = get_param('SYS_MC_EST_GROUP')
|
||||||
@@ -486,14 +406,14 @@ def check_vpe():
|
|||||||
if vision_yaw_w == 0:
|
if vision_yaw_w == 0:
|
||||||
failure('vision yaw weight is zero, change ATT_W_EXT_HDG parameter')
|
failure('vision yaw weight is zero, change ATT_W_EXT_HDG parameter')
|
||||||
else:
|
else:
|
||||||
info('vision yaw weight: %s', ff(vision_yaw_w))
|
info('Vision yaw weight: %.2f', vision_yaw_w)
|
||||||
fuse = get_param('LPE_FUSION')
|
fuse = get_param('LPE_FUSION')
|
||||||
if not fuse & (1 << 2):
|
if not fuse & (1 << 2):
|
||||||
failure('vision position fusion is disabled, change LPE_FUSION parameter')
|
failure('vision position fusion is disabled, change LPE_FUSION parameter')
|
||||||
delay = get_param('LPE_VIS_DELAY')
|
delay = get_param('LPE_VIS_DELAY')
|
||||||
if delay != 0:
|
if delay != 0:
|
||||||
failure('LPE_VIS_DELAY = %s, but it should be zero', delay)
|
failure('LPE_VIS_DELAY parameter is %s, but it should be zero', delay)
|
||||||
info('LPE_VIS_XY = %s m, LPE_VIS_Z = %s m', get_paramf('LPE_VIS_XY'), get_paramf('LPE_VIS_Z'))
|
info('LPE_VIS_XY is %.2f m, LPE_VIS_Z is %.2f m', get_param('LPE_VIS_XY'), get_param('LPE_VIS_Z'))
|
||||||
elif est == 2:
|
elif est == 2:
|
||||||
fuse = get_param('EKF2_AID_MASK')
|
fuse = get_param('EKF2_AID_MASK')
|
||||||
if not fuse & (1 << 3):
|
if not fuse & (1 << 3):
|
||||||
@@ -502,10 +422,10 @@ def check_vpe():
|
|||||||
failure('vision yaw fusion is disabled, change EKF2_AID_MASK parameter')
|
failure('vision yaw fusion is disabled, change EKF2_AID_MASK parameter')
|
||||||
delay = get_param('EKF2_EV_DELAY')
|
delay = get_param('EKF2_EV_DELAY')
|
||||||
if delay != 0:
|
if delay != 0:
|
||||||
failure('EKF2_EV_DELAY = %.2f, but it should be zero', delay)
|
failure('EKF2_EV_DELAY is %.2f, but it should be zero', delay)
|
||||||
info('EKF2_EVA_NOISE = %s, EKF2_EVP_NOISE = %s',
|
info('EKF2_EVA_NOISE is %.3f, EKF2_EVP_NOISE is %.3f',
|
||||||
get_paramf('EKF2_EVA_NOISE', 3),
|
get_param('EKF2_EVA_NOISE'),
|
||||||
get_paramf('EKF2_EVP_NOISE', 3))
|
get_param('EKF2_EVP_NOISE'))
|
||||||
|
|
||||||
if not vis:
|
if not vis:
|
||||||
return
|
return
|
||||||
@@ -563,12 +483,6 @@ def check_local_position():
|
|||||||
failure('roll is %.2f deg; place copter horizontally or redo level horizon calib',
|
failure('roll is %.2f deg; place copter horizontally or redo level horizon calib',
|
||||||
math.degrees(roll))
|
math.degrees(roll))
|
||||||
|
|
||||||
if not tf_buffer.can_transform('base_link', pose.header.frame_id, rospy.get_rostime(), rospy.Duration(0.5)):
|
|
||||||
failure('can\'t transform from %s to base_link (timeout 0.5 s): is TF enabled?', pose.header.frame_id)
|
|
||||||
|
|
||||||
if not tf_buffer.can_transform('body', pose.header.frame_id, rospy.get_rostime(), rospy.Duration(0.5)):
|
|
||||||
failure('can\'t transform from %s to body (timeout 0.5 s)', pose.header.frame_id)
|
|
||||||
|
|
||||||
except rospy.ROSException:
|
except rospy.ROSException:
|
||||||
failure('no local position')
|
failure('no local position')
|
||||||
|
|
||||||
@@ -603,19 +517,13 @@ def check_velocity():
|
|||||||
@check('Global position (GPS)')
|
@check('Global position (GPS)')
|
||||||
def check_global_position():
|
def check_global_position():
|
||||||
try:
|
try:
|
||||||
rospy.wait_for_message('mavros/global_position/global', NavSatFix, timeout=0.8)
|
rospy.wait_for_message('mavros/global_position/global', NavSatFix, timeout=1)
|
||||||
except rospy.ROSException:
|
except rospy.ROSException:
|
||||||
info('no global position')
|
info('no global position')
|
||||||
if get_param('SYS_MC_EST_GROUP') == 2 and (get_param('EKF2_AID_MASK', 0) & (1 << 0)):
|
|
||||||
failure('enabled GPS fusion may suppress vision position aiding')
|
|
||||||
|
|
||||||
|
|
||||||
@check('Optical flow')
|
@check('Optical flow')
|
||||||
def check_optical_flow():
|
def check_optical_flow():
|
||||||
if not is_process_running('optical_flow', full=True):
|
|
||||||
info('optical_flow is not running')
|
|
||||||
return
|
|
||||||
|
|
||||||
# TODO:check FPS!
|
# TODO:check FPS!
|
||||||
try:
|
try:
|
||||||
rospy.wait_for_message('mavros/px4flow/raw/send', OpticalFlowRad, timeout=0.5)
|
rospy.wait_for_message('mavros/px4flow/raw/send', OpticalFlowRad, timeout=0.5)
|
||||||
@@ -623,7 +531,7 @@ def check_optical_flow():
|
|||||||
# check PX4 settings
|
# check PX4 settings
|
||||||
rot = get_param('SENS_FLOW_ROT')
|
rot = get_param('SENS_FLOW_ROT')
|
||||||
if rot != 0:
|
if rot != 0:
|
||||||
failure('SENS_FLOW_ROT = %s, but it should be zero', rot)
|
failure('SENS_FLOW_ROT parameter is %s, but it should be zero', rot)
|
||||||
est = get_param('SYS_MC_EST_GROUP')
|
est = get_param('SYS_MC_EST_GROUP')
|
||||||
if est == 1:
|
if est == 1:
|
||||||
fuse = get_param('LPE_FUSION')
|
fuse = get_param('LPE_FUSION')
|
||||||
@@ -631,36 +539,32 @@ def check_optical_flow():
|
|||||||
failure('optical flow fusion is disabled, change LPE_FUSION parameter')
|
failure('optical flow fusion is disabled, change LPE_FUSION parameter')
|
||||||
if not fuse & (1 << 1):
|
if not fuse & (1 << 1):
|
||||||
failure('flow gyro compensation is disabled, change LPE_FUSION parameter')
|
failure('flow gyro compensation is disabled, change LPE_FUSION parameter')
|
||||||
scale = get_param('LPE_FLW_SCALE', 1)
|
scale = get_param('LPE_FLW_SCALE')
|
||||||
if not numpy.isclose(scale, 1.0):
|
if not numpy.isclose(scale, 1.0):
|
||||||
failure('LPE_FLW_SCALE = %.2f, but it should be 1.0', scale)
|
failure('LPE_FLW_SCALE parameter is %.2f, but it should be 1.0', scale)
|
||||||
|
|
||||||
info('LPE_FLW_QMIN = %s, LPE_FLW_R = %s, LPE_FLW_RR = %s',
|
info('LPE_FLW_QMIN is %s, LPE_FLW_R is %.4f, LPE_FLW_RR is %.4f, SENS_FLOW_MINHGT is %.3f, SENS_FLOW_MAXHGT is %.3f',
|
||||||
get_paramf('LPE_FLW_QMIN'),
|
get_param('LPE_FLW_QMIN'),
|
||||||
get_paramf('LPE_FLW_R', 4),
|
get_param('LPE_FLW_R'),
|
||||||
get_paramf('LPE_FLW_RR', 4))
|
get_param('LPE_FLW_RR'),
|
||||||
|
get_param('SENS_FLOW_MINHGT'),
|
||||||
|
get_param('SENS_FLOW_MAXHGT'))
|
||||||
elif est == 2:
|
elif est == 2:
|
||||||
fuse = get_param('EKF2_AID_MASK', 0)
|
fuse = get_param('EKF2_AID_MASK')
|
||||||
if not fuse & (1 << 1):
|
if not fuse & (1 << 1):
|
||||||
failure('optical flow fusion is disabled, change EKF2_AID_MASK parameter')
|
failure('optical flow fusion is disabled, change EKF2_AID_MASK parameter')
|
||||||
delay = get_param('EKF2_OF_DELAY', 0)
|
delay = get_param('EKF2_OF_DELAY')
|
||||||
if delay != 0:
|
if delay != 0:
|
||||||
failure('EKF2_OF_DELAY = %.2f, but it should be zero', delay)
|
failure('EKF2_OF_DELAY is %.2f, but it should be zero', delay)
|
||||||
info('EKF2_OF_QMIN = %s, EKF2_OF_N_MIN = %s, EKF2_OF_N_MAX = %s',
|
info('EKF2_OF_QMIN is %s, EKF2_OF_N_MIN is %.4f, EKF2_OF_N_MAX is %.4f, SENS_FLOW_MINHGT is %.3f, SENS_FLOW_MAXHGT is %.3f',
|
||||||
get_paramf('EKF2_OF_QMIN'),
|
get_param('EKF2_OF_QMIN'),
|
||||||
get_paramf('EKF2_OF_N_MIN', 4),
|
get_param('EKF2_OF_N_MIN'),
|
||||||
get_paramf('EKF2_OF_N_MAX', 4))
|
get_param('EKF2_OF_N_MAX'),
|
||||||
info('SENS_FLOW_MINHGT = %s, SENS_FLOW_MAXHGT = %s', get_paramf('SENS_FLOW_MINHGT', 3), get_paramf('SENS_FLOW_MAXHGT', 3))
|
get_param('SENS_FLOW_MINHGT'),
|
||||||
|
get_param('SENS_FLOW_MAXHGT'))
|
||||||
|
|
||||||
except rospy.ROSException:
|
except rospy.ROSException:
|
||||||
if rospy.get_param('optical_flow/disable_on_vpe', False):
|
failure('no optical flow data (from Raspberry)')
|
||||||
try:
|
|
||||||
rospy.wait_for_message('mavros/vision_pose/pose', PoseStamped, timeout=1)
|
|
||||||
info('no optical flow as disable_on_vpe is true')
|
|
||||||
except:
|
|
||||||
failure('no optical flow on RPi, disable_on_vpe is true, but no vision pose also')
|
|
||||||
else:
|
|
||||||
failure('no optical flow on RPi')
|
|
||||||
|
|
||||||
|
|
||||||
@check('Rangefinder')
|
@check('Rangefinder')
|
||||||
@@ -684,7 +588,7 @@ def check_rangefinder():
|
|||||||
|
|
||||||
est = get_param('SYS_MC_EST_GROUP')
|
est = get_param('SYS_MC_EST_GROUP')
|
||||||
if est == 1:
|
if est == 1:
|
||||||
fuse = get_param('LPE_FUSION', 0)
|
fuse = get_param('LPE_FUSION')
|
||||||
if not fuse & (1 << 5):
|
if not fuse & (1 << 5):
|
||||||
info('"pub agl as lpos down" in LPE_FUSION is disabled, NOT operating over flat surface')
|
info('"pub agl as lpos down" in LPE_FUSION is disabled, NOT operating over flat surface')
|
||||||
else:
|
else:
|
||||||
@@ -705,41 +609,34 @@ def check_rangefinder():
|
|||||||
|
|
||||||
@check('Boot duration')
|
@check('Boot duration')
|
||||||
def check_boot_duration():
|
def check_boot_duration():
|
||||||
if not os.path.exists('/etc/clover_version'):
|
output = subprocess.check_output('systemd-analyze')
|
||||||
info('skip check')
|
|
||||||
return # Don't check not on Clover's image
|
|
||||||
|
|
||||||
output = subprocess.check_output('systemd-analyze').decode()
|
|
||||||
r = re.compile(r'([\d\.]+)s\s*$', flags=re.MULTILINE)
|
r = re.compile(r'([\d\.]+)s\s*$', flags=re.MULTILINE)
|
||||||
duration = float(r.search(output).groups()[0])
|
duration = float(r.search(output).groups()[0])
|
||||||
if duration > 20:
|
if duration > 15:
|
||||||
failure('long Raspbian boot duration: %ss (systemd-analyze for analyzing)', duration)
|
failure('long Raspbian boot duration: %ss (systemd-analyze for analyzing)', duration)
|
||||||
|
|
||||||
|
|
||||||
@check('CPU usage')
|
@check('CPU usage')
|
||||||
def check_cpu_usage():
|
def check_cpu_usage():
|
||||||
WHITELIST = 'nodelet', 'gzclient', 'gzserver', 'selfcheck.py'
|
WHITELIST = 'nodelet',
|
||||||
CMD = "top -n 1 -b -i | tail -n +8 | awk '{ printf(\"%-8s\\t%-8s\\t%-8s\\n\", $1, $9, $12); }'"
|
CMD = "top -n 1 -b -i | tail -n +8 | awk '{ printf(\"%-8s\\t%-8s\\t%-8s\\n\", $1, $9, $12); }'"
|
||||||
output = subprocess.check_output(CMD, shell=True).decode()
|
output = subprocess.check_output(CMD, shell=True)
|
||||||
processes = output.split('\n')
|
processes = output.split('\n')
|
||||||
for process in processes:
|
for process in processes:
|
||||||
if not process:
|
if not process:
|
||||||
continue
|
continue
|
||||||
pid, cpu, cmd = process.split('\t')
|
pid, cpu, cmd = process.split('\t')
|
||||||
|
|
||||||
if cmd.strip() not in WHITELIST and locale.atof(cpu) > 30:
|
if cmd.strip() not in WHITELIST and float(cpu) > 30:
|
||||||
failure('high CPU usage (%s%%) detected: %s (PID %s)',
|
failure('high CPU usage (%s%%) detected: %s (PID %s)',
|
||||||
cpu.strip(), cmd.strip(), pid.strip())
|
cpu.strip(), cmd.strip(), pid.strip())
|
||||||
|
|
||||||
|
|
||||||
@check('clover.service')
|
@check('clover.service')
|
||||||
def check_clover_service():
|
def check_clover_service():
|
||||||
if not os.path.exists('/etc/clover_version'):
|
|
||||||
return # Don't check not on Clover's image
|
|
||||||
|
|
||||||
try:
|
try:
|
||||||
output = subprocess.check_output('systemctl show -p ActiveState --value clover.service'.split(),
|
output = subprocess.check_output('systemctl show -p ActiveState --value clover.service'.split(),
|
||||||
stderr=subprocess.STDOUT).decode()
|
stderr=subprocess.STDOUT)
|
||||||
except subprocess.CalledProcessError as e:
|
except subprocess.CalledProcessError as e:
|
||||||
failure('systemctl returned %s: %s', e.returncode, e.output)
|
failure('systemctl returned %s: %s', e.returncode, e.output)
|
||||||
return
|
return
|
||||||
@@ -749,22 +646,13 @@ def check_clover_service():
|
|||||||
elif 'failed' in output:
|
elif 'failed' in output:
|
||||||
failure('service failed to run, check your launch-files')
|
failure('service failed to run, check your launch-files')
|
||||||
|
|
||||||
BLACKLIST = 'Unexpected command 520', 'Time jump detected', 'different index:'
|
r = re.compile(r'^(.*)\[(FATAL|ERROR)\] \[\d+.\d+\]: (.*?)(\x1b(.*))?$')
|
||||||
|
|
||||||
r = re.compile(r'^(.*)\[(FATAL|ERROR| WARN)\] \[\d+.\d+\]: (.*?)(\x1b(.*))?$')
|
|
||||||
error_count = OrderedDict()
|
error_count = OrderedDict()
|
||||||
try:
|
try:
|
||||||
for line in open('/tmp/clover.err', 'r'):
|
for line in open('/tmp/clover.err', 'r'):
|
||||||
skip = False
|
|
||||||
for substr in BLACKLIST:
|
|
||||||
if substr in line:
|
|
||||||
skip = True
|
|
||||||
if skip:
|
|
||||||
continue
|
|
||||||
|
|
||||||
node_error = r.search(line)
|
node_error = r.search(line)
|
||||||
if node_error:
|
if node_error:
|
||||||
msg = node_error.groups()[1].strip() + ': ' + node_error.groups()[2]
|
msg = node_error.groups()[1] + ': ' + node_error.groups()[2]
|
||||||
if msg in error_count:
|
if msg in error_count:
|
||||||
error_count[msg] += 1
|
error_count[msg] += 1
|
||||||
else:
|
else:
|
||||||
@@ -787,18 +675,11 @@ def check_image():
|
|||||||
try:
|
try:
|
||||||
info('version: %s', open('/etc/clover_version').read().strip())
|
info('version: %s', open('/etc/clover_version').read().strip())
|
||||||
except IOError:
|
except IOError:
|
||||||
try:
|
info('no /etc/clover_version file, not the Clover image?')
|
||||||
info('VM version: %s', open('/etc/clover_vm_version').read().strip())
|
|
||||||
except IOError:
|
|
||||||
info('no /etc/clover_version file, not the Clover image?')
|
|
||||||
|
|
||||||
|
|
||||||
@check('Preflight status')
|
@check('Preflight status')
|
||||||
def check_preflight_status():
|
def check_preflight_status():
|
||||||
if is_process_running('px4', exact=True):
|
|
||||||
info('can\'t check in SITL')
|
|
||||||
return
|
|
||||||
|
|
||||||
# Make sure the console is available to us
|
# Make sure the console is available to us
|
||||||
mavlink_exec('\n')
|
mavlink_exec('\n')
|
||||||
cmdr_output = mavlink_exec('commander check')
|
cmdr_output = mavlink_exec('commander check')
|
||||||
@@ -820,10 +701,6 @@ def check_preflight_status():
|
|||||||
|
|
||||||
@check('Network')
|
@check('Network')
|
||||||
def check_network():
|
def check_network():
|
||||||
if not os.path.exists('/etc/clover_version'):
|
|
||||||
# TODO:
|
|
||||||
return # Don't check not on Clover's image
|
|
||||||
|
|
||||||
ros_hostname = os.environ.get('ROS_HOSTNAME', '').strip()
|
ros_hostname = os.environ.get('ROS_HOSTNAME', '').strip()
|
||||||
|
|
||||||
if not ros_hostname:
|
if not ros_hostname:
|
||||||
@@ -846,14 +723,6 @@ def check_network():
|
|||||||
|
|
||||||
@check('RPi health')
|
@check('RPi health')
|
||||||
def check_rpi_health():
|
def check_rpi_health():
|
||||||
try:
|
|
||||||
import shutil
|
|
||||||
total, used, free = shutil.disk_usage('/')
|
|
||||||
if free < 1024 * 1024 * 1024:
|
|
||||||
failure('disk space is less than 1 GB; consider removing logs (~/.ros/log/)')
|
|
||||||
except Exception as e:
|
|
||||||
info('could not check the disk free space: %s', str(e))
|
|
||||||
|
|
||||||
# `vcgencmd get_throttled` output codes taken from
|
# `vcgencmd get_throttled` output codes taken from
|
||||||
# https://github.com/raspberrypi/documentation/blob/JamesH65-patch-vcgencmd-vcdbg-docs/raspbian/applications/vcgencmd.md#get_throttled
|
# https://github.com/raspberrypi/documentation/blob/JamesH65-patch-vcgencmd-vcdbg-docs/raspbian/applications/vcgencmd.md#get_throttled
|
||||||
# TODO: support more base platforms?
|
# TODO: support more base platforms?
|
||||||
@@ -882,9 +751,9 @@ def check_rpi_health():
|
|||||||
# <parameter>=<value>
|
# <parameter>=<value>
|
||||||
# In case of `get_throttled`, <value> is a hexadecimal number
|
# In case of `get_throttled`, <value> is a hexadecimal number
|
||||||
# with some of the FLAGs OR'ed together
|
# with some of the FLAGs OR'ed together
|
||||||
output = subprocess.check_output(['vcgencmd', 'get_throttled']).decode()
|
output = subprocess.check_output(['vcgencmd', 'get_throttled'])
|
||||||
except OSError:
|
except OSError:
|
||||||
info('could not call vcgencmd binary; not a Raspberry Pi?')
|
failure('could not call vcgencmd binary; not a Raspberry Pi?')
|
||||||
return
|
return
|
||||||
|
|
||||||
throttle_mask = int(output.split('=')[1], base=16)
|
throttle_mask = int(output.split('=')[1], base=16)
|
||||||
@@ -901,47 +770,26 @@ def check_board():
|
|||||||
info('could not open /proc/device-tree/model, not a Raspberry Pi?')
|
info('could not open /proc/device-tree/model, not a Raspberry Pi?')
|
||||||
|
|
||||||
|
|
||||||
def parallel_for(fns):
|
|
||||||
threads = []
|
|
||||||
for fn in fns:
|
|
||||||
thread = Thread(target=fn)
|
|
||||||
thread.start()
|
|
||||||
threads.append(thread)
|
|
||||||
for thread in threads:
|
|
||||||
thread.join()
|
|
||||||
|
|
||||||
|
|
||||||
def consequentially_for(fns):
|
|
||||||
for fn in fns:
|
|
||||||
fn()
|
|
||||||
|
|
||||||
|
|
||||||
def selfcheck():
|
def selfcheck():
|
||||||
checks = [
|
check_image()
|
||||||
check_image,
|
check_board()
|
||||||
check_board,
|
check_clover_service()
|
||||||
check_clover_service,
|
check_network()
|
||||||
check_network,
|
check_fcu()
|
||||||
check_fcu,
|
check_imu()
|
||||||
check_imu,
|
check_local_position()
|
||||||
check_local_position,
|
check_velocity()
|
||||||
check_velocity,
|
check_global_position()
|
||||||
check_global_position,
|
check_preflight_status()
|
||||||
check_preflight_status,
|
check_main_camera()
|
||||||
check_main_camera,
|
check_aruco()
|
||||||
check_aruco,
|
check_simpleoffboard()
|
||||||
check_simpleoffboard,
|
check_optical_flow()
|
||||||
check_optical_flow,
|
check_vpe()
|
||||||
check_vpe,
|
check_rangefinder()
|
||||||
check_rangefinder,
|
check_rpi_health()
|
||||||
check_rpi_health,
|
check_cpu_usage()
|
||||||
check_cpu_usage,
|
check_boot_duration()
|
||||||
check_boot_duration,
|
|
||||||
]
|
|
||||||
if rospy.get_param('~parallel', False):
|
|
||||||
parallel_for(checks)
|
|
||||||
else:
|
|
||||||
consequentially_for(checks)
|
|
||||||
|
|
||||||
|
|
||||||
if __name__ == '__main__':
|
if __name__ == '__main__':
|
||||||
|
|||||||
@@ -23,7 +23,6 @@
|
|||||||
#include <tf2_ros/static_transform_broadcaster.h>
|
#include <tf2_ros/static_transform_broadcaster.h>
|
||||||
#include <tf2_geometry_msgs/tf2_geometry_msgs.h>
|
#include <tf2_geometry_msgs/tf2_geometry_msgs.h>
|
||||||
#include <std_srvs/Trigger.h>
|
#include <std_srvs/Trigger.h>
|
||||||
#include <geometry_msgs/PointStamped.h>
|
|
||||||
#include <geometry_msgs/PoseStamped.h>
|
#include <geometry_msgs/PoseStamped.h>
|
||||||
#include <geometry_msgs/TwistStamped.h>
|
#include <geometry_msgs/TwistStamped.h>
|
||||||
#include <geometry_msgs/Vector3Stamped.h>
|
#include <geometry_msgs/Vector3Stamped.h>
|
||||||
@@ -38,19 +37,14 @@
|
|||||||
#include <mavros_msgs/State.h>
|
#include <mavros_msgs/State.h>
|
||||||
#include <mavros_msgs/StatusText.h>
|
#include <mavros_msgs/StatusText.h>
|
||||||
#include <mavros_msgs/ManualControl.h>
|
#include <mavros_msgs/ManualControl.h>
|
||||||
#include <mavros_msgs/Altitude.h>
|
|
||||||
|
|
||||||
#include <clover/GetTelemetry.h>
|
#include <clover/GetTelemetry.h>
|
||||||
#include <clover/Navigate.h>
|
#include <clover/Navigate.h>
|
||||||
#include <clover/NavigateGlobal.h>
|
#include <clover/NavigateGlobal.h>
|
||||||
#include <clover/SetAltitude.h>
|
|
||||||
#include <clover/SetYaw.h>
|
|
||||||
#include <clover/SetYawRate.h>
|
|
||||||
#include <clover/SetPosition.h>
|
#include <clover/SetPosition.h>
|
||||||
#include <clover/SetVelocity.h>
|
#include <clover/SetVelocity.h>
|
||||||
#include <clover/SetAttitude.h>
|
#include <clover/SetAttitude.h>
|
||||||
#include <clover/SetRates.h>
|
#include <clover/SetRates.h>
|
||||||
#include <clover/State.h>
|
|
||||||
|
|
||||||
using std::string;
|
using std::string;
|
||||||
using std::isnan;
|
using std::isnan;
|
||||||
@@ -60,7 +54,6 @@ using namespace clover;
|
|||||||
using mavros_msgs::PositionTarget;
|
using mavros_msgs::PositionTarget;
|
||||||
using mavros_msgs::AttitudeTarget;
|
using mavros_msgs::AttitudeTarget;
|
||||||
using mavros_msgs::Thrust;
|
using mavros_msgs::Thrust;
|
||||||
using mavros_msgs::Altitude;
|
|
||||||
|
|
||||||
// tf2
|
// tf2
|
||||||
tf2_ros::Buffer tf_buffer;
|
tf2_ros::Buffer tf_buffer;
|
||||||
@@ -68,7 +61,6 @@ std::shared_ptr<tf2_ros::TransformBroadcaster> transform_broadcaster;
|
|||||||
std::shared_ptr<tf2_ros::StaticTransformBroadcaster> static_transform_broadcaster;
|
std::shared_ptr<tf2_ros::StaticTransformBroadcaster> static_transform_broadcaster;
|
||||||
|
|
||||||
// Parameters
|
// Parameters
|
||||||
string mavros;
|
|
||||||
string local_frame;
|
string local_frame;
|
||||||
string fcu_frame;
|
string fcu_frame;
|
||||||
ros::Duration transform_timeout;
|
ros::Duration transform_timeout;
|
||||||
@@ -88,40 +80,33 @@ bool land_only_in_offboard, nav_from_sp, check_kill_switch;
|
|||||||
std::map<string, string> reference_frames;
|
std::map<string, string> reference_frames;
|
||||||
|
|
||||||
// Publishers
|
// Publishers
|
||||||
ros::Publisher attitude_pub, attitude_raw_pub, position_pub, position_raw_pub, rates_pub, thrust_pub, state_pub;
|
ros::Publisher attitude_pub, attitude_raw_pub, position_pub, position_raw_pub, rates_pub, thrust_pub;
|
||||||
|
|
||||||
// Service clients
|
// Service clients
|
||||||
ros::ServiceClient arming, set_mode;
|
ros::ServiceClient arming, set_mode;
|
||||||
|
|
||||||
// Containers
|
// Containers
|
||||||
ros::Timer setpoint_timer;
|
ros::Timer setpoint_timer;
|
||||||
|
tf::Quaternion tq;
|
||||||
PoseStamped position_msg;
|
PoseStamped position_msg;
|
||||||
PositionTarget position_raw_msg;
|
PositionTarget position_raw_msg;
|
||||||
//TwistStamped rates_msg;
|
AttitudeTarget att_raw_msg;
|
||||||
|
Thrust thrust_msg;
|
||||||
|
TwistStamped rates_msg;
|
||||||
TransformStamped target, setpoint;
|
TransformStamped target, setpoint;
|
||||||
geometry_msgs::TransformStamped body;
|
geometry_msgs::TransformStamped body;
|
||||||
geometry_msgs::TransformStamped terrain;
|
|
||||||
|
|
||||||
// State
|
// State
|
||||||
PoseStamped nav_start;
|
PoseStamped nav_start;
|
||||||
PointStamped setpoint_position;
|
PoseStamped setpoint_position, setpoint_position_transformed;
|
||||||
PointStamped setpoint_altitude;
|
Vector3Stamped setpoint_velocity, setpoint_velocity_transformed;
|
||||||
Vector3Stamped setpoint_velocity;
|
QuaternionStamped setpoint_attitude, setpoint_attitude_transformed;
|
||||||
float setpoint_yaw, setpoint_roll, setpoint_pitch;
|
float setpoint_yaw_rate;
|
||||||
Vector3 setpoint_rates;
|
|
||||||
string yaw_frame_id;
|
|
||||||
float setpoint_thrust;
|
|
||||||
float nav_speed;
|
float nav_speed;
|
||||||
float setpoint_lat = NAN, setpoint_lon = NAN;
|
|
||||||
bool busy = false;
|
bool busy = false;
|
||||||
bool wait_armed = false;
|
bool wait_armed = false;
|
||||||
bool nav_from_sp_flag = false;
|
bool nav_from_sp_flag = false;
|
||||||
|
|
||||||
// Last published
|
|
||||||
PoseStamped setpoint_pose_local;
|
|
||||||
Vector3Stamped setpoint_velocity_local;
|
|
||||||
float yaw_local;
|
|
||||||
|
|
||||||
enum setpoint_type_t {
|
enum setpoint_type_t {
|
||||||
NONE,
|
NONE,
|
||||||
NAVIGATE,
|
NAVIGATE,
|
||||||
@@ -129,10 +114,7 @@ enum setpoint_type_t {
|
|||||||
POSITION,
|
POSITION,
|
||||||
VELOCITY,
|
VELOCITY,
|
||||||
ATTITUDE,
|
ATTITUDE,
|
||||||
RATES,
|
RATES
|
||||||
_ALTITUDE,
|
|
||||||
_YAW,
|
|
||||||
_YAW_RATE,
|
|
||||||
};
|
};
|
||||||
|
|
||||||
enum setpoint_type_t setpoint_type = NONE;
|
enum setpoint_type_t setpoint_type = NONE;
|
||||||
@@ -167,9 +149,6 @@ void handleState(const mavros_msgs::State& s)
|
|||||||
inline void publishBodyFrame()
|
inline void publishBodyFrame()
|
||||||
{
|
{
|
||||||
if (body.child_frame_id.empty()) return;
|
if (body.child_frame_id.empty()) return;
|
||||||
if (!body.header.stamp.isZero() && body.header.stamp == local_position.header.stamp) {
|
|
||||||
return; // avoid TF_REPEATED_DATA warnings
|
|
||||||
}
|
|
||||||
|
|
||||||
tf::Quaternion q;
|
tf::Quaternion q;
|
||||||
q.setRPY(0, 0, tf::getYaw(local_position.pose.orientation));
|
q.setRPY(0, 0, tf::getYaw(local_position.pose.orientation));
|
||||||
@@ -187,7 +166,7 @@ void handleLocalPosition(const PoseStamped& pose)
|
|||||||
{
|
{
|
||||||
local_position = pose;
|
local_position = pose;
|
||||||
publishBodyFrame();
|
publishBodyFrame();
|
||||||
// TODO: home?
|
// TODO: terrain?, home?
|
||||||
}
|
}
|
||||||
|
|
||||||
// wait for transform without interrupting publishing setpoints
|
// wait for transform without interrupting publishing setpoints
|
||||||
@@ -202,24 +181,9 @@ inline bool waitTransform(const string& target, const string& source,
|
|||||||
ros::spinOnce();
|
ros::spinOnce();
|
||||||
r.sleep();
|
r.sleep();
|
||||||
}
|
}
|
||||||
return false;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
void handleAltitude(const Altitude& alt)
|
#define TIMEOUT(msg, timeout) (ros::Time::now() - msg.header.stamp > timeout)
|
||||||
{
|
|
||||||
// publish terrain frame
|
|
||||||
if (!std::isfinite(alt.bottom_clearance)) return;
|
|
||||||
// terrain.header.stamp = alt.header.stamp;
|
|
||||||
|
|
||||||
if (!waitTransform(local_frame, body.child_frame_id, alt.header.stamp, ros::Duration(0.1))) return;
|
|
||||||
|
|
||||||
auto t = tf_buffer.lookupTransform(local_frame, body.child_frame_id, alt.header.stamp);
|
|
||||||
t.child_frame_id = terrain.child_frame_id;
|
|
||||||
t.transform.translation.z -= alt.bottom_clearance;
|
|
||||||
static_transform_broadcaster->sendTransform(t);
|
|
||||||
}
|
|
||||||
|
|
||||||
#define TIMEOUT(msg, timeout) (msg.header.stamp.isZero() || (ros::Time::now() - msg.header.stamp > timeout))
|
|
||||||
|
|
||||||
bool getTelemetry(GetTelemetry::Request& req, GetTelemetry::Response& res)
|
bool getTelemetry(GetTelemetry::Request& req, GetTelemetry::Response& res)
|
||||||
{
|
{
|
||||||
@@ -238,11 +202,11 @@ bool getTelemetry(GetTelemetry::Request& req, GetTelemetry::Response& res)
|
|||||||
res.vx = NAN;
|
res.vx = NAN;
|
||||||
res.vy = NAN;
|
res.vy = NAN;
|
||||||
res.vz = NAN;
|
res.vz = NAN;
|
||||||
res.roll = NAN;
|
|
||||||
res.pitch = NAN;
|
res.pitch = NAN;
|
||||||
|
res.roll = NAN;
|
||||||
res.yaw = NAN;
|
res.yaw = NAN;
|
||||||
res.roll_rate = NAN;
|
|
||||||
res.pitch_rate = NAN;
|
res.pitch_rate = NAN;
|
||||||
|
res.roll_rate = NAN;
|
||||||
res.yaw_rate = NAN;
|
res.yaw_rate = NAN;
|
||||||
res.voltage = NAN;
|
res.voltage = NAN;
|
||||||
res.cell_voltage = NAN;
|
res.cell_voltage = NAN;
|
||||||
@@ -372,20 +336,20 @@ inline float getDistance(const Point& from, const Point& to)
|
|||||||
return hypot(from.x - to.x, from.y - to.y, from.z - to.z);
|
return hypot(from.x - to.x, from.y - to.y, from.z - to.z);
|
||||||
}
|
}
|
||||||
|
|
||||||
void getNavigateSetpoint(const ros::Time& stamp, const float speed, Point& nav_setpoint)
|
void getNavigateSetpoint(const ros::Time& stamp, float speed, Point& nav_setpoint)
|
||||||
{
|
{
|
||||||
if (wait_armed) {
|
if (wait_armed) {
|
||||||
// don't start navigating if we're waiting arming
|
// don't start navigating if we're waiting arming
|
||||||
nav_start.header.stamp = stamp;
|
nav_start.header.stamp = stamp;
|
||||||
}
|
}
|
||||||
|
|
||||||
float distance = getDistance(nav_start.pose.position, setpoint_pose_local.pose.position);
|
float distance = getDistance(nav_start.pose.position, setpoint_position_transformed.pose.position);
|
||||||
float time = distance / speed;
|
float time = distance / speed;
|
||||||
float passed = std::min((stamp - nav_start.header.stamp).toSec() / time, 1.0);
|
float passed = std::min((stamp - nav_start.header.stamp).toSec() / time, 1.0);
|
||||||
|
|
||||||
nav_setpoint.x = nav_start.pose.position.x + (setpoint_pose_local.pose.position.x - nav_start.pose.position.x) * passed;
|
nav_setpoint.x = nav_start.pose.position.x + (setpoint_position_transformed.pose.position.x - nav_start.pose.position.x) * passed;
|
||||||
nav_setpoint.y = nav_start.pose.position.y + (setpoint_pose_local.pose.position.y - nav_start.pose.position.y) * passed;
|
nav_setpoint.y = nav_start.pose.position.y + (setpoint_position_transformed.pose.position.y - nav_start.pose.position.y) * passed;
|
||||||
nav_setpoint.z = nav_start.pose.position.z + (setpoint_pose_local.pose.position.z - nav_start.pose.position.z) * passed;
|
nav_setpoint.z = nav_start.pose.position.z + (setpoint_position_transformed.pose.position.z - nav_start.pose.position.z) * passed;
|
||||||
}
|
}
|
||||||
|
|
||||||
PoseStamped globalToLocal(double lat, double lon)
|
PoseStamped globalToLocal(double lat, double lon)
|
||||||
@@ -416,101 +380,44 @@ PoseStamped globalToLocal(double lat, double lon)
|
|||||||
return pose;
|
return pose;
|
||||||
}
|
}
|
||||||
|
|
||||||
// publish navigate_target frame
|
|
||||||
void publishTarget(ros::Time stamp, bool _static = false)
|
|
||||||
{
|
|
||||||
bool single_frame = (setpoint_position.header.frame_id == setpoint_altitude.header.frame_id);
|
|
||||||
|
|
||||||
// handle yaw for target frame
|
|
||||||
if (setpoint_yaw_type == YAW || setpoint_yaw_type == YAW_RATE) { // use last set yaw for yaw_rate
|
|
||||||
if (setpoint_altitude.header.frame_id == yaw_frame_id) {
|
|
||||||
target.transform.rotation = tf::createQuaternionMsgFromYaw(setpoint_yaw);
|
|
||||||
} else {
|
|
||||||
single_frame = false;
|
|
||||||
target.transform.rotation = tf::createQuaternionMsgFromYaw(yaw_local);
|
|
||||||
}
|
|
||||||
} else if (setpoint_yaw_type == TOWARDS) {
|
|
||||||
single_frame = false;
|
|
||||||
target.transform.rotation = tf::createQuaternionMsgFromYaw(yaw_local);
|
|
||||||
}
|
|
||||||
|
|
||||||
if (_static && single_frame) {
|
|
||||||
// publish at user's command, if all frames are the same
|
|
||||||
target.header.frame_id = setpoint_position.header.frame_id;
|
|
||||||
target.header.stamp = stamp;
|
|
||||||
target.transform.translation.x = setpoint_position.point.x;
|
|
||||||
target.transform.translation.y = setpoint_position.point.y;
|
|
||||||
target.transform.translation.z = setpoint_position.point.z;
|
|
||||||
|
|
||||||
} else if (!_static) {
|
|
||||||
// publish at each iteration, if frames are different
|
|
||||||
target.header = setpoint_pose_local.header;
|
|
||||||
target.transform.translation.x = setpoint_pose_local.pose.position.x;
|
|
||||||
target.transform.translation.y = setpoint_pose_local.pose.position.y;
|
|
||||||
target.transform.translation.z = setpoint_pose_local.pose.position.z;
|
|
||||||
}
|
|
||||||
|
|
||||||
static_transform_broadcaster->sendTransform(target);
|
|
||||||
}
|
|
||||||
|
|
||||||
void publish(const ros::Time stamp)
|
void publish(const ros::Time stamp)
|
||||||
{
|
{
|
||||||
if (setpoint_type == NONE) return;
|
if (setpoint_type == NONE) return;
|
||||||
|
|
||||||
position_raw_msg.header.stamp = stamp;
|
position_raw_msg.header.stamp = stamp;
|
||||||
|
thrust_msg.header.stamp = stamp;
|
||||||
|
rates_msg.header.stamp = stamp;
|
||||||
|
|
||||||
// transform position
|
try {
|
||||||
if (setpoint_type == NAVIGATE || setpoint_type == NAVIGATE_GLOBAL || setpoint_type == POSITION) {
|
// transform position and/or yaw
|
||||||
setpoint_position.header.stamp = stamp;
|
if (setpoint_type == NAVIGATE || setpoint_type == NAVIGATE_GLOBAL || setpoint_type == POSITION || setpoint_type == VELOCITY || setpoint_type == ATTITUDE) {
|
||||||
setpoint_altitude.header.stamp = stamp;
|
setpoint_position.header.stamp = stamp;
|
||||||
// transform xy
|
tf_buffer.transform(setpoint_position, setpoint_position_transformed, local_frame, ros::Duration(0.05));
|
||||||
try {
|
|
||||||
auto xy = tf_buffer.transform(setpoint_position, local_frame, ros::Duration(0.05));
|
|
||||||
setpoint_pose_local.header = xy.header;
|
|
||||||
setpoint_pose_local.pose.position.x = xy.point.x;
|
|
||||||
setpoint_pose_local.pose.position.y = xy.point.y;
|
|
||||||
} catch (tf2::TransformException& ex) {
|
|
||||||
// can't transform xy, use last known
|
|
||||||
ROS_WARN_THROTTLE(10, "can't transform: %s", ex.what());
|
|
||||||
}
|
}
|
||||||
// transform altitude
|
|
||||||
try {
|
// transform velocity
|
||||||
setpoint_pose_local.pose.position.z = tf_buffer.transform(setpoint_altitude, local_frame, ros::Duration(0.05)).point.z;
|
if (setpoint_type == VELOCITY) {
|
||||||
} catch (tf2::TransformException& ex) {
|
setpoint_velocity.header.stamp = stamp;
|
||||||
// can't transform altitude, use last known
|
tf_buffer.transform(setpoint_velocity, setpoint_velocity_transformed, local_frame, ros::Duration(0.05));
|
||||||
ROS_WARN_THROTTLE(10, "can't transform: %s", ex.what());
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
} catch (const tf2::TransformException& e) {
|
||||||
|
ROS_WARN_THROTTLE(10, "can't transform");
|
||||||
}
|
}
|
||||||
|
|
||||||
// transform yaw
|
|
||||||
if (setpoint_yaw_type == YAW) {
|
|
||||||
try {
|
|
||||||
QuaternionStamped q;
|
|
||||||
q.header.stamp = stamp;
|
|
||||||
q.header.frame_id = yaw_frame_id;
|
|
||||||
q.quaternion = tf::createQuaternionMsgFromYaw(setpoint_yaw);
|
|
||||||
yaw_local = tf2::getYaw(tf_buffer.transform(q, local_frame, ros::Duration(0.05)).quaternion);
|
|
||||||
} catch (tf2::TransformException& ex) {
|
|
||||||
// can't transform yaw, use last known
|
|
||||||
ROS_WARN_THROTTLE(10, "can't transform: %s", ex.what());
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// compute navigate setpoint
|
|
||||||
if (setpoint_type == NAVIGATE || setpoint_type == NAVIGATE_GLOBAL) {
|
if (setpoint_type == NAVIGATE || setpoint_type == NAVIGATE_GLOBAL) {
|
||||||
|
position_msg.pose.orientation = setpoint_position_transformed.pose.orientation; // copy yaw
|
||||||
getNavigateSetpoint(stamp, nav_speed, position_msg.pose.position);
|
getNavigateSetpoint(stamp, nav_speed, position_msg.pose.position);
|
||||||
|
|
||||||
if (setpoint_yaw_type == TOWARDS) {
|
if (setpoint_yaw_type == TOWARDS) {
|
||||||
yaw_local = atan2(position_msg.pose.position.y - nav_start.pose.position.y,
|
double yaw_towards = atan2(position_msg.pose.position.y - nav_start.pose.position.y,
|
||||||
position_msg.pose.position.x - nav_start.pose.position.x);
|
position_msg.pose.position.x - nav_start.pose.position.x);
|
||||||
|
position_msg.pose.orientation = tf::createQuaternionMsgFromRollPitchYaw(0, 0, yaw_towards);
|
||||||
}
|
}
|
||||||
|
|
||||||
position_msg.pose.orientation = tf::createQuaternionMsgFromYaw(yaw_local);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
if (setpoint_type == POSITION) {
|
if (setpoint_type == POSITION) {
|
||||||
position_msg = setpoint_pose_local;
|
position_msg = setpoint_position_transformed;
|
||||||
position_msg.pose.orientation = tf::createQuaternionMsgFromYaw(yaw_local);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
if (setpoint_type == POSITION || setpoint_type == NAVIGATE || setpoint_type == NAVIGATE_GLOBAL) {
|
if (setpoint_type == POSITION || setpoint_type == NAVIGATE || setpoint_type == NAVIGATE_GLOBAL) {
|
||||||
@@ -527,17 +434,13 @@ void publish(const ros::Time stamp)
|
|||||||
PositionTarget::IGNORE_AFY +
|
PositionTarget::IGNORE_AFY +
|
||||||
PositionTarget::IGNORE_AFZ +
|
PositionTarget::IGNORE_AFZ +
|
||||||
PositionTarget::IGNORE_YAW;
|
PositionTarget::IGNORE_YAW;
|
||||||
position_raw_msg.yaw_rate = setpoint_rates.z;
|
position_raw_msg.yaw_rate = setpoint_yaw_rate;
|
||||||
position_raw_msg.position = position_msg.pose.position;
|
position_raw_msg.position = position_msg.pose.position;
|
||||||
position_raw_pub.publish(position_raw_msg);
|
position_raw_pub.publish(position_raw_msg);
|
||||||
}
|
}
|
||||||
|
|
||||||
// publish setpoint frame
|
// publish setpoint frame
|
||||||
if (!setpoint.child_frame_id.empty()) {
|
if (!setpoint.child_frame_id.empty()) {
|
||||||
if (setpoint.header.stamp >= position_msg.header.stamp) {
|
|
||||||
return; // avoid TF_REPEATED_DATA warnings
|
|
||||||
}
|
|
||||||
|
|
||||||
setpoint.transform.translation.x = position_msg.pose.position.x;
|
setpoint.transform.translation.x = position_msg.pose.position.x;
|
||||||
setpoint.transform.translation.y = position_msg.pose.position.y;
|
setpoint.transform.translation.y = position_msg.pose.position.y;
|
||||||
setpoint.transform.translation.z = position_msg.pose.position.z;
|
setpoint.transform.translation.z = position_msg.pose.position.z;
|
||||||
@@ -546,22 +449,9 @@ void publish(const ros::Time stamp)
|
|||||||
setpoint.header.stamp = position_msg.header.stamp;
|
setpoint.header.stamp = position_msg.header.stamp;
|
||||||
transform_broadcaster->sendTransform(setpoint);
|
transform_broadcaster->sendTransform(setpoint);
|
||||||
}
|
}
|
||||||
|
|
||||||
// publish dynamic target frame
|
|
||||||
publishTarget(stamp);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
if (setpoint_type == VELOCITY) {
|
if (setpoint_type == VELOCITY) {
|
||||||
// transform velocity to local frame
|
|
||||||
setpoint_velocity.header.stamp = stamp;
|
|
||||||
try {
|
|
||||||
setpoint_velocity_local = tf_buffer.transform(setpoint_velocity, local_frame, ros::Duration(0.05));
|
|
||||||
} catch (tf2::TransformException& ex) {
|
|
||||||
// can't transform velocity, use last known
|
|
||||||
ROS_WARN_THROTTLE(10, "can't transform: %s", ex.what());
|
|
||||||
}
|
|
||||||
|
|
||||||
// publish velocity
|
|
||||||
position_raw_msg.type_mask = PositionTarget::IGNORE_PX +
|
position_raw_msg.type_mask = PositionTarget::IGNORE_PX +
|
||||||
PositionTarget::IGNORE_PY +
|
PositionTarget::IGNORE_PY +
|
||||||
PositionTarget::IGNORE_PZ +
|
PositionTarget::IGNORE_PZ +
|
||||||
@@ -569,22 +459,14 @@ void publish(const ros::Time stamp)
|
|||||||
PositionTarget::IGNORE_AFY +
|
PositionTarget::IGNORE_AFY +
|
||||||
PositionTarget::IGNORE_AFZ;
|
PositionTarget::IGNORE_AFZ;
|
||||||
position_raw_msg.type_mask += setpoint_yaw_type == YAW ? PositionTarget::IGNORE_YAW_RATE : PositionTarget::IGNORE_YAW;
|
position_raw_msg.type_mask += setpoint_yaw_type == YAW ? PositionTarget::IGNORE_YAW_RATE : PositionTarget::IGNORE_YAW;
|
||||||
position_raw_msg.velocity = setpoint_velocity_local.vector;
|
position_raw_msg.velocity = setpoint_velocity_transformed.vector;
|
||||||
position_raw_msg.yaw = yaw_local;
|
position_raw_msg.yaw = tf2::getYaw(setpoint_position_transformed.pose.orientation);
|
||||||
position_raw_msg.yaw_rate = setpoint_rates.z;
|
position_raw_msg.yaw_rate = setpoint_yaw_rate;
|
||||||
position_raw_pub.publish(position_raw_msg);
|
position_raw_pub.publish(position_raw_msg);
|
||||||
}
|
}
|
||||||
|
|
||||||
if (setpoint_type == ATTITUDE) {
|
if (setpoint_type == ATTITUDE) {
|
||||||
PoseStamped msg;
|
attitude_pub.publish(setpoint_position_transformed);
|
||||||
msg.header.stamp = stamp;
|
|
||||||
msg.header.frame_id = local_frame;
|
|
||||||
msg.pose.orientation = tf::createQuaternionMsgFromRollPitchYaw(setpoint_roll, setpoint_pitch, yaw_local);
|
|
||||||
attitude_pub.publish(msg);
|
|
||||||
|
|
||||||
Thrust thrust_msg;
|
|
||||||
thrust_msg.header.stamp = stamp;
|
|
||||||
thrust_msg.thrust = setpoint_thrust;
|
|
||||||
thrust_pub.publish(thrust_msg);
|
thrust_pub.publish(thrust_msg);
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -593,12 +475,11 @@ void publish(const ros::Time stamp)
|
|||||||
// thrust_pub.publish(thrust_msg);
|
// thrust_pub.publish(thrust_msg);
|
||||||
// mavros rates topics waits for rates in local frame
|
// mavros rates topics waits for rates in local frame
|
||||||
// use rates in body frame for simplicity
|
// use rates in body frame for simplicity
|
||||||
AttitudeTarget att_raw_msg;
|
|
||||||
att_raw_msg.header.stamp = stamp;
|
att_raw_msg.header.stamp = stamp;
|
||||||
att_raw_msg.header.frame_id = fcu_frame;
|
att_raw_msg.header.frame_id = fcu_frame;
|
||||||
att_raw_msg.type_mask = AttitudeTarget::IGNORE_ATTITUDE;
|
att_raw_msg.type_mask = AttitudeTarget::IGNORE_ATTITUDE;
|
||||||
att_raw_msg.body_rate = setpoint_rates;
|
att_raw_msg.body_rate = rates_msg.twist.angular;
|
||||||
att_raw_msg.thrust = setpoint_thrust;
|
att_raw_msg.thrust = thrust_msg.thrust;
|
||||||
attitude_raw_pub.publish(att_raw_msg);
|
attitude_raw_pub.publish(att_raw_msg);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -638,59 +519,10 @@ inline void checkState()
|
|||||||
throw std::runtime_error("No connection to FCU, https://clover.coex.tech/connection");
|
throw std::runtime_error("No connection to FCU, https://clover.coex.tech/connection");
|
||||||
}
|
}
|
||||||
|
|
||||||
void publishState()
|
|
||||||
{
|
|
||||||
clover::State msg;
|
|
||||||
msg.mode = setpoint_type;
|
|
||||||
msg.yaw_mode = setpoint_yaw_type;
|
|
||||||
|
|
||||||
if (setpoint_position.header.frame_id.empty()) {
|
|
||||||
msg.x = NAN;
|
|
||||||
msg.y = NAN;
|
|
||||||
msg.z = NAN;
|
|
||||||
} else {
|
|
||||||
msg.x = setpoint_position.point.x;
|
|
||||||
msg.y = setpoint_position.point.y;
|
|
||||||
msg.z = setpoint_altitude.point.z;
|
|
||||||
}
|
|
||||||
|
|
||||||
msg.speed = nav_speed;
|
|
||||||
msg.lat = setpoint_lat;
|
|
||||||
msg.lon = setpoint_lon;
|
|
||||||
msg.vx = setpoint_velocity.vector.x;
|
|
||||||
msg.vy = setpoint_velocity.vector.y;
|
|
||||||
msg.vz = setpoint_velocity.vector.z;
|
|
||||||
msg.roll = setpoint_roll;
|
|
||||||
msg.pitch = setpoint_pitch;
|
|
||||||
msg.yaw = !yaw_frame_id.empty() ? setpoint_yaw : NAN;
|
|
||||||
|
|
||||||
msg.roll_rate = setpoint_rates.x;
|
|
||||||
msg.pitch_rate = setpoint_rates.y;
|
|
||||||
msg.yaw_rate = setpoint_rates.z;
|
|
||||||
msg.thrust = setpoint_thrust;
|
|
||||||
|
|
||||||
if (setpoint_type == VELOCITY) {
|
|
||||||
msg.xy_frame_id = setpoint_velocity.header.frame_id;
|
|
||||||
msg.z_frame_id = setpoint_velocity.header.frame_id;
|
|
||||||
} else {
|
|
||||||
msg.xy_frame_id = setpoint_position.header.frame_id;
|
|
||||||
msg.z_frame_id = setpoint_altitude.header.frame_id;
|
|
||||||
}
|
|
||||||
msg.yaw_frame_id = yaw_frame_id;
|
|
||||||
|
|
||||||
state_pub.publish(msg);
|
|
||||||
}
|
|
||||||
|
|
||||||
inline float safe(float value) {
|
|
||||||
return std::isfinite(value) ? value : 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
#define ENSURE_FINITE(var) { if (!std::isfinite(var)) throw std::runtime_error(#var " argument cannot be NaN or Inf"); }
|
#define ENSURE_FINITE(var) { if (!std::isfinite(var)) throw std::runtime_error(#var " argument cannot be NaN or Inf"); }
|
||||||
|
|
||||||
#define ENSURE_NON_INF(var) { if (std::isinf(var)) throw std::runtime_error(#var " argument cannot be Inf"); }
|
|
||||||
|
|
||||||
bool serve(enum setpoint_type_t sp_type, float x, float y, float z, float vx, float vy, float vz,
|
bool serve(enum setpoint_type_t sp_type, float x, float y, float z, float vx, float vy, float vz,
|
||||||
float roll, float pitch, float yaw, float roll_rate, float pitch_rate, float yaw_rate, // editorconfig-checker-disable-line
|
float pitch, float roll, float yaw, float pitch_rate, float roll_rate, float yaw_rate, // editorconfig-checker-disable-line
|
||||||
float lat, float lon, float thrust, float speed, string frame_id, bool auto_arm, // editorconfig-checker-disable-line
|
float lat, float lon, float thrust, float speed, string frame_id, bool auto_arm, // editorconfig-checker-disable-line
|
||||||
uint8_t& success, string& message) // editorconfig-checker-disable-line
|
uint8_t& success, string& message) // editorconfig-checker-disable-line
|
||||||
{
|
{
|
||||||
@@ -717,40 +549,69 @@ bool serve(enum setpoint_type_t sp_type, float x, float y, float z, float vx, fl
|
|||||||
auto search = reference_frames.find(frame_id);
|
auto search = reference_frames.find(frame_id);
|
||||||
const string& reference_frame = search == reference_frames.end() ? frame_id : search->second;
|
const string& reference_frame = search == reference_frames.end() ? frame_id : search->second;
|
||||||
|
|
||||||
ENSURE_NON_INF(x);
|
// Serve "partial" commands
|
||||||
ENSURE_NON_INF(y);
|
|
||||||
ENSURE_NON_INF(z);
|
|
||||||
ENSURE_NON_INF(speed); // TODO: allow inf
|
|
||||||
ENSURE_NON_INF(vx);
|
|
||||||
ENSURE_NON_INF(vy);
|
|
||||||
ENSURE_NON_INF(vz);
|
|
||||||
ENSURE_NON_INF(roll);
|
|
||||||
ENSURE_NON_INF(pitch);
|
|
||||||
ENSURE_NON_INF(roll_rate);
|
|
||||||
ENSURE_NON_INF(pitch_rate);
|
|
||||||
ENSURE_NON_INF(yaw_rate);
|
|
||||||
ENSURE_NON_INF(thrust);
|
|
||||||
|
|
||||||
if (sp_type == NAVIGATE_GLOBAL) {
|
if (!auto_arm && std::isfinite(yaw) &&
|
||||||
|
isnan(x) && isnan(y) && isnan(z) && isnan(vx) && isnan(vy) && isnan(vz) &&
|
||||||
|
isnan(pitch) && isnan(roll) && isnan(thrust) &&
|
||||||
|
isnan(lat) && isnan(lon)) {
|
||||||
|
// change only the yaw
|
||||||
|
if (setpoint_type == POSITION || setpoint_type == NAVIGATE || setpoint_type == NAVIGATE_GLOBAL || setpoint_type == VELOCITY) {
|
||||||
|
if (!waitTransform(setpoint_position.header.frame_id, frame_id, stamp, transform_timeout))
|
||||||
|
throw std::runtime_error("Can't transform from " + frame_id + " to " + setpoint_position.header.frame_id);
|
||||||
|
|
||||||
|
message = "Changing yaw only";
|
||||||
|
|
||||||
|
QuaternionStamped q;
|
||||||
|
q.header.frame_id = frame_id;
|
||||||
|
q.header.stamp = stamp;
|
||||||
|
q.quaternion = tf::createQuaternionMsgFromYaw(yaw); // TODO: pitch=0, roll=0 is not totally correct
|
||||||
|
setpoint_position.pose.orientation = tf_buffer.transform(q, setpoint_position.header.frame_id).quaternion;
|
||||||
|
setpoint_yaw_type = YAW;
|
||||||
|
goto publish_setpoint;
|
||||||
|
} else {
|
||||||
|
throw std::runtime_error("Setting yaw is possible only when position or velocity setpoints active");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if (!auto_arm && std::isfinite(yaw_rate) &&
|
||||||
|
isnan(x) && isnan(y) && isnan(z) && isnan(vx) && isnan(vy) && isnan(vz) &&
|
||||||
|
isnan(pitch) && isnan(roll) && isnan(yaw) && isnan(thrust) &&
|
||||||
|
isnan(lat) && isnan(lon)) {
|
||||||
|
// change only the yaw rate
|
||||||
|
if (setpoint_type == POSITION || setpoint_type == NAVIGATE || setpoint_type == NAVIGATE_GLOBAL || setpoint_type == VELOCITY) {
|
||||||
|
message = "Changing yaw rate only";
|
||||||
|
|
||||||
|
setpoint_yaw_type = YAW_RATE;
|
||||||
|
setpoint_yaw_rate = yaw_rate;
|
||||||
|
goto publish_setpoint;
|
||||||
|
} else {
|
||||||
|
throw std::runtime_error("Setting yaw rate is possible only when position or velocity setpoints active");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Serve normal commands
|
||||||
|
|
||||||
|
if (sp_type == NAVIGATE || sp_type == POSITION) {
|
||||||
|
ENSURE_FINITE(x);
|
||||||
|
ENSURE_FINITE(y);
|
||||||
|
ENSURE_FINITE(z);
|
||||||
|
} else if (sp_type == NAVIGATE_GLOBAL) {
|
||||||
ENSURE_FINITE(lat);
|
ENSURE_FINITE(lat);
|
||||||
ENSURE_FINITE(lon);
|
ENSURE_FINITE(lon);
|
||||||
}
|
ENSURE_FINITE(z);
|
||||||
|
} else if (sp_type == VELOCITY) {
|
||||||
if (isfinite(x) != isfinite(y)) {
|
ENSURE_FINITE(vx);
|
||||||
throw std::runtime_error("x and y can be set only together");
|
ENSURE_FINITE(vy);
|
||||||
}
|
ENSURE_FINITE(vz);
|
||||||
|
} else if (sp_type == ATTITUDE) {
|
||||||
if (isfinite(yaw_rate)) {
|
ENSURE_FINITE(pitch);
|
||||||
if (sp_type > RATES && setpoint_type == ATTITUDE) {
|
ENSURE_FINITE(roll);
|
||||||
throw std::runtime_error("Yaw rate cannot be set in attitude mode.");
|
ENSURE_FINITE(thrust);
|
||||||
}
|
} else if (sp_type == RATES) {
|
||||||
}
|
ENSURE_FINITE(pitch_rate);
|
||||||
|
ENSURE_FINITE(roll_rate);
|
||||||
// set_altitude
|
ENSURE_FINITE(thrust);
|
||||||
if (sp_type == _ALTITUDE) {
|
|
||||||
if (setpoint_type == VELOCITY || setpoint_type == ATTITUDE || setpoint_type == RATES) {
|
|
||||||
throw std::runtime_error("Altitude cannot be set in velocity, attitude or rates mode.");
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
if (sp_type == NAVIGATE || sp_type == NAVIGATE_GLOBAL) {
|
if (sp_type == NAVIGATE || sp_type == NAVIGATE_GLOBAL) {
|
||||||
@@ -764,13 +625,20 @@ bool serve(enum setpoint_type_t sp_type, float x, float y, float z, float vx, fl
|
|||||||
speed = default_speed;
|
speed = default_speed;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
if (sp_type == NAVIGATE || sp_type == NAVIGATE_GLOBAL || sp_type == POSITION || sp_type == VELOCITY) {
|
||||||
|
if (yaw_rate != 0 && !std::isnan(yaw))
|
||||||
|
throw std::runtime_error("Yaw value should be NaN for setting yaw rate");
|
||||||
|
|
||||||
|
if (std::isnan(yaw_rate) && std::isnan(yaw))
|
||||||
|
throw std::runtime_error("Both yaw and yaw_rate cannot be NaN");
|
||||||
|
}
|
||||||
|
|
||||||
if (sp_type == NAVIGATE_GLOBAL) {
|
if (sp_type == NAVIGATE_GLOBAL) {
|
||||||
if (TIMEOUT(global_position, global_position_timeout))
|
if (TIMEOUT(global_position, global_position_timeout))
|
||||||
throw std::runtime_error("No global position");
|
throw std::runtime_error("No global position");
|
||||||
}
|
}
|
||||||
|
|
||||||
// if any value need to be transformed to reference frame
|
if (sp_type == NAVIGATE || sp_type == NAVIGATE_GLOBAL || sp_type == POSITION || sp_type == VELOCITY || sp_type == ATTITUDE) {
|
||||||
if (isfinite(x) || isfinite(y) || isfinite(z) || isfinite(vx) || isfinite(vy) || isfinite(vz) || isfinite(yaw)) {
|
|
||||||
// make sure transform from frame_id to reference frame available
|
// make sure transform from frame_id to reference frame available
|
||||||
if (!waitTransform(reference_frame, frame_id, stamp, transform_timeout))
|
if (!waitTransform(reference_frame, frame_id, stamp, transform_timeout))
|
||||||
throw std::runtime_error("Can't transform from " + frame_id + " to " + reference_frame);
|
throw std::runtime_error("Can't transform from " + frame_id + " to " + reference_frame);
|
||||||
@@ -787,26 +655,15 @@ bool serve(enum setpoint_type_t sp_type, float x, float y, float z, float vx, fl
|
|||||||
auto xy_in_req_frame = tf_buffer.transform(pose_local, frame_id);
|
auto xy_in_req_frame = tf_buffer.transform(pose_local, frame_id);
|
||||||
x = xy_in_req_frame.pose.position.x;
|
x = xy_in_req_frame.pose.position.x;
|
||||||
y = xy_in_req_frame.pose.position.y;
|
y = xy_in_req_frame.pose.position.y;
|
||||||
setpoint_lat = lat;
|
|
||||||
setpoint_lon = lon;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// Everything fine - switch setpoint type
|
// Everything fine - switch setpoint type
|
||||||
if (sp_type <= RATES) {
|
setpoint_type = sp_type;
|
||||||
setpoint_type = sp_type;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (setpoint_type != NAVIGATE && setpoint_type != NAVIGATE_GLOBAL) {
|
if (sp_type != NAVIGATE && sp_type != NAVIGATE_GLOBAL) {
|
||||||
nav_from_sp_flag = false;
|
nav_from_sp_flag = false;
|
||||||
}
|
}
|
||||||
|
|
||||||
if (auto_arm || setpoint_type == VELOCITY || setpoint_type == ATTITUDE || setpoint_type == RATES) {
|
|
||||||
// invalidate position setpoint
|
|
||||||
setpoint_position.header.frame_id = "";
|
|
||||||
setpoint_altitude.header.frame_id = "";
|
|
||||||
yaw_frame_id = "";
|
|
||||||
}
|
|
||||||
|
|
||||||
if (sp_type == NAVIGATE || sp_type == NAVIGATE_GLOBAL) {
|
if (sp_type == NAVIGATE || sp_type == NAVIGATE_GLOBAL) {
|
||||||
// starting point
|
// starting point
|
||||||
if (nav_from_sp && nav_from_sp_flag) {
|
if (nav_from_sp && nav_from_sp_flag) {
|
||||||
@@ -815,139 +672,84 @@ bool serve(enum setpoint_type_t sp_type, float x, float y, float z, float vx, fl
|
|||||||
} else {
|
} else {
|
||||||
nav_start = local_position;
|
nav_start = local_position;
|
||||||
}
|
}
|
||||||
|
nav_speed = speed;
|
||||||
if (!isnan(speed)) {
|
|
||||||
nav_speed = speed;
|
|
||||||
}
|
|
||||||
|
|
||||||
nav_from_sp_flag = true;
|
nav_from_sp_flag = true;
|
||||||
}
|
}
|
||||||
|
|
||||||
// handle position
|
// if (sp_type == NAVIGATE || sp_type == NAVIGATE_GLOBAL || sp_type == POSITION || sp_type == VELOCITY) {
|
||||||
if (setpoint_type == NAVIGATE || setpoint_type == NAVIGATE_GLOBAL || setpoint_type == POSITION) {
|
// if (std::isnan(yaw) && yaw_rate == 0) {
|
||||||
|
// // keep yaw unchanged
|
||||||
|
// // TODO: this is incorrect, because we need yaw in desired frame
|
||||||
|
// yaw = tf2::getYaw(local_position.pose.orientation);
|
||||||
|
// }
|
||||||
|
// }
|
||||||
|
|
||||||
PointStamped desired;
|
if (sp_type == POSITION || sp_type == NAVIGATE || sp_type == NAVIGATE_GLOBAL || sp_type == VELOCITY || sp_type == ATTITUDE) {
|
||||||
desired.header.frame_id = frame_id;
|
// destination point and/or yaw
|
||||||
desired.header.stamp = stamp;
|
PoseStamped ps;
|
||||||
desired.point.x = safe(x);
|
ps.header.frame_id = frame_id;
|
||||||
desired.point.y = safe(y);
|
ps.header.stamp = stamp;
|
||||||
desired.point.z = safe(z);
|
ps.pose.position.x = x;
|
||||||
|
ps.pose.position.y = y;
|
||||||
|
ps.pose.position.z = z;
|
||||||
|
ps.pose.orientation.w = 1.0; // Ensure quaternion is always valid
|
||||||
|
|
||||||
// transform to reference frame
|
if (std::isnan(yaw)) {
|
||||||
desired = tf_buffer.transform(desired, reference_frame);
|
setpoint_yaw_type = YAW_RATE;
|
||||||
|
setpoint_yaw_rate = yaw_rate;
|
||||||
// set horizontal position
|
} else if (std::isinf(yaw) && yaw > 0) {
|
||||||
if (isfinite(x) && isfinite(y)) {
|
|
||||||
setpoint_position = desired;
|
|
||||||
} else if (setpoint_position.header.frame_id.empty()) {
|
|
||||||
// TODO: use transform for current stamp
|
|
||||||
setpoint_position.header = local_position.header;
|
|
||||||
setpoint_position.point = local_position.pose.position;
|
|
||||||
}
|
|
||||||
|
|
||||||
// set altitude
|
|
||||||
if (isfinite(z)) {
|
|
||||||
setpoint_altitude = desired;
|
|
||||||
} else if (setpoint_altitude.header.frame_id.empty()) {
|
|
||||||
setpoint_altitude.header = local_position.header;
|
|
||||||
setpoint_altitude.point = local_position.pose.position;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// handle velocity
|
|
||||||
if (sp_type == VELOCITY) {
|
|
||||||
// TODO: allow setting different modes by altitude and xy
|
|
||||||
Vector3Stamped desired;
|
|
||||||
desired.header.frame_id = frame_id;
|
|
||||||
desired.header.stamp = stamp;
|
|
||||||
desired.vector.x = safe(vx);
|
|
||||||
desired.vector.y = safe(vy);
|
|
||||||
desired.vector.z = safe(vz);
|
|
||||||
|
|
||||||
// transform to reference frame
|
|
||||||
desired = tf_buffer.transform(desired, reference_frame);
|
|
||||||
setpoint_velocity.header = desired.header;
|
|
||||||
|
|
||||||
// set horizontal velocity
|
|
||||||
if (isfinite(vx) && isfinite(vy)) {
|
|
||||||
setpoint_velocity.vector.x = desired.vector.x;
|
|
||||||
setpoint_velocity.vector.y = desired.vector.y;
|
|
||||||
}
|
|
||||||
|
|
||||||
// set vertical velocity
|
|
||||||
if (isfinite(vz)) {
|
|
||||||
setpoint_velocity.vector.z = desired.vector.z;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// handle yaw
|
|
||||||
if (sp_type == NAVIGATE || sp_type == NAVIGATE_GLOBAL || sp_type == POSITION || sp_type == VELOCITY || sp_type == ATTITUDE || sp_type == _YAW) {
|
|
||||||
if (isfinite(yaw)) {
|
|
||||||
setpoint_yaw_type = YAW;
|
|
||||||
QuaternionStamped desired;
|
|
||||||
desired.header.frame_id = frame_id;
|
|
||||||
desired.header.stamp = stamp;
|
|
||||||
desired.quaternion = tf::createQuaternionMsgFromYaw(yaw);
|
|
||||||
|
|
||||||
// transform to reference frame
|
|
||||||
desired = tf_buffer.transform(desired, reference_frame);
|
|
||||||
setpoint_yaw = tf2::getYaw(desired.quaternion);
|
|
||||||
yaw_frame_id = reference_frame;
|
|
||||||
|
|
||||||
} else if (isinf(yaw) && yaw > 0) {
|
|
||||||
// yaw towards
|
// yaw towards
|
||||||
setpoint_yaw_type = TOWARDS;
|
setpoint_yaw_type = TOWARDS;
|
||||||
|
yaw = 0;
|
||||||
} else if (yaw_frame_id.empty() || sp_type == _YAW) {
|
setpoint_yaw_rate = 0;
|
||||||
// yaw is nan and not set previously OR set_yaw(yaw=nan) was called
|
} else {
|
||||||
setpoint_yaw_type = YAW;
|
setpoint_yaw_type = YAW;
|
||||||
setpoint_yaw = tf2::getYaw(local_position.pose.orientation); // set yaw to current yaw
|
setpoint_yaw_rate = 0;
|
||||||
yaw_frame_id = local_position.header.frame_id;
|
ps.pose.orientation = tf::createQuaternionMsgFromYaw(yaw);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
tf_buffer.transform(ps, setpoint_position, reference_frame);
|
||||||
}
|
}
|
||||||
|
|
||||||
// handle roll
|
if (sp_type == VELOCITY) {
|
||||||
if (isfinite(roll)) {
|
static Vector3Stamped vel;
|
||||||
setpoint_roll = roll;
|
vel.header.frame_id = frame_id;
|
||||||
|
vel.header.stamp = stamp;
|
||||||
|
vel.vector.x = vx;
|
||||||
|
vel.vector.y = vy;
|
||||||
|
vel.vector.z = vz;
|
||||||
|
tf_buffer.transform(vel, setpoint_velocity, reference_frame);
|
||||||
}
|
}
|
||||||
|
|
||||||
// handle pitch
|
if (sp_type == ATTITUDE || sp_type == RATES) {
|
||||||
if (isfinite(pitch)) {
|
thrust_msg.thrust = thrust;
|
||||||
setpoint_pitch = pitch;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// handle yaw rate
|
if (sp_type == RATES) {
|
||||||
if (isfinite(yaw_rate)) {
|
rates_msg.twist.angular.x = roll_rate;
|
||||||
setpoint_yaw_type = YAW_RATE;
|
rates_msg.twist.angular.y = pitch_rate;
|
||||||
setpoint_rates.z = yaw_rate;
|
rates_msg.twist.angular.z = yaw_rate;
|
||||||
}
|
|
||||||
|
|
||||||
// handle pitch rate
|
|
||||||
if (isfinite(roll_rate)) {
|
|
||||||
setpoint_rates.x = roll_rate;
|
|
||||||
}
|
|
||||||
|
|
||||||
// handle roll rate
|
|
||||||
if (isfinite(pitch_rate)) {
|
|
||||||
setpoint_rates.y = pitch_rate;
|
|
||||||
}
|
|
||||||
|
|
||||||
// handle thrust
|
|
||||||
if (isfinite(thrust)) {
|
|
||||||
setpoint_thrust = thrust;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
wait_armed = auto_arm;
|
wait_armed = auto_arm;
|
||||||
|
|
||||||
|
publish_setpoint:
|
||||||
publish(stamp); // calculate initial transformed messages first
|
publish(stamp); // calculate initial transformed messages first
|
||||||
setpoint_timer.start();
|
setpoint_timer.start();
|
||||||
|
|
||||||
if (setpoint_type == NAVIGATE || setpoint_type == NAVIGATE_GLOBAL || setpoint_type == POSITION) {
|
// publish target frame
|
||||||
publishTarget(stamp, true);
|
if (!target.child_frame_id.empty()) {
|
||||||
|
if (setpoint_type == NAVIGATE || setpoint_type == NAVIGATE_GLOBAL || setpoint_type == POSITION) {
|
||||||
|
target.header.frame_id = setpoint_position.header.frame_id;
|
||||||
|
target.header.stamp = stamp;
|
||||||
|
target.transform.translation.x = setpoint_position.pose.position.x;
|
||||||
|
target.transform.translation.y = setpoint_position.pose.position.y;
|
||||||
|
target.transform.translation.z = setpoint_position.pose.position.z;
|
||||||
|
target.transform.rotation = setpoint_position.pose.orientation;
|
||||||
|
static_transform_broadcaster->sendTransform(target);
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
publishState();
|
|
||||||
|
|
||||||
if (auto_arm) {
|
if (auto_arm) {
|
||||||
offboardAndArm();
|
offboardAndArm();
|
||||||
wait_armed = false;
|
wait_armed = false;
|
||||||
@@ -972,39 +774,27 @@ bool serve(enum setpoint_type_t sp_type, float x, float y, float z, float vx, fl
|
|||||||
}
|
}
|
||||||
|
|
||||||
bool navigate(Navigate::Request& req, Navigate::Response& res) {
|
bool navigate(Navigate::Request& req, Navigate::Response& res) {
|
||||||
return serve(NAVIGATE, req.x, req.y, req.z, NAN, NAN, NAN, NAN, NAN, req.yaw, NAN, NAN, NAN, NAN, NAN, NAN, req.speed, req.frame_id, req.auto_arm, res.success, res.message);
|
return serve(NAVIGATE, req.x, req.y, req.z, NAN, NAN, NAN, NAN, NAN, req.yaw, NAN, NAN, req.yaw_rate, NAN, NAN, NAN, req.speed, req.frame_id, req.auto_arm, res.success, res.message);
|
||||||
}
|
}
|
||||||
|
|
||||||
bool navigateGlobal(NavigateGlobal::Request& req, NavigateGlobal::Response& res) {
|
bool navigateGlobal(NavigateGlobal::Request& req, NavigateGlobal::Response& res) {
|
||||||
return serve(NAVIGATE_GLOBAL, NAN, NAN, req.z, NAN, NAN, NAN, NAN, NAN, req.yaw, NAN, NAN, NAN, req.lat, req.lon, NAN, req.speed, req.frame_id, req.auto_arm, res.success, res.message);
|
return serve(NAVIGATE_GLOBAL, NAN, NAN, req.z, NAN, NAN, NAN, NAN, NAN, req.yaw, NAN, NAN, req.yaw_rate, req.lat, req.lon, NAN, req.speed, req.frame_id, req.auto_arm, res.success, res.message);
|
||||||
}
|
|
||||||
|
|
||||||
bool setAltitude(SetAltitude::Request& req, SetAltitude::Response& res) {
|
|
||||||
return serve(_ALTITUDE, NAN, NAN, req.z, NAN, NAN, NAN, NAN, NAN, NAN, NAN, NAN, NAN, NAN, NAN, NAN, NAN, req.frame_id, false, res.success, res.message);
|
|
||||||
}
|
|
||||||
|
|
||||||
bool setYaw(SetYaw::Request& req, SetYaw::Response& res) {
|
|
||||||
return serve(_YAW, NAN, NAN, NAN, NAN, NAN, NAN, NAN, NAN, req.yaw, NAN, NAN, NAN, NAN, NAN, NAN, NAN, req.frame_id, false, res.success, res.message);
|
|
||||||
}
|
|
||||||
|
|
||||||
bool setYawRate(SetYawRate::Request& req, SetYawRate::Response& res) {
|
|
||||||
return serve(_YAW_RATE, NAN, NAN, NAN, NAN, NAN, NAN, NAN, NAN, NAN, NAN, NAN, req.yaw_rate, NAN, NAN, NAN, NAN, "", false, res.success, res.message);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
bool setPosition(SetPosition::Request& req, SetPosition::Response& res) {
|
bool setPosition(SetPosition::Request& req, SetPosition::Response& res) {
|
||||||
return serve(POSITION, req.x, req.y, req.z, NAN, NAN, NAN, NAN, NAN, req.yaw, NAN, NAN, NAN, NAN, NAN, NAN, NAN, req.frame_id, req.auto_arm, res.success, res.message);
|
return serve(POSITION, req.x, req.y, req.z, NAN, NAN, NAN, NAN, NAN, req.yaw, NAN, NAN, req.yaw_rate, NAN, NAN, NAN, NAN, req.frame_id, req.auto_arm, res.success, res.message);
|
||||||
}
|
}
|
||||||
|
|
||||||
bool setVelocity(SetVelocity::Request& req, SetVelocity::Response& res) {
|
bool setVelocity(SetVelocity::Request& req, SetVelocity::Response& res) {
|
||||||
return serve(VELOCITY, NAN, NAN, NAN, req.vx, req.vy, req.vz, NAN, NAN, req.yaw, NAN, NAN, NAN, NAN, NAN, NAN, NAN, req.frame_id, req.auto_arm, res.success, res.message);
|
return serve(VELOCITY, NAN, NAN, NAN, req.vx, req.vy, req.vz, NAN, NAN, req.yaw, NAN, NAN, req.yaw_rate, NAN, NAN, NAN, NAN, req.frame_id, req.auto_arm, res.success, res.message);
|
||||||
}
|
}
|
||||||
|
|
||||||
bool setAttitude(SetAttitude::Request& req, SetAttitude::Response& res) {
|
bool setAttitude(SetAttitude::Request& req, SetAttitude::Response& res) {
|
||||||
return serve(ATTITUDE, NAN, NAN, NAN, NAN, NAN, NAN, req.roll, req.pitch, req.yaw, NAN, NAN, NAN, NAN, NAN, req.thrust, NAN, req.frame_id, req.auto_arm, res.success, res.message);
|
return serve(ATTITUDE, NAN, NAN, NAN, NAN, NAN, NAN, req.pitch, req.roll, req.yaw, NAN, NAN, NAN, NAN, NAN, req.thrust, NAN, req.frame_id, req.auto_arm, res.success, res.message);
|
||||||
}
|
}
|
||||||
|
|
||||||
bool setRates(SetRates::Request& req, SetRates::Response& res) {
|
bool setRates(SetRates::Request& req, SetRates::Response& res) {
|
||||||
return serve(RATES, NAN, NAN, NAN, NAN, NAN, NAN, NAN, NAN, NAN, req.roll_rate, req.pitch_rate, req.yaw_rate, NAN, NAN, req.thrust, NAN, "", req.auto_arm, res.success, res.message);
|
return serve(RATES, NAN, NAN, NAN, NAN, NAN, NAN, NAN, NAN, NAN, req.pitch_rate, req.roll_rate, req.yaw_rate, NAN, NAN, req.thrust, NAN, "", req.auto_arm, res.success, res.message);
|
||||||
}
|
}
|
||||||
|
|
||||||
bool land(std_srvs::Trigger::Request& req, std_srvs::Trigger::Response& res)
|
bool land(std_srvs::Trigger::Request& req, std_srvs::Trigger::Response& res)
|
||||||
@@ -1036,7 +826,9 @@ bool land(std_srvs::Trigger::Request& req, std_srvs::Trigger::Response& res)
|
|||||||
auto start = ros::Time::now();
|
auto start = ros::Time::now();
|
||||||
while (ros::ok()) {
|
while (ros::ok()) {
|
||||||
if (state.mode == "AUTO.LAND") {
|
if (state.mode == "AUTO.LAND") {
|
||||||
break;
|
res.success = true;
|
||||||
|
busy = false;
|
||||||
|
return true;
|
||||||
}
|
}
|
||||||
if (ros::Time::now() - start > land_timeout)
|
if (ros::Time::now() - start > land_timeout)
|
||||||
throw std::runtime_error("Land request timed out");
|
throw std::runtime_error("Land request timed out");
|
||||||
@@ -1045,37 +837,12 @@ bool land(std_srvs::Trigger::Request& req, std_srvs::Trigger::Response& res)
|
|||||||
r.sleep();
|
r.sleep();
|
||||||
}
|
}
|
||||||
|
|
||||||
// stop setpoints and invalidate position setpoint
|
|
||||||
setpoint_timer.stop();
|
|
||||||
setpoint_type = NONE;
|
|
||||||
setpoint_position.header.frame_id = "";
|
|
||||||
setpoint_altitude.header.frame_id = "";
|
|
||||||
yaw_frame_id = "";
|
|
||||||
publishState();
|
|
||||||
|
|
||||||
res.success = true;
|
|
||||||
busy = false;
|
|
||||||
return true;
|
|
||||||
|
|
||||||
} catch (const std::exception& e) {
|
} catch (const std::exception& e) {
|
||||||
res.message = e.what();
|
res.message = e.what();
|
||||||
ROS_INFO("%s", e.what());
|
ROS_INFO("%s", e.what());
|
||||||
busy = false;
|
busy = false;
|
||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
return false;
|
|
||||||
}
|
|
||||||
|
|
||||||
bool release(std_srvs::Trigger::Request& req, std_srvs::Trigger::Response& res)
|
|
||||||
{
|
|
||||||
setpoint_timer.stop();
|
|
||||||
setpoint_type = NONE;
|
|
||||||
setpoint_position.header.frame_id = "";
|
|
||||||
setpoint_altitude.header.frame_id = "";
|
|
||||||
yaw_frame_id = "";
|
|
||||||
publishState();
|
|
||||||
res.success = true;
|
|
||||||
return true;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
int main(int argc, char **argv)
|
int main(int argc, char **argv)
|
||||||
@@ -1088,9 +855,8 @@ int main(int argc, char **argv)
|
|||||||
static_transform_broadcaster = std::make_shared<tf2_ros::StaticTransformBroadcaster>();
|
static_transform_broadcaster = std::make_shared<tf2_ros::StaticTransformBroadcaster>();
|
||||||
|
|
||||||
// Params
|
// Params
|
||||||
nh_priv.param("mavros", mavros, string("mavros")); // for case of using multiple connections
|
nh.param<string>("mavros/local_position/tf/frame_id", local_frame, "map");
|
||||||
nh.param<string>(mavros + "/local_position/tf/frame_id", local_frame, "map");
|
nh.param<string>("mavros/local_position/tf/child_frame_id", fcu_frame, "base_link");
|
||||||
nh.param<string>(mavros + "/local_position/tf/child_frame_id", fcu_frame, "base_link");
|
|
||||||
nh_priv.param("target_frame", target.child_frame_id, string("navigate_target"));
|
nh_priv.param("target_frame", target.child_frame_id, string("navigate_target"));
|
||||||
nh_priv.param("setpoint", setpoint.child_frame_id, string("setpoint"));
|
nh_priv.param("setpoint", setpoint.child_frame_id, string("setpoint"));
|
||||||
nh_priv.param("auto_release", auto_release, true);
|
nh_priv.param("auto_release", auto_release, true);
|
||||||
@@ -1099,16 +865,8 @@ int main(int argc, char **argv)
|
|||||||
nh_priv.param("check_kill_switch", check_kill_switch, true);
|
nh_priv.param("check_kill_switch", check_kill_switch, true);
|
||||||
nh_priv.param("default_speed", default_speed, 0.5f);
|
nh_priv.param("default_speed", default_speed, 0.5f);
|
||||||
nh_priv.param<string>("body_frame", body.child_frame_id, "body");
|
nh_priv.param<string>("body_frame", body.child_frame_id, "body");
|
||||||
nh_priv.param<string>("terrain_frame", terrain.child_frame_id, "terrain");
|
|
||||||
nh_priv.getParam("reference_frames", reference_frames);
|
nh_priv.getParam("reference_frames", reference_frames);
|
||||||
|
|
||||||
// Default reference frames
|
|
||||||
std::map<string, string> default_reference_frames;
|
|
||||||
default_reference_frames[body.child_frame_id] = local_frame;
|
|
||||||
default_reference_frames[fcu_frame] = local_frame;
|
|
||||||
if (!target.child_frame_id.empty()) default_reference_frames[target.child_frame_id] = local_frame;
|
|
||||||
reference_frames.insert(default_reference_frames.begin(), default_reference_frames.end()); // merge defaults
|
|
||||||
|
|
||||||
state_timeout = ros::Duration(nh_priv.param("state_timeout", 3.0));
|
state_timeout = ros::Duration(nh_priv.param("state_timeout", 3.0));
|
||||||
local_position_timeout = ros::Duration(nh_priv.param("local_position_timeout", 2.0));
|
local_position_timeout = ros::Duration(nh_priv.param("local_position_timeout", 2.0));
|
||||||
velocity_timeout = ros::Duration(nh_priv.param("velocity_timeout", 2.0));
|
velocity_timeout = ros::Duration(nh_priv.param("velocity_timeout", 2.0));
|
||||||
@@ -1123,48 +881,35 @@ int main(int argc, char **argv)
|
|||||||
arming_timeout = ros::Duration(nh_priv.param("arming_timeout", 4.0));
|
arming_timeout = ros::Duration(nh_priv.param("arming_timeout", 4.0));
|
||||||
|
|
||||||
// Service clients
|
// Service clients
|
||||||
arming = nh.serviceClient<mavros_msgs::CommandBool>(mavros + "/cmd/arming");
|
arming = nh.serviceClient<mavros_msgs::CommandBool>("mavros/cmd/arming");
|
||||||
set_mode = nh.serviceClient<mavros_msgs::SetMode>(mavros + "/set_mode");
|
set_mode = nh.serviceClient<mavros_msgs::SetMode>("mavros/set_mode");
|
||||||
|
|
||||||
// Telemetry subscribers
|
// Telemetry subscribers
|
||||||
auto state_sub = nh.subscribe(mavros + "/state", 1, &handleState);
|
auto state_sub = nh.subscribe("mavros/state", 1, &handleState);
|
||||||
auto velocity_sub = nh.subscribe(mavros + "/local_position/velocity_body", 1, &handleMessage<TwistStamped, velocity>);
|
auto velocity_sub = nh.subscribe("mavros/local_position/velocity_body", 1, &handleMessage<TwistStamped, velocity>);
|
||||||
auto global_position_sub = nh.subscribe(mavros + "/global_position/global", 1, &handleMessage<NavSatFix, global_position>);
|
auto global_position_sub = nh.subscribe("mavros/global_position/global", 1, &handleMessage<NavSatFix, global_position>);
|
||||||
auto battery_sub = nh.subscribe(mavros + "/battery", 1, &handleMessage<BatteryState, battery>);
|
auto battery_sub = nh.subscribe("mavros/battery", 1, &handleMessage<BatteryState, battery>);
|
||||||
auto statustext_sub = nh.subscribe(mavros + "/statustext/recv", 1, &handleMessage<mavros_msgs::StatusText, statustext>);
|
auto statustext_sub = nh.subscribe("mavros/statustext/recv", 1, &handleMessage<mavros_msgs::StatusText, statustext>);
|
||||||
auto manual_control_sub = nh.subscribe(mavros + "/manual_control/control", 1, &handleMessage<mavros_msgs::ManualControl, manual_control>);
|
auto manual_control_sub = nh.subscribe("mavros/manual_control/control", 1, &handleMessage<mavros_msgs::ManualControl, manual_control>);
|
||||||
auto local_position_sub = nh.subscribe(mavros + "/local_position/pose", 1, &handleLocalPosition);
|
auto local_position_sub = nh.subscribe("mavros/local_position/pose", 1, &handleLocalPosition);
|
||||||
|
|
||||||
ros::Subscriber altitude_sub;
|
|
||||||
if (!body.child_frame_id.empty() && !terrain.child_frame_id.empty()) {
|
|
||||||
terrain.header.frame_id = local_frame;
|
|
||||||
altitude_sub = nh.subscribe(mavros + "/altitude", 1, &handleAltitude);
|
|
||||||
}
|
|
||||||
|
|
||||||
// Setpoint publishers
|
// Setpoint publishers
|
||||||
position_pub = nh.advertise<PoseStamped>(mavros + "/setpoint_position/local", 1);
|
position_pub = nh.advertise<PoseStamped>("mavros/setpoint_position/local", 1);
|
||||||
position_raw_pub = nh.advertise<PositionTarget>(mavros + "/setpoint_raw/local", 1);
|
position_raw_pub = nh.advertise<PositionTarget>("mavros/setpoint_raw/local", 1);
|
||||||
attitude_pub = nh.advertise<PoseStamped>(mavros + "/setpoint_attitude/attitude", 1);
|
attitude_pub = nh.advertise<PoseStamped>("mavros/setpoint_attitude/attitude", 1);
|
||||||
attitude_raw_pub = nh.advertise<AttitudeTarget>(mavros + "/setpoint_raw/attitude", 1);
|
attitude_raw_pub = nh.advertise<AttitudeTarget>("mavros/setpoint_raw/attitude", 1);
|
||||||
rates_pub = nh.advertise<TwistStamped>(mavros + "/setpoint_attitude/cmd_vel", 1);
|
rates_pub = nh.advertise<TwistStamped>("mavros/setpoint_attitude/cmd_vel", 1);
|
||||||
thrust_pub = nh.advertise<Thrust>(mavros + "/setpoint_attitude/thrust", 1);
|
thrust_pub = nh.advertise<Thrust>("mavros/setpoint_attitude/thrust", 1);
|
||||||
|
|
||||||
// State publisher
|
|
||||||
state_pub = nh_priv.advertise<clover::State>("state", 1, true);
|
|
||||||
|
|
||||||
// Service servers
|
// Service servers
|
||||||
auto gt_serv = nh.advertiseService("get_telemetry", &getTelemetry);
|
auto gt_serv = nh.advertiseService("get_telemetry", &getTelemetry);
|
||||||
auto na_serv = nh.advertiseService("navigate", &navigate);
|
auto na_serv = nh.advertiseService("navigate", &navigate);
|
||||||
auto ng_serv = nh.advertiseService("navigate_global", &navigateGlobal);
|
auto ng_serv = nh.advertiseService("navigate_global", &navigateGlobal);
|
||||||
auto sl_serv = nh.advertiseService("set_altitude", &setAltitude);
|
|
||||||
auto ya_serv = nh.advertiseService("set_yaw", &setYaw);
|
|
||||||
auto yr_serv = nh.advertiseService("set_yaw_rate", &setYawRate);
|
|
||||||
auto sp_serv = nh.advertiseService("set_position", &setPosition);
|
auto sp_serv = nh.advertiseService("set_position", &setPosition);
|
||||||
auto sv_serv = nh.advertiseService("set_velocity", &setVelocity);
|
auto sv_serv = nh.advertiseService("set_velocity", &setVelocity);
|
||||||
auto sa_serv = nh.advertiseService("set_attitude", &setAttitude);
|
auto sa_serv = nh.advertiseService("set_attitude", &setAttitude);
|
||||||
auto sr_serv = nh.advertiseService("set_rates", &setRates);
|
auto sr_serv = nh.advertiseService("set_rates", &setRates);
|
||||||
auto ld_serv = nh.advertiseService("land", &land);
|
auto ld_serv = nh.advertiseService("land", &land);
|
||||||
auto rl_serv = nh_priv.advertiseService("release", &release);
|
|
||||||
|
|
||||||
// Setpoint timer
|
// Setpoint timer
|
||||||
setpoint_timer = nh.createTimer(ros::Duration(1 / nh_priv.param("setpoint_rate", 30.0)), &publishSetpoint, false, false);
|
setpoint_timer = nh.createTimer(ros::Duration(1 / nh_priv.param("setpoint_rate", 30.0)), &publishSetpoint, false, false);
|
||||||
@@ -1172,7 +917,7 @@ int main(int argc, char **argv)
|
|||||||
position_msg.header.frame_id = local_frame;
|
position_msg.header.frame_id = local_frame;
|
||||||
position_raw_msg.header.frame_id = local_frame;
|
position_raw_msg.header.frame_id = local_frame;
|
||||||
position_raw_msg.coordinate_frame = PositionTarget::FRAME_LOCAL_NED;
|
position_raw_msg.coordinate_frame = PositionTarget::FRAME_LOCAL_NED;
|
||||||
//rates_msg.header.frame_id = fcu_frame;
|
rates_msg.header.frame_id = fcu_frame;
|
||||||
|
|
||||||
ROS_INFO("ready");
|
ROS_INFO("ready");
|
||||||
ros::spin();
|
ros::spin();
|
||||||
|
|||||||
@@ -25,7 +25,7 @@
|
|||||||
using std::string;
|
using std::string;
|
||||||
using namespace geometry_msgs;
|
using namespace geometry_msgs;
|
||||||
|
|
||||||
bool reset_flag = true; // offset should be reset on the start
|
bool reset_flag = false;
|
||||||
string local_frame_id, frame_id, child_frame_id, offset_frame_id;
|
string local_frame_id, frame_id, child_frame_id, offset_frame_id;
|
||||||
tf2_ros::Buffer tf_buffer;
|
tf2_ros::Buffer tf_buffer;
|
||||||
ros::Publisher vpe_pub;
|
ros::Publisher vpe_pub;
|
||||||
@@ -33,14 +33,14 @@ ros::Subscriber local_position_sub;
|
|||||||
ros::Timer zero_timer;
|
ros::Timer zero_timer;
|
||||||
PoseStamped vpe, pose;
|
PoseStamped vpe, pose;
|
||||||
ros::Time got_local_pos(0);
|
ros::Time got_local_pos(0);
|
||||||
ros::Duration publish_zero_timeout, publish_zero_duration, offset_timeout;
|
ros::Duration publish_zero_timout, publish_zero_duration, offset_timeout;
|
||||||
TransformStamped offset;
|
TransformStamped offset;
|
||||||
|
|
||||||
void publishZero(const ros::TimerEvent& e)
|
void publishZero(const ros::TimerEvent& e)
|
||||||
{
|
{
|
||||||
if (!vpe.header.stamp.isZero() && e.current_real - vpe.header.stamp < publish_zero_timeout) return; // have vpe
|
if (e.current_real - vpe.header.stamp < publish_zero_timout) return; // have vpe
|
||||||
|
|
||||||
if (!pose.header.stamp.isZero() && e.current_real - pose.header.stamp < publish_zero_timeout) { // have local position
|
if (e.current_real - pose.header.stamp < publish_zero_timout) { // have local position
|
||||||
if (got_local_pos.isZero()) {
|
if (got_local_pos.isZero()) {
|
||||||
ROS_INFO("got local position");
|
ROS_INFO("got local position");
|
||||||
got_local_pos = e.current_real;
|
got_local_pos = e.current_real;
|
||||||
@@ -53,7 +53,7 @@ void publishZero(const ros::TimerEvent& e)
|
|||||||
}
|
}
|
||||||
|
|
||||||
ROS_INFO_THROTTLE(10, "publish zero");
|
ROS_INFO_THROTTLE(10, "publish zero");
|
||||||
geometry_msgs::PoseStamped zero;
|
static geometry_msgs::PoseStamped zero;
|
||||||
zero.header.frame_id = local_frame_id;
|
zero.header.frame_id = local_frame_id;
|
||||||
zero.header.stamp = e.current_real;
|
zero.header.stamp = e.current_real;
|
||||||
zero.pose.orientation.w = 1;
|
zero.pose.orientation.w = 1;
|
||||||
@@ -124,8 +124,8 @@ int main(int argc, char **argv) {
|
|||||||
|
|
||||||
nh_priv.param<string>("frame_id", frame_id, "");
|
nh_priv.param<string>("frame_id", frame_id, "");
|
||||||
nh_priv.param<string>("offset_frame_id", offset_frame_id, "");
|
nh_priv.param<string>("offset_frame_id", offset_frame_id, "");
|
||||||
nh.param<string>("mavros/local_position/frame_id", local_frame_id, "map");
|
nh_priv.param<string>("mavros/local_position/frame_id", local_frame_id, "map");
|
||||||
nh.param<string>("mavros/local_position/tf/child_frame_id", child_frame_id, "base_link");
|
nh_priv.param<string>("mavros/local_position/tf/child_frame_id", child_frame_id, "base_link");
|
||||||
offset_timeout = ros::Duration(nh_priv.param("offset_timeout", 3.0));
|
offset_timeout = ros::Duration(nh_priv.param("offset_timeout", 3.0));
|
||||||
|
|
||||||
if (!frame_id.empty()) {
|
if (!frame_id.empty()) {
|
||||||
@@ -141,11 +141,11 @@ int main(int argc, char **argv) {
|
|||||||
vpe_pub = nh_priv.advertise<PoseStamped>("vpe", 1);
|
vpe_pub = nh_priv.advertise<PoseStamped>("vpe", 1);
|
||||||
//vpe_cov_pub = nh_priv_.advertise<PoseStamped>("pose_cov_pub", 1);
|
//vpe_cov_pub = nh_priv_.advertise<PoseStamped>("pose_cov_pub", 1);
|
||||||
|
|
||||||
if (nh_priv.param("force_init", false) || nh_priv.param("publish_zero", false)) { // publish_zero is old name
|
if (nh_priv.param("publish_zero", false)) {
|
||||||
// publish zero to initialize the local position
|
// publish zero to initialize the local position
|
||||||
zero_timer = nh.createTimer(ros::Duration(0.1), &publishZero);
|
zero_timer = nh.createTimer(ros::Duration(0.1), &publishZero);
|
||||||
publish_zero_timeout = ros::Duration(nh_priv.param("force_init_timeout", 5.0));
|
publish_zero_timout = ros::Duration(nh_priv.param("publish_zero_timout", 5.0));
|
||||||
publish_zero_duration = ros::Duration(nh_priv.param("force_init_duration", 5.0));
|
publish_zero_duration = ros::Duration(nh_priv.param("publish_zero_duration", 5.0));
|
||||||
local_position_sub = nh.subscribe("mavros/local_position/pose", 1, &localPositionCallback);
|
local_position_sub = nh.subscribe("mavros/local_position/pose", 1, &localPositionCallback);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -1,9 +0,0 @@
|
|||||||
#!/usr/bin/env bash
|
|
||||||
|
|
||||||
# $ ./waitfile <file> <command> <args...>
|
|
||||||
# wait until <file> appears and then invoke <command> with <args>
|
|
||||||
|
|
||||||
echo "wait for file $1"
|
|
||||||
while [ ! -e "$1" ]; do sleep 1; done;
|
|
||||||
echo "file $1 appeared"
|
|
||||||
exec "${@:2}"
|
|
||||||
@@ -1,4 +0,0 @@
|
|||||||
#!/usr/bin/env bash
|
|
||||||
|
|
||||||
export ROSWWW_DEFAULT=clover
|
|
||||||
rosrun roswww_static update
|
|
||||||
@@ -13,11 +13,11 @@ float32 alt
|
|||||||
float32 vx
|
float32 vx
|
||||||
float32 vy
|
float32 vy
|
||||||
float32 vz
|
float32 vz
|
||||||
float32 roll
|
|
||||||
float32 pitch
|
float32 pitch
|
||||||
|
float32 roll
|
||||||
float32 yaw
|
float32 yaw
|
||||||
float32 roll_rate
|
|
||||||
float32 pitch_rate
|
float32 pitch_rate
|
||||||
|
float32 roll_rate
|
||||||
float32 yaw_rate
|
float32 yaw_rate
|
||||||
float32 voltage
|
float32 voltage
|
||||||
float32 cell_voltage
|
float32 cell_voltage
|
||||||
|
|||||||
@@ -2,6 +2,7 @@ float32 x
|
|||||||
float32 y
|
float32 y
|
||||||
float32 z
|
float32 z
|
||||||
float32 yaw
|
float32 yaw
|
||||||
|
float32 yaw_rate
|
||||||
float32 speed
|
float32 speed
|
||||||
string frame_id
|
string frame_id
|
||||||
bool auto_arm
|
bool auto_arm
|
||||||
|
|||||||
@@ -2,6 +2,7 @@ float64 lat
|
|||||||
float64 lon
|
float64 lon
|
||||||
float32 z
|
float32 z
|
||||||
float32 yaw
|
float32 yaw
|
||||||
|
float32 yaw_rate
|
||||||
float32 speed
|
float32 speed
|
||||||
string frame_id
|
string frame_id
|
||||||
bool auto_arm
|
bool auto_arm
|
||||||
|
|||||||
@@ -1,5 +0,0 @@
|
|||||||
float32 z
|
|
||||||
string frame_id
|
|
||||||
---
|
|
||||||
bool success
|
|
||||||
string message
|
|
||||||
@@ -1,5 +1,5 @@
|
|||||||
float32 roll
|
|
||||||
float32 pitch
|
float32 pitch
|
||||||
|
float32 roll
|
||||||
float32 yaw
|
float32 yaw
|
||||||
float32 thrust
|
float32 thrust
|
||||||
string frame_id
|
string frame_id
|
||||||
|
|||||||
@@ -2,6 +2,7 @@ float32 x
|
|||||||
float32 y
|
float32 y
|
||||||
float32 z
|
float32 z
|
||||||
float32 yaw
|
float32 yaw
|
||||||
|
float32 yaw_rate
|
||||||
string frame_id
|
string frame_id
|
||||||
bool auto_arm
|
bool auto_arm
|
||||||
---
|
---
|
||||||
|
|||||||
@@ -1,5 +1,5 @@
|
|||||||
float32 roll_rate
|
|
||||||
float32 pitch_rate
|
float32 pitch_rate
|
||||||
|
float32 roll_rate
|
||||||
float32 yaw_rate
|
float32 yaw_rate
|
||||||
float32 thrust
|
float32 thrust
|
||||||
bool auto_arm
|
bool auto_arm
|
||||||
|
|||||||
@@ -2,6 +2,7 @@ float32 vx
|
|||||||
float32 vy
|
float32 vy
|
||||||
float32 vz
|
float32 vz
|
||||||
float32 yaw
|
float32 yaw
|
||||||
|
float32 yaw_rate
|
||||||
string frame_id
|
string frame_id
|
||||||
bool auto_arm
|
bool auto_arm
|
||||||
---
|
---
|
||||||
|
|||||||
@@ -1,5 +0,0 @@
|
|||||||
float32 yaw
|
|
||||||
string frame_id
|
|
||||||
---
|
|
||||||
bool success
|
|
||||||
string message
|
|
||||||
@@ -1,4 +0,0 @@
|
|||||||
float32 yaw_rate
|
|
||||||
---
|
|
||||||
bool success
|
|
||||||
string message
|
|
||||||
@@ -3,7 +3,6 @@ import rospy
|
|||||||
import pytest
|
import pytest
|
||||||
from mavros_msgs.msg import State
|
from mavros_msgs.msg import State
|
||||||
from clover import srv
|
from clover import srv
|
||||||
import time
|
|
||||||
|
|
||||||
@pytest.fixture()
|
@pytest.fixture()
|
||||||
def node():
|
def node():
|
||||||
@@ -25,54 +24,23 @@ def test_simple_offboard_services_available():
|
|||||||
rospy.wait_for_service('set_attitude', timeout=5)
|
rospy.wait_for_service('set_attitude', timeout=5)
|
||||||
rospy.wait_for_service('set_rates', timeout=5)
|
rospy.wait_for_service('set_rates', timeout=5)
|
||||||
rospy.wait_for_service('land', timeout=5)
|
rospy.wait_for_service('land', timeout=5)
|
||||||
rospy.wait_for_service('simple_offboard/release', timeout=5)
|
|
||||||
|
|
||||||
def test_web_video_server(node):
|
def test_web_video_server(node):
|
||||||
try:
|
import urllib2
|
||||||
# Python 2
|
urllib2.urlopen("http://localhost:8080").read()
|
||||||
import urllib2 as urllib
|
|
||||||
except ModuleNotFoundError:
|
|
||||||
# Python 3
|
|
||||||
import urllib.request as urllib
|
|
||||||
urllib.urlopen("http://localhost:8080").read()
|
|
||||||
|
|
||||||
def test_blocks(node):
|
def test_shell(node):
|
||||||
rospy.wait_for_service('clover_blocks/run', timeout=5)
|
execute = rospy.ServiceProxy('exec', srv.Execute)
|
||||||
rospy.wait_for_service('clover_blocks/stop', timeout=5)
|
execute.wait_for_service(5)
|
||||||
rospy.wait_for_service('clover_blocks/load', timeout=5)
|
|
||||||
rospy.wait_for_service('clover_blocks/store', timeout=5)
|
|
||||||
|
|
||||||
from std_msgs.msg import String
|
res = execute(cmd='echo foo')
|
||||||
from clover_blocks.srv import Run
|
assert res.code == 0
|
||||||
|
assert res.output == 'foo\n'
|
||||||
|
|
||||||
def wait_print():
|
res = execute(cmd='foo')
|
||||||
try:
|
assert res.code == 32512
|
||||||
wait_print.result = rospy.wait_for_message('clover_blocks/print', String, timeout=5).data
|
assert res.output == ''
|
||||||
except Exception as e:
|
|
||||||
wait_print.result = str(e)
|
|
||||||
|
|
||||||
import threading
|
res = execute(cmd='ls foo')
|
||||||
t = threading.Thread(target=wait_print)
|
assert res.code == 512
|
||||||
t.start()
|
assert res.output == ''
|
||||||
rospy.sleep(0.1)
|
|
||||||
|
|
||||||
run = rospy.ServiceProxy('clover_blocks/run', Run)
|
|
||||||
assert run(code='print("test")').success == True
|
|
||||||
|
|
||||||
t.join()
|
|
||||||
assert wait_print.result == 'test'
|
|
||||||
|
|
||||||
def test_long_callback():
|
|
||||||
from clover import long_callback
|
|
||||||
from time import sleep
|
|
||||||
|
|
||||||
# very basic test for long_callback
|
|
||||||
@long_callback
|
|
||||||
def cb(i):
|
|
||||||
cb.counter += i
|
|
||||||
cb.counter = 0
|
|
||||||
cb(2)
|
|
||||||
sleep(0.1)
|
|
||||||
cb(3)
|
|
||||||
sleep(1)
|
|
||||||
assert cb.counter == 5
|
|
||||||
|
|||||||
@@ -23,7 +23,10 @@
|
|||||||
|
|
||||||
<node pkg="tf2_ros" type="static_transform_publisher" name="map_flipped_frame" args="0 0 0 3.1415926 3.1415926 0 map map_flipped" required="true"/>
|
<node pkg="tf2_ros" type="static_transform_publisher" name="map_flipped_frame" args="0 0 0 3.1415926 3.1415926 0 map map_flipped" required="true"/>
|
||||||
|
|
||||||
<node name="simple_offboard" pkg="clover" type="simple_offboard" required="true" output="screen"/>
|
<node name="simple_offboard" pkg="clover" type="simple_offboard" required="true" output="screen">
|
||||||
|
<param name="reference_frames/body" value="map"/>
|
||||||
|
<param name="reference_frames/base_link" value="map"/>
|
||||||
|
</node>
|
||||||
|
|
||||||
<node name="tf2_web_republisher" pkg="tf2_web_republisher" type="tf2_web_republisher" required="true"/>
|
<node name="tf2_web_republisher" pkg="tf2_web_republisher" type="tf2_web_republisher" required="true"/>
|
||||||
|
|
||||||
@@ -35,11 +38,6 @@
|
|||||||
<rosparam param="notify">startup: { r: 255, g: 255, b: 255 }</rosparam>
|
<rosparam param="notify">startup: { r: 255, g: 255, b: 255 }</rosparam>
|
||||||
</node>
|
</node>
|
||||||
|
|
||||||
<node name="clover_blocks" pkg="clover_blocks" type="clover_blocks" output="screen" required="true"/>
|
|
||||||
|
|
||||||
<node pkg="topic_tools" name="main_camera_throttle" type="throttle" ns="main_camera"
|
|
||||||
args="messages image_raw 5.0 image_raw_throttled" required="true"/>
|
|
||||||
|
|
||||||
<param name="test_module" value="$(find clover)/test/basic.py"/>
|
<param name="test_module" value="$(find clover)/test/basic.py"/>
|
||||||
<test test-name="basic_test" pkg="ros_pytest" type="ros_pytest_runner"/>
|
<test test-name="basic_test" pkg="ros_pytest" type="ros_pytest_runner"/>
|
||||||
</launch>
|
</launch>
|
||||||
|
|||||||
@@ -1,402 +0,0 @@
|
|||||||
import rospy
|
|
||||||
import pytest
|
|
||||||
from pytest import approx
|
|
||||||
import threading
|
|
||||||
import mavros_msgs.msg
|
|
||||||
from geometry_msgs.msg import PoseStamped
|
|
||||||
from clover import srv
|
|
||||||
from clover.msg import State
|
|
||||||
from math import nan, inf
|
|
||||||
import tf2_ros
|
|
||||||
import tf2_geometry_msgs
|
|
||||||
|
|
||||||
@pytest.fixture()
|
|
||||||
def node():
|
|
||||||
return rospy.init_node('offboard_test', anonymous=True)
|
|
||||||
|
|
||||||
@pytest.fixture
|
|
||||||
def tf_buffer():
|
|
||||||
buf = tf2_ros.Buffer()
|
|
||||||
tf2_ros.TransformListener(buf)
|
|
||||||
return buf
|
|
||||||
|
|
||||||
def get_state():
|
|
||||||
return rospy.wait_for_message('/simple_offboard/state', State, timeout=1)
|
|
||||||
|
|
||||||
def get_navigate_target(tf_buffer):
|
|
||||||
target = tf_buffer.lookup_transform('map', 'navigate_target', rospy.get_rostime(), rospy.Duration(1))
|
|
||||||
assert target.child_frame_id == 'navigate_target'
|
|
||||||
return target
|
|
||||||
|
|
||||||
def test_offboard(node, tf_buffer):
|
|
||||||
navigate = rospy.ServiceProxy('navigate', srv.Navigate)
|
|
||||||
set_position = rospy.ServiceProxy('set_position', srv.SetPosition)
|
|
||||||
set_altitude = rospy.ServiceProxy('set_altitude', srv.SetAltitude)
|
|
||||||
set_yaw = rospy.ServiceProxy('set_yaw', srv.SetYaw)
|
|
||||||
set_yaw_rate = rospy.ServiceProxy('set_yaw_rate', srv.SetYawRate)
|
|
||||||
set_velocity = rospy.ServiceProxy('set_velocity', srv.SetVelocity)
|
|
||||||
set_attitude = rospy.ServiceProxy('set_attitude', srv.SetAttitude)
|
|
||||||
set_rates = rospy.ServiceProxy('set_rates', srv.SetRates)
|
|
||||||
get_telemetry = rospy.ServiceProxy('get_telemetry', srv.GetTelemetry)
|
|
||||||
res = navigate()
|
|
||||||
assert res.success == False
|
|
||||||
assert res.message.startswith('State timeout')
|
|
||||||
|
|
||||||
telem = get_telemetry()
|
|
||||||
assert telem.connected == False
|
|
||||||
|
|
||||||
state_pub = rospy.Publisher('/mavros/state', mavros_msgs.msg.State, latch=True, queue_size=1)
|
|
||||||
state_msg = mavros_msgs.msg.State(mode='OFFBOARD', armed=True)
|
|
||||||
|
|
||||||
def publish_state():
|
|
||||||
r = rospy.Rate(2)
|
|
||||||
while not rospy.is_shutdown():
|
|
||||||
state_msg.header.stamp = rospy.Time.now()
|
|
||||||
state_pub.publish(state_msg)
|
|
||||||
r.sleep()
|
|
||||||
|
|
||||||
# start publishing state
|
|
||||||
threading.Thread(target=publish_state, daemon=True).start()
|
|
||||||
rospy.sleep(0.5)
|
|
||||||
|
|
||||||
telem = get_telemetry()
|
|
||||||
assert telem.connected == False
|
|
||||||
|
|
||||||
res = navigate()
|
|
||||||
assert res.success == False
|
|
||||||
assert res.message.startswith('No connection to FCU')
|
|
||||||
|
|
||||||
state_msg.connected = True
|
|
||||||
rospy.sleep(1)
|
|
||||||
|
|
||||||
telem = get_telemetry()
|
|
||||||
assert telem.connected == True
|
|
||||||
|
|
||||||
res = navigate()
|
|
||||||
assert res.success == False
|
|
||||||
assert res.message.startswith('No local position')
|
|
||||||
|
|
||||||
local_position_pub = rospy.Publisher('/mavros/local_position/pose', PoseStamped, latch=True, queue_size=1)
|
|
||||||
local_position_msg = PoseStamped()
|
|
||||||
local_position_msg.header.frame_id = 'map'
|
|
||||||
local_position_msg.pose.position.x = 1
|
|
||||||
local_position_msg.pose.position.y = 2
|
|
||||||
local_position_msg.pose.position.z = 3
|
|
||||||
local_position_msg.pose.orientation.w = 1
|
|
||||||
|
|
||||||
def publish_local_position():
|
|
||||||
r = rospy.Rate(30)
|
|
||||||
while not rospy.is_shutdown():
|
|
||||||
local_position_msg.header.stamp = rospy.Time.now()
|
|
||||||
local_position_pub.publish(local_position_msg)
|
|
||||||
r.sleep()
|
|
||||||
|
|
||||||
# start publishing local position
|
|
||||||
threading.Thread(target=publish_local_position, daemon=True).start()
|
|
||||||
rospy.sleep(0.5)
|
|
||||||
|
|
||||||
# check body frame
|
|
||||||
body = tf_buffer.lookup_transform('map', 'body', rospy.get_rostime(), rospy.Duration(1))
|
|
||||||
assert body.child_frame_id == 'body'
|
|
||||||
assert body.transform.translation.x == approx(1)
|
|
||||||
assert body.transform.translation.y == approx(2)
|
|
||||||
assert body.transform.translation.z == approx(3)
|
|
||||||
|
|
||||||
res = navigate(x=3, y=2, z=1, frame_id='map')
|
|
||||||
assert res.success == True
|
|
||||||
state = get_state()
|
|
||||||
assert state.mode == State.MODE_NAVIGATE
|
|
||||||
assert state.yaw_mode == State.YAW_MODE_YAW
|
|
||||||
assert state.x == 3
|
|
||||||
assert state.y == 2
|
|
||||||
assert state.z == 1
|
|
||||||
assert state.yaw == 0
|
|
||||||
assert state.xy_frame_id == 'map'
|
|
||||||
assert state.z_frame_id == 'map'
|
|
||||||
assert state.yaw_frame_id == 'map'
|
|
||||||
target = get_navigate_target(tf_buffer)
|
|
||||||
assert target.header.frame_id == 'map'
|
|
||||||
assert target.transform.translation.x == approx(3)
|
|
||||||
assert target.transform.translation.y == approx(2)
|
|
||||||
assert target.transform.translation.z == approx(1)
|
|
||||||
assert target.transform.rotation.x == 0
|
|
||||||
assert target.transform.rotation.y == 0
|
|
||||||
assert target.transform.rotation.z == 0
|
|
||||||
assert target.transform.rotation.w == 1
|
|
||||||
|
|
||||||
# try to set only the y
|
|
||||||
res = navigate(x=nan, y=1, z=nan)
|
|
||||||
assert res.success == False
|
|
||||||
assert res.message.startswith('x and y can be set only together')
|
|
||||||
|
|
||||||
# set z in body frame
|
|
||||||
res = navigate(x=nan, y=nan, z=1, frame_id='body')
|
|
||||||
assert res.success == True
|
|
||||||
state = get_state()
|
|
||||||
assert state.mode == State.MODE_NAVIGATE
|
|
||||||
assert state.yaw_mode == State.YAW_MODE_YAW
|
|
||||||
assert state.x == 3
|
|
||||||
assert state.y == 2
|
|
||||||
assert state.z == 4
|
|
||||||
assert state.yaw == 0
|
|
||||||
assert state.xy_frame_id == 'map'
|
|
||||||
assert state.z_frame_id == 'map'
|
|
||||||
assert state.yaw_frame_id == 'map'
|
|
||||||
|
|
||||||
# set xy in test frame
|
|
||||||
res = navigate(x=1, y=2, z=nan, frame_id='test')
|
|
||||||
assert res.success == True
|
|
||||||
state = get_state()
|
|
||||||
assert state.mode == State.MODE_NAVIGATE
|
|
||||||
assert state.yaw_mode == State.YAW_MODE_YAW
|
|
||||||
assert state.x == 1
|
|
||||||
assert state.y == 2
|
|
||||||
assert state.z == 4
|
|
||||||
assert state.yaw == 0
|
|
||||||
assert state.xy_frame_id == 'test'
|
|
||||||
assert state.z_frame_id == 'map'
|
|
||||||
assert state.yaw_frame_id == 'test'
|
|
||||||
|
|
||||||
# auto_arm should invalidate the setpoint
|
|
||||||
res = navigate(x=nan, y=nan, z=1, frame_id='map', auto_arm=True)
|
|
||||||
assert res.success == True
|
|
||||||
state = get_state()
|
|
||||||
assert state.mode == State.MODE_NAVIGATE
|
|
||||||
assert state.yaw_mode == State.YAW_MODE_YAW
|
|
||||||
assert state.x == 1
|
|
||||||
assert state.y == 2
|
|
||||||
assert state.z == 1
|
|
||||||
assert state.yaw == 0
|
|
||||||
assert state.xy_frame_id == 'map'
|
|
||||||
assert state.z_frame_id == 'map'
|
|
||||||
assert state.yaw_frame_id == 'map'
|
|
||||||
|
|
||||||
# set_attitude should invalidate the setpoint
|
|
||||||
res = set_attitude()
|
|
||||||
assert res.success == True
|
|
||||||
|
|
||||||
res = navigate(x=5, y=6, z=nan, yaw=nan, frame_id='map')
|
|
||||||
assert res.success == True
|
|
||||||
state = get_state()
|
|
||||||
assert state.mode == State.MODE_NAVIGATE
|
|
||||||
assert state.yaw_mode == State.YAW_MODE_YAW
|
|
||||||
assert state.x == 5
|
|
||||||
assert state.y == 6
|
|
||||||
assert state.z == 3
|
|
||||||
assert state.yaw == 0
|
|
||||||
assert state.xy_frame_id == 'map'
|
|
||||||
assert state.z_frame_id == 'map'
|
|
||||||
assert state.yaw_frame_id == 'map'
|
|
||||||
|
|
||||||
# test set_altitude
|
|
||||||
res = set_altitude(z=7, frame_id='test')
|
|
||||||
assert res.success == True
|
|
||||||
state = get_state()
|
|
||||||
assert state.mode == State.MODE_NAVIGATE
|
|
||||||
assert state.yaw_mode == State.YAW_MODE_YAW
|
|
||||||
assert state.x == 5
|
|
||||||
assert state.y == 6
|
|
||||||
assert state.z == 7
|
|
||||||
assert state.yaw == 0
|
|
||||||
assert state.xy_frame_id == 'map'
|
|
||||||
assert state.z_frame_id == 'test'
|
|
||||||
assert state.yaw_frame_id == 'map'
|
|
||||||
|
|
||||||
# test set_yaw
|
|
||||||
res = set_yaw(yaw=0.5, frame_id='test2')
|
|
||||||
assert res.success == True
|
|
||||||
state = get_state()
|
|
||||||
assert state.mode == State.MODE_NAVIGATE
|
|
||||||
assert state.yaw_mode == State.YAW_MODE_YAW
|
|
||||||
assert state.x == 5
|
|
||||||
assert state.y == 6
|
|
||||||
assert state.z == 7
|
|
||||||
assert state.yaw == 0.5
|
|
||||||
assert state.xy_frame_id == 'map'
|
|
||||||
assert state.z_frame_id == 'test'
|
|
||||||
assert state.yaw_frame_id == 'test2'
|
|
||||||
|
|
||||||
# test set_yaw_rate
|
|
||||||
res = set_yaw_rate(yaw_rate=2)
|
|
||||||
assert res.success == True
|
|
||||||
state = get_state()
|
|
||||||
assert state.mode == State.MODE_NAVIGATE
|
|
||||||
assert state.yaw_mode == State.YAW_MODE_YAW_RATE
|
|
||||||
assert state.x == 5
|
|
||||||
assert state.y == 6
|
|
||||||
assert state.z == 7
|
|
||||||
assert state.yaw_rate == 2
|
|
||||||
assert state.xy_frame_id == 'map'
|
|
||||||
assert state.z_frame_id == 'test'
|
|
||||||
|
|
||||||
# navigate(yaw=nan) should keep yaw rate mode
|
|
||||||
res = navigate(x=nan, y=nan, z=nan, yaw=nan)
|
|
||||||
assert res.success == True
|
|
||||||
state = get_state()
|
|
||||||
assert state.mode == State.MODE_NAVIGATE
|
|
||||||
assert state.yaw_mode == State.YAW_MODE_YAW_RATE
|
|
||||||
assert state.x == 5
|
|
||||||
assert state.y == 6
|
|
||||||
assert state.z == 7
|
|
||||||
assert state.yaw_rate == 2
|
|
||||||
assert state.xy_frame_id == 'map'
|
|
||||||
assert state.z_frame_id == 'test'
|
|
||||||
|
|
||||||
# set_yaw(nan) should change back to yaw mode
|
|
||||||
res = set_yaw(yaw=nan)
|
|
||||||
assert res.success == True
|
|
||||||
state = get_state()
|
|
||||||
assert state.mode == State.MODE_NAVIGATE
|
|
||||||
assert state.yaw_mode == State.YAW_MODE_YAW
|
|
||||||
assert state.yaw == 0
|
|
||||||
assert state.yaw_frame_id == 'map'
|
|
||||||
|
|
||||||
# test set_position
|
|
||||||
res = set_position(x=nan, y=nan, z=13, yaw=nan, frame_id='test2')
|
|
||||||
assert res.success == True
|
|
||||||
state = get_state()
|
|
||||||
assert state.mode == State.MODE_POSITION
|
|
||||||
assert state.yaw_mode == State.YAW_MODE_YAW
|
|
||||||
assert state.x == 5
|
|
||||||
assert state.y == 6
|
|
||||||
assert state.z == 13
|
|
||||||
assert state.yaw == 0
|
|
||||||
assert state.xy_frame_id == 'map'
|
|
||||||
assert state.z_frame_id == 'test2'
|
|
||||||
assert state.yaw_frame_id == 'map'
|
|
||||||
|
|
||||||
# set_altitude should not change the mode
|
|
||||||
res = set_altitude(z=3, frame_id='test')
|
|
||||||
assert res.success == True
|
|
||||||
state = get_state()
|
|
||||||
assert state.mode == State.MODE_POSITION
|
|
||||||
assert state.yaw_mode == State.YAW_MODE_YAW
|
|
||||||
assert state.x == 5
|
|
||||||
assert state.y == 6
|
|
||||||
assert state.z == 3
|
|
||||||
assert state.yaw == 0
|
|
||||||
assert state.xy_frame_id == 'map'
|
|
||||||
assert state.z_frame_id == 'test'
|
|
||||||
assert state.yaw_frame_id == 'map'
|
|
||||||
|
|
||||||
# set_yaw should not change the main mode
|
|
||||||
res = set_yaw(yaw=1, frame_id='test2')
|
|
||||||
assert res.success == True
|
|
||||||
state = get_state()
|
|
||||||
assert state.mode == State.MODE_POSITION
|
|
||||||
assert state.yaw_mode == State.YAW_MODE_YAW
|
|
||||||
assert state.x == 5
|
|
||||||
assert state.y == 6
|
|
||||||
assert state.z == 3
|
|
||||||
assert state.yaw == 1
|
|
||||||
assert state.xy_frame_id == 'map'
|
|
||||||
assert state.z_frame_id == 'test'
|
|
||||||
assert state.yaw_frame_id == 'test2'
|
|
||||||
|
|
||||||
# test set_velocity
|
|
||||||
res = set_velocity(vx=1, frame_id='body')
|
|
||||||
state = get_state()
|
|
||||||
assert state.mode == State.MODE_VELOCITY
|
|
||||||
assert state.yaw_mode == State.YAW_MODE_YAW
|
|
||||||
assert state.vx == 1
|
|
||||||
assert state.vy == 0
|
|
||||||
assert state.vz == 0
|
|
||||||
assert state.yaw == 0
|
|
||||||
assert state.xy_frame_id == 'map'
|
|
||||||
assert state.z_frame_id == 'map'
|
|
||||||
assert state.yaw_frame_id == 'map'
|
|
||||||
|
|
||||||
# set_altitude should not work in velocity mode
|
|
||||||
res = set_altitude(z=3, frame_id='test')
|
|
||||||
assert res.success == False
|
|
||||||
assert res.message.startswith('Altitude cannot be set in')
|
|
||||||
|
|
||||||
# test set_attitude
|
|
||||||
res = set_attitude(roll=0.1, pitch=0.2, yaw=0.3, thrust=0.5)
|
|
||||||
assert res.success == True
|
|
||||||
state = get_state()
|
|
||||||
assert state.mode == State.MODE_ATTITUDE
|
|
||||||
assert state.yaw_mode == State.YAW_MODE_YAW
|
|
||||||
assert state.roll == approx(0.1)
|
|
||||||
assert state.pitch == approx(0.2)
|
|
||||||
assert state.yaw == approx(0.3)
|
|
||||||
assert state.thrust == approx(0.5)
|
|
||||||
assert state.yaw_frame_id == 'map'
|
|
||||||
msg = rospy.wait_for_message('/mavros/setpoint_attitude/attitude', PoseStamped, timeout=3)
|
|
||||||
# Tait-Bryan ZYX angle (rzyx) converted to quaternion
|
|
||||||
assert msg.pose.orientation.x == approx(0.0342708)
|
|
||||||
assert msg.pose.orientation.y == approx(0.10602051)
|
|
||||||
assert msg.pose.orientation.z == approx(0.14357218)
|
|
||||||
assert msg.pose.orientation.w == approx(0.98334744)
|
|
||||||
msg = rospy.wait_for_message('/mavros/setpoint_attitude/thrust', mavros_msgs.msg.Thrust, timeout=3)
|
|
||||||
assert msg.thrust == approx(0.5)
|
|
||||||
|
|
||||||
# set_yaw should work in attitude mode
|
|
||||||
res = set_yaw(yaw=0.7, frame_id='test2')
|
|
||||||
assert res.success == True
|
|
||||||
state = get_state()
|
|
||||||
assert state.mode == State.MODE_ATTITUDE
|
|
||||||
assert state.yaw_mode == State.YAW_MODE_YAW
|
|
||||||
assert state.roll == approx(0.1)
|
|
||||||
assert state.pitch == approx(0.2)
|
|
||||||
assert state.yaw == approx(0.7)
|
|
||||||
assert state.thrust == approx(0.5)
|
|
||||||
assert state.yaw_frame_id == 'test2'
|
|
||||||
|
|
||||||
# set_yaw_rate should not work in attitude mode
|
|
||||||
res = set_yaw_rate(yaw_rate=0.3)
|
|
||||||
assert res.success == False
|
|
||||||
assert res.message.startswith('Yaw rate cannot be set in')
|
|
||||||
|
|
||||||
# test set_rates
|
|
||||||
res = set_rates(roll_rate=nan, pitch_rate=nan, yaw_rate=0.3, thrust=0.6)
|
|
||||||
assert res.success == True
|
|
||||||
state = get_state()
|
|
||||||
assert state.mode == State.MODE_RATES
|
|
||||||
assert state.yaw_mode == State.YAW_MODE_YAW_RATE
|
|
||||||
assert state.roll_rate == approx(0)
|
|
||||||
assert state.pitch_rate == approx(0)
|
|
||||||
assert state.yaw_rate == approx(0.3)
|
|
||||||
assert state.thrust == approx(0.6)
|
|
||||||
msg = rospy.wait_for_message('/mavros/setpoint_raw/attitude', mavros_msgs.msg.AttitudeTarget, timeout=3)
|
|
||||||
assert msg.thrust == approx(0.6)
|
|
||||||
|
|
||||||
res = set_rates(roll_rate=0.3, pitch_rate=0.2, yaw_rate=0.1, thrust=0.4)
|
|
||||||
assert res.success == True
|
|
||||||
state = get_state()
|
|
||||||
assert state.mode == State.MODE_RATES
|
|
||||||
assert state.yaw_mode == State.YAW_MODE_YAW_RATE
|
|
||||||
assert state.roll_rate == approx(0.3)
|
|
||||||
assert state.pitch_rate == approx(0.2)
|
|
||||||
assert state.yaw_rate == approx(0.1)
|
|
||||||
assert state.thrust == approx(0.4)
|
|
||||||
|
|
||||||
res = set_rates(roll_rate=nan, pitch_rate=nan, yaw_rate=nan, thrust=0.3)
|
|
||||||
assert res.success == True
|
|
||||||
state = get_state()
|
|
||||||
assert state.mode == State.MODE_RATES
|
|
||||||
assert state.yaw_mode == State.YAW_MODE_YAW_RATE
|
|
||||||
assert state.roll_rate == approx(0.3)
|
|
||||||
assert state.pitch_rate == approx(0.2)
|
|
||||||
assert state.yaw_rate == approx(0.1)
|
|
||||||
assert state.thrust == approx(0.3)
|
|
||||||
msg = rospy.wait_for_message('/mavros/setpoint_raw/attitude', mavros_msgs.msg.AttitudeTarget, timeout=3)
|
|
||||||
assert msg.type_mask == mavros_msgs.msg.AttitudeTarget.IGNORE_ATTITUDE
|
|
||||||
assert msg.body_rate.x == approx(0.3)
|
|
||||||
assert msg.body_rate.y == approx(0.2)
|
|
||||||
assert msg.body_rate.z == approx(0.1)
|
|
||||||
|
|
||||||
# set_yaw_rate should work in rates mode
|
|
||||||
res = set_yaw_rate(yaw_rate=0.4)
|
|
||||||
assert res.success == True
|
|
||||||
state = get_state()
|
|
||||||
assert state.mode == State.MODE_RATES
|
|
||||||
assert state.yaw_mode == State.YAW_MODE_YAW_RATE
|
|
||||||
assert state.roll_rate == approx(0.3)
|
|
||||||
assert state.pitch_rate == approx(0.2)
|
|
||||||
assert state.yaw_rate == approx(0.4)
|
|
||||||
assert state.thrust == approx(0.3)
|
|
||||||
|
|
||||||
res = set_rates(roll_rate=inf)
|
|
||||||
assert res.success == False
|
|
||||||
assert res.message == 'roll_rate argument cannot be Inf'
|
|
||||||
@@ -1,10 +0,0 @@
|
|||||||
<launch>
|
|
||||||
<node name="simple_offboard" pkg="clover" type="simple_offboard" required="true" output="screen"/>
|
|
||||||
|
|
||||||
<node pkg="tf2_ros" type="static_transform_publisher" name="test_frame" args="10 20 30 0 0 0 map test"/>
|
|
||||||
|
|
||||||
<node pkg="tf2_ros" type="static_transform_publisher" name="test2_frame" args="100 200 300 0 0 0 map test2"/>
|
|
||||||
|
|
||||||
<param name="test_module" value="$(find clover)/test/offboard.py"/>
|
|
||||||
<test test-name="offboard_test" pkg="ros_pytest" type="ros_pytest_runner"/>
|
|
||||||
</launch>
|
|
||||||
@@ -1 +0,0 @@
|
|||||||
/var/log/clover.log
|
|
||||||
Some files were not shown because too many files have changed in this diff Show More
Reference in New Issue
Block a user