Examples with WaterloggedRemover com.sk89q.worldedit.function.pattern.WaterloggedRemover used on opensource projects

Example 1 with WaterloggedRemover

use of com.sk89q.worldedit.function.pattern.WaterloggedRemover in project FastAsyncWorldEdit by IntellectualSites.

the class EditSession method drainArea.

/**
 * Drain nearby pools of water or lava, optionally removed waterlogged states from blocks.
 *
 * @param origin      the origin to drain from, which will search a 3x3 area
 * @param radius      the radius of the removal, where a value should be 0 or greater
 * @param waterlogged true to make waterlogged blocks non-waterlogged as well
 * @param plants      true to remove underwater plants
 * @return number of blocks affected
 * @throws MaxChangedBlocksException thrown if too many blocks are changed
 */
public int drainArea(BlockVector3 origin, double radius, boolean waterlogged, boolean plants) throws MaxChangedBlocksException {
    checkNotNull(origin);
    checkArgument(radius >= 0, "radius >= 0 required");
    // FAWE start - liquidmask
    Mask liquidMask;
    if (plants) {
        liquidMask = new BlockTypeMask(this, BlockTypes.LAVA, BlockTypes.WATER, BlockTypes.BUBBLE_COLUMN, BlockTypes.KELP_PLANT, BlockTypes.KELP, BlockTypes.SEAGRASS, BlockTypes.TALL_SEAGRASS);
    } else {
        liquidMask = new BlockMaskBuilder().addTypes(BlockTypes.WATER, BlockTypes.LAVA, BlockTypes.BUBBLE_COLUMN).build(this);
    }
    // FAWE end
    if (waterlogged) {
        Map<String, String> stateMap = new HashMap<>();
        stateMap.put("waterlogged", "true");
        // FAWE start
        liquidMask = new MaskUnion(liquidMask, new BlockStateMask(this, stateMap, true));
    // FAWE end
    }
    Mask mask = new MaskIntersection(new BoundedHeightMask(minY, maxY), new RegionMask(new EllipsoidRegion(null, origin, Vector3.at(radius, radius, radius))), // FAWE start
    liquidMask);
    // FAWE end
    BlockReplace replace;
    if (waterlogged) {
        replace = new BlockReplace(this, new WaterloggedRemover(this));
    } else {
        replace = new BlockReplace(this, BlockTypes.AIR.getDefaultState());
    }
    // FAWE start - provide extent for preloading, min/max y
    RecursiveVisitor visitor = new RecursiveVisitor(mask, replace, (int) (radius * 2 + 1), minY, maxY, this);
    // Around the origin in a 3x3 block
    for (BlockVector3 position : CuboidRegion.fromCenter(origin, 1)) {
        if (mask.test(position)) {
            visitor.visit(position);
        }
    }
    Operations.completeLegacy(visitor);
    // FAWE start
    return this.changes = visitor.getAffected();
// FAWE end
}