HealFightTree.java

package colloid.model.control;

import java.util.ArrayList;
import java.util.Comparator;
import java.util.Iterator;

import colloid.model.event.Ability;
import colloid.model.event.Actor;
import colloid.model.event.Fight;
import colloid.model.event.CombatEvent.EventType;

public class HealFightTree extends FightTree {

    public HealFightTree(Fight fight) {
        super(fight);
        init(fight.new ActorHealDoneComparator());
    }

    @Override
    protected double valueDone(Actor actor, Fight fight) {
        return actor.getHealDone(fight);
    }

    @Override
    protected ArrayList<Ability> getAbilities(Actor actor) {
        ArrayList<Ability> items = new ArrayList<Ability>();

        Iterator<Ability> iterAbility = actor.getAbilities().iterator();
        while (iterAbility.hasNext()) {
            Ability ability = iterAbility.next();
            if (ability.getHealDone() > 0) {
                items.add(ability);
            }
        }
        return items;
    }

    @Override
    protected String abilityInfo(Ability ability, double totalVal) {
        return ability.healInfo(totalVal);
    }

    @Override
    protected Comparator<Ability> getAbilityComparator(Actor actor) {
        return actor.new AbilityHealDoneComparator();
    }
}