move projvars to stpmanager

meelgroup · Nov 24, 2024 · 4202ac9 · 4202ac9
1 parent 567a0f9
commit 4202ac9
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 1 deletion.
diff --git a/include/stp/STPManager/STPManager.h b/include/stp/STPManager/STPManager.h
@@ -72,6 +72,8 @@ class STPMgr
   // Table for variable names, let names etc.
   ASTSymbolSet _symbol_unique_table;
 
+  ASTNodeSet _proj_symbol_list;
+
   // Table to uniquefy bvconst
   ASTBVConstSet _bvconst_unique_table;
 
@@ -391,6 +393,19 @@ class STPMgr
     return CurrentSymbol;
   }
 
+  bool addProjSymbol(ASTNode& s)
+  {
+    _proj_symbol_list.insert(s);
+    std::cout << "Adding proj symbol: " << s << std::endl;
+    return true;
+  }
+
+  bool isProjSymbol(ASTNode& s)
+  {
+    return _proj_symbol_list.find(s) != _proj_symbol_list.end();
+  }
+
+
   bool FoundIntroducedSymbolSet(const ASTNode& in)
   {
     if (Introduced_SymbolsSet.find(in) != Introduced_SymbolsSet.end())

diff --git a/lib/Interface/cpp_interface.cpp b/lib/Interface/cpp_interface.cpp
@@ -341,7 +341,7 @@ void Cpp_interface::addSymbol(ASTNode& s)
 
 void Cpp_interface::addProjSymbol(ASTNode& s)
 {
-  getCurrentSymbols().push_back(s);
+    bm.addProjSymbol(s);
 }
 
 void Cpp_interface::success()