insilmaril@825: #include <QtGui>
insilmaril@825: 
insilmaril@825: #include "findresultitem.h"
insilmaril@825: #include "findresultmodel.h"
insilmaril@829: #include "treeitem.h"
insilmaril@825: 
insilmaril@825: FindResultModel::FindResultModel( QObject *parent)
insilmaril@825:     : QAbstractItemModel(parent)
insilmaril@825: {
insilmaril@825:     QVector<QVariant> rootData;
insilmaril@829: 	rootData << "Heading";
insilmaril@825:     rootItem = new FindResultItem(rootData);
insilmaril@825: }
insilmaril@825: 
insilmaril@825: FindResultModel::~FindResultModel()
insilmaril@825: {
insilmaril@825:     delete rootItem;
insilmaril@825: }
insilmaril@825: 
insilmaril@829: void FindResultModel::clear()
insilmaril@829: {
insilmaril@829: 	if (rootItem->childCount()>0)
insilmaril@829: 		removeRows (0,rowCount (QModelIndex ()));
insilmaril@829: }
insilmaril@829: 
insilmaril@825: int FindResultModel::columnCount(const QModelIndex & /* parent */) const
insilmaril@825: {
insilmaril@825:     return rootItem->columnCount();
insilmaril@825: }
insilmaril@825: 
insilmaril@825: QVariant FindResultModel::data(const QModelIndex &index, int role) const
insilmaril@825: {
insilmaril@825:     if (!index.isValid())
insilmaril@825:         return QVariant();
insilmaril@825: 
insilmaril@825:     if (role != Qt::DisplayRole && role != Qt::EditRole)
insilmaril@825:         return QVariant();
insilmaril@825: 
insilmaril@825:     FindResultItem *item = getItem(index);
insilmaril@825: 
insilmaril@825:     return item->data(index.column());
insilmaril@825: }
insilmaril@825: 
insilmaril@825: Qt::ItemFlags FindResultModel::flags(const QModelIndex &index) const
insilmaril@825: {
insilmaril@825:     if (!index.isValid())
insilmaril@825:         return 0;
insilmaril@825: 
insilmaril@825:     return Qt::ItemIsEditable | Qt::ItemIsEnabled | Qt::ItemIsSelectable;
insilmaril@825: }
insilmaril@825: 
insilmaril@825: QVariant FindResultModel::headerData(int section, Qt::Orientation orientation,
insilmaril@825:                                int role) const
insilmaril@825: {
insilmaril@825:     if (orientation == Qt::Horizontal && role == Qt::DisplayRole)
insilmaril@825:         return rootItem->data(section);
insilmaril@825: 
insilmaril@825:     return QVariant();
insilmaril@825: }
insilmaril@825: 
insilmaril@829: QModelIndex FindResultModel::index (FindResultItem *fri)
insilmaril@829: {
insilmaril@829: 	if (!fri->parent())
insilmaril@829: 		return QModelIndex();
insilmaril@829: 	else	
insilmaril@829: 		return createIndex (fri->row(),0,fri);
insilmaril@829: }
insilmaril@829: 
insilmaril@829: 
insilmaril@825: QModelIndex FindResultModel::index(int row, int column, const QModelIndex &parent) const
insilmaril@825: {
insilmaril@825:     if (parent.isValid() && parent.column() != 0)
insilmaril@825:         return QModelIndex();
insilmaril@825: 
insilmaril@825:     FindResultItem *parentItem = getItem(parent);
insilmaril@825: 
insilmaril@825:     FindResultItem *childItem = parentItem->child(row);
insilmaril@825:     if (childItem)
insilmaril@825:         return createIndex(row, column, childItem);
insilmaril@825:     else
insilmaril@825:         return QModelIndex();
insilmaril@825: }
insilmaril@825: 
insilmaril@825: bool FindResultModel::insertColumns(int position, int columns, const QModelIndex &parent)
insilmaril@825: {
insilmaril@825:     bool success;
insilmaril@825: 
insilmaril@825:     beginInsertColumns(parent, position, position + columns - 1);
insilmaril@825:     success = rootItem->insertColumns(position, columns);
insilmaril@825:     endInsertColumns();
insilmaril@825: 
insilmaril@825:     return success;
insilmaril@825: }
insilmaril@825: 
insilmaril@825: bool FindResultModel::insertRows(int position, int rows, const QModelIndex &parent)
insilmaril@825: {
insilmaril@825:     FindResultItem *parentItem = getItem(parent);
insilmaril@825:     bool success;
insilmaril@825: 
insilmaril@825:     beginInsertRows(parent, position, position + rows - 1);
insilmaril@825:     success = parentItem->insertChildren(position, rows, rootItem->columnCount());
insilmaril@825:     endInsertRows();
insilmaril@825: 
insilmaril@825:     return success;
insilmaril@825: }
insilmaril@825: 
insilmaril@825: QModelIndex FindResultModel::parent(const QModelIndex &index) const
insilmaril@825: {
insilmaril@825:     if (!index.isValid())
insilmaril@825:         return QModelIndex();
insilmaril@825: 
insilmaril@825:     FindResultItem *childItem = getItem(index);
insilmaril@825:     FindResultItem *parentItem = childItem->parent();
insilmaril@825: 
insilmaril@825:     if (parentItem == rootItem)
insilmaril@825:         return QModelIndex();
insilmaril@825: 
insilmaril@825:     return createIndex(parentItem->childNumber(), 0, parentItem);
insilmaril@825: }
insilmaril@825: 
insilmaril@825: bool FindResultModel::removeColumns(int position, int columns, const QModelIndex &parent)
insilmaril@825: {
insilmaril@825:     bool success;
insilmaril@825: 
insilmaril@825:     beginRemoveColumns(parent, position, position + columns - 1);
insilmaril@825:     success = rootItem->removeColumns(position, columns);
insilmaril@825:     endRemoveColumns();
insilmaril@825: 
insilmaril@825:     if (rootItem->columnCount() == 0)
insilmaril@825:         removeRows(0, rowCount());
insilmaril@825: 
insilmaril@825:     return success;
insilmaril@825: }
insilmaril@825: 
insilmaril@825: bool FindResultModel::removeRows(int position, int rows, const QModelIndex &parent)
insilmaril@825: {
insilmaril@825:     FindResultItem *parentItem = getItem(parent);
insilmaril@825:     bool success = true;
insilmaril@825: 
insilmaril@825:     beginRemoveRows(parent, position, position + rows - 1);
insilmaril@825:     success = parentItem->removeChildren(position, rows);
insilmaril@825:     endRemoveRows();
insilmaril@825: 
insilmaril@825:     return success;
insilmaril@825: }
insilmaril@825: 
insilmaril@825: int FindResultModel::rowCount(const QModelIndex &parent) const
insilmaril@825: {
insilmaril@825:     FindResultItem *parentItem = getItem(parent);
insilmaril@825: 
insilmaril@825:     return parentItem->childCount();
insilmaril@825: }
insilmaril@825: 
insilmaril@825: bool FindResultModel::setData(const QModelIndex &index, const QVariant &value,
insilmaril@825:                         int role)
insilmaril@825: {
insilmaril@825:     if (role != Qt::EditRole)
insilmaril@825:         return false;
insilmaril@825: 
insilmaril@825:     FindResultItem *item = getItem(index);
insilmaril@825:     bool result = item->setData(index.column(), value);
insilmaril@825: 
insilmaril@825:     if (result)
insilmaril@825:         emit dataChanged(index, index);
insilmaril@825: 
insilmaril@825:     return result;
insilmaril@825: }
insilmaril@825: 
insilmaril@825: bool FindResultModel::setHeaderData(int section, Qt::Orientation orientation,
insilmaril@825:                               const QVariant &value, int role)
insilmaril@825: {
insilmaril@825:     if (role != Qt::EditRole || orientation != Qt::Horizontal)
insilmaril@825:         return false;
insilmaril@825: 
insilmaril@825:     bool result = rootItem->setData(section, value);
insilmaril@825: 
insilmaril@825:     if (result)
insilmaril@825:         emit headerDataChanged(orientation, section, section);
insilmaril@825: 
insilmaril@825:     return result;
insilmaril@825: }
insilmaril@825: 
insilmaril@841: FindResultItem *FindResultModel::getItem(const QModelIndex &index) const
insilmaril@841: {
insilmaril@841:     if (index.isValid()) {
insilmaril@841:         FindResultItem *item = static_cast<FindResultItem*>(index.internalPointer());
insilmaril@841:         if (item) return item;
insilmaril@841:     }
insilmaril@841:     return rootItem;
insilmaril@841: }
insilmaril@841: 
insilmaril@841: FindResultItem *FindResultModel::findTreeItem(TreeItem *ti)
insilmaril@841: {
insilmaril@841: }
insilmaril@841: 
insilmaril@829: void FindResultModel::addItem (TreeItem *ti)
insilmaril@825: {
insilmaril@829: 	if (ti)
insilmaril@829: 	{
insilmaril@829: 		QModelIndex ix (index (rootItem));
insilmaril@829: 		
insilmaril@829: 		if (!insertRow(ix.row()+1, ix.parent()))
insilmaril@829: 			return;
insilmaril@825: 
insilmaril@841: 		for (int column = 0; column < columnCount(ix.parent()); ++column) 
insilmaril@841: 		{
insilmaril@829: 			QModelIndex child = index(ix.row()+1, column, ix.parent());
insilmaril@829: 			setData(child, QVariant(ti->getHeading()), Qt::EditRole);
insilmaril@829: 			getItem(child)->setOriginal (ti);
insilmaril@829: 		}
insilmaril@829: 	}
insilmaril@829: }
insilmaril@825: 
insilmaril@829: void FindResultModel::addItem (const QString &s)
insilmaril@829: {
insilmaril@829: 	if (!s.isEmpty())
insilmaril@829: 	{
insilmaril@829: 		QModelIndex ix ( index (rootItem));
insilmaril@829: 		
insilmaril@829: 		if (!insertRow(ix.row()+1, ix.parent()))
insilmaril@829: 			return;
insilmaril@825: 
insilmaril@829: 		for (int column = 0; column < columnCount(ix.parent()); ++column) {
insilmaril@829: 			QModelIndex child = index(ix.row()+1, column, ix.parent());
insilmaril@829: 			setData(child, QVariant(s), Qt::EditRole);
insilmaril@829: 		}
insilmaril@829: 	}
insilmaril@829: }
insilmaril@825: 
insilmaril@841: void FindResultModel::addSubItem (TreeItem *parent,const QString &s, TreeItem *ti, int i)
insilmaril@837: {
insilmaril@841: 	if (ti)
insilmaril@841: 	{
insilmaril@841: 		QModelIndex ix ( index (rootItem));
insilmaril@841: 		
insilmaril@841: 		if (!insertRow(ix.row()+1, ix.parent()))
insilmaril@841: 			return;
insilmaril@841: 
insilmaril@841: 		for (int column = 0; column < columnCount(ix.parent()); ++column) {
insilmaril@841: 			QModelIndex child = index(ix.row()+1, column, ix.parent());
insilmaril@841: 			setData(child, QVariant("Note: "+ti->getHeading()), Qt::EditRole);
insilmaril@841: 			getItem(child)->setOriginal (ti);
insilmaril@841: 		}
insilmaril@841: 	}
insilmaril@837: }
insilmaril@825: