| 12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076 | 
							- /*
 
-  * CCreatureSet.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 "CCreatureSet.h"
 
- #include "CCreatureHandler.h"
 
- #include "VCMI_Lib.h"
 
- #include "CModHandler.h"
 
- #include "mapObjects/CGHeroInstance.h"
 
- #include "IGameCallback.h"
 
- #include "CGameState.h"
 
- #include "CGeneralTextHandler.h"
 
- #include "spells/CSpellHandler.h"
 
- #include "CHeroHandler.h"
 
- #include "IBonusTypeHandler.h"
 
- #include "serializer/JsonSerializeFormat.h"
 
- #include "NetPacksBase.h"
 
- bool CreatureSlotComparer::operator()(const TPairCreatureSlot & lhs, const TPairCreatureSlot & rhs)
 
- {
 
- 	return lhs.first->getAIValue() < rhs.first->getAIValue(); // Descendant order sorting
 
- }
 
- const CStackInstance &CCreatureSet::operator[](SlotID slot) const
 
- {
 
- 	auto i = stacks.find(slot);
 
- 	if (i != stacks.end())
 
- 		return *i->second;
 
- 	else
 
- 		throw std::runtime_error("That slot is empty!");
 
- }
 
- const CCreature* CCreatureSet::getCreature(SlotID slot) const
 
- {
 
- 	auto i = stacks.find(slot);
 
- 	if (i != stacks.end())
 
- 		return i->second->type;
 
- 	else
 
- 		return nullptr;
 
- }
 
- bool CCreatureSet::setCreature(SlotID slot, CreatureID type, TQuantity quantity) /*slots 0 to 6 */
 
- {
 
- 	if(!slot.validSlot())
 
- 	{
 
- 		logGlobal->error("Cannot set slot %d", slot.getNum());
 
- 		return false;
 
- 	}
 
- 	if(!quantity)
 
- 	{
 
- 		logGlobal->warn("Using set creature to delete stack?");
 
- 		eraseStack(slot);
 
- 		return true;
 
- 	}
 
- 	if(hasStackAtSlot(slot)) //remove old creature
 
- 		eraseStack(slot);
 
- 	auto armyObj = castToArmyObj();
 
- 	bool isHypotheticArmy = armyObj ? armyObj->isHypothetic() : false;
 
- 	putStack(slot, new CStackInstance(type, quantity, isHypotheticArmy));
 
- 	return true;
 
- }
 
- SlotID CCreatureSet::getSlotFor(CreatureID creature, ui32 slotsAmount) const /*returns -1 if no slot available */
 
- {
 
- 	return getSlotFor(VLC->creh->objects[creature], slotsAmount);
 
- }
 
- SlotID CCreatureSet::getSlotFor(const CCreature *c, ui32 slotsAmount) const
 
- {
 
- 	assert(c && c->valid());
 
- 	for(auto & elem : stacks)
 
- 	{
 
- 		assert(elem.second->type->valid());
 
- 		if(elem.second->type == c)
 
- 		{
 
- 			return elem.first; //if there is already such creature we return its slot id
 
- 		}
 
- 	}
 
- 	return getFreeSlot(slotsAmount);
 
- }
 
- bool CCreatureSet::hasCreatureSlots(const CCreature * c, SlotID exclude) const
 
- {
 
- 	assert(c && c->valid());
 
- 	for(auto & elem : stacks) // elem is const
 
- 	{
 
- 		if(elem.first == exclude) // Check slot
 
- 			continue;
 
- 		if(!elem.second || !elem.second->type) // Check creature
 
- 			continue;
 
- 		assert(elem.second->type->valid());
 
- 		if(elem.second->type == c)
 
- 			return true;
 
- 	}
 
- 	return false;
 
- }
 
- std::vector<SlotID> CCreatureSet::getCreatureSlots(const CCreature * c, SlotID exclude, TQuantity ignoreAmount) const
 
- {
 
- 	assert(c && c->valid());
 
- 	std::vector<SlotID> result;
 
- 	for(auto & elem : stacks)
 
- 	{
 
- 		if(elem.first == exclude)
 
- 			continue;
 
- 		if(!elem.second || !elem.second->type || elem.second->type != c)
 
- 			continue;
 
- 		if(elem.second->count == ignoreAmount || elem.second->count < 1)
 
- 			continue;
 
- 		assert(elem.second->type->valid());
 
- 		result.push_back(elem.first);
 
- 	}
 
- 	return result;
 
- }
 
- bool CCreatureSet::isCreatureBalanced(const CCreature * c, TQuantity ignoreAmount) const
 
- {
 
- 	assert(c && c->valid());
 
- 	TQuantity max = 0;
 
- 	TQuantity min = std::numeric_limits<TQuantity>::max();
 
- 	for(auto & elem : stacks)
 
- 	{
 
- 		if(!elem.second || !elem.second->type || elem.second->type != c)
 
- 			continue;
 
- 		const auto count = elem.second->count;
 
- 		if(count == ignoreAmount || count < 1)
 
- 			continue;
 
- 		assert(elem.second->type->valid());
 
- 		if(count > max)
 
- 			max = count;
 
- 		if(count < min)
 
- 			min = count;
 
- 		if(max - min > 1)
 
- 			return false;
 
- 	}
 
- 	return true;
 
- }
 
- SlotID CCreatureSet::getFreeSlot(ui32 slotsAmount) const
 
- {
 
- 	for(ui32 i=0; i<slotsAmount; i++)
 
- 	{
 
- 		if(!vstd::contains(stacks, SlotID(i)))
 
- 		{
 
- 			return SlotID(i); //return first free slot
 
- 		}
 
- 	}
 
- 	return SlotID(); //no slot available
 
- }
 
- std::vector<SlotID> CCreatureSet::getFreeSlots(ui32 slotsAmount) const
 
- {
 
- 	std::vector<SlotID> freeSlots;
 
- 	for(ui32 i = 0; i < slotsAmount; i++)
 
- 	{
 
- 		auto slot = SlotID(i);
 
- 		if(!vstd::contains(stacks, slot))
 
- 			freeSlots.push_back(slot);
 
- 	}
 
- 	return freeSlots;
 
- }
 
- std::queue<SlotID> CCreatureSet::getFreeSlotsQueue(ui32 slotsAmount) const
 
- {
 
- 	std::queue<SlotID> freeSlots;
 
- 	for (ui32 i = 0; i < slotsAmount; i++)
 
- 	{
 
- 		auto slot = SlotID(i);
 
- 		if(!vstd::contains(stacks, slot))
 
- 			freeSlots.push(slot);
 
- 	}
 
- 	return freeSlots;
 
- }
 
- TMapCreatureSlot CCreatureSet::getCreatureMap() const
 
- {
 
- 	TMapCreatureSlot creatureMap;
 
- 	TMapCreatureSlot::key_compare keyComp = creatureMap.key_comp();
 
- 	// https://stackoverflow.com/questions/97050/stdmap-insert-or-stdmap-find
 
- 	// https://www.cplusplus.com/reference/map/map/key_comp/
 
- 	for(auto pair : stacks)
 
- 	{
 
- 		auto creature = pair.second->type;
 
- 		auto slot = pair.first;
 
- 		TMapCreatureSlot::iterator lb = creatureMap.lower_bound(creature);
 
- 		if(lb != creatureMap.end() && !(keyComp(creature, lb->first)))
 
- 			continue;
 
- 		creatureMap.insert(lb, TMapCreatureSlot::value_type(creature, slot));
 
- 	}
 
- 	return creatureMap;
 
- }
 
- TCreatureQueue CCreatureSet::getCreatureQueue(SlotID exclude) const
 
- {
 
- 	TCreatureQueue creatureQueue;
 
- 	for(auto pair : stacks)
 
- 	{
 
- 		if(pair.first == exclude)
 
- 			continue;
 
- 		creatureQueue.push(std::make_pair(pair.second->type, pair.first));
 
- 	}
 
- 	return creatureQueue;
 
- }
 
- TQuantity CCreatureSet::getStackCount(SlotID slot) const
 
- {
 
- 	auto i = stacks.find(slot);
 
- 	if (i != stacks.end())
 
- 		return i->second->count;
 
- 	else
 
- 		return 0; //TODO? consider issuing a warning
 
- }
 
- TExpType CCreatureSet::getStackExperience(SlotID slot) const
 
- {
 
- 	auto i = stacks.find(slot);
 
- 	if (i != stacks.end())
 
- 		return i->second->experience;
 
- 	else
 
- 		return 0; //TODO? consider issuing a warning
 
- }
 
- bool CCreatureSet::mergableStacks(std::pair<SlotID, SlotID> &out, SlotID preferable) const /*looks for two same stacks, returns slot positions */
 
- {
 
- 	//try to match creature to our preferred stack
 
- 	if(preferable.validSlot() &&  vstd::contains(stacks, preferable))
 
- 	{
 
- 		const CCreature *cr = stacks.find(preferable)->second->type;
 
- 		for(auto & elem : stacks)
 
- 		{
 
- 			if(cr == elem.second->type && elem.first != preferable)
 
- 			{
 
- 				out.first = preferable;
 
- 				out.second = elem.first;
 
- 				return true;
 
- 			}
 
- 		}
 
- 	}
 
- 	for(auto i=stacks.begin(); i!=stacks.end(); ++i)
 
- 	{
 
- 		for(auto & elem : stacks)
 
- 		{
 
- 			if(i->second->type == elem.second->type  &&  i->first != elem.first)
 
- 			{
 
- 				out.first = i->first;
 
- 				out.second = elem.first;
 
- 				return true;
 
- 			}
 
- 		}
 
- 	}
 
- 	return false;
 
- }
 
- void CCreatureSet::sweep()
 
- {
 
- 	for(auto i=stacks.begin(); i!=stacks.end(); ++i)
 
- 	{
 
- 		if(!i->second->count)
 
- 		{
 
- 			stacks.erase(i);
 
- 			sweep();
 
- 			break;
 
- 		}
 
- 	}
 
- }
 
- void CCreatureSet::addToSlot(SlotID slot, CreatureID cre, TQuantity count, bool allowMerging)
 
- {
 
- 	const CCreature *c = VLC->creh->objects[cre];
 
- 	if(!hasStackAtSlot(slot))
 
- 	{
 
- 		setCreature(slot, cre, count);
 
- 	}
 
- 	else if(getCreature(slot) == c && allowMerging) //that slot was empty or contained same type creature
 
- 	{
 
- 		setStackCount(slot, getStackCount(slot) + count);
 
- 	}
 
- 	else
 
- 	{
 
- 		logGlobal->error("Failed adding to slot!");
 
- 	}
 
- }
 
- void CCreatureSet::addToSlot(SlotID slot, CStackInstance *stack, bool allowMerging)
 
- {
 
- 	assert(stack->valid(true));
 
- 	if(!hasStackAtSlot(slot))
 
- 	{
 
- 		putStack(slot, stack);
 
- 	}
 
- 	else if(allowMerging && stack->type == getCreature(slot))
 
- 	{
 
- 		joinStack(slot, stack);
 
- 	}
 
- 	else
 
- 	{
 
- 		logGlobal->error("Cannot add to slot %d stack %s", slot.getNum(), stack->nodeName());
 
- 	}
 
- }
 
- bool CCreatureSet::validTypes(bool allowUnrandomized) const
 
- {
 
- 	for(auto & elem : stacks)
 
- 	{
 
- 		if(!elem.second->valid(allowUnrandomized))
 
- 			return false;
 
- 	}
 
- 	return true;
 
- }
 
- bool CCreatureSet::slotEmpty(SlotID slot) const
 
- {
 
- 	return !hasStackAtSlot(slot);
 
- }
 
- bool CCreatureSet::needsLastStack() const
 
- {
 
- 	return false;
 
- }
 
- ui64 CCreatureSet::getArmyStrength() const
 
- {
 
- 	ui64 ret = 0;
 
- 	for(auto & elem : stacks)
 
- 		ret += elem.second->getPower();
 
- 	return ret;
 
- }
 
- ui64 CCreatureSet::getPower (SlotID slot) const
 
- {
 
- 	return getStack(slot).getPower();
 
- }
 
- std::string CCreatureSet::getRoughAmount(SlotID slot, int mode) const
 
- {
 
- 	/// Mode represent return string format
 
- 	/// "Pack" - 0, "A pack of" - 1, "a pack of" - 2
 
- 	int quantity = CCreature::getQuantityID(getStackCount(slot));
 
- 	if(quantity)
 
- 		return VLC->generaltexth->arraytxt[(174 + mode) + 3*CCreature::getQuantityID(getStackCount(slot))];
 
- 	return "";
 
- }
 
- std::string CCreatureSet::getArmyDescription() const
 
- {
 
- 	std::string text;
 
- 	std::vector<std::string> guards;
 
- 	for(auto & elem : stacks)
 
- 	{
 
- 		auto str = boost::str(boost::format("%s %s") % getRoughAmount(elem.first, 2) % getCreature(elem.first)->namePl);
 
- 		guards.push_back(str);
 
- 	}
 
- 	if(guards.size())
 
- 	{
 
- 		for(int i = 0; i < guards.size(); i++)
 
- 		{
 
- 			text += guards[i];
 
- 			if(i + 2 < guards.size())
 
- 				text += ", ";
 
- 			else if(i + 2 == guards.size())
 
- 				text += VLC->generaltexth->allTexts[237];
 
- 		}
 
- 	}
 
- 	return text;
 
- }
 
- int CCreatureSet::stacksCount() const
 
- {
 
- 	return static_cast<int>(stacks.size());
 
- }
 
- void CCreatureSet::setFormation(bool tight)
 
- {
 
- 	formation = tight;
 
- }
 
- void CCreatureSet::setStackCount(SlotID slot, TQuantity count)
 
- {
 
- 	assert(hasStackAtSlot(slot));
 
- 	assert(stacks[slot]->count + count > 0);
 
- 	if (VLC->modh->modules.STACK_EXP && count > stacks[slot]->count)
 
- 		stacks[slot]->experience = static_cast<TExpType>(stacks[slot]->experience * (count / static_cast<double>(stacks[slot]->count)));
 
- 	stacks[slot]->count = count;
 
- 	armyChanged();
 
- }
 
- void CCreatureSet::giveStackExp(TExpType exp)
 
- {
 
- 	for(TSlots::const_iterator i = stacks.begin(); i != stacks.end(); i++)
 
- 		i->second->giveStackExp(exp);
 
- }
 
- void CCreatureSet::setStackExp(SlotID slot, TExpType exp)
 
- {
 
- 	assert(hasStackAtSlot(slot));
 
- 	stacks[slot]->experience = exp;
 
- }
 
- void CCreatureSet::clear()
 
- {
 
- 	while(!stacks.empty())
 
- 	{
 
- 		eraseStack(stacks.begin()->first);
 
- 	}
 
- }
 
- const CStackInstance& CCreatureSet::getStack(SlotID slot) const
 
- {
 
- 	assert(hasStackAtSlot(slot));
 
- 	return *getStackPtr(slot);
 
- }
 
- const CStackInstance* CCreatureSet::getStackPtr(SlotID slot) const
 
- {
 
- 	if(hasStackAtSlot(slot))
 
- 		return stacks.find(slot)->second;
 
- 	else return nullptr;
 
- }
 
- void CCreatureSet::eraseStack(SlotID slot)
 
- {
 
- 	assert(hasStackAtSlot(slot));
 
- 	CStackInstance *toErase = detachStack(slot);
 
- 	vstd::clear_pointer(toErase);
 
- }
 
- bool CCreatureSet::contains(const CStackInstance *stack) const
 
- {
 
- 	if(!stack)
 
- 		return false;
 
- 	for(auto & elem : stacks)
 
- 		if(elem.second == stack)
 
- 			return true;
 
- 	return false;
 
- }
 
- SlotID CCreatureSet::findStack(const CStackInstance *stack) const
 
- {
 
- 	auto h = dynamic_cast<const CGHeroInstance *>(this);
 
- 	if (h && h->commander == stack)
 
- 		return SlotID::COMMANDER_SLOT_PLACEHOLDER;
 
- 	if(!stack)
 
- 		return SlotID();
 
- 	for(auto & elem : stacks)
 
- 		if(elem.second == stack)
 
- 			return elem.first;
 
- 	return SlotID();
 
- }
 
- CArmedInstance * CCreatureSet::castToArmyObj()
 
- {
 
- 	return dynamic_cast<CArmedInstance *>(this);
 
- }
 
- void CCreatureSet::putStack(SlotID slot, CStackInstance *stack)
 
- {
 
- 	assert(slot.getNum() < GameConstants::ARMY_SIZE);
 
- 	assert(!hasStackAtSlot(slot));
 
- 	stacks[slot] = stack;
 
- 	stack->setArmyObj(castToArmyObj());
 
- 	armyChanged();
 
- }
 
- void CCreatureSet::joinStack(SlotID slot, CStackInstance * stack)
 
- {
 
- 	const CCreature *c = getCreature(slot);
 
- 	assert(c == stack->type);
 
- 	assert(c);
 
- 	UNUSED(c);
 
- 	//TODO move stuff
 
- 	changeStackCount(slot, stack->count);
 
- 	vstd::clear_pointer(stack);
 
- }
 
- void CCreatureSet::changeStackCount(SlotID slot, TQuantity toAdd)
 
- {
 
- 	setStackCount(slot, getStackCount(slot) + toAdd);
 
- }
 
- CCreatureSet::CCreatureSet()
 
- {
 
- 	formation = false;
 
- }
 
- CCreatureSet::CCreatureSet(const CCreatureSet&)
 
- {
 
- 	assert(0);
 
- }
 
- CCreatureSet::~CCreatureSet()
 
- {
 
- 	clear();
 
- }
 
- void CCreatureSet::setToArmy(CSimpleArmy &src)
 
- {
 
- 	clear();
 
- 	while(src)
 
- 	{
 
- 		auto i = src.army.begin();
 
- 		putStack(i->first, new CStackInstance(i->second.first, i->second.second));
 
- 		src.army.erase(i);
 
- 	}
 
- }
 
- CStackInstance * CCreatureSet::detachStack(SlotID slot)
 
- {
 
- 	assert(hasStackAtSlot(slot));
 
- 	CStackInstance *ret = stacks[slot];
 
- 	//if(CArmedInstance *armedObj = castToArmyObj())
 
- 	if(ret)
 
- 	{
 
- 		ret->setArmyObj(nullptr); //detaches from current armyobj
 
- 		assert(!ret->armyObj); //we failed detaching?
 
- 	}
 
- 	stacks.erase(slot);
 
- 	armyChanged();
 
- 	return ret;
 
- }
 
- void CCreatureSet::setStackType(SlotID slot, CreatureID type)
 
- {
 
- 	assert(hasStackAtSlot(slot));
 
- 	CStackInstance *s = stacks[slot];
 
- 	s->setType(type);
 
- 	armyChanged();
 
- }
 
- bool CCreatureSet::canBeMergedWith(const CCreatureSet &cs, bool allowMergingStacks) const
 
- {
 
- 	if(!allowMergingStacks)
 
- 	{
 
- 		int freeSlots = stacksCount() - GameConstants::ARMY_SIZE;
 
- 		std::set<const CCreature*> cresToAdd;
 
- 		for(auto & elem : cs.stacks)
 
- 		{
 
- 			SlotID dest = getSlotFor(elem.second->type);
 
- 			if(!dest.validSlot() || hasStackAtSlot(dest))
 
- 				cresToAdd.insert(elem.second->type);
 
- 		}
 
- 		return cresToAdd.size() <= freeSlots;
 
- 	}
 
- 	else
 
- 	{
 
- 		CCreatureSet cres;
 
- 		SlotID j;
 
- 		//get types of creatures that need their own slot
 
- 		for(auto & elem : cs.stacks)
 
- 			if ((j = cres.getSlotFor(elem.second->type)).validSlot())
 
- 				cres.addToSlot(j, elem.second->type->idNumber, 1, true);  //merge if possible
 
- 			//cres.addToSlot(elem.first, elem.second->type->idNumber, 1, true);
 
- 		for(auto & elem : stacks)
 
- 		{
 
- 			if ((j = cres.getSlotFor(elem.second->type)).validSlot())
 
- 				cres.addToSlot(j, elem.second->type->idNumber, 1, true);  //merge if possible
 
- 			else
 
- 				return false; //no place found
 
- 		}
 
- 		return true; //all stacks found their slots
 
- 	}
 
- }
 
- bool CCreatureSet::hasStackAtSlot(SlotID slot) const
 
- {
 
- 	return vstd::contains(stacks, slot);
 
- }
 
- CCreatureSet & CCreatureSet::operator=(const CCreatureSet&cs)
 
- {
 
- 	assert(0);
 
- 	return *this;
 
- }
 
- void CCreatureSet::armyChanged()
 
- {
 
- }
 
- void CCreatureSet::serializeJson(JsonSerializeFormat & handler, const std::string & fieldName, const boost::optional<int> fixedSize)
 
- {
 
- 	if(handler.saving && stacks.empty())
 
- 		return;
 
- 	auto a = handler.enterArray(fieldName);
 
- 	if(handler.saving)
 
- 	{
 
- 		size_t sz = 0;
 
- 		for(const auto & p : stacks)
 
- 			vstd::amax(sz, p.first.getNum()+1);
 
- 		if(fixedSize)
 
- 			vstd::amax(sz, fixedSize.get());
 
- 		a.resize(sz, JsonNode::JsonType::DATA_STRUCT);
 
- 		for(const auto & p : stacks)
 
- 		{
 
- 			auto s = a.enterStruct(p.first.getNum());
 
- 			p.second->serializeJson(handler);
 
- 		}
 
- 	}
 
- 	else
 
- 	{
 
- 		for(size_t idx = 0; idx < a.size(); idx++)
 
- 		{
 
- 			auto s = a.enterStruct(idx);
 
- 			TQuantity amount = 0;
 
- 			handler.serializeInt("amount", amount);
 
- 			if(amount > 0)
 
- 			{
 
- 				CStackInstance * new_stack = new CStackInstance();
 
- 				new_stack->serializeJson(handler);
 
- 				putStack(SlotID((si32)idx), new_stack);
 
- 			}
 
- 		}
 
- 	}
 
- }
 
- CStackInstance::CStackInstance()
 
- 	: armyObj(_armyObj)
 
- {
 
- 	init();
 
- }
 
- CStackInstance::CStackInstance(CreatureID id, TQuantity Count, bool isHypothetic)
 
- 	: CBonusSystemNode(isHypothetic), armyObj(_armyObj)
 
- {
 
- 	init();
 
- 	setType(id);
 
- 	count = Count;
 
- }
 
- CStackInstance::CStackInstance(const CCreature *cre, TQuantity Count, bool isHypothetic)
 
- 	: CBonusSystemNode(isHypothetic), armyObj(_armyObj)
 
- {
 
- 	init();
 
- 	setType(cre);
 
- 	count = Count;
 
- }
 
- void CStackInstance::init()
 
- {
 
- 	experience = 0;
 
- 	count = 0;
 
- 	type = nullptr;
 
- 	idRand = -1;
 
- 	_armyObj = nullptr;
 
- 	setNodeType(STACK_INSTANCE);
 
- }
 
- int CStackInstance::getQuantityID() const
 
- {
 
- 	return CCreature::getQuantityID(count);
 
- }
 
- int CStackInstance::getExpRank() const
 
- {
 
- 	if (!VLC->modh->modules.STACK_EXP)
 
- 		return 0;
 
- 	int tier = type->level;
 
- 	if (vstd::iswithin(tier, 1, 7))
 
- 	{
 
- 		for (int i = (int)VLC->creh->expRanks[tier].size()-2; i >-1; --i)//sic!
 
- 		{ //exp values vary from 1st level to max exp at 11th level
 
- 			if (experience >= VLC->creh->expRanks[tier][i])
 
- 				return ++i; //faster, but confusing - 0 index mean 1st level of experience
 
- 		}
 
- 		return 0;
 
- 	}
 
- 	else //higher tier
 
- 	{
 
- 		for (int i = (int)VLC->creh->expRanks[0].size()-2; i >-1; --i)
 
- 		{
 
- 			if (experience >= VLC->creh->expRanks[0][i])
 
- 				return ++i;
 
- 		}
 
- 		return 0;
 
- 	}
 
- }
 
- int CStackInstance::getLevel() const
 
- {
 
- 	return std::max (1, (int)type->level);
 
- }
 
- si32 CStackInstance::magicResistance() const
 
- {
 
- 	si32 val = valOfBonuses(Selector::type()(Bonus::MAGIC_RESISTANCE));
 
- 	if (const CGHeroInstance * hero = dynamic_cast<const CGHeroInstance *>(_armyObj))
 
- 	{
 
- 		//resistance skill
 
- 		val += hero->valOfBonuses(Bonus::SECONDARY_SKILL_PREMY, SecondarySkill::RESISTANCE);
 
- 	}
 
- 	vstd::amin (val, 100);
 
- 	return val;
 
- }
 
- void CStackInstance::giveStackExp(TExpType exp)
 
- {
 
- 	int level = type->level;
 
- 	if (!vstd::iswithin(level, 1, 7))
 
- 		level = 0;
 
- 	CCreatureHandler * creh = VLC->creh;
 
- 	ui32 maxExp = creh->expRanks[level].back();
 
- 	vstd::amin(exp, (TExpType)maxExp); //prevent exp overflow due to different types
 
- 	vstd::amin(exp, (maxExp * creh->maxExpPerBattle[level])/100);
 
- 	vstd::amin(experience += exp, maxExp); //can't get more exp than this limit
 
- }
 
- void CStackInstance::setType(CreatureID creID)
 
- {
 
- 	if(creID >= 0 && creID < VLC->creh->objects.size())
 
- 		setType(VLC->creh->objects[creID]);
 
- 	else
 
- 		setType((const CCreature*)nullptr);
 
- }
 
- void CStackInstance::setType(const CCreature *c)
 
- {
 
- 	if(type)
 
- 	{
 
- 		detachFrom(const_cast<CCreature*>(type));
 
- 		if (type->isMyUpgrade(c) && VLC->modh->modules.STACK_EXP)
 
- 			experience = static_cast<TExpType>(experience * VLC->creh->expAfterUpgrade / 100.0);
 
- 	}
 
- 	CStackBasicDescriptor::setType(c);
 
- 	if(type)
 
- 		attachTo(const_cast<CCreature*>(type));
 
- }
 
- std::string CStackInstance::bonusToString(const std::shared_ptr<Bonus>& bonus, bool description) const
 
- {
 
- 	if(Bonus::MAGIC_RESISTANCE == bonus->type)
 
- 	{
 
- 		return "";
 
- 	}
 
- 	else
 
- 	{
 
- 		return VLC->getBth()->bonusToString(bonus, this, description);
 
- 	}
 
- }
 
- std::string CStackInstance::bonusToGraphics(const std::shared_ptr<Bonus>& bonus) const
 
- {
 
- 	return VLC->getBth()->bonusToGraphics(bonus);
 
- }
 
- void CStackInstance::setArmyObj(const CArmedInstance * ArmyObj)
 
- {
 
- 	if(_armyObj)
 
- 		detachFrom(const_cast<CArmedInstance*>(_armyObj));
 
- 	_armyObj = ArmyObj;
 
- 	if(ArmyObj)
 
- 		attachTo(const_cast<CArmedInstance*>(_armyObj));
 
- }
 
- std::string CStackInstance::getQuantityTXT(bool capitalized) const
 
- {
 
- 	int quantity = getQuantityID();
 
- 	if (quantity)
 
- 		return VLC->generaltexth->arraytxt[174 + quantity*3 - 1 - capitalized];
 
- 	else
 
- 		return "";
 
- }
 
- bool CStackInstance::valid(bool allowUnrandomized) const
 
- {
 
- 	bool isRand = (idRand != -1);
 
- 	if(!isRand)
 
- 	{
 
- 		return (type  &&  type == VLC->creh->objects[type->idNumber]);
 
- 	}
 
- 	else
 
- 		return allowUnrandomized;
 
- }
 
- CStackInstance::~CStackInstance()
 
- {
 
- }
 
- std::string CStackInstance::nodeName() const
 
- {
 
- 	std::ostringstream oss;
 
- 	oss << "Stack of " << count << " of ";
 
- 	if(type)
 
- 		oss << type->namePl;
 
- 	else if(idRand >= 0)
 
- 		oss << "[no type, idRand=" << idRand << "]";
 
- 	else
 
- 		oss << "[UNDEFINED TYPE]";
 
- 	return oss.str();
 
- }
 
- PlayerColor CStackInstance::getOwner() const
 
- {
 
- 	return _armyObj ? _armyObj->getOwner() : PlayerColor::NEUTRAL;
 
- }
 
- void CStackInstance::deserializationFix()
 
- {
 
- 	const CCreature *backup = type;
 
- 	type = nullptr;
 
- 		setType(backup);
 
- 	const CArmedInstance *armyBackup = _armyObj;
 
- 	_armyObj = nullptr;
 
- 	setArmyObj(armyBackup);
 
- 	artDeserializationFix(this);
 
- }
 
- CreatureID CStackInstance::getCreatureID() const
 
- {
 
- 	if(type)
 
- 		return type->idNumber;
 
- 	else
 
- 		return CreatureID::NONE;
 
- }
 
- std::string CStackInstance::getName() const
 
- {
 
- 	return (count > 1) ? type->namePl : type->nameSing;
 
- }
 
- ui64 CStackInstance::getPower() const
 
- {
 
- 	assert(type);
 
- 	return type->AIValue * count;
 
- }
 
- ArtBearer::ArtBearer CStackInstance::bearerType() const
 
- {
 
- 	return ArtBearer::CREATURE;
 
- }
 
- void CStackInstance::putArtifact(ArtifactPosition pos, CArtifactInstance * art)
 
- {
 
- 	assert(!getArt(pos));
 
- 	art->putAt(ArtifactLocation(this, pos));
 
- }
 
- void CStackInstance::serializeJson(JsonSerializeFormat & handler)
 
- {
 
- 	//todo: artifacts
 
- 	CStackBasicDescriptor::serializeJson(handler);//must be first
 
- 	if(handler.saving)
 
- 	{
 
- 		if(idRand > -1)
 
- 		{
 
- 			int level = (int)idRand / 2;
 
- 			boost::logic::tribool upgraded = (idRand % 2) > 0;
 
- 			handler.serializeInt("level", level, 0);
 
- 			handler.serializeBool("upgraded", upgraded);
 
- 		}
 
- 	}
 
- 	else
 
- 	{
 
- 		//type set by CStackBasicDescriptor::serializeJson
 
- 		if(type == nullptr)
 
- 		{
 
- 			int level = 0;
 
- 			bool upgraded = false;
 
- 			handler.serializeInt("level", level, 0);
 
- 			handler.serializeBool("upgraded", upgraded);
 
- 			idRand = level * 2 + (int)(bool)upgraded;
 
- 		}
 
- 	}
 
- }
 
- CCommanderInstance::CCommanderInstance()
 
- {
 
- 	init();
 
- 	name = "Unnamed";
 
- }
 
- CCommanderInstance::CCommanderInstance (CreatureID id)
 
- {
 
- 	init();
 
- 	setType(id);
 
- 	name = "Commando"; //TODO - parse them
 
- }
 
- void CCommanderInstance::init()
 
- {
 
- 	alive = true;
 
- 	experience = 0;
 
- 	level = 1;
 
- 	count = 1;
 
- 	type = nullptr;
 
- 	idRand = -1;
 
- 	_armyObj = nullptr;
 
- 	setNodeType (CBonusSystemNode::COMMANDER);
 
- 	secondarySkills.resize (ECommander::SPELL_POWER + 1);
 
- }
 
- CCommanderInstance::~CCommanderInstance()
 
- {
 
- }
 
- void CCommanderInstance::setAlive (bool Alive)
 
- {
 
- 	//TODO: helm of immortality
 
- 	alive = Alive;
 
- 	if (!alive)
 
- 	{
 
- 		removeBonusesRecursive(Bonus::UntilCommanderKilled);
 
- 	}
 
- }
 
- void CCommanderInstance::giveStackExp (TExpType exp)
 
- {
 
- 	if (alive)
 
- 		experience += exp;
 
- }
 
- int CCommanderInstance::getExpRank() const
 
- {
 
- 	return VLC->heroh->level (experience);
 
- }
 
- int CCommanderInstance::getLevel() const
 
- {
 
- 	return std::max (1, getExpRank());
 
- }
 
- void CCommanderInstance::levelUp ()
 
- {
 
- 	level++;
 
- 	for (auto bonus : VLC->creh->commanderLevelPremy)
 
- 	{ //grant all regular level-up bonuses
 
- 		accumulateBonus(bonus);
 
- 	}
 
- }
 
- ArtBearer::ArtBearer CCommanderInstance::bearerType() const
 
- {
 
- 	return ArtBearer::COMMANDER;
 
- }
 
- bool CCommanderInstance::gainsLevel() const
 
- {
 
- 	return experience >= (TExpType)VLC->heroh->reqExp(level+1);
 
- }
 
- CStackBasicDescriptor::CStackBasicDescriptor()
 
- {
 
- 	type = nullptr;
 
- 	count = -1;
 
- }
 
- CStackBasicDescriptor::CStackBasicDescriptor(CreatureID id, TQuantity Count)
 
- 	: type (VLC->creh->objects[id]), count(Count)
 
- {
 
- }
 
- CStackBasicDescriptor::CStackBasicDescriptor(const CCreature *c, TQuantity Count)
 
- 	: type(c), count(Count)
 
- {
 
- }
 
- const Creature * CStackBasicDescriptor::getType() const
 
- {
 
- 	return type;
 
- }
 
- TQuantity CStackBasicDescriptor::getCount() const
 
- {
 
- 	return count;
 
- }
 
- void CStackBasicDescriptor::setType(const CCreature * c)
 
- {
 
- 	type = c;
 
- }
 
- void CStackBasicDescriptor::serializeJson(JsonSerializeFormat & handler)
 
- {
 
- 	handler.serializeInt("amount", count);
 
- 	if(handler.saving)
 
- 	{
 
- 		if(type)
 
- 		{
 
- 			std::string typeName = type->identifier;
 
- 			handler.serializeString("type", typeName);
 
- 		}
 
- 	}
 
- 	else
 
- 	{
 
- 		std::string typeName("");
 
- 		handler.serializeString("type", typeName);
 
- 		if(typeName != "")
 
- 			setType(VLC->creh->getCreature("core", typeName));
 
- 	}
 
- }
 
- void CSimpleArmy::clear()
 
- {
 
- 	army.clear();
 
- }
 
- CSimpleArmy::operator bool() const
 
- {
 
- 	return army.size();
 
- }
 
- bool CSimpleArmy::setCreature(SlotID slot, CreatureID cre, TQuantity count)
 
- {
 
- 	assert(!vstd::contains(army, slot));
 
- 	army[slot] = std::make_pair(cre, count);
 
- 	return true;
 
- }
 
 
  |