Add new particle renderer (#1682)

* feat: add new particle renderer
galacean · Sep 7, 2023 · a1ce393 · a1ce393
1 parent c4cbb2c
commit a1ce393
Show file tree

Hide file tree

Showing 88 changed files with 4,323 additions and 1,401 deletions.
diff --git a/packages/core/src/Camera.ts b/packages/core/src/Camera.ts
@@ -37,6 +37,8 @@ export class Camera extends Component {
 
   private static _inverseViewMatrixProperty = ShaderProperty.getByName("camera_ViewInvMat");
   private static _cameraPositionProperty = ShaderProperty.getByName("camera_Position");
+  private static _cameraForwardProperty = ShaderProperty.getByName("camera_Forward");
+  private static _cameraUpProperty = ShaderProperty.getByName("camera_Up");
   private static _cameraDepthBufferParamsProperty = ShaderProperty.getByName("camera_DepthBufferParams");
 
   /** Whether to enable frustum culling, it is enabled by default. */
@@ -625,13 +627,17 @@ export class Camera extends Component {
   }
 
   private _updateShaderData(): void {
+    const shaderData = this.shaderData;
+
+    const transform = this._transform;
+    shaderData.setMatrix(Camera._inverseViewMatrixProperty, transform.worldMatrix);
+    shaderData.setVector3(Camera._cameraPositionProperty, transform.worldPosition);
+    shaderData.setVector3(Camera._cameraForwardProperty, transform.worldForward);
+    shaderData.setVector3(Camera._cameraUpProperty, transform.worldUp);
+
     const depthBufferParams = this._depthBufferParams;
     const farDivideNear = this._farClipPlane / this._nearClipPlane;
     depthBufferParams.set(1.0 - farDivideNear, farDivideNear, 0, 0);
-
-    const shaderData = this.shaderData;
-    shaderData.setMatrix(Camera._inverseViewMatrixProperty, this._transform.worldMatrix);
-    shaderData.setVector3(Camera._cameraPositionProperty, this._transform.worldPosition);
     shaderData.setVector4(Camera._cameraDepthBufferParamsProperty, depthBufferParams);
   }
 

diff --git a/packages/core/src/Engine.ts b/packages/core/src/Engine.ts
@@ -5,8 +5,8 @@ import { Canvas } from "./Canvas";
 import { EngineSettings } from "./EngineSettings";
 import { Entity } from "./Entity";
 import { ClassPool } from "./RenderPipeline/ClassPool";
-import { MeshRenderData } from "./RenderPipeline/MeshRenderData";
 import { RenderContext } from "./RenderPipeline/RenderContext";
+import { RenderData } from "./RenderPipeline/RenderData";
 import { RenderElement } from "./RenderPipeline/RenderElement";
 import { SpriteMaskManager } from "./RenderPipeline/SpriteMaskManager";
 import { SpriteMaskRenderData } from "./RenderPipeline/SpriteMaskRenderData";
@@ -21,6 +21,7 @@ import { GLCapabilityType } from "./base/Constant";
 import { ColorSpace } from "./enums/ColorSpace";
 import { InputManager } from "./input";
 import { Material } from "./material/Material";
+import { ParticleBufferUtils } from "./particle/ParticleBufferUtils";
 import { PhysicsScene } from "./physics/PhysicsScene";
 import { ColliderShape } from "./physics/shape/ColliderShape";
 import { IHardwareRenderer } from "./renderingHardwareInterface";
@@ -55,6 +56,8 @@ export class Engine extends EventDispatcher {
   /** Input manager of Engine. */
   readonly inputManager: InputManager;
 
+  /** @internal */
+  _particleBufferUtils: ParticleBufferUtils;
   /** @internal */
   _physicsInitialized: boolean = false;
   /** @internal */
@@ -69,7 +72,7 @@ export class Engine extends EventDispatcher {
   /* @internal */
   _renderElementPool: ClassPool<RenderElement> = new ClassPool(RenderElement);
   /* @internal */
-  _meshRenderDataPool: ClassPool<MeshRenderData> = new ClassPool(MeshRenderData);
+  _renderDataPool: ClassPool<RenderData> = new ClassPool(RenderData);
   /* @internal */
   _spriteRenderDataPool: ClassPool<SpriteRenderData> = new ClassPool(SpriteRenderData);
   /* @internal */
@@ -263,6 +266,8 @@ export class Engine extends EventDispatcher {
     const colorSpace = configuration.colorSpace || ColorSpace.Linear;
     colorSpace === ColorSpace.Gamma && this._macroCollection.enable(Engine._gammaMacro);
     innerSettings.colorSpace = colorSpace;
+
+    this._particleBufferUtils = new ParticleBufferUtils(this);
   }
 
   /**
@@ -308,7 +313,7 @@ export class Engine extends EventDispatcher {
     this._frameInProcess = true;
 
     this._renderElementPool.resetPool();
-    this._meshRenderDataPool.resetPool();
+    this._renderDataPool.resetPool();
     this._spriteRenderDataPool.resetPool();
     this._spriteMaskRenderDataPool.resetPool();
     this._textRenderDataPool.resetPool();
@@ -644,6 +649,8 @@ export class Engine extends EventDispatcher {
 
   private _onDeviceLost(): void {
     this._isDeviceLost = true;
+    // Lose graphic resources
+    this.resourceManager._lostGraphicResources();
     console.log("Device lost.");
     this.dispatch("devicelost", this);
   }
@@ -660,6 +667,7 @@ export class Engine extends EventDispatcher {
     console.log("Graphic resource restored.");
 
     // Restore resources content
+    this._particleBufferUtils.setBufferData();
     resourceManager
       ._restoreResourcesContent()
       .then(() => {
@@ -674,7 +682,7 @@ export class Engine extends EventDispatcher {
 
   private _gc(): void {
     this._renderElementPool.garbageCollection();
-    this._meshRenderDataPool.garbageCollection();
+    this._renderDataPool.garbageCollection();
     this._spriteRenderDataPool.garbageCollection();
     this._spriteMaskRenderDataPool.garbageCollection();
     this._textRenderDataPool.garbageCollection();

diff --git a/packages/core/src/RenderPipeline/MeshRenderData.ts b/packages/core/src/RenderPipeline/MeshRenderData.ts
diff --git a/packages/core/src/RenderPipeline/RenderData.ts b/packages/core/src/RenderPipeline/RenderData.ts
@@ -1,9 +1,29 @@
+import { SubMesh } from "../graphic";
+import { Primitive } from "../graphic/Primitive";
 import { Material } from "../material";
 import { Renderer } from "../Renderer";
+import { IPoolElement } from "./IPoolElement";
 
-export class RenderData {
+export class RenderData implements IPoolElement {
   component: Renderer;
   material: Material;
+  primitive: Primitive;
+  subPrimitive: SubMesh;
 
   multiRenderData: boolean;
+
+  setX(component: Renderer, material: Material, primitive: Primitive, subPrimitive: SubMesh): void {
+    this.component = component;
+    this.material = material;
+
+    this.primitive = primitive;
+    this.subPrimitive = subPrimitive;
+  }
+
+  dispose(): void {
+    this.component = null;
+    this.material = null;
+    this.primitive = null;
+    this.subPrimitive = null;
+  }
 }
diff --git a/packages/core/src/RenderPipeline/RenderQueue.ts b/packages/core/src/RenderPipeline/RenderQueue.ts
@@ -3,7 +3,6 @@ import { Engine } from "../Engine";
 import { Layer } from "../Layer";
 import { Shader } from "../shader";
 import { ShaderMacroCollection } from "../shader/ShaderMacroCollection";
-import { MeshRenderData } from "./MeshRenderData";
 import { RenderContext } from "./RenderContext";
 import { RenderElement } from "./RenderElement";
 import { SpriteBatcher } from "./SpriteBatcher";
@@ -70,13 +69,13 @@ export class RenderQueue {
         continue;
       }
 
-      if (!!(data as MeshRenderData).mesh) {
+      if (data.primitive) {
         this._spriteBatcher.flush(camera);
 
         const compileMacros = Shader._compileMacros;
-        const meshData = <MeshRenderData>data;
-        const renderer = meshData.component;
-        const material = meshData.material.destroyed ? engine._magentaMaterial : meshData.material;
+        const primitive = data.primitive;
+        const renderer = data.component;
+        const material = data.material.destroyed ? engine._magentaMaterial : data.material;
         const rendererData = renderer.shaderData;
         const materialData = material.shaderData;
 
@@ -157,7 +156,7 @@ export class RenderQueue {
             material.shaderData
           );
 
-          rhi.drawPrimitive(meshData.mesh, meshData.subMesh, program);
+          rhi.drawPrimitive(primitive, data.subPrimitive, program);
         }
       } else {
         this._spriteBatcher.drawElement(element, camera);

diff --git a/packages/core/src/RenderPipeline/SpriteBatcher.ts b/packages/core/src/RenderPipeline/SpriteBatcher.ts
@@ -126,7 +126,7 @@ export class SpriteBatcher extends Basic2DBatcher {
       program.uploadAll(program.materialUniformBlock, material.shaderData);
 
       material.renderState._apply(engine, false, shaderPass._renderStateDataMap, material.shaderData);
-      engine._hardwareRenderer.drawPrimitive(mesh, subMesh, program);
+      engine._hardwareRenderer.drawPrimitive(mesh._primitive, subMesh, program);
 
       maskManager.postRender(renderer);
     }

diff --git a/packages/core/src/RenderPipeline/SpriteMaskRenderData.ts b/packages/core/src/RenderPipeline/SpriteMaskRenderData.ts
@@ -19,7 +19,7 @@ export class SpriteMaskRenderData extends RenderData implements IPoolElement {
     this.verticesData = verticesData;
   }
 
-  dispose(): void {
+  override dispose(): void {
     this.component = this.material = this.verticesData = null;
   }
 }
diff --git a/packages/core/src/RenderPipeline/SpriteRenderData.ts b/packages/core/src/RenderPipeline/SpriteRenderData.ts
@@ -30,7 +30,7 @@ export class SpriteRenderData extends RenderData implements IPoolElement {
     this.dataIndex = dataIndex;
   }
 
-  dispose(): void {
+  override dispose(): void {
     this.component = this.material = this.verticesData = this.texture = null;
   }
 }
diff --git a/packages/core/src/RenderPipeline/TextRenderData.ts b/packages/core/src/RenderPipeline/TextRenderData.ts
@@ -10,7 +10,7 @@ export class TextRenderData extends RenderData implements IPoolElement {
     this.multiRenderData = true;
   }
 
-  dispose(): void {
+  override dispose(): void {
     this.component = this.material = null;
     this.charsData.length = 0;
   }

diff --git a/packages/core/src/Renderer.ts b/packages/core/src/Renderer.ts
@@ -6,7 +6,7 @@ import { Entity } from "./Entity";
 import { RenderContext } from "./RenderPipeline/RenderContext";
 import { Transform, TransformModifyFlags } from "./Transform";
 import { assignmentClone, deepClone, ignoreClone } from "./clone/CloneManager";
-import { ICustomClone } from "./clone/ComponentCloner";
+import { IComponentCustomClone } from "./clone/ComponentCloner";
 import { Material } from "./material";
 import { ShaderMacro, ShaderProperty } from "./shader";
 import { ShaderData } from "./shader/ShaderData";
@@ -18,7 +18,7 @@ import { ShaderDataGroup } from "./shader/enums/ShaderDataGroup";
  * @decorator `@dependentComponents(Transform, DependentMode.CheckOnly)`
  */
 @dependentComponents(Transform, DependentMode.CheckOnly)
-export class Renderer extends Component implements ICustomClone {
+export class Renderer extends Component implements IComponentCustomClone {
   private static _tempVector0 = new Vector3();
 
   private static _receiveShadowMacro = ShaderMacro.getByName("RENDERER_IS_RECEIVE_SHADOWS");
@@ -67,7 +67,7 @@ export class Renderer extends Component implements ICustomClone {
   private _normalMatrix: Matrix = new Matrix();
   @ignoreClone
   private _materialsInstanced: boolean[] = [];
-  @ignoreClone
+  @assignmentClone
   private _priority: number = 0;
   @assignmentClone
   private _receiveShadows: boolean = true;

diff --git a/packages/core/src/animation/animationCurve/AnimationCurve.ts b/packages/core/src/animation/animationCurve/AnimationCurve.ts
@@ -82,7 +82,7 @@ export abstract class AnimationCurve<V extends KeyframeValueType> {
     let newLength = 0;
     for (let i = keys.length - 1; i >= 0; i--) {
       const key = keys[i];
-      if (key.time > length) {
+      if (key.time > this._length) {
         newLength = key.time;
       }
     }

diff --git a/packages/core/src/asset/GraphicsResource.ts b/packages/core/src/asset/GraphicsResource.ts
@@ -2,6 +2,16 @@ import { Engine } from "../Engine";
 import { ReferResource } from "./ReferResource";
 
 export abstract class GraphicsResource extends ReferResource {
+  /** @internal */
+  _isContentLost: boolean = false;
+
+  /**
+   * Whether the content of the resource is lost.
+   */
+  get isContentLost(): boolean {
+    return this._isContentLost;
+  }
+
   protected constructor(engine: Engine) {
     super(engine);
     engine.resourceManager._addGraphicResource(this);

diff --git a/packages/core/src/asset/ResourceManager.ts b/packages/core/src/asset/ResourceManager.ts
@@ -258,6 +258,16 @@ export class ResourceManager {
     }
   }
 
+  /**
+   * @internal
+   */
+  _lostGraphicResources(): void {
+    const graphicResourcePool = this._graphicResourcePool;
+    for (const id in graphicResourcePool) {
+      graphicResourcePool[id]._isContentLost = true;
+    }
+  }
+
   /**
    * @internal
    */