keep track of channel initializations in an ordered set

Nuand · Nov 29, 2023 · 27de289 · 27de289
1 parent 810afe3
commit 27de289
Showing 1 changed file with 15 additions and 6 deletions.
diff --git a/lib/channel_store.h b/lib/channel_store.h
@@ -1,21 +1,30 @@
 #include <map>
+#include <set>
 
 template<typename T>
 class channel_store
 {
   std::map<size_t, T> values_;
+  std::set<size_t> initialized_channels_;
+
 public:
   template <typename pred>
   T set_if_not_equal(T value, size_t n, size_t max_n, pred fn)
   {
       if( n < max_n)
       {
-          auto &val = values_[n];
-          if(val != value)
-          {
-              val = fn();
-          }
-          return val;
+         if( initialized_channels_.find(n) == initialized_channels_.end())
+         {
+             values_[n] = fn();
+             initialized_channels_.insert(n);
+         } else {
+             auto &val = values_[n];
+             if( val != value)
+             {
+                 val = fn();
+             }
+         }
+         return values_[n];
       }
       return T{};
   }