evil-inside/src/ei/game/player/Human.java

package ei.game.player;

import java.util.ArrayList;

import ei.engine.scene.Node;
import ei.game.scene.GameEntity;
import ei.game.scene.units.Unit;

public class Human {
	private ArrayList<GameEntity> units;
	private Node unitsNode;
	
	public Human(){
		units = new ArrayList<GameEntity>();
		unitsNode = new Node("UnitsNode");
	}
	
	public void addUnit(Unit u){
		units.add(u);
		unitsNode.add(u.getSprite());
	}
	
	public void removeUnit(Unit u){
		units.remove(u);
		unitsNode.remove(u.getSprite());
	}
	
	public Node getNode(){
		return unitsNode;
	}
}