#include "ftnoir_tracker_sm.h"
#include "mainwindow.h"

using namespace sm::faceapi;
using namespace sm::faceapi::qt;
using namespace sm::faceapi::samplecode;
//using namespace sm::faceapi::qt;

FTNoIR_Tracker_SM::FTNoIR_Tracker_SM()
{
	//allocate memory for the parameters
	parameterValueAsFloat.clear();
	parameterRange.clear();

	// Add the parameters to the list
	parameterRange.append(std::pair<float,float>(1000.0f,9999.0f));
	parameterValueAsFloat.append(0.0f);
	setParameterValue(kPortAddress,5551.0f);
}

FTNoIR_Tracker_SM::~FTNoIR_Tracker_SM()
{
	qDebug() << "stopTracker says: terminating";
	//if ( _display ) {
	//	_display->disconnect();
	//	qDebug() << "stopTracker says: display disconnected";
	//	delete _display;
	//	qDebug() << "stopTracker says: display deleted";
	//	_display = 0;
	//	delete l;
	//	l = 0;
	//	qDebug() << "stopTracker says: l deleted";
	//}

	_engine->stop();
	smAPIQuit();
}

void FTNoIR_Tracker_SM::Release()
{
	qDebug() << "FTNoIR_Tracker_SM::Release says: Starting ";
    delete this;
}

void FTNoIR_Tracker_SM::Initialize( QFrame *videoframe )
{
	qDebug() << "FTNoIR_Tracker_SM::Initialize says: Starting ";
	loadSettings();

	try {
		// Initialize the faceAPI Qt library
		sm::faceapi::qt::initialize();
		smLoggingSetFileOutputEnable( false );

		// Initialize the API
		faceapi_scope = new APIScope();

		//if (APIScope::internalQtGuiIsDisabled()){
		//	QMessageBox::warning(0,"faceAPI Error","Something Bad",QMessageBox::Ok,QMessageBox::NoButton);
		//}

		// Create head-tracking engine v2 using first detected webcam
		CameraInfo::registerType(SM_API_CAMERA_TYPE_WDM);
		_engine = QSharedPointer<HeadTrackerV2>(new HeadTrackerV2());	

	} 
	catch (sm::faceapi::Error &e)
	{
		/* ERROR with camera */
		QMessageBox::warning(0,"faceAPI Error",e.what(),QMessageBox::Ok,QMessageBox::NoButton);
	}



	// Show the video widget
	qDebug() << "FTNoIR_Tracker_SM::Initialize says: videoframe = " << videoframe;

	// QMainWindow derived class. See mainwindow.h
    QSharedPointer<CameraBase> camera;
    main_window = new MainWindow(camera,_engine,0);
    main_window->show();

	//videoframe->show();
	//_display = new VideoDisplayWidget( _engine, videoframe, 0 );
	//l = new QVBoxLayout(videoframe);
	//l->setMargin(0);
	//l->setSpacing(0);
	//l->addWidget(_display);

	return;
}

void FTNoIR_Tracker_SM::StartTracker( HWND parent_window )
{

	// starts the faceapi engine
	if (_engine->state() != SM_API_ENGINE_STATE_HT_TRACKING) {
		_engine->start();
	}

	// some parameteres [optional]
	smHTSetHeadPosePredictionEnabled( _engine->handle(), false);
	smHTSetLipTrackingEnabled( _engine->handle(), false);
	smLoggingSetFileOutputEnable( false );
	return;
}

void FTNoIR_Tracker_SM::StopTracker()
{

	qDebug() << "FTNoIR_Tracker_SM::StopTracker says: Starting ";
	// stops the faceapi engine
	_engine->stop();
	return;
}

bool FTNoIR_Tracker_SM::GiveHeadPoseData(THeadPoseData *data)
{
	smEngineHeadPoseData head_pose;					// headpose from faceAPI
	smEngineHeadPoseData temp_head_pose;			// headpose from faceAPI

    smReturnCode smret = smHTCurrentHeadPose(_engine->handle(), &temp_head_pose);
	memcpy(&head_pose, &temp_head_pose, sizeof(smEngineHeadPoseData));

	data->x     = head_pose.head_pos.x * 100.0f;					// From meters to centimeters
	data->y     = head_pose.head_pos.y * 100.0f;
	data->z     = head_pose.head_pos.z * 100.0f;
	data->yaw   = head_pose.head_rot.y_rads * 57.295781f;		// From rads to degrees
	data->pitch = head_pose.head_rot.x_rads * 57.295781f;
	data->roll  = head_pose.head_rot.z_rads * 57.295781f;

	return ( head_pose.confidence > 0 );
}

bool FTNoIR_Tracker_SM::setParameterValue(const int index, const float newvalue)
{
	if ((index >= 0) && (index < parameterValueAsFloat.size()))
	{
		//
		// Limit the new value, using the defined range.
		//
		if (newvalue < parameterRange[index].first) {
			parameterValueAsFloat[index] = parameterRange[index].first;
		}
		else {
			if (newvalue > parameterRange[index].second) {
				parameterValueAsFloat[index] = parameterRange[index].second;
			}
			else {
				parameterValueAsFloat[index] = newvalue;
			}
		}

//		updateParameterString(index);
		return true;
	}
	else
	{
		return false;
	}
};

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

	qDebug() << "FTNoIR_Tracker_SM::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() << "FTNoIR_Tracker_SM::loadSettings says: iniFile = " << currentFile;

	iniFile.beginGroup ( "FTNClient" );
	setParameterValue(kPortAddress, (float) iniFile.value ( "PortNumber", 5550 ).toInt());
	iniFile.endGroup ();
}


////////////////////////////////////////////////////////////////////////////////
// Factory function that creates instances if the Tracker object.

// Export both decorated and undecorated names.
//   GetTracker     - Undecorated name, which can be easily used with GetProcAddress
//                Win32 API function.
//   _GetTracker@0  - Common name decoration for __stdcall functions in C language.
#pragma comment(linker, "/export:GetTracker=_GetTracker@0")

FTNOIR_TRACKER_BASE_EXPORT TRACKERHANDLE __stdcall GetTracker()
{
	return new FTNoIR_Tracker_SM;
}