|
@@ -7,7 +7,7 @@
|
|
|
boost::mutex Queries::mx;
|
|
|
|
|
|
template <typename Container>
|
|
|
-std::string formatContainer(const Container &c, std::string delimeter=", ", std::string opener="(", std::string closer=")")
|
|
|
+std::string formatContainer(const Container & c, std::string delimeter = ", ", std::string opener = "(", std::string closer=")")
|
|
|
{
|
|
|
std::string ret = opener;
|
|
|
auto itr = std::begin(c);
|
|
@@ -24,17 +24,18 @@ std::string formatContainer(const Container &c, std::string delimeter=", ", std:
|
|
|
return ret;
|
|
|
}
|
|
|
|
|
|
-std::ostream & operator<<(std::ostream &out, const CQuery &query)
|
|
|
+std::ostream & operator<<(std::ostream & out, const CQuery & query)
|
|
|
{
|
|
|
return out << query.toString();
|
|
|
}
|
|
|
|
|
|
-std::ostream & operator<<(std::ostream &out, QueryPtr query)
|
|
|
+std::ostream & operator<<(std::ostream & out, QueryPtr query)
|
|
|
{
|
|
|
return out << "[" << query.get() << "] " << query->toString();
|
|
|
}
|
|
|
|
|
|
-CQuery::CQuery(void)
|
|
|
+CQuery::CQuery(Queries * Owner):
|
|
|
+ owner(Owner)
|
|
|
{
|
|
|
boost::unique_lock<boost::mutex> l(Queries::mx);
|
|
|
|
|
@@ -53,9 +54,7 @@ CQuery::~CQuery(void)
|
|
|
void CQuery::addPlayer(PlayerColor color)
|
|
|
{
|
|
|
if(color.isValidPlayer())
|
|
|
- {
|
|
|
players.push_back(color);
|
|
|
- }
|
|
|
}
|
|
|
|
|
|
std::string CQuery::toString() const
|
|
@@ -69,31 +68,59 @@ bool CQuery::endsByPlayerAnswer() const
|
|
|
return false;
|
|
|
}
|
|
|
|
|
|
-void CQuery::onRemoval(CGameHandler *gh, PlayerColor color)
|
|
|
+void CQuery::onRemoval(PlayerColor color)
|
|
|
{
|
|
|
+
|
|
|
}
|
|
|
|
|
|
-bool CQuery::blocksPack(const CPack *pack) const
|
|
|
+bool CQuery::blocksPack(const CPack * pack) const
|
|
|
{
|
|
|
return false;
|
|
|
}
|
|
|
|
|
|
-void CQuery::notifyObjectAboutRemoval(const CObjectVisitQuery &objectVisit) const
|
|
|
+void CQuery::notifyObjectAboutRemoval(const CObjectVisitQuery & objectVisit) const
|
|
|
{
|
|
|
+
|
|
|
}
|
|
|
|
|
|
-void CQuery::onExposure(CGameHandler *gh, QueryPtr topQuery)
|
|
|
+void CQuery::onExposure(QueryPtr topQuery)
|
|
|
{
|
|
|
- gh->queries.popQuery(*this);
|
|
|
+ logGlobal->trace("Exposed query with id %d", queryID);
|
|
|
+ owner->popQuery(*this);
|
|
|
}
|
|
|
|
|
|
-void CQuery::onAdding(CGameHandler *gh, PlayerColor color)
|
|
|
+void CQuery::onAdding(PlayerColor color)
|
|
|
{
|
|
|
|
|
|
}
|
|
|
|
|
|
-CObjectVisitQuery::CObjectVisitQuery(const CGObjectInstance *Obj, const CGHeroInstance *Hero, int3 Tile)
|
|
|
- : visitedObject(Obj), visitingHero(Hero), tile(Tile), removeObjectAfterVisit(false)
|
|
|
+void CQuery::onAdded(PlayerColor color)
|
|
|
+{
|
|
|
+
|
|
|
+}
|
|
|
+
|
|
|
+void CQuery::setReply(const JsonNode & reply)
|
|
|
+{
|
|
|
+
|
|
|
+}
|
|
|
+
|
|
|
+bool CQuery::blockAllButReply(const CPack * pack) const
|
|
|
+{
|
|
|
+ //We accept only query replies from correct player
|
|
|
+ if(auto reply = dynamic_ptr_cast<QueryReply>(pack))
|
|
|
+ return !vstd::contains(players, reply->player);
|
|
|
+
|
|
|
+ return true;
|
|
|
+}
|
|
|
+
|
|
|
+CGhQuery::CGhQuery(CGameHandler * owner):
|
|
|
+ CQuery(&owner->queries), gh(owner)
|
|
|
+{
|
|
|
+
|
|
|
+}
|
|
|
+
|
|
|
+CObjectVisitQuery::CObjectVisitQuery(CGameHandler * owner, const CGObjectInstance * Obj, const CGHeroInstance * Hero, int3 Tile):
|
|
|
+ CGhQuery(owner), visitedObject(Obj), visitingHero(Hero), tile(Tile), removeObjectAfterVisit(false)
|
|
|
{
|
|
|
addPlayer(Hero->tempOwner);
|
|
|
}
|
|
@@ -105,7 +132,7 @@ bool CObjectVisitQuery::blocksPack(const CPack *pack) const
|
|
|
return true;
|
|
|
}
|
|
|
|
|
|
-void CObjectVisitQuery::onRemoval(CGameHandler *gh, PlayerColor color)
|
|
|
+void CObjectVisitQuery::onRemoval(PlayerColor color)
|
|
|
{
|
|
|
gh->objectVisitEnded(*this);
|
|
|
|
|
@@ -115,13 +142,13 @@ void CObjectVisitQuery::onRemoval(CGameHandler *gh, PlayerColor color)
|
|
|
gh->removeObject(visitedObject);
|
|
|
}
|
|
|
|
|
|
-void CObjectVisitQuery::onExposure(CGameHandler *gh, QueryPtr topQuery)
|
|
|
+void CObjectVisitQuery::onExposure(QueryPtr topQuery)
|
|
|
{
|
|
|
//Object may have been removed and deleted.
|
|
|
if(gh->isValidObject(visitedObject))
|
|
|
topQuery->notifyObjectAboutRemoval(*this);
|
|
|
|
|
|
- gh->queries.popQuery(*this);
|
|
|
+ owner->popQuery(*this);
|
|
|
}
|
|
|
|
|
|
void Queries::popQuery(PlayerColor player, QueryPtr query)
|
|
@@ -136,13 +163,11 @@ void Queries::popQuery(PlayerColor player, QueryPtr query)
|
|
|
queries[player] -= query;
|
|
|
auto nextQuery = topQuery(player);
|
|
|
|
|
|
- query->onRemoval(gh, player);
|
|
|
+ query->onRemoval(player);
|
|
|
|
|
|
//Exposure on query below happens only if removal didn't trigger any new query
|
|
|
if(nextQuery && nextQuery == topQuery(player))
|
|
|
- {
|
|
|
- nextQuery->onExposure(gh, query);
|
|
|
- }
|
|
|
+ nextQuery->onExposure(query);
|
|
|
}
|
|
|
|
|
|
void Queries::popQuery(const CQuery &query)
|
|
@@ -170,12 +195,15 @@ void Queries::addQuery(QueryPtr query)
|
|
|
{
|
|
|
for(auto player : query->players)
|
|
|
addQuery(player, query);
|
|
|
+
|
|
|
+ for(auto player : query->players)
|
|
|
+ query->onAdded(player);
|
|
|
}
|
|
|
|
|
|
void Queries::addQuery(PlayerColor player, QueryPtr query)
|
|
|
{
|
|
|
//LOG_TRACE_PARAMS(logGlobal, "player='%d', query='%s'", player.getNum() % query);
|
|
|
- query->onAdding(gh, player);
|
|
|
+ query->onAdding(player);
|
|
|
queries[player].push_back(query);
|
|
|
}
|
|
|
|
|
@@ -193,7 +221,7 @@ void Queries::popIfTop(QueryPtr query)
|
|
|
popIfTop(*query);
|
|
|
}
|
|
|
|
|
|
-void Queries::popIfTop(const CQuery &query)
|
|
|
+void Queries::popIfTop(const CQuery & query)
|
|
|
{
|
|
|
for(PlayerColor color : query.players)
|
|
|
if(topQuery(color).get() == &query)
|
|
@@ -203,8 +231,8 @@ void Queries::popIfTop(const CQuery &query)
|
|
|
std::vector<std::shared_ptr<const CQuery>> Queries::allQueries() const
|
|
|
{
|
|
|
std::vector<std::shared_ptr<const CQuery>> ret;
|
|
|
- for(auto &playerQueries : queries)
|
|
|
- for(auto &query : playerQueries.second)
|
|
|
+ for(auto & playerQueries : queries)
|
|
|
+ for(auto & query : playerQueries.second)
|
|
|
ret.push_back(query);
|
|
|
|
|
|
return ret;
|
|
@@ -214,53 +242,55 @@ std::vector<std::shared_ptr<CQuery>> Queries::allQueries()
|
|
|
{
|
|
|
//TODO code duplication with const function :(
|
|
|
std::vector<std::shared_ptr<CQuery>> ret;
|
|
|
- for(auto &playerQueries : queries)
|
|
|
- for(auto &query : playerQueries.second)
|
|
|
- ret.push_back(query);
|
|
|
+ for(auto & playerQueries : queries)
|
|
|
+ for(auto & query : playerQueries.second)
|
|
|
+ ret.push_back(query);
|
|
|
|
|
|
return ret;
|
|
|
}
|
|
|
|
|
|
-void CBattleQuery::notifyObjectAboutRemoval(const CObjectVisitQuery &objectVisit) const
|
|
|
+void CBattleQuery::notifyObjectAboutRemoval(const CObjectVisitQuery & objectVisit) const
|
|
|
{
|
|
|
assert(result);
|
|
|
objectVisit.visitedObject->battleFinished(objectVisit.visitingHero, *result);
|
|
|
}
|
|
|
|
|
|
-CBattleQuery::CBattleQuery(const BattleInfo *Bi)
|
|
|
+CBattleQuery::CBattleQuery(CGameHandler * owner, const BattleInfo * Bi):
|
|
|
+ CGhQuery(owner)
|
|
|
{
|
|
|
belligerents[0] = Bi->sides[0].armyObject;
|
|
|
belligerents[1] = Bi->sides[1].armyObject;
|
|
|
|
|
|
bi = Bi;
|
|
|
|
|
|
- for(auto &side : bi->sides)
|
|
|
+ for(auto & side : bi->sides)
|
|
|
addPlayer(side.color);
|
|
|
}
|
|
|
|
|
|
-CBattleQuery::CBattleQuery()
|
|
|
- :bi(nullptr)
|
|
|
+CBattleQuery::CBattleQuery(CGameHandler * owner):
|
|
|
+ CGhQuery(owner), bi(nullptr)
|
|
|
{
|
|
|
belligerents[0] = belligerents[1] = nullptr;
|
|
|
}
|
|
|
|
|
|
-bool CBattleQuery::blocksPack(const CPack *pack) const
|
|
|
+bool CBattleQuery::blocksPack(const CPack * pack) const
|
|
|
{
|
|
|
const char * name = typeid(*pack).name();
|
|
|
return strcmp(name, typeid(MakeAction).name()) && strcmp(name, typeid(MakeCustomAction).name());
|
|
|
}
|
|
|
|
|
|
-void CBattleQuery::onRemoval(CGameHandler *gh, PlayerColor color)
|
|
|
+void CBattleQuery::onRemoval(PlayerColor color)
|
|
|
{
|
|
|
gh->battleAfterLevelUp(*result);
|
|
|
}
|
|
|
|
|
|
-void CGarrisonDialogQuery::notifyObjectAboutRemoval(const CObjectVisitQuery &objectVisit) const
|
|
|
+void CGarrisonDialogQuery::notifyObjectAboutRemoval(const CObjectVisitQuery & objectVisit) const
|
|
|
{
|
|
|
objectVisit.visitedObject->garrisonDialogClosed(objectVisit.visitingHero);
|
|
|
}
|
|
|
|
|
|
-CGarrisonDialogQuery::CGarrisonDialogQuery(const CArmedInstance *up, const CArmedInstance *down)
|
|
|
+CGarrisonDialogQuery::CGarrisonDialogQuery(CGameHandler * owner, const CArmedInstance * up, const CArmedInstance * down):
|
|
|
+ CDialogQuery(owner)
|
|
|
{
|
|
|
exchangingArmies[0] = up;
|
|
|
exchangingArmies[1] = down;
|
|
@@ -269,18 +299,16 @@ CGarrisonDialogQuery::CGarrisonDialogQuery(const CArmedInstance *up, const CArme
|
|
|
addPlayer(down->tempOwner);
|
|
|
}
|
|
|
|
|
|
-bool CGarrisonDialogQuery::blocksPack(const CPack *pack) const
|
|
|
+bool CGarrisonDialogQuery::blocksPack(const CPack * pack) const
|
|
|
{
|
|
|
std::set<ObjectInstanceID> ourIds;
|
|
|
ourIds.insert(this->exchangingArmies[0]->id);
|
|
|
ourIds.insert(this->exchangingArmies[1]->id);
|
|
|
|
|
|
- if (auto stacks = dynamic_ptr_cast<ArrangeStacks>(pack))
|
|
|
- {
|
|
|
+ if(auto stacks = dynamic_ptr_cast<ArrangeStacks>(pack))
|
|
|
return !vstd::contains(ourIds, stacks->id1) || !vstd::contains(ourIds, stacks->id2);
|
|
|
- }
|
|
|
|
|
|
- if (auto arts = dynamic_ptr_cast<ExchangeArtifacts>(pack))
|
|
|
+ if(auto arts = dynamic_ptr_cast<ExchangeArtifacts>(pack))
|
|
|
{
|
|
|
if(auto id1 = boost::apply_visitor(GetEngagedHeroIds(), arts->src.artHolder))
|
|
|
if(!vstd::contains(ourIds, *id1))
|
|
@@ -291,36 +319,32 @@ bool CGarrisonDialogQuery::blocksPack(const CPack *pack) const
|
|
|
return true;
|
|
|
return false;
|
|
|
}
|
|
|
- if (auto dismiss = dynamic_ptr_cast<DisbandCreature>(pack))
|
|
|
- {
|
|
|
+ if(auto dismiss = dynamic_ptr_cast<DisbandCreature>(pack))
|
|
|
return !vstd::contains(ourIds, dismiss->id);
|
|
|
- }
|
|
|
|
|
|
- if (auto dismiss = dynamic_ptr_cast<AssembleArtifacts>(pack))
|
|
|
- {
|
|
|
+ if(auto dismiss = dynamic_ptr_cast<AssembleArtifacts>(pack))
|
|
|
return !vstd::contains(ourIds, dismiss->heroID);
|
|
|
- }
|
|
|
|
|
|
if(auto upgrade = dynamic_ptr_cast<UpgradeCreature>(pack))
|
|
|
- {
|
|
|
return !vstd::contains(ourIds, upgrade->id);
|
|
|
- }
|
|
|
+
|
|
|
return CDialogQuery::blocksPack(pack);
|
|
|
}
|
|
|
|
|
|
-void CBlockingDialogQuery::notifyObjectAboutRemoval(const CObjectVisitQuery &objectVisit) const
|
|
|
+void CBlockingDialogQuery::notifyObjectAboutRemoval(const CObjectVisitQuery & objectVisit) const
|
|
|
{
|
|
|
assert(answer);
|
|
|
objectVisit.visitedObject->blockingDialogAnswered(objectVisit.visitingHero, *answer);
|
|
|
}
|
|
|
|
|
|
-CBlockingDialogQuery::CBlockingDialogQuery(const BlockingDialog &bd)
|
|
|
+CBlockingDialogQuery::CBlockingDialogQuery(CGameHandler * owner, const BlockingDialog & bd):
|
|
|
+ CDialogQuery(owner)
|
|
|
{
|
|
|
this->bd = bd;
|
|
|
addPlayer(bd.player);
|
|
|
}
|
|
|
|
|
|
-void CTeleportDialogQuery::notifyObjectAboutRemoval(const CObjectVisitQuery &objectVisit) const
|
|
|
+void CTeleportDialogQuery::notifyObjectAboutRemoval(const CObjectVisitQuery & objectVisit) const
|
|
|
{
|
|
|
// do not change to dynamic_ptr_cast - SIGSEGV!
|
|
|
auto obj = dynamic_cast<const CGTeleport*>(objectVisit.visitedObject);
|
|
@@ -330,71 +354,80 @@ void CTeleportDialogQuery::notifyObjectAboutRemoval(const CObjectVisitQuery &obj
|
|
|
logGlobal->error("Invalid instance in teleport query");
|
|
|
}
|
|
|
|
|
|
-CTeleportDialogQuery::CTeleportDialogQuery(const TeleportDialog &td)
|
|
|
+CTeleportDialogQuery::CTeleportDialogQuery(CGameHandler * owner, const TeleportDialog & td):
|
|
|
+ CDialogQuery(owner)
|
|
|
{
|
|
|
this->td = td;
|
|
|
addPlayer(td.hero->tempOwner);
|
|
|
}
|
|
|
|
|
|
-CHeroLevelUpDialogQuery::CHeroLevelUpDialogQuery(const HeroLevelUp &Hlu)
|
|
|
+CHeroLevelUpDialogQuery::CHeroLevelUpDialogQuery(CGameHandler * owner, const HeroLevelUp & Hlu):
|
|
|
+ CDialogQuery(owner)
|
|
|
{
|
|
|
hlu = Hlu;
|
|
|
addPlayer(hlu.hero->tempOwner);
|
|
|
}
|
|
|
|
|
|
-void CHeroLevelUpDialogQuery::onRemoval(CGameHandler *gh, PlayerColor color)
|
|
|
+void CHeroLevelUpDialogQuery::onRemoval(PlayerColor color)
|
|
|
{
|
|
|
assert(answer);
|
|
|
logGlobal->trace("Completing hero level-up query. %s gains skill %d", hlu.hero->getObjectName(), answer.get());
|
|
|
gh->levelUpHero(hlu.hero, hlu.skills[*answer]);
|
|
|
}
|
|
|
|
|
|
-void CHeroLevelUpDialogQuery::notifyObjectAboutRemoval(const CObjectVisitQuery &objectVisit) const
|
|
|
+void CHeroLevelUpDialogQuery::notifyObjectAboutRemoval(const CObjectVisitQuery & objectVisit) const
|
|
|
{
|
|
|
objectVisit.visitedObject->heroLevelUpDone(objectVisit.visitingHero);
|
|
|
}
|
|
|
|
|
|
-CCommanderLevelUpDialogQuery::CCommanderLevelUpDialogQuery(const CommanderLevelUp &Clu)
|
|
|
+CCommanderLevelUpDialogQuery::CCommanderLevelUpDialogQuery(CGameHandler * owner, const CommanderLevelUp & Clu):
|
|
|
+ CDialogQuery(owner)
|
|
|
{
|
|
|
clu = Clu;
|
|
|
addPlayer(clu.hero->tempOwner);
|
|
|
}
|
|
|
|
|
|
-void CCommanderLevelUpDialogQuery::onRemoval(CGameHandler *gh, PlayerColor color)
|
|
|
+void CCommanderLevelUpDialogQuery::onRemoval(PlayerColor color)
|
|
|
{
|
|
|
assert(answer);
|
|
|
logGlobal->trace("Completing commander level-up query. Commander of hero %s gains skill %s", clu.hero->getObjectName(), answer.get());
|
|
|
gh->levelUpCommander(clu.hero->commander, clu.skills[*answer]);
|
|
|
}
|
|
|
|
|
|
-void CCommanderLevelUpDialogQuery::notifyObjectAboutRemoval(const CObjectVisitQuery &objectVisit) const
|
|
|
+void CCommanderLevelUpDialogQuery::notifyObjectAboutRemoval(const CObjectVisitQuery & objectVisit) const
|
|
|
{
|
|
|
objectVisit.visitedObject->heroLevelUpDone(objectVisit.visitingHero);
|
|
|
}
|
|
|
|
|
|
+CDialogQuery::CDialogQuery(CGameHandler * owner):
|
|
|
+ CGhQuery(owner)
|
|
|
+{
|
|
|
+
|
|
|
+}
|
|
|
+
|
|
|
bool CDialogQuery::endsByPlayerAnswer() const
|
|
|
{
|
|
|
return true;
|
|
|
}
|
|
|
|
|
|
-bool CDialogQuery::blocksPack(const CPack *pack) const
|
|
|
+bool CDialogQuery::blocksPack(const CPack * pack) const
|
|
|
{
|
|
|
- //We accept only query replies from correct player
|
|
|
- if(auto reply = dynamic_ptr_cast<QueryReply>(pack))
|
|
|
- {
|
|
|
- return !vstd::contains(players, reply->player);
|
|
|
- }
|
|
|
+ return blockAllButReply(pack);
|
|
|
+}
|
|
|
|
|
|
- return true;
|
|
|
+void CDialogQuery::setReply(const JsonNode & reply)
|
|
|
+{
|
|
|
+ if(reply.getType() == JsonNode::DATA_INTEGER)
|
|
|
+ answer = reply.Integer();
|
|
|
}
|
|
|
|
|
|
-CHeroMovementQuery::CHeroMovementQuery(const TryMoveHero &Tmh, const CGHeroInstance *Hero, bool VisitDestAfterVictory)
|
|
|
- : tmh(Tmh), visitDestAfterVictory(VisitDestAfterVictory), hero(Hero)
|
|
|
+CHeroMovementQuery::CHeroMovementQuery(CGameHandler * owner, const TryMoveHero & Tmh, const CGHeroInstance * Hero, bool VisitDestAfterVictory):
|
|
|
+ CGhQuery(owner), tmh(Tmh), visitDestAfterVictory(VisitDestAfterVictory), hero(Hero)
|
|
|
{
|
|
|
players.push_back(hero->tempOwner);
|
|
|
}
|
|
|
|
|
|
-void CHeroMovementQuery::onExposure(CGameHandler *gh, QueryPtr topQuery)
|
|
|
+void CHeroMovementQuery::onExposure(QueryPtr topQuery)
|
|
|
{
|
|
|
assert(players.size() == 1);
|
|
|
|
|
@@ -407,10 +440,10 @@ void CHeroMovementQuery::onExposure(CGameHandler *gh, QueryPtr topQuery)
|
|
|
gh->visitObjectOnTile(*gh->getTile(CGHeroInstance::convertPosition(tmh.end, false)), hero);
|
|
|
}
|
|
|
|
|
|
- gh->queries.popIfTop(*this);
|
|
|
+ owner->popIfTop(*this);
|
|
|
}
|
|
|
|
|
|
-void CHeroMovementQuery::onRemoval(CGameHandler *gh, PlayerColor color)
|
|
|
+void CHeroMovementQuery::onRemoval(PlayerColor color)
|
|
|
{
|
|
|
PlayerBlocked pb;
|
|
|
pb.player = color;
|
|
@@ -419,7 +452,7 @@ void CHeroMovementQuery::onRemoval(CGameHandler *gh, PlayerColor color)
|
|
|
gh->sendAndApply(&pb);
|
|
|
}
|
|
|
|
|
|
-void CHeroMovementQuery::onAdding(CGameHandler *gh, PlayerColor color)
|
|
|
+void CHeroMovementQuery::onAdding(PlayerColor color)
|
|
|
{
|
|
|
PlayerBlocked pb;
|
|
|
pb.player = color;
|
|
@@ -427,3 +460,29 @@ void CHeroMovementQuery::onAdding(CGameHandler *gh, PlayerColor color)
|
|
|
pb.startOrEnd = PlayerBlocked::BLOCKADE_STARTED;
|
|
|
gh->sendAndApply(&pb);
|
|
|
}
|
|
|
+
|
|
|
+CGenericQuery::CGenericQuery(Queries * Owner, PlayerColor color, std::function<void(const JsonNode &)> Callback):
|
|
|
+ CQuery(Owner), callback(Callback)
|
|
|
+{
|
|
|
+ addPlayer(color);
|
|
|
+}
|
|
|
+
|
|
|
+bool CGenericQuery::blocksPack(const CPack * pack) const
|
|
|
+{
|
|
|
+ return blockAllButReply(pack);
|
|
|
+}
|
|
|
+
|
|
|
+bool CGenericQuery::endsByPlayerAnswer() const
|
|
|
+{
|
|
|
+ return true;
|
|
|
+}
|
|
|
+
|
|
|
+void CGenericQuery::onExposure(QueryPtr topQuery)
|
|
|
+{
|
|
|
+ //do nothing
|
|
|
+}
|
|
|
+
|
|
|
+void CGenericQuery::setReply(const JsonNode & reply)
|
|
|
+{
|
|
|
+ callback(reply);
|
|
|
+}
|