Temp commit

Author: JulianThijssen

src/ScatterplotPlugin.cpp

@@ -12,6 +12,7 @@
 #include <ClusterData/ClusterData.h>
 #include <ColorData/ColorData.h>
 #include <PointData/PointData.h>
+#include <SelectionGroup.h>
 
 #include <graphics/Vector3f.h>
 
@@ -231,7 +232,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
@@ -242,6 +242,22 @@ ScatterplotPlugin::ScatterplotPlugin(const PluginFactory* factory) :
                 else {
                     if (candidateDataset.isValid())
                     {
+                        bool foundGroup = false;
+                        // 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(_positionDataset->getParent(), candidateDataset->getParent()))
+                            {
+                                // Use the clusters set for points color
+                                dropRegions << new DropWidget::DropRegion(this, "Color", description, "palette", true, [this, candidateDataset]() {
+                                    _settingsAction->getColoringAction().addColorDataset(candidateDataset);
+                                    _settingsAction->getColoringAction().setCurrentColorDataset(candidateDataset);
+                                 });
+                                foundGroup = true;
+                            }
+                        }
                         // Check to set whether the number of data points comprised throughout all clusters is the same number
                         // as the number of data points in the dataset we are trying to color
                         int totalNumIndices = 0;
@@ -255,8 +271,12 @@ ScatterplotPlugin::ScatterplotPlugin(const PluginFactory* factory) :
                             totalNumPoints = _positionSourceDataset->getFullDataset<Points>()->getNumPoints();
                         else
                             totalNumPoints = _positionDataset->getFullDataset<Points>()->getNumPoints();
+                        qDebug() << "Found group" << foundGroup << candidateDataset->getGuiName() << candidateDataset->getParent()->getGuiName();
+                        if (foundGroup)
+                        {
 
-                        if (totalNumIndices == totalNumPoints)
+                        }
+                        else if (totalNumIndices == totalNumPoints)
                         {
                             // Use the clusters set for points color
                             dropRegions << new DropWidget::DropRegion(this, "Color", description, "palette", true, [this, candidateDataset]() {
@@ -845,6 +865,40 @@ 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(_positionDataset->getParent(), clusters->getParent()))
+        {
+            const auto& clusterVec = clusters->getClusters();
+            std::vector<Vector3f> localColors(_numPoints, Vector3f(1, 0, 1));
+
+            std::vector<int> mappedIndices = selectionGroup.getMappingBetweenDatasets(clusters->getParent(), _positionDataset->getParent());
+
+            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
     int totalNumPoints = 0;
     if (_positionDataset->isDerivedData())