vcmi/lib/battle/SideInBattle.cpp

/*
 * SideInBattle.cpp, part of VCMI engine
 *
 * Authors: listed in file AUTHORS in main folder
 *
 * License: GNU General Public License v2.0 or later
 * Full text of license available in license.txt file, in main folder
 *
 */
#include "StdInc.h"
#include "SideInBattle.h"
#include "../mapObjects/CArmedInstance.h"

VCMI_LIB_NAMESPACE_BEGIN

void SideInBattle::init(const CGHeroInstance * Hero, const CArmedInstance * Army)
{
	hero = Hero;
	armyObject = Army;
	color = armyObject->getOwner();

	if(armyObject->ID == Obj::CREATURE_GENERATOR1
		|| armyObject->ID == Obj::CREATURE_GENERATOR2
		|| armyObject->ID == Obj::CREATURE_GENERATOR3
		|| armyObject->ID == Obj::CREATURE_GENERATOR4)
	{
		color = PlayerColor::NEUTRAL;
	}

	if(color == PlayerColor::UNFLAGGABLE)
		color = PlayerColor::NEUTRAL;
}

VCMI_LIB_NAMESPACE_END