518 lines
17 KiB
Java
518 lines
17 KiB
Java
/*
|
|
* Minecraft Forge
|
|
* Copyright (c) 2016-2018.
|
|
*
|
|
* This library is free software; you can redistribute it and/or
|
|
* modify it under the terms of the GNU Lesser General Public
|
|
* License as published by the Free Software Foundation version 2.1
|
|
* of the License.
|
|
*
|
|
* This library is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
* Lesser General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU Lesser General Public
|
|
* License along with this library; if not, write to the Free Software
|
|
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
|
|
*/
|
|
|
|
package net.minecraftforge.common;
|
|
|
|
import java.io.File;
|
|
import java.util.ArrayList;
|
|
import java.util.Arrays;
|
|
import java.util.BitSet;
|
|
import java.util.Hashtable;
|
|
import java.util.IdentityHashMap;
|
|
import java.util.List;
|
|
import java.util.ListIterator;
|
|
import java.util.Map;
|
|
import java.util.concurrent.ConcurrentMap;
|
|
import java.util.stream.Stream;
|
|
|
|
import it.unimi.dsi.fastutil.ints.Int2ObjectLinkedOpenHashMap;
|
|
import it.unimi.dsi.fastutil.ints.Int2ObjectMap;
|
|
import it.unimi.dsi.fastutil.ints.Int2ObjectMaps;
|
|
import it.unimi.dsi.fastutil.ints.IntIterator;
|
|
import it.unimi.dsi.fastutil.ints.IntLinkedOpenHashSet;
|
|
import it.unimi.dsi.fastutil.ints.IntOpenHashSet;
|
|
import it.unimi.dsi.fastutil.ints.IntRBTreeSet;
|
|
import it.unimi.dsi.fastutil.ints.IntSet;
|
|
import it.unimi.dsi.fastutil.ints.IntSets;
|
|
import it.unimi.dsi.fastutil.ints.IntSortedSet;
|
|
|
|
import com.google.common.collect.HashMultiset;
|
|
import com.google.common.collect.Lists;
|
|
import com.google.common.collect.MapMaker;
|
|
import com.google.common.collect.Multiset;
|
|
|
|
import net.minecraft.nbt.NBTTagCompound;
|
|
import net.minecraft.server.MinecraftServer;
|
|
import net.minecraft.world.World;
|
|
import net.minecraft.world.ServerWorldEventHandler;
|
|
import net.minecraft.world.WorldServer;
|
|
import net.minecraft.world.WorldServerMulti;
|
|
import net.minecraft.world.dimension.Dimension;
|
|
import net.minecraft.world.dimension.DimensionType;
|
|
import net.minecraft.world.storage.ISaveHandler;
|
|
import net.minecraftforge.event.world.WorldEvent;
|
|
import org.apache.logging.log4j.LogManager;
|
|
import org.apache.logging.log4j.Logger;
|
|
import org.apache.logging.log4j.Marker;
|
|
import org.apache.logging.log4j.MarkerManager;
|
|
|
|
import javax.annotation.Nullable;
|
|
|
|
public class DimensionManager
|
|
{
|
|
private static final Logger LOGGER = LogManager.getLogger();
|
|
private static final Marker DIMMGR = MarkerManager.getMarker("DIMS");
|
|
private static class DimensionData
|
|
{
|
|
private final DimensionType type;
|
|
private int ticksWaited;
|
|
private DimensionData(DimensionType type)
|
|
{
|
|
this.type = type;
|
|
this.ticksWaited = 0;
|
|
}
|
|
}
|
|
|
|
private static boolean hasInit = false;
|
|
|
|
private static final Int2ObjectMap<WorldServer> worlds = Int2ObjectMaps.synchronize(new Int2ObjectLinkedOpenHashMap<>());
|
|
private static final Int2ObjectMap<DimensionData> dimensions = Int2ObjectMaps.synchronize(new Int2ObjectLinkedOpenHashMap<>());
|
|
private static final IntSet keepLoaded = IntSets.synchronize(new IntOpenHashSet());
|
|
private static final IntSet unloadQueue = IntSets.synchronize(new IntLinkedOpenHashSet());
|
|
private static final BitSet dimensionMap = new BitSet(Long.SIZE << 4);
|
|
private static final ConcurrentMap<World, World> weakWorldMap = new MapMaker().weakKeys().weakValues().makeMap();
|
|
private static final Multiset<Integer> leakedWorlds = HashMultiset.create();
|
|
|
|
/**
|
|
* Returns a list of dimensions associated with this DimensionType.
|
|
*/
|
|
public static int[] getDimensions(DimensionType type)
|
|
{
|
|
int[] ret = new int[dimensions.size()];
|
|
int x = 0;
|
|
for (Int2ObjectMap.Entry<DimensionData> ent : dimensions.int2ObjectEntrySet())
|
|
{
|
|
if (ent.getValue().type == type)
|
|
{
|
|
ret[x++] = ent.getIntKey();
|
|
}
|
|
}
|
|
|
|
return Arrays.copyOf(ret, x);
|
|
}
|
|
|
|
public static Map<DimensionType, IntSortedSet> getRegisteredDimensions()
|
|
{
|
|
Map<DimensionType, IntSortedSet> map = new IdentityHashMap<>();
|
|
for (Int2ObjectMap.Entry<DimensionData> entry : dimensions.int2ObjectEntrySet())
|
|
{
|
|
map.computeIfAbsent(entry.getValue().type, k -> new IntRBTreeSet()).add(entry.getIntKey());
|
|
}
|
|
return map;
|
|
}
|
|
|
|
public static void init()
|
|
{
|
|
if (hasInit)
|
|
{
|
|
return;
|
|
}
|
|
|
|
hasInit = true;
|
|
|
|
registerDimension( 0, DimensionType.OVERWORLD);
|
|
registerDimension(-1, DimensionType.NETHER);
|
|
registerDimension( 1, DimensionType.THE_END);
|
|
}
|
|
|
|
public static void registerDimension(int id, DimensionType type)
|
|
{
|
|
DimensionType.getById(type.getId()); //Check if type is invalid {will throw an error} No clue how it would be invalid tho...
|
|
if (dimensions.containsKey(id))
|
|
{
|
|
throw new IllegalArgumentException(String.format("Failed to register dimension for id %d, One is already registered", id));
|
|
}
|
|
dimensions.put(id, new DimensionData(type));
|
|
if (id >= 0)
|
|
{
|
|
dimensionMap.set(id);
|
|
}
|
|
}
|
|
|
|
/**
|
|
* For unregistering a dimension when the save is changed (disconnected from a server or loaded a new save
|
|
*/
|
|
public static void unregisterDimension(int id)
|
|
{
|
|
if (!dimensions.containsKey(id))
|
|
{
|
|
throw new IllegalArgumentException(String.format("Failed to unregister dimension for id %d; No provider registered", id));
|
|
}
|
|
dimensions.remove(id);
|
|
}
|
|
|
|
public static boolean isDimensionRegistered(int dim)
|
|
{
|
|
return dimensions.containsKey(dim);
|
|
}
|
|
|
|
public static DimensionType getProviderType(int dim)
|
|
{
|
|
if (!dimensions.containsKey(dim))
|
|
{
|
|
throw new IllegalArgumentException(String.format("Could not get provider type for dimension %d, does not exist", dim));
|
|
}
|
|
return dimensions.get(dim).type;
|
|
}
|
|
|
|
public static Dimension getProvider(int dim)
|
|
{
|
|
return getWorld(dim).provider;
|
|
}
|
|
|
|
public static Integer[] getIDs(boolean check)
|
|
{
|
|
if (check)
|
|
{
|
|
List<World> allWorlds = Lists.newArrayList(weakWorldMap.keySet());
|
|
allWorlds.removeAll(worlds.values());
|
|
for (ListIterator<World> li = allWorlds.listIterator(); li.hasNext(); )
|
|
{
|
|
World w = li.next();
|
|
leakedWorlds.add(System.identityHashCode(w));
|
|
}
|
|
for (World w : allWorlds)
|
|
{
|
|
int leakCount = leakedWorlds.count(System.identityHashCode(w));
|
|
if (leakCount == 5)
|
|
{
|
|
LOGGER.debug(DIMMGR,"The world {} ({}) may have leaked: first encounter (5 occurrences).\n", Integer.toHexString(System.identityHashCode(w)), w.getWorldInfo().getWorldName());
|
|
}
|
|
else if (leakCount % 5 == 0)
|
|
{
|
|
LOGGER.debug(DIMMGR,"The world {} ({}) may have leaked: seen {} times.\n", Integer.toHexString(System.identityHashCode(w)), w.getWorldInfo().getWorldName(), leakCount);
|
|
}
|
|
}
|
|
}
|
|
return getIDs();
|
|
}
|
|
|
|
public static Integer[] getIDs()
|
|
{
|
|
return worlds.keySet().toArray(new Integer[0]); // Only loaded dims, since usually used to cycle through loaded worlds
|
|
}
|
|
|
|
public static Stream<Integer> getIDStream()
|
|
{
|
|
return worlds.keySet().stream();
|
|
}
|
|
|
|
public static void setWorld(int id, @Nullable WorldServer world, MinecraftServer server)
|
|
{
|
|
if (world != null)
|
|
{
|
|
worlds.put(id, world);
|
|
weakWorldMap.put(world, world);
|
|
LOGGER.info(DIMMGR,"Loading dimension {} ({}) ({})", id, world.getWorldInfo().getWorldName(), world.getMinecraftServer());
|
|
}
|
|
else
|
|
{
|
|
worlds.remove(id);
|
|
LOGGER.info(DIMMGR,"Unloading dimension {}", id);
|
|
}
|
|
|
|
ArrayList<WorldServer> tmp = new ArrayList<WorldServer>();
|
|
if (worlds.get( 0) != null)
|
|
tmp.add(worlds.get( 0));
|
|
if (worlds.get(-1) != null)
|
|
tmp.add(worlds.get(-1));
|
|
if (worlds.get( 1) != null)
|
|
tmp.add(worlds.get( 1));
|
|
|
|
for (Int2ObjectMap.Entry<WorldServer> entry : worlds.int2ObjectEntrySet())
|
|
{
|
|
int dim = entry.getIntKey();
|
|
if (dim >= -1 && dim <= 1)
|
|
{
|
|
continue;
|
|
}
|
|
tmp.add(entry.getValue());
|
|
}
|
|
|
|
server.worlds = tmp.toArray(new WorldServer[0]);
|
|
}
|
|
|
|
public static void initDimension(int dim)
|
|
{
|
|
WorldServer overworld = getWorld(0);
|
|
if (overworld == null)
|
|
{
|
|
throw new RuntimeException("Cannot Hotload Dim: Overworld is not Loaded!");
|
|
}
|
|
try
|
|
{
|
|
DimensionManager.getProviderType(dim);
|
|
}
|
|
catch (Exception e)
|
|
{
|
|
LOGGER.error(DIMMGR,"Cannot Hotload Dim: {}", dim, e);
|
|
return; // If a provider hasn't been registered then we can't hotload the dim
|
|
}
|
|
MinecraftServer mcServer = overworld.getMinecraftServer();
|
|
ISaveHandler savehandler = overworld.getSaveHandler();
|
|
//WorldSettings worldSettings = new WorldSettings(overworld.getWorldInfo());
|
|
|
|
WorldServer world = (dim == 0 ? overworld : (WorldServer)(new WorldServerMulti(mcServer, savehandler, dim, overworld, mcServer.profiler).init()));
|
|
world.addEventListener(new ServerWorldEventHandler(mcServer, world));
|
|
MinecraftForge.EVENT_BUS.post(new WorldEvent.Load(world));
|
|
if (!mcServer.isSinglePlayer())
|
|
{
|
|
world.getWorldInfo().setGameType(mcServer.getGameType());
|
|
}
|
|
|
|
mcServer.setDifficultyForAllWorlds(mcServer.getDifficulty());
|
|
}
|
|
|
|
public static WorldServer getWorld(int id)
|
|
{
|
|
return getWorld(id, false);
|
|
}
|
|
|
|
public static WorldServer getWorld(int id, boolean resetUnloadDelay)
|
|
{
|
|
return getWorld(id, resetUnloadDelay, false);
|
|
}
|
|
|
|
public static WorldServer getWorld(int id, boolean resetUnloadDelay, boolean forceLoad)
|
|
{
|
|
if (resetUnloadDelay && unloadQueue.contains(id))
|
|
{
|
|
dimensions.get(id).ticksWaited = 0;
|
|
}
|
|
WorldServer ret = worlds.get(id);
|
|
if (ret == null && forceLoad)
|
|
{
|
|
initDimension(id);
|
|
ret = worlds.get(id);
|
|
}
|
|
return ret;
|
|
}
|
|
|
|
public static WorldServer[] getWorlds()
|
|
{
|
|
return worlds.values().toArray(new WorldServer[0]);
|
|
}
|
|
|
|
static
|
|
{
|
|
init();
|
|
}
|
|
|
|
/**
|
|
* Not public API: used internally to get dimensions that should load at
|
|
* server startup
|
|
*/
|
|
public static Integer[] getStaticDimensionIDs()
|
|
{
|
|
return dimensions.keySet().toArray(new Integer[0]);
|
|
}
|
|
|
|
public static Dimension createProviderFor(int dim)
|
|
{
|
|
try
|
|
{
|
|
if (dimensions.containsKey(dim))
|
|
{
|
|
Dimension ret = getProviderType(dim).createDimension();
|
|
ret.setId(dim);
|
|
return ret;
|
|
}
|
|
else
|
|
{
|
|
throw new RuntimeException(String.format("No WorldProvider bound for dimension %d", dim)); //It's going to crash anyway at this point. Might as well be informative
|
|
}
|
|
}
|
|
catch (Exception e)
|
|
{
|
|
LOGGER.error(DIMMGR,"An error occurred trying to create an instance of WorldProvider {} ({})",
|
|
dim, getProviderType(dim), e);
|
|
throw new RuntimeException(e);
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Sets if a dimension should stay loaded.
|
|
* @param dim the dimension ID
|
|
* @param keep whether or not the dimension should be kept loaded
|
|
* @return true iff the dimension's status changed
|
|
*/
|
|
public static boolean keepDimensionLoaded(int dim, boolean keep)
|
|
{
|
|
return keep ? keepLoaded.add(dim) : keepLoaded.remove(dim);
|
|
}
|
|
|
|
private static boolean canUnloadWorld(WorldServer world)
|
|
{
|
|
return ForgeChunkManager.getPersistentChunksFor(world).isEmpty()
|
|
&& world.playerEntities.isEmpty()
|
|
&& !world.provider.getDimensionType().shouldLoadSpawn()
|
|
&& !keepLoaded.contains(world.provider.getId());
|
|
}
|
|
|
|
/**
|
|
* Queues a dimension to unload, if it can be unloaded.
|
|
* @param id The id of the dimension
|
|
*/
|
|
public static void unloadWorld(int id)
|
|
{
|
|
WorldServer world = worlds.get(id);
|
|
if (world == null || !canUnloadWorld(world)) return;
|
|
|
|
if (unloadQueue.add(id))
|
|
{
|
|
LOGGER.debug(DIMMGR,"Queueing dimension {} to unload", id);
|
|
}
|
|
}
|
|
|
|
public static boolean isWorldQueuedToUnload(int id)
|
|
{
|
|
return unloadQueue.contains(id);
|
|
}
|
|
|
|
/*
|
|
* To be called by the server at the appropriate time, do not call from mod code.
|
|
*/
|
|
public static void unloadWorlds()
|
|
{
|
|
IntIterator queueIterator = unloadQueue.iterator();
|
|
while (queueIterator.hasNext())
|
|
{
|
|
int id = queueIterator.nextInt();
|
|
DimensionData dimension = dimensions.get(id);
|
|
if (dimension.ticksWaited < ForgeMod.dimensionUnloadQueueDelay)
|
|
{
|
|
dimension.ticksWaited++;
|
|
continue;
|
|
}
|
|
WorldServer w = worlds.get(id);
|
|
queueIterator.remove();
|
|
dimension.ticksWaited = 0;
|
|
// Don't unload the world if the status changed
|
|
if (w == null || !canUnloadWorld(w))
|
|
{
|
|
LOGGER.debug(DIMMGR,"Aborting unload for dimension {} as status changed", id);
|
|
continue;
|
|
}
|
|
try
|
|
{
|
|
w.saveAllChunks(true, null);
|
|
}
|
|
catch (Exception e)
|
|
{
|
|
LOGGER.error(DIMMGR,"Caught an exception while saving all chunks:", e);
|
|
}
|
|
finally
|
|
{
|
|
MinecraftForge.EVENT_BUS.post(new WorldEvent.Unload(w));
|
|
w.close();
|
|
setWorld(id, null, w.getMinecraftServer());
|
|
}
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Return the next free dimension ID. Note: you are not guaranteed a contiguous
|
|
* block of free ids. Always call for each individual ID you wish to get.
|
|
* @return the next free dimension ID
|
|
*/
|
|
public static int getNextFreeDimId() {
|
|
int next = 0;
|
|
while (true)
|
|
{
|
|
next = dimensionMap.nextClearBit(next);
|
|
if (dimensions.containsKey(next))
|
|
{
|
|
dimensionMap.set(next);
|
|
}
|
|
else
|
|
{
|
|
return next;
|
|
}
|
|
}
|
|
}
|
|
|
|
public static NBTTagCompound saveDimensionDataMap()
|
|
{
|
|
int[] data = new int[(dimensionMap.length() + Integer.SIZE - 1 )/ Integer.SIZE];
|
|
NBTTagCompound dimMap = new NBTTagCompound();
|
|
for (int i = 0; i < data.length; i++)
|
|
{
|
|
int val = 0;
|
|
for (int j = 0; j < Integer.SIZE; j++)
|
|
{
|
|
val |= dimensionMap.get(i * Integer.SIZE + j) ? (1 << j) : 0;
|
|
}
|
|
data[i] = val;
|
|
}
|
|
dimMap.setIntArray("DimensionArray", data);
|
|
return dimMap;
|
|
}
|
|
|
|
public static void loadDimensionDataMap(@Nullable NBTTagCompound compoundTag)
|
|
{
|
|
dimensionMap.clear();
|
|
if (compoundTag == null)
|
|
{
|
|
IntIterator iterator = dimensions.keySet().iterator();
|
|
while (iterator.hasNext())
|
|
{
|
|
int id = iterator.nextInt();
|
|
if (id >= 0)
|
|
{
|
|
dimensionMap.set(id);
|
|
}
|
|
}
|
|
}
|
|
else
|
|
{
|
|
int[] intArray = compoundTag.getIntArray("DimensionArray");
|
|
for (int i = 0; i < intArray.length; i++)
|
|
{
|
|
for (int j = 0; j < Integer.SIZE; j++)
|
|
{
|
|
dimensionMap.set(i * Integer.SIZE + j, (intArray[i] & (1 << j)) != 0);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Return the current root directory for the world save. Accesses getSaveHandler from the overworld
|
|
* @return the root directory of the save
|
|
*/
|
|
@Nullable
|
|
public static File getCurrentSaveRootDirectory()
|
|
{
|
|
if (DimensionManager.getWorld(0) != null)
|
|
{
|
|
return DimensionManager.getWorld(0).getSaveHandler().getWorldDirectory();
|
|
}/*
|
|
else if (MinecraftServer.getServer() != null)
|
|
{
|
|
MinecraftServer srv = MinecraftServer.getServer();
|
|
SaveHandler saveHandler = (SaveHandler) srv.getActiveAnvilConverter().getSaveLoader(srv.getFolderName(), false);
|
|
return saveHandler.getWorldDirectory();
|
|
}*/
|
|
else
|
|
{
|
|
return null;
|
|
}
|
|
}
|
|
}
|