IESaveData.java

package blusunrize.immersiveengineering.common;

import blusunrize.immersiveengineering.api.DimensionChunkCoords;
import blusunrize.immersiveengineering.api.energy.wires.IICProxy;
import blusunrize.immersiveengineering.api.energy.wires.ImmersiveNetHandler;
import blusunrize.immersiveengineering.api.energy.wires.ImmersiveNetHandler.Connection;
import blusunrize.immersiveengineering.api.shader.ShaderRegistry;
import blusunrize.immersiveengineering.api.tool.ExcavatorHandler;
import blusunrize.immersiveengineering.api.tool.ExcavatorHandler.MineralWorldInfo;
import net.minecraft.nbt.NBTTagCompound;
import net.minecraft.nbt.NBTTagList;
import net.minecraft.nbt.NBTTagString;
import net.minecraft.world.World;
import net.minecraft.world.WorldSavedData;
import net.minecraftforge.fml.common.FMLCommonHandler;
import net.minecraftforge.fml.relauncher.Side;

import java.util.Map;

public class IESaveData extends WorldSavedData
{
	//	private static HashMap<Integer, IESaveData> INSTANCE = new HashMap<Integer, IESaveData>();
	private static IESaveData INSTANCE;
	public static final String dataName = "ImmersiveEngineering-SaveData";

	public IESaveData(String s)
	{
		super(s);
	}

	@Override
	public void readFromNBT(NBTTagCompound nbt)
	{
		int[] savedDimensions = nbt.getIntArray("savedDimensions");
		for(int dim: savedDimensions)
		{
			NBTTagList connectionList = nbt.getTagList("connectionList"+dim, 10);
			ImmersiveNetHandler.INSTANCE.clearAllConnections(dim);
			for(int i=0; i<connectionList.tagCount(); i++)
			{
				NBTTagCompound conTag = connectionList.getCompoundTagAt(i);
				Connection con = Connection.readFromNBT(conTag);
				if(con!=null)
				{
					ImmersiveNetHandler.INSTANCE.addConnection(dim, con.start, con);
				}
			}
		}

		NBTTagList proxies = nbt.getTagList("iicProxies", 10);
		for (int i = 0;i<proxies.tagCount();i++)
			ImmersiveNetHandler.INSTANCE.addProxy(IICProxy.readFromNBT(proxies.getCompoundTagAt(i)));
		
		EventHandler.validateConnsNextTick = true;

		NBTTagList mineralList = nbt.getTagList("mineralDepletion", 10);
		ExcavatorHandler.mineralCache.clear();		
		for(int i=0; i<mineralList.tagCount(); i++)
		{
			NBTTagCompound tag = mineralList.getCompoundTagAt(i);
			DimensionChunkCoords coords = DimensionChunkCoords.readFromNBT(tag);
			if(coords!=null)
			{
				MineralWorldInfo info = MineralWorldInfo.readFromNBT(tag.getCompoundTag("info"));
				ExcavatorHandler.mineralCache.put(coords, info);
			}
		}


		NBTTagList receivedShaderList = nbt.getTagList("receivedShaderList", 10);
		for(int i=0; i<receivedShaderList.tagCount(); i++)
		{
			NBTTagCompound tag = receivedShaderList.getCompoundTagAt(i);
			String player = tag.getString("player"); 
			ShaderRegistry.receivedShaders.get(player).clear();

			NBTTagList playerReceived = tag.getTagList("received", 8);
			for(int j=0; j<playerReceived.tagCount(); j++)
			{
				String s = playerReceived.getStringTagAt(j);
				if(s!=null && !s.isEmpty())
					ShaderRegistry.receivedShaders.put(player, s);
			}
		}
	}

	@Override
	public NBTTagCompound writeToNBT(NBTTagCompound nbt)
	{
		Integer[] relDim = ImmersiveNetHandler.INSTANCE.getRelevantDimensions().toArray(new Integer[0]);
		int[] savedDimensions = new int[relDim.length];
		for(int ii=0; ii<relDim.length; ii++)
			savedDimensions[ii] = relDim[ii];

		nbt.setIntArray("savedDimensions", savedDimensions);
		for(int dim: savedDimensions)
		{
			NBTTagList connectionList = new NBTTagList();
			for(Connection con : ImmersiveNetHandler.INSTANCE.getAllConnections(dim))
			{
				connectionList.appendTag(con.writeToNBT());
			}
			nbt.setTag("connectionList"+dim, connectionList);
		}
		
		NBTTagList proxies = new NBTTagList();
		for (IICProxy iic:ImmersiveNetHandler.INSTANCE.proxies.values())
			proxies.appendTag(iic.writeToNBT());
		nbt.setTag("iicProxies", proxies);

		NBTTagList mineralList = new NBTTagList();
		for(Map.Entry<DimensionChunkCoords,MineralWorldInfo> e: ExcavatorHandler.mineralCache.entrySet())
			if(e.getKey()!=null && e.getValue()!=null)
			{
				NBTTagCompound tag = e.getKey().writeToNBT();
				tag.setTag("info", e.getValue().writeToNBT());
				mineralList.appendTag(tag);
			}
		nbt.setTag("mineralDepletion", mineralList);


		NBTTagList receivedShaderList = new NBTTagList();
		for(String player : ShaderRegistry.receivedShaders.keySet())
		{
			NBTTagCompound tag = new NBTTagCompound();
			tag.setString("player", player);
			NBTTagList playerReceived = new NBTTagList();
			for(String shader : ShaderRegistry.receivedShaders.get(player))
				if(shader!=null && !shader.isEmpty())
					playerReceived.appendTag(new NBTTagString(shader));
			tag.setTag("received", playerReceived);
			receivedShaderList.appendTag(tag);
		}
		nbt.setTag("receivedShaderList", receivedShaderList);
		return nbt;
	}


	public static void setDirty(int dimension)
	{
		//		if(FMLCommonHandler.instance().getEffectiveSide()==Side.SERVER && INSTANCE.get(dimension)!=null)
		//		{
		//			INSTANCE.get(dimension).markDirty();
		//		}
		if(FMLCommonHandler.instance().getEffectiveSide()==Side.SERVER && INSTANCE!=null)
			INSTANCE.markDirty();
	}
	public static void setInstance(int dimension, IESaveData in)
	{
		//		if(FMLCommonHandler.instance().getEffectiveSide()==Side.SERVER)
		//			INSTANCE.put(dimension, in);
		if(FMLCommonHandler.instance().getEffectiveSide()==Side.SERVER)
			INSTANCE=in;
	}

}