export default function graphForceLayout()

in src/chart/graph/forceLayout.ts [38:167]
119 lines of code
26 McCabe index (conditional complexity)

export default function graphForceLayout(ecModel: GlobalModel) {
    ecModel.eachSeriesByType('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;
                }
                const edgeModel = edge.getModel<GraphEdgeItemOption>();
                const curveness = zrUtil.retrieve3(
                    edge.getModel<GraphEdgeItemOption>().get(['lineStyle', 'curveness']),
                    -getCurvenessForEdge(edge, graphSeries, idx, true),
                    0
                );
                return {
                    n1: nodes[edge.node1.dataIndex],
                    n2: nodes[edge.node2.dataIndex],
                    d: d,
                    curveness,
                    ignoreForceLayout: edgeModel.get('ignoreForceLayout')
                };
            });

            // let coordSys = graphSeries.coordinateSystem;
            const rect = coordSys.getBoundingRect();
            const forceInstance = forceLayout(nodes, edges, {
                rect: rect,
                gravity: forceModel.get('gravity'),
                friction: forceModel.get('friction')
            });
            forceInstance.beforeStep(function (nodes, edges) {
                for (let i = 0, l = nodes.length; i < l; i++) {
                    if (nodes[i].fixed) {
                        // Write back to layout instance
                        vec2.copy(
                            nodes[i].p,
                            graph.getNodeByIndex(i).getLayout() as number[]
                        );
                    }
                }
            });
            forceInstance.afterStep(function (nodes, edges, stopped) {
                for (let i = 0, l = nodes.length; i < l; i++) {
                    if (!nodes[i].fixed) {
                        graph.getNodeByIndex(i).setLayout(nodes[i].p);
                    }
                    preservedPoints[nodeData.getId(i)] = nodes[i].p;
                }
                for (let i = 0, l = edges.length; i < l; i++) {
                    const e = edges[i];
                    const edge = graph.getEdgeByIndex(i);
                    const p1 = e.n1.p;
                    const p2 = e.n2.p;
                    let points = edge.getLayout() as number[][];
                    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;

            // Step to get the layout
            forceInstance.step();
        }
        else {
            // Remove prev injected forceLayout instance
            graphSeries.forceLayout = null;
        }
    });
}