maxGraph/packages/html/stories/Constituent.stories.js

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({});