|
/*
|
* Licensed to the Apache Software Foundation (ASF) under one
|
* or more contributor license agreements. See the NOTICE file
|
* distributed with this work for additional information
|
* regarding copyright ownership. The ASF licenses this file
|
* to you under the Apache License, Version 2.0 (the
|
* "License"); you may not use this file except in compliance
|
* with the License. You may obtain a copy of the License at
|
*
|
* http://www.apache.org/licenses/LICENSE-2.0
|
*
|
* Unless required by applicable law or agreed to in writing,
|
* software distributed under the License is distributed on an
|
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
|
* KIND, either express or implied. See the License for the
|
* specific language governing permissions and limitations
|
* under the License.
|
*/
|
|
|
/**
|
* AUTO-GENERATED FILE. DO NOT MODIFY.
|
*/
|
|
/*
|
* Licensed to the Apache Software Foundation (ASF) under one
|
* or more contributor license agreements. See the NOTICE file
|
* distributed with this work for additional information
|
* regarding copyright ownership. The ASF licenses this file
|
* to you under the Apache License, Version 2.0 (the
|
* "License"); you may not use this file except in compliance
|
* with the License. You may obtain a copy of the License at
|
*
|
* http://www.apache.org/licenses/LICENSE-2.0
|
*
|
* Unless required by applicable law or agreed to in writing,
|
* software distributed under the License is distributed on an
|
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
|
* KIND, either express or implied. See the License for the
|
* specific language governing permissions and limitations
|
* under the License.
|
*/
|
import { forceLayout } from './forceHelper.js';
|
import { simpleLayout } from './simpleLayoutHelper.js';
|
import { circularLayout } from './circularLayoutHelper.js';
|
import { linearMap } from '../../util/number.js';
|
import * as vec2 from 'zrender/lib/core/vector.js';
|
import * as zrUtil from 'zrender/lib/core/util.js';
|
import { getCurvenessForEdge } from '../helper/multipleGraphEdgeHelper.js';
|
export default function graphForceLayout(ecModel) {
|
ecModel.eachSeriesByType('graph', function (graphSeries) {
|
var coordSys = graphSeries.coordinateSystem;
|
if (coordSys && coordSys.type !== 'view') {
|
return;
|
}
|
if (graphSeries.get('layout') === 'force') {
|
var preservedPoints_1 = graphSeries.preservedPoints || {};
|
var graph_1 = graphSeries.getGraph();
|
var nodeData_1 = graph_1.data;
|
var edgeData = graph_1.edgeData;
|
var forceModel = graphSeries.getModel('force');
|
var initLayout = forceModel.get('initLayout');
|
if (graphSeries.preservedPoints) {
|
nodeData_1.each(function (idx) {
|
var id = nodeData_1.getId(idx);
|
nodeData_1.setItemLayout(idx, preservedPoints_1[id] || [NaN, NaN]);
|
});
|
} else if (!initLayout || initLayout === 'none') {
|
simpleLayout(graphSeries);
|
} else if (initLayout === 'circular') {
|
circularLayout(graphSeries, 'value');
|
}
|
var nodeDataExtent_1 = nodeData_1.getDataExtent('value');
|
var edgeDataExtent_1 = edgeData.getDataExtent('value');
|
// let edgeDataExtent = edgeData.getDataExtent('value');
|
var repulsion = forceModel.get('repulsion');
|
var edgeLength = forceModel.get('edgeLength');
|
var repulsionArr_1 = zrUtil.isArray(repulsion) ? repulsion : [repulsion, repulsion];
|
var edgeLengthArr_1 = zrUtil.isArray(edgeLength) ? edgeLength : [edgeLength, edgeLength];
|
// Larger value has smaller length
|
edgeLengthArr_1 = [edgeLengthArr_1[1], edgeLengthArr_1[0]];
|
var nodes_1 = nodeData_1.mapArray('value', function (value, idx) {
|
var point = nodeData_1.getItemLayout(idx);
|
var rep = linearMap(value, nodeDataExtent_1, repulsionArr_1);
|
if (isNaN(rep)) {
|
rep = (repulsionArr_1[0] + repulsionArr_1[1]) / 2;
|
}
|
return {
|
w: rep,
|
rep: rep,
|
fixed: nodeData_1.getItemModel(idx).get('fixed'),
|
p: !point || isNaN(point[0]) || isNaN(point[1]) ? null : point
|
};
|
});
|
var edges = edgeData.mapArray('value', function (value, idx) {
|
var edge = graph_1.getEdgeByIndex(idx);
|
var d = linearMap(value, edgeDataExtent_1, edgeLengthArr_1);
|
if (isNaN(d)) {
|
d = (edgeLengthArr_1[0] + edgeLengthArr_1[1]) / 2;
|
}
|
var edgeModel = edge.getModel();
|
var curveness = zrUtil.retrieve3(edge.getModel().get(['lineStyle', 'curveness']), -getCurvenessForEdge(edge, graphSeries, idx, true), 0);
|
return {
|
n1: nodes_1[edge.node1.dataIndex],
|
n2: nodes_1[edge.node2.dataIndex],
|
d: d,
|
curveness: curveness,
|
ignoreForceLayout: edgeModel.get('ignoreForceLayout')
|
};
|
});
|
// let coordSys = graphSeries.coordinateSystem;
|
var rect = coordSys.getBoundingRect();
|
var forceInstance = forceLayout(nodes_1, edges, {
|
rect: rect,
|
gravity: forceModel.get('gravity'),
|
friction: forceModel.get('friction')
|
});
|
forceInstance.beforeStep(function (nodes, edges) {
|
for (var i = 0, l = nodes.length; i < l; i++) {
|
if (nodes[i].fixed) {
|
// Write back to layout instance
|
vec2.copy(nodes[i].p, graph_1.getNodeByIndex(i).getLayout());
|
}
|
}
|
});
|
forceInstance.afterStep(function (nodes, edges, stopped) {
|
for (var i = 0, l = nodes.length; i < l; i++) {
|
if (!nodes[i].fixed) {
|
graph_1.getNodeByIndex(i).setLayout(nodes[i].p);
|
}
|
preservedPoints_1[nodeData_1.getId(i)] = nodes[i].p;
|
}
|
for (var i = 0, l = edges.length; i < l; i++) {
|
var e = edges[i];
|
var edge = graph_1.getEdgeByIndex(i);
|
var p1 = e.n1.p;
|
var p2 = e.n2.p;
|
var points = edge.getLayout();
|
points = points ? points.slice() : [];
|
points[0] = points[0] || [];
|
points[1] = points[1] || [];
|
vec2.copy(points[0], p1);
|
vec2.copy(points[1], p2);
|
if (+e.curveness) {
|
points[2] = [(p1[0] + p2[0]) / 2 - (p1[1] - p2[1]) * e.curveness, (p1[1] + p2[1]) / 2 - (p2[0] - p1[0]) * e.curveness];
|
}
|
edge.setLayout(points);
|
}
|
});
|
graphSeries.forceLayout = forceInstance;
|
graphSeries.preservedPoints = preservedPoints_1;
|
// Step to get the layout
|
forceInstance.step();
|
} else {
|
// Remove prev injected forceLayout instance
|
graphSeries.forceLayout = null;
|
}
|
});
|
}
|
