    001package net.minecraft.util;
    002
    003import java.util.Random;
    004import net.minecraft.inventory.IInventory;
    005import net.minecraft.item.ItemStack;
    006import net.minecraft.tileentity.TileEntityDispenser;
    007import net.minecraftforge.common.ChestGenHooks;
    008import net.minecraftforge.common.DungeonHooks;
    009import cpw.mods.fml.common.FMLLog;
    010
    011public class WeightedRandomChestContent extends WeightedRandomItem
    012{
    013    /** The Item/Block ID to generate in the Chest. */
    014    public ItemStack theItemId = null;
    015
    016    /** The minimum chance of item generating. */
    017    public int theMinimumChanceToGenerateItem;
    018
    019    /** The maximum chance of item generating. */
    020    public int theMaximumChanceToGenerateItem;
    021
    022    public WeightedRandomChestContent(int par1, int par2, int par3, int par4, int par5)
    023    {
    024        super(par5);
    025        this.theItemId = new ItemStack(par1, 1, par2);
    026        this.theMinimumChanceToGenerateItem = par3;
    027        this.theMaximumChanceToGenerateItem = par4;
    028    }
    029
    030    public WeightedRandomChestContent(ItemStack par1ItemStack, int par2, int par3, int par4)
    031    {
    032        super(par4);
    033        this.theItemId = par1ItemStack;
    034        this.theMinimumChanceToGenerateItem = par2;
    035        this.theMaximumChanceToGenerateItem = par3;
    036    }
    037
    038    /**
    039     * Generates the Chest contents.
    040     */
    041    public static void generateChestContents(Random par0Random, WeightedRandomChestContent[] par1ArrayOfWeightedRandomChestContent, IInventory par2IInventory, int par3)
    042    {
    043        for (int j = 0; j < par3; ++j)
    044        {
    045            WeightedRandomChestContent weightedrandomchestcontent = (WeightedRandomChestContent)WeightedRandom.getRandomItem(par0Random, par1ArrayOfWeightedRandomChestContent);
    046            int k = weightedrandomchestcontent.theMinimumChanceToGenerateItem + par0Random.nextInt(weightedrandomchestcontent.theMaximumChanceToGenerateItem - weightedrandomchestcontent.theMinimumChanceToGenerateItem + 1);
    047
    048            ItemStack[] stacks = ChestGenHooks.generateStacks(par0Random, weightedrandomchestcontent.theItemId, weightedrandomchestcontent.theMinimumChanceToGenerateItem, weightedrandomchestcontent.theMaximumChanceToGenerateItem);
    049
    050            for (ItemStack item : stacks)
    051            {
    052                par2IInventory.setInventorySlotContents(par0Random.nextInt(par2IInventory.getSizeInventory()), item);
    053            }
    054        }
    055    }
    056
    057    /**
    058     * Generates the Dispenser contents.
    059     */
    060    public static void generateDispenserContents(Random par0Random, WeightedRandomChestContent[] par1ArrayOfWeightedRandomChestContent, TileEntityDispenser par2TileEntityDispenser, int par3)
    061    {
    062        for (int j = 0; j < par3; ++j)
    063        {
    064            WeightedRandomChestContent weightedrandomchestcontent = (WeightedRandomChestContent)WeightedRandom.getRandomItem(par0Random, par1ArrayOfWeightedRandomChestContent);
    065            ItemStack[] stacks = ChestGenHooks.generateStacks(par0Random, weightedrandomchestcontent.theItemId, weightedrandomchestcontent.theMinimumChanceToGenerateItem, weightedrandomchestcontent.theMaximumChanceToGenerateItem);
    066
    067            for (ItemStack item : stacks)
    068            {
    069                par2TileEntityDispenser.setInventorySlotContents(par0Random.nextInt(par2TileEntityDispenser.getSizeInventory()), item);
    070            }
    071        }
    072    }
    073
    074    public static WeightedRandomChestContent[] func_92080_a(WeightedRandomChestContent[] par0ArrayOfWeightedRandomChestContent, WeightedRandomChestContent ... par1ArrayOfWeightedRandomChestContent)
    075    {
    076        WeightedRandomChestContent[] aweightedrandomchestcontent1 = new WeightedRandomChestContent[par0ArrayOfWeightedRandomChestContent.length + par1ArrayOfWeightedRandomChestContent.length];
    077        int i = 0;
    078
    079        for (int j = 0; j < par0ArrayOfWeightedRandomChestContent.length; ++j)
    080        {
    081            aweightedrandomchestcontent1[i++] = par0ArrayOfWeightedRandomChestContent[j];
    082        }
    083
    084        WeightedRandomChestContent[] aweightedrandomchestcontent2 = par1ArrayOfWeightedRandomChestContent;
    085        int k = par1ArrayOfWeightedRandomChestContent.length;
    086
    087        for (int l = 0; l < k; ++l)
    088        {
    089            WeightedRandomChestContent weightedrandomchestcontent1 = aweightedrandomchestcontent2[l];
    090            aweightedrandomchestcontent1[i++] = weightedrandomchestcontent1;
    091        }
    092
    093        return aweightedrandomchestcontent1;
    094    }
    095
    096    // -- Forge hooks
    097    /**
    098     * Allow a mod to submit a custom implementation that can delegate item stack generation beyond simple stack lookup
    099     *
    100     * @param random The current random for generation
    101     * @param newInventory The inventory being generated (do not populate it, but you can refer to it)
    102     * @return An array of {@link ItemStack} to put into the chest
    103     */
    104    protected ItemStack[] generateChestContent(Random random, IInventory newInventory)
    105    {
    106        return ChestGenHooks.generateStacks(random, theItemId, theMinimumChanceToGenerateItem, theMaximumChanceToGenerateItem);
    107    }
    108
    109}