ObjectTable Save/Restore (and overloads/PurgeAllObjects/GetAllObjects)

src/xenia/kernel/util/object_table.h

@@ -18,6 +18,10 @@
 #include "xenia/kernel/xobject.h"
 #include "xenia/xbox.h"
 
+namespace xe {
+class ByteStream;
+}  // namespace xe
+
 namespace xe {
 namespace kernel {
 namespace util {
@@ -35,6 +39,13 @@ class ObjectTable {
   X_STATUS ReleaseHandle(X_HANDLE handle);
   X_STATUS RemoveHandle(X_HANDLE handle);
 
+  bool Save(ByteStream* stream);
+  bool Restore(ByteStream* stream);
+
+  // Restores a XObject reference with a handle. Mainly for internal use - do not
+  // use.
+  X_STATUS RestoreHandle(X_HANDLE handle, XObject* object);
+
   template <typename T>
   object_ref<T> LookupObject(X_HANDLE handle) {
     auto object = LookupObject(handle, false);
@@ -57,6 +68,18 @@ class ObjectTable {
     return results;
   }
 
+  template <typename T>
+  std::vector<object_ref<T>> GetObjectsByType() {
+    std::vector<object_ref<T>> results;
+    GetObjectsByType(
+        T::kType,
+        reinterpret_cast<std::vector<object_ref<XObject>>*>(&results));
+    return results;
+  }
+
+  std::vector<object_ref<XObject>> GetAllObjects();
+  void PurgeAllObjects(); // Purges the object table of all guest objects
+
  private:
   typedef struct {
     int handle_ref_count = 0;
@@ -70,6 +93,7 @@ class ObjectTable {
 
   X_HANDLE TranslateHandle(X_HANDLE handle);
   X_STATUS FindFreeSlot(uint32_t* out_slot);
+  bool Resize(uint32_t new_capacity);
 
   xe::global_critical_region global_critical_region_;
   uint32_t table_capacity_ = 0;