/*
* This file is part of Skript.
*
* Skript 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.
*
* Skript 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 Skript. If not, see <http://www.gnu.org/licenses/>.
*
*
* Copyright 2011-2014 Peter Güttinger
*
*/
package ch.njol.skript.expressions;
import java.lang.reflect.Array;
import java.util.ArrayList;
import java.util.Iterator;
import java.util.List;
import org.bukkit.Bukkit;
import org.bukkit.Location;
import org.bukkit.World;
import org.bukkit.entity.Entity;
import org.bukkit.entity.ExperienceOrb;
import org.bukkit.entity.Player;
import org.bukkit.event.Event;
import org.eclipse.jdt.annotation.Nullable;
import ch.njol.skript.Skript;
import ch.njol.skript.doc.Description;
import ch.njol.skript.doc.Examples;
import ch.njol.skript.doc.Name;
import ch.njol.skript.doc.Since;
import ch.njol.skript.entity.EntityData;
import ch.njol.skript.lang.Expression;
import ch.njol.skript.lang.ExpressionType;
import ch.njol.skript.lang.Literal;
import ch.njol.skript.lang.SkriptParser.ParseResult;
import ch.njol.skript.lang.util.SimpleExpression;
import ch.njol.skript.log.BlockingLogHandler;
import ch.njol.skript.log.LogHandler;
import ch.njol.skript.log.SkriptLogger;
import ch.njol.util.Kleenean;
import ch.njol.util.NullableChecker;
import ch.njol.util.StringUtils;
import ch.njol.util.coll.iterator.CheckedIterator;
import ch.njol.util.coll.iterator.NonNullIterator;
/**
* @author Peter Güttinger
*/
@Name("Entities")
@Description("all entities in all world, in a specific world or in a radius around a certain location, e.g. 'all players', 'all creepers in the player's world', or 'players in radius 100 of the player'.")
@Examples({"kill all creepers in the player's world",
"send \"Psst!\" to all players witin 100 meters of the player",
"give a diamond to all ops",
"heal all tamed wolves in radius 2000 around {town center}"})
@Since("1.2.1")
public class ExprEntities extends SimpleExpression<Entity> {
static {
Skript.registerExpression(ExprEntities.class, Entity.class, ExpressionType.PATTERN_MATCHES_EVERYTHING,
"[all] %*entitydatas% [(in|of) [world[s]] %-worlds%]",
"[all] entities of type[s] %entitydatas% [(in|of) [world[s]] %-worlds%]",
"[all] %*entitydatas% (within|[with]in radius) %number% [(block[s]|met(er|re)[s])] (of|around) %location%",
"[all] entities of type[s] %entitydatas% in radius %number% (of|around) %location%");
}
@SuppressWarnings("null")
Expression<? extends EntityData<?>> types;
@Nullable
Expression<World> worlds;
@Nullable
private Expression<Number> radius;
@Nullable
private Expression<Location> center;
@Nullable
private Expression<? extends Entity> centerEntity;
Class<? extends Entity> returnType = Entity.class;
private int matchedPattern;
@SuppressWarnings({"unchecked", "null"})
@Override
public boolean init(final Expression<?>[] exprs, final int matchedPattern, final Kleenean isDelayed, final ParseResult parseResult) {
this.matchedPattern = matchedPattern;
types = (Expression<? extends EntityData<?>>) exprs[0];
if (matchedPattern % 2 == 0) {
for (final EntityData<?> d : ((Literal<EntityData<?>>) types).getAll()) {
if (d.isPlural().isFalse() || d.isPlural().isUnknown() && !StringUtils.startsWithIgnoreCase(parseResult.expr, "all"))
return false;
}
}
if (matchedPattern < 2) {
worlds = (Expression<World>) exprs[exprs.length - 1];
} else {
radius = (Expression<Number>) exprs[exprs.length - 2];
center = (Expression<Location>) exprs[exprs.length - 1];
final BlockingLogHandler log = SkriptLogger.startLogHandler(new BlockingLogHandler());
try {
centerEntity = center.getSource().getConvertedExpression(Entity.class);
} finally {
log.stop();
}
}
if (types instanceof Literal && ((Literal<EntityData<?>>) types).getAll().length == 1) {
returnType = ((Literal<EntityData<?>>) types).getSingle().getType();
}
return true;
}
@Override
public boolean isSingle() {
return false;
}
@Override
public Class<? extends Entity> getReturnType() {
return returnType;
}
@Override
@Nullable
protected Entity[] get(final Event e) {
if (matchedPattern >= 2) {
final Iterator<? extends Entity> iter = iterator(e);
if (iter == null || !iter.hasNext())
return new Entity[0];
final List<Entity> l = new ArrayList<Entity>();
while (iter.hasNext())
l.add(iter.next());
return l.toArray((Entity[]) Array.newInstance(returnType, l.size()));
} else {
return EntityData.getAll(types.getAll(e), returnType, worlds != null ? worlds.getArray(e) : null);
}
}
@SuppressWarnings("unchecked")
@Override
public boolean isLoopOf(final String s) {
if (!(types instanceof Literal<?>))
return false;
final LogHandler h = SkriptLogger.startLogHandler(new BlockingLogHandler());
try {
final EntityData<?> d = EntityData.parseWithoutIndefiniteArticle(s);
if (d != null) {
for (final EntityData<?> t : ((Literal<EntityData<?>>) types).getAll()) {
assert t != null;
if (!d.isSupertypeOf(t))
return false;
}
return true;
}
} finally {
h.stop();
}
return false;
}
@SuppressWarnings("null")
@Override
@Nullable
public Iterator<? extends Entity> iterator(final Event e) {
if (matchedPattern >= 2) {
final Entity en;
final Location l;
if (centerEntity != null) {
en = centerEntity.getSingle(e);
if (en == null)
return null;
l = en.getLocation();
} else {
assert center != null;
l = center.getSingle(e);
if (l == null)
return null;
en = l.getWorld().spawn(l, ExperienceOrb.class);
}
assert radius != null;
final Number n = radius.getSingle(e);
if (n == null)
return null;
final double d = n.doubleValue();
final List<Entity> es = en.getNearbyEntities(d, d, d);
if (centerEntity == null)
en.remove();
final double radiusSquared = d * d * Skript.EPSILON_MULT;
final EntityData<?>[] ts = types.getAll(e);
return new CheckedIterator<Entity>(es.iterator(), new NullableChecker<Entity>() {
@Override
public boolean check(final @Nullable Entity e) {
if (e == null || e.getLocation().distanceSquared(l) > radiusSquared)
return false;
for (final EntityData<?> t : ts) {
if (t.isInstance(e))
return true;
}
return false;
}
});
} else {
if (worlds == null && returnType == Player.class)
return super.iterator(e);
return new NonNullIterator<Entity>() {
private final World[] ws = worlds == null ? Bukkit.getWorlds().toArray(new World[0]) : worlds.getArray(e);
private int w = -1;
private final EntityData<?>[] ts = types.getAll(e);
@Nullable
private Iterator<? extends Entity> curIter = null;
@Override
@Nullable
protected Entity getNext() {
while (true) {
while (curIter == null || !curIter.hasNext()) {
w++;
if (w == ws.length)
return null;
curIter = ws[w].getEntitiesByClass(returnType).iterator();
}
while (curIter.hasNext()) {
final Entity current = curIter.next();
for (final EntityData<?> t : ts) {
if (t.isInstance(current))
return current;
}
}
}
}
};
}
}
@SuppressWarnings("null")
@Override
public String toString(final @Nullable Event e, final boolean debug) {
return "all entities of types " + types.toString(e, debug) +
(worlds != null ? " in " + worlds.toString(e, debug) :
radius != null && center != null ? " in radius " + radius.toString(e, debug) + " around " + center.toString(e, debug) : "");
}
}