#include <QtGui>

#include "attributedelegate.h"

#include <iostream>

using namespace::std;

AttributeDelegate::AttributeDelegate(QObject *parent)

    : QItemDelegate(parent)

{

}

QWidget *AttributeDelegate::createEditor(QWidget *parent,

    const QStyleOptionViewItem &/* option */,

    const QModelIndex & index ) const

{

	int col=index.column();

	int row=index.row();

	if (col==0 && row==index.model()->rowCount() -1 )

	{

		//We are editing a new attribute, starting with attribute name

		QComboBox *editor = new QComboBox(parent);

		editor->insertItems (0,attributeTable->getKeys());

		return editor;

	}

	if (col==1 && row==index.model()->rowCount() -1 )

	{

		cout << "Edit value now..."<<endl;

		//We are editing a new attribute, starting with attribute name

		QComboBox *editor = new QComboBox(parent);

		editor->insertItems (0,attributeTable->getKeys());

		return editor;

	}

	// Is there already an atttribute defined or 

	// do we need to create a new one?

	QVariant var=index.model()->data(index.model()->index(row,2,QModelIndex()));

	QString typeName=var.toString();

	cout << "AttrDel::createEditor type="<<qPrintable (typeName)<<endl;

	if (typeName=="IntList")

	{

		QSpinBox *editor = new QSpinBox(parent);

		editor->setMinimum(0);

		editor->setMaximum(5);

		return editor;

	} else if (typeName=="FreeInt")

	{

		QSpinBox *editor = new QSpinBox(parent);

		editor->setMinimum(0);

		editor->setMaximum(100);

		return editor;

	} else if (typeName=="FreeString")

	{

		QComboBox *editor = new QComboBox(parent);

		return editor;

	} else if (typeName=="StringList")

	{

		QComboBox *editor = new QComboBox(parent);

		return editor;

	} 

    return NULL;

}

void AttributeDelegate::setEditorData(QWidget *editor,

                                    const QModelIndex &index) const

{

	QVariant value= index.model()->data(index, Qt::DisplayRole);

	switch (value.type())

	{

		case QVariant::Int:

		{

			int value = index.model()->data(index, Qt::DisplayRole).toInt();

			QSpinBox *spinBox = static_cast<QSpinBox*>(editor);

			spinBox->setValue(value);

			break;

		}	

		/*

		{

			QString value = index.model()->data(index, Qt::DisplayRole).toString();

			QLineEdit *le= static_cast<QLineEdit*>(editor);

			le->setText(value);

			break;

		}

		*/

		case QVariant::String:

		{

			QComboBox *cb= static_cast<QComboBox*>(editor);

			QStringList sl;

			sl<< index.model()->data(index, Qt::DisplayRole).toString();

			cb->insertStringList (sl);

			break;

		}

		default: 

			break;

	}		

}

void AttributeDelegate::setModelData(QWidget *editor, QAbstractItemModel *model,

                                   const QModelIndex &index) const

{

	QVariant value= index.model()->data(index, Qt::DisplayRole);

	switch (value.type())

	{

		case QVariant::Int:

		{

			QSpinBox *spinBox = static_cast<QSpinBox*>(editor);

			spinBox->interpretText();

			model->setData(index, spinBox->value(), Qt::EditRole);

			break;

		}	

		case QVariant::String:

		{

			QComboBox *cb = static_cast<QComboBox*>(editor);

			model->setData(index, cb->currentText(), Qt::EditRole);

			break;

		}

		default:

			break;

	}		

}

void AttributeDelegate::updateEditorGeometry(QWidget *editor,

    const QStyleOptionViewItem &option, const QModelIndex &/* index */) const

{

    editor->setGeometry(option.rect);

}

void AttributeDelegate::setAttributeTable (AttributeTable *table)

{

	attributeTable=table;

}