diff --git a/src/main/java/com/sk89q/worldedit/masks/BlockMask.java b/src/main/java/com/sk89q/worldedit/masks/BlockMask.java index e852a9990..ee52d511a 100644 --- a/src/main/java/com/sk89q/worldedit/masks/BlockMask.java +++ b/src/main/java/com/sk89q/worldedit/masks/BlockMask.java @@ -1,14 +1,15 @@ package com.sk89q.worldedit.masks; -import java.util.Collection; -import java.util.HashSet; -import java.util.Set; - import com.sk89q.worldedit.EditSession; import com.sk89q.worldedit.Vector; import com.sk89q.worldedit.blocks.BaseBlock; +import java.util.Collection; +import java.util.HashSet; +import java.util.Set; + public class BlockMask extends AbstractMask { + private final Set blocks; public BlockMask() { @@ -19,6 +20,13 @@ public class BlockMask extends AbstractMask { this.blocks = types; } + public BlockMask(BaseBlock... block) { + blocks = new HashSet(); + for (BaseBlock b : block) { + add(b); + } + } + public BlockMask(BaseBlock block) { this(); add(block); @@ -35,7 +43,8 @@ public class BlockMask extends AbstractMask { @Override public boolean matches(EditSession editSession, Vector pos) { BaseBlock block = editSession.getBlock(pos); - return blocks.contains(block) + return blocks.contains(block) || blocks.contains(new BaseBlock(block.getType(), -1)); - } + } + }