Examples with MaskUnion com.fastasyncworldedit.core.function.mask.MaskUnion used on opensource projects

Example 1 with MaskUnion

use of com.fastasyncworldedit.core.function.mask.MaskUnion in project FastAsyncWorldEdit by IntellectualSites.

the class EditSession method fixLiquid.

/**
 * Fix liquids so that they turn into stationary blocks and extend outward.
 *
 * @param origin the original position
 * @param radius the radius to fix
 * @param fluid  the type of the fluid
 * @return number of blocks affected
 * @throws MaxChangedBlocksException thrown if too many blocks are changed
 */
public int fixLiquid(BlockVector3 origin, double radius, BlockType fluid) throws MaxChangedBlocksException {
    checkNotNull(origin);
    checkArgument(radius >= 0, "radius >= 0 required");
    // Our origins can only be liquids
    Mask liquidMask = new SingleBlockTypeMask(this, fluid);
    // But we will also visit air blocks
    MaskIntersection blockMask = new MaskUnion(liquidMask, Masks.negate(new ExistingBlockMask(this)));
    // There are boundaries that the routine needs to stay in
    Mask mask = new MaskIntersection(new BoundedHeightMask(minY, Math.min(origin.getBlockY(), maxY)), new RegionMask(new EllipsoidRegion(null, origin, Vector3.at(radius, radius, radius))), blockMask);
    BlockReplace replace = new BlockReplace(this, fluid.getDefaultState());
    // FAWE start - provide extent for preloading, world min/maxY
    NonRisingVisitor visitor = new NonRisingVisitor(mask, replace, Integer.MAX_VALUE, minY, maxY, this);
    // Around the origin in a 3x3 block
    for (BlockVector3 position : CuboidRegion.fromCenter(origin, 1)) {
        if (liquidMask.test(position)) {
            visitor.visit(position);
        }
    }
    Operations.completeLegacy(visitor);
    return visitor.getAffected();
}

Example 2 with MaskUnion

use of com.fastasyncworldedit.core.function.mask.MaskUnion 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
}