From e70f8af448a3aaf214c5eb66f72fbf045db704ed Mon Sep 17 00:00:00 2001
From: Varun Agrawal <varagrawal@gmail.com>
Date: Thu, 15 Jun 2023 17:55:33 -0400
Subject: [PATCH] leftover classes to re-add

---
 gtsam/base/base.i     | 40 ++++++++++++++++++++++++++++++++++++++++
 python/CMakeLists.txt |  1 +
 2 files changed, 41 insertions(+)

diff --git a/gtsam/base/base.i b/gtsam/base/base.i
index d8cf9d8ea..31d09cb3d 100644
--- a/gtsam/base/base.i
+++ b/gtsam/base/base.i
@@ -41,9 +41,49 @@ class DSFMap {
   std::map<KEY, This::Set> sets();
 };
 
+// Used in Matlab wrapper
+class IndexPairSet {
+  IndexPairSet();
+  // common STL methods
+  size_t size() const;
+  bool empty() const;
+  void clear();
+
+  // structure specific methods
+  void insert(gtsam::IndexPair key);
+  bool erase(gtsam::IndexPair key);        // returns true if value was removed
+  bool count(gtsam::IndexPair key) const;  // returns true if value exists
+};
+
+// Used in Matlab wrapper
+class IndexPairVector {
+  IndexPairVector();
+  IndexPairVector(const gtsam::IndexPairVector& other);
+
+  // common STL methods
+  size_t size() const;
+  bool empty() const;
+  void clear();
+
+  // structure specific methods
+  gtsam::IndexPair at(size_t i) const;
+  void push_back(gtsam::IndexPair key) const;
+};
 
 gtsam::IndexPairVector IndexPairSetAsArray(gtsam::IndexPairSet& set);
 
+// Used in Matlab wrapper
+class IndexPairSetMap {
+  IndexPairSetMap();
+  // common STL methods
+  size_t size() const;
+  bool empty() const;
+  void clear();
+
+  // structure specific methods
+  gtsam::IndexPairSet at(gtsam::IndexPair& key);
+};
+
 #include <gtsam/base/Matrix.h>
 #include <gtsam/base/MatrixSerialization.h>
 bool linear_independent(Matrix A, Matrix B, double tol);
diff --git a/python/CMakeLists.txt b/python/CMakeLists.txt
index 684cf31a1..2b2abf507 100644
--- a/python/CMakeLists.txt
+++ b/python/CMakeLists.txt
@@ -48,6 +48,7 @@ set(ignore
     gtsam::ISAM2ThresholdMapValue
     gtsam::FactorIndices
     gtsam::FactorIndexSet
+    gtsam::IndexPairSet
     gtsam::IndexPairSetMap
     gtsam::IndexPairVector
     gtsam::BetweenFactorPose2s