Fix loading colors from cluster in minimized scatterplot in project

src/ColoringAction.cpp

@@ -127,24 +127,29 @@ ColoringAction::ColoringAction(QObject* parent, const QString& title) :
     connect(&_scatterplotPlugin->getPositionDataset(), &Dataset<Points>::childAdded, this, &ColoringAction::updateColorByActionOptions);
     connect(&_scatterplotPlugin->getPositionDataset(), &Dataset<Points>::childRemoved, this, &ColoringAction::updateColorByActionOptions);
 
-    connect(&_scatterplotPlugin->getScatterplotWidget(), &ScatterplotWidget::renderModeChanged, this, &ColoringAction::updateScatterPlotWidgetColors);
-    connect(&_scatterplotPlugin->getScatterplotWidget(), &ScatterplotWidget::coloringModeChanged, this, &ColoringAction::updateScatterPlotWidgetColors);
+    connect(&_scatterplotPlugin->getScatterplotWidget(), &ScatterplotWidget::coloringModeChanged, this, [this](const ScatterplotWidget::ColoringMode& coloringMode) {
+        updateScatterPlotWidgetColors();
+        updateColorMapActionsReadOnly();
+        });
+
+    connect(&_scatterplotPlugin->getScatterplotWidget(), &ScatterplotWidget::renderModeChanged, this, [this](const ScatterplotWidget::RenderMode& renderMode) {
+        updateScatterPlotWidgetColors();
+        updateColorMapActionsReadOnly();
+        });
 
-    connect(&_dimensionAction, &DimensionPickerAction::currentDimensionIndexChanged, this, &ColoringAction::updateScatterPlotWidgetColors);
-    connect(&_dimensionAction, &DimensionPickerAction::currentDimensionIndexChanged, this, &ColoringAction::updateColorMapActionScalarRange);
+    connect(&_dimensionAction, &DimensionPickerAction::currentDimensionIndexChanged, this, [this](const int32_t& currentDimensionIndex) {
+        updateScatterPlotWidgetColors();
+        updateColorMapActionsReadOnly();
+        updateColorMapActionScalarRange();
+        });
 
     connect(&_constantColorAction, &ColorAction::colorChanged, this, &ColoringAction::updateScatterplotWidgetColorMap);
     connect(&_colorMap1DAction, &ColorMapAction::imageChanged, this, &ColoringAction::updateScatterplotWidgetColorMap);
     connect(&_colorMap2DAction, &ColorMapAction::imageChanged, this, &ColoringAction::updateScatterplotWidgetColorMap);
-    connect(&_scatterplotPlugin->getScatterplotWidget(), &ScatterplotWidget::coloringModeChanged, this, &ColoringAction::updateScatterplotWidgetColorMap);
-    connect(&_scatterplotPlugin->getScatterplotWidget(), &ScatterplotWidget::renderModeChanged, this, &ColoringAction::updateScatterplotWidgetColorMap);
 
     connect(&_colorMap1DAction.getRangeAction(ColorMapAction::Axis::X), &DecimalRangeAction::rangeChanged, this, &ColoringAction::updateScatterPlotWidgetColorMapRange);
     connect(&_colorMap2DAction.getRangeAction(ColorMapAction::Axis::X), &DecimalRangeAction::rangeChanged, this, &ColoringAction::updateScatterPlotWidgetColorMapRange);
 
-    connect(&_scatterplotPlugin->getScatterplotWidget(), &ScatterplotWidget::coloringModeChanged, this, &ColoringAction::updateColorMapActionsReadOnly);
-    connect(&_scatterplotPlugin->getScatterplotWidget(), &ScatterplotWidget::renderModeChanged, this, &ColoringAction::updateColorMapActionsReadOnly);
-
     const auto updateReadOnly = [this]() {
         setEnabled(_scatterplotPlugin->getPositionDataset().isValid() && _scatterplotPlugin->getScatterplotWidget().getRenderMode() == ScatterplotWidget::SCATTERPLOT);
     };
@@ -276,7 +281,8 @@ void ColoringAction::updateScatterplotWidgetColorMap()
     {
         case ScatterplotWidget::SCATTERPLOT:
         {
-            if (_colorByAction.getCurrentIndex() == 0) {
+            const int32_t currentIndex = _colorByAction.getCurrentIndex();
+            if (currentIndex == 0) {
                 QPixmap colorPixmap(1, 1);
 
                 colorPixmap.fill(_constantColorAction.getColor());
@@ -285,7 +291,7 @@ void ColoringAction::updateScatterplotWidgetColorMap()
                 scatterplotWidget.setScalarEffect(PointEffect::Color);
                 scatterplotWidget.setColoringMode(ScatterplotWidget::ColoringMode::Constant);
             }
-            else if (_colorByAction.getCurrentIndex() == 1) {
+            else if (currentIndex == 1) {
                 scatterplotWidget.setColorMap(_colorMap2DAction.getColorMapImage());
                 scatterplotWidget.setScalarEffect(PointEffect::Color2D);
                 scatterplotWidget.setColoringMode(ScatterplotWidget::ColoringMode::Scatter);

src/ScatterplotPlugin.cpp

@@ -797,25 +797,24 @@ void ScatterplotPlugin::loadColors(const Dataset<Clusters>& clusters)
     if (!clusters.isValid() || !_positionDataset.isValid())
         return;
 
-    // Mapping from local to global indices
-    std::vector<std::uint32_t> globalIndices;
-
     // Get global indices from the position dataset
     int totalNumPoints = 0;
     if (_positionDataset->isDerivedData())
         totalNumPoints = _positionSourceDataset->getFullDataset<Points>()->getNumPoints();
     else
         totalNumPoints = _positionDataset->getFullDataset<Points>()->getNumPoints();
 
+    // Mapping from local to global indices
+    std::vector<std::uint32_t> globalIndices;
     _positionDataset->getGlobalIndices(globalIndices);
 
     // Generate color buffer for global and local colors
     std::vector<Vector3f> globalColors(totalNumPoints);
-    std::vector<Vector3f> localColors(_positions.size());
+    std::vector<Vector3f> localColors(_numPoints);
 
     const auto& clusterVec = clusters->getClusters();
 
-    if (totalNumPoints == _positions.size() && clusterVec.size() == totalNumPoints)
+    if (totalNumPoints == _numPoints && clusterVec.size() == totalNumPoints)
     {
         for (size_t i = 0; i < static_cast<size_t>(clusterVec.size()); i++)
         {
@@ -826,14 +825,15 @@ void ScatterplotPlugin::loadColors(const Dataset<Clusters>& clusters)
         }
 
     }
-    else
+    else if(globalIndices.size() == _numPoints)
     {
         // Loop over all clusters and populate global colors
         for (const auto& cluster : clusterVec)
         {
-            const auto color = cluster.getColor();
+            const auto color  = cluster.getColor();
+            const auto colVec = Vector3f(color.redF(), color.greenF(), color.blueF());
             for (const auto& index : cluster.getIndices())
-                globalColors[index] = Vector3f(color.redF(), color.greenF(), color.blueF());
+                globalColors[index] = colVec;
 
         }
 

src/ScatterplotWidget.cpp

@@ -41,6 +41,7 @@ ScatterplotWidget::ScatterplotWidget(mv::plugin::ViewPlugin* parentPlugin) :
     _pointRenderer(this),
     _isInitialized(false),
     _renderMode(SCATTERPLOT),
+    _scalarEffect(PointEffect::Color),
     _backgroundColor(255, 255, 255, 255),
     _coloringMode(ColoringMode::Constant),
     _dataRectangleAction(this, "Data rectangle"),
@@ -334,7 +335,7 @@ void ScatterplotWidget::setScalars(const std::vector<float>& scalars)
 void ScatterplotWidget::setColors(const std::vector<Vector3f>& colors)
 {
     _pointRenderer.setColors(colors);
-    _pointRenderer.setScalarEffect(None);
+    setScalarEffect(PointEffect::None);
 
     update();
 }
@@ -367,6 +368,7 @@ void ScatterplotWidget::setPointScaling(mv::gui::PointScaling scalingMode)
 void ScatterplotWidget::setScalarEffect(PointEffect effect)
 {
     _pointRenderer.setScalarEffect(effect);
+    _scalarEffect = effect;
 
     update();
 }
@@ -621,7 +623,7 @@ void ScatterplotWidget::initializeGL()
     _densityRenderer.init();
 
     // Set a default color map for both renderers
-    _pointRenderer.setScalarEffect(PointEffect::Color);
+    _pointRenderer.setScalarEffect(_scalarEffect);
 
     _pointRenderer.setPointScaling(Absolute);
     _pointRenderer.setSelectionOutlineColor(Vector3f(1, 0, 0));

src/ScatterplotWidget.h

@@ -296,6 +296,7 @@ private slots:
 private:
     bool                        _isInitialized;                 /** Boolean determining whether the widget it properly initialized or not */
     RenderMode                  _renderMode;                    /** Current render mode */
+    PointEffect                 _scalarEffect;                  /** Current scalar effect */
     QColor                      _backgroundColor;               /** Background color */
     ColoringMode                _coloringMode;                  /** Type of point/density coloring */
     DecimalRectangleAction      _dataRectangleAction;           /** Rectangle action for the bounds of the loaded data */