summaryrefslogtreecommitdiff
path: root/assets/trading-in-the-rain/SeriesComposer.js
blob: 134c64ce13187f31cfcf5f8722bd10ebd13ef0d0 (plain)
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
function SeriesComposer(resource, rainCanvas, color) {
  this.rainCanvas = rainCanvas;
  this.color = color;

  this.priceDist = new Distributor(200);
  this.volumeDist = new Distributor(200);
  this.musicBox = new MusicBox(this.priceDist, this.volumeDist);

  this.enabled = false;
  this.setEnabled = (enabled) => this.enabled = enabled;
  this.getEnabled = () => { return this.enabled; }

  this.totalTrades = 0;
  this.getTotalTrades = () => { return this.totalTrades; }

  this.cw = new CW(resource);
  this.cw.ontrades = (trades) => {
    if (this.totalTrades > 0 && this.enabled) {
      let priceVols = {}; // sum volumes by price, for deduplication
      for (let i in trades) {
        let price = trades[i].price, volume = trades[i].volume;
        if (!priceVols[price]) priceVols[price] = 0;
        priceVols[price] += volume;
      }

      trades = []; // overwrite trades with deduplicated ones.
      for (let price in priceVols) {
        let volume = priceVols[price];
        let intensity = this.volumeDist.distribute(volume, 0, 1);
        this.rainCanvas.newDrop({
          x: this.priceDist.distribute(price, 0, 1),
          y: Math.random(),
          intensity: intensity,
          color: this.color,
        });

        trades.push({price: price, volume: volume});
      }

      this.musicBox.playTrades(trades);
    }

    for (let i in trades) {
      this.priceDist.add(trades[i].price);
      this.volumeDist.add(trades[i].volume);
    }

    this.totalTrades += trades.length;
    if (this.ontrades) this.ontrades(trades);
  };
}