mirror of
https://github.com/mcMMO-Dev/mcMMO.git
synced 2025-04-02 17:56:23 +02:00
303 lines
9.6 KiB
Java
303 lines
9.6 KiB
Java
package com.gmail.nossr50.util.blockmeta;
|
|
|
|
import java.io.EOFException;
|
|
import java.io.File;
|
|
import java.io.FileInputStream;
|
|
import java.io.FileOutputStream;
|
|
import java.io.IOException;
|
|
import java.io.ObjectInputStream;
|
|
import java.io.ObjectOutputStream;
|
|
import java.io.StreamCorruptedException;
|
|
import java.io.UTFDataFormatException;
|
|
import java.util.HashMap;
|
|
|
|
import org.bukkit.Bukkit;
|
|
import org.bukkit.World;
|
|
import org.bukkit.block.Block;
|
|
|
|
import com.gmail.nossr50.mcMMO;
|
|
|
|
public class HashChunkletManager implements ChunkletManager {
|
|
private HashMap<String, ChunkletStore> store = new HashMap<String, ChunkletStore>();
|
|
|
|
@Override
|
|
public void loadChunklet(int cx, int cy, int cz, World world) {
|
|
File dataDir = new File(world.getWorldFolder(), "mcmmo_data");
|
|
File cxDir = new File(dataDir, "" + cx);
|
|
if(!cxDir.exists()) return;
|
|
File czDir = new File(cxDir, "" + cz);
|
|
if(!czDir.exists()) return;
|
|
File yFile = new File(czDir, "" + cy);
|
|
if(!yFile.exists()) return;
|
|
|
|
ChunkletStore in = deserializeChunkletStore(yFile);
|
|
if(in != null) {
|
|
store.put(world.getName() + "," + cx + "," + cz + "," + cy, in);
|
|
}
|
|
}
|
|
|
|
@Override
|
|
public void chunkLoaded(int cx, int cz, World world) {
|
|
//File dataDir = new File(world.getWorldFolder(), "mcmmo_data");
|
|
//File cxDir = new File(dataDir, "" + cx);
|
|
//if(!cxDir.exists()) return;
|
|
//File czDir = new File(cxDir, "" + cz);
|
|
//if(!czDir.exists()) return;
|
|
|
|
//for(int y = 0; y < 4; y++) {
|
|
// File yFile = new File(czDir, "" + y);
|
|
// if(!yFile.exists()) {
|
|
// continue;
|
|
// } else {
|
|
// ChunkletStore in = deserializeChunkletStore(yFile);
|
|
// if(in != null) {
|
|
// store.put(world.getName() + "," + cx + "," + cz + "," + y, in);
|
|
// }
|
|
// }
|
|
//}
|
|
}
|
|
|
|
@Override
|
|
public void chunkUnloaded(int cx, int cz, World world) {
|
|
File dataDir = new File(world.getWorldFolder(), "mcmmo_data");
|
|
|
|
for(int y = 0; y < 4; y++) {
|
|
if(store.containsKey(world.getName() + "," + cx + "," + cz + "," + y)) {
|
|
File cxDir = new File(dataDir, "" + cx);
|
|
if(!cxDir.exists()) cxDir.mkdir();
|
|
File czDir = new File(cxDir, "" + cz);
|
|
if(!czDir.exists()) czDir.mkdir();
|
|
File yFile = new File(czDir, "" + y);
|
|
|
|
ChunkletStore out = store.get(world.getName() + "," + cx + "," + cz + "," + y);
|
|
serializeChunkletStore(out, yFile);
|
|
store.remove(world.getName() + "," + cx + "," + cz + "," + y);
|
|
}
|
|
}
|
|
}
|
|
|
|
@Override
|
|
public void saveWorld(World world) {
|
|
String worldName = world.getName();
|
|
File dataDir = new File(world.getWorldFolder(), "mcmmo_data");
|
|
|
|
for(String key : store.keySet()) {
|
|
String[] info = key.split(",");
|
|
if(worldName.equals(info[0])) {
|
|
File cxDir = new File(dataDir, "" + info[1]);
|
|
if(!cxDir.exists()) cxDir.mkdir();
|
|
File czDir = new File(cxDir, "" + info[2]);
|
|
if(!czDir.exists()) czDir.mkdir();
|
|
|
|
File yFile = new File(czDir, "" + info[3]);
|
|
serializeChunkletStore(store.get(key), yFile);
|
|
}
|
|
}
|
|
}
|
|
|
|
@Override
|
|
public void unloadWorld(World world) {
|
|
saveWorld(world);
|
|
|
|
String worldName = world.getName();
|
|
|
|
for(String key : store.keySet()) {
|
|
String tempWorldName = key.split(",")[0];
|
|
if(tempWorldName.equals(worldName)) {
|
|
store.remove(key);
|
|
return;
|
|
}
|
|
}
|
|
}
|
|
|
|
@Override
|
|
public void loadWorld(World world) {
|
|
//for(Chunk chunk : world.getLoadedChunks()) {
|
|
// this.chunkLoaded(chunk.getX(), chunk.getZ(), world);
|
|
//}
|
|
}
|
|
|
|
@Override
|
|
public void saveAll() {
|
|
for(World world : Bukkit.getWorlds()) {
|
|
saveWorld(world);
|
|
}
|
|
}
|
|
|
|
@Override
|
|
public void unloadAll() {
|
|
saveAll();
|
|
for(World world : Bukkit.getWorlds()) {
|
|
unloadWorld(world);
|
|
}
|
|
}
|
|
|
|
@Override
|
|
public boolean isTrue(int x, int y, int z, World world) {
|
|
int cx = x / 16;
|
|
int cz = z / 16;
|
|
int cy = y / 64;
|
|
String key = world.getName() + "," + cx + "," + cz + "," + cy;
|
|
|
|
if (!store.containsKey(key)) {
|
|
loadChunklet(cx, cy, cz, world);
|
|
}
|
|
|
|
if (!store.containsKey(key)) {
|
|
return false;
|
|
}
|
|
|
|
ChunkletStore check = store.get(world.getName() + "," + cx + "," + cz + "," + cy);
|
|
int ix = Math.abs(x) % 16;
|
|
int iz = Math.abs(z) % 16;
|
|
int iy = Math.abs(y) % 64;
|
|
|
|
return check.isTrue(ix, iy, iz);
|
|
}
|
|
|
|
@Override
|
|
public boolean isTrue(Block block) {
|
|
return isTrue(block.getX(), block.getY(), block.getZ(), block.getWorld());
|
|
}
|
|
|
|
@Override
|
|
public void setTrue(int x, int y, int z, World world) {
|
|
int cx = x / 16;
|
|
int cz = z / 16;
|
|
int cy = y / 64;
|
|
|
|
int ix = Math.abs(x) % 16;
|
|
int iz = Math.abs(z) % 16;
|
|
int iy = Math.abs(y) % 64;
|
|
|
|
String key = world.getName() + "," + cx + "," + cz + "," + cy;
|
|
|
|
if (!store.containsKey(key)) {
|
|
loadChunklet(cx, cy, cz, world);
|
|
}
|
|
|
|
ChunkletStore cStore = store.get(key);
|
|
|
|
if (cStore == null) {
|
|
cStore = ChunkletStoreFactory.getChunkletStore();
|
|
|
|
store.put(world.getName() + "," + cx + "," + cz + "," + cy, cStore);
|
|
}
|
|
|
|
cStore.setTrue(ix, iy, iz);
|
|
}
|
|
|
|
@Override
|
|
public void setTrue(Block block) {
|
|
setTrue(block.getX(), block.getY(), block.getZ(), block.getWorld());
|
|
}
|
|
|
|
@Override
|
|
public void setFalse(int x, int y, int z, World world) {
|
|
int cx = x / 16;
|
|
int cz = z / 16;
|
|
int cy = y / 64;
|
|
|
|
int ix = Math.abs(x) % 16;
|
|
int iz = Math.abs(z) % 16;
|
|
int iy = Math.abs(y) % 64;
|
|
|
|
String key = world.getName() + "," + cx + "," + cz + "," + cy;
|
|
|
|
if (!store.containsKey(key)) {
|
|
loadChunklet(cx, cy, cz, world);
|
|
}
|
|
|
|
ChunkletStore cStore = store.get(key);
|
|
|
|
if (cStore == null) {
|
|
return; //No need to make a store for something we will be setting to false
|
|
}
|
|
|
|
cStore.setFalse(ix, iy, iz);
|
|
}
|
|
|
|
@Override
|
|
public void setFalse(Block block) {
|
|
setFalse(block.getX(), block.getY(), block.getZ(), block.getWorld());
|
|
}
|
|
|
|
@Override
|
|
public void cleanUp() {
|
|
for(String key : store.keySet()) {
|
|
if(store.get(key).isEmpty()) {
|
|
String[] info = key.split(",");
|
|
File dataDir = new File(Bukkit.getWorld(info[0]).getWorldFolder(), "mcmmo_data");
|
|
|
|
File cxDir = new File(dataDir, "" + info[1]);
|
|
if(!cxDir.exists()) continue;
|
|
File czDir = new File(cxDir, "" + info[2]);
|
|
if(!czDir.exists()) continue;
|
|
|
|
File yFile = new File(czDir, "" + info[3]);
|
|
yFile.delete();
|
|
|
|
//Delete empty directories
|
|
if(czDir.list().length == 0) czDir.delete();
|
|
if(cxDir.list().length == 0) cxDir.delete();
|
|
}
|
|
}
|
|
}
|
|
|
|
/**
|
|
* @param cStore ChunkletStore to save
|
|
* @param location Where on the disk to put it
|
|
*/
|
|
private void serializeChunkletStore(ChunkletStore cStore, File location) {
|
|
try {
|
|
FileOutputStream fileOut = new FileOutputStream(location);
|
|
ObjectOutputStream objOut = new ObjectOutputStream(fileOut);
|
|
objOut.writeObject(cStore);
|
|
objOut.close();
|
|
fileOut.close();
|
|
} catch (IOException ex) {
|
|
ex.printStackTrace();
|
|
}
|
|
}
|
|
|
|
/**
|
|
* @param location Where on the disk to read from
|
|
* @return ChunkletStore from the specified location
|
|
*/
|
|
private ChunkletStore deserializeChunkletStore(File location) {
|
|
ChunkletStore storeIn = null;
|
|
|
|
try {
|
|
FileInputStream fileIn = new FileInputStream(location);
|
|
ObjectInputStream objIn = new ObjectInputStream(fileIn);
|
|
storeIn = (ChunkletStore) objIn.readObject();
|
|
objIn.close();
|
|
fileIn.close();
|
|
}
|
|
catch (IOException ex) {
|
|
if (ex instanceof EOFException) {
|
|
// EOF should only happen on Chunklets that somehow have been corrupted.
|
|
mcMMO.p.getLogger().severe("Chunklet data at " + location.toString() + " could not be read due to an EOFException, data in this area will be lost.");
|
|
return ChunkletStoreFactory.getChunkletStore();
|
|
}
|
|
else if (ex instanceof StreamCorruptedException) {
|
|
// StreamCorrupted happens when the Chunklet is no good.
|
|
mcMMO.p.getLogger().severe("Chunklet data at " + location.toString() + " is corrupted, data in this area will be lost.");
|
|
return ChunkletStoreFactory.getChunkletStore();
|
|
}
|
|
else if (ex instanceof UTFDataFormatException) {
|
|
// UTF happens when the Chunklet cannot be read or is corrupted
|
|
mcMMO.p.getLogger().severe("Chunklet data at " + location.toString() + " could not be read due to an UTFDataFormatException, data in this area will be lost.");
|
|
return ChunkletStoreFactory.getChunkletStore();
|
|
}
|
|
|
|
ex.printStackTrace();
|
|
}
|
|
catch (ClassNotFoundException ex) {
|
|
ex.printStackTrace();
|
|
}
|
|
|
|
return storeIn;
|
|
}
|
|
}
|