Newer
Older
PrismSoftware / ECTrainer2 / ECTrainer.h
#pragma once

#include <Windows.h>
#include "myOpenCV.h"

#define ADDR "192.168.71.50"

class ECTrainerGUI;
class SceneCamera;
class Stimulus;
class Marker;
class ImageProc;
class EyeTrack;
class TobiiREST;
class BitalMonitor;
class BaseProcess;

class ECTrainer
{
private:
	enum PROC {GUI, IMGPROC, SCNCAM, STIM, EYETR, REST, BITAL, NUM  };
	HINSTANCE _hInstance;
	BaseProcess* _pProcs[PROC::NUM];
	Marker* _pMarker;
	bool _Running;			// 実行中フラグ
	bool _HomographyOK;		// 変換行列の獲得有無
	int _CalibResult;		// キャリブレーション結果 0:未実施 1:成功 -1:失敗
	cv::Size _SceneSize;	// 視界カメラの画像サイズ
	std::wstring _MovieToShow;	// 再生する動画ファイル
	int _HeartBeat;			// 心拍数

	// スレッド開始点
	static DWORD WINAPI ThreadEntry(LPVOID lpParameter);
	static DWORD WINAPI KeepAliveThreadEntry(LPVOID lpParameter);

public:
	ECTrainer(HINSTANCE hInstance);
	~ECTrainer();
	bool Process();
	void CalibStart();
	int CheckCalibResult();
	void StartStim();
	void StopStim();
	void NextStim();
	void SetSceneBuffer(cv::Mat& img);
	void SetDispBuffer(cv::Mat& img);
	void SetGazeV(cv::Point gazeV);
	int BatteryLevel();
	cv::Point GetGazeV();
	cv::Point2f GetGazeI();
	cv::Point2f GetEyeR();
	cv::Point2f GetEyeL();

	// インライン関数
	bool IsRunning() { return _Running; }
	void Stop() { _Running = false; }
	void SetHomographyStatus(bool ok) { _HomographyOK = ok; }
	bool GetHomographyStatus() { return _HomographyOK; }
	void SetSceneSize(cv::Size s) { _SceneSize = s; }
	cv::Size GetSceneSize() { return _SceneSize; }
	void SetCalibResult(int result) { _CalibResult = result; }
	void SetMovieToShow(std::wstring file) { _MovieToShow = file; }
	std::wstring GetMovieToShow() { return _MovieToShow; }
	void SetHeartBeat(int hb) { _HeartBeat = hb; }
	int GetHeartBeat() { return _HeartBeat; }

};