Function to replace data in lsPointData

ViennaTools · Dec 20, 2024 · 9231eb7 · 9231eb7
1 parent 2fffc10
commit 9231eb7
Showing 1 changed file with 40 additions and 0 deletions.
diff --git a/include/viennals/lsPointData.hpp b/include/viennals/lsPointData.hpp
@@ -81,6 +81,46 @@ class PointData {
     vectorDataLabels.push_back(label);
   }
 
+  /// insert or replace scalar data array
+  void insertReplaceScalarData(const ScalarDataType &scalars,
+                               std::string label = "Scalars") {
+    if (int i = getScalarDataIndex(label); i != -1) {
+      scalarData[i] = scalars;
+    } else {
+      insertNextScalarData(scalars, label);
+    }
+  }
+
+  /// insert or replace scalar data array
+  void insertReplaceScalarData(ScalarDataType &&scalars,
+                               std::string label = "Scalars") {
+    if (int i = getScalarDataIndex(label); i != -1) {
+      scalarData[i] = std::move(scalars);
+    } else {
+      insertNextScalarData(std::move(scalars), label);
+    }
+  }
+
+  /// insert or replace vector data array
+  void insertNextVectorData(const VectorDataType &vectors,
+                            std::string label = "Vectors") {
+    if (int i = getVectorDataIndex(label); i != -1) {
+      vectorData[i] = vectors;
+    } else {
+      insertNextVectorData(vectors, label);
+    }
+  }
+
+  /// insert new vector data array
+  void insertNextVectorData(VectorDataType &&vectors,
+                            std::string label = "Vectors") {
+    if (int i = getVectorDataIndex(label); i != -1) {
+      vectorData[i] = std::move(vectors);
+    } else {
+      insertNextVectorData(std::move(vectors), label);
+    }
+  }
+
   /// get the number of different scalar data arrays saved
   unsigned getScalarDataSize() const { return scalarData.size(); }