AttributeCounter.java

package net.demilich.metastone.game.spells.desc.valueprovider;

import java.util.List;

import net.demilich.metastone.game.Attribute;
import net.demilich.metastone.game.GameContext;
import net.demilich.metastone.game.Player;
import net.demilich.metastone.game.entities.Entity;
import net.demilich.metastone.game.targeting.EntityReference;

public class AttributeCounter extends ValueProvider {

	public AttributeCounter(ValueProviderDesc desc) {
		super(desc);
	}

	@Override
	protected int provideValue(GameContext context, Player player, Entity target, Entity host) {
		EntityReference source = desc.getSource();
		List<Entity> relevantEntities = context.resolveTarget(player, host, source);
		int count = 0;
		Attribute attribute = (Attribute) desc.get(ValueProviderArg.ATTRIBUTE);
		for (Entity entity : relevantEntities) {
			if (entity.hasAttribute(attribute)) {
				count++;
			}
		}
		return count;
	}

}