#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;
		std::cout << st.filename << st.eyes[0] << st.eyes[1] << std::endl;
		if (st.filename.size() > 0) _StimImages.push_back(st);
	}

	return true;
}

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

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

	return true;
}


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

// 開始
void Stimulus::Start() {
	_StimNo = 0;
	SetStimulus();
}

// 次へ
void Stimulus::Next() {
	_StimNo = (_StimNo + 1) % _StimImages.size();
	SetStimulus();
}

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