Files
Applied-Energistics-2/src/main/java/appeng/core/Registration.java
T
2022-09-17 08:08:02 -03:00

527 lines
23 KiB
Java

/*
* This file is part of Applied Energistics 2.
* Copyright (c) 2013 - 2015, AlgorithmX2, All rights reserved.
*
* Applied Energistics 2 is free software: you can redistribute it and/or modify
* it under the terms of the GNU Lesser General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* Applied Energistics 2 is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public License
* along with Applied Energistics 2. If not, see <http://www.gnu.org/licenses/lgpl>.
*/
package appeng.core;
import appeng.api.config.Upgrades;
import appeng.api.definitions.IBlocks;
import appeng.api.definitions.IItems;
import appeng.api.definitions.IParts;
import appeng.api.features.IRecipeHandlerRegistry;
import appeng.api.features.IRegistryContainer;
import appeng.api.features.IWirelessTermHandler;
import appeng.api.features.IWorldGen.WorldGenType;
import appeng.api.movable.IMovableRegistry;
import appeng.api.networking.IGridCacheRegistry;
import appeng.api.networking.crafting.ICraftingGrid;
import appeng.api.networking.energy.IEnergyGrid;
import appeng.api.networking.pathing.IPathingGrid;
import appeng.api.networking.security.ISecurityGrid;
import appeng.api.networking.spatial.ISpatialCache;
import appeng.api.networking.storage.IStorageGrid;
import appeng.api.networking.ticking.ITickManager;
import appeng.bootstrap.ICriterionTriggerRegistry;
import appeng.bootstrap.IModelRegistry;
import appeng.bootstrap.components.*;
import appeng.capabilities.Capabilities;
import appeng.core.features.AEFeature;
import appeng.core.features.registries.P2PTunnelRegistry;
import appeng.core.features.registries.cell.BasicCellHandler;
import appeng.core.features.registries.cell.BasicItemCellGuiHandler;
import appeng.core.features.registries.cell.CreativeCellHandler;
import appeng.core.localization.GuiText;
import appeng.core.localization.PlayerMessages;
import appeng.core.stats.AdvancementTriggers;
import appeng.core.stats.PartItemPredicate;
import appeng.core.stats.Stats;
import appeng.core.worlddata.SpatialDimensionManager;
import appeng.fluids.registries.BasicFluidCellGuiHandler;
import appeng.hooks.TickHandler;
import appeng.items.materials.ItemMaterial;
import appeng.items.parts.ItemFacade;
import appeng.items.parts.ItemPart;
import appeng.loot.ChestLoot;
import appeng.me.cache.*;
import appeng.parts.PartPlacement;
import appeng.recipes.AEItemResolver;
import appeng.recipes.AERecipeLoader;
import appeng.recipes.game.DisassembleRecipe;
import appeng.recipes.game.FacadeRecipe;
import appeng.recipes.ores.OreDictionaryHandler;
import appeng.spatial.BiomeGenStorage;
import appeng.spatial.StorageWorldProvider;
import appeng.tile.AEBaseTile;
import appeng.worldgen.MeteoriteWorldGen;
import appeng.worldgen.QuartzWorldGen;
import com.google.common.base.Preconditions;
import net.minecraft.advancements.CriteriaTriggers;
import net.minecraft.advancements.ICriterionInstance;
import net.minecraft.advancements.ICriterionTrigger;
import net.minecraft.block.Block;
import net.minecraft.client.renderer.ItemMeshDefinition;
import net.minecraft.client.renderer.block.model.ModelResourceLocation;
import net.minecraft.client.renderer.block.statemap.IStateMapper;
import net.minecraft.item.Item;
import net.minecraft.item.crafting.IRecipe;
import net.minecraft.util.ResourceLocation;
import net.minecraft.world.DimensionType;
import net.minecraft.world.World;
import net.minecraft.world.biome.Biome;
import net.minecraftforge.client.event.ModelRegistryEvent;
import net.minecraftforge.client.model.ModelLoader;
import net.minecraftforge.common.DimensionManager;
import net.minecraftforge.common.MinecraftForge;
import net.minecraftforge.event.AttachCapabilitiesEvent;
import net.minecraftforge.event.RegistryEvent;
import net.minecraftforge.fml.common.FMLCommonHandler;
import net.minecraftforge.fml.common.event.FMLInitializationEvent;
import net.minecraftforge.fml.common.event.FMLPostInitializationEvent;
import net.minecraftforge.fml.common.event.FMLPreInitializationEvent;
import net.minecraftforge.fml.common.eventhandler.SubscribeEvent;
import net.minecraftforge.fml.common.registry.EntityEntry;
import net.minecraftforge.fml.common.registry.GameRegistry;
import net.minecraftforge.fml.relauncher.ReflectionHelper;
import net.minecraftforge.fml.relauncher.Side;
import net.minecraftforge.fml.relauncher.SideOnly;
import net.minecraftforge.registries.IForgeRegistry;
import javax.annotation.Nonnull;
import java.io.File;
import java.lang.reflect.InvocationTargetException;
import java.lang.reflect.Method;
import java.util.HashSet;
import java.util.Set;
final class Registration {
DimensionType storageDimensionType;
int storageDimensionID;
Biome storageBiome;
AdvancementTriggers advancementTriggers;
void preInitialize(final FMLPreInitializationEvent event) {
Capabilities.register();
final Api api = Api.INSTANCE;
final IRecipeHandlerRegistry recipeRegistry = api.registries().recipes();
this.registerCraftHandlers(recipeRegistry);
MinecraftForge.EVENT_BUS.register(OreDictionaryHandler.INSTANCE);
ApiDefinitions definitions = api.definitions();
// Register
definitions.getRegistry().getBootstrapComponents(IPreInitComponent.class).forEachRemaining(b -> b.preInitialize(event.getSide()));
}
private void registerSpatialBiome(IForgeRegistry<Biome> registry) {
if (!AEConfig.instance().isFeatureEnabled(AEFeature.SPATIAL_IO)) {
return;
}
if (this.storageBiome == null) {
this.storageBiome = new BiomeGenStorage();
}
registry.register(this.storageBiome.setRegistryName("appliedenergistics2:storage_biome"));
}
private void registerSpatialDimension() {
final AEConfig config = AEConfig.instance();
if (!config.isFeatureEnabled(AEFeature.SPATIAL_IO)) {
return;
}
if (config.getStorageProviderID() == -1) {
final Set<Integer> ids = new HashSet<>();
for (DimensionType type : DimensionType.values()) {
ids.add(type.getId());
}
int newId = -11;
while (ids.contains(newId)) {
--newId;
}
config.setStorageProviderID(newId);
config.save();
}
this.storageDimensionType = DimensionType.register("Storage Cell", "_cell", config.getStorageProviderID(), StorageWorldProvider.class, true);
if (config.getStorageDimensionID() == -1) {
config.setStorageDimensionID(DimensionManager.getNextFreeDimId());
config.save();
}
this.storageDimensionID = config.getStorageDimensionID();
DimensionManager.registerDimension(this.storageDimensionID, this.storageDimensionType);
}
private void registerCraftHandlers(final IRecipeHandlerRegistry registry) {
registry.addNewSubItemResolver(new AEItemResolver());
}
public void initialize(@Nonnull final FMLInitializationEvent event, @Nonnull final File recipeDirectory) {
Preconditions.checkNotNull(event);
Preconditions.checkNotNull(recipeDirectory);
Preconditions.checkArgument(!recipeDirectory.isFile());
final Api api = Api.INSTANCE;
final IRegistryContainer registries = api.registries();
ApiDefinitions definitions = api.definitions();
definitions.getRegistry().getBootstrapComponents(IInitComponent.class).forEachRemaining(b -> b.initialize(event.getSide()));
MinecraftForge.EVENT_BUS.register(TickHandler.INSTANCE);
MinecraftForge.EVENT_BUS.register(new PartPlacement());
if (AEConfig.instance().isFeatureEnabled(AEFeature.CHEST_LOOT)) {
MinecraftForge.EVENT_BUS.register(new ChestLoot());
}
final IGridCacheRegistry gcr = registries.gridCache();
gcr.registerGridCache(ITickManager.class, TickManagerCache.class);
gcr.registerGridCache(IEnergyGrid.class, EnergyGridCache.class);
gcr.registerGridCache(IPathingGrid.class, PathGridCache.class);
gcr.registerGridCache(IStorageGrid.class, GridStorageCache.class);
gcr.registerGridCache(P2PCache.class, P2PCache.class);
gcr.registerGridCache(ISpatialCache.class, SpatialPylonCache.class);
gcr.registerGridCache(ISecurityGrid.class, SecurityCache.class);
gcr.registerGridCache(ICraftingGrid.class, CraftingGridCache.class);
registries.cell().addCellHandler(new BasicCellHandler());
registries.cell().addCellHandler(new CreativeCellHandler());
registries.cell().addCellGuiHandler(new BasicItemCellGuiHandler());
registries.cell().addCellGuiHandler(new BasicFluidCellGuiHandler());
api.definitions().materials().matterBall().maybeStack(1).ifPresent(ammoStack ->
{
final double weight = 32;
registries.matterCannon().registerAmmo(ammoStack, weight);
});
PartItemPredicate.register();
Stats.register();
this.advancementTriggers = new AdvancementTriggers(new CriterionTrigggerRegistry());
}
@SubscribeEvent
public void registerBiomes(RegistryEvent.Register<Biome> event) {
final IForgeRegistry<Biome> registry = event.getRegistry();
this.registerSpatialBiome(registry);
}
@SubscribeEvent
@SideOnly(Side.CLIENT)
public void modelRegistryEvent(ModelRegistryEvent event) {
final ApiDefinitions definitions = Api.INSTANCE.definitions();
final IModelRegistry registry = new ModelLoaderWrapper();
final Side side = FMLCommonHandler.instance().getEffectiveSide();
definitions.getRegistry().getBootstrapComponents(IModelRegistrationComponent.class).forEachRemaining(b -> b.modelRegistration(side, registry));
}
@SubscribeEvent
public void registerBlocks(RegistryEvent.Register<Block> event) {
final IForgeRegistry<Block> registry = event.getRegistry();
final ApiDefinitions definitions = Api.INSTANCE.definitions();
final Side side = FMLCommonHandler.instance().getEffectiveSide();
definitions.getRegistry().getBootstrapComponents(IBlockRegistrationComponent.class).forEachRemaining(b -> b.blockRegistration(side, registry));
}
@SubscribeEvent
public void registerItems(RegistryEvent.Register<Item> event) {
final IForgeRegistry<Item> registry = event.getRegistry();
final ApiDefinitions definitions = Api.INSTANCE.definitions();
final Side side = FMLCommonHandler.instance().getEffectiveSide();
definitions.getRegistry().getBootstrapComponents(IItemRegistrationComponent.class).forEachRemaining(b -> b.itemRegistration(side, registry));
// register oredicts
definitions.getRegistry().getBootstrapComponents(IOreDictComponent.class).forEachRemaining(b -> b.oreRegistration(side));
ItemMaterial.instance.registerOredicts();
ItemPart.instance.registerOreDicts();
}
@SubscribeEvent
public void registerRecipes(RegistryEvent.Register<IRecipe> event) {
final IForgeRegistry<IRecipe> registry = event.getRegistry();
final Api api = Api.INSTANCE;
final ApiDefinitions definitions = api.definitions();
final Side side = FMLCommonHandler.instance().getEffectiveSide();
if (AEConfig.instance().isFeatureEnabled(AEFeature.ENABLE_DISASSEMBLY_CRAFTING)) {
DisassembleRecipe r = new DisassembleRecipe();
registry.register(r.setRegistryName(AppEng.MOD_ID.toLowerCase(), "disassemble"));
}
if (AEConfig.instance().isFeatureEnabled(AEFeature.ENABLE_FACADE_CRAFTING)) {
definitions.items().facade().maybeItem().ifPresent(facadeItem ->
{
FacadeRecipe f = new FacadeRecipe((ItemFacade) facadeItem);
registry.register(f.setRegistryName(AppEng.MOD_ID.toLowerCase(), "facade"));
});
}
definitions.getRegistry().getBootstrapComponents(IRecipeRegistrationComponent.class).forEachRemaining(b -> b.recipeRegistration(side, registry));
final AERecipeLoader ldr = new AERecipeLoader();
ldr.loadProcessingRecipes();
}
@SubscribeEvent
public void registerEntities(RegistryEvent.Register<EntityEntry> event) {
final IForgeRegistry<EntityEntry> registry = event.getRegistry();
final ApiDefinitions definitions = Api.INSTANCE.definitions();
definitions.getRegistry().getBootstrapComponents(IEntityRegistrationComponent.class).forEachRemaining(b -> b.entityRegistration(registry));
}
@SubscribeEvent
public void attachSpatialDimensionManager(AttachCapabilitiesEvent<World> event) {
if (AEConfig.instance()
.isFeatureEnabled(AEFeature.SPATIAL_IO) && event.getObject() == DimensionManager.getWorld(AEConfig.instance().getStorageDimensionID())) {
event.addCapability(new ResourceLocation("appliedenergistics2:spatial_dimension_manager"), new SpatialDimensionManager(event.getObject()));
}
}
void postInit(final FMLPostInitializationEvent event) {
final IRegistryContainer registries = Api.INSTANCE.registries();
ApiDefinitions definitions = Api.INSTANCE.definitions();
final IParts parts = definitions.parts();
final IBlocks blocks = definitions.blocks();
final IItems items = definitions.items();
this.registerSpatialDimension();
// default settings..
((P2PTunnelRegistry) registries.p2pTunnel()).configure();
// add to localization..
PlayerMessages.values();
GuiText.values();
definitions.getRegistry().getBootstrapComponents(IPostInitComponent.class).forEachRemaining(b -> b.postInitialize(event.getSide()));
// Interface
Upgrades.CRAFTING.registerItem(parts.iface(), 1);
Upgrades.CRAFTING.registerItem(blocks.iface(), 1);
Upgrades.PATTERN_EXPANSION.registerItem(parts.iface(), 3);
Upgrades.PATTERN_EXPANSION.registerItem(blocks.iface(), 3);
// IO Port!
Upgrades.SPEED.registerItem(blocks.iOPort(), 3);
Upgrades.REDSTONE.registerItem(blocks.iOPort(), 1);
// Level Emitter!
Upgrades.FUZZY.registerItem(parts.levelEmitter(), 1);
Upgrades.CRAFTING.registerItem(parts.levelEmitter(), 1);
// Import Bus
Upgrades.FUZZY.registerItem(parts.importBus(), 1);
Upgrades.REDSTONE.registerItem(parts.importBus(), 1);
Upgrades.CAPACITY.registerItem(parts.importBus(), 2);
Upgrades.SPEED.registerItem(parts.importBus(), 4);
// Fluid Import Bus
Upgrades.CAPACITY.registerItem(parts.fluidImportBus(), 2);
Upgrades.REDSTONE.registerItem(parts.fluidImportBus(), 1);
Upgrades.SPEED.registerItem(parts.fluidImportBus(), 4);
// Export Bus
Upgrades.FUZZY.registerItem(parts.exportBus(), 1);
Upgrades.REDSTONE.registerItem(parts.exportBus(), 1);
Upgrades.CAPACITY.registerItem(parts.exportBus(), 2);
Upgrades.SPEED.registerItem(parts.exportBus(), 4);
Upgrades.CRAFTING.registerItem(parts.exportBus(), 1);
// Fluid Export Bus
Upgrades.CAPACITY.registerItem(parts.fluidExportBus(), 2);
Upgrades.REDSTONE.registerItem(parts.fluidExportBus(), 1);
Upgrades.SPEED.registerItem(parts.fluidExportBus(), 4);
// Storage Cells
Upgrades.FUZZY.registerItem(items.cell1k(), 1);
Upgrades.INVERTER.registerItem(items.cell1k(), 1);
Upgrades.FUZZY.registerItem(items.cell4k(), 1);
Upgrades.INVERTER.registerItem(items.cell4k(), 1);
Upgrades.FUZZY.registerItem(items.cell16k(), 1);
Upgrades.INVERTER.registerItem(items.cell16k(), 1);
Upgrades.FUZZY.registerItem(items.cell64k(), 1);
Upgrades.INVERTER.registerItem(items.cell64k(), 1);
Upgrades.FUZZY.registerItem(items.portableCell(), 1);
Upgrades.INVERTER.registerItem(items.portableCell(), 1);
Upgrades.FUZZY.registerItem(items.viewCell(), 1);
Upgrades.INVERTER.registerItem(items.viewCell(), 1);
// Storage Bus
Upgrades.FUZZY.registerItem(parts.storageBus(), 1);
Upgrades.INVERTER.registerItem(parts.storageBus(), 1);
Upgrades.CAPACITY.registerItem(parts.storageBus(), 5);
// Storage Bus Fluids
Upgrades.INVERTER.registerItem(parts.fluidStorageBus(), 1);
Upgrades.CAPACITY.registerItem(parts.fluidStorageBus(), 5);
// Formation Plane
Upgrades.FUZZY.registerItem(parts.formationPlane(), 1);
Upgrades.INVERTER.registerItem(parts.formationPlane(), 1);
Upgrades.CAPACITY.registerItem(parts.formationPlane(), 5);
// Matter Cannon
Upgrades.FUZZY.registerItem(items.massCannon(), 1);
Upgrades.INVERTER.registerItem(items.massCannon(), 1);
Upgrades.SPEED.registerItem(items.massCannon(), 4);
// Molecular Assembler
Upgrades.SPEED.registerItem(blocks.molecularAssembler(), 5);
// Inscriber
Upgrades.SPEED.registerItem(blocks.inscriber(), 3);
// Wireless Terminal Handler
items.wirelessTerminal().maybeItem().ifPresent(terminal -> registries.wireless().registerWirelessHandler((IWirelessTermHandler) terminal));
// Charge Rates
items.chargedStaff().maybeItem().ifPresent(chargedStaff -> registries.charger().addChargeRate(chargedStaff, 320d));
items.portableCell().maybeItem().ifPresent(chargedStaff -> registries.charger().addChargeRate(chargedStaff, 800d));
items.colorApplicator().maybeItem().ifPresent(colorApplicator -> registries.charger().addChargeRate(colorApplicator, 800d));
items.wirelessTerminal().maybeItem().ifPresent(terminal -> registries.charger().addChargeRate(terminal, 8000d));
items.entropyManipulator().maybeItem().ifPresent(entropyManipulator -> registries.charger().addChargeRate(entropyManipulator, 8000d));
items.massCannon().maybeItem().ifPresent(massCannon -> registries.charger().addChargeRate(massCannon, 8000d));
blocks.energyCell().maybeItem().ifPresent(cell -> registries.charger().addChargeRate(cell, 8000d));
blocks.energyCellDense().maybeItem().ifPresent(cell -> registries.charger().addChargeRate(cell, 16000d));
// add villager trading to black smiths for a few basic materials
if (AEConfig.instance().isFeatureEnabled(AEFeature.VILLAGER_TRADING)) {
// TODO: VILLAGER TRADING
// VillagerRegistry.instance().getRegisteredVillagers().registerVillageTradeHandler( 3, new AETrading() );
}
if (AEConfig.instance().isFeatureEnabled(AEFeature.CERTUS_QUARTZ_WORLD_GEN)) {
GameRegistry.registerWorldGenerator(new QuartzWorldGen(), 0);
}
if (AEConfig.instance().isFeatureEnabled(AEFeature.METEORITE_WORLD_GEN)) {
GameRegistry.registerWorldGenerator(new MeteoriteWorldGen(), 0);
}
final IMovableRegistry mr = registries.movable();
/*
* You can't move bed rock.
*/
mr.blacklistBlock(net.minecraft.init.Blocks.BEDROCK);
/*
* White List Vanilla...
*/
mr.whiteListTileEntity(net.minecraft.tileentity.TileEntityBanner.class);
mr.whiteListTileEntity(net.minecraft.tileentity.TileEntityBeacon.class);
mr.whiteListTileEntity(net.minecraft.tileentity.TileEntityBrewingStand.class);
mr.whiteListTileEntity(net.minecraft.tileentity.TileEntityChest.class);
mr.whiteListTileEntity(net.minecraft.tileentity.TileEntityCommandBlock.class);
mr.whiteListTileEntity(net.minecraft.tileentity.TileEntityComparator.class);
mr.whiteListTileEntity(net.minecraft.tileentity.TileEntityDaylightDetector.class);
mr.whiteListTileEntity(net.minecraft.tileentity.TileEntityDispenser.class);
mr.whiteListTileEntity(net.minecraft.tileentity.TileEntityDropper.class);
mr.whiteListTileEntity(net.minecraft.tileentity.TileEntityEnchantmentTable.class);
mr.whiteListTileEntity(net.minecraft.tileentity.TileEntityEnderChest.class);
mr.whiteListTileEntity(net.minecraft.tileentity.TileEntityEndPortal.class);
mr.whiteListTileEntity(net.minecraft.tileentity.TileEntityFlowerPot.class);
mr.whiteListTileEntity(net.minecraft.tileentity.TileEntityFurnace.class);
mr.whiteListTileEntity(net.minecraft.tileentity.TileEntityHopper.class);
mr.whiteListTileEntity(net.minecraft.tileentity.TileEntityMobSpawner.class);
mr.whiteListTileEntity(net.minecraft.tileentity.TileEntityNote.class);
mr.whiteListTileEntity(net.minecraft.tileentity.TileEntityPiston.class);
mr.whiteListTileEntity(net.minecraft.tileentity.TileEntityShulkerBox.class);
mr.whiteListTileEntity(net.minecraft.tileentity.TileEntitySign.class);
mr.whiteListTileEntity(net.minecraft.tileentity.TileEntitySkull.class);
/*
* Whitelist AE2
*/
mr.whiteListTileEntity(AEBaseTile.class);
/*
* world gen
*/
for (final WorldGenType type : WorldGenType.values()) {
registries.worldgen().disableWorldGenForProviderID(type, StorageWorldProvider.class);
// nether
registries.worldgen().disableWorldGenForDimension(type, -1);
// end
registries.worldgen().disableWorldGenForDimension(type, 1);
}
// whitelist from config
for (final int dimension : AEConfig.instance().getMeteoriteDimensionWhitelist()) {
registries.worldgen().enableWorldGenForDimension(WorldGenType.METEORITES, dimension);
}
}
private static class ModelLoaderWrapper implements IModelRegistry {
@Override
public void registerItemVariants(Item item, ResourceLocation... names) {
ModelLoader.registerItemVariants(item, names);
}
@Override
public void setCustomModelResourceLocation(Item item, int metadata, ModelResourceLocation model) {
ModelLoader.setCustomModelResourceLocation(item, metadata, model);
}
@Override
public void setCustomMeshDefinition(Item item, ItemMeshDefinition meshDefinition) {
ModelLoader.setCustomMeshDefinition(item, meshDefinition);
}
@Override
public void setCustomStateMapper(Block block, IStateMapper mapper) {
ModelLoader.setCustomStateMapper(block, mapper);
}
}
private static class CriterionTrigggerRegistry implements ICriterionTriggerRegistry {
private final Method method;
CriterionTrigggerRegistry() {
this.method = ReflectionHelper.findMethod(CriteriaTriggers.class, "register", "func_192118_a", ICriterionTrigger.class);
this.method.setAccessible(true);
}
@Override
public void register(ICriterionTrigger<? extends ICriterionInstance> trigger) {
try {
this.method.invoke(null, trigger);
} catch (IllegalAccessException | IllegalArgumentException | InvocationTargetException e) {
AELog.debug(e);
}
}
}
}