polygonjs-engine/dist/src/core/graph/DirtyController.js

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

export class DirtyController {
  constructor(node) {
    this.node = node;
    this._dirty_count = 0;
    this._dirty = true;
  }
  dispose() {
    this._cached_successors = void 0;
    this._post_dirty_hooks = void 0;
    this._post_dirty_hook_names = void 0;
  }
  isDirty() {
    return this._dirty === true;
  }
  dirtyTimestamp() {
    return this._dirty_timestamp;
  }
  dirtyCount() {
    return this._dirty_count;
  }
  addPostDirtyHook(name, method) {
    this._post_dirty_hook_names = this._post_dirty_hook_names || [];
    this._post_dirty_hooks = this._post_dirty_hooks || [];
    if (!this._post_dirty_hook_names.includes(name)) {
      this._post_dirty_hook_names.push(name);
      this._post_dirty_hooks.push(method);
    } else {
      console.warn(`hook with name ${name} already exists`, this.node);
    }
  }
  removePostDirtyHook(name) {
    if (this._post_dirty_hook_names && this._post_dirty_hooks) {
      const index = this._post_dirty_hook_names.indexOf(name);
      if (index >= 0) {
        this._post_dirty_hook_names.splice(index, 1);
        this._post_dirty_hooks.splice(index, 1);
      }
    }
  }
  has_hook(name) {
    if (this._post_dirty_hook_names) {
      return this._post_dirty_hook_names.includes(name);
    }
    return false;
  }
  removeDirtyState() {
    this._dirty = false;
  }
  set_forbidden_trigger_nodes(nodes) {
    this._forbidden_trigger_nodes = nodes.map((n) => n.graphNodeId());
  }
  set_dirty(original_trigger_graph_node, propagate) {
    if (propagate == null) {
      propagate = true;
    }
    if (original_trigger_graph_node && this._forbidden_trigger_nodes && this._forbidden_trigger_nodes.includes(original_trigger_graph_node.graphNodeId())) {
      return;
    }
    if (original_trigger_graph_node == null) {
      original_trigger_graph_node = this.node;
    }
    this._dirty = true;
    this._dirty_timestamp = performance.now();
    this._dirty_count += 1;
    this.run_post_dirty_hooks(original_trigger_graph_node);
    if (propagate === true) {
      this.set_successors_dirty(original_trigger_graph_node);
    }
  }
  run_post_dirty_hooks(original_trigger_graph_node) {
    if (this._post_dirty_hooks) {
      const cooker = this.node.scene().cooker;
      if (cooker.blocked) {
        cooker.enqueue(this.node, original_trigger_graph_node);
      } else {
        for (let hook of this._post_dirty_hooks) {
          hook(original_trigger_graph_node);
        }
      }
    }
  }
  set_successors_dirty(original_trigger_graph_node) {
    const propagate = false;
    this._cached_successors = this._cached_successors || this.node.graphAllSuccessors();
    for (let successor of this._cached_successors) {
      successor.dirtyController.set_dirty(original_trigger_graph_node, propagate);
    }
  }
  clear_successors_cache() {
    this._cached_successors = void 0;
  }
  clear_successors_cache_with_predecessors() {
    this.clear_successors_cache();
    for (let predecessor of this.node.graphAllPredecessors()) {
      predecessor.dirtyController.clear_successors_cache();
    }
  }
}