Merge branch 'master' into 1.16

src/main/java/carpet/logging/HUDController.java

@@ -0,0 +1,138 @@
+package carpet.logging;
+
+import carpet.CarpetServer;
+import carpet.helpers.HopperCounter;
+import carpet.helpers.TickSpeed;
+import carpet.logging.LoggerRegistry;
+import carpet.logging.logHelpers.PacketCounter;
+import carpet.mixins.PlayerListHeaderS2CPacketMixin;
+import carpet.utils.Messenger;
+import carpet.utils.SpawnReporter;
+import net.minecraft.entity.player.PlayerEntity;
+import net.minecraft.server.network.ServerPlayerEntity;
+import net.minecraft.network.packet.s2c.play.PlayerListHeaderS2CPacket;
+import net.minecraft.server.MinecraftServer;
+import net.minecraft.util.math.MathHelper;
+import net.minecraft.text.BaseText;
+import net.minecraft.text.LiteralText;
+import net.minecraft.util.registry.RegistryKey;
+import net.minecraft.world.World;
+import net.minecraft.world.dimension.DimensionType;
+
+import java.util.ArrayList;
+import java.util.Arrays;
+import java.util.HashMap;
+import java.util.List;
+import java.util.Locale;
+import java.util.Map;
+import java.util.function.Consumer;
+
+public class HUDController
+{
+    private static List<Consumer<MinecraftServer>> HUDListeners = new ArrayList<>();
+
+    /**
+     * Adds listener to be called when HUD is updated for logging information
+     * @param listener - a method to be called when new HUD inforation are collected
+     */
+    public void register(Consumer<MinecraftServer> listener)
+    {
+        HUDListeners.add(listener);
+    }
+
+    public static Map<PlayerEntity, List<BaseText>> player_huds = new HashMap<>();
+
+    public static void addMessage(PlayerEntity player, BaseText hudMessage)
+    {
+        if (!player_huds.containsKey(player))
+        {
+            player_huds.put(player, new ArrayList<>());
+        }
+        else
+        {
+            player_huds.get(player).add(new LiteralText("\n"));
+        }
+        player_huds.get(player).add(hudMessage);
+    }
+    public static void clear_player(PlayerEntity player)
+    {
+        PlayerListHeaderS2CPacket packet = new PlayerListHeaderS2CPacket();
+        ((PlayerListHeaderS2CPacketMixin)packet).setHeader(new LiteralText(""));
+        ((PlayerListHeaderS2CPacketMixin)packet).setFooter(new LiteralText(""));
+        ((ServerPlayerEntity)player).networkHandler.sendPacket(packet);
+    }
+
+
+    public static void update_hud(MinecraftServer server)
+    {
+        if(server.getTicks() % 20 != 0)
+            return;
+
+        player_huds.clear();
+
+        if (LoggerRegistry.__tps)
+            LoggerRegistry.getLogger("tps").log(()-> send_tps_display(server));
+
+        if (LoggerRegistry.__mobcaps)
+            LoggerRegistry.getLogger("mobcaps").log((option, player) -> {
+                RegistryKey<World> dim = player.world.getRegistryKey(); //getDimType
+                switch (option)
+                {
+                    case "overworld":
+                        dim = World.OVERWORLD; // OW
+                        break;
+                    case "nether":
+                        dim = World.NETHER; // nether
+                        break;
+                    case "end":
+                        dim = World.END; // end
+                        break;
+                }
+                return new BaseText[]{SpawnReporter.printMobcapsForDimension(server.getWorld(dim), false).get(0)};
+            });
+
+        if(LoggerRegistry.__counter)
+            LoggerRegistry.getLogger("counter").log((option)->send_counter_info(server, option));
+
+        if (LoggerRegistry.__packets)
+            LoggerRegistry.getLogger("packets").log(()-> packetCounter());
+
+        // extensions have time to pitch in.
+        HUDListeners.forEach(l -> l.accept(server));
+
+        for (PlayerEntity player: player_huds.keySet())
+        {
+            PlayerListHeaderS2CPacket packet = new PlayerListHeaderS2CPacket();
+            ((PlayerListHeaderS2CPacketMixin)packet).setHeader(new LiteralText(""));
+            ((PlayerListHeaderS2CPacketMixin)packet).setFooter(Messenger.c(player_huds.get(player).toArray(new Object[0])));
+            ((ServerPlayerEntity)player).networkHandler.sendPacket(packet);
+        }
+    }
+    private static BaseText [] send_tps_display(MinecraftServer server)
+    {
+        double MSPT = MathHelper.average(server.lastTickLengths) * 1.0E-6D;
+        double TPS = 1000.0D / Math.max((TickSpeed.time_warp_start_time != 0)?0.0:TickSpeed.mspt, MSPT);
+        String color = Messenger.heatmap_color(MSPT,TickSpeed.mspt);
+        return new BaseText[]{Messenger.c(
+                "g TPS: ", String.format(Locale.US, "%s %.1f",color, TPS),
+                "g  MSPT: ", String.format(Locale.US,"%s %.1f", color, MSPT))};
+    }
+
+    private static BaseText [] send_counter_info(MinecraftServer server, String color)
+    {
+        List <BaseText> res = new ArrayList<>();
+        Arrays.asList(color.split(",")).forEach(c ->{
+            HopperCounter counter = HopperCounter.getCounter(c);
+            if (counter != null) res.addAll(counter.format(server, false, true));
+        });
+        return res.toArray(new BaseText[0]);
+    }
+    private static BaseText [] packetCounter()
+    {
+        BaseText [] ret =  new BaseText[]{
+                Messenger.c("w I/" + PacketCounter.totalIn + " O/" + PacketCounter.totalOut),
+        };
+        PacketCounter.reset();
+        return ret;
+    }
+}