Fix bolt fletchling color

SlimeKnights · Oct 10, 2014 · 4f9f63a · 4f9f63a
1 parent 8fe42be
commit 4f9f63a
Showing 1 changed file with 18 additions and 0 deletions.
diff --git a/src/main/java/tconstruct/weaponry/ammo/BoltAmmo.java b/src/main/java/tconstruct/weaponry/ammo/BoltAmmo.java
@@ -1,5 +1,9 @@
 package tconstruct.weaponry.ammo;
 
+import tconstruct.library.TConstructRegistry;
+import tconstruct.library.tools.CustomMaterial;
+import tconstruct.library.tools.FletchingMaterial;
+import tconstruct.library.tools.FletchlingLeafMaterial;
 import tconstruct.weaponry.TinkerWeaponry;
 import tconstruct.library.tools.DualMaterialToolPart;
 import tconstruct.library.weaponry.AmmoItem;
@@ -89,4 +93,18 @@ public void buildTool (int id, String name, List list)
             list.add(tool);
         }
     }
+
+    @Override
+    protected int getDefaultColor(int renderPass, int materialID) {
+        if(renderPass != 2)
+            return super.getDefaultColor(renderPass, materialID);
+
+        CustomMaterial mat = TConstructRegistry.getCustomMaterial(materialID, FletchingMaterial.class);
+        if(mat == null)
+            TConstructRegistry.getCustomMaterial(materialID, FletchlingLeafMaterial.class);
+        if(mat == null)
+            return 0xffffff;
+
+        return mat.color;
+    }
 }