Amend f441590c

[lyx.git] / src / frontends / qt4 / TocModel.h

diff --git a/src/frontends/qt4/TocModel.h b/src/frontends/qt4/TocModel.h
index b41e1006b9ec1ad0dd8ceee2a271783568a73170..4d4d442719bb86d8fef5c7e7141e5fbd48cdfc67 100644 (file)
--- a/src/frontends/qt4/TocModel.h
+++ b/src/frontends/qt4/TocModel.h
@@ -12,12 +12,10 @@
 #ifndef TOCMODEL_H
 #define TOCMODEL_H
 
-#include "qt_helpers.h"
+#include "support/shared_ptr.h"
 
 #include <QHash>
-#include <QStandardItemModel>
-
-class QSortFilterProxyModel;
+#include <QSortFilterProxyModel>
 
 namespace lyx {
 
@@ -36,11 +34,14 @@ class TocTypeModel;
 /// There is one TocModel per list in the TocBackend.
 class TocModel
 {
+       /// noncopyable
+       TocModel(TocModel const &);
+       void operator=(TocModel const &);
 public:
        ///
        TocModel(QObject * parent);
        ///
-       void reset(Toc const & toc);
+       void reset(shared_ptr<Toc const>);
        ///
        void reset();
        ///
@@ -72,7 +73,7 @@ private:
        ///
        bool is_sorted_;
        ///
-       Toc const * toc_;
+       shared_ptr<Toc const> toc_;
        ///
        int maxdepth_;
        ///
@@ -80,6 +81,30 @@ private:
 };
 
 
+/// A filter to sort the models alphabetically but with
+/// the table of contents on top.
+class TocModelSortProxyModel : public QSortFilterProxyModel
+{
+public:
+       TocModelSortProxyModel(QObject * w) 
+               : QSortFilterProxyModel(w)
+       {}
+
+       bool lessThan (const QModelIndex & left, const QModelIndex & right) const
+       {
+               if (left.model()->data(left, Qt::UserRole).toString()
+                         == QString("tableofcontents"))
+                       return true;
+               else if (right.model()->data(right, Qt::UserRole).toString()
+                         == QString("tableofcontents"))
+                       return false;
+               else
+                       return QSortFilterProxyModel::lessThan(left, right);
+       }
+};
+
+
+
 /// A container for the different TocModels.
 class TocModels : public QObject
 {
@@ -102,8 +127,6 @@ public:
        ///
        void init(Buffer const & buffer);
        ///
-       void updateBackend() const;
-       ///
        void updateItem(QString const & type, DocIterator const & dit);
        ///
        void sort(QString const & type, bool sort_it);
@@ -129,7 +152,7 @@ private:
        ///
        TocTypeModel * names_;
        ///
-       QSortFilterProxyModel * names_sorted_;
+       TocModelSortProxyModel * names_sorted_;
 };
 
 } // namespace frontend