package info.u_team.u_team_test.init;

import info.u_team.u_team_core.util.registry.MainThreadWorker;
import info.u_team.u_team_test.TestMod;
import info.u_team.u_team_test.screen.*;
import net.minecraft.client.gui.ScreenManager;
import net.minecraftforge.api.distmarker.Dist;
import net.minecraftforge.eventbus.api.SubscribeEvent;
import net.minecraftforge.fml.common.Mod.EventBusSubscriber;
import net.minecraftforge.fml.common.Mod.EventBusSubscriber.Bus;
import net.minecraftforge.fml.event.lifecycle.FMLClientSetupEvent;

@EventBusSubscriber(modid = TestMod.MODID, bus = Bus.MOD, value = Dist.CLIENT)
public class TestScreens {
	
	@SubscribeEvent
	public static void register(FMLClientSetupEvent event) {
		MainThreadWorker.run(() -> {
			ScreenManager.registerFactory(TestContainers.BASIC.get(), BasicTileEntityScreen::new);
			ScreenManager.registerFactory(TestContainers.BASIC_ENERGY_CREATOR.get(), BasicEnergyCreatorScreen::new);
			ScreenManager.registerFactory(TestContainers.BASIC_FLUID_INVENTORY.get(), BasicFluidInventoryScreen::new);
		});
	}
}

@@ -77,32 +77,17 @@
 egistry.
-MainThreadWorker
+*
 ;%0Aimport
@@ -165,55 +165,8 @@
 .*;%0A
-import net.minecraft.client.gui.ScreenManager;%0A
 impo
@@ -678,37 +678,37 @@
 %0A%09%09%09
-ScreenManager.registerFactory
+ClientRegistry.registryScreen
 (Tes
@@ -724,22 +724,16 @@
 rs.BASIC
-.get()
 , BasicT
@@ -762,37 +762,37 @@
 %0A%09%09%09
-ScreenManager.registerFactory
+ClientRegistry.registryScreen
 (Tes
@@ -823,22 +823,16 @@
 _CREATOR
-.get()
 , BasicE
@@ -864,37 +864,37 @@
 %0A%09%09%09
-ScreenManager.registerFactory
+ClientRegistry.registryScreen
 (Tes
@@ -930,14 +930,8 @@
 TORY
-.get()
 , Ba

Use the ClientRegistry Method in screen register to make it allow the registry object supplier