12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208 |
- /*
- * CArtHandler.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 "CArtHandler.h"
- #include "filesystem/Filesystem.h"
- #include "CGeneralTextHandler.h"
- #include "VCMI_Lib.h"
- #include "CModHandler.h"
- #include "CSpellHandler.h"
- #include "mapObjects/MapObjects.h"
- #include "NetPacksBase.h"
- #include "GameConstants.h"
- #include "CRandomGenerator.h"
- #include "mapObjects/CObjectClassesHandler.h"
- using namespace boost::assign;
- // Note: list must match entries in ArtTraits.txt
- #define ART_POS_LIST \
- ART_POS(SPELLBOOK) \
- ART_POS(MACH4) \
- ART_POS(MACH3) \
- ART_POS(MACH2) \
- ART_POS(MACH1) \
- ART_POS(MISC5) \
- ART_POS(MISC4) \
- ART_POS(MISC3) \
- ART_POS(MISC2) \
- ART_POS(MISC1) \
- ART_POS(FEET) \
- ART_POS(LEFT_RING) \
- ART_POS(RIGHT_RING) \
- ART_POS(TORSO) \
- ART_POS(LEFT_HAND) \
- ART_POS(RIGHT_HAND) \
- ART_POS(NECK) \
- ART_POS(SHOULDERS) \
- ART_POS(HEAD)
- const std::string & CArtifact::Name() const
- {
- return name;
- }
- const std::string & CArtifact::Description() const
- {
- return description;
- }
- const std::string & CArtifact::EventText() const
- {
- return eventText;
- }
- bool CArtifact::isBig () const
- {
- return VLC->arth->isBigArtifact(id);
- }
- CArtifact::CArtifact()
- {
- setNodeType(ARTIFACT);
- possibleSlots[ArtBearer::HERO]; //we want to generate map entry even if it will be empty
- possibleSlots[ArtBearer::CREATURE]; //we want to generate map entry even if it will be empty
- possibleSlots[ArtBearer::COMMANDER];
- }
- CArtifact::~CArtifact()
- {
- }
- int CArtifact::getArtClassSerial() const
- {
- if(id == 1)
- return 4;
- switch(aClass)
- {
- case ART_TREASURE:
- return 0;
- case ART_MINOR:
- return 1;
- case ART_MAJOR:
- return 2;
- case ART_RELIC:
- return 3;
- case ART_SPECIAL:
- return 5;
- }
- return -1;
- }
- std::string CArtifact::nodeName() const
- {
- return "Artifact: " + Name();
- }
- void CArtifact::addNewBonus(Bonus *b)
- {
- b->source = Bonus::ARTIFACT;
- b->duration = Bonus::PERMANENT;
- b->description = name;
- CBonusSystemNode::addNewBonus(b);
- }
- void CGrowingArtifact::levelUpArtifact (CArtifactInstance * art)
- {
- Bonus b;
- b.type = Bonus::LEVEL_COUNTER;
- b.val = 1;
- b.duration = Bonus::COMMANDER_KILLED;
- art->accumulateBonus (b);
- for (auto bonus : bonusesPerLevel)
- {
- if (art->valOfBonuses(Bonus::LEVEL_COUNTER) % bonus.first == 0) //every n levels
- {
- art->accumulateBonus (bonus.second);
- }
- }
- for (auto bonus : thresholdBonuses)
- {
- if (art->valOfBonuses(Bonus::LEVEL_COUNTER) == bonus.first) //every n levels
- {
- art->addNewBonus (&bonus.second);
- }
- }
- }
- CArtHandler::CArtHandler()
- {
- //VLC->arth = this;
- // War machines are the default big artifacts.
- for (ArtifactID i = ArtifactID::CATAPULT; i <= ArtifactID::FIRST_AID_TENT; i.advance(1))
- bigArtifacts.insert(i);
- }
- CArtHandler::~CArtHandler()
- {
- for(CArtifact * art : artifacts)
- delete art;
- }
- std::vector<JsonNode> CArtHandler::loadLegacyData(size_t dataSize)
- {
- artifacts.resize(dataSize);
- std::vector<JsonNode> h3Data;
- h3Data.reserve(dataSize);
- #define ART_POS(x) #x ,
- const std::vector<std::string> artSlots = { ART_POS_LIST };
- #undef ART_POS
- static std::map<char, std::string> classes =
- {{'S',"SPECIAL"}, {'T',"TREASURE"},{'N',"MINOR"},{'J',"MAJOR"},{'R',"RELIC"},};
- CLegacyConfigParser parser("DATA/ARTRAITS.TXT");
- CLegacyConfigParser events("DATA/ARTEVENT.TXT");
- parser.endLine(); // header
- parser.endLine();
- for (size_t i = 0; i < dataSize; i++)
- {
- JsonNode artData;
- artData["text"]["name"].String() = parser.readString();
- artData["text"]["event"].String() = events.readString();
- artData["value"].Float() = parser.readNumber();
- for(auto & artSlot : artSlots)
- {
- if(parser.readString() == "x")
- {
- artData["slot"].Vector().push_back(JsonNode());
- artData["slot"].Vector().back().String() = artSlot;
- }
- }
- artData["class"].String() = classes[parser.readString()[0]];
- artData["text"]["description"].String() = parser.readString();
- parser.endLine();
- events.endLine();
- h3Data.push_back(artData);
- }
- return h3Data;
- }
- void CArtHandler::loadObject(std::string scope, std::string name, const JsonNode & data)
- {
- auto object = loadFromJson(data);
- object->id = ArtifactID(artifacts.size());
- object->iconIndex = object->id + 5;
- artifacts.push_back(object);
- VLC->modh->identifiers.registerObject(scope, "artifact", name, object->id);
- }
- void CArtHandler::loadObject(std::string scope, std::string name, const JsonNode & data, size_t index)
- {
- auto object = loadFromJson(data);
- object->id = ArtifactID(index);
- object->iconIndex = object->id;
- assert(artifacts[index] == nullptr); // ensure that this id was not loaded before
- artifacts[index] = object;
- VLC->modh->identifiers.registerObject(scope, "artifact", name, object->id);
- }
- CArtifact * CArtHandler::loadFromJson(const JsonNode & node)
- {
- CArtifact * art;
- if (!VLC->modh->modules.COMMANDERS || node["growing"].isNull())
- art = new CArtifact();
- else
- {
- auto growing = new CGrowingArtifact();
- loadGrowingArt(growing, node);
- art = growing;
- }
- const JsonNode & text = node["text"];
- art->name = text["name"].String();
- art->description = text["description"].String();
- art->eventText = text["event"].String();
- const JsonNode & graphics = node["graphics"];
- art->image = graphics["image"].String();
- if (!graphics["large"].isNull())
- art->large = graphics["large"].String();
- else
- art->large = art->image;
- art->advMapDef = graphics["map"].String();
- art->price = node["value"].Float();
- loadSlots(art, node);
- loadClass(art, node);
- loadType(art, node);
- loadComponents(art, node);
- for (auto b : node["bonuses"].Vector())
- {
- auto bonus = JsonUtils::parseBonus (b);
- art->addNewBonus(bonus);
- }
- return art;
- }
- ArtifactPosition CArtHandler::stringToSlot(std::string slotName)
- {
- #define ART_POS(x) { #x, ArtifactPosition::x },
- static const std::map<std::string, ArtifactPosition> artifactPositionMap = { ART_POS_LIST };
- #undef ART_POS
- auto it = artifactPositionMap.find (slotName);
- if (it != artifactPositionMap.end())
- return it->second;
- logGlobal->warnStream() << "Warning! Artifact slot " << slotName << " not recognized!";
- return ArtifactPosition::PRE_FIRST;
- }
- void CArtHandler::addSlot(CArtifact * art, const std::string & slotID)
- {
- if (slotID == "MISC")
- {
- art->possibleSlots[ArtBearer::HERO] += ArtifactPosition::MISC1, ArtifactPosition::MISC2, ArtifactPosition::MISC3, ArtifactPosition::MISC4, ArtifactPosition::MISC5;
- }
- else if (slotID == "RING")
- {
- art->possibleSlots[ArtBearer::HERO] += ArtifactPosition::LEFT_RING, ArtifactPosition::RIGHT_RING;
- }
- else
- {
- auto slot = stringToSlot(slotID);
- if (slot != ArtifactPosition::PRE_FIRST)
- art->possibleSlots[ArtBearer::HERO].push_back (slot);
- }
- }
- void CArtHandler::loadSlots(CArtifact * art, const JsonNode & node)
- {
- if (!node["slot"].isNull()) //we assume non-hero slots are irrelevant?
- {
- if (node["slot"].getType() == JsonNode::DATA_STRING)
- addSlot(art, node["slot"].String());
- else
- {
- for (const JsonNode & slot : node["slot"].Vector())
- addSlot(art, slot.String());
- }
- }
- }
- CArtifact::EartClass CArtHandler::stringToClass(std::string className)
- {
- static const std::map<std::string, CArtifact::EartClass> artifactClassMap =
- {
- {"TREASURE", CArtifact::ART_TREASURE},
- {"MINOR", CArtifact::ART_MINOR},
- {"MAJOR", CArtifact::ART_MAJOR},
- {"RELIC", CArtifact::ART_RELIC},
- {"SPECIAL", CArtifact::ART_SPECIAL}
- };
- auto it = artifactClassMap.find (className);
- if (it != artifactClassMap.end())
- return it->second;
- logGlobal->warnStream() << "Warning! Artifact rarity " << className << " not recognized!";
- return CArtifact::ART_SPECIAL;
- }
- void CArtHandler::loadClass(CArtifact * art, const JsonNode & node)
- {
- art->aClass = stringToClass(node["class"].String());
- }
- void CArtHandler::loadType(CArtifact * art, const JsonNode & node)
- {
- #define ART_BEARER(x) { #x, ArtBearer::x },
- static const std::map<std::string, int> artifactBearerMap = { ART_BEARER_LIST };
- #undef ART_BEARER
- for (const JsonNode & b : node["type"].Vector())
- {
- auto it = artifactBearerMap.find (b.String());
- if (it != artifactBearerMap.end())
- {
- int bearerType = it->second;
- switch (bearerType)
- {
- case ArtBearer::HERO://TODO: allow arts having several possible bearers
- break;
- case ArtBearer::COMMANDER:
- makeItCommanderArt (art); //original artifacts should have only one bearer type
- break;
- case ArtBearer::CREATURE:
- makeItCreatureArt (art);
- break;
- }
- }
- else
- logGlobal->warnStream() << "Warning! Artifact type " << b.String() << " not recognized!";
- }
- }
- void CArtHandler::loadComponents(CArtifact * art, const JsonNode & node)
- {
- if (!node["components"].isNull())
- {
- art->constituents.reset(new std::vector<CArtifact *>());
- for (auto component : node["components"].Vector())
- {
- VLC->modh->identifiers.requestIdentifier("artifact", component, [=](si32 id)
- {
- // when this code is called both combinational art as well as component are loaded
- // so it is safe to access any of them
- art->constituents->push_back(VLC->arth->artifacts[id]);
- VLC->arth->artifacts[id]->constituentOf.push_back(art);
- });
- }
- }
- }
- void CArtHandler::loadGrowingArt(CGrowingArtifact * art, const JsonNode & node)
- {
- for (auto b : node["growing"]["bonusesPerLevel"].Vector())
- {
- art->bonusesPerLevel.push_back (std::pair <ui16, Bonus> (b["level"].Float(), *JsonUtils::parseBonus (b["bonus"])));
- }
- for (auto b : node["growing"]["thresholdBonuses"].Vector())
- {
- art->thresholdBonuses.push_back (std::pair <ui16, Bonus> (b["level"].Float(), *JsonUtils::parseBonus (b["bonus"])));
- }
- }
- //TODO: use bimap
- ArtifactID CArtHandler::creatureToMachineID(CreatureID id)
- {
- switch (id)
- {
- case CreatureID::CATAPULT: //Catapult
- return ArtifactID::CATAPULT;
- break;
- case CreatureID::BALLISTA: //Ballista
- return ArtifactID::BALLISTA;
- break;
- case CreatureID::FIRST_AID_TENT: //First Aid tent
- return ArtifactID::FIRST_AID_TENT;
- break;
- case CreatureID::AMMO_CART: //Ammo cart
- return ArtifactID::AMMO_CART;
- break;
- }
- return ArtifactID::NONE; //this creature is not artifact
- }
- CreatureID CArtHandler::machineIDToCreature(ArtifactID id)
- {
- switch (id)
- {
- case ArtifactID::CATAPULT:
- return CreatureID::CATAPULT;
- break;
- case ArtifactID::BALLISTA:
- return CreatureID::BALLISTA;
- break;
- case ArtifactID::FIRST_AID_TENT:
- return CreatureID::FIRST_AID_TENT;
- break;
- case ArtifactID::AMMO_CART:
- return CreatureID::AMMO_CART;
- break;
- }
- return CreatureID::NONE; //this artifact is not a creature
- }
- ArtifactID CArtHandler::pickRandomArtifact(CRandomGenerator & rand, int flags, std::function<bool(ArtifactID)> accepts)
- {
- auto getAllowedArts = [&](std::vector<ConstTransitivePtr<CArtifact> > &out, std::vector<CArtifact*> *arts, CArtifact::EartClass flag)
- {
- if (arts->empty()) //restock available arts
- fillList(*arts, flag);
- for (auto & arts_i : *arts)
- {
- if (accepts(arts_i->id))
- {
- CArtifact *art = arts_i;
- out.push_back(art);
- }
- }
- };
- auto getAllowed = [&](std::vector<ConstTransitivePtr<CArtifact> > &out)
- {
- if (flags & CArtifact::ART_TREASURE)
- getAllowedArts (out, &treasures, CArtifact::ART_TREASURE);
- if (flags & CArtifact::ART_MINOR)
- getAllowedArts (out, &minors, CArtifact::ART_MINOR);
- if (flags & CArtifact::ART_MAJOR)
- getAllowedArts (out, &majors, CArtifact::ART_MAJOR);
- if (flags & CArtifact::ART_RELIC)
- getAllowedArts (out, &relics, CArtifact::ART_RELIC);
- if (!out.size()) //no artifact of specified rarity, we need to take another one
- {
- getAllowedArts (out, &treasures, CArtifact::ART_TREASURE);
- getAllowedArts (out, &minors, CArtifact::ART_MINOR);
- getAllowedArts (out, &majors, CArtifact::ART_MAJOR);
- getAllowedArts (out, &relics, CArtifact::ART_RELIC);
- }
- if (!out.size()) //no arts are available at all
- {
- out.resize (64);
- std::fill_n (out.begin(), 64, artifacts[2]); //Give Grail - this can't be banned (hopefully)
- }
- };
- std::vector<ConstTransitivePtr<CArtifact> > out;
- getAllowed(out);
- ArtifactID artID = (*RandomGeneratorUtil::nextItem(out, rand))->id;
- erasePickedArt(artID);
- return artID;
- }
- ArtifactID CArtHandler::pickRandomArtifact(CRandomGenerator & rand, std::function<bool(ArtifactID)> accepts)
- {
- return pickRandomArtifact(rand, 0xff, accepts);
- }
- ArtifactID CArtHandler::pickRandomArtifact(CRandomGenerator & rand, int flags)
- {
- return pickRandomArtifact(rand, flags, [](ArtifactID){ return true;});
- }
- Bonus *createBonus(Bonus::BonusType type, int val, int subtype, Bonus::ValueType valType, shared_ptr<ILimiter> limiter = shared_ptr<ILimiter>(), int additionalInfo = 0)
- {
- auto added = new Bonus(Bonus::PERMANENT,type,Bonus::ARTIFACT,val,-1,subtype);
- added->additionalInfo = additionalInfo;
- added->valType = valType;
- added->limiter = limiter;
- return added;
- }
- Bonus *createBonus(Bonus::BonusType type, int val, int subtype, shared_ptr<IPropagator> propagator = shared_ptr<IPropagator>(), int additionalInfo = 0)
- {
- auto added = new Bonus(Bonus::PERMANENT,type,Bonus::ARTIFACT,val,-1,subtype);
- added->additionalInfo = additionalInfo;
- added->valType = Bonus::BASE_NUMBER;
- added->propagator = propagator;
- return added;
- }
- void CArtHandler::giveArtBonus( ArtifactID aid, Bonus::BonusType type, int val, int subtype, Bonus::ValueType valType, shared_ptr<ILimiter> limiter, int additionalInfo)
- {
- giveArtBonus(aid, createBonus(type, val, subtype, valType, limiter, additionalInfo));
- }
- void CArtHandler::giveArtBonus(ArtifactID aid, Bonus::BonusType type, int val, int subtype, shared_ptr<IPropagator> propagator /*= nullptr*/, int additionalInfo)
- {
- giveArtBonus(aid, createBonus(type, val, subtype, propagator, additionalInfo));
- }
- void CArtHandler::giveArtBonus(ArtifactID aid, Bonus *bonus)
- {
- bonus->sid = aid;
- if(bonus->subtype == Bonus::MORALE || bonus->type == Bonus::LUCK)
- bonus->description = artifacts[aid]->Name() + (bonus->val > 0 ? " +" : " ") + boost::lexical_cast<std::string>(bonus->val);
- else
- bonus->description = artifacts[aid]->Name();
- artifacts[aid]->addNewBonus(bonus);
- }
- void CArtHandler::makeItCreatureArt (CArtifact * a, bool onlyCreature /*=true*/)
- {
- if (onlyCreature)
- {
- a->possibleSlots[ArtBearer::HERO].clear();
- a->possibleSlots[ArtBearer::COMMANDER].clear();
- }
- a->possibleSlots[ArtBearer::CREATURE].push_back(ArtifactPosition::CREATURE_SLOT);
- }
- void CArtHandler::makeItCreatureArt (ArtifactID aid, bool onlyCreature /*=true*/)
- {
- CArtifact *a = artifacts[aid];
- makeItCreatureArt (a, onlyCreature);
- }
- void CArtHandler::makeItCommanderArt (CArtifact * a, bool onlyCommander /*= true*/ )
- {
- if (onlyCommander)
- {
- a->possibleSlots[ArtBearer::HERO].clear();
- a->possibleSlots[ArtBearer::CREATURE].clear();
- }
- for (int i = ArtifactPosition::COMMANDER1; i <= ArtifactPosition::COMMANDER6; ++i)
- a->possibleSlots[ArtBearer::COMMANDER].push_back(ArtifactPosition(i));
- }
- void CArtHandler::makeItCommanderArt( ArtifactID aid, bool onlyCommander /*= true*/ )
- {
- CArtifact *a = artifacts[aid];
- makeItCommanderArt (a, onlyCommander);
- }
- bool CArtHandler::legalArtifact(ArtifactID id)
- {
- auto art = artifacts[id];
- //assert ( (!art->constituents) || art->constituents->size() ); //artifacts is not combined or has some components
- return (art->possibleSlots[ArtBearer::HERO].size() ||
- (art->possibleSlots[ArtBearer::COMMANDER].size() && VLC->modh->modules.COMMANDERS) ||
- (art->possibleSlots[ArtBearer::CREATURE].size() && VLC->modh->modules.STACK_ARTIFACT)) &&
- !(art->constituents); //no combo artifacts spawning
- }
- void CArtHandler::initAllowedArtifactsList(const std::vector<bool> &allowed)
- {
- allowedArtifacts.clear();
- treasures.clear();
- minors.clear();
- majors.clear();
- relics.clear();
- for (ArtifactID i=ArtifactID::SPELLBOOK; i<ArtifactID::ART_SELECTION; i.advance(1))
- {
- if (allowed[i] && legalArtifact(i))
- allowedArtifacts.push_back(artifacts[i]);
- }
- if (VLC->modh->modules.COMMANDERS) //allow all commander artifacts for testing
- {
- for (int i = 146; i <= 155; ++i)
- {
- allowedArtifacts.push_back(artifacts[i]);
- }
- }
- for (int i = GameConstants::ARTIFACTS_QUANTITY; i < artifacts.size(); ++i) //allow all new artifacts by default
- {
- if (legalArtifact(ArtifactID(i)))
- allowedArtifacts.push_back(artifacts[i]);
- //keep im mind that artifact can be worn by more than one type of bearer
- }
- }
- std::vector<bool> CArtHandler::getDefaultAllowed() const
- {
- std::vector<bool> allowedArtifacts;
- allowedArtifacts.resize(127, true);
- allowedArtifacts.resize(141, false);
- allowedArtifacts.resize(GameConstants::ARTIFACTS_QUANTITY, true);
- return allowedArtifacts;
- }
- void CArtHandler::erasePickedArt(ArtifactID id)
- {
- CArtifact *art = artifacts[id];
- if(auto artifactList = listFromClass(art->aClass))
- {
- if(artifactList->empty())
- fillList(*artifactList, art->aClass);
- auto itr = vstd::find(*artifactList, art);
- if(itr != artifactList->end())
- {
- artifactList->erase(itr);
- }
- else
- logGlobal->warnStream() << "Problem: cannot erase artifact " << art->Name() << " from list, it was not present";
- }
- else
- logGlobal->warnStream() << "Problem: cannot find list for artifact " << art->Name() << ", strange class. (special?)";
- }
- boost::optional<std::vector<CArtifact*>&> CArtHandler::listFromClass( CArtifact::EartClass artifactClass )
- {
- switch(artifactClass)
- {
- case CArtifact::ART_TREASURE:
- return treasures;
- case CArtifact::ART_MINOR:
- return minors;
- case CArtifact::ART_MAJOR:
- return majors;
- case CArtifact::ART_RELIC:
- return relics;
- default: //special artifacts should not be erased
- return nullptr;
- }
- }
- void CArtHandler::fillList( std::vector<CArtifact*> &listToBeFilled, CArtifact::EartClass artifactClass )
- {
- assert(listToBeFilled.empty());
- for (auto & elem : allowedArtifacts)
- {
- if (elem->aClass == artifactClass)
- listToBeFilled.push_back(elem);
- }
- }
- void CArtHandler::afterLoadFinalization()
- {
- //All artifacts have their id, so we can properly update their bonuses' source ids.
- for(auto &art : artifacts)
- {
- for(auto &bonus : art->getExportedBonusList())
- {
- assert(art == artifacts[art->id]);
- assert(bonus->source == Bonus::ARTIFACT);
- bonus->sid = art->id;
- }
- }
- for (CArtifact * art : artifacts)
- {
- VLC->objtypeh->loadSubObject(art->Name(), JsonNode(), Obj::ARTIFACT, art->id.num);
- if (!art->advMapDef.empty())
- {
- JsonNode templ;
- templ["animation"].String() = art->advMapDef;
- // add new template.
- // Necessary for objects added via mods that don't have any templates in H3
- VLC->objtypeh->getHandlerFor(Obj::ARTIFACT, art->id)->addTemplate(templ);
- }
- // object does not have any templates - this is not usable object (e.g. pseudo-art like lock)
- if (VLC->objtypeh->getHandlerFor(Obj::ARTIFACT, art->id)->getTemplates().empty())
- VLC->objtypeh->removeSubObject(Obj::ARTIFACT, art->id);
- }
- }
- CArtifactInstance::CArtifactInstance()
- {
- init();
- }
- CArtifactInstance::CArtifactInstance( CArtifact *Art)
- {
- init();
- setType(Art);
- }
- void CArtifactInstance::setType( CArtifact *Art )
- {
- artType = Art;
- attachTo(Art);
- }
- std::string CArtifactInstance::nodeName() const
- {
- return "Artifact instance of " + (artType ? artType->Name() : std::string("uninitialized")) + " type";
- }
- CArtifactInstance * CArtifactInstance::createScroll( const CSpell *s)
- {
- auto ret = new CArtifactInstance(VLC->arth->artifacts[1]);
- auto b = new Bonus(Bonus::PERMANENT, Bonus::SPELL, Bonus::ARTIFACT_INSTANCE, -1, 1, s->id);
- ret->addNewBonus(b);
- return ret;
- }
- void CArtifactInstance::init()
- {
- id = ArtifactInstanceID();
- id = static_cast<ArtifactInstanceID>(ArtifactID::NONE); //to be randomized
- setNodeType(ARTIFACT_INSTANCE);
- }
- ArtifactPosition CArtifactInstance::firstAvailableSlot(const CArtifactSet *h) const
- {
- for(auto slot : artType->possibleSlots.at(h->bearerType()))
- {
- if(canBePutAt(h, slot)) //if(artType->fitsAt(h->artifWorn, slot))
- {
- //we've found a free suitable slot.
- return slot;
- }
- }
- //if haven't find proper slot, use backpack
- return firstBackpackSlot(h);
- }
- ArtifactPosition CArtifactInstance::firstBackpackSlot(const CArtifactSet *h) const
- {
- if(!artType->isBig()) //discard big artifact
- return ArtifactPosition(
- GameConstants::BACKPACK_START + h->artifactsInBackpack.size());
- return ArtifactPosition::PRE_FIRST;
- }
- bool CArtifactInstance::canBePutAt(const ArtifactLocation & al, bool assumeDestRemoved /*= false*/) const
- {
- return canBePutAt(al.getHolderArtSet(), al.slot, assumeDestRemoved);
- }
- bool CArtifactInstance::canBePutAt(const CArtifactSet *artSet, ArtifactPosition slot, bool assumeDestRemoved /*= false*/) const
- {
- if(slot >= GameConstants::BACKPACK_START)
- {
- if(artType->isBig())
- return false;
- //TODO backpack limit
- return true;
- }
- auto possibleSlots = artType->possibleSlots.find(artSet->bearerType());
- if(possibleSlots == artType->possibleSlots.end())
- {
- logGlobal->warnStream() << "Warning: artifact " << artType->Name() << " doesn't have defined allowed slots for bearer of type "
- << artSet->bearerType();
- return false;
- }
- if(!vstd::contains(possibleSlots->second, slot))
- return false;
- return artSet->isPositionFree(slot, assumeDestRemoved);
- }
- void CArtifactInstance::putAt(ArtifactLocation al)
- {
- assert(canBePutAt(al));
- al.getHolderArtSet()->setNewArtSlot(al.slot, this, false);
- if(al.slot < GameConstants::BACKPACK_START)
- al.getHolderNode()->attachTo(this);
- }
- void CArtifactInstance::removeFrom(ArtifactLocation al)
- {
- assert(al.getHolderArtSet()->getArt(al.slot) == this);
- al.getHolderArtSet()->eraseArtSlot(al.slot);
- if(al.slot < GameConstants::BACKPACK_START)
- al.getHolderNode()->detachFrom(this);
- //TODO delete me?
- }
- bool CArtifactInstance::canBeDisassembled() const
- {
- return bool(artType->constituents);
- }
- std::vector<const CArtifact *> CArtifactInstance::assemblyPossibilities(const CArtifactSet *h) const
- {
- std::vector<const CArtifact *> ret;
- if(artType->constituents) //combined artifact already: no combining of combined artifacts... for now.
- return ret;
- for(const CArtifact * artifact : artType->constituentOf)
- {
- assert(artifact->constituents);
- bool possible = true;
- for(const CArtifact * constituent : *artifact->constituents) //check if all constituents are available
- {
- if(!h->hasArt(constituent->id, true)) //constituent must be equipped
- {
- possible = false;
- break;
- }
- }
- if(possible)
- ret.push_back(artifact);
- }
- return ret;
- }
- void CArtifactInstance::move(ArtifactLocation src, ArtifactLocation dst)
- {
- removeFrom(src);
- putAt(dst);
- }
- CArtifactInstance * CArtifactInstance::createNewArtifactInstance(CArtifact *Art)
- {
- if(!Art->constituents)
- {
- auto ret = new CArtifactInstance(Art);
- if (dynamic_cast<CGrowingArtifact *>(Art))
- {
- auto bonus = new Bonus;
- bonus->type = Bonus::LEVEL_COUNTER;
- bonus->val = 0;
- ret->addNewBonus (bonus);
- }
- return ret;
- }
- else
- {
- auto ret = new CCombinedArtifactInstance(Art);
- ret->createConstituents();
- return ret;
- }
- }
- CArtifactInstance * CArtifactInstance::createNewArtifactInstance(int aid)
- {
- return createNewArtifactInstance(VLC->arth->artifacts[aid]);
- }
- void CArtifactInstance::deserializationFix()
- {
- setType(artType);
- }
- SpellID CArtifactInstance::getGivenSpellID() const
- {
- const Bonus * b = getBonusLocalFirst(Selector::type(Bonus::SPELL));
- if(!b)
- {
- logGlobal->warnStream() << "Warning: " << nodeName() << " doesn't bear any spell!";
- return SpellID::NONE;
- }
- return SpellID(b->subtype);
- }
- bool CArtifactInstance::isPart(const CArtifactInstance *supposedPart) const
- {
- return supposedPart == this;
- }
- bool CCombinedArtifactInstance::canBePutAt(const CArtifactSet *artSet, ArtifactPosition slot, bool assumeDestRemoved /*= false*/) const
- {
- bool canMainArtifactBePlaced = CArtifactInstance::canBePutAt(artSet, slot, assumeDestRemoved);
- if(!canMainArtifactBePlaced)
- return false; //no is no...
- if(slot >= GameConstants::BACKPACK_START)
- return true; //we can always remove combined art to the backapck
- assert(artType->constituents);
- std::vector<ConstituentInfo> constituentsToBePlaced = constituentsInfo; //we'll remove constituents from that list, as we find a suitable slot for them
- //it may be that we picked a combined artifact in hero screen (though technically it's still there) to move it
- //so we remove from the list all constituents that are already present on dst hero in the form of locks
- for(const ConstituentInfo &constituent : constituentsInfo)
- {
- if(constituent.art == artSet->getArt(constituent.slot, false)) //no need to worry about locked constituent
- constituentsToBePlaced -= constituent;
- }
- //we iterate over all active slots and check if constituents fits them
- for (int i = 0; i < GameConstants::BACKPACK_START; i++)
- {
- for(auto art = constituentsToBePlaced.begin(); art != constituentsToBePlaced.end(); art++)
- {
- if(art->art->canBePutAt(artSet, ArtifactPosition(i), i == slot)) // i == al.slot because we can remove already worn artifact only from that slot that is our main destination
- {
- constituentsToBePlaced.erase(art);
- break;
- }
- }
- }
- return constituentsToBePlaced.empty();
- }
- bool CCombinedArtifactInstance::canBeDisassembled() const
- {
- return true;
- }
- CCombinedArtifactInstance::CCombinedArtifactInstance(CArtifact *Art)
- : CArtifactInstance(Art) //TODO: seems unued, but need to be written
- {
- }
- CCombinedArtifactInstance::CCombinedArtifactInstance()
- {
- }
- void CCombinedArtifactInstance::createConstituents()
- {
- assert(artType);
- assert(artType->constituents);
- for(const CArtifact * art : *artType->constituents)
- {
- addAsConstituent(CArtifactInstance::createNewArtifactInstance(art->id), ArtifactPosition::PRE_FIRST);
- }
- }
- void CCombinedArtifactInstance::addAsConstituent(CArtifactInstance *art, ArtifactPosition slot)
- {
- assert(vstd::contains(*artType->constituents, art->artType.get()));
- assert(art->getParentNodes().size() == 1 && art->getParentNodes().front() == art->artType);
- constituentsInfo.push_back(ConstituentInfo(art, slot));
- attachTo(art);
- }
- void CCombinedArtifactInstance::putAt(ArtifactLocation al)
- {
- if(al.slot >= GameConstants::BACKPACK_START)
- {
- CArtifactInstance::putAt(al);
- for(ConstituentInfo &ci : constituentsInfo)
- ci.slot = ArtifactPosition::PRE_FIRST;
- }
- else
- {
- CArtifactInstance *mainConstituent = figureMainConstituent(al); //it'll be replaced with combined artifact, not a lock
- CArtifactInstance::putAt(al); //puts combined art (this)
- for(ConstituentInfo &ci : constituentsInfo)
- {
- if(ci.art != mainConstituent)
- {
- const ArtifactLocation suggestedPos(al.artHolder, ci.slot);
- const bool inActiveSlot = vstd::isbetween(ci.slot, 0, GameConstants::BACKPACK_START);
- const bool suggestedPosValid = ci.art->canBePutAt(suggestedPos);
- ArtifactPosition pos = ArtifactPosition::PRE_FIRST;
- if(inActiveSlot && suggestedPosValid) //there is a valid suggestion where to place lock
- pos = ci.slot;
- else
- ci.slot = pos = ci.art->firstAvailableSlot(al.getHolderArtSet());
- assert(pos < GameConstants::BACKPACK_START);
- al.getHolderArtSet()->setNewArtSlot(pos, ci.art, true); //sets as lock
- }
- else
- {
- ci.slot = ArtifactPosition::PRE_FIRST;
- }
- }
- }
- }
- void CCombinedArtifactInstance::removeFrom(ArtifactLocation al)
- {
- if(al.slot >= GameConstants::BACKPACK_START)
- {
- CArtifactInstance::removeFrom(al);
- }
- else
- {
- for(ConstituentInfo &ci : constituentsInfo)
- {
- if(ci.slot >= 0)
- {
- al.getHolderArtSet()->eraseArtSlot(ci.slot);
- ci.slot = ArtifactPosition::PRE_FIRST;
- }
- else
- {
- //main constituent
- CArtifactInstance::removeFrom(al);
- }
- }
- }
- }
- CArtifactInstance * CCombinedArtifactInstance::figureMainConstituent(const ArtifactLocation al)
- {
- CArtifactInstance *mainConstituent = nullptr; //it'll be replaced with combined artifact, not a lock
- for(ConstituentInfo &ci : constituentsInfo)
- if(ci.slot == al.slot)
- mainConstituent = ci.art;
- if(!mainConstituent)
- {
- for(ConstituentInfo &ci : constituentsInfo)
- {
- if(vstd::contains(ci.art->artType->possibleSlots[al.getHolderArtSet()->bearerType()], al.slot))
- {
- mainConstituent = ci.art;
- }
- }
- }
- return mainConstituent;
- }
- void CCombinedArtifactInstance::deserializationFix()
- {
- for(ConstituentInfo &ci : constituentsInfo)
- attachTo(ci.art);
- }
- bool CCombinedArtifactInstance::isPart(const CArtifactInstance *supposedPart) const
- {
- bool me = CArtifactInstance::isPart(supposedPart);
- if(me)
- return true;
- //check for constituents
- for(const ConstituentInfo &constituent : constituentsInfo)
- if(constituent.art == supposedPart)
- return true;
- return false;
- }
- CCombinedArtifactInstance::ConstituentInfo::ConstituentInfo(CArtifactInstance *Art /*= nullptr*/, ArtifactPosition Slot /*= -1*/)
- {
- art = Art;
- slot = Slot;
- }
- bool CCombinedArtifactInstance::ConstituentInfo::operator==(const ConstituentInfo &rhs) const
- {
- return art == rhs.art && slot == rhs.slot;
- }
- const CArtifactInstance* CArtifactSet::getArt(ArtifactPosition pos, bool excludeLocked /*= true*/) const
- {
- if(const ArtSlotInfo *si = getSlot(pos))
- {
- if(si->artifact && (!excludeLocked || !si->locked))
- return si->artifact;
- }
- return nullptr;
- }
- CArtifactInstance* CArtifactSet::getArt(ArtifactPosition pos, bool excludeLocked /*= true*/)
- {
- return const_cast<CArtifactInstance*>((const_cast<const CArtifactSet*>(this))->getArt(pos, excludeLocked));
- }
- ArtifactPosition CArtifactSet::getArtPos(int aid, bool onlyWorn /*= true*/) const
- {
- for(auto i = artifactsWorn.cbegin(); i != artifactsWorn.cend(); i++)
- if(i->second.artifact->artType->id == aid)
- return i->first;
- if(onlyWorn)
- return ArtifactPosition::PRE_FIRST;
- for(int i = 0; i < artifactsInBackpack.size(); i++)
- if(artifactsInBackpack[i].artifact->artType->id == aid)
- return ArtifactPosition(GameConstants::BACKPACK_START + i);
- return ArtifactPosition::PRE_FIRST;
- }
- ArtifactPosition CArtifactSet::getArtPos(const CArtifactInstance *art) const
- {
- for(auto i : artifactsWorn)
- if(i.second.artifact == art)
- return i.first;
- for(int i = 0; i < artifactsInBackpack.size(); i++)
- if(artifactsInBackpack[i].artifact == art)
- return ArtifactPosition(GameConstants::BACKPACK_START + i);
- return ArtifactPosition::PRE_FIRST;
- }
- const CArtifactInstance * CArtifactSet::getArtByInstanceId( ArtifactInstanceID artInstId ) const
- {
- for(auto i : artifactsWorn)
- if(i.second.artifact->id == artInstId)
- return i.second.artifact;
- for(auto i : artifactsInBackpack)
- if(i.artifact->id == artInstId)
- return i.artifact;
- return nullptr;
- }
- bool CArtifactSet::hasArt(ui32 aid, bool onlyWorn /*= false*/) const
- {
- return getArtPos(aid, onlyWorn) != ArtifactPosition::PRE_FIRST;
- }
- const ArtSlotInfo * CArtifactSet::getSlot(ArtifactPosition pos) const
- {
- if(vstd::contains(artifactsWorn, pos))
- return &artifactsWorn.at(pos);
- if(pos >= ArtifactPosition::AFTER_LAST )
- {
- int backpackPos = (int)pos - GameConstants::BACKPACK_START;
- if(backpackPos < 0 || backpackPos >= artifactsInBackpack.size())
- return nullptr;
- else
- return &artifactsInBackpack[backpackPos];
- }
- return nullptr;
- }
- bool CArtifactSet::isPositionFree(ArtifactPosition pos, bool onlyLockCheck /*= false*/) const
- {
- if(const ArtSlotInfo *s = getSlot(pos))
- return (onlyLockCheck || !s->artifact) && !s->locked;
- return true; //no slot means not used
- }
- si32 CArtifactSet::getArtTypeId(ArtifactPosition pos) const
- {
- const CArtifactInstance * const a = getArt(pos);
- if(!a)
- {
- logGlobal->warnStream() << (dynamic_cast<const CGHeroInstance*>(this))->name << " has no artifact at " << pos << " (getArtTypeId)";
- return -1;
- }
- return a->artType->id;
- }
- CArtifactSet::~CArtifactSet()
- {
- }
- ArtSlotInfo & CArtifactSet::retreiveNewArtSlot(ArtifactPosition slot)
- {
- assert(!vstd::contains(artifactsWorn, slot));
- ArtSlotInfo &ret = slot < GameConstants::BACKPACK_START
- ? artifactsWorn[slot]
- : *artifactsInBackpack.insert(artifactsInBackpack.begin() + (slot - GameConstants::BACKPACK_START), ArtSlotInfo());
- return ret;
- }
- void CArtifactSet::setNewArtSlot(ArtifactPosition slot, CArtifactInstance *art, bool locked)
- {
- ArtSlotInfo &asi = retreiveNewArtSlot(slot);
- asi.artifact = art;
- asi.locked = locked;
- }
- void CArtifactSet::eraseArtSlot(ArtifactPosition slot)
- {
- if(slot < GameConstants::BACKPACK_START)
- {
- artifactsWorn.erase(slot);
- }
- else
- {
- slot = ArtifactPosition(slot - GameConstants::BACKPACK_START);
- artifactsInBackpack.erase(artifactsInBackpack.begin() + slot);
- }
- }
- void CArtifactSet::artDeserializationFix(CBonusSystemNode *node)
- {
- for(auto & elem : artifactsWorn)
- if(elem.second.artifact && !elem.second.locked)
- node->attachTo(elem.second.artifact);
- }
|