-
Notifications
You must be signed in to change notification settings - Fork 45
/
LayoutEngine.ts
52 lines (48 loc) · 1.99 KB
/
LayoutEngine.ts
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
/*******************************************************************************
* Copyright (c) 2023 Obeo.
* This program and the accompanying materials
* are made available under the terms of the Eclipse Public License v2.0
* which accompanies this distribution, and is available at
* https://www.eclipse.org/legal/epl-2.0/
*
* SPDX-License-Identifier: EPL-2.0
*
* Contributors:
* Obeo - initial API and implementation
*******************************************************************************/
import { Node } from 'reactflow';
import { Diagram, NodeData } from '../DiagramRenderer.types';
import { DiagramNodeType } from '../node/NodeTypes.types';
import { IconLabelNodeLayoutHandler } from './IconLabelNodeLayoutHandler';
import { ImageNodeLayoutHandler } from './ImageNodeLayoutHandler';
import { ILayoutEngine, INodeLayoutHandler } from './LayoutEngine.types';
import { ListNodeLayoutHandler } from './ListNodeLayoutHandler';
import { RectangleNodeLayoutHandler } from './RectangleNodeLayoutHandler';
export class LayoutEngine implements ILayoutEngine {
nodeLayoutHandlers: INodeLayoutHandler<NodeData>[] = [
new RectangleNodeLayoutHandler(),
new ListNodeLayoutHandler(),
new ImageNodeLayoutHandler(),
new IconLabelNodeLayoutHandler(),
];
public layoutNodes(
previousDiagram: Diagram | null,
visibleNodes: Node<NodeData, DiagramNodeType>[],
nodesToLayout: Node<NodeData, DiagramNodeType>[]
) {
nodesToLayout.forEach((node) => {
const nodeLayoutHandler: INodeLayoutHandler<NodeData> | undefined = this.nodeLayoutHandlers.find((handler) =>
handler.canHandle(node)
);
if (nodeLayoutHandler) {
const directChildren = visibleNodes.filter((visibleNode) => visibleNode.parentNode === node.id);
nodeLayoutHandler.handle(this, previousDiagram, node, visibleNodes, directChildren);
node.style = {
...node.style,
width: `${node.width}px`,
height: `${node.height}px`,
};
}
});
}
}