polygonjs-engine/src/engine/nodes/gl/code/utils/TextureVariable.ts

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

import {TextureAllocation} from './TextureAllocation';
import {PolyScene} from '../../../../scene/PolyScene';
import {CoreGraphNodeId} from '../../../../../core/graph/CoreGraph';

export interface TextureVariableData {
	name: string;
	size: number;
	nodes: string[];
}

export class TextureVariable {
	private _allocation: TextureAllocation | undefined;
	private _position: number = -1;

	private _graph_node_ids: Map<CoreGraphNodeId, boolean> | undefined;

	constructor(private _name: string, private _size: number) {
		if (!_name) {
			throw 'TextureVariable requires a name';
		}
	}

	set_allocation(allocation: TextureAllocation) {
		this._allocation = allocation;
	}
	get allocation() {
		return this._allocation;
	}

	get graph_node_ids() {
		return this._graph_node_ids;
	}
	add_graph_node_id(id: CoreGraphNodeId) {
		this._graph_node_ids = this._graph_node_ids || new Map();
		this._graph_node_ids.set(id, true);
	}
	name() {
		return this._name;
	}
	get size() {
		return this._size;
	}

	setPosition(position: number) {
		this._position = position;
	}
	get position() {
		return this._position;
	}
	get component(): string {
		return 'xyzw'.split('').splice(this._position, this._size).join('');
	}

	static from_json(data: TextureVariableData): TextureVariable {
		return new TextureVariable(data.name, data.size);
	}

	toJSON(scene: PolyScene): TextureVariableData {
		const names: string[] = [];
		if (this._graph_node_ids) {
			this._graph_node_ids.forEach((boolean, node_id) => {
				const name = scene.graph.node_from_id(node_id)?.name();
				if (name) {
					names.push(name);
				}
			});
		}

		return {
			name: this.name(),
			size: this.size,
			nodes: names.sort(),
		};
	}
}