Also track ANARILight objects for leak detection

KhronosGroup · Oct 22, 2024 · f389af1 · f389af1
1 parent 17aba42
commit f389af1
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 0 deletions.
diff --git a/src/helium/BaseDevice.cpp b/src/helium/BaseDevice.cpp
@@ -280,6 +280,7 @@ BaseDevice::~BaseDevice()
   reportLeaks(state.objectCounts.worlds, "ANARIWorld");
   reportLeaks(state.objectCounts.instances, "ANARIInstance");
   reportLeaks(state.objectCounts.groups, "ANARIGroup");
+  reportLeaks(state.objectCounts.lights, "ANARILight");
   reportLeaks(state.objectCounts.surfaces, "ANARISurface");
   reportLeaks(state.objectCounts.geometries, "ANARIGeometry");
   reportLeaks(state.objectCounts.materials, "ANARIMaterial");

diff --git a/src/helium/BaseGlobalDeviceState.h b/src/helium/BaseGlobalDeviceState.h
@@ -51,6 +51,7 @@ struct BaseGlobalDeviceState
     std::atomic<size_t> worlds{0};
     std::atomic<size_t> instances{0};
     std::atomic<size_t> groups{0};
+    std::atomic<size_t> lights{0};
     std::atomic<size_t> surfaces{0};
     std::atomic<size_t> geometries{0};
     std::atomic<size_t> materials{0};

diff --git a/src/helium/BaseObject.cpp b/src/helium/BaseObject.cpp
@@ -144,6 +144,9 @@ void BaseObject::incrementObjectCount()
   case ANARI_GROUP:
     s->objectCounts.groups++;
     break;
+  case ANARI_LIGHT:
+    s->objectCounts.lights++;
+    break;
   case ANARI_SURFACE:
     s->objectCounts.surfaces++;
     break;
@@ -200,6 +203,9 @@ void BaseObject::decrementObjectCount()
   case ANARI_GROUP:
     s->objectCounts.groups--;
     break;
+  case ANARI_LIGHT:
+    s->objectCounts.lights--;
+    break;
   case ANARI_SURFACE:
     s->objectCounts.surfaces--;
     break;