@polygonjs/polygonjs/src/engine/nodes/cop/Lut.ts

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

/**
 * Imports a LUT to be used in Post nodes
 *
 */
import {ParamEvent} from './../../poly/ParamEvent';
import {ClampToEdgeWrapping, LinearFilter, Texture, TextureLoader} from 'three';
import {TypedCopNode} from './_Base';
import {BaseNodeType} from '../_Base';
import {BaseParamType} from '../../params/_Base';
import {NodeParamsConfig, ParamConfig} from '../utils/params/ParamsConfig';
import {LUT3dlLoader, LUTCubeLoader} from 'postprocessing';
import {LOADING_MANAGER} from '../../../core/loader/_Base';
import {CopType} from '../../poly/registers/nodes/types/Cop';
import {EXTENSIONS_BY_NODE_TYPE_BY_CONTEXT} from '../../../core/loader/FileExtensionRegister';
import {NodeContext} from '../../poly/NodeContext';

type OnTextureLoad = (texture: Texture) => void;
class LutCopParamsConfig extends NodeParamsConfig {
	/** @param url to fetch the lut from */
	url = ParamConfig.STRING('', {
		fileBrowse: {extensions: EXTENSIONS_BY_NODE_TYPE_BY_CONTEXT[NodeContext.COP][CopType.LUT]},
	});
	/** @param reload the image */
	reload = ParamConfig.BUTTON(null, {
		callback: (node: BaseNodeType, param: BaseParamType) => {
			LutCopNode.PARAM_CALLBACK_reload(node as LutCopNode);
		},
	});
}

const ParamsConfig = new LutCopParamsConfig();

export class LutCopNode extends TypedCopNode<LutCopParamsConfig> {
	override paramsConfig = ParamsConfig;
	static override type() {
		return CopType.LUT;
	}

	override initializeNode() {
		this.io.inputs.setCount(0);
	}
	override async cook(input_contents: Texture[]) {
		const textureLoader = new TextureLoader(LOADING_MANAGER);

		let loader: TextureLoader | LUT3dlLoader | LUTCubeLoader = textureLoader;
		let onLoad: OnTextureLoad | null = (texture) => {
			texture.generateMipmaps = false;
			texture.minFilter = LinearFilter;
			texture.magFilter = LinearFilter;
			texture.wrapS = ClampToEdgeWrapping;
			texture.wrapT = ClampToEdgeWrapping;
			texture.flipY = false;
		};
		const {url} = this.pv;
		if (/.3dl$/im.test(url)) {
			loader = new LUT3dlLoader(LOADING_MANAGER);
			onLoad = null;
		}
		if (/.cube$/im.test(url)) {
			loader = new LUTCubeLoader(LOADING_MANAGER);
			onLoad = null;
		}
		loader.load(this.pv.url, (texture: Texture) => {
			console.log(texture);
			if (loader instanceof TextureLoader)
				if (onLoad) {
					// texture.name = entry[0];
					onLoad(texture);
				}

			this.setTexture(texture);
		});
	}
	static PARAM_CALLBACK_reload(node: LutCopNode) {
		node.paramCallbackReload();
	}
	private paramCallbackReload() {
		// set the param dirty is preferable to just the successors, in case the expression result needs to be updated
		// this.p.url.set_successors_dirty();
		this.p.url.setDirty();
		this.p.url.emit(ParamEvent.ASSET_RELOAD_REQUEST);
	}
}