94 lines
3.0 KiB
TypeScript
94 lines
3.0 KiB
TypeScript
import React, { memo } from 'react';
|
|
import cc from 'classcat';
|
|
|
|
import { useStoreState } from '../../store/hooks';
|
|
import { getRectOfNodes, getBoundsofRects } from '../../utils/graph';
|
|
import { Node, Rect } from '../../types';
|
|
import MiniMapNode from './MiniMapNode';
|
|
|
|
import './style.css';
|
|
|
|
type StringFunc = (node: Node) => string;
|
|
|
|
interface MiniMapProps extends React.HTMLAttributes<SVGSVGElement> {
|
|
nodeColor?: string | StringFunc;
|
|
nodeBorderRadius?: number;
|
|
maskColor?: string;
|
|
}
|
|
|
|
const defaultWidth = 200;
|
|
const defaultHeight = 150;
|
|
|
|
const MiniMap = ({
|
|
style = { backgroundColor: '#f8f8f8' },
|
|
className,
|
|
nodeColor = '#ddd',
|
|
nodeBorderRadius = 5,
|
|
maskColor = 'rgba(10, 10, 10, .25)',
|
|
}: MiniMapProps) => {
|
|
const containerWidth = useStoreState((s) => s.width);
|
|
const containerHeight = useStoreState((s) => s.height);
|
|
const [tX, tY, tScale] = useStoreState((s) => s.transform);
|
|
const nodes = useStoreState((s) => s.nodes);
|
|
|
|
const mapClasses = cc(['react-flow__minimap', className]);
|
|
const elementWidth = (style.width || defaultWidth)! as number;
|
|
const elementHeight = (style.height || defaultHeight)! as number;
|
|
const nodeColorFunc = (nodeColor instanceof Function ? nodeColor : () => nodeColor) as StringFunc;
|
|
const hasNodes = nodes && nodes.length;
|
|
const bb = getRectOfNodes(nodes);
|
|
const viewBB: Rect = {
|
|
x: -tX / tScale,
|
|
y: -tY / tScale,
|
|
width: containerWidth / tScale,
|
|
height: containerHeight / tScale,
|
|
};
|
|
const boundingRect = hasNodes ? getBoundsofRects(bb, viewBB) : viewBB;
|
|
const scaledWidth = boundingRect.width / elementWidth;
|
|
const scaledHeight = boundingRect.height / elementHeight;
|
|
const viewScale = Math.max(scaledWidth, scaledHeight);
|
|
const viewWidth = viewScale * elementWidth;
|
|
const viewHeight = viewScale * elementHeight;
|
|
const offset = 5 * viewScale;
|
|
const x = boundingRect.x - (viewWidth - boundingRect.width) / 2 - offset;
|
|
const y = boundingRect.y - (viewHeight - boundingRect.height) / 2 - offset;
|
|
const width = viewWidth + offset * 2;
|
|
const height = viewHeight + offset * 2;
|
|
|
|
return (
|
|
<svg
|
|
width={elementWidth}
|
|
height={elementHeight}
|
|
viewBox={`${x} ${y} ${width} ${height}`}
|
|
style={style}
|
|
className={mapClasses}
|
|
>
|
|
{nodes
|
|
.filter((node) => !node.isHidden)
|
|
.map((node) => (
|
|
<MiniMapNode
|
|
key={node.id}
|
|
x={node.__rf.position.x}
|
|
y={node.__rf.position.y}
|
|
width={node.__rf.width}
|
|
height={node.__rf.height}
|
|
style={node.style}
|
|
color={nodeColorFunc(node)}
|
|
borderRadius={nodeBorderRadius}
|
|
/>
|
|
))}
|
|
<path
|
|
className="react-flow__minimap-mask"
|
|
d={`M${x - offset},${y - offset}h${width + offset * 2}v${height + offset * 2}h${-width - offset * 2}z
|
|
M${viewBB.x},${viewBB.y}h${viewBB.width}v${viewBB.height}h${-viewBB.width}z`}
|
|
fill={maskColor}
|
|
fillRule="evenodd"
|
|
/>
|
|
</svg>
|
|
);
|
|
};
|
|
|
|
MiniMap.displayName = 'MiniMap';
|
|
|
|
export default memo(MiniMap);
|