package net.minecraftforge.fluids; import java.util.HashMap; import java.util.Map; import net.minecraft.block.Block; import net.minecraftforge.common.MinecraftForge; import net.minecraftforge.event.Event; import com.google.common.collect.BiMap; import com.google.common.collect.HashBiMap; import com.google.common.collect.ImmutableMap; /** * Handles Fluid registrations. Fluids MUST be registered in order to function. * * @author King Lemming, CovertJaguar (LiquidDictionary) * */ public abstract class FluidRegistry { static int maxID = 0; static HashMap fluids = new HashMap(); static BiMap fluidIDs = HashBiMap.create(); static BiMap fluidBlocks; public static final Fluid WATER = new Fluid("water").setBlockID(Block.waterStill.blockID).setUnlocalizedName(Block.waterStill.getUnlocalizedName()); public static final Fluid LAVA = new Fluid("lava").setBlockID(Block.lavaStill.blockID).setLuminosity(15).setDensity(3000).setViscosity(6000).setTemperature(1300).setUnlocalizedName(Block.lavaStill.getUnlocalizedName()); public static int renderIdFluid = -1; static { registerFluid(WATER); registerFluid(LAVA); } private FluidRegistry(){} /** * Called by Forge to prepare the ID map for server -> client sync. */ static void initFluidIDs(BiMap newfluidIDs) { maxID = newfluidIDs.size(); fluidIDs.clear(); fluidIDs.putAll(newfluidIDs); } /** * Register a new Fluid. If a fluid with the same name already exists, registration is denied. * * @param fluid * The fluid to register. * @return True if the fluid was successfully registered; false if there is a name clash. */ public static boolean registerFluid(Fluid fluid) { if (fluidIDs.containsKey(fluid.getName())) { return false; } fluids.put(fluid.getName(), fluid); fluidIDs.put(fluid.getName(), ++maxID); MinecraftForge.EVENT_BUS.post(new FluidRegisterEvent(fluid.getName(), maxID)); return true; } public static boolean isFluidRegistered(Fluid fluid) { return fluidIDs.containsKey(fluid.getName()); } public static boolean isFluidRegistered(String fluidName) { return fluidIDs.containsKey(fluidName); } public static Fluid getFluid(String fluidName) { return fluids.get(fluidName); } public static Fluid getFluid(int fluidID) { return fluids.get(getFluidName(fluidID)); } public static String getFluidName(int fluidID) { return fluidIDs.inverse().get(fluidID); } public static String getFluidName(FluidStack stack) { return getFluidName(stack.fluidID); } public static int getFluidID(String fluidName) { return fluidIDs.get(fluidName); } public static FluidStack getFluidStack(String fluidName, int amount) { if (!fluidIDs.containsKey(fluidName)) { return null; } return new FluidStack(getFluidID(fluidName), amount); } /** * Returns a read-only map containing Fluid Names and their associated Fluids. */ public static Map getRegisteredFluids() { return ImmutableMap.copyOf(fluids); } /** * Returns a read-only map containing Fluid Names and their associated IDs. */ public static Map getRegisteredFluidIDs() { return ImmutableMap.copyOf(fluidIDs); } public static Fluid lookupFluidForBlock(Block block) { if (fluidBlocks == null) { for (Fluid fluid : fluids.values()) { if (fluid.canBePlacedInWorld() && Block.blocksList[fluid.getBlockID()] != null) { fluidBlocks.put(Block.blocksList[fluid.getBlockID()], fluid); } } } return fluidBlocks.get(block); } public static class FluidRegisterEvent extends Event { public final String fluidName; public final int fluidID; public FluidRegisterEvent(String fluidName, int fluidID) { this.fluidName = fluidName; this.fluidID = fluidID; } } }