TechReborn/src/main/java/techreborn/tiles/idsu/IDSUManager.java

200 lines
5.2 KiB
Java
Raw Normal View History

package techreborn.tiles.idsu;
2015-06-15 21:09:16 +02:00
import com.google.gson.Gson;
import com.google.gson.GsonBuilder;
import com.google.gson.reflect.TypeToken;
2015-06-15 21:09:16 +02:00
import net.minecraft.world.World;
import net.minecraft.world.storage.ISaveHandler;
import net.minecraftforge.event.world.WorldEvent;
2015-11-23 20:19:18 +01:00
import net.minecraftforge.fml.common.eventhandler.EventPriority;
import net.minecraftforge.fml.common.eventhandler.SubscribeEvent;
2015-06-15 21:09:16 +02:00
import java.io.*;
import java.lang.reflect.Type;
import java.util.HashMap;
import java.util.TreeMap;
2015-06-15 21:09:16 +02:00
public class IDSUManager {
2016-03-25 10:47:34 +01:00
public static final String savename = "idsu.json";
public static IDSUManager INSTANCE;
public HashMap<World, IDSUWorldSaveData> worldData = new HashMap<>();
2016-03-25 10:47:34 +01:00
@SubscribeEvent(priority = EventPriority.HIGHEST)
public void worldSave(WorldEvent.Save event) {
if (event.getWorld() != null && event.getWorld().getSaveHandler() != null
&& event.getWorld().getSaveHandler().getWorldDirectory() != null) {
if (worldData.containsKey(event.getWorld())) {
worldData.get(event.getWorld()).save();
2016-03-25 10:47:34 +01:00
}
}
}
@SubscribeEvent(priority = EventPriority.HIGHEST)
public void worldLoad(WorldEvent.Load event) {
if (event.getWorld() != null && event.getWorld().getSaveHandler() != null
&& event.getWorld().getSaveHandler().getWorldDirectory() != null) {
if (worldData.containsKey(event.getWorld())) {
worldData.get(event.getWorld()).load();
} else {
IDSUWorldSaveData worldSaveData = new IDSUWorldSaveData(event.getWorld());
worldData.put(event.getWorld(), worldSaveData);
2016-03-25 10:47:34 +01:00
worldSaveData.load();
}
}
}
@SubscribeEvent(priority = EventPriority.LOWEST)
public void worldClosed(WorldEvent.Unload event) {
if (event.getWorld() != null && event.getWorld().getSaveHandler() != null
&& event.getWorld().getSaveHandler().getWorldDirectory() != null) {
if (worldData.containsKey(event.getWorld())) {
worldData.get(event.getWorld()).save();
2016-03-25 10:47:34 +01:00
}
}
// this clears the data ready for a new world
worldData.clear();
}
public IDSUValueSaveData getSaveDataForWorld(World world, String channel) {
if (worldData.containsKey(world)) {
2016-03-25 10:47:34 +01:00
return worldData.get(world).getSaves(channel);
} else {
2016-03-25 10:47:34 +01:00
IDSUWorldSaveData worldSaveData = new IDSUWorldSaveData(world);
worldData.put(world, worldSaveData);
worldSaveData.load();
return worldSaveData.getSaves(channel);
}
}
public IDSUWorldSaveData getWorldDataFormWorld(World world) {
if (worldData.containsKey(world)) {
2016-03-25 10:47:34 +01:00
return worldData.get(world);
} else {
2016-03-25 10:47:34 +01:00
IDSUWorldSaveData worldSaveData = new IDSUWorldSaveData(world);
worldData.put(world, worldSaveData);
worldSaveData.load();
return worldSaveData;
}
}
public void loadFromString(String json, World world) {
if (json.equals("EMPTY")) {
2016-03-25 10:47:34 +01:00
return;
}
IDSUWorldSaveData worldSaveData;
if (worldData.containsKey(world)) {
2016-03-25 10:47:34 +01:00
worldSaveData = worldData.get(world);
} else {
2016-03-25 10:47:34 +01:00
worldSaveData = new IDSUWorldSaveData(world);
worldData.put(world, worldSaveData);
}
Gson gson = new Gson();
Type typeOfHashMap = new TypeToken<TreeMap<Integer, IDSUValueSaveData>>() {
2016-03-25 10:47:34 +01:00
}.getType();
worldSaveData.idsuValues.clear();
worldSaveData.idsuValues = gson.fromJson(json, typeOfHashMap);
}
public class IDSUWorldSaveData {
2016-03-25 10:47:34 +01:00
public TreeMap<String, IDSUValueSaveData> idsuValues = new TreeMap<>();
2016-03-25 10:47:34 +01:00
public World world;
ISaveHandler saveHandler;
File folder;
File file;
public IDSUWorldSaveData(World world) {
2016-03-25 10:47:34 +01:00
this.world = world;
this.saveHandler = world.getSaveHandler();
folder = new File(saveHandler.getWorldDirectory(), "idsuData");
file = new File(folder, savename);
}
public IDSUValueSaveData getSaves(String udid) {
if (udid == null) {
2016-03-25 10:47:34 +01:00
return null;
}
if (idsuValues.containsKey(udid)) {
2016-03-25 10:47:34 +01:00
return idsuValues.get(udid);
} else {
2016-03-25 10:47:34 +01:00
IDSUValueSaveData data = new IDSUValueSaveData();
idsuValues.put(udid, data);
return data;
}
}
public void load() {
if (!file.exists()) {
2016-03-25 10:47:34 +01:00
return;
}
try {
2016-03-25 10:47:34 +01:00
Gson gson = new Gson();
BufferedReader reader = new BufferedReader(new FileReader(file));
Type typeOfHashMap = new TypeToken<TreeMap<String, IDSUValueSaveData>>() {
2016-03-25 10:47:34 +01:00
}.getType();
idsuValues.clear();
idsuValues = gson.fromJson(reader, typeOfHashMap);
} catch (Exception e) {
2016-03-25 10:47:34 +01:00
e.printStackTrace();
}
}
public void save() {
if (idsuValues.isEmpty()) {
2016-03-25 10:47:34 +01:00
return;
}
if (!file.exists()) {
if (!folder.exists()) {
2016-03-25 10:47:34 +01:00
folder.mkdirs();
}
try {
2016-03-25 10:47:34 +01:00
file.createNewFile();
} catch (IOException e) {
2016-03-25 10:47:34 +01:00
e.printStackTrace();
}
}
Gson gson = new GsonBuilder().setPrettyPrinting().create();
String json = gson.toJson(idsuValues);
try {
2016-03-25 10:47:34 +01:00
FileWriter writer = new FileWriter(file);
writer.write(json);
writer.close();
} catch (IOException e) {
2016-03-25 10:47:34 +01:00
e.printStackTrace();
}
}
}
public class IDSUValueSaveData {
2016-03-25 10:47:34 +01:00
public double storedPower = 0;
public IDSUValueSaveData(double storedPower) {
2016-03-25 10:47:34 +01:00
this.storedPower = storedPower;
}
public IDSUValueSaveData() {
2016-03-25 10:47:34 +01:00
}
public double getStoredPower() {
2016-03-25 10:47:34 +01:00
return storedPower;
}
public void setStoredPower(double storedPower) {
2016-03-25 10:47:34 +01:00
this.storedPower = storedPower;
}
public void addEnergy(double storedPower) {
2016-03-25 10:47:34 +01:00
this.storedPower += storedPower;
}
}
2015-06-15 21:09:16 +02:00
}