trustgraph/src/components/graph/GraphCanvas.tsx
elpresidank ad40332d56 Squashed 'ai-context/trustgraph-templates/' content from commit 338a8ffa
git-subtree-dir: ai-context/trustgraph-templates
git-subtree-split: 338a8ffadb1439013071ae922e55ed2421f17025
2026-04-05 21:08:57 -05:00

596 lines
21 KiB
TypeScript

import { useEffect, useRef, useCallback, useState, MouseEvent } from "react";
import type { DomainKey, Entity, GraphNode, OntologyType, Relationship } from "../../types";
import { ZoomControls } from "./ZoomControls";
import { border } from "../../theme";
interface GraphCanvasProps {
entities: Entity[];
relationships: Relationship[];
ontology: OntologyType;
highlightedEntities: string[];
onNodeClick: (node: GraphNode) => void;
activeFilter: DomainKey | null;
}
const SETTLE_TIME = 10000; // 10 seconds until nodes settle
const FRAME_INTERVAL = 1000 / 30; // 30fps
export function GraphCanvas({ entities, relationships, ontology, highlightedEntities, onNodeClick, activeFilter }: GraphCanvasProps) {
const containerRef = useRef<HTMLDivElement>(null);
const staticCanvasRef = useRef<HTMLCanvasElement>(null);
const nodesCanvasRef = useRef<HTMLCanvasElement>(null);
const edgesCanvasRef = useRef<HTMLCanvasElement>(null);
const nodesRef = useRef<GraphNode[]>([]);
const animRef = useRef<number>(0);
const hoveredRef = useRef<string | null>(null);
const settledRef = useRef<boolean>(false);
const startTimeRef = useRef<number>(0);
const timeRef = useRef<number>(0);
const lastFrameTimeRef = useRef<number>(0);
// Store view state in refs to avoid triggering resets
const highlightedRef = useRef<string[]>(highlightedEntities);
const activeFilterRef = useRef<DomainKey | null>(activeFilter);
const relationshipsRef = useRef<Relationship[]>(relationships);
const ontologyRef = useRef<OntologyType>(ontology);
const [hovered, setHovered] = useState<string | null>(null);
const [containerSize, setContainerSize] = useState({ width: 0, height: 0 });
// Zoom and pan state
const [zoom, setZoom] = useState(1);
const [pan, setPan] = useState({ x: 0, y: 0 });
const zoomRef = useRef(1);
const panRef = useRef({ x: 0, y: 0 });
const isPanningRef = useRef(false);
const lastPanPosRef = useRef({ x: 0, y: 0 });
// Keep zoom/pan refs in sync
zoomRef.current = zoom;
panRef.current = pan;
// Keep refs in sync with props
useEffect(() => {
highlightedRef.current = highlightedEntities;
}, [highlightedEntities]);
useEffect(() => {
activeFilterRef.current = activeFilter;
}, [activeFilter]);
useEffect(() => {
relationshipsRef.current = relationships;
}, [relationships]);
useEffect(() => {
ontologyRef.current = ontology;
}, [ontology]);
// Track container size changes
useEffect(() => {
const container = containerRef.current;
if (!container) return;
const resizeObserver = new ResizeObserver((entries) => {
const entry = entries[0];
if (entry) {
setContainerSize({
width: entry.contentRect.width,
height: entry.contentRect.height,
});
}
});
resizeObserver.observe(container);
return () => resizeObserver.disconnect();
}, []);
// Draw static layer (grid + domain labels)
const drawStaticLayer = useCallback((ctx: CanvasRenderingContext2D, canvas: HTMLCanvasElement, domainPositions: Record<DomainKey, { x: number; y: number }>) => {
ctx.clearRect(0, 0, canvas.width, canvas.height);
// Grid stays fixed (no transform)
ctx.strokeStyle = border.grid;
ctx.lineWidth = 1;
for (let x = 0; x < canvas.width; x += 60) {
ctx.beginPath(); ctx.moveTo(x, 0); ctx.lineTo(x, canvas.height); ctx.stroke();
}
for (let y = 0; y < canvas.height; y += 60) {
ctx.beginPath(); ctx.moveTo(0, y); ctx.lineTo(canvas.width, y); ctx.stroke();
}
// Domain labels with zoom/pan transform
ctx.save();
ctx.translate(panRef.current.x, panRef.current.y);
ctx.scale(zoomRef.current, zoomRef.current);
const currentOntology = ontologyRef.current;
(Object.entries(domainPositions) as [DomainKey, { x: number; y: number }][]).forEach(([domain, pos]) => {
const data = currentOntology[domain];
ctx.font = "bold 22px 'IBM Plex Mono', monospace";
ctx.fillStyle = data.color + "44";
ctx.textAlign = "center";
ctx.fillText(data.label.toUpperCase(), pos.x, pos.y - Math.min(canvas.width, canvas.height) * 0.14);
});
ctx.restore();
}, []);
// Draw nodes layer - reads from refs
const drawNodesLayer = useCallback((ctx: CanvasRenderingContext2D, canvas: HTMLCanvasElement, time: number) => {
ctx.clearRect(0, 0, canvas.width, canvas.height);
// Apply zoom/pan transform
ctx.save();
ctx.translate(panRef.current.x, panRef.current.y);
ctx.scale(zoomRef.current, zoomRef.current);
const nodes = nodesRef.current;
const settled = settledRef.current;
const highlighted = highlightedRef.current;
const filter = activeFilterRef.current;
const rels = relationshipsRef.current;
nodes.forEach((node) => {
const isHighlighted = highlighted && highlighted.includes(node.id);
const isHovered = hoveredRef.current === node.id;
const isDimmed = highlighted && highlighted.length > 0 && !isHighlighted;
const isFiltered = filter && node.domain !== filter && !rels.some(
r => r.domain.includes(filter) && (r.from === node.id || r.to === node.id)
);
const alpha = isFiltered ? 0.15 : isDimmed ? 0.3 : 1;
const r = isHighlighted || isHovered ? node.r * 1.4 : node.r;
const pulseR = isHighlighted && !settled ? Math.sin(time * 3) * 3 : 0;
// Glow
if ((isHighlighted || isHovered) && !isFiltered) {
ctx.beginPath();
ctx.arc(node.x, node.y, r + 12 + pulseR, 0, Math.PI * 2);
const grd = ctx.createRadialGradient(node.x, node.y, r, node.x, node.y, r + 12 + pulseR);
grd.addColorStop(0, node.glow);
grd.addColorStop(1, "rgba(0,0,0,0)");
ctx.fillStyle = grd;
ctx.fill();
}
// Node circle
ctx.beginPath();
ctx.arc(node.x, node.y, r, 0, Math.PI * 2);
ctx.fillStyle = node.color + Math.round(alpha * 255 * 0.2).toString(16).padStart(2, "0");
ctx.fill();
ctx.strokeStyle = node.color + Math.round(alpha * 255).toString(16).padStart(2, "0");
ctx.lineWidth = isHighlighted ? 2.5 : 1.5;
ctx.stroke();
// Label
ctx.font = `${isHighlighted ? "bold " : ""}${isHovered ? 17 : 14}px 'IBM Plex Sans', sans-serif`;
ctx.fillStyle = `rgba(255,255,255,${alpha * (isHighlighted ? 1 : 0.75)})`;
ctx.textAlign = "center";
ctx.fillText(node.label, node.x, node.y + r + 18);
// Update node positions (spring physics + drift) - only if not settled
if (!settled) {
node.x += (node.targetX - node.x) * 0.02;
node.y += (node.targetY - node.y) * 0.02;
node.x += Math.sin(time + node.targetX * 0.01) * 0.3;
node.y += Math.cos(time + node.targetY * 0.01) * 0.3;
}
});
ctx.restore();
}, []);
// Draw edges layer - reads from refs
const drawEdgesLayer = useCallback((ctx: CanvasRenderingContext2D, canvas: HTMLCanvasElement, time: number) => {
ctx.clearRect(0, 0, canvas.width, canvas.height);
// Apply zoom/pan transform
ctx.save();
ctx.translate(panRef.current.x, panRef.current.y);
ctx.scale(zoomRef.current, zoomRef.current);
const nodes = nodesRef.current;
const highlighted = highlightedRef.current;
const filter = activeFilterRef.current;
const rels = relationshipsRef.current;
const filteredRels = filter
? rels.filter((r) => r.domain.includes(filter))
: rels;
filteredRels.forEach((rel) => {
const fromNode = nodes.find((n) => n.id === rel.from);
const toNode = nodes.find((n) => n.id === rel.to);
if (!fromNode || !toNode) return;
const isHighlighted =
highlighted &&
highlighted.includes(rel.from) &&
highlighted.includes(rel.to);
const baseAlpha = isHighlighted ? 0.7 : 0.12;
const pulse = isHighlighted ? Math.sin(time * 4) * 0.15 + 0.15 : 0;
ctx.beginPath();
ctx.moveTo(fromNode.x, fromNode.y);
// Curved edges
const mx = (fromNode.x + toNode.x) / 2 + (fromNode.y - toNode.y) * 0.1;
const my = (fromNode.y + toNode.y) / 2 + (toNode.x - fromNode.x) * 0.1;
ctx.quadraticCurveTo(mx, my, toNode.x, toNode.y);
const gradient = ctx.createLinearGradient(fromNode.x, fromNode.y, toNode.x, toNode.y);
gradient.addColorStop(0, fromNode.color + Math.round((baseAlpha + pulse) * 255).toString(16).padStart(2, "0"));
gradient.addColorStop(1, toNode.color + Math.round((baseAlpha + pulse) * 255).toString(16).padStart(2, "0"));
ctx.strokeStyle = gradient;
ctx.lineWidth = isHighlighted ? 3 : 1.5;
ctx.stroke();
// Animated particles on highlighted edges
if (isHighlighted) {
const t = (time * 2) % 1;
const px = (1 - t) * (1 - t) * fromNode.x + 2 * (1 - t) * t * mx + t * t * toNode.x;
const py = (1 - t) * (1 - t) * fromNode.y + 2 * (1 - t) * t * my + t * t * toNode.y;
ctx.beginPath();
ctx.arc(px, py, 3, 0, Math.PI * 2);
ctx.fillStyle = "#fff";
ctx.fill();
}
});
ctx.restore();
}, []);
// Animation loop function - separate from setup
const runAnimation = useCallback(() => {
const nodesCanvas = nodesCanvasRef.current;
const edgesCanvas = edgesCanvasRef.current;
const nodesCtx = nodesCanvas?.getContext("2d");
const edgesCtx = edgesCanvas?.getContext("2d");
if (!nodesCtx || !nodesCanvas || !edgesCtx || !edgesCanvas) return;
// Capture validated references for the closure
const validNodesCtx = nodesCtx;
const validNodesCanvas = nodesCanvas;
const validEdgesCtx = edgesCtx;
const validEdgesCanvas = edgesCanvas;
function animate(currentTime: number) {
// Throttle to target fps
if (currentTime - lastFrameTimeRef.current < FRAME_INTERVAL) {
animRef.current = requestAnimationFrame(animate);
return;
}
lastFrameTimeRef.current = currentTime;
timeRef.current += 0.01;
// Check if we should settle
if (!settledRef.current && currentTime - startTimeRef.current > SETTLE_TIME) {
settledRef.current = true;
}
const hasHighlights = highlightedRef.current && highlightedRef.current.length > 0;
const isSettled = settledRef.current;
// Draw edges layer
drawEdgesLayer(validEdgesCtx, validEdgesCanvas, timeRef.current);
// Draw nodes layer
if (!isSettled || hasHighlights || hoveredRef.current) {
drawNodesLayer(validNodesCtx, validNodesCanvas, timeRef.current);
}
// Continue animation if not settled, or if there are highlights
if (!isSettled || hasHighlights) {
animRef.current = requestAnimationFrame(animate);
} else {
// Settled with no highlights - do one final draw and stop
drawNodesLayer(validNodesCtx, validNodesCanvas, timeRef.current);
drawEdgesLayer(validEdgesCtx, validEdgesCanvas, timeRef.current);
animRef.current = 0;
}
}
animRef.current = requestAnimationFrame(animate);
}, [drawNodesLayer, drawEdgesLayer]);
// Main setup - only runs when data or size changes
useEffect(() => {
const staticCanvas = staticCanvasRef.current;
const nodesCanvas = nodesCanvasRef.current;
const edgesCanvas = edgesCanvasRef.current;
if (!staticCanvas || !nodesCanvas || !edgesCanvas || containerSize.width === 0) return;
// Cancel any existing animation
if (animRef.current) {
cancelAnimationFrame(animRef.current);
animRef.current = 0;
}
// Setup all canvases
[staticCanvas, nodesCanvas, edgesCanvas].forEach(canvas => {
canvas.width = containerSize.width * 2;
canvas.height = containerSize.height * 2;
canvas.style.width = containerSize.width + "px";
canvas.style.height = containerSize.height + "px";
});
const cx = staticCanvas.width / 2;
const cy = staticCanvas.height / 2;
// Position nodes in domain clusters
const domainKeys = Object.keys(ontology);
const domainPositions: Record<DomainKey, { x: number; y: number }> = {};
domainKeys.forEach((domain, i) => {
const angle = (Math.PI * 2 * i) / domainKeys.length - Math.PI / 2;
const radius = Math.min(cx, cy) * 0.45;
domainPositions[domain] = {
x: cx + Math.cos(angle) * radius,
y: cy + Math.sin(angle) * radius,
};
});
nodesRef.current = entities.map((e) => {
const dp = domainPositions[e.domain];
const subIdx = ontology[e.domain].subclasses.findIndex((s) => s.id === e.id);
const total = ontology[e.domain].subclasses.length;
const angle = ((Math.PI * 2) / total) * subIdx - Math.PI / 2;
const radius = Math.min(staticCanvas.width, staticCanvas.height) * 0.1;
return {
...e,
x: dp.x + Math.cos(angle) * radius,
y: dp.y + Math.sin(angle) * radius,
vx: 0,
vy: 0,
targetX: dp.x + Math.cos(angle) * radius,
targetY: dp.y + Math.sin(angle) * radius,
r: 18,
};
});
const staticCtx = staticCanvas.getContext("2d");
if (!staticCtx) return;
// Draw static layer once
drawStaticLayer(staticCtx, staticCanvas, domainPositions);
// Reset animation state
settledRef.current = false;
startTimeRef.current = performance.now();
timeRef.current = 0;
lastFrameTimeRef.current = 0;
// Start animation
runAnimation();
return () => {
if (animRef.current) {
cancelAnimationFrame(animRef.current);
animRef.current = 0;
}
};
}, [entities, ontology, containerSize, drawStaticLayer, runAnimation]);
// Restart animation when highlights change (without resetting positions)
useEffect(() => {
const hasHighlights = highlightedEntities && highlightedEntities.length > 0;
// If we have highlights and animation isn't running, restart it
if (hasHighlights && animRef.current === 0) {
runAnimation();
}
}, [highlightedEntities, runAnimation]);
// Redraw on filter change (without resetting)
useEffect(() => {
const nodesCanvas = nodesCanvasRef.current;
const edgesCanvas = edgesCanvasRef.current;
const nodesCtx = nodesCanvas?.getContext("2d");
const edgesCtx = edgesCanvas?.getContext("2d");
if (nodesCtx && nodesCanvas && edgesCtx && edgesCanvas && settledRef.current && animRef.current === 0) {
drawNodesLayer(nodesCtx, nodesCanvas, timeRef.current);
drawEdgesLayer(edgesCtx, edgesCanvas, timeRef.current);
}
}, [activeFilter, drawNodesLayer, drawEdgesLayer]);
const handleMouseMove = useCallback((e: MouseEvent<HTMLCanvasElement>) => {
// Handle panning first
if (isPanningRef.current) {
const dx = (e.clientX - lastPanPosRef.current.x) * 2;
const dy = (e.clientY - lastPanPosRef.current.y) * 2;
lastPanPosRef.current = { x: e.clientX, y: e.clientY };
setPan(p => ({ x: p.x + dx, y: p.y + dy }));
return;
}
const canvas = nodesCanvasRef.current;
if (!canvas) return;
const rect = canvas.getBoundingClientRect();
// Transform screen coordinates to world coordinates (accounting for zoom/pan)
const screenX = (e.clientX - rect.left) * 2;
const screenY = (e.clientY - rect.top) * 2;
const x = (screenX - panRef.current.x) / zoomRef.current;
const y = (screenY - panRef.current.y) / zoomRef.current;
const nodes = nodesRef.current;
let found: string | null = null;
for (const node of nodes) {
const dx = node.x - x;
const dy = node.y - y;
if (Math.sqrt(dx * dx + dy * dy) < node.r * 1.5) {
found = node.id;
break;
}
}
const wasHovered = hoveredRef.current;
hoveredRef.current = found;
setHovered(found);
canvas.style.cursor = isPanningRef.current ? "grabbing" : (found ? "pointer" : "default");
// Redraw if hover state changed and we're settled
if (wasHovered !== found && settledRef.current) {
const nodesCanvas = nodesCanvasRef.current;
const edgesCanvas = edgesCanvasRef.current;
const nodesCtx = nodesCanvas?.getContext("2d");
const edgesCtx = edgesCanvas?.getContext("2d");
if (nodesCtx && nodesCanvas && edgesCtx && edgesCanvas) {
drawNodesLayer(nodesCtx, nodesCanvas, timeRef.current);
drawEdgesLayer(edgesCtx, edgesCanvas, timeRef.current);
}
}
}, [drawNodesLayer, drawEdgesLayer]);
const handleClick = useCallback((e: MouseEvent<HTMLCanvasElement>) => {
// Don't trigger click if we were panning
if (e.shiftKey) return;
if (hoveredRef.current && onNodeClick) {
const node = nodesRef.current.find((n) => n.id === hoveredRef.current);
if (node) onNodeClick(node);
}
}, [onNodeClick]);
// Redraw all layers (used when zoom/pan changes)
const redrawAllLayers = useCallback(() => {
const staticCanvas = staticCanvasRef.current;
const nodesCanvas = nodesCanvasRef.current;
const edgesCanvas = edgesCanvasRef.current;
const staticCtx = staticCanvas?.getContext("2d");
const nodesCtx = nodesCanvas?.getContext("2d");
const edgesCtx = edgesCanvas?.getContext("2d");
if (!staticCtx || !staticCanvas || !nodesCtx || !nodesCanvas || !edgesCtx || !edgesCanvas) return;
// Recalculate domain positions for static layer redraw
const cx = staticCanvas.width / 2;
const cy = staticCanvas.height / 2;
const domainKeys = Object.keys(ontologyRef.current);
const domainPositions: Record<DomainKey, { x: number; y: number }> = {};
domainKeys.forEach((domain, i) => {
const angle = (Math.PI * 2 * i) / domainKeys.length - Math.PI / 2;
const radius = Math.min(cx, cy) * 0.45;
domainPositions[domain] = {
x: cx + Math.cos(angle) * radius,
y: cy + Math.sin(angle) * radius,
};
});
drawStaticLayer(staticCtx, staticCanvas, domainPositions);
drawEdgesLayer(edgesCtx, edgesCanvas, timeRef.current);
drawNodesLayer(nodesCtx, nodesCanvas, timeRef.current);
}, [drawStaticLayer, drawEdgesLayer, drawNodesLayer]);
// Zoom handler - zoom towards cursor position
const handleWheel = useCallback((e: React.WheelEvent) => {
e.preventDefault();
const delta = e.deltaY > 0 ? 0.9 : 1.1;
const newZoom = Math.min(4, Math.max(0.25, zoomRef.current * delta));
const canvas = nodesCanvasRef.current;
if (!canvas) return;
const rect = canvas.getBoundingClientRect();
const cursorX = (e.clientX - rect.left) * 2; // Account for 2x canvas scaling
const cursorY = (e.clientY - rect.top) * 2;
// Adjust pan to zoom towards cursor
const zoomRatio = newZoom / zoomRef.current;
const newPanX = cursorX - (cursorX - panRef.current.x) * zoomRatio;
const newPanY = cursorY - (cursorY - panRef.current.y) * zoomRatio;
setZoom(newZoom);
setPan({ x: newPanX, y: newPanY });
}, []);
// Pan handlers
const handleMouseDown = useCallback((e: MouseEvent<HTMLCanvasElement>) => {
if (e.button === 1 || (e.button === 0 && e.shiftKey)) {
e.preventDefault();
isPanningRef.current = true;
lastPanPosRef.current = { x: e.clientX, y: e.clientY };
}
}, []);
const handleMouseUp = useCallback(() => {
isPanningRef.current = false;
}, []);
// Reset zoom/pan
const handleResetView = useCallback(() => {
setZoom(1);
setPan({ x: 0, y: 0 });
}, []);
// Redraw when zoom/pan changes
useEffect(() => {
if (containerSize.width > 0) {
redrawAllLayers();
}
}, [zoom, pan, containerSize, redrawAllLayers]);
const canvasStyle: React.CSSProperties = {
position: "absolute",
top: 0,
left: 0,
width: "100%",
height: "100%",
};
return (
<div
ref={containerRef}
style={{ position: "relative", width: "100%", height: "100%", overflow: "hidden" }}
onMouseUp={handleMouseUp}
onMouseLeave={handleMouseUp}
>
{/* Layer 1: Static (grid + domain labels) */}
<canvas ref={staticCanvasRef} style={canvasStyle} />
{/* Layer 2: Edges */}
<canvas ref={edgesCanvasRef} style={canvasStyle} />
{/* Layer 3: Nodes (on top for interaction) */}
<canvas
ref={nodesCanvasRef}
onMouseMove={handleMouseMove}
onMouseDown={handleMouseDown}
onClick={handleClick}
onWheel={handleWheel}
style={canvasStyle}
/>
<ZoomControls
zoom={zoom}
onZoomIn={() => setZoom(z => Math.min(4, z * 1.2))}
onZoomOut={() => setZoom(z => Math.max(0.25, z / 1.2))}
onReset={handleResetView}
/>
{/* Tooltip */}
{hovered && (() => {
const node = nodesRef.current.find((n) => n.id === hovered);
if (!node) return null;
// Transform node position to screen coordinates
const sx = (node.x * zoomRef.current + panRef.current.x) / 2;
const sy = (node.y * zoomRef.current + panRef.current.y) / 2;
return (
<div style={{
position: "absolute", left: sx + 20, top: sy - 20,
background: "rgba(15,15,20,0.95)", border: `1px solid ${node.color}44`,
borderRadius: 8, padding: "10px 14px", pointerEvents: "none",
backdropFilter: "blur(12px)", zIndex: 10, minWidth: 180,
}}>
<div style={{ color: node.color, fontWeight: 700, fontSize: 13, fontFamily: "'IBM Plex Mono', monospace" }}>
{node.icon} {node.label}
</div>
<div style={{ color: "#888", fontSize: 11, marginTop: 4, fontFamily: "'IBM Plex Mono', monospace" }}>
{Object.entries(node.props || {}).map(([k, v]) => (
<div key={k}><span style={{ color: "#666" }}>{k}:</span> <span style={{ color: "#ccc" }}>{String(v)}</span></div>
))}
</div>
</div>
);
})()}
</div>
);
}