Few fixes for FastModeExtent.

This commit is contained in:
Matthew Miller 2018-11-15 19:18:22 +10:00
parent 36430863a1
commit b3f5bc030e
2 changed files with 14 additions and 6 deletions

View File

@ -246,6 +246,9 @@ public class EditSession implements Extent, AutoCloseable {
if (isBatchingChunks() && chunkBatchingExtent.commitRequired()) {
return true;
}
if (hasFastMode() && fastModeExtent.commitRequired()) {
return true;
}
return false;
}

View File

@ -31,10 +31,9 @@ import com.sk89q.worldedit.world.World;
import com.sk89q.worldedit.world.block.BlockStateHolder;
import com.sk89q.worldedit.world.block.BlockTypes;
import java.util.ArrayDeque;
import java.util.HashSet;
import java.util.Iterator;
import java.util.List;
import java.util.Queue;
import java.util.Set;
/**
@ -43,7 +42,7 @@ import java.util.Set;
public class FastModeExtent extends AbstractDelegateExtent {
private final World world;
private final Queue<BlockVector3> positions = new ArrayDeque<>();
private final Set<BlockVector3> positions = new HashSet<>();
private final Set<BlockVector2> dirtyChunks = new HashSet<>();
private boolean enabled = true;
private boolean postEditSimulation;
@ -106,7 +105,7 @@ public class FastModeExtent extends AbstractDelegateExtent {
if (world.setBlock(location, block, false)) {
if (postEditSimulation) {
positions.offer(location);
positions.add(location);
}
return true;
}
@ -117,6 +116,10 @@ public class FastModeExtent extends AbstractDelegateExtent {
}
}
public boolean commitRequired() {
return !dirtyChunks.isEmpty() || !positions.isEmpty();
}
@Override
protected Operation commitBefore() {
return new Operation() {
@ -127,9 +130,11 @@ public class FastModeExtent extends AbstractDelegateExtent {
}
if (postEditSimulation) {
while (run.shouldContinue() && !positions.isEmpty()) {
BlockVector3 position = positions.poll(); // Remove from queue
Iterator<BlockVector3> positionIterator = positions.iterator();
while (run.shouldContinue() && positionIterator.hasNext()) {
BlockVector3 position = positionIterator.next();
world.notifyAndLightBlock(position, BlockTypes.AIR.getDefaultState());
positionIterator.remove();
}
return !positions.isEmpty() ? this : null;