71 lines
2.7 KiB
Java
71 lines
2.7 KiB
Java
package net.minecraft.gametest.framework;
|
|
|
|
import java.util.ArrayList;
|
|
import java.util.Collection;
|
|
import java.util.Optional;
|
|
import net.minecraft.core.BlockPos;
|
|
import net.minecraft.server.level.ServerLevel;
|
|
import net.minecraft.world.level.levelgen.structure.BoundingBox;
|
|
import net.minecraft.world.phys.AABB;
|
|
|
|
public class StructureGridSpawner implements GameTestRunner.StructureSpawner {
|
|
private static final int SPACE_BETWEEN_COLUMNS = 5;
|
|
private static final int SPACE_BETWEEN_ROWS = 6;
|
|
private final int testsPerRow;
|
|
private int currentRowCount;
|
|
private AABB rowBounds;
|
|
private final BlockPos.MutableBlockPos nextTestNorthWestCorner;
|
|
private final BlockPos firstTestNorthWestCorner;
|
|
private final boolean clearOnBatch;
|
|
private float maxX = -1.0F;
|
|
private final Collection<GameTestInfo> testInLastBatch = new ArrayList();
|
|
|
|
public StructureGridSpawner(BlockPos northTestNorthWestCorner, int testsPerRow, boolean clearOnBatch) {
|
|
this.testsPerRow = testsPerRow;
|
|
this.nextTestNorthWestCorner = northTestNorthWestCorner.mutable();
|
|
this.rowBounds = new AABB(this.nextTestNorthWestCorner);
|
|
this.firstTestNorthWestCorner = northTestNorthWestCorner;
|
|
this.clearOnBatch = clearOnBatch;
|
|
}
|
|
|
|
@Override
|
|
public void onBatchStart(ServerLevel level) {
|
|
if (this.clearOnBatch) {
|
|
this.testInLastBatch.forEach(gameTestInfo -> {
|
|
BoundingBox boundingBox = gameTestInfo.getTestInstanceBlockEntity().getStructureBoundingBox();
|
|
StructureUtils.clearSpaceForStructure(boundingBox, level);
|
|
});
|
|
this.testInLastBatch.clear();
|
|
this.rowBounds = new AABB(this.firstTestNorthWestCorner);
|
|
this.nextTestNorthWestCorner.set(this.firstTestNorthWestCorner);
|
|
}
|
|
}
|
|
|
|
@Override
|
|
public Optional<GameTestInfo> spawnStructure(GameTestInfo gameTestInfo) {
|
|
BlockPos blockPos = new BlockPos(this.nextTestNorthWestCorner);
|
|
gameTestInfo.setTestBlockPos(blockPos);
|
|
GameTestInfo gameTestInfo2 = gameTestInfo.prepareTestStructure();
|
|
if (gameTestInfo2 == null) {
|
|
return Optional.empty();
|
|
} else {
|
|
gameTestInfo2.startExecution(1);
|
|
AABB aABB = gameTestInfo.getTestInstanceBlockEntity().getStructureBounds();
|
|
this.rowBounds = this.rowBounds.minmax(aABB);
|
|
this.nextTestNorthWestCorner.move((int)aABB.getXsize() + 5, 0, 0);
|
|
if (this.nextTestNorthWestCorner.getX() > this.maxX) {
|
|
this.maxX = this.nextTestNorthWestCorner.getX();
|
|
}
|
|
|
|
if (++this.currentRowCount >= this.testsPerRow) {
|
|
this.currentRowCount = 0;
|
|
this.nextTestNorthWestCorner.move(0, 0, (int)this.rowBounds.getZsize() + 6);
|
|
this.nextTestNorthWestCorner.setX(this.firstTestNorthWestCorner.getX());
|
|
this.rowBounds = new AABB(this.nextTestNorthWestCorner);
|
|
}
|
|
|
|
this.testInLastBatch.add(gameTestInfo);
|
|
return Optional.of(gameTestInfo);
|
|
}
|
|
}
|
|
}
|