From f0ef1096587e6103c5c2b8a6de4f03ce7bdd88e3 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Manuel=20V=C3=B6gele?= <develop@manuel-voegele.de>
Date: Thu, 29 Apr 2021 00:03:05 +0200
Subject: [PATCH] Track the movement of all tokens that are moved
 simultaneously in one batch (huge performance bump if many tokens are moved)

---
 src/foundry_imports.js   |  2 +-
 src/movement_tracking.js | 18 +++++++++++++-----
 src/socket.js            | 12 +++++++-----
 3 files changed, 21 insertions(+), 11 deletions(-)

diff --git a/src/foundry_imports.js b/src/foundry_imports.js
index b39bcb0..eec7977 100644
--- a/src/foundry_imports.js
+++ b/src/foundry_imports.js
@@ -64,8 +64,8 @@ async function animateTokens(tokens, draggedToken, draggedRays, wasPaused) {
 		return {token, rays: offsetRays, dx, dy};
 	});
 
+	trackRays(tokenAnimationData.map(({token}) => token), tokenAnimationData.map(({rays}) => rays));
 	for (const {token, rays} of tokenAnimationData) {
-		trackRays(token, rays);
 		token._noAnimate = true;
 	}
 	for (let i = 0;i < tokenAnimationData[0].rays.length; i++) {
diff --git a/src/movement_tracking.js b/src/movement_tracking.js
index 940e4df..4b2528b 100644
--- a/src/movement_tracking.js
+++ b/src/movement_tracking.js
@@ -1,6 +1,6 @@
 import {measureDistances} from "./compatibility.js";
 import {updateCombatantDragRulerFlags} from "./socket.js";
-import {getTokenShape} from "./util.js";
+import {getTokenShape, zip} from "./util.js";
 
 function initTrackingFlag(combatant) {
 	const initialFlag = {passedWaypoints: [], trackedRound: 0};
@@ -23,13 +23,21 @@ function getInitializedCombatant(token, combat) {
 	return combatant;
 }
 
-export async function trackRays(token, rays) {
-	// Only track movement if the current token is participating in the active combat
+export async function trackRays(tokens, tokenRays) {
 	const combat = game.combat;
 	if (!combat)
 		return;
 	if (!combat.started)
 		return;
+	if (!(tokens instanceof Array)) {
+		tokens = [tokens];
+		tokenRays = [tokenRays];
+	}
+	const updates = Array.from(zip(tokens, tokenRays)).map(([token, rays]) => calculateUpdate(combat, token, rays)).filter(Boolean);
+	await updateCombatantDragRulerFlags(combat, updates);
+}
+
+function calculateUpdate(combat, token, rays) {
 	const combatant = getInitializedCombatant(token, combat);
 	if (!combatant)
 		return;
@@ -55,7 +63,7 @@ export async function trackRays(token, rays) {
 			waypoints.push(ray.A);
 		}
 	}
-	await updateCombatantDragRulerFlags(combat, combatant, dragRulerFlags);
+	return {_id: combatant._id, dragRulerFlags};
 }
 
 export function getMovementHistory(token) {
@@ -81,5 +89,5 @@ export async function resetMovementHistory(combat, combatantId) {
 	dragRulerFlags.passedWaypoints = undefined;
 	dragRulerFlags.trackedRound = undefined;
 	dragRulerFlags.rulerState = undefined;
-	await updateCombatantDragRulerFlags(combat, combatant, dragRulerFlags);
+	await updateCombatantDragRulerFlags(combat, [{_id: combatantId, dragRulerFlags}]);
 }
diff --git a/src/socket.js b/src/socket.js
index a8c6b95..6a8d460 100644
--- a/src/socket.js
+++ b/src/socket.js
@@ -5,13 +5,15 @@ Hooks.once("socketlib.ready", () => {
 	socket.register("updateCombatantDragRulerFlags", _socketUpdateCombatantDragRulerFlags);
 });
 
-export function updateCombatantDragRulerFlags(combat, combatant, flags) {
+export function updateCombatantDragRulerFlags(combat, updates) {
 	const combatId = combat.id;
-	const combatantId = combatant._id;
-	return socket.executeAsGM(_socketUpdateCombatantDragRulerFlags, combatId, combatantId, flags);
+	return socket.executeAsGM(_socketUpdateCombatantDragRulerFlags, combatId, updates);
 }
 
-async function _socketUpdateCombatantDragRulerFlags(combatId, combatantId, flags) {
+async function _socketUpdateCombatantDragRulerFlags(combatId, updates) {
 	const combat = game.combats.get(combatId);
-	await combat.updateEmbeddedEntity("Combatant", {_id: combatantId, flags: {dragRuler: flags}}, {diff: false});
+	updates = updates.map(update => {
+		return {_id: update._id, flags: {dragRuler: update.dragRulerFlags}};
+	});
+	await combat.updateEmbeddedEntity("Combatant", updates, {diff: false});
 }