@polygonjs/polygonjs/src/engine/operations/sop/utils/attribCreate/AttribCreateVertex.ts

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

import {CoreGroup} from '../../../../../core/geometry/Group';
import {CoreAttribute} from '../../../../../core/geometry/Attribute';
import {AttribCreateSopParams} from '../../AttribCreate';
import {AttribType} from '../../../../../core/geometry/Constant';
import {TypeAssert} from '../../../../poly/Assert';
import {
	verticesFromObject,
	verticesFromObjectFromGroup,
} from '../../../../../core/geometry/entities/vertex/CoreVertexUtils';
import {coreVertexClassFactory} from '../../../../../core/geometry/CoreObjectFactory';
import {CoreObjectType, ObjectContent} from '../../../../../core/geometry/ObjectContent';
import {CoreVertex} from '../../../../../core/geometry/entities/vertex/CoreVertex';

export function addVertexAttribute(attribType: AttribType, coreGroup: CoreGroup, params: AttribCreateSopParams) {
	const objects = coreGroup.allObjects();
	switch (attribType) {
		case AttribType.NUMERIC: {
			for (let object of objects) {
				_addAttributeToVertices(object, params, false);
			}
			return;
		}
		case AttribType.STRING: {
			for (let object of objects) {
				_addAttributeToVertices(object, params, true);
			}
			return;
		}
	}
	TypeAssert.unreachable(attribType);
}

const _vertices: CoreVertex<CoreObjectType>[] = [];
function _addAttributeToVertices<T extends CoreObjectType>(
	object: ObjectContent<T>,
	params: AttribCreateSopParams,
	isString: boolean
) {
	const value = isString
		? params.string
		: [params.value1, params.value2, params.value3, params.value4][params.size - 1];
	const attribName = CoreAttribute.remapName(params.name);

	// add default if not found
	const vertexClass = coreVertexClassFactory(object);
	let attribute = vertexClass.attribute(object, attribName);
	if (!attribute) {
		const verticesCount = vertexClass.entitiesCount(object);
		const values = new Array(verticesCount * params.size).fill(value);
		attribute = {array: values, itemSize: params.size, isString: isString};
		vertexClass.addAttribute(object, attribName, attribute);
	}

	// set values
	if (params.group) {
		verticesFromObjectFromGroup(object, params.group, _vertices);
		for (let vertex of _vertices) {
			vertex.setAttribValue(attribName, value);
		}
	} else {
		verticesFromObject(object, _vertices);
		for (let vertex of _vertices) {
			vertex.setAttribValue(attribName, value);
		}
	}
}