134 lines
7 KiB
Java
134 lines
7 KiB
Java
package com.atsuishio.superbwarfare.init;
|
|
|
|
import com.atsuishio.superbwarfare.Mod;
|
|
import com.atsuishio.superbwarfare.item.*;
|
|
import com.atsuishio.superbwarfare.tools.NBTTool;
|
|
import net.minecraft.core.HolderLookup;
|
|
import net.minecraft.core.registries.Registries;
|
|
import net.minecraft.network.chat.Component;
|
|
import net.minecraft.world.flag.FeatureFlagSet;
|
|
import net.minecraft.world.item.CreativeModeTab;
|
|
import net.minecraft.world.item.CreativeModeTabs;
|
|
import net.minecraft.world.item.Item;
|
|
import net.minecraft.world.item.ItemStack;
|
|
import net.minecraft.world.item.alchemy.Potion;
|
|
import net.minecraft.world.item.alchemy.PotionContents;
|
|
import net.neoforged.bus.api.SubscribeEvent;
|
|
import net.neoforged.fml.common.EventBusSubscriber;
|
|
import net.neoforged.neoforge.capabilities.Capabilities;
|
|
import net.neoforged.neoforge.event.BuildCreativeModeTabContentsEvent;
|
|
import net.neoforged.neoforge.registries.DeferredHolder;
|
|
import net.neoforged.neoforge.registries.DeferredRegister;
|
|
|
|
import java.util.function.Supplier;
|
|
|
|
@EventBusSubscriber(bus = EventBusSubscriber.Bus.MOD)
|
|
@SuppressWarnings("unused")
|
|
public class ModTabs {
|
|
|
|
public static final DeferredRegister<CreativeModeTab> TABS = DeferredRegister.create(Registries.CREATIVE_MODE_TAB, Mod.MODID);
|
|
|
|
public static final DeferredHolder<CreativeModeTab, CreativeModeTab> GUN_TAB = TABS.register("guns",
|
|
() -> CreativeModeTab.builder()
|
|
.title(Component.translatable("item_group.superbwarfare.guns"))
|
|
.icon(() -> new ItemStack(ModItems.TASER.get()))
|
|
.displayItems((param, output) -> ModItems.GUNS.getEntries().forEach(registryObject -> {
|
|
// 普通枪械
|
|
var stack = new ItemStack(registryObject.get());
|
|
var id = stack.getDescriptionId();
|
|
var tag = NBTTool.getTag(stack);
|
|
tag.putString("id", id.substring(id.lastIndexOf(".") + 1));
|
|
NBTTool.saveTag(stack, tag);
|
|
|
|
output.accept(stack);
|
|
|
|
// 充电后枪械
|
|
var newStack = stack.copy();
|
|
var cap = newStack.getCapability(Capabilities.EnergyStorage.ITEM);
|
|
|
|
if (cap != null) {
|
|
cap.receiveEnergy(Integer.MAX_VALUE, false);
|
|
output.accept(newStack);
|
|
}
|
|
}))
|
|
.build());
|
|
|
|
public static final DeferredHolder<CreativeModeTab, CreativeModeTab> PERK_TAB = TABS.register("perk",
|
|
() -> CreativeModeTab.builder()
|
|
.title(Component.translatable("item_group.superbwarfare.perk"))
|
|
.icon(() -> new ItemStack(ModItems.AP_BULLET.get()))
|
|
.withTabsBefore(GUN_TAB.getKey())
|
|
.displayItems((param, output) -> ModItems.PERKS.getEntries().forEach(registryObject -> output.accept(registryObject.get())))
|
|
.build());
|
|
|
|
public static final DeferredHolder<CreativeModeTab, CreativeModeTab> AMMO_TAB = TABS.register("ammo",
|
|
() -> CreativeModeTab.builder()
|
|
.title(Component.translatable("item_group.superbwarfare.ammo"))
|
|
.icon(() -> new ItemStack(ModItems.SHOTGUN_AMMO_BOX.get()))
|
|
.withTabsBefore(PERK_TAB.getKey())
|
|
.displayItems((param, output) -> {
|
|
ModItems.AMMO.getEntries().forEach(registryObject -> {
|
|
if (registryObject.get() != ModItems.POTION_MORTAR_SHELL.get()) {
|
|
output.accept(registryObject.get());
|
|
|
|
if (registryObject.get() == ModItems.C4_BOMB.get()) {
|
|
output.accept(C4Bomb.makeInstance());
|
|
}
|
|
}
|
|
});
|
|
|
|
param.holders().lookup(Registries.POTION)
|
|
.ifPresent(potion -> generatePotionEffectTypes(output, potion, ModItems.POTION_MORTAR_SHELL.get(),
|
|
CreativeModeTab.TabVisibility.PARENT_AND_SEARCH_TABS,
|
|
param.enabledFeatures()));
|
|
})
|
|
.build());
|
|
|
|
public static final DeferredHolder<CreativeModeTab, CreativeModeTab> ITEM_TAB = TABS.register("item",
|
|
() -> CreativeModeTab.builder()
|
|
.title(Component.translatable("item_group.superbwarfare.item"))
|
|
.icon(() -> new ItemStack(ModItems.TARGET_DEPLOYER.get()))
|
|
.displayItems((param, output) -> ModItems.ITEMS.getEntries().forEach(registryObject -> {
|
|
output.accept(registryObject.get());
|
|
if (registryObject.get() == ModItems.ARMOR_PLATE.get()) {
|
|
output.accept(ArmorPlate.getInfiniteInstance());
|
|
}
|
|
if (registryObject.get() instanceof BatteryItem batteryItem) {
|
|
output.accept(batteryItem.makeFullEnergyStack());
|
|
}
|
|
}))
|
|
.build());
|
|
|
|
public static final DeferredHolder<CreativeModeTab, CreativeModeTab> BLOCK_TAB = TABS.register("block",
|
|
() -> CreativeModeTab.builder()
|
|
.title(Component.translatable("item_group.superbwarfare.block"))
|
|
.icon(() -> new ItemStack(ModItems.SANDBAG.get()))
|
|
.withTabsBefore(ITEM_TAB.getKey())
|
|
.displayItems((param, output) -> ModItems.BLOCKS.getEntries().forEach(registryObject -> {
|
|
if (registryObject.get() == ModItems.CONTAINER.get()) {
|
|
ContainerBlockItem.CONTAINER_ENTITIES.stream().map(Supplier::get).forEach(output::accept);
|
|
} else if (registryObject.get() == ModItems.SMALL_CONTAINER.get()) {
|
|
output.accept(registryObject.get());
|
|
SmallContainerBlockItem.SMALL_CONTAINER_LOOT_TABLES.stream().map(Supplier::get).forEach(output::accept);
|
|
} else {
|
|
output.accept(registryObject.get());
|
|
}
|
|
}))
|
|
.build());
|
|
|
|
@SubscribeEvent
|
|
public static void buildTabContentsVanilla(BuildCreativeModeTabContentsEvent tabData) {
|
|
if (tabData.getTabKey() == CreativeModeTabs.SPAWN_EGGS) {
|
|
tabData.accept(ModItems.SENPAI_SPAWN_EGG.get());
|
|
}
|
|
}
|
|
|
|
private static void generatePotionEffectTypes(
|
|
CreativeModeTab.Output output, HolderLookup<Potion> potions, Item item, CreativeModeTab.TabVisibility visibility, FeatureFlagSet requiredFeatures
|
|
) {
|
|
potions.listElements()
|
|
.filter(potion -> potion.value().isEnabled(requiredFeatures))
|
|
.map(potion -> PotionContents.createItemStack(item, potion))
|
|
.forEach(itemStack -> output.accept(itemStack, visibility));
|
|
}
|
|
}
|