1 files changed, 96 insertions, 0 deletions
diff --git a/src/main/java/gregtech/common/pollution/BlockMatcher.java b/src/main/java/gregtech/common/pollution/BlockMatcher.java
new file mode 100644
index 0000000000..a701d323bd
--- /dev/null
+++ b/src/main/java/gregtech/common/pollution/BlockMatcher.java
@@ -0,0 +1,96 @@
+package gregtech.common.pollution;
+
+import java.util.Map;
+import java.util.Set;
+
+import net.minecraft.block.Block;
+import net.minecraft.client.multiplayer.WorldClient;
+import net.minecraftforge.event.world.WorldEvent;
+
+import com.google.common.collect.Maps;
+import com.google.common.collect.Sets;
+
+import cpw.mods.fml.common.eventhandler.SubscribeEvent;
+import cpw.mods.fml.common.registry.FMLControlledNamespacedRegistry;
+import cpw.mods.fml.common.registry.GameData;
+import gregtech.GTMod;
+
+// Shamelessly Taken from BetterFoliage by octarine-noise
+public class BlockMatcher {
+
+    public Map<Class<?>, ColorOverrideType> whiteList = Maps.newHashMap();
+    public Set<Class<?>> blackList = Sets.newHashSet();
+    public Map<Integer, ColorOverrideType> blockIDs = Maps.newHashMap();
+
+    public ColorOverrideType matchesID(int blockId) {
+        return blockIDs.get(blockId);
+    }
+
+    public ColorOverrideType matchesID(Block block) {
+        return blockIDs.get(Block.blockRegistry.getIDForObject(block));
+    }
+
+    public void updateClassList(String[] cfg) {
+        whiteList.clear();
+        blackList.clear();
+        for (String line : cfg) {
+            GTMod.GT_FML_LOGGER.info("Checking for block:" + line);
+            String[] lines = line.split(":");
+            ColorOverrideType type = null;
+            if (lines.length > 1) {
+                try {
+                    type = ColorOverrideType.fromName(lines[1].trim());
+                } catch (NumberFormatException e) {
+                    GTMod.GT_FML_LOGGER.error(String.format("Invalid type [%s]", line));
+                    continue;
+                }
+            }
+
+            if (lines[0].startsWith("-")) {
+                try {
+                    blackList.add(Class.forName(lines[0].substring(1)));
+                    GTMod.GT_FML_LOGGER.info("\t added blacklist:" + lines[0].substring(1));
+                } catch (ClassNotFoundException ignored) {}
+            } else {
+                if (type == null) {
+                    GTMod.GT_FML_LOGGER.error(String.format("Invalid type [%s]", line));
+                    continue;
+                }
+
+                try {
+                    whiteList.put(Class.forName(lines[0]), type);
+                    GTMod.GT_FML_LOGGER.info("\t added whitelist:" + lines[0]);
+                } catch (ClassNotFoundException ignored) {}
+            }
+        }
+        // updateBlockIDs();
+    }
+
+    private void updateBlockIDs() {
+        blockIDs.clear();
+        FMLControlledNamespacedRegistry<Block> blockRegistry = GameData.getBlockRegistry();
+        for (Block block : blockRegistry.typeSafeIterable()) {
+            ColorOverrideType t = matchesClass(block);
+            if (t != null) blockIDs.put(Block.blockRegistry.getIDForObject(block), t);
+        }
+    }
+
+    private ColorOverrideType matchesClass(Block block) {
+        for (Class<?> clazz : blackList) if (clazz.isAssignableFrom(block.getClass())) return null;
+        for (Class<?> clazz : whiteList.keySet())
+            if (clazz.isAssignableFrom(block.getClass())) return whiteList.get(clazz);
+        return null;
+    }
+
+    /**
+     * Caches block IDs on world load for fast lookup
+     *
+     * @param event
+     */
+    @SubscribeEvent
+    public void handleWorldLoad(WorldEvent.Load event) {
+        if (event.world instanceof WorldClient) {
+            updateBlockIDs();
+        }
+    }
+}