Use similar stream + electrode names if no exact match

Author: medengineer

Plugins/BasicSpikeDisplay/SpikeDisplayNode/SpikeDisplayCanvas.cpp

@@ -112,6 +112,17 @@ SpikeDisplayCanvas::SpikeDisplayCanvas(SpikeDisplayNode* processor_) :
     cache = std::make_unique<SpikeDisplayCache>();
 }
 
+void SpikeDisplayCanvas::applyCachedDisplaySettings(int plotIdx, std::string key)
+{
+    spikeDisplay->setMonitorStateForPlot(plotIdx, cache->isMonitored(key));
+
+    for (int j = 0; j < processor->getNumberOfChannelsForElectrode(plotIdx); j++)
+    {
+        spikeDisplay->setThresholdForWaveAxis(plotIdx,j,cache->getThreshold(key,j));
+        spikeDisplay->setRangeForWaveAxis(plotIdx,j,cache->getRange(key, j));
+    }
+}
+
 void SpikeDisplayCanvas::update()
 {
 
@@ -132,17 +143,16 @@ void SpikeDisplayCanvas::update()
 
         std::string cacheKey = processor->getSpikeChannel(i)->getIdentifier().toStdString();
 
-        if (cache && cache->hasCachedDisplaySettings(cacheKey))
+        if (cache)
         {
-
-            spikeDisplay->setMonitorStateForPlot(i, cache->isMonitored(cacheKey));
-
-            for (int j = 0; j < processor->getNumberOfChannelsForElectrode(i); j++)
+            if (cache->hasCachedDisplaySettings(cacheKey))
             {
-                spikeDisplay->setThresholdForWaveAxis(i,j,cache->getThreshold(cacheKey,j));
-                spikeDisplay->setRangeForWaveAxis(i,j,cache->getRange(cacheKey, j));
+                applyCachedDisplaySettings(i, cacheKey);
+            }
+            else if (cache->findSimilarKey(cacheKey).size() > 0)
+            {
+                applyCachedDisplaySettings(i, cache->findSimilarKey(cacheKey));
             }
-
         }
 
     }

Plugins/BasicSpikeDisplay/SpikeDisplayNode/SpikeDisplayCanvas.h

@@ -81,6 +81,19 @@ class SpikeDisplayCache
         return thresholds.count(cacheKey) > 0;
     };
 
+    std::string findSimilarKey(std::string key)
+    {
+        std::vector<std::string> keys = extract_keys(ranges);
+        for (int i = 0; i < keys.size(); i++)
+        {
+            std::string partToMatch = key.substr(3, key.length() - 3);
+            std::string possibleMatch = keys[i].substr(3, keys[i].length() - 3);
+            if (partToMatch.compare(possibleMatch) == 0)
+                return keys[i];
+        }
+        return "";
+    }
+
 private:
 
     std::map<std::string, std::map<int, double>> ranges;
@@ -178,6 +191,9 @@ class SpikeDisplayCanvas : public Visualizer,
     /** Loads display parameters */
     void loadCustomParametersFromXml(XmlElement* xml);
 
+    /** Apply cached settings */
+    void applyCachedDisplaySettings(int plotIdx, std::string cacheKey);
+
     /** Pointer to the underlying SpikeDisplayNode*/
     SpikeDisplayNode* processor;