package darkknight.jewelrycraft.affixes;

import java.util.*;

import cpw.mods.fml.common.event.FMLPreInitializationEvent;
import darkknight.jewelrycraft.api.ModifierEffect;
import darkknight.jewelrycraft.random.WeightedRandomAffix;
import net.minecraft.util.WeightedRandom;

public class AffixMods {
	private static Map<String, ModifierEffect> prefixes;
	private static Map<String, ModifierEffect> suffixes;

	private static List<WeightedRandomAffix> generalPrefixGen;
	private static List<WeightedRandomAffix> generalSuffixGen;

	private static List<WeightedRandomAffix> ringPrefixGen;
	private static List<WeightedRandomAffix> ringSuffixGen;

	private static List<WeightedRandomAffix> braceletPrefixGen;
	private static List<WeightedRandomAffix> braceletSuffixGen;

	private static List<WeightedRandomAffix> necklacePrefixGen;
	private static List<WeightedRandomAffix> necklaceSuffixGen;

	private static List<WeightedRandomAffix> earringsPrefixGen;
	private static List<WeightedRandomAffix> earringsSuffixGen;

	public static void initializeAffixes(FMLPreInitializationEvent fpie) {
		prefixes = new HashMap<String, ModifierEffect>();
		suffixes = new HashMap<String, ModifierEffect>();

		generalPrefixGen = new LinkedList<WeightedRandomAffix>();
		generalSuffixGen = new LinkedList<WeightedRandomAffix>();

		ringPrefixGen = new LinkedList<WeightedRandomAffix>();
		ringSuffixGen = new LinkedList<WeightedRandomAffix>();

		braceletPrefixGen = new LinkedList<WeightedRandomAffix>();
		braceletSuffixGen = new LinkedList<WeightedRandomAffix>();

		necklacePrefixGen = new LinkedList<WeightedRandomAffix>();
		necklaceSuffixGen = new LinkedList<WeightedRandomAffix>();

		earringsPrefixGen = new LinkedList<WeightedRandomAffix>();
		earringsSuffixGen = new LinkedList<WeightedRandomAffix>();

		ModifierEffect health1 = new HealthAffix(1);
		ModifierEffect health2 = new HealthAffix(2);
		ModifierEffect health3 = new HealthAffix(3);
		ModifierEffect health4 = new HealthAffix(4);
		ModifierEffect health5 = new HealthAffix(5);
		ModifierEffect health6 = new HealthAffix(6);

		ModifierEffect hitheal1 = new FlatLifeHitAffix(1);
		ModifierEffect hitheal2 = new FlatLifeHitAffix(2);
		ModifierEffect hitheal3 = new FlatLifeHitAffix(3);

		prefixes.put("health1", health1);
		prefixes.put("health2", health2);
		prefixes.put("health3", health3);
		prefixes.put("health4", health4);
		prefixes.put("health5", health5);
		prefixes.put("health6", health6);

		suffixes.put("health1", health1);
		suffixes.put("health2", health2);
		suffixes.put("health3", health3);
		suffixes.put("hitheal1", hitheal1);
		suffixes.put("hitheal2", hitheal2);
		suffixes.put("hitheal3", hitheal3);

		WeightedRandomAffix wraHealth1 = new WeightedRandomAffix("health1", 10);
		WeightedRandomAffix wraHealth2 = new WeightedRandomAffix("health2", 8);
		WeightedRandomAffix wraHealth3 = new WeightedRandomAffix("health3", 6);
		WeightedRandomAffix wraHealth4 = new WeightedRandomAffix("health4", 4);
		WeightedRandomAffix wraHealth5 = new WeightedRandomAffix("health5", 2);
		WeightedRandomAffix wraHealth6 = new WeightedRandomAffix("health6", 1);

		WeightedRandomAffix wraHitheal1 = new WeightedRandomAffix("hitheal1", 9);
		WeightedRandomAffix wraHitheal2 = new WeightedRandomAffix("hitheal2", 8);
		WeightedRandomAffix wraHitheal3 = new WeightedRandomAffix("hitheal3", 7);

		generalPrefixGen.add(wraHealth1);
		generalPrefixGen.add(wraHealth2);
		generalPrefixGen.add(wraHealth3);
		ringPrefixGen.add(wraHealth1);
		ringPrefixGen.add(wraHealth2);
		braceletPrefixGen.add(wraHealth2);
		braceletPrefixGen.add(wraHealth3);
		necklacePrefixGen.add(wraHealth4);
		necklacePrefixGen.add(wraHealth5);
		earringsPrefixGen.add(wraHealth5);
		earringsPrefixGen.add(wraHealth6);

		generalSuffixGen.add(wraHealth1);
		generalSuffixGen.add(wraHealth2);
		generalSuffixGen.add(wraHealth3);
		ringSuffixGen.add(wraHealth1);
		ringSuffixGen.add(wraHealth2);
		braceletSuffixGen.add(wraHealth2);
		braceletSuffixGen.add(wraHealth3);

		generalSuffixGen.add(wraHitheal1);
		braceletSuffixGen.add(wraHitheal1);
		braceletSuffixGen.add(wraHitheal2);
		necklaceSuffixGen.add(wraHitheal2);
		necklaceSuffixGen.add(wraHitheal3);
	}

	public static ModifierEffect getPrefix(String prefix) {
		return prefixes.get(prefix);
	}

	public static ModifierEffect getSuffix(String suffix) {
		return suffixes.get(suffix);
	}

	public static String pickPrefix(Random random, int jewelType) {
		List<WeightedRandomAffix> pickList = generalPrefixGen;

		if (random.nextInt(3) != 0) {
			switch (jewelType) {
			case 0:
				pickList = ringPrefixGen;
				break;
			case 1:
				pickList = braceletPrefixGen;
				break;
			case 2:
				pickList = necklacePrefixGen;
				break;
			case 3:
				pickList = earringsPrefixGen;
				break;
			case -1:
			default:
				pickList = generalPrefixGen;
			}
		}

		WeightedRandomAffix wra = ((WeightedRandomAffix) WeightedRandom.getRandomItem(random, pickList));

		return wra.getAffix(random);
	}

	public static String pickSuffix(Random random, int jewelType) {
		List<WeightedRandomAffix> pickList = generalSuffixGen;

		if (random.nextInt(3) != 0) {
			switch (jewelType) {
			case 0:
				pickList = ringSuffixGen;
				break;
			case 1:
				pickList = braceletSuffixGen;
				break;
			case 2:
				pickList = necklaceSuffixGen;
				break;
			case 3:
				pickList = earringsSuffixGen;
				break;
			case -1:
			default:
				pickList = generalSuffixGen;
			}
		}

		WeightedRandomAffix wra = ((WeightedRandomAffix) WeightedRandom.getRandomItem(random, pickList));

		return wra.getAffix(random);
	}
}