/******************************************************************************** * 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) 2012 Wim Vriend (Developing) * * Ron Hendriks (Researching and Testing) * * * * Homepage: http://facetracknoir.sourceforge.net/home/default.htm * * * * 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 . * *********************************************************************************/ /* Modifications (last one on top): 20120427 - WVR: The Protocol-code was already in separate DLLs, but the ListBox was still filled ´statically´. Now, a Dir() of the EXE-folder is done, to locate Protocol-DLLs. The Icons were also moved to the DLLs 20120317 - WVR: The Filter and Tracker-code was moved to separate DLLs. The calling-method was changed accordingly. The face-tracker member-functions NotifyZeroed and refreshVideo were added, as requested by Stanislaw. 20110411 - WVR: Finished moving all Protocols to separate C++ projects. Every protocol now has it's own Class, that's inside it's own DLL. This reduces the size of the program, makes it more structured and enables a more sophisticated installer. 20110328 - WVR: Changed the camera-structs into class-instances. This makes initialisation easier and hopefully solves the remaining 'start-up problem'. 20110313 - WVR: Removed 'set_initial'. Less is more. 20110109 - WVR: Added setZero option to define behaviour after STOP tracking via shortkey. 20110104 - WVR: Removed a few nasty bugs (it was impossible to stop tracker without crash). 20101224 - WVR: Removed the QThread inheritance of the Base Class for the protocol-servers. Again, this drastically simplifies the code in the protocols. 20101217 - WVR: Created Base Class for the protocol-servers. This drastically simplifies the code needed here. 20101024 - WVR: Added shortkey to disable/enable one or more axis during tracking. 20101021 - WVR: Added FSUIPC server for FS2004. 20101011 - WVR: Added SimConnect server. 20101007 - WVR: Created 6DOF-curves and drastically changed the tracker for that. Also eliminated a 'glitch' in the process. 20100607 - WVR: Re-installed Rotation Neutral Zone and improved reaction after 'start/stop'. MessageBeep when confidence is back... 20100604 - WVR: Created structure for DOF-data and changed timing of ReceiveHeadPose end run(). 20100602 - WVR: Implemented EWMA-filtering, according to the example of Melchior Franz. Works like a charm... 20100601 - WVR: Added DirectInput keyboard-handling. '=' used for center, 'BACK' for start (+center)/stop. 20100517 - WVR: Added upstream command(s) from FlightGear 20100523 - WVR: Checkboxes to invert 6DOF's was implemented. Multiply by 1 or (-1). */ #include "tracker.h" #include "FaceTrackNoIR.h" // // Definitions for testing purposes // //#define USE_HEADPOSE_CALLBACK //#define USE_DEBUG_CLIENT // Flags bool Tracker::confid = false; bool Tracker::do_tracking = true; bool Tracker::do_center = false; bool Tracker::do_inhibit = false; bool Tracker::do_game_zero = false; bool Tracker::do_axis_reverse = false; bool Tracker::useFilter = false; bool Tracker::setZero = true; bool Tracker::setEngineStop = true; HANDLE Tracker::hTrackMutex = 0; bool Tracker::useAxisReverse = false; // Use Axis Reverse float Tracker::YawAngle4ReverseAxis = 40.0f; // Axis Reverse settings float Tracker::Z_Pos4ReverseAxis = -20.0f; float Tracker::Z_PosWhenReverseAxis = 50.0f; T6DOF Tracker::current_camera(0,0,0,0,0,0); // Used for filtering T6DOF Tracker::target_camera(0,0,0,0,0,0); T6DOF Tracker::new_camera(0,0,0,0,0,0); T6DOF Tracker::output_camera(0,0,0,0,0,0); // Position sent to game protocol THeadPoseDOF Tracker::Pitch; // One structure for each of 6DOF's THeadPoseDOF Tracker::Yaw; THeadPoseDOF Tracker::Roll; THeadPoseDOF Tracker::X; THeadPoseDOF Tracker::Y; THeadPoseDOF Tracker::Z; TShortKey Tracker::CenterKey; // ShortKey to Center headposition TShortKey Tracker::StartStopKey; // ShortKey to Start/stop tracking TShortKey Tracker::InhibitKey; // ShortKey to inhibit axis while tracking TShortKey Tracker::GameZeroKey; // ShortKey to Set Game Zero //TShortKey Tracker::AxisReverseKey; // ShortKey to start/stop axis reverse while tracking ITrackerPtr Tracker::pTracker; // Pointer to Tracker instance (in DLL) IProtocolPtr Tracker::pProtocol; // Pointer to Protocol instance (in DLL) IFilterPtr Tracker::pFilter; // Pointer to Filter instance (in DLL) /** constructor **/ Tracker::Tracker( FaceTrackNoIR *parent ) { QString libName; importGetTracker getIT; QLibrary *trackerLib; importGetFilter getFilter; QLibrary *filterLib; importGetProtocol getProtocol; QLibrary *protocolLib; QFrame *video_frame; // Retieve the pointer to the parent mainApp = parent; // Create events m_StopThread = CreateEvent(0, TRUE, FALSE, 0); m_WaitThread = CreateEvent(0, TRUE, FALSE, 0); Tracker::hTrackMutex = CreateMutexA(NULL, false, "HeadPose_mutex"); // // Initialize the headpose-data // Tracker::Yaw.initHeadPoseData(); Tracker::Pitch.initHeadPoseData(); Tracker::Roll.initHeadPoseData(); Tracker::X.initHeadPoseData(); Tracker::Y.initHeadPoseData(); Tracker::Z.initHeadPoseData(); // // Locate the video-frame, for the DLL // video_frame = 0; video_frame = mainApp->getVideoWidget(); qDebug() << "Tracker::setup VideoFrame = " << video_frame; // // Load the Tracker-engine DLL, get the tracker-class from it and do stuff... // libName = mainApp->getCurrentTrackerName(); if (!libName.isEmpty()) { trackerLib = new QLibrary(libName); getIT = (importGetTracker) trackerLib->resolve("GetTracker"); if (getIT) { ITrackerPtr ptrXyz(getIT()); // Get the Class if (ptrXyz) { pTracker = ptrXyz; pTracker->Initialize( video_frame ); qDebug() << "Tracker::setup Function Resolved!"; } } else { QMessageBox::warning(0,"FaceTrackNoIR Error", "Facetracker DLL not loaded",QMessageBox::Ok,QMessageBox::NoButton); } } // // Load the DLL with the protocol-logic and retrieve a pointer to the Protocol-class. // libName = mainApp->getCurrentProtocolName(); if (!libName.isEmpty()) { protocolLib = new QLibrary(libName); getProtocol = (importGetProtocol) protocolLib->resolve("GetProtocol"); if (getProtocol) { IProtocolPtr ptrXyz(getProtocol()); if (ptrXyz) { pProtocol = ptrXyz; pProtocol->Initialize(); qDebug() << "Protocol::setup Function Resolved!"; } } else { QMessageBox::warning(0,"FaceTrackNoIR Error", "Protocol-DLL not loaded",QMessageBox::Ok,QMessageBox::NoButton); return; } } # ifdef USE_DEBUG_CLIENT debug_Client = QSharedPointer(new ExcelServer ( this )); // Create Excel protocol-server # endif // // Load the DLL with the filter-logic and retrieve a pointer to the Filter-class. // libName = mainApp->getCurrentFilterName(); filterLib = new QLibrary(libName); getFilter = (importGetFilter) filterLib->resolve("GetFilter"); if (getFilter) { IFilterPtr ptrXyz(getFilter()); if (ptrXyz) { pFilter = ptrXyz; qDebug() << "Filter::setup Function Resolved!"; } } else { QMessageBox::warning(0,"FaceTrackNoIR Error", "Filter-DLL not loaded",QMessageBox::Ok,QMessageBox::NoButton); return; } // Load the settings from the INI-file loadSettings(); } /** destructor empty **/ Tracker::~Tracker() { // Stop the Tracker if (pTracker) { pTracker->StopTracker( true ); } // Trigger thread to stop ::SetEvent(m_StopThread); // Wait until thread finished if (isRunning()) { ::WaitForSingleObject(m_WaitThread, INFINITE); } // Close handles ::CloseHandle(m_StopThread); ::CloseHandle(m_WaitThread); if (Tracker::hTrackMutex != 0) { ::CloseHandle( Tracker::hTrackMutex ); } # ifdef USE_DEBUG_CLIENT debug_Client->deleteLater(); // Delete Excel protocol-server # endif qDebug() << "Tracker::~Tracker Finished..."; } /** setting up the tracker engine **/ void Tracker::setup() { bool DLL_Ok; // retrieve pointers to the User Interface and the main Application pTracker->StartTracker( mainApp->winId() ); // // Check if the Protocol-server files were installed OK. // Some servers also create a memory-mapping, for Inter Process Communication. // The handle of the MainWindow is sent to 'The Game', so it can send a message back. // if (pProtocol) { DLL_Ok = pProtocol->checkServerInstallationOK( mainApp->winId() ); if (!DLL_Ok) { QMessageBox::information(mainApp, "FaceTrackNoIR error", "Protocol is not (correctly) installed!"); } } # ifdef USE_DEBUG_CLIENT DLL_Ok = debug_Client->checkServerInstallationOK( mainApp->winId() ); // Check installation if (!DLL_Ok) { QMessageBox::information(mainApp, "FaceTrackNoIR error", "Excel Protocol is not (correctly) installed!"); } # endif } /** QThread run method @override **/ void Tracker::run() { /** Direct Input variables **/ LPDIRECTINPUT8 din; // the pointer to our DirectInput interface LPDIRECTINPUTDEVICE8 dinkeyboard; // the pointer to the keyboard device BYTE keystate[256]; // the storage for the key-information HRESULT retAcquire; bool lastCenterKey = false; // Remember state, to detect rising edge bool lastStartStopKey = false; bool lastInhibitKey = false; bool lastGameZeroKey = false; bool waitAxisReverse = false; bool waitThroughZero = false; double actualYaw = 0.0f; double actualZ = 0.0f; T6DOF offset_camera(0,0,0,0,0,0); T6DOF gamezero_camera(0,0,0,0,0,0); T6DOF gameoutput_camera(0,0,0,0,0,0); Tracker::do_tracking = true; // Start initially Tracker::do_center = true; // Center initially current_camera.initHeadPoseData(); target_camera.initHeadPoseData(); new_camera.initHeadPoseData(); output_camera.initHeadPoseData(); // // Test some Filter-stuff // if (pFilter) { QString filterName; pFilter->getFullName(&filterName); qDebug() << "Tracker::run() FilterName = " << filterName; } // // Setup the DirectInput for keyboard strokes // // create the DirectInput interface if (DirectInput8Create(GetModuleHandle(NULL), DIRECTINPUT_VERSION, IID_IDirectInput8, (void**)&din, NULL) != DI_OK) { // COM stuff, so we'll set it to NULL qDebug() << "Tracker::setup DirectInput8 Creation failed!" << GetLastError(); } // create the keyboard device if (din->CreateDevice(GUID_SysKeyboard, &dinkeyboard, NULL) != DI_OK) { qDebug() << "Tracker::setup CreateDevice function failed!" << GetLastError(); } // set the data format to keyboard format if (dinkeyboard->SetDataFormat(&c_dfDIKeyboard) != DI_OK) { qDebug() << "Tracker::setup SetDataFormat function failed!" << GetLastError(); } // set the control you will have over the keyboard if (dinkeyboard->SetCooperativeLevel(mainApp->winId(), DISCL_NONEXCLUSIVE | DISCL_BACKGROUND) != DI_OK) { qDebug() << "Tracker::setup SetCooperativeLevel function failed!" << GetLastError(); } forever { // Check event for stop thread if(::WaitForSingleObject(m_StopThread, 0) == WAIT_OBJECT_0) { dinkeyboard->Unacquire(); // Unacquire keyboard din->Release(); // Release DirectInput // Set event ::SetEvent(m_WaitThread); qDebug() << "Tracker::run terminated run()"; return; } // // Check the keyboard // // get access if we don't have it already retAcquire = dinkeyboard->Acquire(); if ( (retAcquire != DI_OK) && (retAcquire != S_FALSE) ) { qDebug() << "Tracker::run Acquire function failed!" << GetLastError(); } else { // get the input data if (dinkeyboard->GetDeviceState(256, (LPVOID)keystate) != DI_OK) { qDebug() << "Tracker::run GetDeviceState function failed!" << GetLastError(); } else { // // Check the state of the Start/Stop key // if ( isShortKeyPressed( &StartStopKey, &keystate[0] ) && (!lastStartStopKey) ) { Tracker::do_tracking = !Tracker::do_tracking; // // To start tracking again and to be at '0', execute Center command too // if (Tracker::do_tracking) { Tracker::do_center = true; Tracker::confid = false; Pitch.rawList.clear(); Pitch.prevPos = 0.0f; Yaw.rawList.clear(); Yaw.prevPos = 0.0f; Roll.rawList.clear(); Roll.prevPos = 0.0f; X.rawList.clear(); X.prevPos = 0.0f; Y.rawList.clear(); Y.prevPos = 0.0f; Z.rawList.clear(); Z.prevPos = 0.0f; current_camera.initHeadPoseData(); target_camera.initHeadPoseData(); new_camera.initHeadPoseData(); offset_camera.initHeadPoseData(); pTracker->StartTracker( mainApp->winId() ); } else { if (setEngineStop) { // Only stop engine when option is checked pTracker->StopTracker( false ); } } qDebug() << "Tracker::run() says StartStop pressed, do_tracking =" << Tracker::do_tracking; } lastStartStopKey = isShortKeyPressed( &StartStopKey, &keystate[0] ); // Remember // // Check the state of the Center key // if ( isShortKeyPressed( &CenterKey, &keystate[0] ) && (!lastCenterKey) ) { Tracker::do_center = true; qDebug() << "Tracker::run() says Center pressed"; } lastCenterKey = isShortKeyPressed( &CenterKey, &keystate[0] ); // Remember // // Check the state of the GameZero key // if ( isShortKeyPressed( &GameZeroKey, &keystate[0] ) && (!lastGameZeroKey) ) { Tracker::do_game_zero = true; qDebug() << "Tracker::run() says GameZero pressed"; } lastGameZeroKey = isShortKeyPressed( &GameZeroKey, &keystate[0] ); // Remember // // Check the state of the Inhibit key // if ( isShortKeyPressed( &InhibitKey, &keystate[0] ) && (!lastInhibitKey) ) { Tracker::do_inhibit = !Tracker::do_inhibit; qDebug() << "Tracker::run() says Inhibit pressed"; // // Execute Center command too, when inhibition ends. // if (!Tracker::do_inhibit) { Tracker::do_center = true; } } lastInhibitKey = isShortKeyPressed( &InhibitKey, &keystate[0] ); // Remember } } // // Reset the 'wait' flag. Moving above 90 with the key pressed, will (de-)activate Axis Reverse. // // qDebug() << "Tracker::run() says actualZ = " << actualZ << ", terwijl Z_Pos4 = " << Z_Pos4ReverseAxis; if (useAxisReverse) { Tracker::do_axis_reverse = ((fabs(actualYaw) > YawAngle4ReverseAxis) && (actualZ < Z_Pos4ReverseAxis)); } else { Tracker::do_axis_reverse = false; } if (WaitForSingleObject(Tracker::hTrackMutex, 100) == WAIT_OBJECT_0) { THeadPoseData newpose; Tracker::confid = pTracker->GiveHeadPoseData(&newpose); if ( Tracker::confid ) { addHeadPose(newpose); } // // If Center is pressed, copy the current values to the offsets. // if (Tracker::confid && Tracker::do_center) { MessageBeep (MB_ICONASTERISK); offset_camera.position.x = getSmoothFromList( &X.rawList ); offset_camera.position.y = getSmoothFromList( &Y.rawList ); offset_camera.position.z = getSmoothFromList( &Z.rawList ); offset_camera.position.pitch = getSmoothFromList( &Pitch.rawList ); offset_camera.position.yaw = getSmoothFromList( &Yaw.rawList ); offset_camera.position.roll = getSmoothFromList( &Roll.rawList ); Tracker::do_center = false; } // // If Set Game Zero is pressed, copy the current values to the offsets. // Change requested by Stanislaw // if (Tracker::confid && Tracker::do_game_zero) { if (!pTracker->notifyZeroed()) gamezero_camera.position = gameoutput_camera.position; // gamezero_camera.position = gameoutput_camera.position; Tracker::do_game_zero = false; } if (Tracker::do_tracking && Tracker::confid) { // Pitch target_camera.position.x = getSmoothFromList( &X.rawList ); target_camera.position.y = getSmoothFromList( &Y.rawList ); target_camera.position.z = getSmoothFromList( &Z.rawList ); target_camera.position.pitch = getSmoothFromList( &Pitch.rawList ); target_camera.position.yaw = getSmoothFromList( &Yaw.rawList ); target_camera.position.roll = getSmoothFromList( &Roll.rawList ); target_camera = target_camera - offset_camera; if (Tracker::useFilter && pFilter) { pFilter->FilterHeadPoseData(¤t_camera.position, &target_camera.position, &new_camera.position, Tracker::Pitch.newSample); } else { new_camera.position.x = getSmoothFromList( &X.rawList ) - X.offset_headPos; new_camera.position.y = getSmoothFromList( &Y.rawList ) - Y.offset_headPos; new_camera.position.z = getSmoothFromList( &Z.rawList ) - Z.offset_headPos; new_camera.position.pitch = getSmoothFromList( &Pitch.rawList ) - Pitch.offset_headPos; new_camera.position.yaw = getSmoothFromList( &Yaw.rawList ) - Yaw.offset_headPos; new_camera.position.roll = getSmoothFromList( &Roll.rawList ) - Roll.offset_headPos; } output_camera.position.x = X.invert * getOutputFromCurve(&X.curve, new_camera.position.x, X.NeutralZone, X.MaxInput); output_camera.position.y = Y.invert * getOutputFromCurve(&Y.curve, new_camera.position.y, Y.NeutralZone, Y.MaxInput); output_camera.position.z = Z.invert * getOutputFromCurve(&Z.curve, new_camera.position.z, Z.NeutralZone, Z.MaxInput); output_camera.position.pitch = Pitch.invert * getOutputFromCurve(&Pitch.curve, new_camera.position.pitch, Pitch.NeutralZone, Pitch.MaxInput); output_camera.position.yaw = Yaw.invert * getOutputFromCurve(&Yaw.curve, new_camera.position.yaw, Yaw.NeutralZone, Yaw.MaxInput); output_camera.position.roll = Roll.invert * getOutputFromCurve(&Roll.curve, new_camera.position.roll, Roll.NeutralZone, Roll.MaxInput); // // Reverse Axis. // actualYaw = output_camera.position.yaw; // Save the actual Yaw, otherwise we can't check for +90 actualZ = output_camera.position.z; // Also the Z if (Tracker::do_axis_reverse) { output_camera.position.z = Z_PosWhenReverseAxis; // Set the desired Z-position } // // Reset value for the selected axis, if inhibition is active // if (Tracker::do_inhibit) { if (InhibitKey.doPitch) output_camera.position.pitch = 0.0f; if (InhibitKey.doYaw) output_camera.position.yaw = 0.0f; if (InhibitKey.doRoll) output_camera.position.roll = 0.0f; if (InhibitKey.doX) output_camera.position.x = 0.0f; if (InhibitKey.doY) output_camera.position.y = 0.0f; if (InhibitKey.doZ) output_camera.position.z = 0.0f; } // All Protocol server(s) if (pProtocol) { gameoutput_camera = output_camera + gamezero_camera; pProtocol->sendHeadposeToGame( &gameoutput_camera ); // degrees & centimeters } # ifdef USE_DEBUG_CLIENT debug_Client->setHeadRotX( Tracker::Pitch.headPos ); // degrees debug_Client->setHeadRotY( Tracker::Yaw.headPos ); debug_Client->setHeadRotZ( Tracker::Roll.headPos ); debug_Client->setHeadPosX( Tracker::X.headPos ); // centimeters debug_Client->setHeadPosY( Tracker::Y.headPos ); debug_Client->setHeadPosZ( Tracker::Z.headPos ); debug_Client->setVirtRotX ( new_camera.position.pitch ); // degrees debug_Client->setVirtRotY ( new_camera.position.yaw ); debug_Client->setVirtRotZ ( new_camera.position.roll ); debug_Client->setVirtPosX ( new_camera.position.x ); // centimeters debug_Client->setVirtPosY ( new_camera.position.y ); debug_Client->setVirtPosZ ( new_camera.position.z ); # endif } else { // // Go to initial position // if (pProtocol && setZero) { output_camera.position.pitch = 0.0f; output_camera.position.yaw = 0.0f; output_camera.position.roll = 0.0f; output_camera.position.x = 0.0f; output_camera.position.y = 0.0f; output_camera.position.z = 0.0f; gameoutput_camera = output_camera + gamezero_camera; pProtocol->sendHeadposeToGame( &gameoutput_camera ); // degrees & centimeters } } } # ifdef USE_DEBUG_CLIENT debug_Client->confidence = Tracker::Pitch.confidence; debug_Client->newSample = Tracker::Pitch.newSample; debug_Client->smoothvalue = getSmoothFromList( &Pitch.rawList ); debug_Client->prev_value = Tracker::Pitch.prevPos; debug_Client->dT = dT; debug_Client->sendHeadposeToGame(); // Log to Excel # endif Tracker::Pitch.newSample = false; ReleaseMutex(Tracker::hTrackMutex); //for lower cpu load usleep(10000); yieldCurrentThread(); } } /** Add the headpose-data to the Lists **/ void Tracker::addHeadPose( THeadPoseData head_pose ) { // Pitch Tracker::Pitch.headPos = head_pose.pitch; // degrees addRaw2List ( &Pitch.rawList, Pitch.maxItems, Tracker::Pitch.headPos ); // Tracker::Pitch.confidence = head_pose.confidence; // Just this one ... Tracker::Pitch.newSample = true; // Yaw Tracker::Yaw.headPos = head_pose.yaw; // degrees addRaw2List ( &Yaw.rawList, Yaw.maxItems, Tracker::Yaw.headPos ); // Roll Tracker::Roll.headPos = head_pose.roll; // degrees addRaw2List ( &Roll.rawList, Roll.maxItems, Tracker::Roll.headPos ); // X-position Tracker::X.headPos = head_pose.x; // centimeters addRaw2List ( &X.rawList, X.maxItems, Tracker::X.headPos ); // Y-position Tracker::Y.headPos = head_pose.y; // centimeters addRaw2List ( &Y.rawList, Y.maxItems, Tracker::Y.headPos ); // Z-position (distance to camera, absolute!) Tracker::Z.headPos = head_pose.z; // centimeters addRaw2List ( &Z.rawList, Z.maxItems, Tracker::Z.headPos ); } // // Get the ProgramName from the Game and return it. // QString Tracker::getGameProgramName() { QString str; char dest[100]; str = QString("No protocol active?"); if (pProtocol) { pProtocol->getNameFromGame( dest ); str = QString( dest ); } return str; } // // Handle the command, send upstream by the game. // Valid values are: // 1 = reset Headpose // bool Tracker::handleGameCommand ( int command ) { qDebug() << "handleGameCommand says: Command =" << command; switch ( command ) { case 1: // reset headtracker Tracker::do_center = true; break; default: break; } return false; } // // Add the new Raw value to the QList. // Remove the last item(s), depending on the set maximum list-items. // void Tracker::addRaw2List ( QList *rawList, float maxIndex, float raw ) { // // Remove old values from the end of the QList. // If the setting for MaxItems was lowered, the QList is shortened here... // while (rawList->size() >= maxIndex) { rawList->removeLast(); } // // Insert the newest at the beginning. // rawList->prepend ( raw ); } // // Get the raw headpose, so it can be displayed. // void Tracker::getHeadPose( THeadPoseData *data ) { data->x = Tracker::X.headPos - Tracker::X.offset_headPos; // centimeters data->y = Tracker::Y.headPos - Tracker::Y.offset_headPos; data->z = Tracker::Z.headPos - Tracker::Z.offset_headPos; data->pitch = Tracker::Pitch.headPos- Tracker::Pitch.offset_headPos; // degrees data->yaw = Tracker::Yaw.headPos- Tracker::Yaw.offset_headPos; data->roll = Tracker::Roll.headPos - Tracker::Roll.offset_headPos; } // // Get the output-headpose, so it can be displayed. // void Tracker::getOutputHeadPose( THeadPoseData *data ) { data->x = output_camera.position.x; // centimeters data->y = output_camera.position.y; data->z = output_camera.position.z; data->pitch = output_camera.position.pitch; // degrees data->yaw = output_camera.position.yaw; data->roll = output_camera.position.roll; } // // Get the Smoothed value from the QList. // float Tracker::getSmoothFromList ( QList *rawList ) { float sum = 0; if (rawList->isEmpty()) return 0.0f; // // Add the Raw values and divide. // for ( int i = 0; i < rawList->size(); i++) { sum += rawList->at(i); } return sum / rawList->size(); } // // Correct the Raw value, with the Neutral Zone supplied // //float Tracker::getCorrectedNewRaw ( float NewRaw, float rotNeutral ) { // // // // // Return 0, if NewRaw is within the Neutral Zone // // // if ( fabs( NewRaw ) < rotNeutral ) { // return 0.0f; // } // // // // // NewRaw is outside the zone. // // Substract rotNeutral from the NewRaw // // // if ( NewRaw > 0.0f ) { // return (NewRaw - rotNeutral); // } // else { // return (NewRaw + rotNeutral); // Makes sense? // } // //} // // Implementation of an Exponentially Weighted Moving Average, used to serve as a low-pass filter. // The code was adopted from Melchior Franz, who created it for FlightGear (aircraft.nas). // // The function takes the new value, the delta-time (sec) and a weighing coefficient (>0 and <1) // All previous values are taken into account, the weight of this is determined by 'coeff'. // //float Tracker::lowPassFilter ( float newvalue, float *oldvalue, float dt, float coeff) { //float c = 0.0f; //float fil = 0.0f; // // c = dt / (coeff + dt); // fil = (newvalue * c) + (*oldvalue * (1 - c)); // *oldvalue = fil; // // return fil; //} // // Implementation of a Rate Limiter, used to eliminate spikes in the raw data. // // The function takes the new value, the delta-time (sec) and the positive max. slew-rate (engineering units/sec) // //float Tracker::rateLimiter ( float newvalue, float *oldvalue, float dt, float max_rate) { //float rate = 0.0f; //float clamped_value = 0.0f; // // rate = (newvalue - *oldvalue) / dt; // clamped_value = newvalue; // If all is well, the newvalue is returned // // // // // One max-rate is used for ramp-up and ramp-down // // If the rate exceeds max_rate, return the maximum value that the max_rate allows // // // if (fabs(rate) > max_rate) { // // // // For ramp-down, apply a factor -1 to the max_rate // // // if (rate < 0.0f) { // clamped_value = (-1.0f * dt * max_rate) + *oldvalue; // } // else { // clamped_value = (dt * max_rate) + *oldvalue; // } // } // *oldvalue = clamped_value; // // return clamped_value; //} // // Get the output from the curve. // float Tracker::getOutputFromCurve ( QPainterPath *curve, float input, float neutralzone, float maxinput ) { float sign; sign = 1.0f; if (input < 0.0f) { sign = -1.0f; } // // Always return 0 inside the NeutralZone // Always return max. when input larger than expected // if (fabs(input) > maxinput) return sign * curve->pointAtPercent(1.0).x(); // // Return the value, derived from the Bezier-curve // return sign * curve->pointAtPercent((fabs(input))/maxinput).x(); } // // Load the current Settings from the currently 'active' INI-file. // void Tracker::loadSettings() { int NeutralZone; int sensYaw, sensPitch, sensRoll; int sensX, sensY, sensZ; QPointF point1, point2, point3, point4; qDebug() << "Tracker::loadSettings says: Starting "; QSettings settings("Abbequerque Inc.", "FaceTrackNoIR"); // Registry settings (in HK_USER) QString currentFile = settings.value ( "SettingsFile", QCoreApplication::applicationDirPath() + "/Settings/default.ini" ).toString(); QSettings iniFile( currentFile, QSettings::IniFormat ); // Application settings (in INI-file) qDebug() << "loadSettings says: iniFile = " << currentFile; // // Read the Tracking settings, to fill the curves. // iniFile.beginGroup ( "Tracking" ); NeutralZone = iniFile.value ( "NeutralZone", 5 ).toInt(); sensYaw = iniFile.value ( "sensYaw", 100 ).toInt(); sensPitch = iniFile.value ( "sensPitch", 100 ).toInt(); sensRoll = iniFile.value ( "sensRoll", 100 ).toInt(); sensX = iniFile.value ( "sensX", 100 ).toInt(); sensY = iniFile.value ( "sensY", 100 ).toInt(); sensZ = iniFile.value ( "sensZ", 100 ).toInt(); iniFile.endGroup (); // // Read the curve-settings from the file. Use the (deprecated) settings, if the curves are not there. // iniFile.beginGroup ( "Curves" ); // // Create a new path and assign it to the curve. // getCurvePoints( &iniFile, "Yaw_", &point1, &point2, &point3, &point4, NeutralZone, sensYaw, 50, 180 ); QPainterPath newYawCurve; newYawCurve.moveTo( QPointF(0,0) ); newYawCurve.lineTo( point1 ); newYawCurve.cubicTo(point2, point3, point4); Yaw.NeutralZone = point1.y(); // Get the Neutral Zone Yaw.MaxInput = point4.y(); // Get Maximum Input Yaw.curve = newYawCurve; qDebug() << "loadSettings says: curve-elementcount = " << Yaw.curve.elementCount(); // Pitch getCurvePoints( &iniFile, "Pitch_", &point1, &point2, &point3, &point4, NeutralZone, sensPitch, 50, 180 ); QPainterPath newPitchCurve; newPitchCurve.moveTo( QPointF(0,0) ); newPitchCurve.lineTo( point1 ); newPitchCurve.cubicTo(point2, point3, point4); Pitch.NeutralZone = point1.y(); // Get the Neutral Zone Pitch.MaxInput = point4.y(); // Get Maximum Input Pitch.curve = newPitchCurve; // Roll getCurvePoints( &iniFile, "Roll_", &point1, &point2, &point3, &point4, NeutralZone, sensRoll, 50, 180 ); QPainterPath newRollCurve; newRollCurve.moveTo( QPointF(0,0) ); newRollCurve.lineTo( point1 ); newRollCurve.cubicTo(point2, point3, point4); Roll.NeutralZone = point1.y(); // Get the Neutral Zone Roll.MaxInput = point4.y(); // Get Maximum Input Roll.curve = newRollCurve; // X getCurvePoints( &iniFile, "X_", &point1, &point2, &point3, &point4, NeutralZone, sensX, 50, 180 ); QPainterPath newXCurve; newXCurve.moveTo( QPointF(0,0) ); newXCurve.lineTo( point1 ); newXCurve.cubicTo(point2, point3, point4); X.NeutralZone = point1.y(); // Get the Neutral Zone X.MaxInput = point4.y(); // Get Maximum Input X.curve = newXCurve; // Y getCurvePoints( &iniFile, "Y_", &point1, &point2, &point3, &point4, NeutralZone, sensY, 50, 180 ); QPainterPath newYCurve; newYCurve.moveTo( QPointF(0,0) ); newYCurve.lineTo( point1 ); newYCurve.cubicTo(point2, point3, point4); Y.NeutralZone = point1.y(); // Get the Neutral Zone Y.MaxInput = point4.y(); // Get Maximum Input Y.curve = newYCurve; // Z getCurvePoints( &iniFile, "Z_", &point1, &point2, &point3, &point4, NeutralZone, sensZ, 50, 180 ); QPainterPath newZCurve; newZCurve.moveTo( QPointF(0,0) ); newZCurve.lineTo( point1 ); newZCurve.cubicTo(point2, point3, point4); Z.NeutralZone = point1.y(); // Get the Neutral Zone Z.MaxInput = point4.y(); // Get Maximum Input Z.curve = newZCurve; iniFile.endGroup (); // // Read the keyboard shortcuts. // iniFile.beginGroup ( "KB_Shortcuts" ); // Center key CenterKey.keycode = iniFile.value ( "Keycode_Center", 0 ).toInt(); CenterKey.shift = iniFile.value ( "Shift_Center", 0 ).toBool(); CenterKey.ctrl = iniFile.value ( "Ctrl_Center", 0 ).toBool(); CenterKey.alt = iniFile.value ( "Alt_Center", 0 ).toBool(); // StartStop key StartStopKey.keycode = iniFile.value ( "Keycode_StartStop", 0 ).toInt(); StartStopKey.shift = iniFile.value ( "Shift_StartStop", 0 ).toBool(); StartStopKey.ctrl = iniFile.value ( "Ctrl_StartStop", 0 ).toBool(); StartStopKey.alt = iniFile.value ( "Alt_StartStop", 0 ).toBool(); setZero = iniFile.value ( "SetZero", 1 ).toBool(); setEngineStop = iniFile.value ( "SetEngineStop", 1 ).toBool(); // Inhibit key InhibitKey.keycode = iniFile.value ( "Keycode_Inhibit", 0 ).toInt(); InhibitKey.shift = iniFile.value ( "Shift_Inhibit", 0 ).toBool(); InhibitKey.ctrl = iniFile.value ( "Ctrl_Inhibit", 0 ).toBool(); InhibitKey.alt = iniFile.value ( "Alt_Inhibit", 0 ).toBool(); InhibitKey.doPitch = iniFile.value ( "Inhibit_Pitch", 0 ).toBool(); InhibitKey.doYaw = iniFile.value ( "Inhibit_Yaw", 0 ).toBool(); InhibitKey.doRoll = iniFile.value ( "Inhibit_Roll", 0 ).toBool(); InhibitKey.doX = iniFile.value ( "Inhibit_X", 0 ).toBool(); InhibitKey.doY = iniFile.value ( "Inhibit_Y", 0 ).toBool(); InhibitKey.doZ = iniFile.value ( "Inhibit_Z", 0 ).toBool(); // Game Zero key GameZeroKey.keycode = iniFile.value ( "Keycode_GameZero", 0 ).toInt(); GameZeroKey.shift = iniFile.value ( "Shift_GameZero", 0 ).toBool(); GameZeroKey.ctrl = iniFile.value ( "Ctrl_GameZero", 0 ).toBool(); GameZeroKey.alt = iniFile.value ( "Alt_GameZero", 0 ).toBool(); // Axis Reverse key //AxisReverseKey.keycode = DIK_R; //AxisReverseKey.shift = false; //AxisReverseKey.ctrl = false; //AxisReverseKey.alt = false; // Reverse Axis useAxisReverse = iniFile.value ( "Enable_ReverseAxis", 0 ).toBool(); YawAngle4ReverseAxis = iniFile.value ( "RA_Yaw", 40 ).toInt(); Z_Pos4ReverseAxis = iniFile.value ( "RA_ZPos", 50 ).toInt(); Z_PosWhenReverseAxis = iniFile.value ( "RA_ToZPos", 80 ).toInt(); iniFile.endGroup (); } // // Determine if the ShortKey (incl. CTRL, SHIFT and/or ALT) is pressed. // bool Tracker::isShortKeyPressed( TShortKey *key, BYTE *keystate ){ bool shift; bool ctrl; bool alt; // // First, check if the right key is pressed. If so, check the modifiers // if (keystate[key->keycode] & 0x80) { shift = ( (keystate[DIK_LSHIFT] & 0x80) || (keystate[DIK_RSHIFT] & 0x80) ); ctrl = ( (keystate[DIK_LCONTROL] & 0x80) || (keystate[DIK_RCONTROL] & 0x80) ); alt = ( (keystate[DIK_LALT] & 0x80) || (keystate[DIK_RALT] & 0x80) ); // // If one of the modifiers is needed and not pressed, return false. // if (key->shift && !shift) return false; if (key->ctrl && !ctrl) return false; if (key->alt && !alt) return false; // // All is well! // return true; } else { return false; } }