From 0000000000000000000000000000000000000000 Mon Sep 17 00:00:00 2001
From: Taiyou06 <kaandindar21@gmail.com>
Date: Tue, 9 Nov 2077 00:00:00 +0800
Subject: [PATCH] Collect then startEachNonRunningBehavior in Brain


diff --git a/net/minecraft/world/entity/ai/Brain.java b/net/minecraft/world/entity/ai/Brain.java
index b946faf899561b9e4f8399e16600eede54bb55a5..b2e17d387558875f45a8458cf874a5b056820432 100644
--- a/net/minecraft/world/entity/ai/Brain.java
+++ b/net/minecraft/world/entity/ai/Brain.java
@@ -462,20 +462,29 @@ public class Brain<E extends LivingEntity> {
     }
 
     private void startEachNonRunningBehavior(ServerLevel level, E entity) {
-        long gameTime = level.getGameTime();
+        // Leaf start - Collect then startEachNonRunningBehavior in Brain
+        final long gameTime = level.getGameTime();
+        List<BehaviorControl<? super E>> behaviorsToStart = new ObjectArrayList<>();
 
-        for (Map<Activity, Set<BehaviorControl<? super E>>> map : this.availableBehaviorsByPriority.values()) {
-            for (Entry<Activity, Set<BehaviorControl<? super E>>> entry : map.entrySet()) {
-                Activity activity = entry.getKey();
-                if (this.activeActivities.contains(activity)) {
-                    for (BehaviorControl<? super E> behaviorControl : entry.getValue()) {
+        for (Activity activeActivity : this.activeActivities) {
+            for (Map<Activity, Set<BehaviorControl<? super E>>> priorityMap : this.availableBehaviorsByPriority.values()) {
+                Set<BehaviorControl<? super E>> behaviors = priorityMap.get(activeActivity);
+
+                if (behaviors != null && !behaviors.isEmpty()) {
+                    for (BehaviorControl<? super E> behaviorControl : behaviors) {
                         if (behaviorControl.getStatus() == Behavior.Status.STOPPED) {
-                            behaviorControl.tryStart(level, entity, gameTime);
+                            behaviorsToStart.add(behaviorControl);
                         }
                     }
                 }
             }
         }
+        if (!behaviorsToStart.isEmpty()) {
+            for (BehaviorControl<? super E> behaviorControl : behaviorsToStart) {
+                behaviorControl.tryStart(level, entity, gameTime);
+            }
+        }
+        // Leaf end - Collect then startEachNonRunningBehavior in Brain
     }
 
     private void tickEachRunningBehavior(ServerLevel level, E entity) {