+#include <QtGui>
+
+#include "findresultitem.h"
+#include "findresultmodel.h"
+#include "settings.h"
+#include "treeitem.h"
+
+extern Settings settings;
+
+FindResultModel::FindResultModel(QObject *parent) : QAbstractItemModel(parent)
+{
+ QVector<QVariant> rootData;
+ rootData << "Heading";
+ rootItem = new FindResultItem(rootData);
+ showParentsLevel =
+ settings.value("/satellite/findResults/showParentsLevel", 1).toInt();
+}
+
+FindResultModel::~FindResultModel() { delete rootItem; }
+
+void FindResultModel::clear()
+{
+ if (rootItem->childCount() > 0)
+ removeRows(0, rowCount(QModelIndex()));
+}
+
+int FindResultModel::columnCount(const QModelIndex & /* parent */) const
+{
+ return rootItem->columnCount();
+}
+
+QVariant FindResultModel::data(const QModelIndex &index, int role) const
+{
+ if (!index.isValid())
+ return QVariant();
+
+ if (role != Qt::DisplayRole && role != Qt::EditRole)
+ return QVariant();
+
+ FindResultItem *item = getItem(index);
+
+ return item->data(index.column());
+}
+
+Qt::ItemFlags FindResultModel::flags(const QModelIndex &index) const
+{
+ if (!index.isValid())
+ return Qt::NoItemFlags;
+
+ return Qt::ItemIsEditable | Qt::ItemIsEnabled | Qt::ItemIsSelectable;
+}
+
+QVariant FindResultModel::headerData(int section, Qt::Orientation orientation,
+ int role) const
+{
+ if (orientation == Qt::Horizontal && role == Qt::DisplayRole)
+ return rootItem->data(section);
+
+ return QVariant();
+}
+
+QModelIndex FindResultModel::index(FindResultItem *fri)
+{
+ if (!fri->parent())
+ return QModelIndex();
+ else
+ return createIndex(fri->row(), 0, fri);
+}
+
+QModelIndex FindResultModel::index(int row, int column,
+ const QModelIndex &parent) const
+{
+ if (parent.isValid() && parent.column() != 0)
+ return QModelIndex();
+
+ FindResultItem *parentItem = getItem(parent);
+
+ FindResultItem *childItem = parentItem->child(row);
+ if (childItem)
+ return createIndex(row, column, childItem);
+ else
+ return QModelIndex();
+}
+
+bool FindResultModel::insertColumns(int position, int columns,
+ const QModelIndex &parent)
+{
+ bool success;
+
+ beginInsertColumns(parent, position, position + columns - 1);
+ success = rootItem->insertColumns(position, columns);
+ endInsertColumns();
+
+ return success;
+}
+
+bool FindResultModel::insertRows(int position, int rows,
+ const QModelIndex &parent)
+{
+ FindResultItem *parentItem = getItem(parent);
+ bool success;
+
+ beginInsertRows(parent, position, position + rows - 1);
+ success =
+ parentItem->insertChildren(position, rows, rootItem->columnCount());
+ endInsertRows();
+
+ return success;
+}
+
+QModelIndex FindResultModel::parent(const QModelIndex &index) const
+{
+ if (!index.isValid())
+ return QModelIndex();
+
+ FindResultItem *childItem = getItem(index);
+ FindResultItem *parentItem = childItem->parent();
+
+ if (parentItem == rootItem)
+ return QModelIndex();
+
+ return createIndex(parentItem->childNumber(), 0, parentItem);
+}
+
+bool FindResultModel::removeColumns(int position, int columns,
+ const QModelIndex &parent)
+{
+ bool success;
+
+ beginRemoveColumns(parent, position, position + columns - 1);
+ success = rootItem->removeColumns(position, columns);
+ endRemoveColumns();
+
+ if (rootItem->columnCount() == 0)
+ removeRows(0, rowCount());
+
+ return success;
+}
+
+bool FindResultModel::removeRows(int position, int rows,
+ const QModelIndex &parent)
+{
+ FindResultItem *parentItem = getItem(parent);
+ bool success = true;
+
+ beginRemoveRows(parent, position, position + rows - 1);
+ success = parentItem->removeChildren(position, rows);
+ endRemoveRows();
+
+ return success;
+}
+
+int FindResultModel::rowCount(const QModelIndex &parent) const
+{
+ FindResultItem *parentItem = getItem(parent);
+
+ return parentItem->childCount();
+}
+
+bool FindResultModel::setData(const QModelIndex &index, const QVariant &value,
+ int role)
+{
+ if (role != Qt::EditRole)
+ return false;
+
+ FindResultItem *item = getItem(index);
+ bool result = item->setData(index.column(), value);
+
+ if (result)
+ emit dataChanged(index, index);
+
+ return result;
+}
+
+bool FindResultModel::setHeaderData(int section, Qt::Orientation orientation,
+ const QVariant &value, int role)
+{
+ if (role != Qt::EditRole || orientation != Qt::Horizontal)
+ return false;
+
+ bool result = rootItem->setData(section, value);
+
+ if (result)
+ emit headerDataChanged(orientation, section, section);
+
+ return result;
+}
+
+FindResultItem *FindResultModel::getItem(const QModelIndex &index) const
+{
+ if (index.isValid()) {
+ FindResultItem *item =
+ static_cast<FindResultItem *>(index.internalPointer());
+ if (item)
+ return item;
+ }
+ return rootItem;
+}
+
+FindResultItem *FindResultModel::addItem(TreeItem *ti)
+{
+ FindResultItem *ni = NULL;
+ if (ti) {
+ QModelIndex parix(index(rootItem));
+
+ emit(layoutAboutToBeChanged());
+
+ int n = rowCount(parix);
+ beginInsertRows(parix, n, n);
+ if (rootItem->insertChildren(n, 1, 0)) {
+ QString h = ti->getHeadingPlainWithParents(showParentsLevel);
+ QModelIndex ix = index(n, 0, QModelIndex());
+ setData(ix, QVariant(h), Qt::EditRole);
+ ni = getItem(ix);
+ ni->setOriginal(ti);
+ }
+ endInsertRows();
+
+ emit(layoutChanged());
+ }
+ return ni;
+}
+
+FindResultItem *FindResultModel::addSubItem(FindResultItem *parent,
+ const QString &s, TreeItem *pi,
+ int i)
+{
+ FindResultItem *ni = NULL;
+ if (pi && parent) {
+ QModelIndex parix(index(parent));
+
+ emit(layoutAboutToBeChanged());
+
+ int n = rowCount(parix);
+ beginInsertRows(parix, n, n);
+
+ QModelIndex ix;
+ if (parent->insertChildren(n, 1, 0)) {
+ ix = index(n, 0, parix);
+ setData(ix, QVariant(s), Qt::EditRole);
+ ni = getItem(ix);
+ ni->setOriginal(pi);
+ ni->setOriginalIndex(i);
+ }
+ endInsertRows();
+ emit(layoutChanged());
+ }
+ return ni;
+}
+
+void FindResultModel::setSearchString(const QString &s) { searchString = s; }
+
+QString FindResultModel::getSearchString() { return searchString; }
+
+void FindResultModel::setSearchFlags(QTextDocument::FindFlags f)
+{
+ searchFlags = f;
+}
+
+QTextDocument::FindFlags FindResultModel::getSearchFlags()
+{
+ return searchFlags;
+}
+
+void FindResultModel::setShowParentsLevel(uint i)
+{
+ showParentsLevel = i;
+ settings.setValue("/findResults/showParentsLevel", showParentsLevel);
+}
+
+uint FindResultModel::getShowParentsLevel() { return showParentsLevel; }