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@842: 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@842: FindResultItem *FindResultModel::findTreeItem(TreeItem *)
insilmaril@841: {
insilmaril@842: 	return NULL;	// FIXME-3 not used so far
insilmaril@841: }
insilmaril@841: 
insilmaril@842: FindResultItem*  FindResultModel::addItem (TreeItem *ti)
insilmaril@825: {
insilmaril@842: 	FindResultItem *ni=NULL;
insilmaril@829: 	if (ti)
insilmaril@829: 	{
insilmaril@842: 		QModelIndex parix (index (rootItem));
insilmaril@829: 		
insilmaril@842: 		emit (layoutAboutToBeChanged() );
insilmaril@825: 
insilmaril@842: 		int n=rowCount (parix);
insilmaril@842: 		beginInsertRows (parix,n,n);
insilmaril@842: 		if (rootItem->insertChildren (n,1,0) )
insilmaril@841: 		{
insilmaril@842: 			QModelIndex ix=index(n,0,QModelIndex());
insilmaril@842: 			setData (ix,QVariant(ti->getHeading()),Qt::EditRole);
insilmaril@842: 			ni=getItem(ix);
insilmaril@842: 			ni->setOriginal (ti);
insilmaril@829: 		}
insilmaril@842: 		endInsertRows ();
insilmaril@842: 
insilmaril@842: 		emit (layoutChanged() );
insilmaril@829: 	}
insilmaril@842: 	return ni;
insilmaril@829: }
insilmaril@825: 
insilmaril@842: FindResultItem*  FindResultModel::addSubItem (FindResultItem *parent,const QString &s, TreeItem *pi, int i)
insilmaril@829: {
insilmaril@842: 	FindResultItem *ni=NULL;
insilmaril@842: 	if (pi && parent)
insilmaril@829: 	{
insilmaril@842: 		QModelIndex parix ( index (parent));
insilmaril@829: 		
insilmaril@842: 		emit (layoutAboutToBeChanged() );
insilmaril@825: 
insilmaril@842: 		int n=rowCount (parix);
insilmaril@842: 		beginInsertRows (parix,n,n);
insilmaril@842: 
insilmaril@842: 		QModelIndex ix;
insilmaril@842: 		if (parent->insertChildren (n,1,0))
insilmaril@842: 		{
insilmaril@842: 			ix=index(n,0,parix);
insilmaril@842: 			setData (ix,QVariant(s),Qt::EditRole);
insilmaril@842: 			ni=getItem(ix);
insilmaril@842: 			ni->setOriginal (pi);
insilmaril@842: 			ni->setOriginalIndex (i);
insilmaril@829: 		}
insilmaril@842: 		endInsertRows ();
insilmaril@842: 		emit (layoutChanged() );
insilmaril@829: 	}
insilmaril@842: 	return ni;
insilmaril@829: }
insilmaril@825: 
insilmaril@842: void FindResultModel::setSearchString( const QString &s)
insilmaril@837: {
insilmaril@842: 	searchString=s;
insilmaril@837: }
insilmaril@825: 
insilmaril@842: QString FindResultModel::getSearchString()
insilmaril@842: {
insilmaril@842: 	return searchString;
insilmaril@842: }
insilmaril@842: 
insilmaril@842: void FindResultModel::setSearchFlags( QTextDocument::FindFlags f)
insilmaril@842: {
insilmaril@842: 	searchFlags=f;
insilmaril@842: }
insilmaril@842: 
insilmaril@842: QTextDocument::FindFlags FindResultModel::getSearchFlags()
insilmaril@842: {
insilmaril@842: 	return searchFlags;
insilmaril@842: }
insilmaril@842: