polygonjs-engine/dist/src/core/performance/CorePerformance.js

node-based webgl 3D engine https://polygonjs.com

import {PerformanceNode as PerformanceNode2} from "./PerformanceNode";
import {ArrayUtils as ArrayUtils2} from "../ArrayUtils";
import {ObjectUtils as ObjectUtils2} from "../ObjectUtils";
export class CorePerformance {
  constructor() {
    this._started = false;
    this._start_time = 0;
    this._previous_timestamp = 0;
    this._nodes_cook_data = {};
    this._durations_by_name = {};
    this._durations_count_by_name = {};
  }
  profile(name, method) {
    const start_time = performance.now();
    method();
    const total_time = performance.now() - start_time;
    console.log(`${name}: ${total_time}`);
  }
  start() {
    if (!this._started) {
      this.reset();
      this._started = true;
      this._start_time = performance.now();
      this._nodes_cook_data = {};
      this._previous_timestamp = this._start_time;
    }
  }
  stop() {
    this.reset();
  }
  reset() {
    this._started = false;
    this._start_time = null;
    this._durations_by_name = {};
    this._durations_count_by_name = {};
    this._nodes_cook_data = {};
  }
  started() {
    return this._started;
  }
  record_node_cook_data(node, performance_data) {
    const id = node.graphNodeId();
    if (this._nodes_cook_data[id] == null) {
      this._nodes_cook_data[id] = new PerformanceNode2(node);
    }
    this._nodes_cook_data[id].update_cook_data(performance_data);
  }
  record(name) {
    if (!this.started()) {
      this.start();
    }
    const current_timestamp = performance.now();
    if (this._durations_by_name[name] == null) {
      this._durations_by_name[name] = 0;
    }
    this._durations_by_name[name] += current_timestamp - this._previous_timestamp;
    if (this._durations_count_by_name[name] == null) {
      this._durations_count_by_name[name] = 0;
    }
    this._durations_count_by_name[name] += 1;
    return this._previous_timestamp = current_timestamp;
  }
  print() {
    this.print_node_cook_data();
    this.print_recordings();
  }
  print_node_cook_data() {
    let performance_nodes = Object.values(this._nodes_cook_data);
    performance_nodes = ArrayUtils2.sortBy(performance_nodes, (performance_node) => performance_node.total_cook_time());
    const print_objects = performance_nodes.map((performance_node) => performance_node.print_object());
    console.log("--------------- NODES COOK TIME -----------");
    const table_entries = [];
    const sorted_print_objects = ArrayUtils2.sortBy(print_objects, (print_object) => -print_object.total_cook_time);
    for (let print_object of sorted_print_objects) {
      table_entries.push(print_object);
    }
    console.table(table_entries);
    return print_objects;
  }
  print_recordings() {
    const durations_by_name = ObjectUtils2.clone(this._durations_by_name);
    const durations_count_by_name = ObjectUtils2.clone(this._durations_count_by_name);
    const durations = [];
    const names_by_duration = {};
    for (let name of Object.keys(durations_by_name)) {
      const duration = durations_by_name[name];
      durations.push(duration);
      if (names_by_duration[duration] == null) {
        names_by_duration[duration] = [];
      }
      names_by_duration[duration].push(name);
    }
    durations.sort((a, b) => a - b);
    const sorted_durations = ArrayUtils2.uniq(durations);
    console.log("--------------- PERF RECORDINGS -----------");
    const table_entries = [];
    for (let duration of sorted_durations) {
      const names = names_by_duration[duration];
      for (let name of names) {
        const count = durations_count_by_name[name];
        const duration_per_iteration = duration / count;
        const entry = {duration, name, count, duration_per_iteration};
        table_entries.push(entry);
      }
    }
    console.table(table_entries);
    return table_entries;
  }
}