psn-js/benchmark.js

const { Encoder, Decoder, Tracker } = require('./dist/cjs');

const testSizes = [1, 10, 100, 1000];

const encoder = new Encoder('test encoder', 2, 3);
const decoder = new Decoder();

function getNTrackers(n) {
  const trackers = [];
  for (let index = 0; index < n; index += 1) {
    const tracker = new Tracker(index, `Tracker ${index}`);
    tracker.setPos(0, 0, 0);
    tracker.setSpeed(0, 0, 0);
    tracker.setOri(0, 0, 0);
    tracker.setAccel(0, 0, 0);
    tracker.setTrgtPos(0, 0, 0);
    tracker.setStatus(1.0);
    tracker.setTimestamp(Date.now());
    trackers.push(tracker);
  }
  return trackers;
}

function benchmark(trackerCount, iterations) {
  const benchmarkResults = {
    data: {},
    info: {},
  };
  const trackers = getNTrackers(trackerCount);
  console.log(
    `processing ${trackers.length} tracker${trackers.length > 1 ? 's' : ''} ${iterations} time${
      iterations > 1 ? 's' : ''
    }`
  );

  let latestEncodedPackets;

  // DATA
  const dataEncoderStart = performance.now();
  for (let index = 0; index < iterations; index += 1) {
    latestEncodedPackets = encoder.getDataPackets(Date.now(), trackers);
  }
  benchmarkResults.data.encode = `${(performance.now() - dataEncoderStart).toFixed(2)}ms`;

  const dataDecodedStart = performance.now();
  for (let index = 0; index < iterations; index += 1) {
    latestEncodedPackets.forEach((packet) => {
      decoder.decode(packet);
    });
  }
  benchmarkResults.data.decode = `${(performance.now() - dataDecodedStart).toFixed(2)}ms`;

  // INFO
  const infoEncoderStart = performance.now();
  for (let index = 0; index < iterations; index += 1) {
    latestEncodedPackets = encoder.getInfoPackets(Date.now(), trackers);
  }
  benchmarkResults.info.encode = `${(performance.now() - infoEncoderStart).toFixed(2)}ms`;

  const infoDecodeStart = performance.now();
  for (let index = 0; index < iterations; index += 1) {
    latestEncodedPackets.forEach((packet) => {
      decoder.decode(packet);
    });
  }
  benchmarkResults.info.decode = `${(performance.now() - infoDecodeStart).toFixed(2)}ms`;

  console.table(benchmarkResults);
}

testSizes.forEach((iterations) => {
  testSizes.forEach((trackerCount) => {
    benchmark(trackerCount, iterations);
  });
});