1 files changed, 55 insertions, 0 deletions
diff --git a/src/main/java/makamys/neodymium/util/SpriteUtil.java b/src/main/java/makamys/neodymium/util/SpriteUtil.java
new file mode 100644
index 0000000..4219802
--- /dev/null
+++ b/src/main/java/makamys/neodymium/util/SpriteUtil.java
@@ -0,0 +1,55 @@
+package makamys.neodymium.util;
+
+import java.util.HashMap;
+import java.util.List;
+import java.util.Map;
+import java.util.stream.Collectors;
+
+import net.minecraft.client.Minecraft;
+import net.minecraft.client.renderer.texture.TextureAtlasSprite;
+import net.minecraft.client.renderer.texture.TextureMap;
+import net.minecraft.world.ChunkCoordIntPair;
+
+public class SpriteUtil {
+    
+    private static int[] spriteIndexMap;
+    public static List<TextureAtlasSprite> sprites;
+    
+    private static Map<Long, Integer> uv2spriteIndex = new HashMap<>();
+    
+    private static int findSpriteIndexForUV(float u, float v) {
+        Map<String, TextureAtlasSprite> uploadedSprites = ((TextureMap)Minecraft.getMinecraft().getTextureManager().getTexture(TextureMap.locationBlocksTexture)).mapUploadedSprites;
+        
+        int spriteIndex = 0;
+        for(TextureAtlasSprite tas : uploadedSprites.values()) {
+            if(tas.getMinU() <= u && u <= tas.getMaxU() && tas.getMinV() <= v && v <= tas.getMaxV()) {
+                break;
+            }
+            spriteIndex++;
+        }
+        return spriteIndex;
+    }
+    
+    public static int getSpriteIndexForUV(float u, float v){
+        long key = ChunkCoordIntPair.chunkXZ2Int((int)(u * Integer.MAX_VALUE), (int)(v * Integer.MAX_VALUE));
+        int index = uv2spriteIndex.getOrDefault(key, -1);
+        if(index == -1) {
+            index = findSpriteIndexForUV(u, v);
+            uv2spriteIndex.put(key, index);
+        }
+        return index;
+    }
+    
+    public static TextureAtlasSprite getSprite(int i){
+        if(i >= 0 && i < sprites.size()) {
+            return sprites.get(i);
+        } else {
+            return null;
+        }
+    }
+    
+    public static void init() {
+        Map<String, TextureAtlasSprite> uploadedSprites = ((TextureMap)Minecraft.getMinecraft().getTextureManager().getTexture(TextureMap.locationBlocksTexture)).mapUploadedSprites;
+        sprites = uploadedSprites.values().stream().collect(Collectors.toList());
+    }
+}