136 lines
3.6 KiB
Java
136 lines
3.6 KiB
Java
package resonantinduction.electrical.armbot.task;
|
|
|
|
import java.util.List;
|
|
|
|
import net.minecraft.entity.Entity;
|
|
import net.minecraft.entity.EntityAgeable;
|
|
import net.minecraft.entity.player.EntityPlayer;
|
|
import net.minecraft.entity.projectile.EntityArrow;
|
|
import net.minecraft.nbt.NBTTagCompound;
|
|
import net.minecraft.util.AxisAlignedBB;
|
|
import resonantinduction.core.ArgumentData;
|
|
import resonantinduction.electrical.armbot.IArmbot;
|
|
import resonantinduction.electrical.armbot.TaskBaseProcess;
|
|
import calclavia.api.resonantinduction.electrical.ArmbotEntity;
|
|
|
|
public class TaskGrabEntity extends TaskGrabPrefab
|
|
{
|
|
/**
|
|
* If the grab command is specific to one entity this tell whether or not to grab the child
|
|
* version of that entity.
|
|
*/
|
|
public boolean child = false;
|
|
/** The item to be collected. */
|
|
private Class<? extends Entity> entityToInclude;
|
|
|
|
@SuppressWarnings("unchecked")
|
|
public TaskGrabEntity()
|
|
{
|
|
super("Grab-Entity");
|
|
this.args.add(new ArgumentData("child", false));
|
|
// this.defautlArguments.add(new ArgumentListData<Class<? extends Entity>>("Entity",
|
|
// Entity.class, (Class<? extends Entity>[]) EntityDictionary.getList().toArray(new
|
|
// Object[1])));
|
|
}
|
|
|
|
@Override
|
|
public ProcessReturn onMethodCalled()
|
|
{
|
|
if (super.onMethodCalled() == ProcessReturn.CONTINUE)
|
|
{
|
|
this.entityToInclude = Entity.class;
|
|
try
|
|
{
|
|
if (this.getArg("Entity") instanceof Class)
|
|
{
|
|
this.entityToInclude = (Class<? extends Entity>) this.getArg("Entity");
|
|
}
|
|
}
|
|
catch (Exception e)
|
|
{
|
|
|
|
}
|
|
if (this.getArg("child") instanceof Boolean)
|
|
{
|
|
this.child = (Boolean) this.getArg("child");
|
|
}
|
|
|
|
return ProcessReturn.CONTINUE;
|
|
}
|
|
return ProcessReturn.GENERAL_ERROR;
|
|
}
|
|
|
|
@Override
|
|
public ProcessReturn onUpdate()
|
|
{
|
|
if (super.onUpdate() == ProcessReturn.CONTINUE)
|
|
{
|
|
if (((IArmbot) this.program.getMachine()).getHeldObject() != null)
|
|
{
|
|
return ProcessReturn.DONE;
|
|
}
|
|
@SuppressWarnings("unchecked")
|
|
List<Entity> found = this.program.getMachine().getLocation().left().getEntitiesWithinAABB(entityToInclude, AxisAlignedBB.getBoundingBox(this.armPos.x - radius, this.armPos.y - radius, this.armPos.z - radius, this.armPos.x + radius, this.armPos.y + radius, this.armPos.z + radius));
|
|
|
|
if (found != null && found.size() > 0)
|
|
{
|
|
for (Entity entity : found)
|
|
{
|
|
if ((entity != null && !(entity instanceof EntityArrow) && !(entity instanceof EntityPlayer) && (!(entity instanceof EntityAgeable) || (entity instanceof EntityAgeable && child == ((EntityAgeable) entity).isChild()))))
|
|
{
|
|
((IArmbot) this.program.getMachine()).grabObject(entity);
|
|
if (this.belt != null)
|
|
{
|
|
belt.ignoreEntity(entity);
|
|
}
|
|
return ProcessReturn.DONE;
|
|
}
|
|
}
|
|
}
|
|
|
|
return ProcessReturn.CONTINUE;
|
|
}
|
|
return ProcessReturn.GENERAL_ERROR;
|
|
}
|
|
|
|
@Override
|
|
public void load(NBTTagCompound taskCompound)
|
|
{
|
|
super.loadProgress(taskCompound);
|
|
this.child = taskCompound.getBoolean("child");
|
|
this.entityToInclude = ArmbotEntity.get(taskCompound.getString("name"));
|
|
}
|
|
|
|
@Override
|
|
public void save(NBTTagCompound taskCompound)
|
|
{
|
|
super.saveProgress(taskCompound);
|
|
taskCompound.setBoolean("child", child);
|
|
taskCompound.setString("name", ((this.entityToInclude != null) ? ArmbotEntity.get(this.entityToInclude) : ""));
|
|
}
|
|
|
|
@Override
|
|
public String toString()
|
|
{
|
|
String baby = "";
|
|
String entity = "";
|
|
if (this.entityToInclude != null)
|
|
{
|
|
entity = ArmbotEntity.get(this.entityToInclude);
|
|
if (this.child)
|
|
{
|
|
// TODO do check for EntityAgable
|
|
baby = "baby ";
|
|
}
|
|
}
|
|
return "GRAB " + baby + entity;
|
|
}
|
|
|
|
@Override
|
|
public TaskBaseProcess clone()
|
|
{
|
|
return new TaskGrabEntity();
|
|
}
|
|
|
|
}
|