lightning-algorithm/step.js at main · vitorsss/lightning-algorithm · 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
class Step {
  _matrix;
  _positions;
  constructor({positions}) {
    this._positions = positions;
    this._matrix = [];
    const {
      minDepth,
      maxDepth,
    } = positions.reduce(({minDepth, maxDepth}, {y}) => {
      return {
        minDepth: Math.min(minDepth, y),
        maxDepth: Math.max(maxDepth, y),
      };
    }, {
      minDepth: Infinity,
      maxDepth: -Infinity,
    });
    const probabilityPerIndex = 1/(maxDepth-minDepth);
    positions.forEach(({x,y}) => {
      this._matrix[x] = this._matrix[x] || [];
      this._matrix[x][y] = (y - minDepth) * probabilityPerIndex;
    });
  }

  getProbability({
    x,
    y,
  }) {
    return (this._matrix[x] || [])[y] || 0;
  }

  getPositions() {
    return this._positions;
  }
}