From 1118b975450545bb6ac6a30964b0157448e78b82 Mon Sep 17 00:00:00 2001
From: Soar Qin <soarchin@gmail.com>
Date: Fri, 24 Nov 2023 21:48:42 +0800
Subject: [PATCH] WIP

---
 UXAssist/UXAssist.cs | 38 +++++++++++++++++++++++++++++++++++++-
 1 file changed, 37 insertions(+), 1 deletion(-)

diff --git a/UXAssist/UXAssist.cs b/UXAssist/UXAssist.cs
index 55c563b..84fe668 100644
--- a/UXAssist/UXAssist.cs
+++ b/UXAssist/UXAssist.cs
@@ -1,4 +1,5 @@
-using System.Collections.Generic;
+using System;
+using System.Collections.Generic;
 using System.Reflection.Emit;
 using BepInEx;
 using BepInEx.Configuration;
@@ -238,4 +239,39 @@ public class UXAssist : BaseUnityPlugin
         );
         return matcher.InstructionEnumeration();
     }
+
+    [HarmonyPostfix]
+    [HarmonyPatch(typeof(BlueprintUtils), nameof(BlueprintUtils.GenerateBlueprintData))]
+    private static void BlueprintUtils_GenerateBlueprintData_Postfix(BlueprintData _blueprintData)
+    {
+        var buildings = _blueprintData.buildings;
+        Array.Sort(buildings, (a, b) =>
+        {
+            var tmpItemId = a.itemId - b.itemId;
+            if(tmpItemId != 0)
+                return tmpItemId;
+
+            var tmpModelIndex = a.modelIndex - b.modelIndex;
+            if(tmpModelIndex != 0)
+                return tmpModelIndex;
+
+            var tmpRecipeId = a.recipeId - b.recipeId;
+            if(tmpRecipeId != 0)
+                return tmpRecipeId;
+
+            var tmpAreaIndex = a.areaIndex - b.areaIndex;
+            if(tmpAreaIndex != 0)
+                return tmpAreaIndex;
+
+            const double ky = 256.0;
+            const double kx = 1024.0;
+            var scorePosA = (a.localOffset_y * ky + a.localOffset_x) * kx + a.localOffset_z;
+            var scorePosB = (b.localOffset_y * ky + b.localOffset_x) * kx + b.localOffset_z;
+            return scorePosA < scorePosB ? 1 : -1;
+        });
+        for (var i = buildings.Length - 1; i >= 0; i--)
+        {
+            buildings[i].index = i;
+        }
+    }
 }