Added sorting to tree items by clicking the header

Added names to header
Renamed buttons to 'Download' and 'Import'

ApplicationCode/WellPathImportSsihub/RimWellPathImport.cpp

@@ -17,6 +17,7 @@
 /////////////////////////////////////////////////////////////////////////////////
 
 #include "RimWellPathImport.h"
+#include "cafPdmUiTreeViewEditor.h"
 
 namespace caf {
 
@@ -186,3 +187,19 @@ void RimWellPathImport::fieldChangedByUi(const caf::PdmFieldHandle* changedField
     }
 }
 
+//--------------------------------------------------------------------------------------------------
+/// 
+//--------------------------------------------------------------------------------------------------
+void RimWellPathImport::defineObjectEditorAttribute(QString uiConfigName, caf::PdmUiEditorAttribute * attribute)
+{
+    caf::PdmUiTreeViewEditorAttribute* myAttr = dynamic_cast<caf::PdmUiTreeViewEditorAttribute*>(attribute);
+    if (myAttr)
+    {
+        QStringList colHeaders;
+        colHeaders << "Region";
+        myAttr->columnHeaders = colHeaders;
+    }
+}
+
+
+

ApplicationCode/WellPathImportSsihub/RimWellPathImport.h

@@ -57,9 +57,11 @@ public:
 
     virtual void initAfterRead();
     virtual void fieldChangedByUi( const caf::PdmFieldHandle* changedField, const QVariant& oldValue, const QVariant& newValue );
+    virtual void defineObjectEditorAttribute( QString uiConfigName, caf::PdmUiEditorAttribute * attribute );
 
-//private:
     void updateFieldVisibility();
 
+
+
 };
 

ApplicationCode/WellPathImportSsihub/RiuWellImportWizard.cpp

@@ -32,6 +32,7 @@
 #include "RimWellPathImport.h"
 
 #include "RifJsonEncodeDecode.h"
+#include "cafPdmUiTreeViewEditor.h"
 
 
 //--------------------------------------------------------------------------------------------------
@@ -809,6 +810,31 @@ void FieldSelectionPage::initializePage()
 
 
 
+//--------------------------------------------------------------------------------------------------
+/// Helper class used to define column headers
+//--------------------------------------------------------------------------------------------------
+class ObjectGroupWithHeaders : public caf::PdmObjectGroup
+{
+public:
+    ObjectGroupWithHeaders() : caf::PdmObjectGroup()
+    {
+
+    }
+
+    //--------------------------------------------------------------------------------------------------
+    /// 
+    //--------------------------------------------------------------------------------------------------
+    virtual void defineObjectEditorAttribute(QString uiConfigName, caf::PdmUiEditorAttribute * attribute)
+    {
+        caf::PdmUiTreeViewEditorAttribute* myAttr = dynamic_cast<caf::PdmUiTreeViewEditorAttribute*>(attribute);
+        if (myAttr)
+        {
+            QStringList colHeaders;
+            colHeaders << "Wells";
+            myAttr->columnHeaders = colHeaders;
+        }
+    }
+};
 
 
 
@@ -830,9 +856,7 @@ WellSelectionPage::WellSelectionPage(RimWellPathImport* wellPathImport, QWidget*
 
     m_wellPathImportObject = wellPathImport;
 
-    m_regionsWithVisibleWells = new caf::PdmObjectGroup;
+    m_regionsWithVisibleWells = new ObjectGroupWithHeaders;
-
-    //m_wellSelectionTreeView->setPdmObject(wellPathImport);
 }
 
 //--------------------------------------------------------------------------------------------------
@@ -844,6 +868,8 @@ void WellSelectionPage::initializePage()
     if (!wiz) return;
 
     wiz->downloadWells();
+
+    setButtonText(QWizard::NextButton, "Download");
 }
 
 //--------------------------------------------------------------------------------------------------
@@ -903,6 +929,8 @@ WellSummaryPage::WellSummaryPage(RimWellPathImport* wellPathImport, QWidget* par
     m_listView->hide();
 
     m_objectGroup = new caf::PdmObjectGroup;
+
+    setButtonText(QWizard::FinishButton, "Import");
 }
 
 //--------------------------------------------------------------------------------------------------
@@ -942,17 +970,19 @@ void WellSummaryPage::updateSummaryPage()
                     for (size_t wIdx = 0; wIdx < oilField->wells.size(); wIdx++)
                     {
                         RimWellPathEntry* wellPathEntry = oilField->wells[wIdx];
-
+                        if (wellPathEntry->selected)
-                        if (QFile::exists(oilField->wells[wIdx]->wellPathFilePath))
                         {
-                            wellPathCount++;
+                            if (QFile::exists(oilField->wells[wIdx]->wellPathFilePath))
-                        }
+                            {
-                        else
+                                wellPathCount++;
-                        {
+                            }
-                            errorString += QString("Failed to get file '%1' from well '%2'\n").arg(oilField->wells[wIdx]->wellPathFilePath).arg(oilField->wells[wIdx]->name);
+                            else
-                        }
+                            {
+                                errorString += QString("Failed to get file '%1' from well '%2'\n").arg(oilField->wells[wIdx]->wellPathFilePath).arg(oilField->wells[wIdx]->name);
+                            }
 
-                        m_objectGroup->objects.push_back(wellPathEntry);
+                            m_objectGroup->objects.push_back(wellPathEntry);
+                        }
                     }
                 }
             }
@@ -960,7 +990,7 @@ void WellSummaryPage::updateSummaryPage()
     }
 
 
-    m_textEdit->setText(QString("Downloaded successfully %1 well paths.\nPlease push 'Finish' button to import well paths into ResInsight.\n\n").arg(wellPathCount));
+    m_textEdit->setText(QString("Downloaded successfully %1 well paths.\nPlease push 'Import' button to import well paths into ResInsight.\n\n").arg(wellPathCount));
     if (!errorString.isEmpty())
     {
         m_textEdit->append("Detected following errors during well path download. See details below.");

ApplicationCode/WellPathImportSsihub/RiuWellImportWizard.h

@@ -72,6 +72,10 @@ public:
 };
 
 
+
+class ObjectGroupWithHeaders;
+
+
 //--------------------------------------------------------------------------------------------------
 /// 
 //--------------------------------------------------------------------------------------------------
@@ -87,7 +91,7 @@ public:
     void buildWellTreeView();
 
 private:
-    caf::PdmObjectGroup*  m_regionsWithVisibleWells;
+    ObjectGroupWithHeaders*  m_regionsWithVisibleWells;
     RimWellPathImport*  m_wellPathImportObject;
     caf::PdmUiTreeView* m_wellSelectionTreeView;
 

Fwk/AppFwk/cafUserInterface/cafPdmUiTreeViewEditor.cpp

@@ -45,6 +45,29 @@
 #include <QWidget>
 #include <QGridLayout>
 #include <QTreeView>
+#include <QSortFilterProxyModel>
+
+
+//--------------------------------------------------------------------------------------------------
+/// 
+//--------------------------------------------------------------------------------------------------
+class MySortFilterProxyModel : public QSortFilterProxyModel
+{
+public:
+    MySortFilterProxyModel(QObject *parent = 0)
+        : QSortFilterProxyModel(parent)
+    {
+
+    }
+
+    void notifyModelChanged()
+    {
+        QModelIndex startModelIdx = index(0,0);
+        QModelIndex endModelIdx = index(rowCount(startModelIdx), 0);
+
+        emit dataChanged(startModelIdx, endModelIdx);
+    }
+};
 
 
 
@@ -84,8 +107,14 @@ QWidget* PdmUiTreeViewEditor::createWidget(QWidget* parent)
 
     m_treeModelPdm = new caf::UiTreeModelPdm(m_mainWidget);
     m_treeView = new QTreeView(m_mainWidget);
-    m_treeView->setHeaderHidden(true);
+
-    m_treeView->setModel(m_treeModelPdm);
+    m_proxyTreeModelPdm = new MySortFilterProxyModel(m_mainWidget);
+    m_proxyTreeModelPdm->setSourceModel(m_treeModelPdm);
+    m_treeView->setModel(m_proxyTreeModelPdm);
+
+    m_treeView->setSortingEnabled(true);
+    m_treeView->sortByColumn(1, Qt::AscendingOrder);
+
     
     m_layout->addWidget(m_treeView);
 
@@ -125,12 +154,11 @@ void PdmUiTreeViewEditor::configureAndUpdateUi(const QString& uiConfigName)
         }
     }
 
-
+    m_treeModelPdm->setColumnHeaders(m_editorAttributes.columnHeaders);
-    PdmUiTreeViewEditorAttribute leab;
-
 
     // Notify all connected views that the complete model is updated
     m_treeModelPdm->notifyModelChanged();
+    m_proxyTreeModelPdm->notifyModelChanged();
 }
 
 //--------------------------------------------------------------------------------------------------

Fwk/AppFwk/cafUserInterface/cafPdmUiTreeViewEditor.h

@@ -43,6 +43,8 @@
 #include <QWidget>
 #include <QPointer>
 
+class MySortFilterProxyModel;
+
 class QGridLayout;
 class QVBoxLayout;
 class QTreeView;
@@ -66,6 +68,7 @@ public:
     }
 
 public:
+    QStringList columnHeaders;
 };
 
 
@@ -90,8 +93,9 @@ private:
     QPointer<QWidget>   m_mainWidget;
     QVBoxLayout*        m_layout;
 
-    QTreeView*          m_treeView;
+    QTreeView*              m_treeView;
-    UiTreeModelPdm*     m_treeModelPdm;
+    UiTreeModelPdm*         m_treeModelPdm;
+    MySortFilterProxyModel* m_proxyTreeModelPdm;
 
     // Forward update events to the tree view editor connected to Pdm root object using a proxy editor
     PdmUiProxyEditorHandle* m_proxyEditor;

Fwk/AppFwk/cafUserInterface/cafUiTreeModelPdm.cpp

@@ -498,6 +498,33 @@ void UiTreeModelPdm::notifyModelChanged()
     emit dataChanged(startModelIdx, endModelIdx);
 }
 
+//--------------------------------------------------------------------------------------------------
+/// 
+//--------------------------------------------------------------------------------------------------
+QVariant UiTreeModelPdm::headerData(int section, Qt::Orientation orientation, int role /*= Qt::DisplayRole */) const
+{
+    if (role != Qt::DisplayRole)
+        return QVariant();
+
+    if (section < m_columnHeaders.size())
+    {
+        return m_columnHeaders[section];
+    }
+
+    return QVariant();
+}
+
+//--------------------------------------------------------------------------------------------------
+/// 
+//--------------------------------------------------------------------------------------------------
+void UiTreeModelPdm::setColumnHeaders(const QStringList& columnHeaders)
+{
+    m_columnHeaders = columnHeaders;
+}
+
+
+
+
 //--------------------------------------------------------------------------------------------------
 /// 
 //--------------------------------------------------------------------------------------------------

Fwk/AppFwk/cafUserInterface/cafUiTreeModelPdm.h

@@ -40,6 +40,8 @@
 #include "cafUiTreeItem.h"
 
 #include <QAbstractItemModel>
+#include <QStringList>
+
 #include <assert.h>
 #include "cafPdmPointer.h"
 
@@ -71,6 +73,7 @@ public:
     void                    updateUiSubTree(PdmObject* root);
 
     void                    notifyModelChanged();
+    void                    setColumnHeaders(const QStringList& columnHeaders);
 
 public:
     // Overrides from QAbstractItemModel
@@ -79,10 +82,12 @@ public:
     virtual int             rowCount(const QModelIndex &parentIndex = QModelIndex( ) ) const;
     virtual int             columnCount(const QModelIndex &parentIndex = QModelIndex( ) ) const;
     virtual QVariant        data(const QModelIndex &index, int role = Qt::DisplayRole ) const;
+    virtual QVariant        headerData(int section, Qt::Orientation orientation, int role = Qt::DisplayRole ) const;
     
     virtual Qt::ItemFlags   flags(const QModelIndex &index) const;
     virtual bool            setData(const QModelIndex &index, const QVariant &value, int role = Qt::EditRole);
 
+
     virtual bool            removeRows_special(int position, int rows, const QModelIndex &parent = QModelIndex());
 
 protected:
@@ -91,6 +96,7 @@ private:
     void                    updateModelSubTree(const QModelIndex& uiSubTreeRootModelIdx, PdmUiTreeItem* uiModelSubTreeRoot, PdmUiTreeItem* updatedPdmSubTreeRoot);
 
     PdmUiTreeItem*          m_treeItemRoot;
+    QStringList             m_columnHeaders;
 };