From aca62084d67d501e89714a16c7d586ba58927ebe Mon Sep 17 00:00:00 2001
From: Andreas Sundquist <asundqui@worldlabs.ai>
Date: Wed, 18 Jun 2025 08:28:15 -0700
Subject: [PATCH] Add types to SparkRenderer (#53)

---
 src/SparkRenderer.ts | 10 ++++++----
 1 file changed, 6 insertions(+), 4 deletions(-)

diff --git a/src/SparkRenderer.ts b/src/SparkRenderer.ts
index 818c0e6..7a94baf 100644
--- a/src/SparkRenderer.ts
+++ b/src/SparkRenderer.ts
@@ -43,19 +43,19 @@ let hasSparkRenderer = false;
 
 let sparkRendererInstance: SparkRenderer;
 
-function containsSplatMesh(object3D) {
+function containsSplatMesh(object3D: THREE.Object3D) {
   let hasSplatMesh = false;
   if (object3D instanceof SplatMesh) {
     return true;
   }
-  object3D.traverse((child) => {
+  object3D.traverse((child: THREE.Object3D) => {
     hasSplatMesh = hasSplatMesh || child instanceof SplatMesh;
   });
   return hasSplatMesh;
 }
 
 const sceneAdd = THREE.Scene.prototype.add;
-THREE.Scene.prototype.add = function (object) {
+THREE.Scene.prototype.add = function (object: THREE.Object3D) {
   hasSplatMesh = hasSplatMesh || containsSplatMesh(object);
   hasSparkRenderer = hasSparkRenderer || object instanceof SparkRenderer;
   sceneAdd.call(this, object);
@@ -63,7 +63,9 @@ THREE.Scene.prototype.add = function (object) {
 };
 
 const sceneOnBeforeRender = THREE.Scene.prototype.onBeforeRender;
-THREE.Scene.prototype.onBeforeRender = function (renderer) {
+THREE.Scene.prototype.onBeforeRender = function (
+  renderer: THREE.WebGLRenderer,
+) {
   if (!hasSplatMesh) {
     return;
   }