#include "cpartlistitemdelegate.h"

#include "cpartdistributor.h"
#include "cpartlistitem.h"

#include <QComboBox>
#include <QLineEdit>
#include <QDoubleSpinBox>
#include <QTextEdit>
#include <QString>
#include <QStringList>
#include <QRect>


cPartListItemDelegate::cPartListItemDelegate(QObject *parent) :
	QStyledItemDelegate(parent)
{
}

cPartListItemDelegate::~cPartListItemDelegate()
{
}

QWidget* cPartListItemDelegate::createEditor( QWidget *parent, const QStyleOptionViewItem &option, const QModelIndex &index ) const
{
	switch(index.column())
	{
	case 0: // reference
	{
		QTextEdit*				lpEdit		= new QTextEdit(parent);
		return(lpEdit);
	}
	case 3:	// distributor
	{
		cPartDistributorList*	lpList		= qvariant_cast<cPartDistributorList*>(index.data(Qt::UserRole+1));
		QComboBox*				lpComboBox	= new QComboBox(parent);
		cPartlistItem*			lpItem		= qvariant_cast<cPartlistItem*>(index.data(Qt::UserRole));

		if(!lpItem || !lpList)
			return(0);

		for(int x = 0;x < lpList->count();x++)
		{
			cPartDistributor*	lpPartDistributor	= lpList->at(x);

			if(lpPartDistributor->part()->id() == lpItem->partID())
				lpComboBox->addItem(lpPartDistributor->distributor()->name(), QVariant::fromValue(lpPartDistributor));
		}
		return(lpComboBox);
	}
	case 5:	// price
	{
		QDoubleSpinBox*	lpSpinBox	= new QDoubleSpinBox(parent);
		lpSpinBox->setMinimum(0);
		lpSpinBox->setMaximum(999999);
		lpSpinBox->setDecimals(2);
		return(lpSpinBox);
	}
	case 6: // description
		return(QStyledItemDelegate::createEditor(parent, option, index));
	default:
		return(0);
	}

	return(0);
}

void cPartListItemDelegate::updateEditorGeometry(QWidget *editor, const QStyleOptionViewItem &option, const QModelIndex &index) const
{
	switch(index.column())
	{
	case 0: // reference
	{
		QRect	rect	= option.rect;
		rect.setWidth(100);
		rect.setHeight(100);
		static_cast<QTextEdit*>(editor)->setGeometry(rect);
	}
		break;
	default:
		editor->setGeometry(option.rect);
		break;
	}
}

void cPartListItemDelegate::setEditorData ( QWidget *editor, const QModelIndex &index ) const
{
	QComboBox*	lpComboBox	= qobject_cast<QComboBox*>(editor);
	QTextEdit*	lpTextEdit	= qobject_cast<QTextEdit*>(editor);

	switch(index.column())
	{
	case 0:
		{
			QStringList	list	= index.data(Qt::EditRole).toString().split(", ");
			lpTextEdit->setText(list.join("\n"));
		}
		break;
	case 3:
	{
		QString	currentText	= index.data(Qt::EditRole).toString();
		int		cbIndex		= lpComboBox->findText(currentText);

		if(cbIndex >= 0)
			lpComboBox->setCurrentIndex(cbIndex);
	}
	default:
		QStyledItemDelegate::setEditorData(editor, index);
	}
}

void cPartListItemDelegate::setModelData ( QWidget *editor, QAbstractItemModel *model, const QModelIndex &index ) const
{
	QComboBox*		lpComboBox	= qobject_cast<QComboBox*>(editor);
	QDoubleSpinBox*	lpSpinBox	= qobject_cast<QDoubleSpinBox*>(editor);
	QTextEdit*		lpTextEdit	= qobject_cast<QTextEdit*>(editor);

	switch(index.column())
	{
	case 0:
	{
		QStringList	list	= lpTextEdit->toPlainText().split("\n");
		QStringList	list1;

		for(int x = 0;x < list.count();x++)
		{
			if(!list.at(x).isEmpty())
				list1.append(list.at(x));
		}
		model->setData(index, list1.join(", "), Qt::EditRole);
	}
		break;
	case 3:
	{
		cPartDistributor*		lpPartDistributor	= qvariant_cast<cPartDistributor*>(lpComboBox->currentData());

		model->setData(index, lpComboBox->currentText(), Qt::EditRole);
		model->setData(index, QVariant::fromValue(lpPartDistributor), Qt::UserRole);
		QStandardItemModel*	lpModel		= (QStandardItemModel*)index.model();
		QStandardItem*		lpItem		= lpModel->itemFromIndex(lpModel->index(index.row(), 4));
		lpItem->setText(QString::number(lpPartDistributor->price(), 'f', 2));
	}
		break;
	case 5:
		if(lpSpinBox)
		{
			QStandardItemModel*	lpModel		= (QStandardItemModel*)index.model();
			QStandardItem*		lpItem		= lpModel->itemFromIndex(index);
			lpItem->setText(QString::number(lpSpinBox->value(), 'f', 2));
		}
		break;
	default:
		QStyledItemDelegate::setModelData(editor, model, index);
		break;
	}
}