FEngine/packages/f-engine/src/shape/smoothPolyline.ts at 700b5fb54e0410340aa0a832a98c53b8382d5ecd · antvis/FEngine · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
import { Path } from '@antv/g-lite';
import { PathArray } from '@antv/util';
import * as Smooth from './util/smooth';


export class SmoothPolyline extends Path {
  static tag = 'smooth-polyline';
  parsedStyle: any;

  constructor(config) {
    super(config);
    this.updatePath();
  }

  setAttribute(name, value, force?: boolean) {
    super.setAttribute(name, value, force);
    if (['smooth', 'points', 'step'].indexOf(name) > -1) {
      this.updatePath();
    }
  }

  private updatePath() {
    const { smooth, points, step } = this.parsedStyle;
    const { points: pos } = points;

    const d: PathArray = [['M', pos[0][0], pos[0][1]]];

    if (smooth) {
      const constaint = [
        [0, 0],
        [1, 1],
      ];
      const sps = Smooth.smooth(
        pos.map((d) => {
          return {
            x: d[0],
            y: d[1],
          };
        }),
        false,
        constaint,
      );

      for (let i = 0, n = sps.length; i < n; i++) {
        const sp = sps[i];
        d.push(['C', sp[1], sp[2], sp[3], sp[4], sp[5], sp[6]]);
      }
    } else if (step) {
      let i;
      let l;
      switch (step) {
        case "start":
          for (i = 1, l = pos.length; i < l; i++) {
            const x = pos[i - 1][0]
            d.push(['L', x, pos[i - 1][1]])
            d.push(['L', x, pos[i][1]])
            d.push(['L', pos[i][0], pos[i][1]]);
          }
          break;
        case "middle":
          for (i = 1, l = pos.length; i < l; i++) {
            const x = (pos[i][0] + pos[i - 1][0]) / 2
            d.push(['L', x, pos[i - 1][1]])
            d.push(['L', x, pos[i][1]])
            d.push(['L', pos[i][0], pos[i][1]]);
          }
          break;
        case "end":
          for (i = 1, l = pos.length; i < l; i++) {
            const x = pos[i][0]
            d.push(['L', x, pos[i - 1][1]])
            d.push(['L', x, pos[i][1]])
            d.push(['L', pos[i][0], pos[i][1]]);
          }
          break;
      }
    } else {
      let i;
      let l;
      for (i = 1, l = pos.length - 1; i < l; i++) {
        d.push(['L', pos[i][0], pos[i][1]]);
      }
      d.push(['L', pos[l][0], pos[l][1]]);
    }
    super.setAttribute('d', d);
  }
}