/******************************************************************************** * FaceTrackNoIR This program is a private project of the some enthusiastic * * gamers from Holland, who don't like to pay much for * * head-tracking. * * * * Copyright (C) 2011 Wim Vriend (Developing) * * Ron Hendriks (Researching and Testing) * * * * Homepage * * * * This program is free software; you can redistribute it and/or modify it * * under the terms of the GNU General Public License as published by the * * Free Software Foundation; either version 3 of the License, or (at your * * option) any later version. * * * * This program is distributed in the hope that it will be useful, but * * WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY * * or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for * * more details. * * * * You should have received a copy of the GNU General Public License along * * with this program; if not, see . * *********************************************************************************/ #include "facetracknoir.h" #include "shortcuts.h" #include "tracker.h" #include "curve-config.h" #include "opentrack-version.h" #include #if defined(_WIN32) # include # include #endif #if defined(__APPLE__) # define SONAME "dylib" #elif defined(_WIN32) # define SONAME "dll" #else # define SONAME "so" #endif #include #ifdef _MSC_VER # define LIB_PREFIX "" #else # define LIB_PREFIX "lib" #endif #if defined(__unix) || defined(__linux) || defined(__APPLE__) # include #endif static bool get_metadata(DynamicLibrary* lib, QString& longName, QIcon& icon) { Metadata* meta; if (!lib->Metadata || ((meta = lib->Metadata()), !meta)) return false; meta->getFullName(&longName); meta->getIcon(&icon); delete meta; return true; } static void fill_combobox(const QString& filter, QList& list, QComboBox* cbx, QComboBox* cbx2) { QDir settingsDir( QCoreApplication::applicationDirPath() ); QStringList filenames = settingsDir.entryList( QStringList() << (LIB_PREFIX + filter + SONAME), QDir::Files, QDir::Name ); for ( int i = 0; i < filenames.size(); i++) { QIcon icon; QString longName; QString str = filenames.at(i); DynamicLibrary* lib = new DynamicLibrary(str); qDebug() << "Loading" << str; std::cout.flush(); if (!get_metadata(lib, longName, icon)) { delete lib; continue; } list.push_back(lib); cbx->addItem(icon, longName); if (cbx2) cbx2->addItem(icon, longName); } } FaceTrackNoIR::FaceTrackNoIR(QWidget *parent) : QMainWindow(parent), #if defined(_WIN32) keybindingWorker(NULL), #else keyCenter(this), keyToggle(this), #endif b(bundle("opentrack-ui")), s(b), pose(std::vector{&s.a_x, &s.a_y, &s.a_z, &s.a_yaw, &s.a_pitch, &s.a_roll}), timUpdateHeadPose(this), pTrackerDialog(NULL), pSecondTrackerDialog(NULL), pProtocolDialog(NULL), pFilterDialog(NULL), kbd_quit(QKeySequence("Ctrl+Q"), this), looping(false) { ui.setupUi(this); setFixedSize(size()); _keyboard_shortcuts = 0; _curve_config = 0; tracker = 0; CurveConfigurationDialog* ccd; if (!_curve_config) { ccd = new CurveConfigurationDialog( this, this ); _curve_config = ccd; } else { ccd = dynamic_cast(_curve_config); } QDir::setCurrent(QCoreApplication::applicationDirPath()); fill_profile_cbx(); connect(ui.btnLoad, SIGNAL(clicked()), this, SLOT(open())); connect(ui.btnSave, SIGNAL(clicked()), this, SLOT(save())); connect(ui.btnSaveAs, SIGNAL(clicked()), this, SLOT(saveAs())); connect(ui.btnEditCurves, SIGNAL(clicked()), this, SLOT(showCurveConfiguration())); connect(ui.btnShortcuts, SIGNAL(clicked()), this, SLOT(showKeyboardShortcuts())); connect(ui.btnShowEngineControls, SIGNAL(clicked()), this, SLOT(showTrackerSettings())); connect(ui.btnShowSecondTrackerSettings, SIGNAL(clicked()), this, SLOT(showSecondTrackerSettings())); connect(ui.btnShowServerControls, SIGNAL(clicked()), this, SLOT(showServerControls())); connect(ui.btnShowFilterControls, SIGNAL(clicked()), this, SLOT(showFilterControls())); ui.cbxSecondTrackerSource->addItem(QIcon(), ""); dlopen_filters.push_back((DynamicLibrary*) NULL); ui.iconcomboFilter->addItem(QIcon(), ""); fill_combobox("opentrack-proto-*.", dlopen_protocols, ui.iconcomboProtocol, NULL); fill_combobox("opentrack-tracker-*.", dlopen_trackers, ui.iconcomboTrackerSource, ui.cbxSecondTrackerSource); fill_combobox("opentrack-filter-*.", dlopen_filters, ui.iconcomboFilter, NULL); tie_setting(s.tracker_dll, ui.iconcomboTrackerSource); tie_setting(s.tracker2_dll, ui.cbxSecondTrackerSource); tie_setting(s.protocol_dll, ui.iconcomboProtocol); tie_setting(s.filter_dll, ui.iconcomboFilter); connect(ui.btnStartTracker, SIGNAL(clicked()), this, SLOT(startTracker())); connect(ui.btnStopTracker, SIGNAL(clicked()), this, SLOT(stopTracker())); GetCameraNameDX(); connect(ui.iconcomboProfile, SIGNAL(currentIndexChanged(int)), this, SLOT(profileSelected(int))); connect(&timUpdateHeadPose, SIGNAL(timeout()), this, SLOT(showHeadPose())); #ifndef _WIN32 connect(&keyCenter, SIGNAL(activated()), this, SLOT(shortcutRecentered())); connect(&keyToggle, SIGNAL(activated()), this, SLOT(shortcutToggled())); #endif connect(&kbd_quit, SIGNAL(activated()), this, SLOT(exit())); kbd_quit.setEnabled(true); } FaceTrackNoIR::~FaceTrackNoIR() { stopTracker(); save(); if (Libraries) delete Libraries; } QFrame* FaceTrackNoIR::get_video_widget() { return ui.video_frame; } void FaceTrackNoIR::GetCameraNameDX() { #if defined(_WIN32) ui.cameraName->setText("No video-capturing device was found in your system: check if it's connected!"); // 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)) { qDebug() << "GetWDM says: CoCreateInstance Failed!"; return; } qDebug() << "GetWDM says: CoCreateInstance succeeded!"; // Obtain a class enumerator for the video compressor category. IEnumMoniker *pEnumCat = NULL; hr = pSysDevEnum->CreateClassEnumerator(CLSID_VideoInputDeviceCategory, &pEnumCat, 0); if (hr == S_OK) { qDebug() << "GetWDM says: CreateClassEnumerator succeeded!"; IMoniker *pMoniker = NULL; ULONG cFetched; if (pEnumCat->Next(1, &pMoniker, &cFetched) == S_OK) { IPropertyBag *pPropBag; hr = pMoniker->BindToStorage(0, 0, IID_IPropertyBag, (void **)&pPropBag); if (SUCCEEDED(hr)) { VARIANT varName; VariantInit(&varName); hr = pPropBag->Read(L"FriendlyName", &varName, 0); if (SUCCEEDED(hr)) { QString str((QChar*)varName.bstrVal, wcslen(varName.bstrVal)); qDebug() << "GetWDM says: Moniker found:" << str; ui.cameraName->setText(str); } VariantClear(&varName); 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) { ui.cameraName->setText(QString(buf)); break; } } #endif } void FaceTrackNoIR::open() { QFileDialog dialog(this); dialog.setFileMode(QFileDialog::ExistingFile); QString fileName = dialog.getOpenFileName( this, tr("Open the settings file"), QCoreApplication::applicationDirPath() + "/settings/", tr("Settings file (*.ini);;All Files (*)"), NULL); if (! fileName.isEmpty() ) { { QSettings settings("opentrack"); settings.setValue ("SettingsFile", QFileInfo(fileName).absoluteFilePath()); } looping = true; fill_profile_cbx(); loadSettings(); looping = false; } } void FaceTrackNoIR::save() { b->save(); QSettings settings("opentrack"); QString currentFile = settings.value("SettingsFile", QCoreApplication::applicationDirPath() + "/settings/default.ini") .toString(); #if defined(__unix) || defined(__linux) QByteArray bytes = QFile::encodeName(currentFile); const char* filename_as_asciiz = bytes.constData(); if (access(filename_as_asciiz, R_OK | W_OK)) { QMessageBox::warning(this, "Something went wrong", "Check permissions and ownership for your .ini file!", QMessageBox::Ok, QMessageBox::NoButton); } #endif } void FaceTrackNoIR::saveAs() { looping = true; QSettings settings("opentrack"); QString oldFile = settings.value ( "SettingsFile", QCoreApplication::applicationDirPath() + "/settings/default.ini" ).toString(); QString fileName = QFileDialog::getSaveFileName(this, tr("Save file"), oldFile, tr("Settings file (*.ini);;All Files (*)")); if (!fileName.isEmpty()) { QFileInfo newFileInfo ( fileName ); if ((newFileInfo.exists()) && (oldFile != fileName)) { QFile newFileFile ( fileName ); newFileFile.remove(); } QFileInfo oldFileInfo ( oldFile ); if (oldFileInfo.exists()) { QFile oldFileFile ( oldFile ); oldFileFile.copy( fileName ); } settings.setValue ("SettingsFile", fileName); save(); } looping = false; fill_profile_cbx(); } void FaceTrackNoIR::loadSettings() { b->reload(); (dynamic_cast(_curve_config))->loadSettings(); } void FaceTrackNoIR::updateButtonState(bool running) { bool e = !running; ui.iconcomboProfile->setEnabled ( e ); ui.btnLoad->setEnabled ( e ); ui.btnSaveAs->setEnabled ( e ); ui.btnStartTracker->setEnabled ( e ); ui.btnStopTracker->setEnabled ( running ); ui.iconcomboProtocol->setEnabled ( e ); ui.btnShowServerControls->setEnabled ( e ); ui.iconcomboFilter->setEnabled ( e ); ui.iconcomboTrackerSource->setEnabled(e); ui.cbxSecondTrackerSource->setEnabled(e); ui.btnStartTracker->setEnabled(e); ui.btnStopTracker->setEnabled(running); } void FaceTrackNoIR::startTracker( ) { b->save(); loadSettings(); bindKeyboardShortcuts(); if (Libraries) delete Libraries; Libraries = new SelectedLibraries(this); if (!Libraries->correct) { QMessageBox::warning(this, "Something went wrong", "Tracking can't be initialized, probably protocol prerequisites missing", QMessageBox::Ok, QMessageBox::NoButton); stopTracker(); return; } #if defined(_WIN32) keybindingWorker = new KeybindingWorker(*this, keyCenter, keyToggle); keybindingWorker->start(); #endif if (tracker) { delete tracker; } { QSettings settings("opentrack"); QString currentFile = settings.value ( "SettingsFile", QCoreApplication::applicationDirPath() + "/settings/default.ini" ).toString(); QSettings iniFile( currentFile, QSettings::IniFormat ); for (int i = 0; i < 6; i++) { axis(i).curve.loadSettings(iniFile); axis(i).curveAlt.loadSettings(iniFile); } } tracker = new Tracker ( this, s ); if (pTrackerDialog && Libraries->pTracker) { pTrackerDialog->registerTracker( Libraries->pTracker ); } if (pFilterDialog && Libraries->pFilter) pFilterDialog->registerFilter(Libraries->pFilter); tracker->start(); ui.video_frame->show(); timUpdateHeadPose.start(50); updateButtonState(true); } void FaceTrackNoIR::stopTracker( ) { ui.game_name->setText("Not connected"); #if defined(_WIN32) if (keybindingWorker) { keybindingWorker->should_quit = true; keybindingWorker->wait(); delete keybindingWorker; keybindingWorker = NULL; } #endif timUpdateHeadPose.stop(); ui.pose_display->rotateBy(0, 0, 0); if (pTrackerDialog) { pTrackerDialog->unRegisterTracker(); delete pTrackerDialog; pTrackerDialog = nullptr; } if (pProtocolDialog) { pProtocolDialog->unRegisterProtocol(); delete pProtocolDialog; pProtocolDialog = nullptr; } if (pFilterDialog) { pFilterDialog->unregisterFilter(); delete pFilterDialog; pFilterDialog = nullptr; } if (pSecondTrackerDialog) { pSecondTrackerDialog->unRegisterTracker(); delete pSecondTrackerDialog; pSecondTrackerDialog = nullptr; } if ( tracker ) { delete tracker; tracker = 0; if (Libraries) { delete Libraries; Libraries = NULL; } } updateButtonState(false); } void FaceTrackNoIR::showHeadPose() { double newdata[6]; tracker->getHeadPose(newdata); ui.lcdNumX->display(newdata[TX]); ui.lcdNumY->display(newdata[TY]); ui.lcdNumZ->display(newdata[TZ]); ui.lcdNumRotX->display(newdata[Yaw]); ui.lcdNumRotY->display(newdata[Pitch]); ui.lcdNumRotZ->display(newdata[Roll]); tracker->getOutputHeadPose(newdata); ui.pose_display->rotateBy(newdata[Yaw], newdata[Roll], newdata[Pitch]); ui.lcdNumOutputPosX->display(newdata[TX]); ui.lcdNumOutputPosY->display(newdata[TY]); ui.lcdNumOutputPosZ->display(newdata[TZ]); ui.lcdNumOutputRotX->display(newdata[Yaw]); ui.lcdNumOutputRotY->display(newdata[Pitch]); ui.lcdNumOutputRotZ->display(newdata[Roll]); if (_curve_config) { _curve_config->update(); } if (Libraries->pProtocol) { const QString name = Libraries->pProtocol->getGameName(); ui.game_name->setText(name); } } void FaceTrackNoIR::showTrackerSettings() { if (pTrackerDialog) { delete pTrackerDialog; pTrackerDialog = NULL; } DynamicLibrary* lib = dlopen_trackers.value(ui.iconcomboTrackerSource->currentIndex(), (DynamicLibrary*) NULL); if (lib) { pTrackerDialog = (ITrackerDialog*) lib->Dialog(); if (pTrackerDialog) { auto foo = dynamic_cast(pTrackerDialog); foo->setFixedSize(foo->size()); if (Libraries && Libraries->pTracker) pTrackerDialog->registerTracker(Libraries->pTracker); dynamic_cast(pTrackerDialog)->show(); } } } void FaceTrackNoIR::showSecondTrackerSettings() { if (pSecondTrackerDialog) { delete pSecondTrackerDialog; pSecondTrackerDialog = NULL; } DynamicLibrary* lib = dlopen_trackers.value(ui.cbxSecondTrackerSource->currentIndex() - 1, (DynamicLibrary*) NULL); if (lib) { pSecondTrackerDialog = (ITrackerDialog*) lib->Dialog(); if (pSecondTrackerDialog) { auto foo = dynamic_cast(pSecondTrackerDialog); foo->setFixedSize(foo->size()); if (Libraries && Libraries->pSecondTracker) pSecondTrackerDialog->registerTracker(Libraries->pSecondTracker); dynamic_cast(pSecondTrackerDialog)->show(); } } } void FaceTrackNoIR::showServerControls() { if (pProtocolDialog) { delete pProtocolDialog; pProtocolDialog = NULL; } DynamicLibrary* lib = dlopen_protocols.value(ui.iconcomboProtocol->currentIndex(), (DynamicLibrary*) NULL); if (lib && lib->Dialog) { pProtocolDialog = (IProtocolDialog*) lib->Dialog(); if (pProtocolDialog) { auto foo = dynamic_cast(pProtocolDialog); foo->setFixedSize(foo->size()); dynamic_cast(pProtocolDialog)->show(); } } } void FaceTrackNoIR::showFilterControls() { if (pFilterDialog) { delete pFilterDialog; pFilterDialog = NULL; } DynamicLibrary* lib = dlopen_filters.value(ui.iconcomboFilter->currentIndex(), (DynamicLibrary*) NULL); if (lib && lib->Dialog) { pFilterDialog = (IFilterDialog*) lib->Dialog(); if (pFilterDialog) { auto foo = dynamic_cast(pFilterDialog); foo->setFixedSize(foo->size()); if (Libraries && Libraries->pFilter) pFilterDialog->registerFilter(Libraries->pFilter); dynamic_cast(pFilterDialog)->show(); } } } void FaceTrackNoIR::showKeyboardShortcuts() { if (!_keyboard_shortcuts) { _keyboard_shortcuts = new KeyboardShortcutDialog( this, this ); } _keyboard_shortcuts->show(); _keyboard_shortcuts->raise(); } void FaceTrackNoIR::showCurveConfiguration() { if (!_curve_config) { _curve_config = new CurveConfigurationDialog( this, this ); } if (_curve_config) { _curve_config->show(); _curve_config->raise(); } } void FaceTrackNoIR::exit() { QCoreApplication::exit(0); } void FaceTrackNoIR::fill_profile_cbx() { if (looping) return; QSettings settings("opentrack"); QString currentFile = settings.value ( "SettingsFile", QCoreApplication::applicationDirPath() + "/settings/default.ini" ).toString(); qDebug() << "Config file now" << currentFile; QFileInfo pathInfo ( currentFile ); setWindowTitle(QString( OPENTRACK_VERSION " :: ") + pathInfo.fileName()); QDir settingsDir( pathInfo.dir() ); QStringList filters; filters << "*.ini"; auto iniFileList = settingsDir.entryList( filters, QDir::Files, QDir::Name ); ui.iconcomboProfile->clear(); for ( int i = 0; i < iniFileList.size(); i++) { ui.iconcomboProfile->addItem(QIcon(":/images/settings16.png"), iniFileList.at(i)); if (iniFileList.at(i) == pathInfo.fileName()) { ui.iconcomboProfile->setCurrentIndex( i ); } } } void FaceTrackNoIR::profileSelected(int index) { QSettings settings("opentrack"); QString currentFile = settings.value ( "SettingsFile", QCoreApplication::applicationDirPath() + "/settings/default.ini" ).toString(); QFileInfo pathInfo ( currentFile ); settings.setValue ("SettingsFile", pathInfo.absolutePath() + "/" + ui.iconcomboProfile->itemText(index)); loadSettings(); } #if !defined(_WIN32) void FaceTrackNoIR::bind_keyboard_shortcut(QxtGlobalShortcut& key, key_opts& k) { key.setShortcut(QKeySequence::fromString("")); key.setDisabled(); const int idx = k.key_index; if (idx > 0) { QString seq(global_key_sequences.value(idx, "")); if (!seq.isEmpty()) { if (k.shift) seq = "Shift+" + seq; if (k.alt) seq = "Alt+" + seq; if (k.ctrl) seq = "Ctrl+" + seq; key.setShortcut(QKeySequence::fromString(seq, QKeySequence::PortableText)); key.setEnabled(); } else { key.setDisabled(); } } } #else static void bind_keyboard_shortcut(Key& key, key_opts& k) { int idx = k.key_index; if (idx > 0) { key.keycode = 0; key.shift = key.alt = key.ctrl = 0; if (idx < global_windows_key_sequences.size()) key.keycode = global_windows_key_sequences[idx]; key.shift = k.shift; key.alt = k.alt; key.ctrl = k.ctrl; } } #endif void FaceTrackNoIR::bindKeyboardShortcuts() { #if !defined(_WIN32) bind_keyboard_shortcut(keyCenter, s.center_key); bind_keyboard_shortcut(keyToggle, s.toggle_key); #else bind_keyboard_shortcut(keyCenter, s.center_key); bind_keyboard_shortcut(keyToggle, s.toggle_key); #endif if (tracker) /* running already */ { #if defined(_WIN32) if (keybindingWorker) { keybindingWorker->should_quit = true; keybindingWorker->wait(); delete keybindingWorker; keybindingWorker = NULL; } keybindingWorker = new KeybindingWorker(*this, keyCenter, keyToggle); keybindingWorker->start(); #endif } } void FaceTrackNoIR::shortcutRecentered() { QApplication::beep(); qDebug() << "Center"; if (tracker) tracker->do_center = true; } void FaceTrackNoIR::shortcutToggled() { QApplication::beep(); qDebug() << "Toggle"; if (tracker) tracker->enabled = !tracker->enabled; }