mirror of
https://github.com/projectstorm/react-diagrams.git
synced 2025-08-20 20:31:16 +08:00
115 lines
3.0 KiB
TypeScript
115 lines
3.0 KiB
TypeScript
import {
|
|
DiagramEngine,
|
|
DefaultNodeFactory,
|
|
DefaultLinkFactory,
|
|
DiagramModel,
|
|
DefaultNodeModel,
|
|
LinkModel,
|
|
NodeModel,
|
|
DefaultPortModel,
|
|
DiagramWidget,
|
|
DefaultNodeInstanceFactory,
|
|
DefaultPortInstanceFactory,
|
|
LinkInstanceFactory
|
|
} from "../../src/main";
|
|
import * as React from "react";
|
|
|
|
/**
|
|
* Tests the grid size
|
|
*/
|
|
class NodeDelayedPosition extends React.Component<any, any> {
|
|
constructor(props) {
|
|
super(props);
|
|
this.cloneSelected = this.cloneSelected.bind(this)
|
|
}
|
|
|
|
updatePosition() {
|
|
const { engine } = this.props;
|
|
let model = engine.getDiagramModel();
|
|
const nodes = model.getNodes();
|
|
let node = nodes[Object.keys(nodes)[0]];
|
|
node.setPosition(node.x + 30, node.y + 30);
|
|
this.forceUpdate();
|
|
}
|
|
|
|
cloneSelected() {
|
|
const {engine} = this.props;
|
|
const offset = {x:100,y:100};
|
|
const model = engine.getDiagramModel();
|
|
const originalItems = model.getSelectedItems();
|
|
const selectedItems = originalItems.reduce((res,i) => {
|
|
if(i instanceof NodeModel) res.nodes.push(i);
|
|
else if(i instanceof LinkModel) res.links.push(i);
|
|
return res;
|
|
},{nodes:[],links:[]});
|
|
let lookupTable = {};
|
|
selectedItems.nodes.forEach(i => {
|
|
|
|
let node = i.clone(lookupTable);
|
|
model.addNode(node);
|
|
node.setPosition(node.x+offset.x,node.y+offset.y)
|
|
});
|
|
selectedItems.links.forEach(i => {
|
|
let link = i.clone(lookupTable);
|
|
link.getPoints().forEach(p => p.updateLocation({x:p.getX()+offset.x,y:p.getY()+offset.y}));
|
|
model.addLink(link);
|
|
});
|
|
originalItems.forEach(i => i.selected = false);
|
|
this.forceUpdate();
|
|
}
|
|
|
|
|
|
render() {
|
|
const { engine } = this.props;
|
|
return (
|
|
<div>
|
|
<DiagramWidget diagramEngine={engine} allowLooseLinks={false} />
|
|
<button onClick={this.cloneSelected}>Clone Selected</button>
|
|
</div>
|
|
);
|
|
}
|
|
}
|
|
|
|
export default () => {
|
|
//1) setup the diagram engine
|
|
var engine = new DiagramEngine();
|
|
engine.registerNodeFactory(new DefaultNodeFactory());
|
|
engine.registerLinkFactory(new DefaultLinkFactory());
|
|
|
|
//2) setup the diagram model
|
|
var model = new DiagramModel();
|
|
|
|
//3-A) create a default node
|
|
var node1 = new DefaultNodeModel("Node 1", "rgb(0,192,255)");
|
|
var port1 = node1.addPort(new DefaultPortModel(false, "out-1", "Out"));
|
|
node1.x = 100;
|
|
node1.y = 100;
|
|
|
|
//3-B) create another default node
|
|
var node2 = new DefaultNodeModel("Node 2", "rgb(192,255,0)");
|
|
var port2 = node2.addPort(new DefaultPortModel(true, "in-1", "IN"));
|
|
node2.x = 400;
|
|
node2.y = 100;
|
|
|
|
//3-C) link the 2 nodes together
|
|
var link1 = new LinkModel();
|
|
link1.setSourcePort(port1);
|
|
link1.setTargetPort(port2);
|
|
|
|
//4) add the models to the root graph
|
|
model.addNode(node1);
|
|
model.addNode(node2);
|
|
model.addLink(link1);
|
|
|
|
//5) load model into engine
|
|
engine.setDiagramModel(model);
|
|
|
|
//we need this to help the system know what models to create form the JSON
|
|
engine.registerInstanceFactory(new DefaultNodeInstanceFactory());
|
|
engine.registerInstanceFactory(new DefaultPortInstanceFactory());
|
|
engine.registerInstanceFactory(new LinkInstanceFactory());
|
|
|
|
//6) render the diagram!
|
|
return <NodeDelayedPosition engine={engine} model={model} />;
|
|
};
|