Multiseries support for selection pointer

Task-number: QTRD-2515

Change-Id: Id3611533c4dd333b68b96669c7230707506b68ab
Reviewed-by: Miikka Heikkinen <miikka.heikkinen@digia.com>

5 files changed, 148 insertions, 54 deletions

diff --git a/src/datavisualization/engine/surface3drenderer.cpp b/src/datavisualization/engine/surface3drenderer.cpp
index 492f4174..2a51094a 100644
--- a/src/datavisualization/engine/surface3drenderer.cpp
+++ b/src/datavisualization/engine/surface3drenderer.cpp
@@ -80,7 +80,6 @@ Surface3DRenderer::Surface3DRenderer(Surface3DController *controller)
       m_selectionResultTexture(0),
       m_shadowQualityToShader(33.3f),
       m_flatSupported(true),
-      m_selectionPointer(0),
       m_selectionActive(false),
       m_xFlipped(false),
       m_zFlipped(false),
@@ -137,8 +136,6 @@ Surface3DRenderer::~Surface3DRenderer()
     delete m_gridLineObj;
     delete m_labelObj;
 
-    delete m_selectionPointer;
-
     foreach (SurfaceSeriesRenderCache *cache, m_renderCacheList)
         delete cache;
     m_renderCacheList.clear();
@@ -268,6 +265,7 @@ void Surface3DRenderer::updateSeries(const QList<QAbstract3DSeries *> &seriesLis
         }
     }
 
+    // Remove non-valid objects from the cache list
     foreach (SurfaceSeriesRenderCache *cache, m_renderCacheList) {
         if (!cache->isValid()) {
             if (cache->series() == m_selectedSeries)
@@ -280,15 +278,23 @@ void Surface3DRenderer::updateSeries(const QList<QAbstract3DSeries *> &seriesLis
         }
     }
 
-    if (m_selectionPointer && m_selectedSeries) {
-        SurfaceSeriesRenderCache *cache =
-                m_renderCacheList.value(const_cast<QSurface3DSeries *>(m_selectedSeries));
-        if (cache) {
-            m_selectionPointer->setHighlightColor(
-                        Utils::vectorFromColor(m_selectedSeries->singleHighlightColor()));
-            // Make sure selection pointer object reference is still good
-            m_selectionPointer->setPointerObject(cache->object());
-            m_selectionPointer->setRotation(cache->meshRotation());
+    // Selection pointer issues
+    if (m_selectedSeries) {
+        foreach (SurfaceSeriesRenderCache *cache, m_renderCacheList) {
+            QVector3D highlightColor =
+                    Utils::vectorFromColor(cache->series()->singleHighlightColor());
+            SelectionPointer *slicePointer = cache->sliceSelectionPointer();
+            if (slicePointer) {
+                slicePointer->setHighlightColor(highlightColor);
+                slicePointer->setPointerObject(cache->object());
+                slicePointer->setRotation(cache->meshRotation());
+            }
+            SelectionPointer *mainPointer = cache->mainSelectionPointer();
+            if (mainPointer) {
+                mainPointer->setHighlightColor(highlightColor);
+                mainPointer->setPointerObject(cache->object());
+                mainPointer->setRotation(cache->meshRotation());
+            }
         }
     }
 }
@@ -419,7 +425,7 @@ void Surface3DRenderer::updateSliceDataModel(const QPoint &point)
 }
 
 QPoint Surface3DRenderer::mapCoordsToSampleSpace(SurfaceSeriesRenderCache *cache,
-                                               const QPointF &coords)
+                                                 const QPointF &coords)
 {
     QPoint point(-1, -1);
 
@@ -674,7 +680,7 @@ void Surface3DRenderer::updateScene(Q3DScene *scene)
 
     Abstract3DRenderer::updateScene(scene);
 
-    if (m_selectionPointer && m_selectionActive
+    if (m_selectionActive
             && m_cachedSelectionMode.testFlag(QAbstract3DGraph::SelectionItem)) {
         m_selectionDirty = true; // Ball may need repositioning if scene changes
     }
@@ -692,9 +698,16 @@ void Surface3DRenderer::render(GLuint defaultFboHandle)
         drawSlicedScene();
 
     // Render selection ball
-    if (m_selectionPointer && m_selectionActive
+    if (m_selectionActive
             && m_cachedSelectionMode.testFlag(QAbstract3DGraph::SelectionItem)) {
-        m_selectionPointer->render(defaultFboHandle);
+        foreach (SurfaceSeriesRenderCache *cache, m_renderCacheList) {
+            if (cache->slicePointerActive() && cache->renderable() &&
+                    m_cachedIsSlicingActivated ) {
+                cache->sliceSelectionPointer()->render(defaultFboHandle);
+            }
+            if (cache->mainPointerActive() && cache->renderable())
+                cache->mainSelectionPointer()->render(defaultFboHandle);
+        }
     }
 }
 
@@ -739,8 +752,7 @@ void Surface3DRenderer::drawSlicedScene()
         bool drawGrid = false;
 
         foreach (SurfaceSeriesRenderCache *cache, m_renderCacheList) {
-            if (cache->sliceSurfaceObject()->indexCount() && cache->isSeriesVisible()
-                    && (cache->surfaceVisible() || cache->surfaceGridVisible())) {
+            if (cache->sliceSurfaceObject()->indexCount() && cache->renderable()) {
                 if (cache->surfaceGridVisible()) {
                     glEnable(GL_POLYGON_OFFSET_FILL);
                     glPolygonOffset(0.5f, 1.0f);
@@ -1189,8 +1201,7 @@ void Surface3DRenderer::drawScene(GLuint defaultFboHandle)
         glDisable(GL_CULL_FACE);
 
         foreach (SurfaceSeriesRenderCache *cache, m_renderCacheList) {
-            if (cache->surfaceObject()->indexCount() && cache->isSeriesVisible()
-                    && (cache->surfaceVisible() || cache->surfaceGridVisible())) {
+            if (cache->surfaceObject()->indexCount() && cache->renderable()) {
                 QMatrix4x4 modelMatrix;
                 QMatrix4x4 MVPMatrix;
 
@@ -2111,48 +2122,101 @@ void Surface3DRenderer::loadGridLineMesh()
 
 void Surface3DRenderer::surfacePointSelected(const QPoint &point)
 {
+    foreach (SurfaceSeriesRenderCache *cache, m_renderCacheList) {
+        cache->setSlicePointerActivity(false);
+        cache->setMainPointerActivity(false);
+    }
+
+    if (m_cachedSelectionMode.testFlag(QAbstract3DGraph::SelectionMultiSeries)) {
+        // Find axis coordinates for the selected point
+        SurfaceSeriesRenderCache *selectedCache =
+            m_renderCacheList.value(const_cast<QSurface3DSeries *>(m_selectedSeries));
+        QSurfaceDataArray &dataArray = selectedCache->dataArray();
+        QSurfaceDataItem item = dataArray.at(point.x())->at(point.y());
+        QPointF coords(item.x(), item.z());
+
+        foreach (SurfaceSeriesRenderCache *cache, m_renderCacheList) {
+            if (cache->series() != m_selectedSeries) {
+                QPoint mappedPoint = mapCoordsToSampleSpace(cache, coords);
+                updateSelectionPoint(cache, mappedPoint, false);
+            } else {
+                updateSelectionPoint(cache, point, true);
+            }
+        }
+    } else {
+        if (m_selectedSeries) {
+            SurfaceSeriesRenderCache *cache =
+                    m_renderCacheList.value(const_cast<QSurface3DSeries *>(m_selectedSeries));
+            if (cache)
+                updateSelectionPoint(cache, point, true);
+        }
+    }
+}
+
+void Surface3DRenderer::updateSelectionPoint(SurfaceSeriesRenderCache *cache, const QPoint &point,
+                                             bool label)
+{
     int row = point.x();
     int column = point.y();
 
-    SurfaceSeriesRenderCache *cache =
-            m_renderCacheList.value(const_cast<QSurface3DSeries *>(m_selectedSeries));
+    if (column < 0 || row < 0)
+        return;
+
     QSurfaceDataArray &dataArray = cache->dataArray();
     float value = dataArray.at(row)->at(column).y();
 
-    if (!m_selectionPointer)
-        m_selectionPointer = new SelectionPointer(m_drawer);
+    SelectionPointer *slicePointer = cache->sliceSelectionPointer();
+    if (!slicePointer && m_cachedIsSlicingActivated) {
+        slicePointer = new SelectionPointer(m_drawer);
+        cache->setSliceSelectionPointer(slicePointer);
+    }
+    SelectionPointer *mainPointer = cache->mainSelectionPointer();
+    if (!mainPointer) {
+        mainPointer = new SelectionPointer(m_drawer);
+        cache->setMainSelectionPointer(mainPointer);
+    }
 
-    QVector3D pos;
     const QVector3D &scale = cache->scale();
     const QVector3D &offset = cache->offset();
+    QString selectionLabel;
+    if (label)
+        selectionLabel = createSelectionLabel(cache, value, column, row);
+
     if (m_cachedIsSlicingActivated) {
+        QVector3D subPos;
         if (m_cachedSelectionMode.testFlag(QAbstract3DGraph::SelectionRow)) {
-            pos = cache->sliceSurfaceObject()->vertexAt(column, 0);
-            pos *= QVector3D(scale.x(), 1.0f, 0.0f);
-            pos += QVector3D(offset.x(), 0.0f, 0.0f);
-            m_selectionPointer->updateBoundingRect(m_secondarySubViewport);
-            m_selectionPointer->updateSliceData(true, m_autoScaleAdjustment);
+            subPos = cache->sliceSurfaceObject()->vertexAt(column, 0);
+            subPos *= QVector3D(scale.x(), 1.0f, 0.0f);
+            subPos += QVector3D(offset.x(), 0.0f, 0.0f);
         } else if (m_cachedSelectionMode.testFlag(QAbstract3DGraph::SelectionColumn)) {
-            pos = cache->sliceSurfaceObject()->vertexAt(row, 0);
-            pos *= QVector3D(scale.z(), 1.0f, 0.0f);
-            pos += QVector3D(-offset.z(), 0.0f, 0.0f);
-            m_selectionPointer->updateBoundingRect(m_secondarySubViewport);
-            m_selectionPointer->updateSliceData(true, m_autoScaleAdjustment);
+            subPos = cache->sliceSurfaceObject()->vertexAt(row, 0);
+            subPos *= QVector3D(scale.z(), 1.0f, 0.0f);
+            subPos += QVector3D(-offset.z(), 0.0f, 0.0f);
         }
-    } else {
-        pos = cache->surfaceObject()->vertexAt(column, row);
-        pos *= scale;
-        pos += offset;
-        m_selectionPointer->updateBoundingRect(m_primarySubViewport);
-        m_selectionPointer->updateSliceData(false, m_autoScaleAdjustment);
-    }
-
-    m_selectionPointer->setPosition(pos);
-    m_selectionPointer->setLabel(createSelectionLabel(value, column, row));
-    m_selectionPointer->setPointerObject(cache->object());
-    m_selectionPointer->setHighlightColor(cache->singleHighlightColor());
-    m_selectionPointer->updateScene(m_cachedScene);
-    m_selectionPointer->setRotation(cache->meshRotation());
+        slicePointer->updateBoundingRect(m_secondarySubViewport);
+        slicePointer->updateSliceData(true, m_autoScaleAdjustment);
+        slicePointer->setPosition(subPos);
+        slicePointer->setLabel(selectionLabel);
+        slicePointer->setPointerObject(cache->object());
+        slicePointer->setHighlightColor(cache->singleHighlightColor());
+        slicePointer->updateScene(m_cachedScene);
+        slicePointer->setRotation(cache->meshRotation());
+        cache->setSlicePointerActivity(true);
+    }
+
+    QVector3D mainPos;
+    mainPos = cache->surfaceObject()->vertexAt(column, row);
+    mainPos *= scale;
+    mainPos += offset;
+    mainPointer->updateBoundingRect(m_primarySubViewport);
+    mainPointer->updateSliceData(false, m_autoScaleAdjustment);
+    mainPointer->setPosition(mainPos);
+    mainPointer->setLabel(selectionLabel);
+    mainPointer->setPointerObject(cache->object());
+    mainPointer->setHighlightColor(cache->singleHighlightColor());
+    mainPointer->updateScene(m_cachedScene);
+    mainPointer->setRotation(cache->meshRotation());
+    cache->setMainPointerActivity(true);
 }
 
 // Maps selection Id to surface point in data array
@@ -2179,10 +2243,9 @@ QPoint Surface3DRenderer::selectionIdToSurfacePoint(uint id)
     return QPoint(row, column);
 }
 
-QString Surface3DRenderer::createSelectionLabel(float value, int column, int row)
+QString Surface3DRenderer::createSelectionLabel(SurfaceSeriesRenderCache *cache, float value,
+                                                int column, int row)
 {
-    SurfaceSeriesRenderCache *cache =
-            m_renderCacheList.value(const_cast<QSurface3DSeries *>(m_selectedSeries));
     QSurfaceDataArray &dataArray = cache->dataArray();
     QString labelText = cache->itemLabelFormat();
     static const QString xTitleTag(QStringLiteral("@xTitle"));
@@ -2290,6 +2353,11 @@ void Surface3DRenderer::updateSlicingActive(bool isSlicing)
 #endif
 
     m_selectionDirty = true;
+
+    foreach (SurfaceSeriesRenderCache *cache, m_renderCacheList) {
+        if (cache->mainSelectionPointer())
+            cache->mainSelectionPointer()->updateBoundingRect(m_primarySubViewport);
+    }
 }
 
 void Surface3DRenderer::loadLabelMesh()
diff --git a/src/datavisualization/engine/surface3drenderer_p.h b/src/datavisualization/engine/surface3drenderer_p.h
index 1d57e85d..2614ecba 100644
--- a/src/datavisualization/engine/surface3drenderer_p.h
+++ b/src/datavisualization/engine/surface3drenderer_p.h
@@ -82,7 +82,6 @@ private:
     GLuint m_selectionResultTexture;
     GLfloat m_shadowQualityToShader;
     bool m_flatSupported;
-    SelectionPointer *m_selectionPointer;
     bool m_selectionActive;
     bool m_xFlipped;
     bool m_zFlipped;
@@ -151,8 +150,9 @@ private:
     void idToRGBA(uint id, uchar *r, uchar *g, uchar *b, uchar *a);
     void fillIdCorner(uchar *p, uchar r, uchar g, uchar b, uchar a, int stride);
     void surfacePointSelected(const QPoint &point);
+    void updateSelectionPoint(SurfaceSeriesRenderCache *cache, const QPoint &point, bool label);
     QPoint selectionIdToSurfacePoint(uint id);
-    QString createSelectionLabel(float value, int column, int row);
+    QString createSelectionLabel(SurfaceSeriesRenderCache *cache, float value, int column, int row);
 #if !defined(QT_OPENGL_ES_2)
     void updateDepthBuffer();
 #endif
diff --git a/src/datavisualization/engine/surfaceseriesrendercache.cpp b/src/datavisualization/engine/surfaceseriesrendercache.cpp
index 59b50358..bc261f05 100644
--- a/src/datavisualization/engine/surfaceseriesrendercache.cpp
+++ b/src/datavisualization/engine/surfaceseriesrendercache.cpp
@@ -39,6 +39,10 @@ SurfaceSeriesRenderCache::SurfaceSeriesRenderCache()
       m_flatStatusDirty(false),
       m_scale(QVector3D(1.0f, 1.0f, 1.0f)),
       m_offset(QVector3D(0.0f, 0.0f, 0.0f)),
+      m_sliceSelectionPointer(0),
+      m_mainSelectionPointer(0),
+      m_slicePointerActive(false),
+      m_mainPointerActive(false),
       m_valid(false),
       m_objectDirty(true)
 {
@@ -55,6 +59,9 @@ SurfaceSeriesRenderCache::~SurfaceSeriesRenderCache()
     for (int i = 0; i < m_sliceDataArray.size(); i++)
         delete m_sliceDataArray.at(i);
     m_sliceDataArray.clear();
+
+    delete m_sliceSelectionPointer;
+    delete m_mainSelectionPointer;
 }
 
 void SurfaceSeriesRenderCache::populate(QSurface3DSeries *series, Abstract3DRenderer *renderer)
diff --git a/src/datavisualization/engine/surfaceseriesrendercache_p.h b/src/datavisualization/engine/surfaceseriesrendercache_p.h
index c0a3cec3..e44691fd 100644
--- a/src/datavisualization/engine/surfaceseriesrendercache_p.h
+++ b/src/datavisualization/engine/surfaceseriesrendercache_p.h
@@ -34,6 +34,7 @@
 #include "qabstract3dseries_p.h"
 #include "qsurface3dseries_p.h"
 #include "surfaceobject_p.h"
+#include "selectionpointer_p.h"
 
 #include <QtGui/QMatrix4x4>
 
@@ -67,6 +68,8 @@ public:
     inline QSurfaceDataArray &dataArray() { return m_dataArray; }
     inline QSurfaceDataArray &sliceDataArray() { return m_sliceDataArray; }
     inline bool isSeriesVisible() const { return m_series->isVisible(); }
+    inline bool renderable() const { return m_series->isVisible() && (m_surfaceVisible ||
+                                                                      m_surfaceGridVisible); }
     inline void setObjectDirty(bool state) { m_objectDirty = state; }
     inline bool objectDirty() const { return m_objectDirty; }
     inline void setSelectionTexture(GLuint texture) { m_selectionTexture = texture; }
@@ -86,6 +89,16 @@ public:
     inline void setMVPMatrix(const QMatrix4x4 &matrix) { m_MVPMatrix = matrix; }
     inline const QMatrix4x4 &MVPMatrix() { return m_MVPMatrix; }
 
+    inline void setSliceSelectionPointer(SelectionPointer *pointer) { m_sliceSelectionPointer = pointer; }
+    inline SelectionPointer *sliceSelectionPointer() const { return m_sliceSelectionPointer; }
+    inline void setMainSelectionPointer(SelectionPointer *pointer) { m_mainSelectionPointer = pointer; }
+    inline SelectionPointer *mainSelectionPointer() const { return m_mainSelectionPointer; }
+
+    inline void setSlicePointerActivity(bool activity) { m_slicePointerActive = activity; }
+    inline bool slicePointerActive() const { return m_slicePointerActive; }
+    inline void setMainPointerActivity(bool activity) { m_mainPointerActive = activity; }
+    inline bool mainPointerActive() const { return m_mainPointerActive; }
+
 protected:
     bool m_surfaceVisible;
     bool m_surfaceGridVisible;
@@ -103,6 +116,10 @@ protected:
     QVector3D m_scale;
     QVector3D m_offset;
     QMatrix4x4 m_MVPMatrix;
+    SelectionPointer *m_sliceSelectionPointer;
+    SelectionPointer *m_mainSelectionPointer;
+    bool m_slicePointerActive;
+    bool m_mainPointerActive;
 
     bool m_valid;
     bool m_objectDirty;
diff --git a/tests/surfacetest/main.cpp b/tests/surfacetest/main.cpp
index f2b187e6..5ebd1a33 100644
--- a/tests/surfacetest/main.cpp
+++ b/tests/surfacetest/main.cpp
@@ -255,6 +255,8 @@ int main(int argc, char *argv[])
                                int(QAbstract3DGraph::SelectionNone));
     selectionMode->addItem(QStringLiteral("Item"),
                                int(QAbstract3DGraph::SelectionItem));
+    selectionMode->addItem(QStringLiteral("Multi: Item"),
+                               int(QAbstract3DGraph::SelectionItem | QAbstract3DGraph::SelectionMultiSeries));
     selectionMode->addItem(QStringLiteral("Row"),
                                int(QAbstract3DGraph::SelectionRow));
     selectionMode->addItem(QStringLiteral("Item and Row"),