[Savestates] Fixed world being unloaded after loadstate

Author: ScribbleTAS

src/main/java/com/minecrafttas/tasmod/mixin/savestates/AccessorPlayerChunkMap.java

@@ -0,0 +1,95 @@
+package com.minecrafttas.tasmod.mixin.savestates;
+
+import java.util.Collections;
+import java.util.Comparator;
+import java.util.Iterator;
+import java.util.List;
+
+import org.spongepowered.asm.mixin.Final;
+import org.spongepowered.asm.mixin.Mixin;
+import org.spongepowered.asm.mixin.Shadow;
+
+import com.google.common.collect.ComparisonChain;
+import com.minecrafttas.tasmod.savestates.handlers.SavestateWorldHandler;
+import com.minecrafttas.tasmod.util.Ducks.PlayerChunkMapDuck;
+
+import net.minecraft.entity.player.EntityPlayerMP;
+import net.minecraft.server.management.PlayerChunkMap;
+import net.minecraft.server.management.PlayerChunkMapEntry;
+import net.minecraft.world.WorldProvider;
+import net.minecraft.world.WorldServer;
+
+@Mixin(PlayerChunkMap.class)
+public class MixinPlayerChunkMap implements PlayerChunkMapDuck {
+
+	@Shadow
+	@Final
+	private List<EntityPlayerMP> players;
+	@Shadow
+	private boolean sortMissingChunks;
+	@Shadow
+	@Final
+	private List<PlayerChunkMapEntry> entriesWithoutChunks;
+	@Shadow
+	private boolean sortSendToPlayers;
+	@Shadow
+	@Final
+	private List<PlayerChunkMapEntry> pendingSendToPlayers;
+	@Shadow
+	@Final
+	private WorldServer world;
+
+	/**
+	 * @return The players from the specified chunk map
+	 * @see SavestateWorldHandler#addPlayerToChunkMap()
+	 */
+	@Override
+	public List<EntityPlayerMP> getPlayers() {
+		return players;
+	}
+
+	/**
+	 * {@inheritDoc}
+	 * @see SavestateWorldHandler#addPlayersToChunkMap()
+	 */
+	@Override
+	public void forceTick() {
+		this.sortMissingChunks = false;
+		Collections.sort(this.entriesWithoutChunks, new Comparator<PlayerChunkMapEntry>() {
+			public int compare(PlayerChunkMapEntry playerChunkMapEntry, PlayerChunkMapEntry playerChunkMapEntry2) {
+				return ComparisonChain.start().compare(playerChunkMapEntry.getClosestPlayerDistance(), playerChunkMapEntry2.getClosestPlayerDistance()).result();
+			}
+		});
+
+		this.sortSendToPlayers = false;
+		Collections.sort(this.pendingSendToPlayers, new Comparator<PlayerChunkMapEntry>() {
+
+			public int compare(PlayerChunkMapEntry playerChunkMapEntry, PlayerChunkMapEntry playerChunkMapEntry2) {
+				return ComparisonChain.start().compare(playerChunkMapEntry.getClosestPlayerDistance(), playerChunkMapEntry2.getClosestPlayerDistance()).result();
+			}
+		});
+
+		if (!this.pendingSendToPlayers.isEmpty()) {
+
+			int i = 81;
+			Iterator<PlayerChunkMapEntry> iterator2 = this.pendingSendToPlayers.iterator();
+
+			while (iterator2.hasNext()) {
+				PlayerChunkMapEntry playerChunkMapEntry3 = (PlayerChunkMapEntry) iterator2.next();
+				if (playerChunkMapEntry3.sendToPlayers()) {
+					iterator2.remove();
+					if (--i < 0) {
+						break;
+					}
+				}
+			}
+		}
+
+		if (this.players.isEmpty()) {
+			WorldProvider worldProvider = this.world.provider;
+			if (!worldProvider.canRespawnHere()) {
+				this.world.getChunkProvider().queueUnloadAll();
+			}
+		}
+	}
+}

src/main/java/com/minecrafttas/tasmod/mixin/savestates/MixinPlayerChunkMap.java

@@ -39,12 +39,6 @@ public enum TASmodKeybinds {
 	TEST1("Various Testing", "TASmod", Keyboard.KEY_F12, () -> {
 	}, VirtualKeybindings::isKeyDown),
 	TEST2("Various Testing2", "TASmod", Keyboard.KEY_F7, () -> {
-		//			try {
-		//				TASmodClient.client = new Client("localhost", TASmod.networkingport - 1, TASmodPackets.values(), mc.getSession().getProfile().getName(), true);
-		//			} catch (Exception e) {
-		//				e.printStackTrace();
-		//			}
-//		TASmodClient.controller.setTASState(TASstate.PLAYBACK);
 	}, VirtualKeybindings::isKeyDown);
 
 	private Keybind keybind;

src/main/java/com/minecrafttas/tasmod/registries/TASmodKeybinds.java

@@ -394,9 +394,7 @@ public void loadState(int savestateIndex, boolean tickrate0, boolean changeIndex
 		server.getPlayerList().sendMessage(new TextComponentString(TextFormatting.GREEN + "Savestate " + indexToLoad + " loaded"));
 
 		// Add players to the chunk
-		server.getPlayerList().getPlayers().forEach(player -> {
-			worldHandler.addPlayerToServerChunk(player);
-		});
+		worldHandler.addPlayersToServerChunks();
 
 		worldHandler.sendChunksToClient();
 

src/main/java/com/minecrafttas/tasmod/savestates/SavestateHandlerServer.java

@@ -4,10 +4,10 @@
 
 import java.util.List;
 
-import com.minecrafttas.tasmod.mixin.savestates.AccessorPlayerChunkMap;
 import com.minecrafttas.tasmod.mixin.savestates.MixinChunkProviderServer;
 import com.minecrafttas.tasmod.savestates.SavestateHandlerClient;
 import com.minecrafttas.tasmod.util.Ducks.ChunkProviderDuck;
+import com.minecrafttas.tasmod.util.Ducks.PlayerChunkMapDuck;
 import com.minecrafttas.tasmod.util.Ducks.WorldServerDuck;
 import com.minecrafttas.tasmod.util.LoggerMarkers;
 
@@ -55,6 +55,15 @@ public void enableLevelSaving() {
 		}
 	}
 
+	/**
+	 * Add players to their respective chunks
+	 */
+	public void addPlayersToServerChunks() {
+		server.getPlayerList().getPlayers().forEach(player -> {
+			addPlayerToServerChunk(player);
+		});
+	}
+
 	/**
 	 * Just like {@link SavestateHandlerClient#addPlayerToClientChunk(EntityPlayer)}, adds the player to the chunk on the server.
 	 * This prevents the player from being able to place block inside of him
@@ -138,6 +147,11 @@ public void addPlayersToChunkMap() {
 					break;
 			}
 		}
+
+		// Tick the player chunk map to send the chunks to the clients
+		for (WorldServer world : worlds) {
+			((PlayerChunkMapDuck) world.getPlayerChunkMap()).forceTick();
+		}
 	}
 
 	/**
@@ -150,15 +164,14 @@ private void addPlayerToChunkMap(WorldServer world, EntityPlayerMP player) {
 		int playerChunkPosY = (int) player.posZ >> 4;
 		PlayerChunkMap playerChunkMap = world.getPlayerChunkMap();
 
-		List<EntityPlayerMP> players = ((AccessorPlayerChunkMap) playerChunkMap).getPlayers();
+		List<EntityPlayerMP> players = ((PlayerChunkMapDuck) playerChunkMap).getPlayers();
 
 		if (players.contains(player)) {
 			LOGGER.debug(LoggerMarkers.Savestate, "Not adding player {} to chunkmap, player already exists", player.getName());
 		} else {
 			playerChunkMap.addPlayer(player);
 		}
-		Chunk chunk = world.getChunkProvider().provideChunk(playerChunkPosX, playerChunkPosY);
-		chunk.addEntity(player);
+		world.getChunkProvider().provideChunk(playerChunkPosX, playerChunkPosY);
 
 		world.spawnEntity(player);
 	}

src/main/java/com/minecrafttas/tasmod/savestates/handlers/SavestateWorldHandler.java

@@ -1,5 +1,11 @@
 package com.minecrafttas.tasmod.util;
 
+import java.util.List;
+
+import com.minecrafttas.tasmod.mixin.savestates.MixinPlayerChunkMap;
+
+import net.minecraft.entity.player.EntityPlayerMP;
+
 /**
  * Oh boy, ducks! I can't help but quack up when they waddle their way into the code. :duck:
  * But let me tell you a little secret: I have a love-hate relationship with ducks. Not the adorable feathered creatures, mind you, but those sneaky little programming devils that swim in the deep waters of out-of-scope variables.
@@ -129,4 +135,24 @@ public static interface WorldClientDuck {
 		 */
 		public void clearEntityList();
 	}
+
+	/**
+	 * Quacks the {@link MixinPlayerChunkMap}
+	 */
+	public static interface PlayerChunkMapDuck {
+		/**
+		 * @return The list of players of this chunk map
+		 */
+		public List<EntityPlayerMP> getPlayers();
+
+		/**
+		 * <p>Forces a tick in the chunk map without being dependent on the world time.
+		 * <p>The chunk map is responsible for sending the necessary chunks to the client.<br>
+		 * However, to properly do that, the chunks have to be sorted first, which happens every few world ticks.
+		 * 
+		 * <p>This method sorts and sends the chunks to the clients, without waiting for the world time
+		 * @see MixinPlayerChunkMap#forceTick()
+		 */
+		public void forceTick();
+	}
 }

src/main/java/com/minecrafttas/tasmod/util/Ducks.java

@@ -12,7 +12,7 @@
 		"savestates.AccessorAnvilChunkLoader",
 		"savestates.AccessorChunkLoader",
 		"savestates.AccessorEntityLivingBase",
-		"savestates.AccessorPlayerChunkMap",
+		"savestates.MixinPlayerChunkMap",
 		"savestates.MixinChunkProviderServer",
 		"savestates.MixinNetHandlerPlayServer",
 		"savestates.MixinScoreboard",