Feature: Starred Mob Highlight + Fels Highlight/Line (#1558)

Co-authored-by: Cal <cwolfson58@gmail.com>
Co-authored-by: hannibal2 <24389977+hannibal00212@users.noreply.github.com>

4 files changed, 241 insertions, 6 deletions

diff --git a/src/main/java/at/hannibal2/skyhanni/config/features/dungeon/DungeonConfig.java b/src/main/java/at/hannibal2/skyhanni/config/features/dungeon/DungeonConfig.java
index b76039da1..1729623d2 100644
--- a/src/main/java/at/hannibal2/skyhanni/config/features/dungeon/DungeonConfig.java
+++ b/src/main/java/at/hannibal2/skyhanni/config/features/dungeon/DungeonConfig.java
@@ -68,6 +68,11 @@ public class DungeonConfig {
     public boolean architectNotifier = true;
 
     @Expose
+    @ConfigOption(name = "Object Highlighter", desc = "Highlights various things in Dungeons.")
+    @Accordion
+    public ObjectHighlighterConfig objectHighlighter = new ObjectHighlighterConfig();
+
+    @Expose
     @ConfigOption(name = "Object Hider", desc = "Hide various things in Dungeons.")
     @Accordion
     public ObjectHiderConfig objectHider = new ObjectHiderConfig();
diff --git a/src/main/java/at/hannibal2/skyhanni/config/features/dungeon/ObjectHighlighterConfig.java b/src/main/java/at/hannibal2/skyhanni/config/features/dungeon/ObjectHighlighterConfig.java
new file mode 100644
index 000000000..385c162c6
--- /dev/null
+++ b/src/main/java/at/hannibal2/skyhanni/config/features/dungeon/ObjectHighlighterConfig.java
@@ -0,0 +1,72 @@
+package at.hannibal2.skyhanni.config.features.dungeon;
+
+import at.hannibal2.skyhanni.config.FeatureToggle;
+import com.google.gson.annotations.Expose;
+import io.github.notenoughupdates.moulconfig.annotations.Accordion;
+import io.github.notenoughupdates.moulconfig.annotations.ConfigEditorBoolean;
+import io.github.notenoughupdates.moulconfig.annotations.ConfigEditorColour;
+import io.github.notenoughupdates.moulconfig.annotations.ConfigEditorInfoText;
+import io.github.notenoughupdates.moulconfig.annotations.ConfigOption;
+import io.github.notenoughupdates.moulconfig.observer.Property;
+
+public class ObjectHighlighterConfig {
+
+    // TODO move some stuff from DungeonConfig into this
+
+    @Expose
+    @ConfigOption(name = "Stared", desc = "")
+    @Accordion
+    public StarredConfig starred = new StarredConfig();
+
+    public static class StarredConfig {
+        @Expose
+        @ConfigOption(name = "Highlight Starred", desc = "Highlights all starred mobs in one colour.")
+        @ConfigEditorBoolean
+        @FeatureToggle
+        public Property<Boolean> highlight = Property.of(true);
+
+        /*
+        TODO for someone who has time
+        @Expose
+        @ConfigOption(name = "Show Outline", desc = "Shows only a outline instead of a full highlight.")
+        @ConfigEditorBoolean
+        public Property<Boolean> showOutline = Property.of(true); */
+
+        @ConfigOption(
+            name = "No Chroma",
+            desc = "§cThe chroma setting for the color is currently not working!"
+        )
+        @ConfigEditorInfoText
+        public String info;
+
+        @Expose
+        @ConfigOption(name = "Colour", desc = "Color in which the stared mobs are highlighted.")
+        @ConfigEditorColour
+        public Property<String> colour = Property.of("0:60:255:255:0");
+    }
+
+    @Expose
+    @ConfigOption(name = "Fels Skull", desc = "")
+    @Accordion
+    public FelConfig fel = new FelConfig();
+
+    public static class FelConfig {
+
+        @Expose
+        @ConfigOption(name = "Highlight Fels Skull", desc = "Highlights fels that are not yet active.")
+        @ConfigEditorBoolean
+        @FeatureToggle
+        public Property<Boolean> highlight = Property.of(true);
+
+        @Expose
+        @ConfigOption(name = "Draw Line", desc = "Draws a line to fels skulls. Works only if the highlight is enabled.")
+        @ConfigEditorBoolean
+        public Boolean line = false;
+
+        @Expose
+        @ConfigOption(name = "Colour", desc = "Color for the fel skull and line.")
+        @ConfigEditorColour
+        public Property<String> colour = Property.of("0:200:255:0:255");
+    }
+
+}
diff --git a/src/main/java/at/hannibal2/skyhanni/data/mob/Mob.kt b/src/main/java/at/hannibal2/skyhanni/data/mob/Mob.kt
index 1d5827490..7e7e9f96c 100644
--- a/src/main/java/at/hannibal2/skyhanni/data/mob/Mob.kt
+++ b/src/main/java/at/hannibal2/skyhanni/data/mob/Mob.kt
@@ -115,10 +115,17 @@ class Mob(
 
     private var highlightColor: Color? = null
 
-    /** If no alpha is set or alpha is set to 255 it will set the alpha to 127 */
-    fun highlight(color: Color) {
-        highlightColor = color.takeIf { it.alpha == 255 }?.addAlpha(127) ?: color
-        internalHighlight()
+    /** If [color] has no alpha or alpha is set to 255 it will set the alpha to 127
+     * If [color] is set to null it removes a highlight*/
+    fun highlight(color: Color?) {
+        if (color == highlightColor) return
+        if (color == null) {
+            internalRemoveColor()
+            highlightColor = null
+        } else {
+            highlightColor = color.takeIf { it.alpha == 255 }?.addAlpha(127) ?: color
+            internalHighlight()
+        }
     }
 
     private fun internalHighlight() {
@@ -162,8 +169,10 @@ class Mob(
     }
 
     private fun makeRelativeBoundingBox() =
-        (baseEntity.entityBoundingBox.union(extraEntities.filter { it !is EntityArmorStand }
-            .mapNotNull { it.entityBoundingBox }))?.offset(-baseEntity.posX, -baseEntity.posY, -baseEntity.posZ)
+        (baseEntity.entityBoundingBox.union(
+            extraEntities.filter { it !is EntityArmorStand }
+                .mapNotNull { it.entityBoundingBox },
+        ))?.offset(-baseEntity.posX, -baseEntity.posY, -baseEntity.posZ)
 
     fun fullEntityList() =
         baseEntity.toSingletonListOrEmpty() +
diff --git a/src/main/java/at/hannibal2/skyhanni/features/dungeon/DungeonMobManager.kt b/src/main/java/at/hannibal2/skyhanni/features/dungeon/DungeonMobManager.kt
new file mode 100644
index 000000000..46450a564
--- /dev/null
+++ b/src/main/java/at/hannibal2/skyhanni/features/dungeon/DungeonMobManager.kt
@@ -0,0 +1,149 @@
+package at.hannibal2.skyhanni.features.dungeon
+
+import at.hannibal2.skyhanni.SkyHanniMod
+import at.hannibal2.skyhanni.data.IslandType
+import at.hannibal2.skyhanni.data.mob.Mob
+import at.hannibal2.skyhanni.data.mob.MobData
+import at.hannibal2.skyhanni.events.ConfigLoadEvent
+import at.hannibal2.skyhanni.events.LorenzRenderWorldEvent
+import at.hannibal2.skyhanni.events.LorenzTickEvent
+import at.hannibal2.skyhanni.events.MobEvent
+import at.hannibal2.skyhanni.skyhannimodule.SkyHanniModule
+import at.hannibal2.skyhanni.utils.ColorUtils.toChromaColor
+import at.hannibal2.skyhanni.utils.ConditionalUtils.onToggle
+import at.hannibal2.skyhanni.utils.LorenzUtils.isInIsland
+import at.hannibal2.skyhanni.utils.RenderUtils.draw3DLine
+import at.hannibal2.skyhanni.utils.RenderUtils.drawWaypointFilled
+import at.hannibal2.skyhanni.utils.RenderUtils.exactPlayerEyeLocation
+import at.hannibal2.skyhanni.utils.getLorenzVec
+import net.minecraftforge.fml.common.eventhandler.SubscribeEvent
+import java.awt.Color
+
+@SkyHanniModule
+object DungeonMobManager {
+
+    private val config get() = SkyHanniMod.feature.dungeon.objectHighlighter
+    private val starredConfig get() = config.starred
+    private val fel get() = config.fel
+
+    private val staredInvisible = mutableSetOf<Mob>()
+    private val felOnTheGround = mutableSetOf<Mob>()
+
+    @SubscribeEvent
+    fun onConfigLoad(event: ConfigLoadEvent) {
+        onToggle(
+            starredConfig.highlight,
+            starredConfig.colour,
+        ) {
+            val color = if (starredConfig.highlight.get()) getStarColor() else null
+            MobData.skyblockMobs.filter { it.hasStar }.forEach {
+                handleStar0(it, color)
+            }
+            if (!starredConfig.highlight.get()) {
+                staredInvisible.clear()
+            }
+        }
+        onToggle(
+            fel.highlight,
+            fel.colour,
+        ) {
+            if (fel.highlight.get()) {
+                if (felOnTheGround.isEmpty()) {
+                    MobData.skyblockMobs.forEach(::handleFel)
+                }
+            } else {
+                felOnTheGround.clear()
+            }
+        }
+    }
+
+    @SubscribeEvent
+    fun onMobSpawn(event: MobEvent.Spawn.SkyblockMob) {
+        if (event.mob.mobType != Mob.Type.DUNGEON) return
+        handleStar(event.mob)
+        handleFel(event.mob)
+    }
+
+    @SubscribeEvent
+    fun onMobDeSpawn(event: MobEvent.DeSpawn.SkyblockMob) {
+        if (event.mob.mobType != Mob.Type.DUNGEON) return
+        if (starredConfig.highlight.get()) {
+            staredInvisible.remove(event.mob)
+        }
+        handleFelDespawn(event.mob)
+    }
+
+    @SubscribeEvent
+    fun onLorenzTick(event: LorenzTickEvent) {
+        if (!IslandType.CATACOMBS.isInIsland()) return
+        handleInvisibleStar()
+    }
+
+    @SubscribeEvent
+    fun onRenderWorld(event: LorenzRenderWorldEvent) {
+        if (!fel.highlight.get()) return
+        if (fel.line) {
+            felOnTheGround.filter { it.canBeSeen() }.forEach {
+                event.draw3DLine(
+                    it.baseEntity.getLorenzVec().add(y = 0.15),
+                    event.exactPlayerEyeLocation(),
+                    fel.colour.get().toChromaColor(),
+                    3,
+                    true,
+                )
+            }
+        }
+
+        felOnTheGround.removeIf { mob ->
+            event.drawWaypointFilled(
+                mob.baseEntity.getLorenzVec().add(-0.5, -0.23, -0.5),
+                fel.colour.get().toChromaColor(),
+                seeThroughBlocks = false,
+                beacon = false,
+                extraSize = -0.2,
+                minimumAlpha = 0.8f,
+                inverseAlphaScale = true,
+            )
+            !mob.isInvisible()
+        }
+    }
+
+    private fun handleStar(mob: Mob) {
+        if (!starredConfig.highlight.get()) return
+        if (!mob.hasStar) return
+        handleStar0(mob, getStarColor())
+    }
+
+    private fun handleInvisibleStar() {
+        if (!starredConfig.highlight.get()) return
+        staredInvisible.removeIf {
+            val visible = !it.isInvisible()
+            if (visible) {
+                it.highlight(getStarColor())
+            }
+            visible
+        }
+    }
+
+    private fun getStarColor(): Color = starredConfig.colour.get().toChromaColor()
+
+    private fun handleStar0(mob: Mob, colour: Color?) {
+        if (mob.isInvisible()) {
+            staredInvisible.add(mob)
+            return
+        }
+        mob.highlight(colour)
+    }
+
+    private fun handleFel(mob: Mob) {
+        if (!fel.highlight.get()) return
+        if (mob.name != "Fels") return
+        if (!mob.isInvisible()) return
+        felOnTheGround.add(mob)
+    }
+
+    private fun handleFelDespawn(mob: Mob) {
+        if (!fel.highlight.get()) return
+        felOnTheGround.remove(mob)
+    }
+}