fix: fix the implementation of polyline in the indented tree case

packages/g6/__tests__/demos/case-indented-tree.ts

@@ -251,7 +251,8 @@ export const caseIndentedTree: TestCase = async (context) => {
   }
 
   class IndentedEdge extends Polyline {
-    protected getControlPoints(attributes: Required<PolylineStyleProps>, sourcePoint: Point, targetPoint: Point) {
+    protected getControlPoints(attributes: Required<PolylineStyleProps>) {
+      const [sourcePoint, targetPoint] = this.getEndpoints(attributes, false);
       const [sx] = sourcePoint;
       const [, ty] = targetPoint;
       return [[sx, ty]] as Point[];

packages/g6/src/elements/edges/polyline.ts

@@ -56,7 +56,7 @@ export class Polyline extends BaseEdge {
     super(mergeOptions({ style: Polyline.defaultStyleProps }, options));
   }
 
-  protected getPoints(attributes: ParsedPolylineStyleProps): Point[] {
+  protected getControlPoints(attributes: ParsedPolylineStyleProps): Point[] {
     const { router } = attributes;
     const { sourceNode, targetNode } = this;
     const [sourcePoint, targetPoint] = this.getEndpoints(attributes, false);
@@ -79,6 +79,12 @@ export class Polyline extends BaseEdge {
       }
     }
 
+    return controlPoints;
+  }
+
+  protected getPoints(attributes: ParsedPolylineStyleProps): Point[] {
+    const controlPoints = this.getControlPoints(attributes);
+
     const [newSourcePoint, newTargetPoint] = this.getEndpoints(attributes, true, controlPoints);
     return [newSourcePoint, ...controlPoints, newTargetPoint];
   }

packages/site/examples/scene-case/default/demo/indented-tree.js

@@ -225,7 +225,8 @@ class IndentedNode extends BaseNode {
 }
 
 class IndentedEdge extends Polyline {
-  getControlPoints(attributes, sourcePoint, targetPoint) {
+  getControlPoints(attributes) {
+    const [sourcePoint, targetPoint] = this.getEndpoints(attributes, false);
     const [sx] = sourcePoint;
     const [, ty] = targetPoint;
     return [[sx, ty]];