playcanvas/build/playcanvas.prf/src/scene/gsplat/gsplat-sogs-resource.js

PlayCanvas WebGL game engine

import { Vec2 } from '../../core/math/vec2.js';
import { GSplatResourceBase } from './gsplat-resource-base.js';

class GSplatSogsResource extends GSplatResourceBase {
		destroy() {
				this.gsplatData.destroy();
				super.destroy();
		}
		configureMaterial(material) {
				const { gsplatData } = this;
				material.setDefine('GSPLAT_SOGS_DATA', true);
				material.setDefine('SH_BANDS', this.gsplatData.shBands);
				[
						'means_l',
						'means_u',
						'quats',
						'scales',
						'sh0',
						'sh_centroids',
						'sh_labels'
				].forEach((name)=>{
						if (gsplatData[name]) {
								material.setParameter(name, gsplatData[name]);
						}
				});
				[
						'means',
						'scales',
						'sh0',
						'shN'
				].forEach((name)=>{
						const v = gsplatData.meta[name];
						if (v) {
								material.setParameter(`${name}_mins`, v.mins);
								material.setParameter(`${name}_maxs`, v.maxs);
						}
				});
		}
		evalTextureSize(count) {
				return new Vec2(this.gsplatData.means_l.width, this.gsplatData.means_l.height);
		}
}

export { GSplatSogsResource };