equi change

najaeda · Oct 18, 2024 · 033c73b · 033c73b
1 parent e528952
commit 033c73b
Showing 1 changed file with 33 additions and 0 deletions.
diff --git a/src/snl/snl/kernel/SNLEquipotential.cpp b/src/snl/snl/kernel/SNLEquipotential.cpp
@@ -12,6 +12,7 @@
 #include "SNLBitTerm.h"
 #include "SNLInstTerm.h"
 #include "SNLInstance.h"
+#include <sstream>  
 
 namespace {
 
@@ -136,4 +137,36 @@ SNLEquipotential::SNLEquipotential(const SNLNetComponentOccurrence& netComponent
   extractor.extractNetFromNetComponentOccurrence(netComponentOccurrence, true);
 }
 
+std::string SNLEquipotential::getString() const {
+  std::ostringstream stream;
+  stream << "InstTermOccurrences: [";
+  bool first = true;
+  for (const auto& instTermOccurrence: instTermOccurrences_) {
+    if (not first) {
+      stream << ", ";
+    }
+    first = false;
+    stream << instTermOccurrence.getString();
+  }
+  stream << "], Terms: [";
+  first = true;
+  for (const auto& term: terms_) {
+    if (not first) {
+      stream << ", ";
+    }
+    first = false;
+    stream << term->getString();
+  }
+  stream << "]";
+  return stream.str();
+}
+
+NajaCollection<SNLBitTerm*> SNLEquipotential::getTerms() const {
+  return NajaCollection(new NajaSTLCollection(&terms_));
+}
+
+NajaCollection<SNLInstTermOccurrence> SNLEquipotential::getInstTermOccurrences() const {
+  return NajaCollection(new NajaSTLCollection(&instTermOccurrences_));
+}
+
 }} // namespace SNL // namespace naja