123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160 |
- /*
- * 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;
- }
- });
- }
|