diff options
author | Stanislaw Halik <sthalik@misaki.pl> | 2013-07-27 09:07:42 +0200 |
---|---|---|
committer | Stanislaw Halik <sthalik@misaki.pl> | 2013-07-27 09:07:42 +0200 |
commit | e4f1be83f42742a2fc9a1606695d89a442999d1c (patch) | |
tree | 61ea5d18c3a0612017beefa02b44dd6541b00833 /ftnoir_tracker_aruco | |
parent | 7c6208d53fa7f96fb1e844b23b9ad341c22f0fc8 (diff) |
Import aruco tracker
Diffstat (limited to 'ftnoir_tracker_aruco')
-rw-r--r-- | ftnoir_tracker_aruco/aruco-tracker.qrc | 5 | ||||
-rw-r--r-- | ftnoir_tracker_aruco/aruco-trackercontrols.ui | 500 | ||||
-rw-r--r-- | ftnoir_tracker_aruco/ftnoir_tracker_aruco.cpp | 492 | ||||
-rw-r--r-- | ftnoir_tracker_aruco/ftnoir_tracker_aruco.h | 85 | ||||
-rw-r--r-- | ftnoir_tracker_aruco/ftnoir_tracker_aruco_dll.h | 20 | ||||
-rw-r--r-- | ftnoir_tracker_aruco/images/aruco.png | bin | 0 -> 987 bytes | |||
-rw-r--r-- | ftnoir_tracker_aruco/include/aruco.h | 134 | ||||
-rw-r--r-- | ftnoir_tracker_aruco/include/arucofidmarkers.h | 119 | ||||
-rw-r--r-- | ftnoir_tracker_aruco/include/board.h | 168 | ||||
-rw-r--r-- | ftnoir_tracker_aruco/include/boarddetector.h | 139 | ||||
-rw-r--r-- | ftnoir_tracker_aruco/include/cameraparameters.h | 137 | ||||
-rw-r--r-- | ftnoir_tracker_aruco/include/cvdrawingutils.h | 52 | ||||
-rw-r--r-- | ftnoir_tracker_aruco/include/exports.h | 46 | ||||
-rw-r--r-- | ftnoir_tracker_aruco/include/marker.h | 143 | ||||
-rw-r--r-- | ftnoir_tracker_aruco/include/markerdetector.h | 358 | ||||
-rw-r--r-- | ftnoir_tracker_aruco/video_widget.cpp | 30 | ||||
-rw-r--r-- | ftnoir_tracker_aruco/video_widget.h | 39 |
17 files changed, 2467 insertions, 0 deletions
diff --git a/ftnoir_tracker_aruco/aruco-tracker.qrc b/ftnoir_tracker_aruco/aruco-tracker.qrc new file mode 100644 index 00000000..70859876 --- /dev/null +++ b/ftnoir_tracker_aruco/aruco-tracker.qrc @@ -0,0 +1,5 @@ +<RCC> + <qresource prefix="/"> + <file>images/aruco.png</file> + </qresource> +</RCC> diff --git a/ftnoir_tracker_aruco/aruco-trackercontrols.ui b/ftnoir_tracker_aruco/aruco-trackercontrols.ui new file mode 100644 index 00000000..173accdf --- /dev/null +++ b/ftnoir_tracker_aruco/aruco-trackercontrols.ui @@ -0,0 +1,500 @@ +<?xml version="1.0" encoding="UTF-8"?> +<ui version="4.0"> + <class>Form</class> + <widget class="QWidget" name="Form"> + <property name="windowModality"> + <enum>Qt::NonModal</enum> + </property> + <property name="geometry"> + <rect> + <x>0</x> + <y>0</y> + <width>593</width> + <height>280</height> + </rect> + </property> + <property name="sizePolicy"> + <sizepolicy hsizetype="Preferred" vsizetype="Preferred"> + <horstretch>0</horstretch> + <verstretch>0</verstretch> + </sizepolicy> + </property> + <property name="maximumSize"> + <size> + <width>750</width> + <height>280</height> + </size> + </property> + <property name="windowTitle"> + <string>HT tracker settings</string> + </property> + <widget class="QLabel" name="label"> + <property name="geometry"> + <rect> + <x>10</x> + <y>10</y> + <width>141</width> + <height>16</height> + </rect> + </property> + <property name="text"> + <string>Horizontal FOV</string> + </property> + </widget> + <widget class="QDoubleSpinBox" name="cameraFOV"> + <property name="geometry"> + <rect> + <x>130</x> + <y>10</y> + <width>251</width> + <height>22</height> + </rect> + </property> + <property name="locale"> + <locale language="English" country="UnitedStates"/> + </property> + <property name="minimum"> + <double>35.000000000000000</double> + </property> + <property name="maximum"> + <double>180.000000000000000</double> + </property> + <property name="value"> + <double>52.000000000000000</double> + </property> + </widget> + <widget class="QLabel" name="label_2"> + <property name="geometry"> + <rect> + <x>10</x> + <y>40</y> + <width>137</width> + <height>16</height> + </rect> + </property> + <property name="text"> + <string>Frames per second</string> + </property> + </widget> + <widget class="QComboBox" name="cameraFPS"> + <property name="geometry"> + <rect> + <x>130</x> + <y>40</y> + <width>251</width> + <height>22</height> + </rect> + </property> + <item> + <property name="text"> + <string notr="true">Default</string> + </property> + </item> + <item> + <property name="text"> + <string>30</string> + </property> + </item> + <item> + <property name="text"> + <string>60</string> + </property> + </item> + <item> + <property name="text"> + <string>120</string> + </property> + </item> + </widget> + <widget class="QLabel" name="label_3"> + <property name="geometry"> + <rect> + <x>10</x> + <y>70</y> + <width>133</width> + <height>16</height> + </rect> + </property> + <property name="text"> + <string>Camera name</string> + </property> + </widget> + <widget class="QPushButton" name="buttonOK"> + <property name="geometry"> + <rect> + <x>430</x> + <y>250</y> + <width>75</width> + <height>23</height> + </rect> + </property> + <property name="text"> + <string>OK</string> + </property> + </widget> + <widget class="QPushButton" name="buttonCancel"> + <property name="geometry"> + <rect> + <x>510</x> + <y>250</y> + <width>75</width> + <height>23</height> + </rect> + </property> + <property name="text"> + <string>Cancel</string> + </property> + </widget> + <widget class="QGroupBox" name="groupBox"> + <property name="geometry"> + <rect> + <x>390</x> + <y>10</y> + <width>101</width> + <height>81</height> + </rect> + </property> + <property name="title"> + <string>Enable axes</string> + </property> + <widget class="QCheckBox" name="rx"> + <property name="geometry"> + <rect> + <x>10</x> + <y>20</y> + <width>70</width> + <height>17</height> + </rect> + </property> + <property name="text"> + <string>RX</string> + </property> + </widget> + <widget class="QCheckBox" name="ry"> + <property name="geometry"> + <rect> + <x>10</x> + <y>40</y> + <width>70</width> + <height>17</height> + </rect> + </property> + <property name="text"> + <string>RY</string> + </property> + </widget> + <widget class="QCheckBox" name="rz"> + <property name="geometry"> + <rect> + <x>10</x> + <y>60</y> + <width>70</width> + <height>17</height> + </rect> + </property> + <property name="text"> + <string>RZ</string> + </property> + </widget> + <widget class="QCheckBox" name="tx"> + <property name="geometry"> + <rect> + <x>60</x> + <y>20</y> + <width>70</width> + <height>17</height> + </rect> + </property> + <property name="text"> + <string>TX</string> + </property> + </widget> + <widget class="QCheckBox" name="ty"> + <property name="geometry"> + <rect> + <x>60</x> + <y>40</y> + <width>70</width> + <height>17</height> + </rect> + </property> + <property name="text"> + <string>TY</string> + </property> + </widget> + <widget class="QCheckBox" name="tz"> + <property name="geometry"> + <rect> + <x>60</x> + <y>60</y> + <width>70</width> + <height>17</height> + </rect> + </property> + <property name="text"> + <string>TZ</string> + </property> + </widget> + </widget> + <widget class="QComboBox" name="cameraName"> + <property name="geometry"> + <rect> + <x>130</x> + <y>70</y> + <width>251</width> + <height>22</height> + </rect> + </property> + </widget> + <widget class="QLabel" name="label_4"> + <property name="geometry"> + <rect> + <x>10</x> + <y>100</y> + <width>128</width> + <height>16</height> + </rect> + </property> + <property name="text"> + <string>Resolution</string> + </property> + </widget> + <widget class="QComboBox" name="resolution"> + <property name="geometry"> + <rect> + <x>130</x> + <y>100</y> + <width>251</width> + <height>22</height> + </rect> + </property> + <item> + <property name="text"> + <string>640x480</string> + </property> + </item> + <item> + <property name="text"> + <string>320x240</string> + </property> + </item> + <item> + <property name="text"> + <string>320x200</string> + </property> + </item> + <item> + <property name="text"> + <string>Default (not recommended!)</string> + </property> + </item> + </widget> + <widget class="QDoubleSpinBox" name="doubleSpinBox"> + <property name="geometry"> + <rect> + <x>130</x> + <y>130</y> + <width>171</width> + <height>22</height> + </rect> + </property> + <property name="frame"> + <bool>true</bool> + </property> + <property name="buttonSymbols"> + <enum>QAbstractSpinBox::NoButtons</enum> + </property> + <property name="decimals"> + <number>24</number> + </property> + <property name="minimum"> + <double>-1000.000000000000000</double> + </property> + <property name="maximum"> + <double>1000.000000000000000</double> + </property> + <property name="singleStep"> + <double>0.000000000000000</double> + </property> + </widget> + <widget class="QLabel" name="label_5"> + <property name="geometry"> + <rect> + <x>10</x> + <y>130</y> + <width>111</width> + <height>16</height> + </rect> + </property> + <property name="text"> + <string>Distortion coefficients</string> + </property> + </widget> + <widget class="QDoubleSpinBox" name="doubleSpinBox_2"> + <property name="geometry"> + <rect> + <x>130</x> + <y>160</y> + <width>171</width> + <height>22</height> + </rect> + </property> + <property name="frame"> + <bool>true</bool> + </property> + <property name="buttonSymbols"> + <enum>QAbstractSpinBox::NoButtons</enum> + </property> + <property name="decimals"> + <number>24</number> + </property> + <property name="minimum"> + <double>-1000.000000000000000</double> + </property> + <property name="maximum"> + <double>1000.000000000000000</double> + </property> + <property name="singleStep"> + <double>0.000000000000000</double> + </property> + </widget> + <widget class="QDoubleSpinBox" name="doubleSpinBox_3"> + <property name="geometry"> + <rect> + <x>130</x> + <y>190</y> + <width>171</width> + <height>22</height> + </rect> + </property> + <property name="frame"> + <bool>true</bool> + </property> + <property name="buttonSymbols"> + <enum>QAbstractSpinBox::NoButtons</enum> + </property> + <property name="decimals"> + <number>24</number> + </property> + <property name="minimum"> + <double>-1000.000000000000000</double> + </property> + <property name="maximum"> + <double>1000.000000000000000</double> + </property> + <property name="singleStep"> + <double>0.000000000000000</double> + </property> + </widget> + <widget class="QDoubleSpinBox" name="doubleSpinBox_4"> + <property name="geometry"> + <rect> + <x>130</x> + <y>220</y> + <width>171</width> + <height>22</height> + </rect> + </property> + <property name="frame"> + <bool>true</bool> + </property> + <property name="buttonSymbols"> + <enum>QAbstractSpinBox::NoButtons</enum> + </property> + <property name="decimals"> + <number>24</number> + </property> + <property name="minimum"> + <double>-1000.000000000000000</double> + </property> + <property name="maximum"> + <double>1000.000000000000000</double> + </property> + <property name="singleStep"> + <double>0.000000000000000</double> + </property> + </widget> + <widget class="QDoubleSpinBox" name="doubleSpinBox_5"> + <property name="geometry"> + <rect> + <x>130</x> + <y>250</y> + <width>171</width> + <height>22</height> + </rect> + </property> + <property name="frame"> + <bool>true</bool> + </property> + <property name="buttonSymbols"> + <enum>QAbstractSpinBox::NoButtons</enum> + </property> + <property name="decimals"> + <number>24</number> + </property> + <property name="minimum"> + <double>-1000.000000000000000</double> + </property> + <property name="maximum"> + <double>1000.000000000000000</double> + </property> + <property name="singleStep"> + <double>0.000000000000000</double> + </property> + </widget> + <widget class="QLabel" name="label_6"> + <property name="geometry"> + <rect> + <x>390</x> + <y>100</y> + <width>58</width> + <height>16</height> + </rect> + </property> + <property name="text"> + <string>Marker size</string> + </property> + </widget> + <widget class="QSpinBox" name="markerSize"> + <property name="geometry"> + <rect> + <x>455</x> + <y>97</y> + <width>70</width> + <height>22</height> + </rect> + </property> + <property name="suffix"> + <string> cm</string> + </property> + <property name="minimum"> + <number>1</number> + </property> + <property name="maximum"> + <number>100</number> + </property> + <property name="value"> + <number>10</number> + </property> + </widget> + </widget> + <resources/> + <connections/> + <designerdata> + <property name="gridDeltaX"> + <number>10</number> + </property> + <property name="gridDeltaY"> + <number>10</number> + </property> + <property name="gridSnapX"> + <bool>false</bool> + </property> + <property name="gridSnapY"> + <bool>false</bool> + </property> + <property name="gridVisible"> + <bool>true</bool> + </property> + </designerdata> +</ui> diff --git a/ftnoir_tracker_aruco/ftnoir_tracker_aruco.cpp b/ftnoir_tracker_aruco/ftnoir_tracker_aruco.cpp new file mode 100644 index 00000000..fb5d91f7 --- /dev/null +++ b/ftnoir_tracker_aruco/ftnoir_tracker_aruco.cpp @@ -0,0 +1,492 @@ +/* Copyright (c) 2013 Stanislaw Halik <sthalik@misaki.pl> + * + * Permission to use, copy, modify, and/or distribute this software for any + * purpose with or without fee is hereby granted, provided that the above + * copyright notice and this permission notice appear in all copies. + */ + +#include "ftnoir_tracker_base/ftnoir_tracker_base.h" +#include "ftnoir_tracker_aruco.h" +#include "ui_aruco-trackercontrols.h" +#include "facetracknoir/global-settings.h" +#include <cmath> +#include <QMutexLocker> +#include <aruco.h> +#include <opencv2/opencv.hpp> +#include <opencv/highgui.h> +#include <vector> + +#if defined(_WIN32) || defined(__WIN32) +#include <dshow.h> +#else +#include <unistd.h> +#endif + +// delicious copypasta +static QList<QString> get_camera_names(void) { + QList<QString> ret; +#if defined(_WIN32) || defined(__WIN32) + // Create the System Device Enumerator. + HRESULT hr; + ICreateDevEnum *pSysDevEnum = NULL; + hr = CoCreateInstance(CLSID_SystemDeviceEnum, NULL, CLSCTX_INPROC_SERVER, IID_ICreateDevEnum, (void **)&pSysDevEnum); + if (FAILED(hr)) + { + return ret; + } + // Obtain a class enumerator for the video compressor category. + IEnumMoniker *pEnumCat = NULL; + hr = pSysDevEnum->CreateClassEnumerator(CLSID_VideoInputDeviceCategory, &pEnumCat, 0); + + if (hr == S_OK) { + // Enumerate the monikers. + IMoniker *pMoniker = NULL; + ULONG cFetched; + while (pEnumCat->Next(1, &pMoniker, &cFetched) == S_OK) { + IPropertyBag *pPropBag; + hr = pMoniker->BindToStorage(0, 0, IID_IPropertyBag, (void **)&pPropBag); + if (SUCCEEDED(hr)) { + // To retrieve the filter's friendly name, do the following: + VARIANT varName; + VariantInit(&varName); + hr = pPropBag->Read(L"FriendlyName", &varName, 0); + if (SUCCEEDED(hr)) + { + // Display the name in your UI somehow. + QString str((QChar*)varName.bstrVal, wcslen(varName.bstrVal)); + ret.append(str); + } + VariantClear(&varName); + + ////// To create an instance of the filter, do the following: + ////IBaseFilter *pFilter; + ////hr = pMoniker->BindToObject(NULL, NULL, IID_IBaseFilter, + //// (void**)&pFilter); + // Now add the filter to the graph. + //Remember to release pFilter later. + pPropBag->Release(); + } + pMoniker->Release(); + } + pEnumCat->Release(); + } + pSysDevEnum->Release(); +#else + for (int i = 0; i < 16; i++) { + char buf[128]; + sprintf(buf, "/dev/video%d", i); + if (access(buf, R_OK | W_OK) == 0) { + ret.append(buf); + } else { + break; + } + } +#endif + return ret; +} + +typedef struct { + int width; + int height; +} resolution_tuple; + +static resolution_tuple resolution_choices[] = { + { 640, 480 }, + { 320, 240 }, + { 320, 200 }, + { 0, 0 } +}; + +void Tracker::load_settings() +{ + QSettings settings("opentrack"); + QString currentFile = settings.value( "SettingsFile", QCoreApplication::applicationDirPath() + "/Settings/default.ini" ).toString(); + QSettings iniFile( currentFile, QSettings::IniFormat ); + + iniFile.beginGroup( "aruco-Tracker" ); + fov = iniFile.value("fov", 56).toFloat(); + force_fps = iniFile.value("fps", 0).toInt(); + camera_index = iniFile.value("camera-index", -1).toInt(); + int res = iniFile.value("resolution", 0).toInt(); + if (res < 0 || res >= (int)(sizeof(resolution_choices) / sizeof(resolution_tuple))) + res = 0; + resolution_tuple r = resolution_choices[res]; + force_width = r.width; + force_height = r.height; + enableRX = iniFile.value("enable-rx", true).toBool(); + enableRY = iniFile.value("enable-ry", true).toBool(); + enableRZ = iniFile.value("enable-rz", true).toBool(); + enableTX = iniFile.value("enable-tx", true).toBool(); + enableTY = iniFile.value("enable-ty", true).toBool(); + enableTZ = iniFile.value("enable-tz", true).toBool(); + marker_size = iniFile.value("marker-size", 10).toInt(); + for (int i = 0; i < 5; i++) + dc[i] = iniFile.value(QString("dc%1").arg(i), 0).toFloat(); + iniFile.endGroup(); +} + +Tracker::Tracker() +{ + fresh = false; + stop = false; + videoWidget = NULL; + layout = NULL; + enableRX = enableRY = enableRZ = enableTX = enableTY = enableTZ = true; + load_settings(); +} + +Tracker::~Tracker() +{ + if (layout) + delete layout; + if (videoWidget) + delete videoWidget; +} + +void Tracker::StartTracker(QFrame* videoframe) +{ + videoframe->show(); + videoWidget = new VideoWidget(videoframe); + QHBoxLayout* layout = new QHBoxLayout(); + layout->setContentsMargins(0, 0, 0, 0); + layout->addWidget(videoWidget); + if (videoframe->layout()) + delete videoframe->layout(); + videoframe->setLayout(layout); + videoWidget->show(); + this->layout = layout; + load_settings(); + connect(&timer, SIGNAL(timeout()), this, SLOT(paint_widget())); + timer.start(40); + start(); + for (int i = 0; i < 6; i++) + pose[i] = 0; +} + +void Tracker::paint_widget() { + if (fresh) { + fresh = false; + videoWidget->update(); + } +} + +#define HT_PI 3.1415926535 + +void Tracker::run() +{ + cv::VideoCapture camera(camera_index); + + if (force_width) + camera.set(CV_CAP_PROP_FRAME_WIDTH, force_width); + if (force_height) + camera.set(CV_CAP_PROP_FRAME_HEIGHT, force_height); + if (force_fps) + camera.set(CV_CAP_PROP_FPS, force_fps); + + aruco::MarkerDetector detector; + detector.setDesiredSpeed(3); + + cv::Mat color, grayscale; + + while (!stop) + { + if (!camera.read(color)) + break; + cv::cvtColor(color, grayscale, cv::COLOR_BGR2GRAY); + const float focal_length_w = 0.5 * grayscale.cols / tan(0.5 * fov * HT_PI / 180); + const float focal_length_h = 0.5 * grayscale.rows / tan(0.5 * fov * grayscale.rows / grayscale.cols * HT_PI / 180.0); + cv::Mat intrinsics = cv::Mat::eye(3, 3, CV_32FC1); + intrinsics.at<float> (0, 0) = focal_length_w; + intrinsics.at<float> (1, 1) = focal_length_h; + intrinsics.at<float> (0, 2) = grayscale.cols/2; + intrinsics.at<float> (1, 2) = grayscale.rows/2; + + cv::Mat dist_coeffs = cv::Mat::zeros(5, 1, CV_32FC1); + + for (int i = 0; i < 5; i++) + dist_coeffs.at<float>(i) = dc[i]; + + aruco::CameraParameters params(intrinsics, dist_coeffs, cv::Size(grayscale.cols, grayscale.rows)); + + std::vector< aruco::Marker > markers; + + detector.detect(grayscale, markers, params, marker_size / 100., false); + + QMutexLocker lck(&mtx); + + if (markers.size() == 1) { + const aruco::Marker& m = markers.at(0); + + cv::Mat rotation_matrix = cv::Mat::zeros(3, 3, CV_64FC1); + + cv::Mat junk1(3, 3, CV_64FC1), junk2(3, 3, CV_64FC1); + + cv::Rodrigues(m.Rvec, rotation_matrix); + + cv::Vec3d foo = cv::RQDecomp3x3(rotation_matrix, junk1, junk2); + + for (int i = 0; i < 3; i++) + { + pose[i+3] = foo[i]; + pose[i] = m.Tvec.at<float>(i); + } + } + + frame = color.clone(); + } +} + +bool Tracker::GiveHeadPoseData(double *data) +{ + bool ret = false; + + QMutexLocker lck(&mtx); + + if (frame.rows > 0) + { + videoWidget->update_image(frame.data, frame.cols, frame.rows); + fresh = true; + } + if (enableRX) + data[Yaw] = pose[Yaw]; + if (enableRY) + data[Pitch] = pose[Pitch]; + if (enableRZ) + data[Roll] = pose[Roll]; + if (enableTX) + data[TX] = pose[TX]; + if (enableTY) + data[TY] = pose[TY]; + if (enableTZ) + data[TZ] = pose[TZ]; + + return true; +} + +class TrackerDll : public Metadata +{ + // ITrackerDll interface + void Initialize() {} + void getFullName(QString *strToBeFilled); + void getShortName(QString *strToBeFilled); + void getDescription(QString *strToBeFilled); + void getIcon(QIcon *icon); +}; + +//----------------------------------------------------------------------------- +void TrackerDll::getFullName(QString *strToBeFilled) +{ + *strToBeFilled = "aruco"; +} + +void TrackerDll::getShortName(QString *strToBeFilled) +{ + *strToBeFilled = "aruco"; +} + +void TrackerDll::getDescription(QString *strToBeFilled) +{ + *strToBeFilled = ""; +} + +void TrackerDll::getIcon(QIcon *icon) +{ + *icon = QIcon(":/images/aruco.png"); +} + + +//----------------------------------------------------------------------------- +//#pragma comment(linker, "/export:GetTrackerDll=_GetTrackerDll@0") + +extern "C" FTNOIR_TRACKER_BASE_EXPORT Metadata* CALLING_CONVENTION GetMetadata() +{ + return new TrackerDll; +} + +//#pragma comment(linker, "/export:GetTracker=_GetTracker@0") + +extern "C" FTNOIR_TRACKER_BASE_EXPORT ITracker* CALLING_CONVENTION GetConstructor() +{ + return new Tracker; +} + +//////////////////////////////////////////////////////////////////////////////// +// Factory function that creates instances if the Tracker-settings dialog object. + +// Export both decorated and undecorated names. +// GetTrackerDialog - Undecorated name, which can be easily used with GetProcAddress +// Win32 API function. +// _GetTrackerDialog@0 - Common name decoration for __stdcall functions in C language. +//#pragma comment(linker, "/export:GetTrackerDialog=_GetTrackerDialog@0") + +extern "C" FTNOIR_TRACKER_BASE_EXPORT ITrackerDialog* CALLING_CONVENTION GetDialog( ) +{ + return new TrackerControls; +} + +TrackerControls::TrackerControls() +{ + ui.setupUi(this); + setAttribute(Qt::WA_NativeWindow, true); + connect(ui.cameraName, SIGNAL(currentIndexChanged(int)), this, SLOT(settingChanged(int))); + connect(ui.cameraFPS, SIGNAL(currentIndexChanged(int)), this, SLOT(settingChanged(int))); + connect(ui.cameraFOV, SIGNAL(valueChanged(double)), this, SLOT(settingChanged(double))); + connect(ui.rx, SIGNAL(stateChanged(int)), this, SLOT(settingChanged(int))); + connect(ui.ry, SIGNAL(stateChanged(int)), this, SLOT(settingChanged(int))); + connect(ui.rz, SIGNAL(stateChanged(int)), this, SLOT(settingChanged(int))); + connect(ui.tx, SIGNAL(stateChanged(int)), this, SLOT(settingChanged(int))); + connect(ui.ty, SIGNAL(stateChanged(int)), this, SLOT(settingChanged(int))); + connect(ui.tz, SIGNAL(stateChanged(int)), this, SLOT(settingChanged(int))); + connect(ui.buttonCancel, SIGNAL(clicked()), this, SLOT(doCancel())); + connect(ui.buttonOK, SIGNAL(clicked()), this, SLOT(doOK())); + //connect(ui.buttonSettings, SIGNAL(clicked()), this, SLOT(cameraSettings())); + loadSettings(); + settingsDirty = false; +} + +TrackerControls::~TrackerControls() +{ +} + +void TrackerControls::showEvent(QShowEvent *event) +{ +} + +void TrackerControls::Initialize(QWidget* parent) +{ + loadSettings(); + show(); +} + +void TrackerControls::loadSettings() +{ + ui.cameraName->clear(); + QList<QString> names = get_camera_names(); + names.prepend("Any available"); + ui.cameraName->addItems(names); + QSettings settings("opentrack"); + QString currentFile = settings.value( "SettingsFile", QCoreApplication::applicationDirPath() + "/Settings/default.ini" ).toString(); + QSettings iniFile( currentFile, QSettings::IniFormat ); + iniFile.beginGroup( "aruco-Tracker" ); + ui.cameraName->setCurrentIndex(iniFile.value("camera-index", -1).toInt() + 1); + ui.cameraFOV->setValue(iniFile.value("fov", 56).toFloat()); + int fps; + switch (iniFile.value("fps", 0).toInt()) + { + default: + case 0: + fps = 0; + break; + case 30: + fps = 1; + break; + case 60: + fps = 2; + break; + case 120: + fps = 3; + break; + } + ui.cameraFPS->setCurrentIndex(fps); + ui.rx->setCheckState(iniFile.value("enable-rx", true).toBool() ? Qt::Checked : Qt::Unchecked); + ui.ry->setCheckState(iniFile.value("enable-ry", true).toBool() ? Qt::Checked : Qt::Unchecked); + ui.rz->setCheckState(iniFile.value("enable-rz", true).toBool() ? Qt::Checked : Qt::Unchecked); + ui.tx->setCheckState(iniFile.value("enable-tx", true).toBool() ? Qt::Checked : Qt::Unchecked); + ui.ty->setCheckState(iniFile.value("enable-ty", true).toBool() ? Qt::Checked : Qt::Unchecked); + ui.tz->setCheckState(iniFile.value("enable-tz", true).toBool() ? Qt::Checked : Qt::Unchecked); + ui.resolution->setCurrentIndex(iniFile.value("resolution", 0).toInt()); + + ui.doubleSpinBox->setValue(iniFile.value("dc0").toDouble()); + ui.doubleSpinBox_2->setValue(iniFile.value("dc1").toDouble()); + ui.doubleSpinBox_3->setValue(iniFile.value("dc2").toDouble()); + ui.doubleSpinBox_4->setValue(iniFile.value("dc3").toDouble()); + ui.doubleSpinBox_5->setValue(iniFile.value("dc4").toDouble()); + + ui.markerSize->setValue(iniFile.value("marker-size", 10).toInt()); + + iniFile.endGroup(); + settingsDirty = false; +} + +void TrackerControls::save() +{ + QSettings settings("opentrack"); + QString currentFile = settings.value( "SettingsFile", QCoreApplication::applicationDirPath() + "/Settings/default.ini" ).toString(); + QSettings iniFile( currentFile, QSettings::IniFormat ); + + iniFile.beginGroup( "aruco-Tracker" ); + iniFile.setValue("fov", ui.cameraFOV->value()); + int fps; + switch (ui.cameraFPS->currentIndex()) + { + case 0: + default: + fps = 0; + break; + case 1: + fps = 30; + break; + case 2: + fps = 60; + break; + case 3: + fps = 120; + break; + } + iniFile.setValue("fps", fps); + iniFile.setValue("camera-index", ui.cameraName->currentIndex() - 1); + iniFile.setValue("enable-rx", ui.rx->checkState() != Qt::Unchecked ? true : false); + iniFile.setValue("enable-ry", ui.ry->checkState() != Qt::Unchecked ? true : false); + iniFile.setValue("enable-rz", ui.rz->checkState() != Qt::Unchecked ? true : false); + iniFile.setValue("enable-tx", ui.tx->checkState() != Qt::Unchecked ? true : false); + iniFile.setValue("enable-ty", ui.ty->checkState() != Qt::Unchecked ? true : false); + iniFile.setValue("enable-tz", ui.tz->checkState() != Qt::Unchecked ? true : false); + iniFile.setValue("resolution", ui.resolution->currentIndex()); + + iniFile.setValue("dc0", ui.doubleSpinBox->value()); + iniFile.setValue("dc1", ui.doubleSpinBox_2->value()); + iniFile.setValue("dc2", ui.doubleSpinBox_3->value()); + iniFile.setValue("dc3", ui.doubleSpinBox_4->value()); + iniFile.setValue("dc4", ui.doubleSpinBox_5->value()); + + iniFile.setValue("marker-size", ui.markerSize->value()); + + iniFile.endGroup(); + settingsDirty = false; +} + +void TrackerControls::doOK() +{ + save(); + this->close(); +} + +void TrackerControls::doCancel() +{ + if (settingsDirty) { + int ret = QMessageBox::question ( this, + "Settings have changed", + "Do you want to save the settings?", + QMessageBox::Save | QMessageBox::Discard | QMessageBox::Cancel, + QMessageBox::Discard ); + + switch (ret) { + case QMessageBox::Save: + save(); + this->close(); + break; + case QMessageBox::Discard: + this->close(); + break; + case QMessageBox::Cancel: + // Cancel was clicked + break; + default: + // should never be reached + break; + } + } + else { + this->close(); + } +} diff --git a/ftnoir_tracker_aruco/ftnoir_tracker_aruco.h b/ftnoir_tracker_aruco/ftnoir_tracker_aruco.h new file mode 100644 index 00000000..cbda40bf --- /dev/null +++ b/ftnoir_tracker_aruco/ftnoir_tracker_aruco.h @@ -0,0 +1,85 @@ +/* Copyright (c) 2013 Stanislaw Halik <sthalik@misaki.pl> + * + * Permission to use, copy, modify, and/or distribute this software for any + * purpose with or without fee is hereby granted, provided that the above + * copyright notice and this permission notice appear in all copies. + */ + +#ifndef FTNOIR_TRACKER_HT_H +#define FTNOIR_TRACKER_HT_H + +#include "ftnoir_tracker_base/ftnoir_tracker_base.h" +#include "ui_aruco-trackercontrols.h" +#include "video_widget.h" +#include <QObject> +#include <QTimer> +#include <QThread> +#include <QMutex> +#include <QHBoxLayout> +#include <opencv2/opencv.hpp> + +class Tracker : public QThread, public ITracker +{ + Q_OBJECT +public: + Tracker(); + ~Tracker(); + void StartTracker(QFrame* frame); + bool GiveHeadPoseData(double *data); + bool enableTX, enableTY, enableTZ, enableRX, enableRY, enableRZ; + bool NeedsTimeToFinish() { + return true; + } + void WaitForExit() { + stop = true; + wait(); + } + void run(); +private: + QMutex mtx; + QTimer timer; + VideoWidget* videoWidget; + QHBoxLayout* layout; + volatile bool fresh, stop; + float fov; + int camera_index; + float dc[5]; + int force_fps, force_width, force_height; + void load_settings(); + double pose[6]; + int marker_size; + cv::Mat frame; +private slots: + void paint_widget(); +}; + +// Widget that has controls for FTNoIR protocol client-settings. +class TrackerControls : public QWidget, public ITrackerDialog +{ + Q_OBJECT +public: + + explicit TrackerControls(); + virtual ~TrackerControls(); + void showEvent ( QShowEvent * event ); + + void Initialize(QWidget *parent); + void registerTracker(ITracker *tracker) {} + void unRegisterTracker() {} + +private: + Ui::Form ui; + void loadSettings(); + void save(); + bool settingsDirty; + +private slots: + void doOK(); + void doCancel(); + void settingChanged() { settingsDirty = true; } + void settingChanged(int) { settingsDirty = true; } + void settingChanged(double) { settingsDirty = true; } +}; + +#endif + diff --git a/ftnoir_tracker_aruco/ftnoir_tracker_aruco_dll.h b/ftnoir_tracker_aruco/ftnoir_tracker_aruco_dll.h new file mode 100644 index 00000000..1e53f802 --- /dev/null +++ b/ftnoir_tracker_aruco/ftnoir_tracker_aruco_dll.h @@ -0,0 +1,20 @@ +/* Copyright (c) 2013 Stanis³aw Halik <sthalik@misaki.pl> + * + * Permission to use, copy, modify, and/or distribute this software for any + * purpose with or without fee is hereby granted, provided that the above + * copyright notice and this permission notice appear in all copies. + */ + +#include "ftnoir_tracker_base/ftnoir_tracker_base.h" +#include "facetracknoir/global-settings.h" + +//----------------------------------------------------------------------------- +class TrackerDll : public Metadata +{ + // ITrackerDll interface + void Initialize() {} + void getFullName(QString *strToBeFilled); + void getShortName(QString *strToBeFilled); + void getDescription(QString *strToBeFilled); + void getIcon(QIcon *icon); +}; diff --git a/ftnoir_tracker_aruco/images/aruco.png b/ftnoir_tracker_aruco/images/aruco.png Binary files differnew file mode 100644 index 00000000..491c39e6 --- /dev/null +++ b/ftnoir_tracker_aruco/images/aruco.png diff --git a/ftnoir_tracker_aruco/include/aruco.h b/ftnoir_tracker_aruco/include/aruco.h new file mode 100644 index 00000000..569b95fb --- /dev/null +++ b/ftnoir_tracker_aruco/include/aruco.h @@ -0,0 +1,134 @@ +/** + +Copyright 2011 Rafael Muñoz Salinas. All rights reserved. + +Redistribution and use in source and binary forms, with or without modification, are +permitted provided that the following conditions are met: + + 1. Redistributions of source code must retain the above copyright notice, this list of + conditions and the following disclaimer. + + 2. Redistributions in binary form must reproduce the above copyright notice, this list + of conditions and the following disclaimer in the documentation and/or other materials + provided with the distribution. + +THIS SOFTWARE IS PROVIDED BY Rafael Muñoz Salinas ''AS IS'' AND ANY EXPRESS OR IMPLIED +WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND +FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL Rafael Muñoz Salinas OR +CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR +CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR +SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON +ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING +NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF +ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. + +The views and conclusions contained in the software and documentation are those of the +authors and should not be interpreted as representing official policies, either expressed +or implied, of Rafael Muñoz Salinas. + + + + \mainpage ArUco: Augmented Reality library from the University of Cordoba + + +ArUco is a minimal C++ library for detection of Augmented Reality markers based on OpenCv exclusively. + +It is an educational project to show student how to detect augmented reality markers and it is provided under BSD license. + + +\section INTRODUCTION INTRODUCTION + +The library relies on the use of coded markers. Each marker has an unique code indicated by the black and white colors in it. The libary detect borders, and analyzes into the rectangular regions which of them are likely to be markers. Then, a decoding is performed and if the code is valid, it is considered that the rectangle is a marker. + +The codification included into the marker is a slighly modified version of the Hamming Code. It has a total a 25 bits didived in 5 rows of 5 bits each. So, we have 5 words of 5 bits. Each word, contains only 2 bits of real information, the rest is for and error detection/correction (error correction is yet to be done). As a conclusion, a marker contains 10 bits of real information wich allows 1024 different markers. + + +\section BOARDS BOARDS + +Aruco allows the possibility to employ board. Boards are markers composed by an array of markers arranged in a known order. The advantages of using boards instead of simple markers are: + - More robusteness. The misdetection of several markers of the board is not a problem as long as a minimum set of them are detected. + - More precision. Since there are a larger number of corners, camera pose estimation becomes more precise. + + +\section APPLICATIONS APPLICATIONS + +The library comes with five applications that will help you to learn how to use the library: + - aruco_create_marker: which creates marker and saves it in a jpg file you can print. + - aruco_simple : simple test aplication that detects the markers in a image + - aruco_test: this is the main application for detection. It reads images either from the camera of from a video and detect markers. Additionally, if you provide the intrinsics of the camera(obtained by OpenCv calibration) and the size of the marker in meters, the library calculates the marker intrinsics so that you can easily create your AR applications. + - aruco_test_gl: shows how to use the library AR applications using OpenGL for rendering + - aruco_create_board: application that helps you to create a board + - aruco_simple_board: simple test aplication that detects a board of markers in a image + - aruco_test_board: application that detects boards + - aruco_test_board_gl: application that detects boards and uses OpenGL to draw + +\section LIBRARY LIBRARY DESCRIPTION: + +The ArUco library contents are divided in two main directories. The src directory, which contains the library itself. And the utils directory which contains the applications. + +The library main classes are: + - aruco::CameraParameters: represent the information of the camera that captures the images. Here you must set the calibration info. + - aruco::Marker: which represent a marker detected in the image + - aruco::MarkerDetector: that is in charge of deteting the markers in a image Detection is done by simple calling the member funcion ArMarkerDetector::detect(). Additionally, the classes contain members to create the required matrices for rendering using OpenGL. See aruco_test_gl for details + - aruco::BoardConfiguration: A board is an array of markers in a known order. BoardConfiguracion is the class that defines a board by indicating the id of its markers. In addition, it has informacion about the distance between the markers so that extrinsica camera computations can be done. + - aruco::Board: This class defines a board detected in a image. The board has the extrinsic camera parameters as public atributes. In addition, it has a method that allows obtain the matrix for getting its position in OpenGL (see aruco_test_board_gl for details). + - aruco::BoardDetector : This is the class in charge of detecting a board in a image. You must pass to it the set of markers detected by ArMarkerDetector and the BoardConfiguracion of the board you want to detect. This class will do the rest for you, even calculating the camera extrinsics. + + +\section COMPILING COMPILING THE LIBRARY: +\subsection Linux +Go to the aruco library and do +\verbatim +>mkdir build +>cd build +>cmake .. +>make +>make install (optional) +\endverbatim + +NOTE ON OPENGL: The library supports eaily the integration with OpenGL. In order to compile with support for OpenGL, you just have installed in your system the develop packages for GL and glut (or freeglut). + +\subsection WINDOWS + +The library has been compiled using MinGW and codeblocks. Below I describe the best way to compile it that I know. If you know better, please let me know. + - step 1) codeblocks + -# Download codeblocks. I recommend to download the version 10.5 with mingw included (codeblocks-10.05mingw-setup.exe) + -# Install and set the PATH variable so that the codeblock/mingw/bin directory is included. In my case c:/codeblocks/mingw/bin. This will allow cmake to find the compiler. + -# The codeblock program will not find the mingw path by deafult. So, run codeblocks and go to setting->Compuiler debugger and set the correct path to the MinGW dir. + - step 2) cmake + -# Download and install the last version of cmake. + - step 3) OpenCv + -# Download the source code and compile it using cmake and codeblocks. Note: install the library in C:\ if you want it to be easily detected by cmake afterwards + - step 4) aruco + -# Download and decompress. + -# Open cmake gui application and set the path to the main library directory and also set a path where the project is going to be built. + -# Generate the codeblock project. + -# Open the project with codeblock and compile then, install. The programs will be probably generated into the bin directory + +OpenGL: by default, the mingw version installed has not the glut library. So, the opengl programs are not compiled. If you want to compile with OpenGL support, you must install glut, or prefereably freeglut. +Thus, + - Download the library (http://www.martinpayne.me.uk/software/development/GLUT/freeglut-MinGW.zip) for mingw. + - Decompress in a directory X. + - Then, rerun cmake setting the variable GLU_PATH to that directory (>cmake .. -DGLUT_PATH="C:\X") + - Finally, recompile and test. Indeed, you should move the freeglut.dll to the directory with the binaries or to any other place in the PATH. + + +CONCLUSION: Move to Linux, things are simpler :P + + +\section Testing + +For testing the applications, the library provides videos and the corresponding camera parameters of these videos. Into the directories you will find information on how to run the examples. + +\section Final Notes + + - REQUIREMENTS: OpenCv >= 2.1.0. and OpenGL for (aruco_test_gl and aruco_test_board_gl) + - CONTACT: Rafael Munoz-Salinas: rmsalinas@uco.es + - This libary is free software and come with no guaratee! + +*/ + +#include "markerdetector.h" +#include "boarddetector.h" +#include "cvdrawingutils.h" + diff --git a/ftnoir_tracker_aruco/include/arucofidmarkers.h b/ftnoir_tracker_aruco/include/arucofidmarkers.h new file mode 100644 index 00000000..7dad4672 --- /dev/null +++ b/ftnoir_tracker_aruco/include/arucofidmarkers.h @@ -0,0 +1,119 @@ +/***************************** +Copyright 2011 Rafael Muñoz Salinas. All rights reserved. + +Redistribution and use in source and binary forms, with or without modification, are +permitted provided that the following conditions are met: + + 1. Redistributions of source code must retain the above copyright notice, this list of + conditions and the following disclaimer. + + 2. Redistributions in binary form must reproduce the above copyright notice, this list + of conditions and the following disclaimer in the documentation and/or other materials + provided with the distribution. + +THIS SOFTWARE IS PROVIDED BY Rafael Muñoz Salinas ''AS IS'' AND ANY EXPRESS OR IMPLIED +WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND +FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL Rafael Muñoz Salinas OR +CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR +CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR +SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON +ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING +NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF +ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. + +The views and conclusions contained in the software and documentation are those of the +authors and should not be interpreted as representing official policies, either expressed +or implied, of Rafael Muñoz Salinas. +********************************/ + +#ifndef ArucoFiducicalMarkerDetector_H +#define ArucoFiducicalMarkerDetector_H +#include <opencv2/core/core.hpp> +#include "exports.h" +#include "marker.h" +#include "board.h" +namespace aruco { + +class ARUCO_EXPORTS FiducidalMarkers { +public: + /** + * \brief Creates an ar marker with the id specified using a modified version of the hamming code. + * There are two type of markers: a) These of 10 bits b) these of 3 bits. The latter are employed for applications + * that need few marker but they must be small. The two type of markers are distinguished by their ids. While the first type + * of markers have ids in the interval [0-1023], the second type ids in the interval [2000-2006]. + * + * + * 10 bits markers + * ----------------------- + * There are a total of 5 rows of 5 cols. Each row encodes a total of 2 bits, so there are 2^10 bits:(0-1023). + * + * The least significative bytes are first (from left-up to to right-bottom) + * + * Example: the id = 110 (decimal) is be represented in binary as : 00 01 10 11 10. + * + * Then, it will generate the following marker: + * + * -# 1st row encodes 00: 1 0 0 0 0 : hex 0x10 + * -# 2nd row encodes 01: 1 0 1 1 1 : hex 0x17 + * -# 3nd row encodes 10: 0 1 0 0 1 : hex 0x09 + * -# 4th row encodes 11: 0 1 1 1 0 : hex 0x0e + * -# 5th row encodes 10: 0 1 0 0 1 : hex 0x09 + * + * Note that : The first bit, is the inverse of the hamming parity. This avoids the 0 0 0 0 0 to be valid + * These marker are detected by the function getFiduciadlMarker_Aruco_Type1 + */ + static cv::Mat createMarkerImage(int id,int size) throw (cv::Exception); + + /** Detection of fiducidal aruco markers (10 bits) + * @param in input image with the patch that contains the possible marker + * @param nRotations number of 90deg rotations in clockwise direction needed to set the marker in correct position + * @return -1 if the image passed is a not a valid marker, and its id in case it really is a marker + */ + static int detect(const cv::Mat &in,int &nRotations); + + /**Similar to createMarkerImage. Instead of returning a visible image, returns a 8UC1 matrix of 0s and 1s with the marker info + */ + static cv::Mat getMarkerMat(int id) throw (cv::Exception); + + + /**Creates a printable image of a board + * @param gridSize grid layout (numer of sqaures in x and Y) + * @param MarkerSize size of markers sides in pixels + * @param MarkerDistance distance between the markers + * @param TInfo output + * @param excludedIds set of ids excluded from the board + */ + static cv::Mat createBoardImage( cv::Size gridSize,int MarkerSize,int MarkerDistance, BoardConfiguration& TInfo ,vector<int> *excludedIds=NULL ) throw (cv::Exception); + + + /**Creates a printable image of a board in chessboard_like manner + * @param gridSize grid layout (numer of sqaures in x and Y) + * @param MarkerSize size of markers sides in pixels + * @param TInfo output + * @param setDataCentered indicates if the center is set at the center of the board. Otherwise it is the left-upper corner + * + */ + static cv::Mat createBoardImage_ChessBoard( cv::Size gridSize,int MarkerSize, BoardConfiguration& TInfo ,bool setDataCentered=true ,vector<int> *excludedIds=NULL) throw (cv::Exception); + + /**Creates a printable image of a board in a frame fashion + * @param gridSize grid layout (numer of sqaures in x and Y) + * @param MarkerSize size of markers sides in pixels + * @param MarkerDistance distance between the markers + * @param TInfo output + * @param setDataCentered indicates if the center is set at the center of the board. Otherwise it is the left-upper corner + * + */ + static cv::Mat createBoardImage_Frame( cv::Size gridSize,int MarkerSize,int MarkerDistance, BoardConfiguration& TInfo ,bool setDataCentered=true,vector<int> *excludedIds=NULL ) throw (cv::Exception); + +private: + + static vector<int> getListOfValidMarkersIds_random(int nMarkers,vector<int> *excluded) throw (cv::Exception); + static cv::Mat rotate(const cv::Mat & in); + static int hammDistMarker(cv::Mat bits); + static int analyzeMarkerImage(cv::Mat &grey,int &nRotations); + static bool correctHammMarker(cv::Mat &bits); +}; + +} + +#endif diff --git a/ftnoir_tracker_aruco/include/board.h b/ftnoir_tracker_aruco/include/board.h new file mode 100644 index 00000000..c1d79292 --- /dev/null +++ b/ftnoir_tracker_aruco/include/board.h @@ -0,0 +1,168 @@ +/***************************** +Copyright 2011 Rafael Muñoz Salinas. All rights reserved. + +Redistribution and use in source and binary forms, with or without modification, are +permitted provided that the following conditions are met: + + 1. Redistributions of source code must retain the above copyright notice, this list of + conditions and the following disclaimer. + + 2. Redistributions in binary form must reproduce the above copyright notice, this list + of conditions and the following disclaimer in the documentation and/or other materials + provided with the distribution. + +THIS SOFTWARE IS PROVIDED BY Rafael Muñoz Salinas ''AS IS'' AND ANY EXPRESS OR IMPLIED +WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND +FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL Rafael Muñoz Salinas OR +CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR +CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR +SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON +ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING +NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF +ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. + +The views and conclusions contained in the software and documentation are those of the +authors and should not be interpreted as representing official policies, either expressed +or implied, of Rafael Muñoz Salinas. +********************************/ +#ifndef _Aruco_board_h +#define _Aruco_board_h +#include <opencv2/opencv.hpp> +#include <string> +#include <vector> +#include "exports.h" +#include "marker.h" +using namespace std; +namespace aruco { +/** + * 3d representation of a marker + */ +struct ARUCO_EXPORTS MarkerInfo:public vector<cv::Point3f> { + MarkerInfo() {} + MarkerInfo(int _id) {id=_id; } + MarkerInfo(const MarkerInfo&MI): vector<cv::Point3f>(MI){id=MI.id; } + MarkerInfo & operator=(const MarkerInfo&MI){ + vector<cv::Point3f> ::operator=(MI); + id=MI.id; + return *this; + } + int id;//maker id +}; + +/**\brief This class defines a board with several markers. + * A Board contains several markers so that they are more robustly detected. + * + * In general, a board is a set of markers. So BoardConfiguration is only a list + * of the id of the markers along with the position of their corners. + * + * The position of the corners can be specified either in pixels (in a non-specific size) or in meters. + * The first is the typical case in which you generate the image of board and the print it. Since you do not know in advance the real + * size of the markers, their corners are specified in pixels, and then, the translation to meters can be made once you know the real size. + * + * On the other hand, you may want to have the information of your boards in meters. The BoardConfiguration allows you to do so. + * + * The point is in the mInfoType variable. It can be either PIX or METERS according to your needs. + * +*/ + + +class ARUCO_EXPORTS BoardConfiguration: public vector<MarkerInfo> +{ + friend class Board; +public: + enum MarkerInfoType {NONE=-1,PIX=0,METERS=1};//indicates if the data in MakersInfo is expressed in meters or in pixels so as to do conversion internally + //variable indicates if the data in MakersInfo is expressed in meters or in pixels so as to do conversion internally + int mInfoType; + /** + */ + BoardConfiguration(); + + /** + */ + BoardConfiguration(const BoardConfiguration &T); + + /** + */ + BoardConfiguration & operator=(const BoardConfiguration &T); + /**Saves the board info to a file + */ + void saveToFile(string sfile)throw (cv::Exception); + /**Reads board info from a file + */ + void readFromFile(string sfile)throw (cv::Exception); + /**Indicates if the corners are expressed in meters + */ + bool isExpressedInMeters()const { + return mInfoType==METERS; + } + /**Indicates if the corners are expressed in meters + */ + bool isExpressedInPixels()const { + return mInfoType==PIX; + } + /**Returns the index of the marker with id indicated, if is in the list + */ + int getIndexOfMarkerId(int id)const; + /**Returns the Info of the marker with id specified. If not in the set, throws exception + */ + const MarkerInfo& getMarkerInfo(int id)const throw (cv::Exception); + /**Set in the list passed the set of the ids + */ + void getIdList(vector<int> &ids,bool append=true)const; +private: + /**Saves the board info to a file + */ + void saveToFile(cv::FileStorage &fs)throw (cv::Exception); + /**Reads board info from a file + */ + void readFromFile(cv::FileStorage &fs)throw (cv::Exception); +}; + +/** +*/ +class ARUCO_EXPORTS Board:public vector<Marker> +{ + +public: + BoardConfiguration conf; + //matrices of rotation and translation respect to the camera + cv::Mat Rvec,Tvec; + /** + */ + Board() + { + Rvec.create(3,1,CV_32FC1); + Tvec.create(3,1,CV_32FC1); + for (int i=0;i<3;i++) + Tvec.at<float>(i,0)=Rvec.at<float>(i,0)=-999999; + } + + /**Given the extrinsic camera parameters returns the GL_MODELVIEW matrix for opengl. + * Setting this matrix, the reference corrdinate system will be set in this board + */ + void glGetModelViewMatrix(double modelview_matrix[16])throw(cv::Exception); + + /** + * Returns position vector and orientation quaternion for an Ogre scene node or entity. + * Use: + * ... + * Ogre::Vector3 ogrePos (position[0], position[1], position[2]); + * Ogre::Quaternion ogreOrient (orientation[0], orientation[1], orientation[2], orientation[3]); + * mySceneNode->setPosition( ogrePos ); + * mySceneNode->setOrientation( ogreOrient ); + * ... + */ + void OgreGetPoseParameters( double position[3], double orientation[4] )throw(cv::Exception); + + + /**Save this from a file + */ + void saveToFile(string filePath)throw(cv::Exception); + /**Read this from a file + */ + void readFromFile(string filePath)throw(cv::Exception); + +}; +} + +#endif diff --git a/ftnoir_tracker_aruco/include/boarddetector.h b/ftnoir_tracker_aruco/include/boarddetector.h new file mode 100644 index 00000000..a0ee2361 --- /dev/null +++ b/ftnoir_tracker_aruco/include/boarddetector.h @@ -0,0 +1,139 @@ +/***************************** +Copyright 2011 Rafael Muñoz Salinas. All rights reserved. + +Redistribution and use in source and binary forms, with or without modification, are +permitted provided that the following conditions are met: + + 1. Redistributions of source code must retain the above copyright notice, this list of + conditions and the following disclaimer. + + 2. Redistributions in binary form must reproduce the above copyright notice, this list + of conditions and the following disclaimer in the documentation and/or other materials + provided with the distribution. + +THIS SOFTWARE IS PROVIDED BY Rafael Muñoz Salinas ''AS IS'' AND ANY EXPRESS OR IMPLIED +WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND +FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL Rafael Muñoz Salinas OR +CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR +CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR +SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON +ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING +NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF +ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. + +The views and conclusions contained in the software and documentation are those of the +authors and should not be interpreted as representing official policies, either expressed +or implied, of Rafael Muñoz Salinas. +********************************/ +#ifndef _Aruco_BoardDetector_H +#define _Aruco_BoardDetector_H +#include <opencv2/opencv.hpp> +#include "exports.h" +#include "board.h" +#include "cameraparameters.h" +#include "markerdetector.h" +using namespace std; + +namespace aruco +{ + +/**\brief This class detects AR boards + * Version 1.2 + * There are two modes for board detection. + * First, the old way. (You first detect markers with MarkerDetector and then call to detect in this class. + * + * Second: New mode, marker detection is included in the class + * \code + + CameraParameters CP; + CP.readFromFile(path_cp) + BoardConfiguration BC; + BC.readFromFile(path_bc); + BoardDetector BD; + BD.setParams(BC,CP); //or only BD.setParams(BC) + //capture image + cv::Mat im; + capture_image(im); + + float prob=BD.detect(im); + if (prob>0.3) + CvDrawingUtils::draw3DAxis(im,BD.getDetectedBoard(),CP); + + \endcode + * +*/ +class ARUCO_EXPORTS BoardDetector +{ +public: + /** See discussion in @see enableRotateXAxis. + * Do not change unless you know what you are doing + */ + BoardDetector(bool setYPerperdicular=true); + + + /** + * Use if you plan to let this class to perform marker detection too + */ + void setParams(const BoardConfiguration &bc,const CameraParameters &cp, float markerSizeMeters=-1); + void setParams(const BoardConfiguration &bc); + /** + * Detect markers, and then, look for the board indicated in setParams() + * @return value indicating the likelihood of having found the marker + */ + float detect(const cv::Mat &im)throw (cv::Exception); + /**Returns a reference to the board detected + */ + Board & getDetectedBoard(){return _boardDetected;} + /**Returns a reference to the internal marker detector + */ + MarkerDetector &getMarkerDetector(){return _mdetector;} + /**Returns the vector of markers detected + */ + vector<Marker> &getDetectedMarkers(){return _vmarkers;} + + + //ALTERNATIVE DETECTION METHOD, BASED ON MARKERS PREVIOUSLY DETECTED + + /** Given the markers detected, determines if there is the board passed + * @param detectedMarkers result provided by aruco::ArMarkerDetector + * @param BConf the board you want to see if is present + * @param Bdetected output information of the detected board + * @param camMatrix camera matrix with intrinsics + * @param distCoeff camera distorsion coeff + * @param camMatrix intrinsic camera information. + * @param distCoeff camera distorsion coefficient. If set Mat() if is assumed no camera distorion + * @param markerSizeMeters size of the marker sides expressed in meters + * @return value indicating the likelihood of having found the marker + */ + float detect(const vector<Marker> &detectedMarkers,const BoardConfiguration &BConf, Board &Bdetected, cv::Mat camMatrix=cv::Mat(),cv::Mat distCoeff=cv::Mat(), float markerSizeMeters=-1 )throw (cv::Exception); + float detect(const vector<Marker> &detectedMarkers,const BoardConfiguration &BConf, Board &Bdetected,const CameraParameters &cp, float markerSizeMeters=-1 )throw (cv::Exception); + + + /** + * By default, the Y axis is set to point up. However this is not the default + * operation mode of opencv, which produces the Z axis pointing up instead. + * So, to achieve this change, we have to rotate the X axis. + */ + void setYPerperdicular(bool enable){_setYPerperdicular=enable;} + + + + +private: + void rotateXAxis(cv::Mat &rotation); + bool _setYPerperdicular; + + //-- Functionality to detect markers inside + bool _areParamsSet; + BoardConfiguration _bconf; + Board _boardDetected; + float _markerSize; + CameraParameters _camParams; + MarkerDetector _mdetector;//internal markerdetector + vector<Marker> _vmarkers;//markers detected in the call to : float detect(const cv::Mat &im); + +}; + +}; +#endif + diff --git a/ftnoir_tracker_aruco/include/cameraparameters.h b/ftnoir_tracker_aruco/include/cameraparameters.h new file mode 100644 index 00000000..c3381a74 --- /dev/null +++ b/ftnoir_tracker_aruco/include/cameraparameters.h @@ -0,0 +1,137 @@ +/***************************** +Copyright 2011 Rafael Muñoz Salinas. All rights reserved. + +Redistribution and use in source and binary forms, with or without modification, are +permitted provided that the following conditions are met: + + 1. Redistributions of source code must retain the above copyright notice, this list of + conditions and the following disclaimer. + + 2. Redistributions in binary form must reproduce the above copyright notice, this list + of conditions and the following disclaimer in the documentation and/or other materials + provided with the distribution. + +THIS SOFTWARE IS PROVIDED BY Rafael Muñoz Salinas ''AS IS'' AND ANY EXPRESS OR IMPLIED +WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND +FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL Rafael Muñoz Salinas OR +CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR +CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR +SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON +ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING +NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF +ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. + +The views and conclusions contained in the software and documentation are those of the +authors and should not be interpreted as representing official policies, either expressed +or implied, of Rafael Muñoz Salinas. +********************************/ +#ifndef _Aruco_CameraParameters_H +#define _Aruco_CameraParameters_H +#include "exports.h" +#include <opencv2/opencv.hpp> +#include <string> +using namespace std; +namespace aruco +{ +/**\brief Parameters of the camera + */ + +class ARUCO_EXPORTS CameraParameters +{ +public: + + // 3x3 matrix (fx 0 cx, 0 fy cy, 0 0 1) + cv::Mat CameraMatrix; + //4x1 matrix (k1,k2,p1,p2) + cv::Mat Distorsion; + //size of the image + cv::Size CamSize; + + /**Empty constructor + */ + CameraParameters() ; + /**Creates the object from the info passed + * @param cameraMatrix 3x3 matrix (fx 0 cx, 0 fy cy, 0 0 1) + * @param distorsionCoeff 4x1 matrix (k1,k2,p1,p2) + * @param size image size + */ + CameraParameters(cv::Mat cameraMatrix,cv::Mat distorsionCoeff,cv::Size size) throw(cv::Exception); + /**Sets the parameters + * @param cameraMatrix 3x3 matrix (fx 0 cx, 0 fy cy, 0 0 1) + * @param distorsionCoeff 4x1 matrix (k1,k2,p1,p2) + * @param size image size + */ + void setParams(cv::Mat cameraMatrix,cv::Mat distorsionCoeff,cv::Size size) throw(cv::Exception); + /**Copy constructor + */ + CameraParameters(const CameraParameters &CI) ; + + /**Indicates whether this object is valid + */ + bool isValid()const { + return CameraMatrix.rows!=0 && CameraMatrix.cols!=0 && Distorsion.rows!=0 && Distorsion.cols!=0 && CamSize.width!=-1 && CamSize.height!=-1; + } + /**Assign operator + */ + CameraParameters & operator=(const CameraParameters &CI); + /**Reads the camera parameters from a file generated using saveToFile. + */ + void readFromFile(string path)throw(cv::Exception); + /**Saves this to a file + */ + void saveToFile(string path,bool inXML=true)throw(cv::Exception); + + /**Reads from a YAML file generated with the opencv2.2 calibration utility + */ + void readFromXMLFile(string filePath)throw(cv::Exception); + + /**Adjust the parameters to the size of the image indicated + */ + void resize(cv::Size size)throw(cv::Exception); + + /**Returns the location of the camera in the reference system given by the rotation and translation vectors passed + * NOT TESTED + */ + static cv::Point3f getCameraLocation(cv::Mat Rvec,cv::Mat Tvec); + + /**Given the intrinsic camera parameters returns the GL_PROJECTION matrix for opengl. + * PLease NOTE that when using OpenGL, it is assumed no camera distorsion! So, if it is not true, you should have + * undistor image + * + * @param orgImgSize size of the original image + * @param size of the image/window where to render (can be different from the real camera image). Please not that it must be related to CamMatrix + * @param proj_matrix output projection matrix to give to opengl + * @param gnear,gfar: visible rendering range + * @param invert: indicates if the output projection matrix has to yield a horizontally inverted image because image data has not been stored in the order of glDrawPixels: bottom-to-top. + */ + void glGetProjectionMatrix( cv::Size orgImgSize, cv::Size size,double proj_matrix[16],double gnear,double gfar,bool invert=false )throw(cv::Exception); + + /** + * setup camera for an Ogre project. + * Use: + * ... + * Ogre::Matrix4 PM(proj_matrix[0], proj_matrix[1], ... , proj_matrix[15]); + * yourCamera->setCustomProjectionMatrix(true, PM); + * yourCamera->setCustomViewMatrix(true, Ogre::Matrix4::IDENTITY); + * ... + * As in OpenGL, it assumes no camera distorsion + */ + void OgreGetProjectionMatrix( cv::Size orgImgSize, cv::Size size,double proj_matrix[16],double gnear,double gfar,bool invert=false )throw(cv::Exception); + + +private: + //GL routines + + static void argConvGLcpara2( double cparam[3][4], int width, int height, double gnear, double gfar, double m[16], bool invert )throw(cv::Exception); + static int arParamDecompMat( double source[3][4], double cpara[3][4], double trans[3][4] )throw(cv::Exception); + static double norm( double a, double b, double c ); + static double dot( double a1, double a2, double a3, + double b1, double b2, double b3 ); + + +}; + +} +#endif + + diff --git a/ftnoir_tracker_aruco/include/cvdrawingutils.h b/ftnoir_tracker_aruco/include/cvdrawingutils.h new file mode 100644 index 00000000..ff67242f --- /dev/null +++ b/ftnoir_tracker_aruco/include/cvdrawingutils.h @@ -0,0 +1,52 @@ +/***************************** +Copyright 2011 Rafael Muñoz Salinas. All rights reserved. + +Redistribution and use in source and binary forms, with or without modification, are +permitted provided that the following conditions are met: + + 1. Redistributions of source code must retain the above copyright notice, this list of + conditions and the following disclaimer. + + 2. Redistributions in binary form must reproduce the above copyright notice, this list + of conditions and the following disclaimer in the documentation and/or other materials + provided with the distribution. + +THIS SOFTWARE IS PROVIDED BY Rafael Muñoz Salinas ''AS IS'' AND ANY EXPRESS OR IMPLIED +WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND +FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL Rafael Muñoz Salinas OR +CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR +CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR +SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON +ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING +NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF +ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. + +The views and conclusions contained in the software and documentation are those of the +authors and should not be interpreted as representing official policies, either expressed +or implied, of Rafael Muñoz Salinas. +********************************/ +#ifndef _ArUco_DrawUtils_H_ +#define _ArUco_DrawUtils_H_ +#include "exports.h" +#include "aruco.h" +namespace aruco +{ + /**\brief A set of functions to draw in opencv images + */ + class ARUCO_EXPORTS CvDrawingUtils + { + public: + + static void draw3dAxis(cv::Mat &Image,Marker &m,const CameraParameters &CP); + + static void draw3dCube(cv::Mat &Image,Marker &m,const CameraParameters &CP); + + static void draw3dAxis(cv::Mat &Image,Board &m,const CameraParameters &CP); + + static void draw3dCube(cv::Mat &Image,Board &m,const CameraParameters &CP); + + }; +}; + +#endif + diff --git a/ftnoir_tracker_aruco/include/exports.h b/ftnoir_tracker_aruco/include/exports.h new file mode 100644 index 00000000..154605ec --- /dev/null +++ b/ftnoir_tracker_aruco/include/exports.h @@ -0,0 +1,46 @@ +/***************************** +Copyright 2011 Rafael Muñoz Salinas. All rights reserved. + +Redistribution and use in source and binary forms, with or without modification, are +permitted provided that the following conditions are met: + + 1. Redistributions of source code must retain the above copyright notice, this list of + conditions and the following disclaimer. + + 2. Redistributions in binary form must reproduce the above copyright notice, this list + of conditions and the following disclaimer in the documentation and/or other materials + provided with the distribution. + +THIS SOFTWARE IS PROVIDED BY Rafael Muñoz Salinas ''AS IS'' AND ANY EXPRESS OR IMPLIED +WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND +FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL Rafael Muñoz Salinas OR +CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR +CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR +SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON +ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING +NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF +ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. + +The views and conclusions contained in the software and documentation are those of the +authors and should not be interpreted as representing official policies, either expressed +or implied, of Rafael Muñoz Salinas. +********************************/ + + + +#ifndef __OPENARUCO_CORE_TYPES_H__ +#define __OPENARUCO_CORE_TYPES_H__ + +#if !defined _CRT_SECURE_NO_DEPRECATE && _MSC_VER > 1300 +#define _CRT_SECURE_NO_DEPRECATE /* to avoid multiple Visual Studio 2005 warnings */ +#endif + + +#if (defined WIN32 || defined _WIN32 || defined WINCE) && defined DSO_EXPORTS + #define ARUCO_EXPORTS __declspec(dllexport) +#else + #define ARUCO_EXPORTS +#endif + + +#endif diff --git a/ftnoir_tracker_aruco/include/marker.h b/ftnoir_tracker_aruco/include/marker.h new file mode 100644 index 00000000..dc6bb28c --- /dev/null +++ b/ftnoir_tracker_aruco/include/marker.h @@ -0,0 +1,143 @@ +/***************************** +Copyright 2011 Rafael Muñoz Salinas. All rights reserved. + +Redistribution and use in source and binary forms, with or without modification, are +permitted provided that the following conditions are met: + + 1. Redistributions of source code must retain the above copyright notice, this list of + conditions and the following disclaimer. + + 2. Redistributions in binary form must reproduce the above copyright notice, this list + of conditions and the following disclaimer in the documentation and/or other materials + provided with the distribution. + +THIS SOFTWARE IS PROVIDED BY Rafael Muñoz Salinas ''AS IS'' AND ANY EXPRESS OR IMPLIED +WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND +FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL Rafael Muñoz Salinas OR +CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR +CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR +SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON +ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING +NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF +ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. + +The views and conclusions contained in the software and documentation are those of the +authors and should not be interpreted as representing official policies, either expressed +or implied, of Rafael Muñoz Salinas. +********************************/ +#ifndef _Aruco_Marker_H +#define _Aruco_Marker_H +#include <vector> +#include <iostream> +#include <opencv2/opencv.hpp> +#include "exports.h" +#include "cameraparameters.h" +using namespace std; +namespace aruco { +/**\brief This class represents a marker. It is a vector of the fours corners ot the marker + * + */ + +class ARUCO_EXPORTS Marker: public std::vector<cv::Point2f> +{ +public: + //id of the marker + int id; + //size of the markers sides in meters + float ssize; + //matrices of rotation and translation respect to the camera + cv::Mat Rvec,Tvec; + + /** + */ + Marker(); + /** + */ + Marker(const Marker &M); + /** + */ + Marker(const std::vector<cv::Point2f> &corners,int _id=-1); + /** + */ + ~Marker() {} + /**Indicates if this object is valid + */ + bool isValid()const{return id!=-1 && size()==4;} + + /**Draws this marker in the input image + */ + void draw(cv::Mat &in, cv::Scalar color, int lineWidth=1,bool writeId=true)const; + + /**Calculates the extrinsics (Rvec and Tvec) of the marker with respect to the camera + * @param markerSize size of the marker side expressed in meters + * @param CP parmeters of the camera + * @param setYPerperdicular If set the Y axis will be perpendicular to the surface. Otherwise, it will be the Z axis + */ + void calculateExtrinsics(float markerSize,const CameraParameters &CP,bool setYPerperdicular=true)throw(cv::Exception); + /**Calculates the extrinsics (Rvec and Tvec) of the marker with respect to the camera + * @param markerSize size of the marker side expressed in meters + * @param CameraMatrix matrix with camera parameters (fx,fy,cx,cy) + * @param Distorsion matrix with distorsion parameters (k1,k2,p1,p2) + * @param setYPerperdicular If set the Y axis will be perpendicular to the surface. Otherwise, it will be the Z axis + */ + void calculateExtrinsics(float markerSize,cv::Mat CameraMatrix,cv::Mat Distorsion=cv::Mat(),bool setYPerperdicular=true)throw(cv::Exception); + + /**Given the extrinsic camera parameters returns the GL_MODELVIEW matrix for opengl. + * Setting this matrix, the reference coordinate system will be set in this marker + */ + void glGetModelViewMatrix( double modelview_matrix[16])throw(cv::Exception); + + /** + * Returns position vector and orientation quaternion for an Ogre scene node or entity. + * Use: + * ... + * Ogre::Vector3 ogrePos (position[0], position[1], position[2]); + * Ogre::Quaternion ogreOrient (orientation[0], orientation[1], orientation[2], orientation[3]); + * mySceneNode->setPosition( ogrePos ); + * mySceneNode->setOrientation( ogreOrient ); + * ... + */ + void OgreGetPoseParameters( double position[3], double orientation[4] )throw(cv::Exception); + + /**Returns the centroid of the marker + */ + cv::Point2f getCenter()const; + /**Returns the perimeter of the marker + */ + float getPerimeter()const; + /**Returns the area + */ + float getArea()const; + /** + */ + /** + */ + friend bool operator<(const Marker &M1,const Marker&M2) + { + return M1.id<M2.id; + } + /** + */ + friend ostream & operator<<(ostream &str,const Marker &M) + { + str<<M.id<<"="; + for (int i=0;i<4;i++) + str<<"("<<M[i].x<< ","<<M[i].y<<") "; + str<<"Txyz="; + for (int i=0;i<3;i++) + str<<M.Tvec.ptr<float>(0)[i]<<" "; + str<<"Rxyz="; + for (int i=0;i<3;i++) + str<<M.Rvec.ptr<float>(0)[i]<<" "; + + return str; + } + + +private: + void rotateXAxis(cv::Mat &rotation); + +}; + +} +#endif diff --git a/ftnoir_tracker_aruco/include/markerdetector.h b/ftnoir_tracker_aruco/include/markerdetector.h new file mode 100644 index 00000000..35369cea --- /dev/null +++ b/ftnoir_tracker_aruco/include/markerdetector.h @@ -0,0 +1,358 @@ +/***************************** +Copyright 2011 Rafael Muñoz Salinas. All rights reserved. + +Redistribution and use in source and binary forms, with or without modification, are +permitted provided that the following conditions are met: + + 1. Redistributions of source code must retain the above copyright notice, this list of + conditions and the following disclaimer. + + 2. Redistributions in binary form must reproduce the above copyright notice, this list + of conditions and the following disclaimer in the documentation and/or other materials + provided with the distribution. + +THIS SOFTWARE IS PROVIDED BY Rafael Muñoz Salinas ''AS IS'' AND ANY EXPRESS OR IMPLIED +WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND +FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL Rafael Muñoz Salinas OR +CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR +CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR +SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON +ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING +NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF +ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. + +The views and conclusions contained in the software and documentation are those of the +authors and should not be interpreted as representing official policies, either expressed +or implied, of Rafael Muñoz Salinas. +********************************/ +#ifndef _ARUCO_MarkerDetector_H +#define _ARUCO_MarkerDetector_H +#include <opencv2/opencv.hpp> +#include <cstdio> +#include <iostream> +#include "cameraparameters.h" +#include "exports.h" +#include "marker.h" +using namespace std; + +namespace aruco +{ + +/**\brief Main class for marker detection + * + */ +class ARUCO_EXPORTS MarkerDetector +{ + //Represent a candidate to be a maker + class MarkerCandidate: public Marker{ + public: + MarkerCandidate(){} + MarkerCandidate(const Marker &M): Marker(M){} + MarkerCandidate(const MarkerCandidate &M): Marker(M){ + contour=M.contour; + idx=M.idx; + } + MarkerCandidate & operator=(const MarkerCandidate &M){ + (*(Marker*)this)=(*(Marker*)&M); + contour=M.contour; + idx=M.idx; + } + + vector<cv::Point> contour;//all the points of its contour + int idx;//index position in the global contour list + }; +public: + + /** + * See + */ + MarkerDetector(); + + /** + */ + ~MarkerDetector(); + + /**Detects the markers in the image passed + * + * If you provide information about the camera parameters and the size of the marker, then, the extrinsics of the markers are detected + * + * @param input input color image + * @param detectedMarkers output vector with the markers detected + * @param camMatrix intrinsic camera information. + * @param distCoeff camera distorsion coefficient. If set Mat() if is assumed no camera distorion + * @param markerSizeMeters size of the marker sides expressed in meters + * @param setYPerperdicular If set the Y axis will be perpendicular to the surface. Otherwise, it will be the Z axis + */ + void detect(const cv::Mat &input,std::vector<Marker> &detectedMarkers,cv::Mat camMatrix=cv::Mat(),cv::Mat distCoeff=cv::Mat(),float markerSizeMeters=-1,bool setYPerperdicular=true) throw (cv::Exception); + /**Detects the markers in the image passed + * + * If you provide information about the camera parameters and the size of the marker, then, the extrinsics of the markers are detected + * + * @param input input color image + * @param detectedMarkers output vector with the markers detected + * @param camParams Camera parameters + * @param markerSizeMeters size of the marker sides expressed in meters + * @param setYPerperdicular If set the Y axis will be perpendicular to the surface. Otherwise, it will be the Z axis + */ + void detect(const cv::Mat &input,std::vector<Marker> &detectedMarkers, CameraParameters camParams,float markerSizeMeters=-1,bool setYPerperdicular=true) throw (cv::Exception); + + /**This set the type of thresholding methods available + */ + + enum ThresholdMethods {FIXED_THRES,ADPT_THRES,CANNY}; + + + + /**Sets the threshold method + */ + void setThresholdMethod(ThresholdMethods m) { + _thresMethod=m; + } + /**Returns the current threshold method + */ + ThresholdMethods getThresholdMethod()const { + return _thresMethod; + } + /** + * Set the parameters of the threshold method + * We are currently using the Adptive threshold ee opencv doc of adaptiveThreshold for more info + * @param param1: blockSize of the pixel neighborhood that is used to calculate a threshold value for the pixel + * @param param2: The constant subtracted from the mean or weighted mean + */ + void setThresholdParams(double param1,double param2) { + _thresParam1=param1; + _thresParam2=param2; + } + /** + * Set the parameters of the threshold method + * We are currently using the Adptive threshold ee opencv doc of adaptiveThreshold for more info + * param1: blockSize of the pixel neighborhood that is used to calculate a threshold value for the pixel + * param2: The constant subtracted from the mean or weighted mean + */ + void getThresholdParams(double ¶m1,double ¶m2)const { + param1=_thresParam1; + param2=_thresParam2; + } + + + /**Returns a reference to the internal image thresholded. It is for visualization purposes and to adjust manually + * the parameters + */ + const cv::Mat & getThresholdedImage() { + return thres; + } + /**Methods for corner refinement + */ + enum CornerRefinementMethod {NONE,HARRIS,SUBPIX,LINES}; + /** + */ + void setCornerRefinementMethod(CornerRefinementMethod method) { + _cornerMethod=method; + } + /** + */ + CornerRefinementMethod getCornerRefinementMethod()const { + return _cornerMethod; + } + /**Specifies the min and max sizes of the markers as a fraction of the image size. By size we mean the maximum + * of cols and rows. + * @param min size of the contour to consider a possible marker as valid (0,1] + * @param max size of the contour to consider a possible marker as valid [0,1) + * + */ + void setMinMaxSize(float min=0.03,float max=0.5)throw(cv::Exception); + + /**reads the min and max sizes employed + * @param min output size of the contour to consider a possible marker as valid (0,1] + * @param max output size of the contour to consider a possible marker as valid [0,1) + * + */ + void getMinMaxSize(float &min,float &max){min=_minSize;max=_maxSize;} + + /**Enables/Disables erosion process that is REQUIRED for chessboard like boards. + * By default, this property is enabled + */ + void enableErosion(bool enable){_doErosion=enable;} + + /** + * Specifies a value to indicate the required speed for the internal processes. If you need maximum speed (at the cost of a lower detection rate), + * use the value 3, If you rather a more precise and slow detection, set it to 0. + * + * Actually, the main differences are that in highspeed mode, we employ setCornerRefinementMethod(NONE) and internally, we use a small canonical + * image to detect the marker. In low speed mode, we use setCornerRefinementMethod(HARRIS) and a bigger size for the canonical marker image + */ + void setDesiredSpeed(int val); + /** + */ + int getDesiredSpeed()const { + return _speed; + } + + /** + * Allows to specify the function that identifies a marker. Therefore, you can create your own type of markers different from these + * employed by default in the library. + * The marker function must have the following structure: + * + * int myMarkerIdentifier(const cv::Mat &in,int &nRotations); + * + * The marker function receives the image 'in' with the region that migh contain one of your markers. These are the rectangular regions with black + * in the image. + * + * As output your marker function must indicate the following information. First, the output parameter nRotations must indicate how many times the marker + * must be rotated clockwise 90 deg to be in its ideal position. (The way you would see it when you print it). This is employed to know + * always which is the corner that acts as reference system. Second, the function must return -1 if the image does not contains one of your markers, and its id otherwise. + * + */ + void setMakerDetectorFunction(int (* markerdetector_func)(const cv::Mat &in,int &nRotations) ) { + markerIdDetector_ptrfunc=markerdetector_func; + } + + /** Use an smaller version of the input image for marker detection. + * If your marker is small enough, you can employ an smaller image to perform the detection without noticeable reduction in the precision. + * Internally, we are performing a pyrdown operation + * + * @param level number of times the image size is divided by 2. Internally, we are performing a pyrdown. + */ + void pyrDown(unsigned int level){pyrdown_level=level;} + + ///------------------------------------------------- + /// Methods you may not need + /// Thesde methods do the hard work. They have been set public in case you want to do customizations + ///------------------------------------------------- + + /** + * Thesholds the passed image with the specified method. + */ + void thresHold(int method,const cv::Mat &grey,cv::Mat &thresImg,double param1=-1,double param2=-1)throw(cv::Exception); + /** + * Detection of candidates to be markers, i.e., rectangles. + * This function returns in candidates all the rectangles found in a thresolded image + */ + void detectRectangles(const cv::Mat &thresImg,vector<std::vector<cv::Point2f> > & candidates); + + /**Returns a list candidates to be markers (rectangles), for which no valid id was found after calling detectRectangles + */ + const vector<std::vector<cv::Point2f> > &getCandidates() { + return _candidates; + } + + /**Given the iput image with markers, creates an output image with it in the canonical position + * @param in input image + * @param out image with the marker + * @param size of out + * @param points 4 corners of the marker in the image in + * @return true if the operation succeed + */ + bool warp(cv::Mat &in,cv::Mat &out,cv::Size size, std::vector<cv::Point2f> points)throw (cv::Exception); + + + + /** Refine MarkerCandidate Corner using LINES method + * @param candidate candidate to refine corners + */ + void refineCandidateLines(MarkerCandidate &candidate); + + + /**DEPRECATED!!! Use the member function in CameraParameters + * + * Given the intrinsic camera parameters returns the GL_PROJECTION matrix for opengl. + * PLease NOTE that when using OpenGL, it is assumed no camera distorsion! So, if it is not true, you should have + * undistor image + * + * @param CamMatrix arameters of the camera specified. + * @param orgImgSize size of the original image + * @param size of the image/window where to render (can be different from the real camera image). Please not that it must be related to CamMatrix + * @param proj_matrix output projection matrix to give to opengl + * @param gnear,gfar: visible rendering range + * @param invert: indicates if the output projection matrix has to yield a horizontally inverted image because image data has not been stored in the order of glDrawPixels: bottom-to-top. + */ + static void glGetProjectionMatrix( CameraParameters & CamMatrix,cv::Size orgImgSize, cv::Size size,double proj_matrix[16],double gnear,double gfar,bool invert=false )throw(cv::Exception); + +private: + + bool _enableCylinderWarp; + bool warp_cylinder ( cv::Mat &in,cv::Mat &out,cv::Size size, MarkerCandidate& mc ) throw ( cv::Exception ); + /** + * Detection of candidates to be markers, i.e., rectangles. + * This function returns in candidates all the rectangles found in a thresolded image + */ + void detectRectangles(const cv::Mat &thresImg,vector<MarkerCandidate> & candidates); + //Current threshold method + ThresholdMethods _thresMethod; + //Threshold parameters + double _thresParam1,_thresParam2; + //Current corner method + CornerRefinementMethod _cornerMethod; + //minimum and maximum size of a contour lenght + float _minSize,_maxSize; + //Speed control + int _speed; + int _markerWarpSize; + bool _doErosion; + //vectr of candidates to be markers. This is a vector with a set of rectangles that have no valid id + vector<std::vector<cv::Point2f> > _candidates; + //level of image reduction + int pyrdown_level; + //Images + cv::Mat grey,thres,thres2,reduced; + //pointer to the function that analizes a rectangular region so as to detect its internal marker + int (* markerIdDetector_ptrfunc)(const cv::Mat &in,int &nRotations); + + /** + */ + bool isInto(cv::Mat &contour,std::vector<cv::Point2f> &b); + /** + */ + int perimeter(std::vector<cv::Point2f> &a); + + +// //GL routines +// +// static void argConvGLcpara2( double cparam[3][4], int width, int height, double gnear, double gfar, double m[16], bool invert )throw(cv::Exception); +// static int arParamDecompMat( double source[3][4], double cpara[3][4], double trans[3][4] )throw(cv::Exception); +// static double norm( double a, double b, double c ); +// static double dot( double a1, double a2, double a3, +// double b1, double b2, double b3 ); +// + + //detection of the + void findBestCornerInRegion_harris(const cv::Mat & grey,vector<cv::Point2f> & Corners,int blockSize); + + + // auxiliar functions to perform LINES refinement + void interpolate2Dline( const vector< cv::Point > &inPoints, cv::Point3f &outLine); + cv::Point2f getCrossPoint(const cv::Point3f& line1, const cv::Point3f& line2); + + + /**Given a vector vinout with elements and a boolean vector indicating the lements from it to remove, + * this function remove the elements + * @param vinout + * @param toRemove + */ + template<typename T> + void removeElements(vector<T> & vinout,const vector<bool> &toRemove) + { + //remove the invalid ones by setting the valid in the positions left by the invalids + size_t indexValid=0; + for (size_t i=0;i<toRemove.size();i++) { + if (!toRemove[i]) { + if (indexValid!=i) vinout[indexValid]=vinout[i]; + indexValid++; + } + } + vinout.resize(indexValid); + } + + //graphical debug + void drawApproxCurve(cv::Mat &in,std::vector<cv::Point> &approxCurve ,cv::Scalar color); + void drawContour(cv::Mat &in,std::vector<cv::Point> &contour,cv::Scalar ); + void drawAllContours(cv::Mat input, std::vector<std::vector<cv::Point> > &contours); + void draw(cv::Mat out,const std::vector<Marker> &markers ); + +}; + + + + +}; +#endif diff --git a/ftnoir_tracker_aruco/video_widget.cpp b/ftnoir_tracker_aruco/video_widget.cpp new file mode 100644 index 00000000..84cba6a3 --- /dev/null +++ b/ftnoir_tracker_aruco/video_widget.cpp @@ -0,0 +1,30 @@ +/* Copyright (c) 2012 Patrick Ruoff + * + * Permission to use, copy, modify, and/or distribute this software for any + * purpose with or without fee is hereby granted, provided that the above + * copyright notice and this permission notice appear in all copies. + */ + +#include "video_widget.h" + +#include <QDebug> + +using namespace std; + +void VideoWidget::update_image(unsigned char *frame, int width, int height) +{ + QMutexLocker foo(&mtx); + QImage qframe = QImage(width, height, QImage::Format_RGB888); + uchar* data = qframe.bits(); + const int pitch = qframe.bytesPerLine(); + for (int y = 0; y < height; y++) + for (int x = 0; x < width; x++) + { + const int pos = 3 * (y*width + x); + data[y * pitch + x * 3 + 0] = frame[pos + 2]; + data[y * pitch + x * 3 + 1] = frame[pos + 1]; + data[y * pitch + x * 3 + 2] = frame[pos + 0]; + } + qframe = qframe.scaled(size(), Qt::IgnoreAspectRatio, Qt::FastTransformation); + pixmap = QPixmap::fromImage(qframe); +} diff --git a/ftnoir_tracker_aruco/video_widget.h b/ftnoir_tracker_aruco/video_widget.h new file mode 100644 index 00000000..87b6278a --- /dev/null +++ b/ftnoir_tracker_aruco/video_widget.h @@ -0,0 +1,39 @@ +/* Copyright (c) 2012 Patrick Ruoff + * + * Permission to use, copy, modify, and/or distribute this software for any + * purpose with or without fee is hereby granted, provided that the above + * copyright notice and this permission notice appear in all copies. + */ + +#ifndef VIDEOWIDGET_H +#define VIDEOWIDGET_H + +#include <QTime> +#include <QWidget> +#include <QMutex> +#include <QMutexLocker> +#include <QLabel> +#include <QPainter> +#include <QPaintEvent> + +// ---------------------------------------------------------------------------- +class VideoWidget : public QWidget +{ + Q_OBJECT + +public: + VideoWidget(QWidget *parent) : QWidget(parent), mtx() { + } + void update_image(unsigned char* frame, int width, int height); +protected slots: + void paintEvent( QPaintEvent* e ) { + QMutexLocker foo(&mtx); + QPainter painter(this); + painter.drawPixmap(e->rect(), pixmap, e->rect()); + } +private: + QMutex mtx; + QPixmap pixmap; +}; + +#endif // VIDEOWIDGET_H |