/**
 * Relational pipes
 * Copyright © 2018 František Kučera (Frantovo.cz, GlobalCode.info)
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program. If not, see <http://www.gnu.org/licenses/>.
 */
#pragma once

#include <vector>

#include <QtCore/QAbstractTableModel>
#include <QtCore/QHash>
#include <QtCore/QRect>
#include <QtCore/QVector>
#include <QtCore/QTime>
#include <QtGui/QColor>

#include <relpipe/reader/typedefs.h>
#include <relpipe/reader/handlers/AttributeMetadata.h>

using namespace relpipe::reader;
using namespace relpipe::reader::handlers;

class RelpipeTableModel : public QAbstractTableModel {
	Q_OBJECT
private:
	std::vector<AttributeMetadata> attributes;
	QList<QVector<QVariant> * > records;
	int attributeCounter = 0;

	/**
	 * @param index
	 * @return whether data at this row/clumn are already loaded (last row might contain columns that are still waiting to be filled)
	 */
	boolean_t isFilled(const QModelIndex &index) const {
		return (index.row() * columnCount() + index.column()) <= attributeCounter;
	}
public:

	RelpipeTableModel(std::vector<AttributeMetadata> attributes, QObject *parent = 0) : QAbstractTableModel(parent), attributes(attributes) {
	}

	virtual ~RelpipeTableModel() {
		qDeleteAll(records);
	}

	int rowCount(const QModelIndex &parent = QModelIndex()) const {
		return records.count();
	}

	int columnCount(const QModelIndex &parent = QModelIndex()) const {
		return attributes.size();
	}

	QVariant headerData(int section, Qt::Orientation orientation, int role = Qt::ItemDataRole::DisplayRole) const {
		if (orientation == Qt::Orientation::Horizontal && role == Qt::ItemDataRole::DisplayRole) return QString::fromStdWString(attributes[section].getAttributeName().c_str());
		if (orientation == Qt::Orientation::Horizontal && role == Qt::ItemDataRole::ToolTipRole) return QString::fromStdWString(attributes[section].getTypeName().c_str());
		if (orientation == Qt::Orientation::Vertical && role == Qt::ItemDataRole::DisplayRole) return QString("%1").arg(section + 1);
		return QVariant();
	}

	QVariant data(const QModelIndex &index, int role = Qt::ItemDataRole::DisplayRole) const {
		if (role == Qt::ItemDataRole::DisplayRole || role == Qt::ItemDataRole::EditRole) return records[index.row()]->at(index.column());
		else return QVariant();
	}

	bool setData(const QModelIndex &index, const QVariant &value, int role = Qt::ItemDataRole::EditRole) {
		if (index.isValid() && role == Qt::ItemDataRole::EditRole) {
			records[index.row()]->replace(index.column(), value);
			emit dataChanged(index, index);
			return true;
		} else {
			return false;
		}
	}

	void addAttribute(const string_t &value) {
		int column = attributeCounter % columnCount();
		int row = attributeCounter / columnCount();
		if (row >= records.size()) records.append(new QVector<QVariant>(columnCount()));

		QString valueString = QString::fromWCharArray(value.c_str());
		QVariant valueVariant;
		boolean_t conversionOk = false;
		if (attributes[column].getTypeId() == TypeId::INTEGER) {
			valueVariant = valueString.toUInt(&conversionOk);
		}
		// TODO: boolean → checkboxes
		// else if (attributes[column].getTypeId() == TypeId::BOOLEAN) {
		//	valueVariant = L"true" == value;
		//	conversionOk = true;
		// }

		if (!conversionOk) valueVariant = valueString;

		setData(index(row, column), valueVariant);
		if (column == 0) emit layoutChanged(); // FIXME: emit other signal ~ begin..., end..., rowsInserted(???  index(0,0), row, row);
		attributeCounter++;
	}
	
	TypeId attributeType(int section) {
		return attributes[section].getTypeId();
	}

	Qt::ItemFlags flags(const QModelIndex &index) const {
		if (isFilled(index)) {
			return QAbstractItemModel::flags(index) | Qt::ItemFlag::ItemIsEditable;
		} else {
			return QAbstractItemModel::flags(index);
		}
	}

};