TinkerUtil.java

package slimeknights.tconstruct.library.utils;

import com.google.common.collect.Lists;

import net.minecraft.item.ItemStack;
import net.minecraft.nbt.NBTTagCompound;
import net.minecraft.nbt.NBTTagList;

import java.util.Arrays;
import java.util.List;

import slimeknights.tconstruct.library.TinkerRegistry;
import slimeknights.tconstruct.library.materials.Material;
import slimeknights.tconstruct.library.modifiers.ModifierNBT;
import slimeknights.tconstruct.library.tinkering.Category;
import slimeknights.tconstruct.library.tinkering.IMaterialItem;

public final class TinkerUtil {

  private TinkerUtil() {
  }

  /**
   * Safe way of getting the material from an itemstack.
   */
  public static Material getMaterialFromStack(ItemStack stack) {
    if(stack == null || stack.getItem() == null) {
      return Material.UNKNOWN;
    }
    if(!(stack.getItem() instanceof IMaterialItem)) {
      return Material.UNKNOWN;
    }

    return ((IMaterialItem) stack.getItem()).getMaterial(stack);
  }

  public static boolean hasCategory(NBTTagCompound root, Category category) {
    return Arrays.stream(TagUtil.getCategories(root)).anyMatch(category::equals);
  }

  public static boolean hasTrait(NBTTagCompound root, String identifier) {
    NBTTagList tagList = TagUtil.getTraitsTagList(root);

    for(int i = 0; i < tagList.tagCount(); i++) {
      if(identifier.equals(tagList.getStringTagAt(i))) {
        return true;
      }
    }

    return false;
  }

  public static boolean hasModifier(NBTTagCompound root, String identifier) {
    NBTTagList tagList = TagUtil.getBaseModifiersTagList(root);

    for(int i = 0; i < tagList.tagCount(); i++) {
      if(identifier.equals(tagList.getStringTagAt(i))) {
        return true;
      }
    }

    return false;
  }

  public static int getIndexInList(NBTTagList tagList, String identifier) {
    if(tagList.getTagType() == TagUtil.TAG_TYPE_STRING) {
      return getIndexInStringList(tagList, identifier);
    }
    else if(tagList.getTagType() == TagUtil.TAG_TYPE_COMPOUND) {
      return getIndexInCompoundList(tagList, identifier);
    }

    // unsupported format
    return -1;
  }

  private static int getIndexInStringList(NBTTagList tagList, String identifier) {
    for(int i = 0; i < tagList.tagCount(); i++) {
      String data = tagList.getStringTagAt(i);
      if(identifier.equals(data)) {
        return i;
      }
    }

    return -1;
  }

  public static int getIndexInCompoundList(NBTTagList tagList, String identifier) {
    // do we already have a tag for this modifier?
    for(int i = 0; i < tagList.tagCount(); i++) {
      ModifierNBT data = ModifierNBT.readTag(tagList.getCompoundTagAt(i));
      if(identifier.equals(data.identifier)) {
        return i;
      }
    }

    return -1;
  }

  public static NBTTagCompound getModifierTag(ItemStack stack, String identifier) {
    NBTTagList tagList = TagUtil.getModifiersTagList(stack);
    int index = getIndexInCompoundList(tagList, identifier);

    // returns new tag if index is out of scope
    return tagList.getCompoundTagAt(index);
  }

  public static NBTTagCompound getModifierTag(NBTTagCompound root, String identifier) {
    NBTTagList tagList = TagUtil.getModifiersTagList(root);
    int index = getIndexInCompoundList(tagList, identifier);

    // returns new tag if index is out of scope
    return tagList.getCompoundTagAt(index);
  }

  public static List<Material> getMaterialsFromTagList(NBTTagList tagList) {
    List<Material> materials = Lists.newLinkedList();
    if(tagList.getTagType() != TagUtil.TAG_TYPE_STRING) {
      //TinkerRegistry.log.error("Incorrect taglist type to get materiallist from TagList");
      return materials;
    }

    for(int i = 0; i < tagList.tagCount(); i++) {
      String identifier = tagList.getStringTagAt(i);
      Material mat = TinkerRegistry.getMaterial(identifier);
      materials.add(mat);
    }

    return materials;
  }

  // balantly stolen from StackOverflow and then optimized
  public static String getRomanNumeral(int value) {
    if(value < 1 || value > 3999) {
      return "Really big";
    }

    StringBuilder sb = new StringBuilder();
    while(value >= 1000) {
      sb.append("M");
      value -= 1000;
    }
    while(value >= 900) {
      sb.append("CM");
      value -= 900;
    }
    while(value >= 500) {
      sb.append("D");
      value -= 500;
    }
    while(value >= 400) {
      sb.append("CD");
      value -= 400;
    }
    while(value >= 100) {
      sb.append("C");
      value -= 100;
    }
    while(value >= 90) {
      sb.append("XC");
      value -= 90;
    }
    while(value >= 50) {
      sb.append("L");
      value -= 50;
    }
    while(value >= 40) {
      sb.append("XL");
      value -= 40;
    }
    while(value >= 10) {
      sb.append("X");
      value -= 10;
    }
    while(value >= 9) {
      sb.append("IX");
      value -= 9;
    }
    while(value >= 5) {
      sb.append("V");
      value -= 5;
    }
    while(value >= 4) {
      sb.append("IV");
      value -= 4;
    }
    while(value >= 1) {
      sb.append("I");
      value -= 1;
    }
    return sb.toString();
  }
}