Quality.java

package greymerk.roguelike.treasure.loot;

import java.util.HashMap;
import java.util.Map;
import java.util.Random;

import greymerk.roguelike.util.IWeighted;
import greymerk.roguelike.util.WeightedChoice;
import greymerk.roguelike.util.WeightedRandomizer;

public enum Quality{
	
	WOOD, STONE, IRON, GOLD, DIAMOND;
	
	private static Map<Integer, IWeighted<Quality>> armourQuality;
	private static Map<Integer, IWeighted<Quality>> weaponQuality;
	private static Map<Integer, IWeighted<Quality>> toolQuality;
	static {
		armourQuality = new HashMap<Integer, IWeighted<Quality>>();
		weaponQuality = new HashMap<Integer, IWeighted<Quality>>();
		toolQuality = new HashMap<Integer, IWeighted<Quality>>();
		
		for(int i = 0; i < 5; ++i){
			WeightedRandomizer<Quality> armour = new WeightedRandomizer<Quality>();
			switch(i){
			case 0:
				armour.add(new WeightedChoice<Quality>(WOOD, 250));
				armour.add(new WeightedChoice<Quality>(STONE, 50));
				armour.add(new WeightedChoice<Quality>(IRON, 20));
				armour.add(new WeightedChoice<Quality>(GOLD, 3));
				armour.add(new WeightedChoice<Quality>(DIAMOND, 1));
				break;
			case 1:
				armour.add(new WeightedChoice<Quality>(WOOD, 150));
				armour.add(new WeightedChoice<Quality>(STONE, 30));
				armour.add(new WeightedChoice<Quality>(IRON, 10));
				armour.add(new WeightedChoice<Quality>(GOLD, 3));
				armour.add(new WeightedChoice<Quality>(DIAMOND, 1));
				break;
			case 2:
				armour.add(new WeightedChoice<Quality>(WOOD, 50));
				armour.add(new WeightedChoice<Quality>(STONE, 30));
				armour.add(new WeightedChoice<Quality>(IRON, 20));
				armour.add(new WeightedChoice<Quality>(GOLD, 3));
				armour.add(new WeightedChoice<Quality>(DIAMOND, 1));
				break;
			case 3:
				armour.add(new WeightedChoice<Quality>(WOOD, 20));
				armour.add(new WeightedChoice<Quality>(STONE, 10));
				armour.add(new WeightedChoice<Quality>(IRON, 10));
				armour.add(new WeightedChoice<Quality>(GOLD, 5));
				armour.add(new WeightedChoice<Quality>(DIAMOND, 3));
				break;
			case 4:
				armour.add(new WeightedChoice<Quality>(WOOD, 2));
				armour.add(new WeightedChoice<Quality>(STONE, 3));
				armour.add(new WeightedChoice<Quality>(IRON, 10));
				armour.add(new WeightedChoice<Quality>(GOLD, 3));
				armour.add(new WeightedChoice<Quality>(DIAMOND, 3));
				break;
			}
			armourQuality.put(i, armour);
			
			WeightedRandomizer<Quality> weapon = new WeightedRandomizer<Quality>();
			switch(i){
			case 0:
				weapon.add(new WeightedChoice<Quality>(WOOD, 200));
				weapon.add(new WeightedChoice<Quality>(STONE, 50));
				weapon.add(new WeightedChoice<Quality>(IRON, 10));
				weapon.add(new WeightedChoice<Quality>(GOLD, 3));
				weapon.add(new WeightedChoice<Quality>(DIAMOND, 1));
				break;
			case 1:
				weapon.add(new WeightedChoice<Quality>(WOOD, 100));
				weapon.add(new WeightedChoice<Quality>(STONE, 30));
				weapon.add(new WeightedChoice<Quality>(IRON, 10));
				weapon.add(new WeightedChoice<Quality>(GOLD, 3));
				weapon.add(new WeightedChoice<Quality>(DIAMOND, 1));
				break;
			case 2:
				weapon.add(new WeightedChoice<Quality>(WOOD, 50));
				weapon.add(new WeightedChoice<Quality>(STONE, 20));
				weapon.add(new WeightedChoice<Quality>(IRON, 10));
				weapon.add(new WeightedChoice<Quality>(GOLD, 3));
				weapon.add(new WeightedChoice<Quality>(DIAMOND, 1));
				break;
			case 3:
				weapon.add(new WeightedChoice<Quality>(WOOD, 1));
				weapon.add(new WeightedChoice<Quality>(STONE, 3));
				weapon.add(new WeightedChoice<Quality>(IRON, 5));
				weapon.add(new WeightedChoice<Quality>(GOLD, 3));
				weapon.add(new WeightedChoice<Quality>(DIAMOND, 1));
				break;
			case 4:
				weapon.add(new WeightedChoice<Quality>(WOOD, 1));
				weapon.add(new WeightedChoice<Quality>(STONE, 2));
				weapon.add(new WeightedChoice<Quality>(IRON, 15));
				weapon.add(new WeightedChoice<Quality>(GOLD, 5));
				weapon.add(new WeightedChoice<Quality>(DIAMOND, 3));
				break;
			}
			weaponQuality.put(i, weapon);
			
			WeightedRandomizer<Quality> tool = new WeightedRandomizer<Quality>();
			switch(i){
			case 0:
				tool.add(new WeightedChoice<Quality>(WOOD, 10));
				tool.add(new WeightedChoice<Quality>(STONE, 20));
				tool.add(new WeightedChoice<Quality>(IRON, 10));
				tool.add(new WeightedChoice<Quality>(GOLD, 3));
				tool.add(new WeightedChoice<Quality>(DIAMOND, 1));
				break;
			case 1:
				tool.add(new WeightedChoice<Quality>(WOOD, 2));
				tool.add(new WeightedChoice<Quality>(STONE, 10));
				tool.add(new WeightedChoice<Quality>(IRON, 10));
				tool.add(new WeightedChoice<Quality>(GOLD, 3));
				tool.add(new WeightedChoice<Quality>(DIAMOND, 1));
				break;
			case 2:
				tool.add(new WeightedChoice<Quality>(WOOD, 1));
				tool.add(new WeightedChoice<Quality>(STONE, 5));
				tool.add(new WeightedChoice<Quality>(IRON, 10));
				tool.add(new WeightedChoice<Quality>(GOLD, 5));
				tool.add(new WeightedChoice<Quality>(DIAMOND, 3));
				break;
			case 3:
				tool.add(new WeightedChoice<Quality>(WOOD, 1));
				tool.add(new WeightedChoice<Quality>(STONE, 3));
				tool.add(new WeightedChoice<Quality>(IRON, 10));
				tool.add(new WeightedChoice<Quality>(GOLD, 5));
				tool.add(new WeightedChoice<Quality>(DIAMOND, 5));
				break;
			case 4:
				tool.add(new WeightedChoice<Quality>(WOOD, 1));
				tool.add(new WeightedChoice<Quality>(STONE, 2));
				tool.add(new WeightedChoice<Quality>(IRON, 10));
				tool.add(new WeightedChoice<Quality>(GOLD, 3));
				tool.add(new WeightedChoice<Quality>(DIAMOND, 5));
				break;
			}
			toolQuality.put(i, tool);
		}
		
	}

	public static Quality get(Random rand, int level, Equipment type) {
		
		switch(type){
		case SWORD:
		case BOW:
			return weaponQuality.get(level).get(rand);
		case HELMET:
		case CHEST:
		case LEGS:
		case FEET:
			return armourQuality.get(level).get(rand);
		case PICK:
		case AXE:
		case SHOVEL:
			return toolQuality.get(level).get(rand);
		}
		return null;
	}
	
	public static Quality get(int level){
		switch(level){
		case 0: return Quality.WOOD;
		case 1: return Quality.STONE;
		case 2: return Quality.IRON;
		case 3: return Quality.GOLD;
		case 4: return Quality.DIAMOND;
		default: return Quality.WOOD;
		}
	}

	public static Quality getArmourQuality(Random rand, int level) {
		return armourQuality.get(level).get(rand);
	}

	public static Quality getToolQuality(Random rand, int level) {
		return toolQuality.get(level).get(rand);
	}

	public static Quality getWeaponQuality(Random rand, int level) {
		return weaponQuality.get(level).get(rand);
	}
	
	
}