尝试写了一个异步渲染粒子, 但是显然失败了

CooStack · CooStack · commit bdd93358337e · 2025-05-28T19:40:33.000+08:00
diff --git a/src/main/java/cn/coostack/cooparticlesapi/mixin/ParticleManagerRenderMixin.java b/src/main/java/cn/coostack/cooparticlesapi/mixin/ParticleManagerRenderMixin.java
@@ -0,0 +1,45 @@
+package cn.coostack.cooparticlesapi.mixin;
+
+import cn.coostack.cooparticlesapi.CooParticleAPI;
+import cn.coostack.cooparticlesapi.utils.ParticleAsyncRenderHelper;
+import com.google.common.collect.EvictingQueue;
+import com.llamalad7.mixinextras.sugar.Local;
+import net.minecraft.client.particle.Particle;
+import net.minecraft.client.particle.ParticleManager;
+import net.minecraft.client.particle.ParticleTextureSheet;
+import net.minecraft.client.render.*;
+import net.minecraft.client.texture.TextureManager;
+import org.spongepowered.asm.mixin.Final;
+import org.spongepowered.asm.mixin.Mixin;
+import org.spongepowered.asm.mixin.Shadow;
+import org.spongepowered.asm.mixin.injection.At;
+import org.spongepowered.asm.mixin.injection.ModifyVariable;
+import org.spongepowered.asm.mixin.injection.Redirect;
+
+import java.util.List;
+import java.util.Queue;
+
+/**
+ * TODO
+ * 异步生成BuiltBuffer时
+ * 在主线程draw会导致 buffer no longer valid的异常
+ * 原理未知
+ */
+@Mixin(ParticleManager.class)
+public class ParticleManagerRenderMixin {
+    @Redirect(method = "renderParticles", at = @At(value = "INVOKE", target = "Lnet/minecraft/client/particle/ParticleTextureSheet;begin(Lnet/minecraft/client/render/Tessellator;Lnet/minecraft/client/texture/TextureManager;)Lnet/minecraft/client/render/BufferBuilder;"))
+    public BufferBuilder renderParticles(ParticleTextureSheet instance,
+                                         Tessellator tessellator,
+                                         TextureManager textureManager,
+                                         @Local(ordinal = 0) Queue<Particle> queue,
+                                         @Local(argsOnly = true) Camera camera,
+                                         @Local(argsOnly = true) float tickDelta,
+                                         @Local(ordinal = 0) ParticleTextureSheet sheet) {
+        Object[] array = queue.toArray();
+        // 异步调用...
+        ParticleAsyncRenderHelper.INSTANCE
+                .renderParticlesAsync(array, instance, textureManager, camera, tickDelta);
+        // 中断原来的判断
+        return null;
+    }
+}
diff --git a/src/main/kotlin/cn/coostack/cooparticlesapi/CooParticleAPIClient.kt b/src/main/kotlin/cn/coostack/cooparticlesapi/CooParticleAPIClient.kt
@@ -32,6 +32,7 @@ import cn.coostack.cooparticlesapi.test.particle.style.ExampleSequencedStyle
 import cn.coostack.cooparticlesapi.test.particle.style.ExampleStyle
 import cn.coostack.cooparticlesapi.test.particle.style.RomaMagicTestStyle
 import cn.coostack.cooparticlesapi.test.particle.style.RotateTestStyle
+import cn.coostack.cooparticlesapi.utils.ParticleAsyncRenderHelper
 import net.fabricmc.api.ClientModInitializer
 import net.fabricmc.fabric.api.client.event.lifecycle.v1.ClientTickEvents
 import net.fabricmc.fabric.api.client.event.lifecycle.v1.ClientWorldEvents
@@ -114,16 +115,19 @@ object CooParticleAPIClient : ClientModInitializer {
             ParticleEmittersManager.clientEmitters.clear()
             ParticleStyleManager.clearAllVisible()
             ClientParticleGroupManager.clearAllVisible()
+            ParticleAsyncRenderHelper.close()
         }
         ClientTickEvents.START_WORLD_TICK.register {
             ClientParticleGroupManager.doClientTick()
             ParticleStyleManager.doTickClient()
             ParticleEmittersManager.doTickClient()
+            ParticleAsyncRenderHelper.reloadIfClosed()
         }
         ClientWorldEvents.AFTER_CLIENT_WORLD_CHANGE.register { _, _ ->
             ParticleEmittersManager.clientEmitters.clear()
             ParticleStyleManager.clearAllVisible()
             ClientParticleGroupManager.clearAllVisible()
+            ParticleAsyncRenderHelper.reloadIfClosed()
         }
     }
 
diff --git a/src/main/kotlin/cn/coostack/cooparticlesapi/config/APIConfig.kt b/src/main/kotlin/cn/coostack/cooparticlesapi/config/APIConfig.kt
@@ -15,4 +15,10 @@ class APIConfig {
      */
     var particleCountLimit = 65536
         get() = max(field, 1)
+
+    /**
+     * Math3DUtil的 threadPool最大线程数
+     */
+    var calculateThreadCount = 4
+        get() = max(field, 1)
 }
diff --git a/src/main/kotlin/cn/coostack/cooparticlesapi/items/TestParticleItem.kt b/src/main/kotlin/cn/coostack/cooparticlesapi/items/TestParticleItem.kt
@@ -10,6 +10,7 @@ import cn.coostack.cooparticlesapi.network.particle.emitters.impl.FireClassParti
 import cn.coostack.cooparticlesapi.network.particle.emitters.impl.LightningClassParticleEmitters
 import cn.coostack.cooparticlesapi.network.particle.emitters.impl.PhysicsParticleEmitters
 import cn.coostack.cooparticlesapi.network.particle.emitters.impl.PresetTestEmitters
+import cn.coostack.cooparticlesapi.network.particle.emitters.impl.SimpleParticleEmitters
 import cn.coostack.cooparticlesapi.network.particle.emitters.type.EmittersShootTypes
 import cn.coostack.cooparticlesapi.network.particle.style.ParticleStyleManager
 import cn.coostack.cooparticlesapi.particles.impl.ControlableCloudEffect
@@ -32,14 +33,20 @@ class TestParticleItem(settings: Settings) : Item(settings) {
         if (world.isClient) {
             return TypedActionResult.success(user.getStackInHand(hand))
         }
-//        testRotate(world, user)
+        testRotate(world, user)
 //        CameraUtil.startShakeCamera(240, 0.25)
-        testRomaCircle(world, user)
-
+//        testRomaCircle(world, user)
         // 线性阻力
         return super.use(world, user, hand)
     }
 
+    private fun testLargeParticles(world: World, user: PlayerEntity) {
+        val emitters = SimpleParticleEmitters(user.eyePos, world, ControlableParticleData())
+        emitters.count = 4096
+
+        ParticleEmittersManager.spawnEmitters(emitters)
+    }
+
     private fun testRotate(world: World, user: PlayerEntity) {
         val style = RotateTestStyle(user.uuid)
         ParticleStyleManager.spawnStyle(world as ServerWorld, user.pos, style)
diff --git a/src/main/kotlin/cn/coostack/cooparticlesapi/test/particle/style/RotateTestStyle.kt b/src/main/kotlin/cn/coostack/cooparticlesapi/test/particle/style/RotateTestStyle.kt
@@ -28,6 +28,7 @@ class RotateTestStyle(val player: UUID, uuid: UUID = UUID.randomUUID()) :
 
     override fun getCurrentFrames(): Map<StyleData, RelativeLocation> {
         return PointsBuilder()
+            .addBall(5.0,200)
             .addLine(
                 RelativeLocation(0.0, -2.0, 0.0),
                 RelativeLocation(0.0, 4.0, 0.0),
diff --git a/src/main/kotlin/cn/coostack/cooparticlesapi/utils/Math3DUtil.kt b/src/main/kotlin/cn/coostack/cooparticlesapi/utils/Math3DUtil.kt
@@ -1,15 +1,21 @@
 package cn.coostack.cooparticlesapi.utils
 
+import cn.coostack.cooparticlesapi.config.APIConfig
+import cn.coostack.cooparticlesapi.config.APIConfigManager
 import net.minecraft.util.math.Vec3d
 import org.joml.Quaterniond
 import org.joml.Vector3d
 import org.joml.Vector3f
 import java.util.ArrayList
+import java.util.concurrent.CopyOnWriteArrayList
+import java.util.concurrent.Executors
+import java.util.concurrent.FutureTask
 import kotlin.math.*
 import kotlin.random.Random
 
 object Math3DUtil {
     private val random = Random(System.currentTimeMillis())
+    private val threadPool = Executors.newFixedThreadPool(APIConfigManager.getConfig().calculateThreadCount)
 
     /**
      * 将RGB值转换为Minecraft粒子使用的 rgb值(/255)
@@ -470,8 +476,16 @@ object Math3DUtil {
      * @param angle 角度 输入一个弧度制角度
      */
     fun rotateAsAxis(locList: List<RelativeLocation>, axis: RelativeLocation, angle: Double): List<RelativeLocation> {
+        val q = Quaterniond()
+        q.rotateAxis(angle, Vector3d(axis.x, axis.y, axis.z))
+        val temp = Vector3d(0.0, 0.0, 0.0)
         for (loc in locList) {
-            RotationMatrix.fromAxisAngle(axis, angle).applyTo(loc)
+//            RotationMatrix.fromAxisAngle(axis, angle).applyTo(loc)
+            temp.set(loc.x, loc.y, loc.z)
+            temp.rotate(q)
+            loc.x = temp.x
+            loc.y = temp.y
+            loc.z = temp.z
         }
         return locList
     }
@@ -484,10 +498,39 @@ object Math3DUtil {
         toPoint: RelativeLocation,
         axis: RelativeLocation
     ): List<RelativeLocation> {
+        return rotatePointsToPointAsync(shape, toPoint, axis, 10)
+    }
+
+    /**
+     * 让图形的对称轴指向某个点(图形跟着转变)
+     *
+     * 使用多线程并发修改shape的值 (FutureTask)
+     */
+    fun rotatePointsToPointAsync(
+        shape: List<RelativeLocation>,
+        toPoint: RelativeLocation,
+        axis: RelativeLocation,
+        threads: Int
+    ): List<RelativeLocation> {
+        val copy = CopyOnWriteArrayList(shape)
         // 同向共线
         if (axis.cross(toPoint).length() in -1e-5..1e-5 && axis.dot(toPoint) > 0) {
             return shape
         }
+        if (copy.isEmpty()) return shape
+        var actualThreads = threads
+        if (threads >= copy.size) {
+            actualThreads = copy.size
+        }
+        // 计算每一个线程处理的点的平均个数
+        val taskPreThreadCount = copy.size / actualThreads
+        var notHandledTaskCount = copy.size % actualThreads
+        // 划分索引范围 从0开始
+        // 索引计算规则如下 从0开始 到 taskPreThreadCount + n 结束 左闭右开
+        // 下一个thread就是 taskPreThreadCount + n 开始 n一般为1或者0
+        var currentIndex = 0
+
+        // 计算旋转四元数
         val q = Quaterniond()
         // 差值
         val na = axis.normalize()
@@ -504,14 +547,36 @@ object Math3DUtil {
         val toQ = Quaterniond()
             .rotateY(-toYaw)
             .rotateX(-toPitch)
-        return shape.onEach {
-            val vector = Vector3d(it.x, it.y, it.z)
-            vector.rotate(q)
-            vector.rotate(toQ)
-            it.x = vector.x
-            it.y = vector.y
-            it.z = vector.z
+
+        // 开始分配旋转任务
+        val tasks = ArrayList<FutureTask<Unit>>()
+        repeat(actualThreads) {
+            var next = currentIndex + taskPreThreadCount // 取到  taskHandledIndexStart ..< next
+            if (notHandledTaskCount > 0) {
+                next++
+                notHandledTaskCount--
+            }
+            val taskHandledIndexStart = currentIndex
+            currentIndex = next
+            // 创建任务
+            val vector = Vector3d(0.0, 0.0, 0.0)
+            val task = FutureTask {
+                for (i in taskHandledIndexStart..<next) {
+                    val it = copy[i]
+                    // 复用节约内存
+                    vector.set(it.x, it.y, it.z)
+                    vector.rotate(q)
+                    vector.rotate(toQ)
+                    it.x = vector.x
+                    it.y = vector.y
+                    it.z = vector.z
+                }
+            }
+            threadPool.submit(task)
+            tasks.add(task)
         }
+        tasks.forEach { it.get() }
+        return shape
     }
 
     /**
diff --git a/src/main/kotlin/cn/coostack/cooparticlesapi/utils/ParticleAsyncRenderHelper.kt b/src/main/kotlin/cn/coostack/cooparticlesapi/utils/ParticleAsyncRenderHelper.kt
@@ -0,0 +1,118 @@
+package cn.coostack.cooparticlesapi.utils
+
+import cn.coostack.cooparticlesapi.config.APIConfigManager
+import net.minecraft.client.particle.Particle
+import net.minecraft.client.particle.ParticleTextureSheet
+import net.minecraft.client.render.BufferBuilder
+import net.minecraft.client.render.BufferRenderer
+import net.minecraft.client.render.BuiltBuffer
+import net.minecraft.client.render.Camera
+import net.minecraft.client.render.Tessellator
+import net.minecraft.client.texture.TextureManager
+import net.minecraft.util.crash.CrashException
+import net.minecraft.util.crash.CrashReport
+import org.joml.Vector3d
+import java.util.ArrayList
+import java.util.Arrays
+import java.util.concurrent.Executors
+import java.util.concurrent.FutureTask
+
+object ParticleAsyncRenderHelper {
+    @JvmStatic
+    val threadCount: Int
+        get() = APIConfigManager.getConfig().calculateThreadCount
+    private var threadPool = Executors.newFixedThreadPool(threadCount)
+
+
+    fun close() {
+        threadPool.shutdownNow()
+    }
+
+    fun reloadIfClosed() {
+        if (!threadPool.isShutdown) {
+            return
+        }
+        threadPool = Executors.newFixedThreadPool(threadCount)
+    }
+
+    /**
+     * 异步渲染粒子
+     * TODO 会因为非法访问内存导致崩溃(原因未知)
+     * 好像是顶点渲染的部分导致的
+     */
+    fun renderParticlesAsync(
+        particles: Array<Any>,
+        instance: ParticleTextureSheet,
+        textureManager: TextureManager,
+        camera: Camera,
+        tickDelta: Float
+    ) {
+        var actualThreads = threadCount
+        val size = particles.size
+        if (size <= actualThreads) {
+            actualThreads = particles.size
+        }
+        // 计算每一个线程处理的点的平均个数
+        val taskPreThreadCount = size / actualThreads
+        var notHandledTaskCount = size % actualThreads
+        // 划分索引范围 从0开始
+        // 索引计算规则如下 从0开始 到 taskPreThreadCount + n 结束 左闭右开
+        // 下一个thread就是 taskPreThreadCount + n 开始 n一般为1或者0
+        var currentIndex = 0
+        val tasks = ArrayList<FutureTask<BufferBuilder?>>()
+        repeat(actualThreads) {
+            val tessellator = Tessellator.getInstance()
+            var next = currentIndex + taskPreThreadCount // 取到  taskHandledIndexStart ..< next
+            if (notHandledTaskCount > 0) {
+                next++
+                notHandledTaskCount--
+            }
+            val taskHandledIndexStart = currentIndex
+            currentIndex = next
+            val builder = instance.begin(tessellator, textureManager) ?: return@repeat
+            // 复用子数组 taskHandledIndexStart - next
+            val array = particles.sliceArray(taskHandledIndexStart..<next)
+            // 创建任务
+            val task = submitParticlesRender(
+                array, builder, camera, tickDelta
+            )
+            task ?: return@repeat
+            tasks.add(task)
+        }
+        tasks.forEach {
+            val builder = it.get() ?: return@forEach
+            val built = builder.endNullable() ?: return@forEach
+            BufferRenderer.drawWithGlobalProgram(built)
+        }
+    }
+
+    fun submitParticlesRender(
+        particles: Array<out Any>,
+        builder: BufferBuilder,
+        camera: Camera,
+        tickDelta: Float
+    ): FutureTask<BufferBuilder?>? {
+        val task = FutureTask {
+            particles.forEach { particle ->
+                render(particle, builder, camera, tickDelta)
+            }
+            return@FutureTask builder
+        }
+        threadPool.submit(task)
+        return task
+    }
+
+    // 渲染一个粒子
+    fun render(p: Any, builder: BufferBuilder, camera: Camera, tickDelta: Float) {
+        if (p !is Particle) return
+        try {
+            p.buildGeometry(builder, camera, tickDelta)
+        } catch (throws: Throwable) {
+            val crash = CrashReport.create(throws, "Rendering Particle Async by CooParticlesAPI")
+            val section = crash.addElement("Particle being rendered")
+            section.add("Particle", p::toString)
+            section.add("Particle Type", p.type::toString)
+            throw CrashException(crash)
+        }
+    }
+}
