package info.u_team.u_team_core.util;

import static net.minecraft.client.renderer.model.ModelBakery.*;

import java.util.*;

import com.google.common.collect.*;

import net.minecraft.block.*;
import net.minecraft.client.renderer.model.RenderMaterial;
import net.minecraft.state.StateContainer;
import net.minecraft.util.ResourceLocation;

public class ModelUtil {
	
	static {
		if (STATE_CONTAINER_OVERRIDES instanceof ImmutableMap) {
			final Map<ResourceLocation, StateContainer<Block, BlockState>> mutableMap = new HashMap<>();
			STATE_CONTAINER_OVERRIDES.forEach(mutableMap::put);
			STATE_CONTAINER_OVERRIDES = mutableMap;
		}
	}
	
	public static void addCustomStateContainer(ResourceLocation location, StateContainer<Block, BlockState> container) {
		STATE_CONTAINER_OVERRIDES.put(location, container);
	}
	
	public static void addTexture(RenderMaterial material) {
		LOCATIONS_BUILTIN_TEXTURES.add(material);
	}
	
	public static class EmptyStateContainer extends StateContainer<Block, BlockState> {
		
		public EmptyStateContainer(Block block) {
			super(Block::getDefaultState, block, BlockState::new, new HashMap<>());
		}
		
		@Override
		public ImmutableList<BlockState> getValidStates() {
			return getOwner().getStateContainer().getValidStates();
		}
		
	}
	
}

@@ -229,30 +229,17 @@
 r.model.
-RenderMaterial
+*
 ;%0Aimport
@@ -319,16 +319,262 @@
 cation;%0A
+import net.minecraftforge.client.event.ModelRegistryEvent;%0A%0A/**%0A * Utility methods for interaction with %7B@link ModelBakery%7D private methods.%3Cbr%3E%0A * All methods should be called in the %7B@link ModelRegistryEvent%7D event.%0A * %0A * @author HyCraftHD%0A */
 %0Apublic 
@@ -861,24 +861,232 @@
 p;%0A%09%09%7D%0A%09%7D%0A%09%0A
+%09/**%0A%09 * Replace the default state container with a custom one. Can remove or add states for the model%0A%09 * %0A%09 * @param location Resource location of the model%0A%09 * @param container Custom state container%0A%09 */%0A
 %09public stat
@@ -1242,32 +1242,162 @@
 ontainer);%0A%09%7D%0A%09%0A
+%09/**%0A%09 * Add %7B@link RenderMaterial%7D to be loaded to the atlas texture and stitched.%0A%09 * %0A%09 * @param material Render Material%0A%09 */%0A
 %09public static v

Add java doc