Set the nav mesh visibility when a nav mesh is loaded

Hubs-Foundation · Oct 12, 2023 · 80ae75d · 80ae75d
1 parent 484a3fb
commit 80ae75d
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/src/bit-systems/audio-debug-system.ts b/src/bit-systems/audio-debug-system.ts
@@ -117,11 +117,11 @@ getScene().then(() => {
 });
 
 const addDebugMaterial = (world: HubsWorld, navEid: number) => {
-  if (nav2mat.has(navEid)) return;
   const obj = world.eid2obj.get(navEid);
   if (obj) {
     const navMesh = obj as Mesh;
     navMesh.visible = isEnabled;
+    if (nav2mat.has(navEid)) return;
     nav2mat.set(navEid, navMesh.material);
     navMesh.material = debugMaterial!;
     navMesh.material.needsUpdate = true;
@@ -130,11 +130,11 @@ const addDebugMaterial = (world: HubsWorld, navEid: number) => {
 };
 
 const removeDebugMaterial = (world: HubsWorld, navEid: number) => {
-  if (!nav2mat.has(navEid)) return;
   const obj = world.eid2obj.get(navEid);
   if (obj) {
     const navMesh = obj as Mesh;
     navMesh.visible = false;
+    if (!nav2mat.has(navEid)) return;
     navMesh.material = nav2mat.get(navEid)!;
     nav2mat.delete(navEid);
     (navMesh.material as Material).needsUpdate = true;
@@ -154,6 +154,13 @@ export function audioDebugSystem(world: HubsWorld) {
   navMeshExitQuery(world).forEach(navEid => {
     removeDebugMaterial(world, navEid);
   });
+  navMeshEnterQuery(world).forEach(navEid => {
+    if (isEnabled) {
+      addDebugMaterial(world, navEid);
+    } else {
+      removeDebugMaterial(world, navEid);
+    }
+  });
   if (isEnabled && uniforms) {
     navMeshEnterQuery(world).forEach(navEid => {
       isEnabled && addDebugMaterial(world, navEid);