feat: added a particle callback function that will be executed when the particle is removed

NexaCore168 · MaTJ · web-flow · commit f3b52e3780f0 · 2023-12-28T12:54:23.000Z
---------

Co-authored-by: MaTJ &lt;tengjiao.ma@weimob.com&gt;
diff --git a/fxgl-entity/src/main/java/com/almasb/fxgl/particle/ParticleEmitter.java b/fxgl-entity/src/main/java/com/almasb/fxgl/particle/ParticleEmitter.java
@@ -249,8 +249,22 @@ public void setParametricEquation(Function<Double, Point2D> parametricEquation)
         this.parametricEquation = parametricEquation;
     }
 
-    private Supplier<Point2D> accelerationFunction = () -> Point2D.ZERO;
+    private Consumer<Particle> onRemoveFunction = particle -> {};
 
+    public Consumer<Particle> getOnRemoveFunction() {
+        return onRemoveFunction;
+    }
+    
+    /**
+     *
+     * @param onRemoveFunction this function is executed when the particle is removed
+     */
+    public void setOnRemoveFunction(Consumer<Particle> onRemoveFunction) {
+        this.onRemoveFunction = onRemoveFunction;
+    }
+    
+    private Supplier<Point2D> accelerationFunction = () -> Point2D.ZERO;
+    
     /**
      *
      * @return gravity function
diff --git a/fxgl-entity/src/main/kotlin/com/almasb/fxgl/particle/ParticleComponent.kt b/fxgl-entity/src/main/kotlin/com/almasb/fxgl/particle/ParticleComponent.kt
@@ -46,6 +46,7 @@ open class ParticleComponent(val emitter: ParticleEmitter) : Component() {
             val p = iter.next()
 
             if (p.update(tpf)) {
+                emitter.getOnRemoveFunction().accept(p)
                 iter.remove()
 
                 parent.viewComponent.removeChild(p.view)
@@ -77,4 +78,4 @@ open class ParticleComponent(val emitter: ParticleEmitter) : Component() {
     }
 
     override fun isComponentInjectionRequired(): Boolean = false
-}
+}
