playcanvas
Version:
Open-source WebGL/WebGPU 3D engine for the web
934 lines (933 loc) • 36 kB
JavaScript
import { now } from "../../core/time.js";
import { Curve } from "../../core/math/curve.js";
import { CurveSet } from "../../core/math/curve-set.js";
import { Mat4 } from "../../core/math/mat4.js";
import { math } from "../../core/math/math.js";
import { Quat } from "../../core/math/quat.js";
import { Vec3 } from "../../core/math/vec3.js";
import { BoundingBox } from "../../core/shape/bounding-box.js";
import {
ADDRESS_CLAMP_TO_EDGE,
BUFFER_DYNAMIC,
CULLFACE_NONE,
FILTER_LINEAR,
FILTER_NEAREST,
INDEXFORMAT_UINT32,
PIXELFORMAT_RGBA8,
PIXELFORMAT_RGBA32F,
PIXELFORMAT_RGBA32U,
PRIMITIVE_TRIANGLES,
SEMANTIC_ATTR0,
SEMANTIC_ATTR1,
SEMANTIC_ATTR2,
SEMANTIC_ATTR3,
SEMANTIC_ATTR4,
SEMANTIC_TEXCOORD0,
TYPE_FLOAT32,
typedArrayIndexFormats,
requiresManualGamma,
PIXELFORMAT_SRGBA8,
SEMANTIC_POSITION,
isIntegerPixelFormat
} from "../../platform/graphics/constants.js";
import { DeviceCache } from "../../platform/graphics/device-cache.js";
import { IndexBuffer } from "../../platform/graphics/index-buffer.js";
import { RenderTarget } from "../../platform/graphics/render-target.js";
import { Texture } from "../../platform/graphics/texture.js";
import { VertexBuffer } from "../../platform/graphics/vertex-buffer.js";
import { VertexFormat } from "../../platform/graphics/vertex-format.js";
import {
BLEND_NORMAL,
EMITTERSHAPE_BOX,
PARTICLEMODE_GPU,
PARTICLEORIENTATION_SCREEN,
PARTICLEORIENTATION_WORLD,
PARTICLESORT_NONE
} from "../constants.js";
import { Mesh } from "../mesh.js";
import { MeshInstance } from "../mesh-instance.js";
import { ShaderUtils } from "../shader-lib/shader-utils.js";
import { ParticleCPUUpdater } from "./cpu-updater.js";
import { ParticleGPUUpdater } from "./gpu-updater.js";
import { ParticleMaterial } from "./particle-material.js";
const particleVerts = [
[-1, -1],
[1, -1],
[1, 1],
[-1, 1]
];
function _createTexture(device, width, height, pixelData, format = PIXELFORMAT_RGBA32F, mult8Bit, filter) {
let mipFilter = FILTER_NEAREST;
if (filter && (format === PIXELFORMAT_RGBA8 || format === PIXELFORMAT_SRGBA8)) {
mipFilter = FILTER_LINEAR;
}
const texture = new Texture(device, {
width,
height,
format,
cubemap: false,
mipmaps: false,
minFilter: mipFilter,
magFilter: mipFilter,
addressU: ADDRESS_CLAMP_TO_EDGE,
addressV: ADDRESS_CLAMP_TO_EDGE,
name: "ParticleSystemTexture"
});
const pixels = texture.lock();
if (format === PIXELFORMAT_RGBA8 || format === PIXELFORMAT_SRGBA8) {
const temp = new Uint8Array(pixelData.length);
for (let i = 0; i < pixelData.length; i++) {
temp[i] = pixelData[i] * mult8Bit * 255;
}
pixelData = temp;
} else if (isIntegerPixelFormat(format)) {
pixelData = new Uint32Array(pixelData.buffer, pixelData.byteOffset, pixelData.length);
}
pixels.set(pixelData);
texture.unlock();
return texture;
}
function saturate(x) {
return Math.max(Math.min(x, 1), 0);
}
const default0Curve = new Curve([0, 0, 1, 0]);
const default1Curve = new Curve([0, 1, 1, 1]);
const default0Curve3 = new CurveSet([0, 0, 1, 0], [0, 0, 1, 0], [0, 0, 1, 0]);
const default1Curve3 = new CurveSet([0, 1, 1, 1], [0, 1, 1, 1], [0, 1, 1, 1]);
let particleTexHeight = 2;
const particleTexChannels = 4;
const extentsInnerRatioUniform = new Float32Array(3);
const spawnMatrix = new Mat4();
const tmpVec3 = new Vec3();
const bMin = new Vec3();
const bMax = new Vec3();
let setPropertyTarget;
let setPropertyOptions;
function setProperty(pName, defaultVal) {
if (setPropertyOptions[pName] !== void 0 && setPropertyOptions[pName] !== null) {
setPropertyTarget[pName] = setPropertyOptions[pName];
} else {
setPropertyTarget[pName] = defaultVal;
}
}
function pack3NFloats(a, b, c) {
const packed = a * 255 << 16 | b * 255 << 8 | c * 255;
return packed / (1 << 24);
}
function packTextureXYZ_NXYZ(qXYZ, qXYZ2) {
const num = qXYZ.length / 3;
const colors = new Array(num * 4);
for (let i = 0; i < num; i++) {
colors[i * 4] = qXYZ[i * 3];
colors[i * 4 + 1] = qXYZ[i * 3 + 1];
colors[i * 4 + 2] = qXYZ[i * 3 + 2];
colors[i * 4 + 3] = pack3NFloats(qXYZ2[i * 3], qXYZ2[i * 3 + 1], qXYZ2[i * 3 + 2]);
}
return colors;
}
function packTextureRGBA(qRGB, qA) {
const colors = new Array(qA.length * 4);
for (let i = 0; i < qA.length; i++) {
colors[i * 4] = qRGB[i * 3];
colors[i * 4 + 1] = qRGB[i * 3 + 1];
colors[i * 4 + 2] = qRGB[i * 3 + 2];
colors[i * 4 + 3] = qA[i];
}
return colors;
}
function packTexture5Floats(qA, qB, qC, qD, qE) {
const colors = new Array(qA.length * 4);
for (let i = 0; i < qA.length; i++) {
colors[i * 4] = qA[i];
colors[i * 4 + 1] = qB[i];
colors[i * 4 + 2] = 0;
colors[i * 4 + 3] = pack3NFloats(qC[i], qD[i], qE[i]);
}
return colors;
}
function packTexture2Floats(qA, qB) {
const colors = new Array(qA.length * 4);
for (let i = 0; i < qA.length; i++) {
colors[i * 4] = qA[i];
colors[i * 4 + 1] = qB[i];
colors[i * 4 + 2] = 0;
colors[i * 4 + 3] = 0;
}
return colors;
}
function calcSpawnDuration(emitter) {
return Math.max(emitter.rate, emitter.rate2) * emitter.numParticles + emitter.lifetime;
}
function subGraph(A, B) {
const r = new Float32Array(A.length);
for (let i = 0; i < A.length; i++) {
r[i] = A[i] - B[i];
}
return r;
}
function maxUnsignedGraphValue(A, outUMax) {
const chans = outUMax.length;
const values = A.length / chans;
for (let i = 0; i < values; i++) {
for (let j = 0; j < chans; j++) {
const a = Math.abs(A[i * chans + j]);
outUMax[j] = Math.max(outUMax[j], a);
}
}
}
function normalizeGraph(A, uMax) {
const chans = uMax.length;
const values = A.length / chans;
for (let i = 0; i < values; i++) {
for (let j = 0; j < chans; j++) {
A[i * chans + j] /= uMax[j] === 0 ? 1 : uMax[j];
A[i * chans + j] *= 0.5;
A[i * chans + j] += 0.5;
}
}
}
function divGraphFrom2Curves(curve1, curve2, outUMax) {
const sub = subGraph(curve2, curve1);
maxUnsignedGraphValue(sub, outUMax);
normalizeGraph(sub, outUMax);
return sub;
}
const particleEmitterDeviceCache = new DeviceCache();
class ParticleEmitter {
material = null;
internalTex0 = null;
internalTex1 = null;
internalTex2 = null;
colorParam = null;
constructor(graphicsDevice, options) {
this.graphicsDevice = graphicsDevice;
const gd = graphicsDevice;
const precision = 32;
this.precision = precision;
this._addTimeTime = 0;
setPropertyTarget = this;
setPropertyOptions = options;
setProperty("numParticles", 1);
if (this.numParticles > graphicsDevice.maxTextureSize) {
this.numParticles = graphicsDevice.maxTextureSize;
}
setProperty("rate", 1);
setProperty("rate2", this.rate);
setProperty("lifetime", 50);
setProperty("emitterExtents", new Vec3(0, 0, 0));
setProperty("emitterExtentsInner", new Vec3(0, 0, 0));
setProperty("emitterRadius", 0);
setProperty("emitterRadiusInner", 0);
setProperty("emitterShape", EMITTERSHAPE_BOX);
setProperty("initialVelocity", 1);
setProperty("wrap", false);
setProperty("localSpace", false);
setProperty("screenSpace", false);
setProperty("wrapBounds", null);
setProperty("colorMap", this.defaultParamTexture);
setProperty("normalMap", null);
setProperty("loop", true);
setProperty("preWarm", false);
setProperty("sort", PARTICLESORT_NONE);
setProperty("mode", PARTICLEMODE_GPU);
setProperty("scene", null);
setProperty("lighting", false);
setProperty("halfLambert", false);
setProperty("intensity", 1);
setProperty("stretch", 0);
setProperty("alignToMotion", false);
setProperty("depthSoftening", 0);
setProperty("mesh", null);
setProperty("particleNormal", new Vec3(0, 1, 0));
setProperty("orientation", PARTICLEORIENTATION_SCREEN);
setProperty("depthWrite", false);
setProperty("noFog", false);
setProperty("blendType", BLEND_NORMAL);
setProperty("node", null);
setProperty("startAngle", 0);
setProperty("startAngle2", this.startAngle);
setProperty("animTilesX", 1);
setProperty("animTilesY", 1);
setProperty("animStartFrame", 0);
setProperty("animNumFrames", 1);
setProperty("animNumAnimations", 1);
setProperty("animIndex", 0);
setProperty("randomizeAnimIndex", false);
setProperty("animSpeed", 1);
setProperty("animLoop", true);
this._gpuUpdater = new ParticleGPUUpdater(this, gd);
this._cpuUpdater = new ParticleCPUUpdater(this);
this.emitterPosUniform = new Float32Array(3);
this.wrapBoundsUniform = new Float32Array(3);
this.emitterScaleUniform = new Float32Array([1, 1, 1]);
setProperty("colorGraph", default1Curve3);
setProperty("colorGraph2", this.colorGraph);
setProperty("scaleGraph", default1Curve);
setProperty("scaleGraph2", this.scaleGraph);
setProperty("alphaGraph", default1Curve);
setProperty("alphaGraph2", this.alphaGraph);
setProperty("localVelocityGraph", default0Curve3);
setProperty("localVelocityGraph2", this.localVelocityGraph);
setProperty("velocityGraph", default0Curve3);
setProperty("velocityGraph2", this.velocityGraph);
setProperty("rotationSpeedGraph", default0Curve);
setProperty("rotationSpeedGraph2", this.rotationSpeedGraph);
setProperty("radialSpeedGraph", default0Curve);
setProperty("radialSpeedGraph2", this.radialSpeedGraph);
setPropertyTarget = null;
setPropertyOptions = null;
this.animTilesParams = new Float32Array(2);
this.animParams = new Float32Array(4);
this.animIndexParams = new Float32Array(2);
this.vbToSort = null;
this.vbOld = null;
this.particleDistance = null;
this.camera = null;
this.swapTex = false;
this.useMesh = true;
this.useCpu = !graphicsDevice.supportsGpuParticles;
this.localBounds = new BoundingBox();
this.worldBoundsNoTrail = new BoundingBox();
this.worldBoundsTrail = [new BoundingBox(), new BoundingBox()];
this.worldBounds = new BoundingBox();
this.worldBoundsSize = new Vec3();
this.prevWorldBoundsSize = new Vec3();
this.prevWorldBoundsCenter = new Vec3();
this.prevEmitterExtents = this.emitterExtents;
this.prevEmitterRadius = this.emitterRadius;
this.timeToSwitchBounds = 0;
this.shaderParticleUpdateRespawn = null;
this.shaderParticleUpdateNoRespawn = null;
this.shaderParticleUpdateOnStop = null;
this.numParticleVerts = 0;
this.numParticleIndices = 0;
this.material = null;
this.meshInstance = null;
this.drawOrder = 0;
this.seed = Math.random();
this.fixedTimeStep = 1 / 60;
this.maxSubSteps = 10;
this.simTime = 0;
this.simTimeTotal = 0;
this.beenReset = false;
this._layer = null;
this.rebuild();
}
get defaultParamTexture() {
return particleEmitterDeviceCache.get(this.graphicsDevice, () => {
const resolution = 16;
const centerPoint = resolution * 0.5 + 0.5;
const dtex = new Float32Array(resolution * resolution * 4);
for (let y = 0; y < resolution; y++) {
for (let x = 0; x < resolution; x++) {
const xgrad = x + 1 - centerPoint;
const ygrad = y + 1 - centerPoint;
const c = saturate(1 - saturate(Math.sqrt(xgrad * xgrad + ygrad * ygrad) / resolution) - 0.5);
const p = y * resolution + x;
dtex[p * 4] = 1;
dtex[p * 4 + 1] = 1;
dtex[p * 4 + 2] = 1;
dtex[p * 4 + 3] = c;
}
}
const texture = _createTexture(this.graphicsDevice, resolution, resolution, dtex, PIXELFORMAT_SRGBA8, 1, true);
texture.minFilter = FILTER_LINEAR;
texture.magFilter = FILTER_LINEAR;
return texture;
});
}
calculateWorldBounds() {
if (!this.node) return;
this.prevWorldBoundsSize.copy(this.worldBoundsSize);
this.prevWorldBoundsCenter.copy(this.worldBounds.center);
if (!this.useCpu) {
let recalculateLocalBounds = false;
if (this.emitterShape === EMITTERSHAPE_BOX) {
recalculateLocalBounds = !this.emitterExtents.equals(this.prevEmitterExtents);
} else {
recalculateLocalBounds = !(this.emitterRadius === this.prevEmitterRadius);
}
if (recalculateLocalBounds) {
this.calculateLocalBounds();
}
}
const nodeWT = this.node.getWorldTransform();
if (this.localSpace) {
this.worldBoundsNoTrail.copy(this.localBounds);
} else {
this.worldBoundsNoTrail.setFromTransformedAabb(this.localBounds, nodeWT);
}
this.worldBoundsTrail[0].add(this.worldBoundsNoTrail);
this.worldBoundsTrail[1].add(this.worldBoundsNoTrail);
const now2 = this.simTimeTotal;
if (now2 >= this.timeToSwitchBounds) {
this.worldBoundsTrail[0].copy(this.worldBoundsTrail[1]);
this.worldBoundsTrail[1].copy(this.worldBoundsNoTrail);
this.timeToSwitchBounds = now2 + this.lifetime;
}
this.worldBounds.copy(this.worldBoundsTrail[0]);
this.worldBoundsSize.copy(this.worldBounds.halfExtents).mulScalar(2);
if (this.localSpace) {
this.meshInstance.aabb.setFromTransformedAabb(this.worldBounds, nodeWT);
this.meshInstance.mesh.aabb.setFromTransformedAabb(this.worldBounds, nodeWT);
} else {
this.meshInstance.aabb.copy(this.worldBounds);
this.meshInstance.mesh.aabb.copy(this.worldBounds);
}
this.meshInstance._aabbVer = 1 - this.meshInstance._aabbVer;
}
resetWorldBounds() {
if (!this.node) return;
this.worldBoundsNoTrail.setFromTransformedAabb(
this.localBounds,
this.localSpace ? Mat4.IDENTITY : this.node.getWorldTransform()
);
this.worldBoundsTrail[0].copy(this.worldBoundsNoTrail);
this.worldBoundsTrail[1].copy(this.worldBoundsNoTrail);
this.worldBounds.copy(this.worldBoundsTrail[0]);
this.worldBoundsSize.copy(this.worldBounds.halfExtents).mulScalar(2);
this.prevWorldBoundsSize.copy(this.worldBoundsSize);
this.prevWorldBoundsCenter.copy(this.worldBounds.center);
this.simTimeTotal = 0;
this.timeToSwitchBounds = 0;
}
calculateLocalBounds() {
let minx = Number.MAX_VALUE;
let miny = Number.MAX_VALUE;
let minz = Number.MAX_VALUE;
let maxx = -Number.MAX_VALUE;
let maxy = -Number.MAX_VALUE;
let maxz = -Number.MAX_VALUE;
let maxR = 0;
let maxScale = 0;
const stepWeight = this.lifetime / this.precision;
const wVels = [this.qVelocity, this.qVelocity2];
const lVels = [this.qLocalVelocity, this.qLocalVelocity2];
const accumX = [0, 0];
const accumY = [0, 0];
const accumZ = [0, 0];
const accumR = [0, 0];
const accumW = [0, 0];
let x, y, z;
for (let i = 0; i < this.precision + 1; i++) {
const index = Math.min(i, this.precision - 1);
for (let j = 0; j < 2; j++) {
x = lVels[j][index * 3 + 0] * stepWeight + accumX[j];
y = lVels[j][index * 3 + 1] * stepWeight + accumY[j];
z = lVels[j][index * 3 + 2] * stepWeight + accumZ[j];
minx = Math.min(x, minx);
miny = Math.min(y, miny);
minz = Math.min(z, minz);
maxx = Math.max(x, maxx);
maxy = Math.max(y, maxy);
maxz = Math.max(z, maxz);
accumX[j] = x;
accumY[j] = y;
accumZ[j] = z;
}
for (let j = 0; j < 2; j++) {
accumW[j] += stepWeight * Math.sqrt(
wVels[j][index * 3 + 0] * wVels[j][index * 3 + 0] + wVels[j][index * 3 + 1] * wVels[j][index * 3 + 1] + wVels[j][index * 3 + 2] * wVels[j][index * 3 + 2]
);
}
accumR[0] += this.qRadialSpeed[index] * stepWeight;
accumR[1] += this.qRadialSpeed2[index] * stepWeight;
maxR = Math.max(maxR, Math.max(Math.abs(accumR[0]), Math.abs(accumR[1])));
maxScale = Math.max(maxScale, this.qScale[index]);
}
if (this.emitterShape === EMITTERSHAPE_BOX) {
x = this.emitterExtents.x * 0.5;
y = this.emitterExtents.y * 0.5;
z = this.emitterExtents.z * 0.5;
} else {
x = this.emitterRadius;
y = this.emitterRadius;
z = this.emitterRadius;
}
const w = Math.max(accumW[0], accumW[1]);
bMin.x = minx - maxScale - x - maxR - w;
bMin.y = miny - maxScale - y - maxR - w;
bMin.z = minz - maxScale - z - maxR - w;
bMax.x = maxx + maxScale + x + maxR + w;
bMax.y = maxy + maxScale + y + maxR + w;
bMax.z = maxz + maxScale + z + maxR + w;
this.localBounds.setMinMax(bMin, bMax);
}
rebuild() {
const gd = this.graphicsDevice;
if (this.colorMap === null) this.colorMap = this.defaultParamTexture;
this.spawnBounds = this.emitterShape === EMITTERSHAPE_BOX ? this.emitterExtents : this.emitterRadius;
this.useCpu = this.useCpu || this.sort > PARTICLESORT_NONE || // force CPU if desirable by user or sorting is enabled
gd.forceCpuParticles;
const wasVisible = this._destroyResources();
particleTexHeight = this.useCpu ? 4 : 2;
this.useMesh = !!this.mesh;
this.numParticlesPot = math.nextPowerOfTwo(this.numParticles);
this.rebuildGraphs();
this.calculateLocalBounds();
this.resetWorldBounds();
if (this.node) {
this.worldBounds.setFromTransformedAabb(
this.localBounds,
this.localSpace ? Mat4.IDENTITY : this.node.getWorldTransform()
);
this.worldBoundsTrail[0].copy(this.worldBounds);
this.worldBoundsTrail[1].copy(this.worldBounds);
this.worldBoundsSize.copy(this.worldBounds.halfExtents).mulScalar(2);
this.prevWorldBoundsSize.copy(this.worldBoundsSize);
this.prevWorldBoundsCenter.copy(this.worldBounds.center);
}
this.vbToSort = new Array(this.numParticles);
for (let iSort = 0; iSort < this.numParticles; iSort++) this.vbToSort[iSort] = [0, 0];
this.particleDistance = new Float32Array(this.numParticles);
this._gpuUpdater.randomize();
this.particleTex = new Float32Array(this.numParticlesPot * particleTexHeight * particleTexChannels);
const emitterPos = this.node === null || this.localSpace ? Vec3.ZERO : this.node.getPosition();
if (this.emitterShape === EMITTERSHAPE_BOX) {
if (this.node === null || this.localSpace) {
spawnMatrix.setTRS(Vec3.ZERO, Quat.IDENTITY, this.spawnBounds);
} else {
spawnMatrix.setTRS(Vec3.ZERO, this.node.getRotation(), tmpVec3.copy(this.spawnBounds).mul(this.node.localScale));
}
extentsInnerRatioUniform[0] = this.emitterExtents.x !== 0 ? this.emitterExtentsInner.x / this.emitterExtents.x : 0;
extentsInnerRatioUniform[1] = this.emitterExtents.y !== 0 ? this.emitterExtentsInner.y / this.emitterExtents.y : 0;
extentsInnerRatioUniform[2] = this.emitterExtents.z !== 0 ? this.emitterExtentsInner.z / this.emitterExtents.z : 0;
}
for (let i = 0; i < this.numParticles; i++) {
this._cpuUpdater.calcSpawnPosition(this.particleTex, spawnMatrix, extentsInnerRatioUniform, emitterPos, i);
if (this.useCpu) this.particleTex[i * particleTexChannels + 3 + this.numParticlesPot * 2 * particleTexChannels] = 1;
}
this.particleTexStart = new Float32Array(this.numParticlesPot * particleTexHeight * particleTexChannels);
for (let i = 0; i < this.particleTexStart.length; i++) {
this.particleTexStart[i] = this.particleTex[i];
}
if (!this.useCpu) {
const particleFormat = gd.textureFloatRenderable ? PIXELFORMAT_RGBA32F : PIXELFORMAT_RGBA32U;
this.particleTexIN = _createTexture(gd, this.numParticlesPot, particleTexHeight, this.particleTex, particleFormat);
this.particleTexOUT = _createTexture(gd, this.numParticlesPot, particleTexHeight, this.particleTex, particleFormat);
this.particleTexStart = _createTexture(gd, this.numParticlesPot, particleTexHeight, this.particleTexStart, particleFormat);
this.rtParticleTexIN = new RenderTarget({
colorBuffer: this.particleTexIN,
depth: false
});
this.rtParticleTexOUT = new RenderTarget({
colorBuffer: this.particleTexOUT,
depth: false
});
this.swapTex = false;
}
const defines = /* @__PURE__ */ new Map();
if (this.localSpace) defines.set("LOCAL_SPACE", "");
if (this.emitterShape === EMITTERSHAPE_BOX) defines.set("EMITTERSHAPE_BOX", "");
const shaderUniqueId = `Shape:${this.emitterShape}-Local:${this.localSpace}`;
const shaderOptions = {
attributes: { vertex_position: SEMANTIC_POSITION },
vertexChunk: "fullscreenQuadVS",
fragmentChunk: "particle_simulationPS",
fragmentDefines: defines,
fragmentOutputTypes: gd.textureFloatRenderable ? "vec4" : "uvec4"
};
shaderOptions.uniqueName = `ParticleUpdateRespawn-${shaderUniqueId}`;
defines.set("RESPAWN", "");
this.shaderParticleUpdateRespawn = ShaderUtils.createShader(gd, shaderOptions);
defines.delete("RESPAWN");
shaderOptions.uniqueName = `ParticleUpdateNoRespawn-${shaderUniqueId}`;
defines.set("NO_RESPAWN", "");
this.shaderParticleUpdateNoRespawn = ShaderUtils.createShader(gd, shaderOptions);
defines.delete("NO_RESPAWN");
shaderOptions.uniqueName = `ParticleUpdateStop-${shaderUniqueId}`;
defines.set("ON_STOP", "");
this.shaderParticleUpdateOnStop = ShaderUtils.createShader(gd, shaderOptions);
this.numParticleVerts = this.useMesh ? this.mesh.vertexBuffer.numVertices : 4;
this.numParticleIndices = this.useMesh ? this.mesh.indexBuffer[0].numIndices : 6;
this._allocate(this.numParticles);
const mesh = new Mesh(gd);
mesh.vertexBuffer = this.vertexBuffer;
mesh.indexBuffer[0] = this.indexBuffer;
mesh.primitive[0].type = PRIMITIVE_TRIANGLES;
mesh.primitive[0].base = 0;
mesh.primitive[0].count = this.numParticles * this.numParticleIndices;
mesh.primitive[0].indexed = true;
this.material = this._createMaterial();
this.resetMaterial();
this.meshInstance = new MeshInstance(mesh, this.material, this.node);
this.meshInstance.pick = false;
this.meshInstance.updateKey();
this.meshInstance.cull = true;
if (this.localSpace) {
this.meshInstance.aabb.setFromTransformedAabb(this.worldBounds, this.node.getWorldTransform());
} else {
this.meshInstance.aabb.copy(this.worldBounds);
}
this.meshInstance._updateAabb = false;
this.meshInstance.visible = wasVisible;
this._setMaterialTextures();
this.resetTime();
this.addTime(0, false);
if (this.preWarm) this.prewarm(this.lifetime);
this.finishFrame();
}
_isAnimated() {
return this.animNumFrames >= 1 && (this.animTilesX > 1 || this.animTilesY > 1) && (this.colorMap && this.colorMap !== this.defaultParamTexture || this.normalMap);
}
rebuildGraphs() {
const precision = this.precision;
const gd = this.graphicsDevice;
this.qLocalVelocity = this.localVelocityGraph.quantize(precision);
this.qVelocity = this.velocityGraph.quantize(precision);
this.qColor = this.colorGraph.quantizeClamped(precision, 0, 1);
this.qRotSpeed = this.rotationSpeedGraph.quantize(precision);
this.qScale = this.scaleGraph.quantize(precision);
this.qAlpha = this.alphaGraph.quantize(precision);
this.qRadialSpeed = this.radialSpeedGraph.quantize(precision);
this.qLocalVelocity2 = this.localVelocityGraph2.quantize(precision);
this.qVelocity2 = this.velocityGraph2.quantize(precision);
this.qColor2 = this.colorGraph2.quantizeClamped(precision, 0, 1);
this.qRotSpeed2 = this.rotationSpeedGraph2.quantize(precision);
this.qScale2 = this.scaleGraph2.quantize(precision);
this.qAlpha2 = this.alphaGraph2.quantize(precision);
this.qRadialSpeed2 = this.radialSpeedGraph2.quantize(precision);
for (let i = 0; i < precision; i++) {
this.qRotSpeed[i] *= math.DEG_TO_RAD;
this.qRotSpeed2[i] *= math.DEG_TO_RAD;
}
this.localVelocityUMax = new Float32Array(3);
this.velocityUMax = new Float32Array(3);
this.colorUMax = new Float32Array(3);
this.rotSpeedUMax = [0];
this.scaleUMax = [0];
this.alphaUMax = [0];
this.radialSpeedUMax = [0];
this.qLocalVelocityDiv = divGraphFrom2Curves(this.qLocalVelocity, this.qLocalVelocity2, this.localVelocityUMax);
this.qVelocityDiv = divGraphFrom2Curves(this.qVelocity, this.qVelocity2, this.velocityUMax);
this.qColorDiv = divGraphFrom2Curves(this.qColor, this.qColor2, this.colorUMax);
this.qRotSpeedDiv = divGraphFrom2Curves(this.qRotSpeed, this.qRotSpeed2, this.rotSpeedUMax);
this.qScaleDiv = divGraphFrom2Curves(this.qScale, this.qScale2, this.scaleUMax);
this.qAlphaDiv = divGraphFrom2Curves(this.qAlpha, this.qAlpha2, this.alphaUMax);
this.qRadialSpeedDiv = divGraphFrom2Curves(this.qRadialSpeed, this.qRadialSpeed2, this.radialSpeedUMax);
if (!this.useCpu) {
this.internalTex0 = _createTexture(gd, precision, 1, packTextureXYZ_NXYZ(this.qLocalVelocity, this.qLocalVelocityDiv));
this.internalTex1 = _createTexture(gd, precision, 1, packTextureXYZ_NXYZ(this.qVelocity, this.qVelocityDiv));
this.internalTex2 = _createTexture(gd, precision, 1, packTexture5Floats(this.qRotSpeed, this.qScale, this.qScaleDiv, this.qRotSpeedDiv, this.qAlphaDiv));
this.internalTex3 = _createTexture(gd, precision, 1, packTexture2Floats(this.qRadialSpeed, this.qRadialSpeedDiv));
}
this.colorParam = _createTexture(gd, precision, 1, packTextureRGBA(this.qColor, this.qAlpha), PIXELFORMAT_SRGBA8, 1, true);
}
_setMaterialTextures() {
if (this.colorMap) {
this.material.setParameter("colorMap", this.colorMap);
if (this.lighting && this.normalMap) {
this.material.setParameter("normalMap", this.normalMap);
}
}
}
_createMaterial() {
const material = new ParticleMaterial(this);
material.name = `EmitterMaterial:${this.node.name}`;
material.cull = CULLFACE_NONE;
material.alphaWrite = false;
material.blendType = this.blendType;
material.depthWrite = this.depthWrite;
return material;
}
resetMaterial() {
const material = this.material;
material.setParameter("stretch", this.stretch);
if (this._isAnimated()) {
material.setParameter("animTexTilesParams", this.animTilesParams);
material.setParameter("animTexParams", this.animParams);
material.setParameter("animTexIndexParams", this.animIndexParams);
}
material.setParameter("colorMult", this.intensity);
if (!this.useCpu) {
material.setParameter("internalTex0", this.internalTex0);
material.setParameter("internalTex1", this.internalTex1);
material.setParameter("internalTex2", this.internalTex2);
material.setParameter("internalTex3", this.internalTex3);
}
material.setParameter("colorParam", this.colorParam);
material.setParameter("numParticles", this.numParticles);
material.setParameter("numParticlesPot", this.numParticlesPot);
material.setParameter("lifetime", this.lifetime);
material.setParameter("rate", this.rate);
material.setParameter("rateDiv", this.rate2 - this.rate);
material.setParameter("seed", this.seed);
material.setParameter("scaleDivMult", this.scaleUMax[0]);
material.setParameter("alphaDivMult", this.alphaUMax[0]);
material.setParameter("radialSpeedDivMult", this.radialSpeedUMax[0]);
material.setParameter("graphNumSamples", this.precision);
material.setParameter("graphSampleSize", 1 / this.precision);
material.setParameter("emitterScale", new Float32Array([1, 1, 1]));
if (this.wrap && this.wrapBounds) {
this.wrapBoundsUniform[0] = this.wrapBounds.x;
this.wrapBoundsUniform[1] = this.wrapBounds.y;
this.wrapBoundsUniform[2] = this.wrapBounds.z;
material.setParameter("wrapBounds", this.wrapBoundsUniform);
}
this._setMaterialTextures();
if (this.depthSoftening > 0) {
material.setParameter("softening", 1 / (this.depthSoftening * this.depthSoftening * 100));
}
if (this.stretch > 0) material.cull = CULLFACE_NONE;
this._compParticleFaceParams();
}
_compParticleFaceParams() {
let tangent, binormal;
if (this.orientation === PARTICLEORIENTATION_SCREEN) {
tangent = new Float32Array([1, 0, 0]);
binormal = new Float32Array([0, 0, 1]);
} else {
let n;
if (this.orientation === PARTICLEORIENTATION_WORLD) {
n = this.particleNormal.normalize();
} else {
const emitterMat = this.node === null ? Mat4.IDENTITY : this.node.getWorldTransform();
n = emitterMat.transformVector(this.particleNormal).normalize();
}
const t = new Vec3(1, 0, 0);
if (Math.abs(t.dot(n)) === 1) {
t.set(0, 0, 1);
}
const b = new Vec3().cross(n, t).normalize();
t.cross(b, n).normalize();
tangent = new Float32Array([t.x, t.y, t.z]);
binormal = new Float32Array([b.x, b.y, b.z]);
}
this.material.setParameter("faceTangent", tangent);
this.material.setParameter("faceBinorm", binormal);
}
getVertexInfo() {
const elements = [];
if (!this.useCpu) {
elements.push({ semantic: SEMANTIC_ATTR0, components: 4, type: TYPE_FLOAT32 });
if (this.useMesh) {
elements.push({ semantic: SEMANTIC_TEXCOORD0, components: 2, type: TYPE_FLOAT32 });
}
} else {
elements.push(
{ semantic: SEMANTIC_ATTR0, components: 4, type: TYPE_FLOAT32 },
{ semantic: SEMANTIC_ATTR1, components: 4, type: TYPE_FLOAT32 },
{ semantic: SEMANTIC_ATTR2, components: 4, type: TYPE_FLOAT32 },
{ semantic: SEMANTIC_ATTR3, components: 1, type: TYPE_FLOAT32 },
{ semantic: SEMANTIC_ATTR4, components: this.useMesh ? 4 : 2, type: TYPE_FLOAT32 }
);
}
return elements;
}
// Declares vertex format, creates VB and IB
_allocate(numParticles) {
const psysVertCount = numParticles * this.numParticleVerts;
const psysIndexCount = numParticles * this.numParticleIndices;
if (this.vertexBuffer === void 0 || this.vertexBuffer.getNumVertices() !== psysVertCount) {
const elements = this.getVertexInfo();
const vertexFormat = new VertexFormat(this.graphicsDevice, elements);
this.vertexBuffer = new VertexBuffer(this.graphicsDevice, vertexFormat, psysVertCount, {
usage: BUFFER_DYNAMIC
});
this.indexBuffer = new IndexBuffer(this.graphicsDevice, INDEXFORMAT_UINT32, psysIndexCount);
const data = new Float32Array(this.vertexBuffer.lock());
let meshData, posOffset, posStride, uvOffset, uvStride;
if (this.useMesh) {
meshData = new Float32Array(this.mesh.vertexBuffer.lock());
const elements2 = this.mesh.vertexBuffer.format.elements;
const posElement = elements2.find((el) => el.name === SEMANTIC_POSITION);
const uvElement = elements2.find((el) => el.name === SEMANTIC_TEXCOORD0);
posOffset = posElement.offset / 4;
posStride = posElement.stride / 4;
uvOffset = uvElement ? uvElement.offset / 4 : -1;
uvStride = uvElement ? uvElement.stride / 4 : 0;
}
for (let i = 0; i < psysVertCount; i++) {
const id = Math.floor(i / this.numParticleVerts);
if (!this.useMesh) {
const vertID = i % 4;
data[i * 4] = particleVerts[vertID][0];
data[i * 4 + 1] = particleVerts[vertID][1];
data[i * 4 + 2] = 0;
data[i * 4 + 3] = id;
} else {
const vert = i % this.numParticleVerts;
const posIndex = vert * posStride + posOffset;
data[i * 6] = meshData[posIndex];
data[i * 6 + 1] = meshData[posIndex + 1];
data[i * 6 + 2] = meshData[posIndex + 2];
data[i * 6 + 3] = id;
if (uvOffset >= 0) {
const uvIndex = vert * uvStride + uvOffset;
data[i * 6 + 4] = meshData[uvIndex];
data[i * 6 + 5] = 1 - meshData[uvIndex + 1];
} else {
data[i * 6 + 4] = 0;
data[i * 6 + 5] = 0;
}
}
}
if (this.useCpu) {
this.vbCPU = new Float32Array(data);
this.vbOld = new Float32Array(this.vbCPU.length);
}
this.vertexBuffer.unlock();
if (this.useMesh) {
this.mesh.vertexBuffer.unlock();
}
let dst = 0;
const indices = new Uint32Array(this.indexBuffer.lock());
if (this.useMesh) {
const ib = this.mesh.indexBuffer[0];
meshData = new typedArrayIndexFormats[ib.format](ib.lock());
}
for (let i = 0; i < numParticles; i++) {
if (!this.useMesh) {
const baseIndex = i * 4;
indices[dst++] = baseIndex;
indices[dst++] = baseIndex + 1;
indices[dst++] = baseIndex + 2;
indices[dst++] = baseIndex;
indices[dst++] = baseIndex + 2;
indices[dst++] = baseIndex + 3;
} else {
for (let j = 0; j < this.numParticleIndices; j++) {
indices[i * this.numParticleIndices + j] = meshData[j] + i * this.numParticleVerts;
}
}
}
this.indexBuffer.unlock();
if (this.useMesh) this.mesh.indexBuffer[0].unlock();
}
}
reset() {
this.beenReset = true;
this.seed = Math.random();
this.material.setParameter("seed", this.seed);
if (this.useCpu) {
for (let i = 0; i < this.particleTexStart.length; i++) {
this.particleTex[i] = this.particleTexStart[i];
}
} else {
this._setMaterialTextures();
}
this.resetWorldBounds();
this.resetTime();
const origLoop = this.loop;
this.loop = true;
this.addTime(0, false);
this.loop = origLoop;
if (this.preWarm) {
this.prewarm(this.lifetime);
}
this.finishFrame();
}
prewarm(time) {
const lifetimeFraction = time / this.lifetime;
const iterations = Math.min(Math.floor(lifetimeFraction * this.precision), this.precision);
const stepDelta = time / iterations;
for (let i = 0; i < iterations; i++) {
this.addTime(stepDelta, false);
}
}
resetTime(duration = calcSpawnDuration(this)) {
this.endTime = this.simTimeTotal + duration;
}
finishFrame() {
if (this.useCpu) this.vertexBuffer.unlock();
}
addTime(delta, isOnStop) {
const device = this.graphicsDevice;
this.simTimeTotal += delta;
this.calculateWorldBounds();
if (this._isAnimated()) {
const tilesParams = this.animTilesParams;
tilesParams[0] = 1 / this.animTilesX;
tilesParams[1] = 1 / this.animTilesY;
const params = this.animParams;
params[0] = this.animStartFrame;
params[1] = this.animNumFrames * this.animSpeed;
params[2] = this.animNumFrames - 1;
params[3] = this.animNumAnimations - 1;
const animIndexParams = this.animIndexParams;
animIndexParams[0] = this.animIndex;
animIndexParams[1] = this.randomizeAnimIndex;
}
if (this.scene) {
this.camera = this.scene._activeCamera;
}
if (this.emitterShape === EMITTERSHAPE_BOX) {
extentsInnerRatioUniform[0] = this.emitterExtents.x !== 0 ? this.emitterExtentsInner.x / this.emitterExtents.x : 0;
extentsInnerRatioUniform[1] = this.emitterExtents.y !== 0 ? this.emitterExtentsInner.y / this.emitterExtents.y : 0;
extentsInnerRatioUniform[2] = this.emitterExtents.z !== 0 ? this.emitterExtentsInner.z / this.emitterExtents.z : 0;
if (this.meshInstance.node === null) {
spawnMatrix.setTRS(Vec3.ZERO, Quat.IDENTITY, this.emitterExtents);
} else {
spawnMatrix.setTRS(Vec3.ZERO, this.meshInstance.node.getRotation(), tmpVec3.copy(this.emitterExtents).mul(this.meshInstance.node.localScale));
}
}
let emitterPos;
const emitterScale = this.meshInstance.node === null ? Vec3.ONE : this.meshInstance.node.localScale;
this.emitterScaleUniform[0] = emitterScale.x;
this.emitterScaleUniform[1] = emitterScale.y;
this.emitterScaleUniform[2] = emitterScale.z;
this.material.setParameter("emitterScale", this.emitterScaleUniform);
if (this.localSpace && this.meshInstance.node) {
emitterPos = this.meshInstance.node.getPosition();
this.emitterPosUniform[0] = emitterPos.x;
this.emitterPosUniform[1] = emitterPos.y;
this.emitterPosUniform[2] = emitterPos.z;
this.material.setParameter("emitterPos", this.emitterPosUniform);
}
this._compParticleFaceParams();
if (!this.useCpu) {
this._gpuUpdater.update(device, spawnMatrix, extentsInnerRatioUniform, delta, isOnStop);
} else {
const data = new Float32Array(this.vertexBuffer.lock());
this._cpuUpdater.update(data, this.vbToSort, this.particleTex, spawnMatrix, extentsInnerRatioUniform, emitterPos, delta, isOnStop);
}
if (!this.loop) {
if (this.simTimeTotal > this.endTime) {
if (this.onFinished) this.onFinished();
this.meshInstance.visible = false;
}
}
if (this.meshInstance) {
this.meshInstance.drawOrder = this.drawOrder;
}
}
_destroyResources() {
this.particleTexIN?.destroy();
this.particleTexIN = null;
this.particleTexOUT?.destroy();
this.particleTexOUT = null;
if (this.particleTexStart && this.particleTexStart.destroy) {
this.particleTexStart.destroy();
this.particleTexStart = null;
}
this.rtParticleTexIN?.destroy();
this.rtParticleTexIN = null;
this.rtParticleTexOUT?.destroy();
this.rtParticleTexOUT = null;
this.internalTex0?.destroy();
this.internalTex0 = null;
this.internalTex1?.destroy();
this.internalTex1 = null;
this.internalTex2?.destroy();
this.internalTex2 = null;
this.internalTex3?.destroy();
this.internalTex3 = null;
this.colorParam?.destroy();
this.colorParam = null;
this.vertexBuffer = void 0;
this.indexBuffer = void 0;
const wasVisible = this.meshInstance?.visible ?? true;
this.meshInstance?.destroy();
this.meshInstance = null;
this.material?.destroy();
this.material = null;
return wasVisible;
}
destroy() {
this.camera = null;
this._destroyResources();
}
}
export {
ParticleEmitter
};