Update CrystalDiffraction.tsx

ImperialCollegeLondon · Dec 12, 2024 · b5ddbb1 · b5ddbb1
1 parent d224f46
commit b5ddbb1
Showing 1 changed file with 12 additions and 8 deletions.
diff --git a/components/CrystalDiffraction.tsx b/components/CrystalDiffraction.tsx
@@ -56,16 +56,20 @@ const CrystalDiffraction = () => {
   const reciprocalScale = 2;
 
   useEffect(() => {
-    let animationFrame;
-    if (isAnimating) {
-      const animate = () => {
-        setRotationZ(prev => (prev + 0.5) % 360);
-        animationFrame = requestAnimationFrame(animate);
-      };
+  let animationFrame: number | undefined;
+  if (isAnimating) {
+    const animate = () => {
+      setRotationZ(prev => (prev + 0.5) % 360);
       animationFrame = requestAnimationFrame(animate);
+    };
+    animationFrame = requestAnimationFrame(animate);
+  }
+  return () => {
+    if (animationFrame) {
+      cancelAnimationFrame(animationFrame);
     }
-    return () => cancelAnimationFrame(animationFrame);
-  }, [isAnimating]);
+  };
+}, [isAnimating]);
 
   // Generate 3D crystal lattice points
   const generateLatticePoints = () => {