125 lines
4.4 KiB
Java
125 lines
4.4 KiB
Java
/*
|
|
* WorldEdit, a Minecraft world manipulation toolkit
|
|
* Copyright (C) sk89q <http://www.sk89q.com>
|
|
* Copyright (C) WorldEdit team and contributors
|
|
*
|
|
* This program is free software: you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation, either version 3 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* This program 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 General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program. If not, see <https://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
package com.sk89q.worldedit.function.block;
|
|
|
|
import com.sk89q.worldedit.WorldEditException;
|
|
import com.sk89q.worldedit.extent.Extent;
|
|
import com.sk89q.worldedit.function.LayerFunction;
|
|
import com.sk89q.worldedit.math.BlockVector3;
|
|
import com.sk89q.worldedit.registry.state.Property;
|
|
import com.sk89q.worldedit.world.block.BlockState;
|
|
import com.sk89q.worldedit.world.block.BlockTypes;
|
|
|
|
public class SnowSimulator implements LayerFunction {
|
|
|
|
private final BlockState ice = BlockTypes.ICE.getDefaultState();
|
|
private final BlockState snow = BlockTypes.SNOW.getDefaultState();
|
|
private final BlockState snowBlock = BlockTypes.SNOW_BLOCK.getDefaultState();
|
|
|
|
private final Property<Integer> snowLayersProperty = BlockTypes.SNOW.getProperty("layers");
|
|
private final Property<Integer> waterLevelProperty = BlockTypes.WATER.getProperty("level");
|
|
|
|
private final Extent extent;
|
|
private final boolean stack;
|
|
|
|
private int affected;
|
|
|
|
public SnowSimulator(Extent extent, boolean stack) {
|
|
this.extent = extent;
|
|
this.stack = stack;
|
|
|
|
this.affected = 0;
|
|
}
|
|
|
|
public int getAffected() {
|
|
return this.affected;
|
|
}
|
|
|
|
@Override
|
|
public boolean isGround(BlockVector3 position) {
|
|
BlockState block = this.extent.getBlock(position);
|
|
|
|
// We're returning the first block we can place *on top of*
|
|
if (block.getBlockType().getMaterial().isAir() || (stack && block.getBlockType() == BlockTypes.SNOW)) {
|
|
return false;
|
|
}
|
|
|
|
// Unless it's water
|
|
if (block.getBlockType() == BlockTypes.WATER) {
|
|
return true;
|
|
}
|
|
|
|
// Can only place on full solid blocks
|
|
return block.getBlockType().getMaterial().isFullCube()
|
|
&& block.getBlockType().getMaterial().isSolid();
|
|
}
|
|
|
|
@Override
|
|
public boolean apply(BlockVector3 position, int depth) throws WorldEditException {
|
|
if (depth > 0) {
|
|
// We only care about the first layer.
|
|
return false;
|
|
}
|
|
|
|
BlockState block = this.extent.getBlock(position);
|
|
|
|
if (block.getBlockType() == BlockTypes.WATER) {
|
|
if (block.getState(waterLevelProperty) == 0) {
|
|
if (this.extent.setBlock(position, ice)) {
|
|
affected++;
|
|
}
|
|
}
|
|
return false;
|
|
}
|
|
|
|
// Can't put snow this far up
|
|
if (position.getBlockY() == this.extent.getMaximumPoint().getBlockY()) {
|
|
return false;
|
|
}
|
|
|
|
BlockVector3 abovePosition = position.add(0, 1, 0);
|
|
BlockState above = this.extent.getBlock(abovePosition);
|
|
|
|
// Can only replace air (or snow in stack mode)
|
|
if (!above.getBlockType().getMaterial().isAir() && (!stack || above.getBlockType() != BlockTypes.SNOW)) {
|
|
return false;
|
|
}
|
|
|
|
if (stack && above.getBlockType() == BlockTypes.SNOW) {
|
|
int currentHeight = above.getState(snowLayersProperty);
|
|
// We've hit the highest layer (If it doesn't contain current + 2 it means it's 1 away from full)
|
|
if (!snowLayersProperty.getValues().contains(currentHeight + 2)) {
|
|
if (this.extent.setBlock(abovePosition, snowBlock)) {
|
|
this.affected++;
|
|
}
|
|
} else {
|
|
if (this.extent.setBlock(abovePosition, above.with(snowLayersProperty, currentHeight + 1))) {
|
|
this.affected++;
|
|
}
|
|
}
|
|
return false;
|
|
}
|
|
if (this.extent.setBlock(abovePosition, snow)) {
|
|
this.affected++;
|
|
}
|
|
return false;
|
|
}
|
|
}
|