104 lines
2.4 KiB
JavaScript
104 lines
2.4 KiB
JavaScript
|
import mxgraph from '@mxgraph/core';
|
||
|
|
||
|
import { globalTypes } from '../.storybook/preview';
|
||
|
|
||
|
export default {
|
||
|
title: 'Layouts/Constituent',
|
||
|
argTypes: {
|
||
|
...globalTypes,
|
||
|
contextMenu: {
|
||
|
type: 'boolean',
|
||
|
defaultValue: false
|
||
|
},
|
||
|
rubberBand: {
|
||
|
type: 'boolean',
|
||
|
defaultValue: true
|
||
|
}
|
||
|
}
|
||
|
};
|
||
|
|
||
|
const Template = ({ label, ...args }) => {
|
||
|
const {
|
||
|
mxGraph,
|
||
|
mxEvent,
|
||
|
mxGraphHandler,
|
||
|
mxRubberband
|
||
|
} = mxgraph;
|
||
|
|
||
|
const container = document.createElement('div');
|
||
|
container.style.position = 'relative';
|
||
|
container.style.overflow = 'hidden';
|
||
|
container.style.width = `${args.width}px`;
|
||
|
container.style.height = `${args.height}px`;
|
||
|
container.style.background = 'url(/images/grid.gif)';
|
||
|
container.style.cursor = 'default';
|
||
|
|
||
|
// Disables the built-in context menu
|
||
|
mxEvent.disableContextMenu(container);
|
||
|
|
||
|
class MyCustomGraphHandler extends mxGraphHandler {
|
||
|
/**
|
||
|
* Redirects start drag to parent.
|
||
|
*/
|
||
|
getInitialCellForEvent(me) {
|
||
|
let cell = super.getInitialCellForEvent(me);
|
||
|
if (this.graph.isPart(cell)) {
|
||
|
cell = this.graph.getModel().getParent(cell);
|
||
|
}
|
||
|
return cell;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
class MyCustomGraph extends mxGraph {
|
||
|
foldingEnabled = false;
|
||
|
|
||
|
recursiveResize = true;
|
||
|
|
||
|
isPart(cell) {
|
||
|
// Helper method to mark parts with constituent=1 in the style
|
||
|
return this.getCurrentCellStyle(cell).constituent == '1';
|
||
|
}
|
||
|
|
||
|
selectCellForEvent(cell, evt) {
|
||
|
// Redirects selection to parent
|
||
|
if (this.isPart(cell)) {
|
||
|
cell = this.model.getParent(cell);
|
||
|
}
|
||
|
super.selectCellForEvent(cell, evt);
|
||
|
}
|
||
|
|
||
|
createGraphHandler() {
|
||
|
return new MyCustomGraphHandler(this);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
// Creates the graph inside the given container
|
||
|
const graph = new MyCustomGraph(container);
|
||
|
|
||
|
// Enables rubberband selection
|
||
|
new mxRubberband(graph);
|
||
|
|
||
|
// Gets the default parent for inserting new cells. This
|
||
|
// is normally the first child of the root (ie. layer 0).
|
||
|
const parent = graph.getDefaultParent();
|
||
|
|
||
|
// Adds cells to the model in a single step
|
||
|
graph.batchUpdate(() => {
|
||
|
const v1 = graph.insertVertex({
|
||
|
parent,
|
||
|
position: [20, 20],
|
||
|
size: [120, 70],
|
||
|
});
|
||
|
const v2 = graph.insertVertex({
|
||
|
parent: v1,
|
||
|
value: 'Constituent',
|
||
|
position: [20, 20],
|
||
|
size: [80, 30],
|
||
|
style: 'constituent=1;',
|
||
|
});
|
||
|
});
|
||
|
|
||
|
return container;
|
||
|
}
|
||
|
|
||
|
export const Default = Template.bind({});
|