WIP: map means to HSNE embedding

Author: alxvth

src/MappingUtils.cpp

@@ -68,6 +68,17 @@ std::pair<const mv::LinkedData*, unsigned int> getSelectionMappingPositionsToCol
 
     return { mapping, numTargetPoints };
 }
+
+std::pair<const mv::LinkedData*, unsigned int> getSelectionMappingPositionSourceToColors(const mv::Dataset<Points>& positions, const mv::Dataset<Points>& colors) {
+    if (!positions->isDerivedData())
+        return { nullptr, 0 };
+
+    const auto fullSourceData = positions->getSourceDataset<Points>()->getFullDataset<Points>();
+
+    if(!fullSourceData.isValid())
+        return { nullptr, 0 };
+
+    return getSelectionMappingPositionsToColors(fullSourceData, colors);
 }
 
 bool checkSurjectiveMapping(const mv::LinkedData& linkedData, const std::uint32_t numPointsInTarget) {
@@ -93,16 +104,17 @@ bool checkSurjectiveMapping(const mv::LinkedData& linkedData, const std::uint32_
 
 bool checkSelectionMapping(const mv::Dataset<Points>& colors, const mv::Dataset<Points>& positions) {
 
-    // Check if there is a mapping
-    auto mapping = getSelectionMappingColorsToPositions(colors, positions);
-    auto numTargetPoints = positions->getNumPoints();
+    printLinkedDataNames(colors);
+    printLinkedDataNames(positions);
 
     // Check if there is a mapping
     auto [mapping, numTargetPoints] = getSelectionMappingColorsToPositions(colors, positions);
 
     if (!mapping)
         std::tie(mapping, numTargetPoints) = getSelectionMappingPositionsToColors(positions, colors);
 
+    if (!mapping)
+        std::tie(mapping, numTargetPoints) = getSelectionMappingPositionSourceToColors(positions, colors);
 
     if (!mapping)
         return false;

src/MappingUtils.h

@@ -39,6 +39,8 @@ std::pair<const mv::LinkedData*, unsigned int> getSelectionMappingColorsToPositi
 
 std::pair<const mv::LinkedData*, unsigned int> getSelectionMappingPositionsToColors(const mv::Dataset<Points>& positions, const mv::Dataset<Points>& colors);
 
+std::pair<const mv::LinkedData*, unsigned int> getSelectionMappingPositionSourceToColors(const mv::Dataset<Points>& positions, const mv::Dataset<Points>& colors);
+
 // Check if the mapping is surjective, i.e. hits all elements in the target
 bool checkSurjectiveMapping(const mv::LinkedData& linkedData, const std::uint32_t numPointsInTarget);
 

src/ScatterplotPlugin.cpp

@@ -723,6 +723,33 @@ void ScatterplotPlugin::loadColors(const Dataset<Points>& pointsColor, const std
                     }
                 }
 
+            }
+            else if ( // mapping from position data set to color data set 
+                const auto [selectionMapping, numPointsTarget] = getSelectionMappingPositionSourceToColors(_positionDataset, pointsColor);
+                /* check if valid */ selectionMapping != nullptr // && numPointsTarget == _numPoints
+                )
+            {   // THIS DOES NOT WORK YET
+
+                // selectionMapping is from full source data (scVI) to pointsColor (means)
+                // we need to use both the global indices and linked data mapping
+
+                const mv::SelectionMap::Map& linkedMap = selectionMapping->getMapping().getMap();
+
+                std::vector<std::uint32_t> globalIndices;
+                _positionDataset->getGlobalIndices(globalIndices);
+
+                for (std::int32_t localColorIndex = 0; localColorIndex < globalIndices.size(); localColorIndex++) {
+
+                    const auto& mappedIndices = linkedMap.at(globalIndices[localColorIndex]);   // from full source (parent) to means
+
+                    for (const auto& mappedIndex : mappedIndices) {
+                        if (mappedScalars[mappedIndex] != std::numeric_limits<float>::lowest())
+                            continue;
+
+                        mappedScalars[localColorIndex] = scalars[mappedIndex];
+                    }
+                }
+
             }
             else {
                 qWarning("Number of points used for coloring does not match number of points in data, aborting attempt to color plot");