export default function DataBlockFunnelToGPU()

in infrastructure/load-balancing-for-inference/components/DataBlockFunnelToGPU.tsx [26:96]
50 lines of code
5 McCabe index (conditional complexity)

export default function DataBlockFunnelToGPU({
  texturePath,
  funnelX,
  funnelY,
  gpuY,
  onComplete,
  fallSpeed = 4,
  gpuIndex = 0,
  funnelNumber = 1
}: DataBlockFunnelToGPUProps) {
  const meshRef = useRef<THREE.Mesh>(null!);
  const groupRef = useRef<THREE.Group>(null!);
  
  // Constants for consistent sizing
  const BLOCK_SIZE = 0.15;
  
  // Get the appropriate X offset based on funnel position
  const xOffset = FUNNEL_POSITION_OFFSETS[funnelNumber] || 0;
  
  // For debugging - log the position data
  useEffect(() => {
    console.log(`Funnel ${funnelNumber} at X=${funnelX}, using offset=${xOffset}, final X=${funnelX + xOffset}`);
  }, [funnelX, xOffset, funnelNumber]);
  
  // Load the data block texture.
  const texture = useLoader(TextureLoader, texturePath);
  
  // On mount, initialize the block's position with the correct offset
  useEffect(() => {
    if (groupRef.current) {
      // Apply the funnel-specific offset to the x position
      groupRef.current.position.set(funnelX + xOffset, funnelY + 0.3, 0.1);
    }
  }, [funnelX, funnelY, xOffset]);
  
  useFrame((_, delta) => {
    if (!groupRef.current) return;
    
    // Smoothly match the funnel's X (with appropriate offset)
    groupRef.current.position.x = THREE.MathUtils.lerp(
      groupRef.current.position.x,
      funnelX + xOffset,
      delta * 4
    );
    
    // Fall downward.
    groupRef.current.position.y -= delta * fallSpeed;
    
    // Once the block is below gpuY, call onComplete to remove it.
    if (groupRef.current.position.y <= gpuY) {
      onComplete();
    }
    
    // Keep rotation at 0 for a 2D effect.
    groupRef.current.rotation.set(0, 0, 0);
  });
  
  return (
    <group ref={groupRef}>
      {/* Actual datablock */}
      <mesh ref={meshRef} position={[0, 0, 0]}>
        <planeGeometry args={[BLOCK_SIZE, BLOCK_SIZE]} />
        <meshStandardMaterial 
          map={texture} 
          transparent 
          opacity={0.9} 
        />
      </mesh>
    </group>
  );
}