@polygonjs/polygonjs/src/engine/nodes/sop/Sort.ts

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

/**
 * Sorts vertices
 *
 *
 */
import {AttribClass} from './../../../core/geometry/Constant';
import {TypedSopNode} from './_Base';
import {CoreGroup} from '../../../core/geometry/Group';
import {InputCloneMode} from '../../poly/InputCloneMode';
import {NodeParamsConfig, ParamConfig} from '../utils/params/ParamsConfig';
import {
	SortSopOperation,
	AXISES,
	SORT_MODES,
	SortMode,
	SORT_TARGET_TYPES,
	SortTargetType,
} from '../../operations/sop/Sort';
import {SopType} from '../../poly/registers/nodes/types/Sop';
const DEFAULT = SortSopOperation.DEFAULT_PARAMS;

class SortSopParamsConfig extends NodeParamsConfig {
	/** @param defines if this node will sort points or objects */
	targetType = ParamConfig.INTEGER(DEFAULT.targetType, {
		menu: {
			entries: SORT_TARGET_TYPES.map((name, value) => ({name, value})),
		},
	});
	/** @param criteria used to sort */
	mode = ParamConfig.INTEGER(DEFAULT.mode, {
		menu: {
			entries: SORT_MODES.map((name, value) => ({name, value})),
		},
	});
	/** @param seed used by the random mode */
	seed = ParamConfig.INTEGER(DEFAULT.seed, {
		range: [0, 100],
		rangeLocked: [false, false],
		visibleIf: {mode: SORT_MODES.indexOf(SortMode.RANDOM)},
	});
	/** @param axis along which points will be sorted */
	axis = ParamConfig.INTEGER(DEFAULT.axis, {
		menu: {
			entries: AXISES.map((name, value) => {
				return {name, value};
			}),
		},
		visibleIf: {mode: SORT_MODES.indexOf(SortMode.AXIS)},
	});
	/** @param attribute */
	attribute = ParamConfig.STRING(DEFAULT.attribute, {
		visibleIf: {mode: SORT_MODES.indexOf(SortMode.ATTRIBUTE)},
	});
	/** @param invert the sort */
	invert = ParamConfig.BOOLEAN(DEFAULT.invert);
}
const ParamsConfig = new SortSopParamsConfig();

export class SortSopNode extends TypedSopNode<SortSopParamsConfig> {
	override paramsConfig = ParamsConfig;
	static override type() {
		return SopType.SORT;
	}

	override initializeNode() {
		this.io.inputs.setCount(1);
		this.io.inputs.initInputsClonedState([InputCloneMode.FROM_NODE]);
	}

	private _operation: SortSopOperation | undefined;
	override cook(inputCoreGroups: CoreGroup[]) {
		this._operation = this._operation || new SortSopOperation(this._scene, this.states, this);
		const coreGroup = this._operation.cook(inputCoreGroups, this.pv);
		this.setCoreGroup(coreGroup);
	}

	setAttribClass(attribClass: SortTargetType) {
		if (SORT_TARGET_TYPES.includes(attribClass)) {
			this.p.targetType.set(SORT_TARGET_TYPES.indexOf(attribClass));
		} else {
			console.warn(`${attribClass} is not possible on this node`);
		}
	}
	attribClass() {
		return SORT_TARGET_TYPES[this.pv.targetType];
	}
	setSortMode(mode: SortMode) {
		this.p.mode.set(SORT_MODES.indexOf(mode));
	}
	setTargetType(targetType: AttribClass.POINT | AttribClass.OBJECT) {
		this.p.targetType.set(SORT_TARGET_TYPES.indexOf(targetType));
	}
}