#include "TPrModel.h"

TPrModel::TPrModel(QObject *parent)
    : V2SvkModel(parent)
{
	m_tests = {
	    {"Temporaladverb Präsens", {"Affe", "Affe", "Schwein", "Schwein", "Gans", "Gans"}},
	    {"SKV: /-e/ o. Stamm", {"Affe", "Affe", "Schwein", "Schwein", "Gans", "Gans"}},
	};
}

unsigned int TPrModel::getV2Points() const
{
	unsigned int points = 0;

	for (auto testIndex : {0})
	{
		const auto &test = m_tests.at(testIndex);

		for (const auto &item : test.items())
		{
			if (item.isChecked())
			{
				points++;
			}
		}
	}

	return points;
}

unsigned int TPrModel::getSvkPoints() const
{
	unsigned int points = 0;

	for (auto testIndex : {1})
	{
		const auto &test = m_tests.at(testIndex);

		for (const auto &item : test.items())
		{
			if (item.isChecked())
			{
				points++;
			}
		}
	}

	return points;
}

void TPrModel::write(ESGRAF48::V2SvkModel &model) const
{
	auto writeTwoVals = [&](ESGRAF48::V2SvkModel::TwoEach *modelData, int testIndex) {
		if (modelData != nullptr)
		{
			const auto &testItems = m_tests.at(testIndex).items();

			modelData->set_affe1(testItems[0].isChecked());
			modelData->set_affe2(testItems[1].isChecked());
			modelData->set_schwein1(testItems[2].isChecked());
			modelData->set_schwein2(testItems[3].isChecked());
			modelData->set_gans1(testItems[4].isChecked());
			modelData->set_gans2(testItems[5].isChecked());
		}
	};

	writeTwoVals(model.mutable_temppraes(), 0);
	writeTwoVals(model.mutable_svke1(), 1);
}

void TPrModel::read(const ESGRAF48::V2SvkModel &model)
{
	auto readTwoVals = [&](const ESGRAF48::V2SvkModel::TwoEach &modelData, int testIndex) {
		auto &testItems = m_tests.at(testIndex).items();

		testItems[0].setState(modelData.affe1());
		testItems[1].setState(modelData.affe2());
		testItems[2].setState(modelData.schwein1());
		testItems[3].setState(modelData.schwein2());
		testItems[4].setState(modelData.gans1());
		testItems[5].setState(modelData.gans2());
	};

	readTwoVals(model.temppraes(), 0);
	readTwoVals(model.svke1(), 1);

	emit dataChanged(index(0, 0), index(rowCount() - 1, columnCount() - 1));
}

std::set<int> TPrModel::v2Tests() const
{
	return {0};
};

std::set<int> TPrModel::svkTests() const
{
	return {1};
};