Merge branch 'release/core_bican_bg/bican_bg' into master

Author: ThomasKroes

src/ScatterplotPlugin.cpp

@@ -13,6 +13,7 @@
 #include <ClusterData/ClusterData.h>
 #include <ColorData/ColorData.h>
 #include <PointData/PointData.h>
+#include <SelectionGroup.h>
 
 #include <graphics/Vector3f.h>
 
@@ -64,6 +65,17 @@ Q_PLUGIN_METADATA(IID "studio.manivault.ScatterplotPlugin")
 using namespace mv;
 using namespace mv::util;
 
+namespace
+{
+    Dataset<DatasetImpl> getTopDataset(const Dataset<DatasetImpl> dataset)
+    {
+        Dataset<DatasetImpl> topDataset = dataset;
+        while (topDataset->getParent().isValid())
+            topDataset = topDataset->getParent();
+        return topDataset;
+    }
+}
+
 ScatterplotPlugin::ScatterplotPlugin(const PluginFactory* factory) :
     ViewPlugin(factory),
     _dropWidget(nullptr),
@@ -250,7 +262,6 @@ ScatterplotPlugin::ScatterplotPlugin(const PluginFactory* factory) :
 
             // Only allow user to color by clusters when there is a positions dataset loaded
             if (_positionDataset.isValid()) {
-
                 if (_settingsAction->getColoringAction().hasColorDataset(candidateDataset)) {
 
                     // The clusters dataset is already loaded
@@ -935,6 +946,44 @@ void ScatterplotPlugin::loadColors(const Dataset<Clusters>& clusters)
     if (!clusters.isValid() || !_positionDataset.isValid())
         return;
 
+    // First check if cross-dataset metadata coloring is possible
+    std::vector<KeyBasedSelectionGroup>& selectionGroups = events().getSelectionGroups();
+    for (int i = 0; i < selectionGroups.size(); i++)
+    {
+        KeyBasedSelectionGroup& selectionGroup = selectionGroups[i];
+        if (selectionGroup.areDatasetsPartOfGroup(getTopDataset(_positionDataset), getTopDataset(clusters)))
+        {
+            const auto& clusterVec = clusters->getClusters();
+            std::vector<Vector3f> localColors(_numPoints, Vector3f(1, 0, 1));
+
+            // If the clusters belong to the same dataset as we're dropping them on, then skip to other path
+            if (clusters->getParent() == _positionDataset)
+                continue;
+
+            std::vector<int> mappedIndices = selectionGroup.getMappingBetweenDatasets(getTopDataset(clusters), _positionDataset);
+
+            for (int j = 0; j < mappedIndices.size(); j++)
+            {
+                int localIndex = mappedIndices[j];
+                if (localIndex == -1) continue;
+                for (const auto& cluster : clusterVec)
+                {
+                    if (std::find(cluster.getIndices().begin(), cluster.getIndices().end(), j) != cluster.getIndices().end())
+                    {
+                        const auto color = cluster.getColor();
+                        localColors[localIndex] = Vector3f(color.redF(), color.greenF(), color.blueF());
+                    }
+                }
+            }
+            // Apply colors to scatter plot widget without modification
+            _scatterPlotWidget->setColors(localColors);
+
+            // Render
+            getWidget().update();
+            return;
+        }
+    }
+
     // Get global indices from the position dataset
     const std::uint64_t totalNumPoints = numTotalPoints();