/**
* Copyright (c) Lambda Innovation, 2013-2015
* 本作品版权由Lambda Innovation所有。
* http://www.li-dev.cn/
* <p/>
* This project is open-source, and it is distributed under
* the terms of GNU General Public License. You can modify
* and distribute freely as long as you follow the license.
* 本项目是一个开源项目，且遵循GNU通用公共授权协议。
* 在遵照该协议的情况下，您可以自由传播和修改。
* http://www.gnu.org/licenses/gpl.html
*/
package cn.academy.energy.internal;
import cn.academy.energy.api.block.*;
import cn.academy.energy.internal.VBlocks.*;
import cn.liutils.util.mc.IBlockFilter;
import cn.liutils.util.mc.WorldUtils;
import net.minecraft.block.Block;
import net.minecraft.nbt.NBTTagCompound;
import net.minecraft.nbt.NBTTagList;
import net.minecraft.tileentity.TileEntity;
import net.minecraft.util.BlockPos;
import net.minecraft.world.World;
import net.minecraft.world.WorldSavedData;
import java.util.*;
/**
* @author WeAthFolD
*
*/
public class WiWorldData extends WorldSavedData {
public static final String ID = "AC_WEN";
//Set by get method, which should be the ONLY way to access WiWorldData
World world;
/***
* Object valid for lookup: VWMatrix, VWNode, String ssid
*/
Map<Object, WirelessNet> netLookup = new HashMap();
//-----WEN-----
Set<WirelessNet> netList = new HashSet();
//-----NodeConn----
Map<Object, NodeConn> nodeLookup = new HashMap();
Set<NodeConn> nodeList = new HashSet();
List<NodeConn> nToRemove = new ArrayList();
private IBlockFilter filterWirelessBlocks = new IBlockFilter() {
@Override
public boolean accepts(World world, int x, int y, int z, Block block) {
TileEntity te = world.getTileEntity(new BlockPos(x, y, z));
return te instanceof IWirelessMatrix || te instanceof IWirelessNode;
}
};
/**
* Internal, used to prevent concurrent modification.
*/
private List<WirelessNet> toRemove = new ArrayList();
public WiWorldData(String www) {
super(ID);
}
public static WiWorldData get(World world) {
if (world.isRemote) {
throw new RuntimeException("Not allowed to create WiWorldData in client");
}
WiWorldData ret = (WiWorldData) world.loadItemData(WiWorldData.class, WiWorldData.ID);
if (ret == null) {
world.setItemData(ID, ret = new WiWorldData(ID));
}
ret.world = world;
return ret;
}
private void tickNetwork() {
for (WirelessNet net : toRemove) {
this.doRemoveNetwork(net);
}
toRemove.clear();
Iterator<WirelessNet> iter = netList.iterator();
while (iter.hasNext()) {
WirelessNet net = iter.next();
if (net.isDisposed()) {
toRemove.add(net);
} else {
net.world = world;
net.tick();
}
}
}
boolean createNetwork(IWirelessMatrix matrix, String ssid, String password) {
if (netLookup.containsKey(ssid)) { //Doesn't allow ssid duplication
return false;
}
// Kill old net of the same matrix, if any
VWMatrix vm = new VWMatrix(matrix);
if (netLookup.containsKey(vm)) {
WirelessNet old = netLookup.get(vm);
doRemoveNetwork(old);
}
//Add new
WirelessNet net = new WirelessNet(this, vm, ssid, password);
doAddNetwork(net);
return true;
}
public Collection<WirelessNet> rangeSearch(int x, int y, int z, double range, int max) {
Collection<BlockPos> bps = WorldUtils.getBlocksWithin(world, x, y, z, range, max, filterWirelessBlocks);
Set<WirelessNet> set = new HashSet();
for (BlockPos bp : bps) {
TileEntity te = world.getTileEntity(bp);
WirelessNet net;
if (te instanceof IWirelessMatrix) {
net = getNetwork((IWirelessMatrix) te);
} else if (te instanceof IWirelessNode) {
net = getNetwork((IWirelessNode) te);
} else {
throw new RuntimeException("Invalid TileEntity");
}
if (net != null && net.isInRange(x, y, z)) {
set.add(net);
if (set.size() >= max)
return set;
}
}
return set;
}
public WirelessNet getNetwork(String ssid) {
return netLookup.get(ssid);
}
public WirelessNet getNetwork(IWirelessMatrix matrix) {
System.out.println(matrix);
return netLookup.get(new VWMatrix(matrix));
}
public WirelessNet getNetwork(IWirelessNode node) {
return netLookup.get(new VWNode(node));
}
private void doRemoveNetwork(WirelessNet net) {
debug("DoRemoveNet" + net.ssid);
netList.remove(net);
net.onCleanup(this);
}
private void doAddNetwork(WirelessNet net) {
netList.add(net);
net.onCreate(this);
}
private void loadNetwork(NBTTagCompound tag) {
NBTTagList list = (NBTTagList) tag.getTag("networks");
for (int i = 0; i < list.tagCount(); ++i) {
NBTTagCompound tag2 = list.getCompoundTagAt(i);
WirelessNet net = new WirelessNet(this, tag2);
doAddNetwork(net);
}
debug("WEN: Loaded " + list.tagCount() + " nets in " + world);
}
private void saveNetwork(NBTTagCompound tag) {
NBTTagList list = new NBTTagList();
for (WirelessNet net : netList) {
if (!net.isDisposed()) {
list.appendTag(net.toNBT());
}
}
tag.setTag("networks", list);
}
/**
* Get the node connection of a node. If not found will create a new one. Never returns null.
*/
public NodeConn getNodeConnection(IWirelessNode node) {
VNNode vnn = new VNNode(node);
NodeConn ret = nodeLookup.get(vnn);
if (ret == null) {
doAddNode(ret = new NodeConn(this, vnn));
}
return ret;
}
public NodeConn getNodeConnection(IWirelessUser user) {
if (user instanceof IWirelessGenerator) {
return nodeLookup.get(new VNGenerator((IWirelessGenerator) user));
} else if (user instanceof IWirelessReceiver) {
return nodeLookup.get(new VNReceiver((IWirelessReceiver) user));
} else
return null;
}
private void tickNode() {
for (NodeConn nc : nToRemove) {
doRemoveNode(nc);
}
nToRemove.clear();
Iterator<NodeConn> iter = nodeList.iterator();
while (iter.hasNext()) {
NodeConn conn = iter.next();
if (conn.isDisposed()) {
nToRemove.add(conn);
} else {
conn.tick();
}
}
}
private void doAddNode(NodeConn conn) {
nodeList.add(conn);
conn.onAdded(this);
}
private void doRemoveNode(NodeConn conn) {
nodeList.remove(conn);
conn.onCleanup(this);
}
private void loadNode(NBTTagCompound tag) {
NBTTagList list = (NBTTagList) tag.getTag("list");
for (int i = 0; i < list.tagCount(); ++i) {
doAddNode(new NodeConn(this, list.getCompoundTagAt(i)));
}
}
private void saveNode(NBTTagCompound tag) {
NBTTagList list = new NBTTagList();
for (NodeConn c : nodeList) {
if (!c.isDisposed()) {
list.appendTag(c.toNBT());
}
}
tag.setTag("list", list);
}
//-----Generic-----
public void tick() {
this.markDirty();
tickNetwork();
tickNode();
}
@Override
public void readFromNBT(NBTTagCompound tag) {
NBTTagCompound tag1 = (NBTTagCompound) tag.getTag("net");
if (tag1 != null)
loadNetwork(tag1);
tag1 = (NBTTagCompound) tag.getTag("node");
if (tag1 != null)
loadNode(tag1);
}
@Override
public void writeToNBT(NBTTagCompound tag) {
NBTTagCompound tag1 = new NBTTagCompound();
saveNetwork(tag1);
tag.setTag("net", tag1);
tag1 = new NBTTagCompound();
saveNode(tag1);
tag.setTag("node", tag1);
}
private void debug(Object msg) {
//AcademyCraft.log.info("WiWorldData: " + msg);
}
private class ChunkCoord {
int cx, cz;
public ChunkCoord(int _cx, int _cz) {
cx = _cx;
cz = _cz;
}
public boolean isLoaded() {
return world.getChunkProvider().chunkExists(cx, cz);
}
@Override
public int hashCode() {
return cx ^ cz;
}
@Override
public boolean equals(Object obj) {
if (obj instanceof ChunkCoord) {
ChunkCoord cc = (ChunkCoord) obj;
return cc.cx == cx && cc.cz == cz;
}
return false;
}
}
}