pixi-stats/src/hooks/TextureHook.ts

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

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

export default class TextureHook {
  public createdTextures: Array<any> = new Array<any>();

  public maxTexturesCount: number = 0;

  public isInit: boolean = false;

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

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

  private gl: any;

  constructor(_gl?: any) {
    if (!_gl) {
      console.warn("[TextureHook] GL can't be NULL");
    } else if (_gl.__proto__.createTexture) {
      this.gl = _gl;
      this.realGLCreateTexture = this.gl.__proto__.createTexture;
      this.realGLDeleteTexture = this.gl.__proto__.deleteTexture;

      // replace with new function
      this.gl.__proto__.createTexture = this.fakeGLCreateTexture.bind(this);
      this.gl.__proto__.deleteTexture = this.fakeGLDeleteTexture.bind(this);
      this.isInit = true;

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

  public get currentTextureCount(): number {
    return this.createdTextures.length;
  }

  public registerTexture(texture: any): void {
    this.createdTextures.push(texture); // ++;
    this.maxTexturesCount = Math.max(
      this.createdTextures.length,
      this.maxTexturesCount
    );
  }

  private fakeGLCreateTexture(): any {
    const texture = this.realGLCreateTexture.call(this.gl);
    this.registerTexture(texture);
    return texture;
  }

  private fakeGLDeleteTexture(texture: any): void {
    const index: number = this.createdTextures.indexOf(texture);
    if (index > -1) {
      this.createdTextures.splice(index, 1);
    }
    this.realGLDeleteTexture.call(this.gl, texture);
  }

  public reset(): void {
    this.createdTextures = new Array<any>();
    this.maxTexturesCount = 0;
  }

  public release(): void {
    if (this.isInit) {
      this.gl.__proto__.createTexture = this.realGLCreateTexture;
      this.gl.__proto__.deleteTexture = this.realGLDeleteTexture;

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

    this.isInit = false;
  }
}