pixi-stats/src/hooks/GLHook.ts

WebGL stats for pixi.js or/and three.js

/* eslint-disable @typescript-eslint/no-explicit-any */

export default class GLHook {
  public drawPasses: number = 0;

  public isInit: boolean = false;

  private realGLDrawElements: (...args: any[]) => void = function () {};

  private gl: any;

  constructor(_gl?: any) {
    if (!_gl) {
      console.warn("[GLHook] GL can't be NULL");
    } else if (_gl.__proto__.drawElements) {
      this.gl = _gl;
      this.realGLDrawElements = this.gl.__proto__.drawElements;

      // replace with new function
      this.gl.__proto__.drawElements = this.fakeGLdrawElements.bind(this);
      this.isInit = true;

      console.info('[GLHook] GL was Hooked!');
    }
  }

  private fakeGLdrawElements(
    mode: any,
    count: any,
    type: any,
    offset: any
  ): void {
    this.drawPasses++;
    this.realGLDrawElements.call(this.gl, mode, count, type, offset);
  }

  public reset(): void {
    this.drawPasses = 0;
  }

  public release(): void {
    if (this.isInit) {
      this.gl.__proto__.drawElements = this.realGLDrawElements;

      console.info('[GLHook] Hook was removed!');
    }

    this.isInit = false;
  }
}