path: root/FTNoIR_Tracker_SM/ftnoir_tracker_faceapi_dialog.cpp

/********************************************************************************
* FaceTrackNoIR		This program is a private project of 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 <http://www.gnu.org/licenses/>.				*
*																				*
********************************************************************************/
#include "ftnoir_tracker_sm.h"
#include <QtGui>

//*******************************************************************************************************
// faceAPI Client Settings-dialog.
//*******************************************************************************************************

//
// Constructor for server-settings-dialog
//
TrackerControls::TrackerControls() :
QWidget()
{
	ui.setupUi( this );

	theTracker = NULL;

	// Connect Qt signals to member-functions
	connect(ui.btnOK, SIGNAL(clicked()), this, SLOT(doOK()));
	connect(ui.btnCancel, SIGNAL(clicked()), this, SLOT(doCancel()));
	connect(ui.btnEngineStart, SIGNAL(clicked()), this, SLOT(doStartEngine()));
	connect(ui.btnEngineStop, SIGNAL(clicked()), this, SLOT(doStopEngine()));

	ui.cbxFilterSetting->addItem("None");
	ui.cbxFilterSetting->addItem("Normal");
	ui.cbxFilterSetting->addItem("High");
	connect(ui.cbxFilterSetting, SIGNAL(currentIndexChanged(int)), this, SLOT(doSetFilter( int )));
	connect(ui.btnCameraSettings, SIGNAL(clicked()), this, SLOT(doShowCam()));

	if (SMCreateMapping()) {
		qDebug() << "TrackerControls::Initialize Mapping created.";
	}
	else {
		QMessageBox::warning(0,"FaceTrackNoIR Error","Memory mapping not created!",QMessageBox::Ok,QMessageBox::NoButton);
	}

	//Setup the timer for showing the headpose.
	timUpdateSettings = new QTimer(this);
    connect(timUpdateSettings, SIGNAL(timeout()), this, SLOT(doTimUpdate()));
	timUpdateSettings->start(100);
	connect(this, SIGNAL(stateChanged( int )), this, SLOT(showSettings( int )));

	connect(ui.chkInvertRoll, SIGNAL(stateChanged(int)), this, SLOT(settingChanged(int)));
	connect(ui.chkInvertPitch, SIGNAL(stateChanged(int)), this, SLOT(settingChanged(int)));
	connect(ui.chkInvertYaw, SIGNAL(stateChanged(int)), this, SLOT(settingChanged(int)));
	connect(ui.chkInvertX, SIGNAL(stateChanged(int)), this, SLOT(settingChanged(int)));
	connect(ui.chkInvertY, SIGNAL(stateChanged(int)), this, SLOT(settingChanged(int)));
	connect(ui.chkInvertZ, SIGNAL(stateChanged(int)), this, SLOT(settingChanged(int)));

	connect(ui.chkEnableRoll, SIGNAL(stateChanged(int)), this, SLOT(settingChanged(int)));
	connect(ui.chkEnablePitch, SIGNAL(stateChanged(int)), this, SLOT(settingChanged(int)));
	connect(ui.chkEnableYaw, SIGNAL(stateChanged(int)), this, SLOT(settingChanged(int)));
	connect(ui.chkEnableX, SIGNAL(stateChanged(int)), this, SLOT(settingChanged(int)));
	connect(ui.chkEnableY, SIGNAL(stateChanged(int)), this, SLOT(settingChanged(int)));
	connect(ui.chkEnableZ, SIGNAL(stateChanged(int)), this, SLOT(settingChanged(int)));

}

//
// Destructor for server-dialog
//
TrackerControls::~TrackerControls() {
	qDebug() << "~TrackerControls() says: started";
}

//
// Initialize tracker-client-dialog
//
void TrackerControls::Initialize(QWidget *parent, int num) {

	QPoint offsetpos(200, 200);
	if (parent) {
		this->move(parent->pos() + offsetpos);
	}

	// Load the settings from the current .INI-file
	numTracker = num;
	loadSettings();

	show();
}

//
// OK clicked on server-dialog
//
void TrackerControls::doOK() {
	save();
	this->close();
}

// override show event
void TrackerControls::showEvent ( QShowEvent * event ) {
	loadSettings();
}

//
// Cancel clicked on server-dialog
//
void TrackerControls::doCancel() {
	//
	// Ask if changed Settings should be saved
	//
	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 );

		qDebug() << "doCancel says: answer =" << ret;

		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();
	}
}

//
// Load the current Settings from the currently 'active' INI-file.
//
void TrackerControls::loadSettings() {

//	qDebug() << "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;

	iniFile.beginGroup ( "SMTracker" );
	ui.cbxFilterSetting->setCurrentIndex(iniFile.value ( "FilterLevel", 1 ).toInt());
	ui.chkInvertRoll->setChecked(iniFile.value ( "InvertRoll", 0 ).toBool());
	ui.chkInvertPitch->setChecked(iniFile.value ( "InvertPitch", 0 ).toBool());
	ui.chkInvertYaw->setChecked(iniFile.value ( "InvertYaw", 0 ).toBool());
	ui.chkInvertX->setChecked(iniFile.value ( "InvertX", 0 ).toBool());
	ui.chkInvertY->setChecked(iniFile.value ( "InvertY", 0 ).toBool());
	ui.chkInvertZ->setChecked(iniFile.value ( "InvertZ", 0 ).toBool());

	iniFile.endGroup ();

	iniFile.beginGroup ( "HeadTracker" );
	//
	// Check if the Tracker is the Primary one.
	// If the property is not found in the INI-file, set the value.
	//
	if (numTracker == 1) {
		numRoll = iniFile.value ( "RollTracker", 1 ).toInt();
		numPitch = iniFile.value ( "PitchTracker", 1 ).toInt();
		numYaw = iniFile.value ( "YawTracker", 1 ).toInt();
		numX = iniFile.value ( "XTracker", 0 ).toInt();
		numY = iniFile.value ( "YTracker", 0 ).toInt();
		numZ = iniFile.value ( "ZTracker", 0 ).toInt();
	}
	else {
		numRoll = iniFile.value ( "RollTracker", 0 ).toInt();
		numPitch = iniFile.value ( "PitchTracker", 0 ).toInt();
		numYaw = iniFile.value ( "YawTracker", 0 ).toInt();
		numX = iniFile.value ( "XTracker", 0 ).toInt();
		numY = iniFile.value ( "YTracker", 0 ).toInt();
		numZ = iniFile.value ( "ZTracker", 0 ).toInt();
	}
	ui.chkEnableRoll->setChecked(numRoll == numTracker);
	ui.chkEnablePitch->setChecked(numPitch == numTracker);
	ui.chkEnableYaw->setChecked(numYaw == numTracker);
	ui.chkEnableX->setChecked(numX == numTracker);
	ui.chkEnableY->setChecked(numY == numTracker);
	ui.chkEnableZ->setChecked(numZ == numTracker);
	iniFile.endGroup ();

	settingsDirty = false;
}

//
// Save the current Settings to the currently 'active' INI-file.
//
void TrackerControls::save() {

	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)

	iniFile.beginGroup ( "SMTracker" );
	iniFile.setValue ( "FilterLevel", ui.cbxFilterSetting->currentIndex() );
	iniFile.setValue ( "InvertRoll", ui.chkInvertRoll->isChecked() );
	iniFile.setValue ( "InvertPitch", ui.chkInvertPitch->isChecked() );
	iniFile.setValue ( "InvertYaw", ui.chkInvertYaw->isChecked() );
	iniFile.setValue ( "InvertX", ui.chkInvertX->isChecked() );
	iniFile.setValue ( "InvertY", ui.chkInvertY->isChecked() );
	iniFile.setValue ( "InvertZ", ui.chkInvertZ->isChecked() );
	iniFile.endGroup ();

	iniFile.beginGroup ( "HeadTracker" );
	if ( ui.chkEnableRoll->isChecked() ) {
		iniFile.setValue ( "RollTracker", numTracker );
	}
	else {
		if (numRoll == numTracker) {
			iniFile.setValue ( "RollTracker", 0 );
		}
	}

	if ( ui.chkEnablePitch->isChecked() ) {
		iniFile.setValue ( "PitchTracker", numTracker );
	}
	else {
		if (numPitch == numTracker) {
			iniFile.setValue ( "PitchTracker", 0 );
		}
	}

	if ( ui.chkEnableYaw->isChecked() ) {
		iniFile.setValue ( "YawTracker", numTracker );
	}
	else {
		if (numYaw == numTracker) {
			iniFile.setValue ( "YawTracker", 0 );
		}
	}
	if ( ui.chkEnableX->isChecked() ) {
		iniFile.setValue ( "XTracker", numTracker );
	}
	else {
		if (numX == numTracker) {
			iniFile.setValue ( "XTracker", 0 );
		}
	}
	if ( ui.chkEnableY->isChecked() ) {
		iniFile.setValue ( "YTracker", numTracker );
	}
	else {
		if (numY == numTracker) {
			iniFile.setValue ( "YTracker", 0 );
		}
	}
	if ( ui.chkEnableZ->isChecked() ) {
		iniFile.setValue ( "ZTracker", numTracker );
	}
	else {
		if (numZ == numTracker) {
			iniFile.setValue ( "ZTracker", 0 );
		}
	}

	iniFile.endGroup ();

	settingsDirty = false;
}

//
// Create a memory-mapping to the faceAPI data.
// It contains the tracking data, a command-code from FaceTrackNoIR
//
//
bool TrackerControls::SMCreateMapping()
{
	qDebug() << "TrackerControls::FTCreateMapping says: Starting Function";

	//
	// A FileMapping is used to create 'shared memory' between the faceAPI and FaceTrackNoIR.
	//
	// Try to create a FileMapping to the Shared Memory.
	// If one already exists: close it.
	//
	hSMMemMap = CreateFileMappingA( INVALID_HANDLE_VALUE , 00 , PAGE_READWRITE , 0 , 
		                           sizeof( TFaceData ) + sizeof( HANDLE ) + 100, 
								   (LPCSTR) SM_MM_DATA );

	if ( hSMMemMap != 0 ) {
		qDebug() << "TrackerControls::FTCreateMapping says: FileMapping Created!";
	}

	if ( ( hSMMemMap != 0 ) && ( (long) GetLastError == ERROR_ALREADY_EXISTS ) ) {
		CloseHandle( hSMMemMap );
		hSMMemMap = 0;
	}

	//
	// Create a new FileMapping, Read/Write access
	//
	hSMMemMap = OpenFileMappingA( FILE_MAP_ALL_ACCESS , false , (LPCSTR) SM_MM_DATA );
	if ( ( hSMMemMap != 0 ) ) {
		qDebug() << "TrackerControls::FTCreateMapping says: FileMapping Created again..." << hSMMemMap;
		pMemData = (SMMemMap *) MapViewOfFile(hSMMemMap, FILE_MAP_ALL_ACCESS, 0, 0, sizeof(TFaceData));
		if (pMemData != NULL) {
			qDebug() << "TrackerControls::FTCreateMapping says: MapViewOfFile OK.";
//			pMemData->handle = handle;	// The game uses the handle, to send a message that the Program-Name was set!
		}
	    hSMMutex = CreateMutexA(NULL, false, SM_MUTEX);
	}
	else {
		qDebug() << "TrackerControls::FTCreateMapping says: Error creating Shared Memory for faceAPI!";
		return false;
	}

	//if (pMemData != NULL) {
	//	pMemData->data.DataID = 1;
	//	pMemData->data.CamWidth = 100;
	//	pMemData->data.CamHeight = 250;
	//}

	return true;
}

//
// Show the current engine-settings etc.
//
void TrackerControls::doTimUpdate()
{
	int state = pMemData->state;
	if ( state != prev_state) {
		emit stateChanged(state);
		prev_state = state;
	}
}

//
// Show the current engine-settings etc.
//
void TrackerControls::showSettings( int newState )
{
	qDebug() << "TrackerControls::showSettings says: Starting Function";
    switch (newState)
    {
    case SM_API_ENGINE_STATE_TERMINATED:
        ui._engine_state_label->setText("TERMINATED");
        break;
    case SM_API_ENGINE_STATE_INVALID:
        ui._engine_state_label->setText("INVALID");
        break;
    case SM_API_ENGINE_STATE_IDLE:
        ui._engine_state_label->setText("IDLE");
        break;
    case SM_API_ENGINE_STATE_HT_INITIALIZING:
        ui._engine_state_label->setText("INITIALIZING");
        break;
    case SM_API_ENGINE_STATE_HT_TRACKING:
        ui._engine_state_label->setText("TRACKING");
        break;
    case SM_API_ENGINE_STATE_HT_SEARCHING:
        ui._engine_state_label->setText("SEARCHING");
        break;
    default:
        ui._engine_state_label->setText("Unknown State!");
        break;
    }

	ui.cbxFilterSetting->setEnabled( (newState == SM_API_ENGINE_STATE_IDLE) );
}

//
// Send a command without parameter-value to the tracking Engine.
//
void TrackerControls::doCommand(int command)
{
	if ( (pMemData != NULL) && (WaitForSingleObject(hSMMutex, 100) == WAIT_OBJECT_0) ) {
		pMemData->command = command;					// Send command
		ReleaseMutex(hSMMutex);
	}
	return;
}

//
// Send a command with integer parameter-value to the tracking Engine.
//
void TrackerControls::doCommand(int command, int value)
{
	if ( (pMemData != NULL) && (WaitForSingleObject(hSMMutex, 100) == WAIT_OBJECT_0) ) {
		pMemData->command = command;					// Send command
		pMemData->par_val_int = value;
		ReleaseMutex(hSMMutex);
	}
	return;
}

////////////////////////////////////////////////////////////////////////////////
// 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")

FTNOIR_TRACKER_BASE_EXPORT ITrackerDialogPtr __stdcall GetTrackerDialog( )
{
	return new TrackerControls;
}