Merge pull request #19 from DeDiamondPro/master

Metal detector crystal hollow solver thing

1 files changed, 139 insertions, 0 deletions

diff --git a/src/main/java/io/github/moulberry/notenoughupdates/miscfeatures/CrystalMetalDetectorSolver.java b/src/main/java/io/github/moulberry/notenoughupdates/miscfeatures/CrystalMetalDetectorSolver.java
new file mode 100644
index 00000000..c6ad2ef1
--- /dev/null
+++ b/src/main/java/io/github/moulberry/notenoughupdates/miscfeatures/CrystalMetalDetectorSolver.java
@@ -0,0 +1,139 @@
+package io.github.moulberry.notenoughupdates.miscfeatures;
+
+import io.github.moulberry.notenoughupdates.NotEnoughUpdates;
+import io.github.moulberry.notenoughupdates.core.util.render.RenderUtils;
+import io.github.moulberry.notenoughupdates.util.SBInfo;
+import net.minecraft.client.Minecraft;
+import net.minecraft.util.*;
+
+import java.util.ArrayList;
+import java.util.List;
+
+public class CrystalMetalDetectorSolver {
+    private static final Minecraft mc = Minecraft.getMinecraft();
+    private static BlockPos prevPos;
+    private static double prevDist = 0;
+    private static List<BlockPos> possibleBlocks = new ArrayList<>();
+    private static final List<BlockPos> locations = new ArrayList<>();
+
+    public static void process(IChatComponent message) {
+        if (SBInfo.getInstance().getLocation() != null && SBInfo.getInstance().getLocation().equals("crystal_hollows")
+                && message.getUnformattedText().contains("TREASURE: ")) {
+            double dist = Double.parseDouble(message.getUnformattedText().split("TREASURE: ")[1].split("m")[0].replaceAll("(?!\\.)\\D", ""));
+            if (NotEnoughUpdates.INSTANCE.config.mining.metalDetectorEnabled && prevDist == dist && prevPos.getX() == mc.thePlayer.getPosition().getX() &&
+                    prevPos.getY() == mc.thePlayer.getPosition().getY() &&
+                    prevPos.getZ() == mc.thePlayer.getPosition().getZ() && !locations.contains(mc.thePlayer.getPosition())) {
+                if (possibleBlocks.size() == 0) {
+                    locations.add(mc.thePlayer.getPosition());
+                    for (int zOffset = (int) Math.floor(-dist); zOffset <= Math.ceil(dist); zOffset++) {
+                        for (int y = 65; y <= 75; y++) {
+                            double calculatedDist = 0;
+                            int xOffset = 0;
+                            while (calculatedDist < dist) {
+                                BlockPos pos = new BlockPos(Math.floor(mc.thePlayer.posX) + xOffset,
+                                        y, Math.floor(mc.thePlayer.posZ) + zOffset);
+                                BlockPos above = new BlockPos(Math.floor(mc.thePlayer.posX) + xOffset,
+                                        y + 1, Math.floor(mc.thePlayer.posZ) + zOffset);
+                                xOffset++;
+                                calculatedDist = getPlayerPos().distanceTo(new Vec3(pos).addVector(0D,1D,0D));
+                                if (inRange(calculatedDist, dist) && treasureAllowed(pos) && !possibleBlocks.contains(pos) &&
+                                        mc.theWorld.getBlockState(above).getBlock().getRegistryName().equals("minecraft:air")) {
+                                    possibleBlocks.add(pos);
+                                }
+                                xOffset++;
+                            }
+                            xOffset = 0;
+                            calculatedDist = 0;
+                            while (calculatedDist < dist) {
+                                BlockPos pos = new BlockPos(Math.floor(mc.thePlayer.posX) - xOffset,
+                                        y, Math.floor(mc.thePlayer.posZ) + zOffset);
+                                BlockPos above = new BlockPos(Math.floor(mc.thePlayer.posX) - xOffset,
+                                        y + 1, Math.floor(mc.thePlayer.posZ) + zOffset);
+                                calculatedDist = getPlayerPos().distanceTo(new Vec3(pos).addVector(0D,1D,0D));
+                                if (inRange(calculatedDist, dist) && treasureAllowed(pos) && !possibleBlocks.contains(pos) &&
+                                        mc.theWorld.getBlockState(above).getBlock().getRegistryName().equals("minecraft:air")) {
+                                    possibleBlocks.add(pos);
+                                }
+                                xOffset++;
+                            }
+                        }
+                    }
+                    sendMessage();
+                } else if (possibleBlocks.size() != 1) {
+                    locations.add(mc.thePlayer.getPosition());
+                    List<BlockPos> temp = new ArrayList<>();
+                    for (BlockPos pos : possibleBlocks) {
+                        if (round(getPlayerPos().distanceTo(new Vec3(pos).addVector(0D,1D,0D)), 1)
+                                == dist) {
+                            temp.add(pos);
+                        }
+                    }
+                    if (temp.size() == 0) {
+                        for (BlockPos pos : possibleBlocks) {
+                            if (inRange(getPlayerPos().distanceTo(new Vec3(pos).addVector(0D,1D,0D)), dist)) {
+                                temp.add(pos);
+                            }
+                        }
+                    }
+                    possibleBlocks = temp;
+                    sendMessage();
+                }
+            }
+            prevPos = mc.thePlayer.getPosition();
+            prevDist = dist;
+        }
+    }
+
+    public static void reset() {
+        possibleBlocks.clear();
+        locations.clear();
+    }
+
+    public static void render(float partialTicks) {
+        if (SBInfo.getInstance().getLocation() != null && SBInfo.getInstance().getLocation().equals("crystal_hollows")) {
+            if (possibleBlocks.size() == 1) {
+                RenderUtils.renderWayPoint("Treasure", possibleBlocks.get(0).add(0, 2.5, 0), partialTicks);
+            } else if (possibleBlocks.size() > 1 && NotEnoughUpdates.INSTANCE.config.mining.metalDetectorShowPossible) {
+                for (BlockPos block : possibleBlocks) {
+                    RenderUtils.renderWayPoint("Possible Treasure Location", block.add(0, 2.5, 0), partialTicks);
+                }
+            }
+        }
+    }
+
+    private static double round(double value, int precision) {
+        int scale = (int) Math.pow(10, precision);
+        return (double) Math.round(value * scale) / scale;
+    }
+
+    private static boolean treasureAllowed(BlockPos pos) {
+        return mc.theWorld.getBlockState(pos).getBlock().getRegistryName().equals("minecraft:gold_block") ||
+                mc.theWorld.getBlockState(pos).getBlock().getRegistryName().equals("minecraft:prismarine") ||
+                mc.theWorld.getBlockState(pos).getBlock().getRegistryName().equals("minecraft:chest") ||
+                mc.theWorld.getBlockState(pos).getBlock().getRegistryName().equals("minecraft:stained_glass") ||
+                mc.theWorld.getBlockState(pos).getBlock().getRegistryName().equals("minecraft:stained_glass_pane") ||
+                mc.theWorld.getBlockState(pos).getBlock().getRegistryName().equals("minecraft:wool") ||
+                mc.theWorld.getBlockState(pos).getBlock().getRegistryName().equals("minecraft:stained_hardened_clay");
+    }
+
+    private static void sendMessage() {
+        if (possibleBlocks.size() > 1) {
+            mc.thePlayer.addChatMessage(new ChatComponentText(EnumChatFormatting.YELLOW + "[NEU] Need another position to find solution. Possible blocks: "
+                    + possibleBlocks.size()));
+        } else if (possibleBlocks.size() == 0) {
+            mc.thePlayer.addChatMessage(new ChatComponentText(EnumChatFormatting.RED + "[NEU] Failed to find solution."));
+            reset();
+        } else {
+            mc.thePlayer.addChatMessage(new ChatComponentText(EnumChatFormatting.YELLOW + "[NEU] Found solution."));
+        }
+    }
+
+
+    private static boolean inRange(double number, double dist) {
+        return dist + 0.1D >= number && dist - 0.1D <= number;
+    }
+
+    private static Vec3 getPlayerPos() {
+        return new Vec3(mc.thePlayer.posX, mc.thePlayer.posY + (mc.thePlayer.getEyeHeight() - mc.thePlayer.getDefaultEyeHeight()), mc.thePlayer.posZ);
+    }
+}