Function graphForceLayout

src/chart/graph/forceLayout.ts:40–169

(ecModel: GlobalModel)

export const graphForceLayoutStageHandler = createSimpleOverallStageHandler(SERIES_TYPE_GRAPH, graphForceLayout);

function graphForceLayout(ecModel: GlobalModel) {
    ecModel.eachSeriesByType(SERIES_TYPE_GRAPH, function (graphSeries: GraphSeriesModel) {
        const coordSys = graphSeries.coordinateSystem;
        if (coordSys && coordSys.type !== 'view') {
            return;
        }
        if (graphSeries.get('layout') === 'force') {
            const preservedPoints = graphSeries.preservedPoints || {};
            const graph = graphSeries.getGraph();
            const nodeData = graph.data;
            const edgeData = graph.edgeData;
            const forceModel = graphSeries.getModel('force');
            const initLayout = forceModel.get('initLayout');
            if (graphSeries.preservedPoints) {
                nodeData.each(function (idx) {
                    const id = nodeData.getId(idx);
                    nodeData.setItemLayout(idx, preservedPoints[id] || [NaN, NaN]);
                });
            }
            else if (!initLayout || initLayout === 'none') {
                simpleLayout(graphSeries);
            }
            else if (initLayout === 'circular') {
                circularLayout(graphSeries, 'value');
            }

            const nodeDataExtent = nodeData.getDataExtent('value');
            const edgeDataExtent = edgeData.getDataExtent('value');
            // let edgeDataExtent = edgeData.getDataExtent('value');
            const repulsion = forceModel.get('repulsion');
            const edgeLength = forceModel.get('edgeLength');
            const repulsionArr = zrUtil.isArray(repulsion)
                ? repulsion : [repulsion, repulsion];
            let edgeLengthArr = zrUtil.isArray(edgeLength)
                ? edgeLength : [edgeLength, edgeLength];

            // Larger value has smaller length
            edgeLengthArr = [edgeLengthArr[1], edgeLengthArr[0]];

            const nodes = nodeData.mapArray('value', function (value: number, idx) {
                const point = nodeData.getItemLayout(idx) as number[];
                let rep = linearMap(value, nodeDataExtent, repulsionArr);
                if (isNaN(rep)) {
                    rep = (repulsionArr[0] + repulsionArr[1]) / 2;
                }
                return {
                    w: rep,
                    rep: rep,
                    fixed: nodeData.getItemModel<GraphNodeItemOption>(idx).get('fixed'),
                    p: (!point || isNaN(point[0]) || isNaN(point[1])) ? null : point
                };
            });
            const edges = edgeData.mapArray('value', function (value: number, idx) {
                const edge = graph.getEdgeByIndex(idx);
                let d = linearMap(value, edgeDataExtent, edgeLengthArr);
                if (isNaN(d)) {
                    d = (edgeLengthArr[0] + edgeLengthArr[1]) / 2;
                }