Fix radius mask

This commit is contained in:
Jesse Boyd 2016-09-25 05:00:47 +10:00
parent f2e576a14f
commit fca33e5e87
3 changed files with 26 additions and 11 deletions

View File

@ -7,29 +7,33 @@ import com.sk89q.worldedit.function.mask.BlockMask;
import java.util.Collection; import java.util.Collection;
public class AdjacentMask extends BlockMask { public class AdjacentMask extends BlockMask {
public AdjacentMask(Extent extent, Collection<BaseBlock> blocks) { private final int required;
public AdjacentMask(Extent extent, Collection<BaseBlock> blocks, int required) {
super(extent, blocks); super(extent, blocks);
this.required = required;
} }
@Override @Override
public boolean test(Vector v) { public boolean test(Vector v) {
int count = 0;
double x = v.x; double x = v.x;
double y = v.y; double y = v.y;
double z = v.z; double z = v.z;
v.x = x + 1; v.x = x + 1;
if (super.test(v)) { v.x = x; return true; } if (super.test(v) && ++count == required) { v.x = x; return true; }
v.x = x - 1; v.x = x - 1;
if (super.test(v)) { v.x = x; return true; } if (super.test(v) && ++count == required) { v.x = x; return true; }
v.x = x; v.x = x;
v.y = y + 1; v.y = y + 1;
if (super.test(v)) { v.y = y; return true; } if (super.test(v) && ++count == required) { v.y = y; return true; }
v.y = y - 1; v.y = y - 1;
if (super.test(v)) { v.y = y; return true; } if (super.test(v) && ++count == required) { v.y = y; return true; }
v.y = y; v.y = y;
v.z = z + 1; v.z = z + 1;
if (super.test(v)) { v.z = z; return true; } if (super.test(v) && ++count == required) { v.z = z; return true; }
v.z = z - 1; v.z = z - 1;
if (super.test(v)) { v.z = z; return true; } if (super.test(v) && ++count == required) { v.z = z; return true; }
v.z = z; v.z = z;
return false; return false;
} }

View File

@ -27,8 +27,8 @@ public class RadiusMask implements Mask, ResettableMask{
pos = new Vector(to); pos = new Vector(to);
} }
int dx = Math.abs((int) (pos.x - to.x)); int dx = Math.abs((int) (pos.x - to.x));
int dy = Math.abs((int) (pos.x - to.x)); int dy = Math.abs((int) (pos.y - to.y));
int dz = Math.abs((int) (pos.x - to.x)); int dz = Math.abs((int) (pos.z - to.z));
int d = dx * dx; int d = dx * dx;
if (d < minSqr || d > maxSqr) { if (d < minSqr || d > maxSqr) {
return false; return false;

View File

@ -45,6 +45,7 @@ import java.util.HashSet;
import java.util.List; import java.util.List;
import java.util.Set; import java.util.Set;
import static com.google.common.base.Preconditions.checkNotNull; import static com.google.common.base.Preconditions.checkNotNull;
/** /**
@ -147,7 +148,7 @@ public class DefaultMaskParser extends InputParser<Mask> {
throw new InputParseException("Unknown angle '" + component + "' (not in form `/#,#`)"); throw new InputParseException("Unknown angle '" + component + "' (not in form `/#,#`)");
} }
} }
case '{': case '{': {
String[] split = component.substring(1).split(","); String[] split = component.substring(1).split(",");
if (split.length != 2) { if (split.length != 2) {
throw new InputParseException("Unknown range '" + component + "' (not in form `{#,#`)"); throw new InputParseException("Unknown range '" + component + "' (not in form `{#,#`)");
@ -159,11 +160,21 @@ public class DefaultMaskParser extends InputParser<Mask> {
} catch (NumberFormatException e) { } catch (NumberFormatException e) {
throw new InputParseException("Unknown range '" + component + "' (not in form `{#,#`)"); throw new InputParseException("Unknown range '" + component + "' (not in form `{#,#`)");
} }
}
case '~': { case '~': {
String[] split = component.substring(1).split("=");
ParserContext tempContext = new ParserContext(context); ParserContext tempContext = new ParserContext(context);
tempContext.setRestricted(false); tempContext.setRestricted(false);
tempContext.setPreferringWildcard(true); tempContext.setPreferringWildcard(true);
return new AdjacentMask(extent, worldEdit.getBlockFactory().parseFromListInput(component.substring(1), tempContext)); try {
int requiredCount = 1;
if (split.length == 2) {
requiredCount = Integer.parseInt(split[1]);
}
return new AdjacentMask(extent, worldEdit.getBlockFactory().parseFromListInput(component.substring(1), tempContext));
} catch (NumberFormatException e) {
throw new InputParseException("Unknown adjacent mask '" + component + "' (not in form `~<ids>[=count]`)");
}
} }
case '>': case '>':
case '<': case '<':