// ... imports ...
import { useEffect, useState, Suspense, Component, ReactNode, useCallback } from 'react';
import { Canvas } from '@react-three/fiber';
import { Html, CameraControls } from '@react-three/drei';
import { layoutApi, Floor3DData, Room3D, Position3D, Section3D } from '../lib/layoutApi';
import { Loader2, ArrowLeft, Maximize, MousePointer2, Layers, Thermometer, Droplets, Activity, Leaf, Eye, EyeOff, Clock, Menu } from 'lucide-react';
import { Link, useSearchParams } from 'react-router-dom';
import { FacilityScene } from '../components/facility3d/FacilityScene';
import { PlantSearch } from '../components/facility3d/PlantSearch';
import { PlantLibrary } from '../components/facility3d/PlantLibrary';
import { HierarchyNav } from '../components/facility3d/HierarchyNav';
import { TimelineSlider } from '../components/facility3d/TimelineSlider';
import { PlantPosition, VisMode, COLORS } from '../components/facility3d/types';
import { CameraPreset, CameraPresetSelector } from '../components/facility3d/CameraPresets';
import { HierarchyBreadcrumb } from '../components/facility3d/HierarchyBreadcrumb';
import { PlantDataCard } from '../components/facility3d/PlantDataCard';
import { SCALE } from '../components/facility3d/coordinates';
import Feature3D from '../components/aura/Feature3D';
// ... calculatePlantCoords and ErrorBoundary remain unchanged ...
function calculatePlantCoords(
section: Section3D,
position: { row: number; column: number; tier: number },
floorWidth: number,
floorHeight: number
): { x: number; y: number; z: number } {
// Scale section to world units (same as SmartRack)
const scaledSection = {
posX: section.posX * SCALE,
posY: section.posY * SCALE,
width: section.width * SCALE,
height: section.height * SCALE,
};
// Calculate grid dimensions (same as SmartRack)
const maxCols = Math.max(...section.positions.map(p => p.column), 1);
const maxRows = Math.max(...section.positions.map(p => p.row), 1);
// Calculate spacing (same as SmartRack)
const colSpacing = scaledSection.width / (maxCols + 1);
const rowSpacing = scaledSection.height / (maxRows + 1);
// Raw position within section (same as SmartRack)
const rawX = scaledSection.posX + (position.column * colSpacing);
const rawZ = scaledSection.posY + (position.row * rowSpacing);
const y = 0.4 + (position.tier * 0.6);
// Apply floor centering (matches FacilityScene offset group)
const floorCenterX = (floorWidth * SCALE) / 2;
const floorCenterZ = (floorHeight * SCALE) / 2;
return {
x: rawX - floorCenterX,
y,
z: rawZ - floorCenterZ,
};
}
// --- Error Boundary ---
interface ErrorBoundaryState { hasError: boolean; error: Error | null; }
class ErrorBoundary extends Component<{ children: ReactNode }, ErrorBoundaryState> {
state: ErrorBoundaryState = { hasError: false, error: null };
static getDerivedStateFromError(error: Error): ErrorBoundaryState {
return { hasError: true, error };
}
render() {
if (this.state.hasError) {
return (
3D Rendering Error
{this.state.error?.message}
);
}
return this.props.children;
}
}
// --- Main Component ---
export default function Facility3DViewerPage() {
const [showIntro, setShowIntro] = useState(true);
const [status, setStatus] = useState('Initializing...');
const [floorData, setFloorData] = useState(null);
const [selectedPlant, setSelectedPlant] = useState(null);
const [visMode, setVisMode] = useState('STANDARD');
// Mobile Nav State
const [isSidebarOpen, setIsSidebarOpen] = useState(false);
// Phase 2: Search state
const [highlightedTags, setHighlightedTags] = useState([]);
const [dimMode, setDimMode] = useState(false);
const [beaconPosition, setBeaconPosition] = useState<[number, number, number] | null>(null);
// Phase 3: Timeline state
const [showTimeline, setShowTimeline] = useState(false);
const [timelineDate, setTimelineDate] = useState(new Date());
const [isTimelinePlaying, setIsTimelinePlaying] = useState(false);
// Floor selection state
const [allFloors, setAllFloors] = useState<{ id: string; name: string; buildingName: string }[]>([]);
const [selectedFloorId, setSelectedFloorId] = useState(null);
// Camera preset state
const [cameraPreset, setCameraPreset] = useState('ISOMETRIC');
const [focusTarget, setFocusTarget] = useState<{ x: number; z: number } | null>(null);
const [searchParams] = useSearchParams();
const targetedPlantTag = searchParams.get('plant');
useEffect(() => {
// If coming from deep link, skip intro
if (targetedPlantTag) {
setShowIntro(false);
}
}, [targetedPlantTag]);
useEffect(() => {
loadData();
}, []);
// Load floor data when selected floor changes
useEffect(() => {
if (selectedFloorId) {
loadFloorData(selectedFloorId);
}
}, [selectedFloorId]);
// Deep link handler
useEffect(() => {
if (floorData && targetedPlantTag) {
for (const room of floorData.rooms) {
for (const section of room.sections) {
const match = section.positions.find((p: Position3D) => p.plant?.tagNumber === targetedPlantTag);
if (match) {
const { x, y, z } = calculatePlantCoords(
section,
{ row: match.row, column: match.column, tier: match.tier },
floorData.floor.width,
floorData.floor.height
);
const plant: PlantPosition = { ...match, x, y, z };
setSelectedPlant(plant);
setFocusTarget({ x, z });
setCameraPreset('ISOMETRIC');
setBeaconPosition([x, y, z]);
setHighlightedTags([targetedPlantTag]);
setDimMode(true);
return;
}
}
}
}
}, [floorData, targetedPlantTag]);
async function loadFloorData(floorId: string) {
setStatus('Fetching 3D scene...');
try {
const data = await layoutApi.getFloor3D(floorId);
setFloorData(data);
setStatus('');
} catch (err) {
setStatus('Error: ' + (err as Error).message);
}
}
async function loadData() {
setStatus('Loading layout...');
try {
const props = await layoutApi.getProperties();
if (props[0]?.buildings?.length > 0) {
// Collect all floors from all buildings
const floors: { id: string; name: string; buildingName: string }[] = [];
for (const building of props[0].buildings) {
for (const floor of building.floors || []) {
floors.push({
id: floor.id,
name: floor.name,
buildingName: building.name
});
}
}
setAllFloors(floors);
// Default to second floor if available (Upper Floor has more plants)
if (floors.length > 0) {
const defaultFloor = floors.length > 1 ? floors[1] : floors[0];
setSelectedFloorId(defaultFloor.id);
}
} else {
setStatus('No floor layout found. Set up a facility first.');
}
} catch (err) {
setStatus('Error: ' + (err as Error).message);
}
}
const focusRoom = (room: Room3D) => {
if (!floorData) return;
const SCALE = 0.1;
// Calculate focus target in scaled world coordinates
const x = (room.posX + room.width / 2) * SCALE - (floorData.floor.width * SCALE / 2);
const z = (room.posY + room.height / 2) * SCALE - (floorData.floor.height * SCALE / 2);
setFocusTarget({ x, z });
setCameraPreset('ISOMETRIC');
// Clear search state when navigating by room
setBeaconPosition(null);
setHighlightedTags([]);
setDimMode(false);
// Mobile UX: Close sidebar after selection
setIsSidebarOpen(false);
};
const resetView = () => {
setCameraPreset('ISOMETRIC');
setFocusTarget(null);
setBeaconPosition(null);
setHighlightedTags([]);
setDimMode(false);
setSelectedPlant(null);
};
// Breadcrumb navigation handler
const handleBreadcrumbNavigate = (level: string) => {
if (!floorData || !selectedPlant) return;
if (level === 'room' && selectedPlant.breadcrumb?.room) {
// Find the room by name and focus on it
const room = floorData.rooms.find(r => r.name === selectedPlant.breadcrumb?.room);
if (room) {
focusRoom(room);
}
} else if (level === 'floor' || level === 'building' || level === 'facility') {
// Return to floor overview
setCameraPreset('ISOMETRIC');
setFocusTarget(null);
}
// Section/tier clicks stay on current view but could highlight
};
// Search result selection handler
const handleSearchSelect = useCallback((result: any) => {
if (!floorData) return;
// Find the section - prefer ID lookup for reliability, fall back to code/name
let section: Section3D | undefined;
for (const room of floorData.rooms) {
if (result.sectionId) {
section = room.sections.find(s => s.id === result.sectionId);
} else {
// Fallback: must match both room and section code
if (room.name === result.roomName) {
section = room.sections.find(s => (s.code || s.name) === result.sectionCode);
}
}
if (section) break;
}
if (!section) {
console.warn('[handleSearchSelect] Could not find section for result:', result);
return;
}
const { x, y, z } = calculatePlantCoords(
section,
{ row: result.position.row, column: result.position.column, tier: result.position.tier },
floorData.floor.width,
floorData.floor.height
);
setFocusTarget({ x, z });
setCameraPreset('ISOMETRIC');
setBeaconPosition([x, y, z]);
setHighlightedTags([result.label]);
setDimMode(true);
// Include breadcrumb data from search result
setSelectedPlant({
...result.plant,
x,
y,
z,
breadcrumb: {
room: result.roomName,
section: result.sectionCode,
},
});
}, [floorData]);
// Highlight handler for search typing
const handleHighlightResults = useCallback((tags: string[]) => {
setHighlightedTags(tags);
if (tags.length > 0) {
setDimMode(true);
} else {
setDimMode(false);
setBeaconPosition(null);
}
}, []);
return (
{/* Intro Overlay */}
{showIntro && (
setShowIntro(false)} />
)}
{/* Header */}
{/* Mobile Menu Toggle */}
Back
Facility 3D
3D View
BETA
{allFloors.length > 1 ? (
) : (
{floorData?.floor.name || 'Loading...'}
)}
• {floorData?.stats.occupiedPositions || 0} Plants
{/* Search Bar - Flexible width on mobile */}
{floorData && (
)}
{/* Camera Preset Selector - Hide on very small screens if crowded */}
{
setCameraPreset(preset);
}}
/>
{/* Legend - Responsive positioning */}
{visMode === 'STANDARD' && (
<>
>
)}
{visMode === 'HEALTH' && (
<>
>
)}
{/* ... other legend items ... */}
{/* Room Navigation Sidebar - Responsive Drawer */}
{floorData && (
{/* Hierarchical Navigation */}
{/* Focus Mode Toggle */}
{highlightedTags.length > 0 && (
)}
{/* Plant Library Browser */}
{
// ... check logic ...
const room = floorData.rooms.find(r => r.name === roomName);
const section = room?.sections.find(s => (s.code || s.name) === sectionCode);
if (section) {
const { x, y, z } = calculatePlantCoords(
section,
{ row: plant.row, column: plant.column, tier: plant.tier },
floorData.floor.width,
floorData.floor.height
);
setFocusTarget({ x, z });
setCameraPreset('ISOMETRIC');
setBeaconPosition([x, y, z]);
setDimMode(false);
setHighlightedTags([plant.plant?.tagNumber || '']);
setSelectedPlant({
...plant,
x,
y,
z,
breadcrumb: {
room: roomName,
section: sectionCode,
},
});
setIsSidebarOpen(false); // Close mobile drawer
}
}}
/>
)}
{/* Sidebar Overlay for Mobile */}
{isSidebarOpen && (
setIsSidebarOpen(false)}
/>
)}
{/* Visualization Mode Buttons - Responsive Position */}
{[
{ mode: 'STANDARD' as VisMode, icon: Layers, label: 'Layout', active: 'bg-primary' },
{ mode: 'HEALTH' as VisMode, icon: Activity, label: 'Health', active: 'bg-red-500' },
{ mode: 'TEMP' as VisMode, icon: Thermometer, label: 'Temp', active: 'bg-orange-500' },
{ mode: 'HUMIDITY' as VisMode, icon: Droplets, label: 'VPD', active: 'bg-blue-500' },
{ mode: 'YIELD' as VisMode, icon: Leaf, label: 'Yield', active: 'bg-green-600' },
].map(({ mode, icon: Icon, label, active }) => (
))}
{/* Timeline Toggle */}
{/* Loading/Error Overlay - Unchanged */}
{status && (
{status}
{status.includes('Error') && (
)}
)}
{/* Plant Data Card - Responsive Bottom Sheet/Side Panel */}
{selectedPlant?.plant && (
{
setSelectedPlant(null);
setBeaconPosition(null);
}}
/>
)}
{/* 3D Canvas - Unchanged */}