ActionRest.java

package Roguelike.Entity.AI.BehaviourTree.Actions;

import java.util.HashSet;

import Roguelike.Global.Statistic;
import Roguelike.Entity.GameEntity;
import Roguelike.Entity.AI.BehaviourTree.BehaviourTree.BehaviourTreeState;
import Roguelike.Entity.Tasks.TaskWait;

import com.badlogic.gdx.utils.XmlReader.Element;

public class ActionRest extends AbstractAction
{
	private HashSet<String> interestedValues = new HashSet<String>();
	private HashSet<String> tempValues = new HashSet<String>();

	int restedTurns = 0;

	@Override
	public BehaviourTreeState evaluate( GameEntity entity )
	{
		tempValues.clear();

		if ( entity.HP < entity.getMaxHP() )
		{
			tempValues.add( "HP" );
		}
		// add ability cooldown checks here

		boolean completed = false;
		for ( String item : interestedValues )
		{
			if ( !tempValues.contains( item ) )
			{
				completed = true;
				break;
			}
		}

		// swap
		HashSet<String> temp = interestedValues;
		interestedValues = tempValues;
		tempValues = temp;

		boolean hasRestingToDo = completed || interestedValues.size() > 0;

		if ( hasRestingToDo )
		{
			entity.tasks.add( new TaskWait( 2 + restedTurns / 5 ) );
			State = BehaviourTreeState.RUNNING;

			restedTurns++;
		}
		else
		{
			State = BehaviourTreeState.SUCCEEDED;
			restedTurns = 0;
		}

		return State;
	}

	@Override
	public void cancel()
	{
		interestedValues.clear();

		restedTurns = 0;
	}

	@Override
	public void parse( Element xmlElement )
	{
	}

}