#include "Stimulus.h"
#include "ECTrainer.h"
#include "Marker.h"
#include <iostream>
#include <fstream>
#include <Windows.h>
#include <tchar.h>

// コンストラクタ
Stimulus::Stimulus(ECTrainer* pEct, Marker* pMarker)
	:BaseProcess(pEct)
	, _pMarker(pMarker)
	, _StimNo(0)
{
}

// 初期化
bool Stimulus::Init() {
	_DispBuffer = cv::imread(OPENING_FILE);

	std::ifstream ifs(DATA_FILE);
	if (!ifs)
	{
		OutputDebugString(_T("Can't open stim data file."));
		return false;
	}
	while (!ifs.eof()) {
		StimImage st;
		ifs >> st.filename >> st.eyes[0].x >> st.eyes[0].y >> st.eyes[1].x >> st.eyes[1].y;
		_StimImages.push_back(st);
	}

	return true;
}

// ループ
bool Stimulus::MainLoop() {
	while (_pEct->IsRunning()) {

		_pEct->SetDispBuffer(_DispBuffer);
		Sleep(30);
	}

	return true;
}


// キャリブレーション開始
void Stimulus::StartCalib() {
	_DispBuffer = cv::imread(CALIB_FILE);
}

// 刺激画像開始
void Stimulus::StartImage() {
	_StimNo = 0;
	SetStimulus();
}

// 刺激をセット
void Stimulus::SetStimulus() {
	_DispBuffer = cv::imread(_StimImages[_StimNo].filename);
	_pMarker->Generate(_DispBuffer.size());
	_pMarker->DrawMarker(_DispBuffer);
}