mirror of
https://gitee.com/antv/g6.git
synced 2024-12-05 13:18:40 +08:00
859 lines
16 KiB
TypeScript
859 lines
16 KiB
TypeScript
import G6, { Graph, TreeGraph } from '../../src';
|
|
|
|
const div = document.createElement('div');
|
|
div.id = 'container';
|
|
document.body.appendChild(div);
|
|
|
|
describe('edge click state', () => {
|
|
it('edge ', () => {
|
|
G6.registerBehavior('active-edge', {
|
|
getEvents() {
|
|
return {
|
|
'edge:click': 'onEdgeSelect',
|
|
'edge:mouseenter': 'onEdgeHover',
|
|
'edge:mouseleave': 'onEdgeLeave',
|
|
'canvas:mousedown': 'clearSelectedEdge',
|
|
};
|
|
},
|
|
onEdgeSelect(evt) {
|
|
const item = evt.item;
|
|
if (item.hasState('select')) {
|
|
this.graph.setItemState(item, 'select', false);
|
|
return;
|
|
}
|
|
this.graph.setItemState(item, 'select', true);
|
|
},
|
|
onEdgeHover(evt) {
|
|
const item = evt.item;
|
|
if (item.hasState('select')) {
|
|
return false;
|
|
}
|
|
this.graph.setItemState(item, 'hover', true);
|
|
},
|
|
onEdgeLeave(evt) {
|
|
const item = evt.item;
|
|
this.graph.setItemState(item, 'hover', false);
|
|
},
|
|
clearSelectedEdge(evt) {
|
|
const edges = this.graph.findAllByState('edge', 'select');
|
|
edges.forEach((edge) => {
|
|
this.graph.setItemState(edge, 'select', false);
|
|
});
|
|
},
|
|
});
|
|
|
|
const graph = new G6.Graph({
|
|
container: 'container',
|
|
width: 500,
|
|
height: 500,
|
|
defaultEdge: {
|
|
type: 'line',
|
|
style: {
|
|
stroke: '#24A0FF',
|
|
radius: 4,
|
|
offset: 15,
|
|
lineWidth: 2,
|
|
lineAppendWidth: 4,
|
|
cursor: 'pointer',
|
|
endArrow: {
|
|
path: 'M 0,0 L 7,3 L 5,0 L 7,-3 Z',
|
|
fill: '#24A0FF',
|
|
},
|
|
},
|
|
},
|
|
edgeStateStyles: {
|
|
hover: {
|
|
stroke: '#0ff',
|
|
endArrow: {
|
|
fill: '#FF7868',
|
|
},
|
|
},
|
|
select: {
|
|
stroke: '#FF7868',
|
|
endArrow: {
|
|
fill: '#FF7868',
|
|
},
|
|
},
|
|
},
|
|
modes: {
|
|
default: ['drag-node', 'active-edge'],
|
|
},
|
|
});
|
|
|
|
graph.read({
|
|
nodes: [
|
|
{ id: '1', x: 100, y: 100 },
|
|
{ id: '2', x: 100, y: 220 },
|
|
],
|
|
edges: [{ source: '1', target: '2' }],
|
|
});
|
|
});
|
|
});
|
|
|
|
describe('dragenter dragleave', () => {
|
|
const data = {
|
|
nodes: [
|
|
{
|
|
id: 'node1',
|
|
x: 100,
|
|
y: 100,
|
|
},
|
|
{
|
|
id: 'node2',
|
|
x: 200,
|
|
y: 100,
|
|
},
|
|
],
|
|
edges: [],
|
|
};
|
|
const graph = new Graph({
|
|
container: 'container',
|
|
width: 500,
|
|
height: 500,
|
|
modes: {
|
|
default: ['drag-node'],
|
|
},
|
|
});
|
|
graph.data(data);
|
|
graph.render();
|
|
it('dragenter', () => {
|
|
graph.on('node:dragenter', (e) => {
|
|
console.log('dragenter');
|
|
});
|
|
graph.on('node:dragleave', (e) => {
|
|
console.log('dragleave');
|
|
});
|
|
graph.on('node:mouseenter', (e) => {
|
|
console.log('mouseenter');
|
|
});
|
|
graph.on('node:mouseleave', (e) => {
|
|
console.log('mouseleave');
|
|
});
|
|
graph.on('node:dragover', (e) => {
|
|
console.log('dragover');
|
|
});
|
|
});
|
|
graph.destroy();
|
|
});
|
|
|
|
// closes: #1026
|
|
describe('empty data array + fitview', () => {
|
|
const data = {
|
|
nodes: [],
|
|
edges: [],
|
|
};
|
|
const graph = new Graph({
|
|
container: 'container',
|
|
width: 500,
|
|
height: 500,
|
|
fitView: true,
|
|
});
|
|
graph.data(data);
|
|
it('empty data array + fitview', () => {
|
|
graph.render();
|
|
graph.destroy();
|
|
});
|
|
});
|
|
|
|
// closes: #1301
|
|
describe('change data with rect node', () => {
|
|
const data = {
|
|
nodes: [
|
|
{
|
|
name: 'source',
|
|
id: 'source',
|
|
label: 'source',
|
|
type: 'rect',
|
|
x: 100,
|
|
y: 100,
|
|
//size: [60, 60],
|
|
style: {
|
|
width: 60,
|
|
height: 20,
|
|
},
|
|
},
|
|
],
|
|
edges: [],
|
|
};
|
|
const graph = new Graph({
|
|
container: 'container',
|
|
width: 500,
|
|
height: 500,
|
|
});
|
|
graph.data(data);
|
|
it('change data', () => {
|
|
graph.render();
|
|
graph.changeData(data);
|
|
graph.destroy();
|
|
});
|
|
});
|
|
|
|
describe('cubic with layout', () => {
|
|
const data = {
|
|
nodes: [
|
|
{
|
|
id: '1',
|
|
label: '冲压',
|
|
degree: 1,
|
|
// x: 0,
|
|
// y: 0
|
|
// x: 100,
|
|
// y: 300
|
|
},
|
|
{
|
|
id: '2',
|
|
label: '电镀',
|
|
degree: 2,
|
|
// x: 0,
|
|
// y: 0
|
|
// x: 600,
|
|
// y: 300
|
|
},
|
|
],
|
|
edges: [
|
|
{
|
|
source: '1',
|
|
target: '2',
|
|
},
|
|
],
|
|
};
|
|
const dashArray = [
|
|
[0, 1],
|
|
[0, 2],
|
|
[1, 2],
|
|
[0, 1, 1, 2],
|
|
[0, 2, 1, 2],
|
|
[1, 2, 1, 2],
|
|
[2, 2, 1, 2],
|
|
[3, 2, 1, 2],
|
|
[4, 2, 1, 2],
|
|
];
|
|
|
|
const lineDash = [4, 2, 1, 2];
|
|
const interval = 9; // lineDash 的和
|
|
G6.registerEdge(
|
|
'line-dash',
|
|
{
|
|
afterDraw(cfg, group) {
|
|
// 获得该边的第一个图形,这里是边的 path
|
|
const shape = group.get('children')[0];
|
|
// 获得边的 path 的总长度
|
|
const length = shape.getTotalLength();
|
|
|
|
let totalArray = [];
|
|
// 计算出整条线的 lineDash
|
|
for (let i = 0; i < length; i += interval) {
|
|
totalArray = totalArray.concat(lineDash);
|
|
}
|
|
|
|
let index = 0;
|
|
// 边 path 图形的动画
|
|
shape.animate(
|
|
(ratio) => {
|
|
const cfg = {
|
|
lineDash: dashArray[index].concat(totalArray),
|
|
};
|
|
// 每次移动 1
|
|
index = (index + 1) % interval;
|
|
// 返回需要修改的参数集,这里只修改了 lineDash
|
|
return cfg;
|
|
},
|
|
{
|
|
repeat: true,
|
|
duration: 3000,
|
|
},
|
|
); // 一次动画的时长为 3000
|
|
},
|
|
},
|
|
'cubic',
|
|
); // 该自定义边继承了内置三阶贝塞尔曲线边 cubic
|
|
const graph = new Graph({
|
|
container: 'container',
|
|
width: 500,
|
|
height: 500,
|
|
//animate: true,
|
|
defaultEdge: {
|
|
size: 2,
|
|
type: 'line-dash', //'line',//'line-dash',//
|
|
color: 'rgb(150, 150, 150)',
|
|
},
|
|
layout: {
|
|
type: 'dagre',
|
|
rankdir: 'LR',
|
|
nodesep: 30,
|
|
//controlPoints: false
|
|
// ranksep: 100
|
|
},
|
|
});
|
|
it('change data', () => {
|
|
graph.data(data);
|
|
graph.render();
|
|
graph.destroy();
|
|
});
|
|
});
|
|
|
|
describe('changdata states', () => {
|
|
it('changeData', () => {
|
|
const data = {
|
|
nodes: [
|
|
{
|
|
id: '0',
|
|
label: '0',
|
|
},
|
|
{
|
|
id: '1',
|
|
label: '1',
|
|
},
|
|
{
|
|
id: '2',
|
|
label: '2',
|
|
},
|
|
{
|
|
id: '3',
|
|
label: '3',
|
|
},
|
|
{
|
|
id: '4',
|
|
label: '4',
|
|
},
|
|
{
|
|
id: '5',
|
|
label: '5',
|
|
},
|
|
{
|
|
id: '6',
|
|
label: '6',
|
|
},
|
|
{
|
|
id: '7',
|
|
label: '7',
|
|
},
|
|
{
|
|
id: '8',
|
|
label: '8',
|
|
},
|
|
{
|
|
id: '9',
|
|
label: '9',
|
|
},
|
|
{
|
|
id: '10',
|
|
label: '10',
|
|
},
|
|
{
|
|
id: '11',
|
|
label: '11',
|
|
},
|
|
{
|
|
id: '12',
|
|
label: '12',
|
|
},
|
|
{
|
|
id: '13',
|
|
label: '13',
|
|
},
|
|
{
|
|
id: '14',
|
|
label: '14',
|
|
},
|
|
{
|
|
id: '15',
|
|
label: '15',
|
|
},
|
|
{
|
|
id: '16',
|
|
label: '16',
|
|
},
|
|
{
|
|
id: '17',
|
|
label: '17',
|
|
},
|
|
{
|
|
id: '18',
|
|
label: '18',
|
|
},
|
|
{
|
|
id: '19',
|
|
label: '19',
|
|
},
|
|
{
|
|
id: '20',
|
|
label: '20',
|
|
},
|
|
{
|
|
id: '21',
|
|
label: '21',
|
|
},
|
|
{
|
|
id: '22',
|
|
label: '22',
|
|
},
|
|
{
|
|
id: '23',
|
|
label: '23',
|
|
},
|
|
{
|
|
id: '24',
|
|
label: '24',
|
|
},
|
|
{
|
|
id: '25',
|
|
label: '25',
|
|
},
|
|
{
|
|
id: '26',
|
|
label: '26',
|
|
},
|
|
{
|
|
id: '27',
|
|
label: '27',
|
|
},
|
|
{
|
|
id: '28',
|
|
label: '28',
|
|
},
|
|
{
|
|
id: '29',
|
|
label: '29',
|
|
},
|
|
{
|
|
id: '30',
|
|
label: '30',
|
|
},
|
|
{
|
|
id: '31',
|
|
label: '31',
|
|
},
|
|
{
|
|
id: '32',
|
|
label: '32',
|
|
},
|
|
{
|
|
id: '33',
|
|
label: '33',
|
|
},
|
|
],
|
|
edges: [
|
|
{
|
|
source: '0',
|
|
target: '1',
|
|
},
|
|
{
|
|
source: '0',
|
|
target: '2',
|
|
},
|
|
{
|
|
source: '0',
|
|
target: '3',
|
|
},
|
|
{
|
|
source: '0',
|
|
target: '4',
|
|
},
|
|
{
|
|
source: '0',
|
|
target: '5',
|
|
},
|
|
{
|
|
source: '0',
|
|
target: '7',
|
|
},
|
|
{
|
|
source: '0',
|
|
target: '8',
|
|
},
|
|
{
|
|
source: '0',
|
|
target: '9',
|
|
},
|
|
{
|
|
source: '0',
|
|
target: '10',
|
|
},
|
|
{
|
|
source: '0',
|
|
target: '11',
|
|
},
|
|
{
|
|
source: '0',
|
|
target: '13',
|
|
},
|
|
{
|
|
source: '0',
|
|
target: '14',
|
|
},
|
|
{
|
|
source: '0',
|
|
target: '15',
|
|
},
|
|
{
|
|
source: '0',
|
|
target: '16',
|
|
},
|
|
{
|
|
source: '2',
|
|
target: '3',
|
|
},
|
|
{
|
|
source: '4',
|
|
target: '5',
|
|
},
|
|
{
|
|
source: '4',
|
|
target: '6',
|
|
},
|
|
{
|
|
source: '5',
|
|
target: '6',
|
|
},
|
|
{
|
|
source: '7',
|
|
target: '13',
|
|
},
|
|
{
|
|
source: '8',
|
|
target: '14',
|
|
},
|
|
{
|
|
source: '9',
|
|
target: '10',
|
|
},
|
|
{
|
|
source: '10',
|
|
target: '22',
|
|
},
|
|
{
|
|
source: '10',
|
|
target: '14',
|
|
},
|
|
{
|
|
source: '10',
|
|
target: '12',
|
|
},
|
|
{
|
|
source: '10',
|
|
target: '24',
|
|
},
|
|
{
|
|
source: '10',
|
|
target: '21',
|
|
},
|
|
{
|
|
source: '10',
|
|
target: '20',
|
|
},
|
|
{
|
|
source: '11',
|
|
target: '24',
|
|
},
|
|
{
|
|
source: '11',
|
|
target: '22',
|
|
},
|
|
{
|
|
source: '11',
|
|
target: '14',
|
|
},
|
|
{
|
|
source: '12',
|
|
target: '13',
|
|
},
|
|
{
|
|
source: '16',
|
|
target: '17',
|
|
},
|
|
{
|
|
source: '16',
|
|
target: '18',
|
|
},
|
|
{
|
|
source: '16',
|
|
target: '21',
|
|
},
|
|
{
|
|
source: '16',
|
|
target: '22',
|
|
},
|
|
{
|
|
source: '17',
|
|
target: '18',
|
|
},
|
|
{
|
|
source: '17',
|
|
target: '20',
|
|
},
|
|
{
|
|
source: '18',
|
|
target: '19',
|
|
},
|
|
{
|
|
source: '19',
|
|
target: '20',
|
|
},
|
|
{
|
|
source: '19',
|
|
target: '33',
|
|
},
|
|
{
|
|
source: '19',
|
|
target: '22',
|
|
},
|
|
{
|
|
source: '19',
|
|
target: '23',
|
|
},
|
|
{
|
|
source: '20',
|
|
target: '21',
|
|
},
|
|
{
|
|
source: '21',
|
|
target: '22',
|
|
},
|
|
{
|
|
source: '22',
|
|
target: '24',
|
|
},
|
|
{
|
|
source: '22',
|
|
target: '25',
|
|
},
|
|
{
|
|
source: '22',
|
|
target: '26',
|
|
},
|
|
{
|
|
source: '22',
|
|
target: '23',
|
|
},
|
|
{
|
|
source: '22',
|
|
target: '28',
|
|
},
|
|
{
|
|
source: '22',
|
|
target: '30',
|
|
},
|
|
{
|
|
source: '22',
|
|
target: '31',
|
|
},
|
|
{
|
|
source: '22',
|
|
target: '32',
|
|
},
|
|
{
|
|
source: '22',
|
|
target: '33',
|
|
},
|
|
{
|
|
source: '23',
|
|
target: '28',
|
|
},
|
|
{
|
|
source: '23',
|
|
target: '27',
|
|
},
|
|
{
|
|
source: '23',
|
|
target: '29',
|
|
},
|
|
{
|
|
source: '23',
|
|
target: '30',
|
|
},
|
|
{
|
|
source: '23',
|
|
target: '31',
|
|
},
|
|
{
|
|
source: '23',
|
|
target: '33',
|
|
},
|
|
{
|
|
source: '32',
|
|
target: '33',
|
|
},
|
|
],
|
|
};
|
|
|
|
const width = 500;
|
|
const height = 500;
|
|
const graph = new G6.Graph({
|
|
container: 'container',
|
|
width,
|
|
height,
|
|
modes: {
|
|
default: ['activate-relations', 'drag-canvas', 'drag-node'],
|
|
},
|
|
layout: {
|
|
type: 'circular',
|
|
},
|
|
nodeStateStyles: {
|
|
active: {
|
|
stroke: 'red',
|
|
},
|
|
},
|
|
edgeStateStyles: {
|
|
active: {
|
|
stroke: '#000',
|
|
opacity: 1,
|
|
},
|
|
inactive: {
|
|
color: '#969696',
|
|
opacity: 0.5,
|
|
},
|
|
},
|
|
animate: true,
|
|
defaultNode: {
|
|
size: 20,
|
|
style: {
|
|
lineWidth: 2,
|
|
fill: 'rgb(239, 244, 255)',
|
|
stroke: 'rgb(95, 149, 255)',
|
|
},
|
|
},
|
|
defaultEdge: {
|
|
size: 1,
|
|
color: 'rgb(150, 150, 150)',
|
|
style: {
|
|
endArrow: {
|
|
path: 'M 0,0 L 8,4 L 8,-4 Z',
|
|
fill: 'rgb(150, 150, 150)',
|
|
},
|
|
},
|
|
},
|
|
});
|
|
graph.data(data);
|
|
graph.render();
|
|
|
|
const node1 = graph.findById('1');
|
|
graph.setItemState(node1, 'active', true);
|
|
const edge1 = graph.getEdges()[0];
|
|
graph.setItemState(edge1, 'active', true);
|
|
|
|
expect(graph.findAllByState('node', 'active').length).toBe(1);
|
|
expect(graph.findAllByState('edge', 'active').length).toBe(1);
|
|
|
|
graph.changeData(data);
|
|
expect(graph.findAllByState('node', 'active').length).toBe(0);
|
|
expect(graph.findAllByState('edge', 'active').length).toBe(0);
|
|
|
|
graph.destroy();
|
|
});
|
|
});
|
|
|
|
describe('defaultStyle states', () => {
|
|
it('label', () => {
|
|
const data = {
|
|
nodes: [
|
|
{
|
|
id: 'node1',
|
|
label: 'node1',
|
|
x: 100,
|
|
y: 100,
|
|
},
|
|
],
|
|
};
|
|
|
|
const graph = new G6.Graph({
|
|
container: 'container',
|
|
width: 500,
|
|
height: 500,
|
|
defaultEdge: {
|
|
color: 'rgb(150, 150, 150)',
|
|
lineAppendWidth: 3,
|
|
},
|
|
defaultNode: {
|
|
style: {
|
|
fill: '#DEE9FF',
|
|
stroke: 'rgb(95, 149, 255)',
|
|
},
|
|
},
|
|
nodeStateStyles: {
|
|
hover: {
|
|
lineWidth: 5,
|
|
fillOpacity: 1,
|
|
'text-shape': {
|
|
fontSize: 20,
|
|
fill: '#003a8c',
|
|
},
|
|
},
|
|
},
|
|
edgeStateStyles: {
|
|
hover: {
|
|
lineWidth: 3,
|
|
},
|
|
},
|
|
});
|
|
graph.data(data);
|
|
graph.render();
|
|
|
|
graph.on('node:mouseenter', function (evt) {
|
|
const node = evt.item;
|
|
const model = node.getModel();
|
|
model.oriLabel = model.label;
|
|
graph.setItemState(node, 'hover', true);
|
|
graph.updateItem(node, {
|
|
label: 'hover 后 ' + model.id,
|
|
labelCfg: {
|
|
style: {
|
|
fill: 'blue',
|
|
},
|
|
},
|
|
});
|
|
});
|
|
|
|
graph.on('node:mouseleave', function (evt) {
|
|
const node = evt.item;
|
|
const model = node.getModel();
|
|
|
|
graph.setItemState(node, 'hover', false);
|
|
|
|
graph.updateItem(node, {
|
|
label: model.oriLabel,
|
|
style: {
|
|
'text-shape': {
|
|
fill: 'red',
|
|
},
|
|
},
|
|
labelCfg: {
|
|
style: {
|
|
fill: 'red',
|
|
},
|
|
},
|
|
});
|
|
});
|
|
|
|
graph.on('edge:mouseenter', function (evt) {
|
|
const edge = evt.item;
|
|
const model = edge.getModel();
|
|
model.oriLabel = model.label;
|
|
graph.setItemState(edge, 'hover', true);
|
|
graph.updateItem(edge, {
|
|
label: 'hover 后',
|
|
labelCfg: {
|
|
style: {
|
|
fill: '#003a8c',
|
|
},
|
|
},
|
|
});
|
|
});
|
|
|
|
graph.on('edge:mouseleave', function (evt) {
|
|
const edge = evt.item;
|
|
graph.setItemState(edge, 'hover', false);
|
|
graph.updateItem(edge, {
|
|
label: 'hover 前的边文本',
|
|
labelCfg: {
|
|
style: {
|
|
fill: '#555',
|
|
},
|
|
},
|
|
});
|
|
});
|
|
|
|
// graph.destroy()
|
|
});
|
|
});
|