IGameCallback.cpp 29 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976
  1. #include "StdInc.h"
  2. #include "IGameCallback.h"
  3. #include <boost/random/linear_congruential.hpp>
  4. #include "CGameState.h"
  5. #include "Mapping/CMap.h"
  6. #include "CObjectHandler.h"
  7. #include "CHeroHandler.h"
  8. #include "StartInfo.h"
  9. #include "CArtHandler.h"
  10. #include "CSpellHandler.h"
  11. #include "VCMI_Lib.h"
  12. #include "CTownHandler.h"
  13. #include "BattleState.h"
  14. #include "NetPacks.h"
  15. #include "CBuildingHandler.h"
  16. #include "GameConstants.h"
  17. #include "CModHandler.h"
  18. /*
  19. * IGameCallback.cpp, part of VCMI engine
  20. *
  21. * Authors: listed in file AUTHORS in main folder
  22. *
  23. * License: GNU General Public License v2.0 or later
  24. * Full text of license available in license.txt file, in main folder
  25. *
  26. */
  27. //TODO make clean
  28. #define ERROR_SILENT_RET_VAL_IF(cond, txt, retVal) do {if(cond){return retVal;}} while(0)
  29. #define ERROR_VERBOSE_OR_NOT_RET_VAL_IF(cond, verbose, txt, retVal) do {if(cond){if(verbose)tlog1 << BOOST_CURRENT_FUNCTION << ": " << txt << std::endl; return retVal;}} while(0)
  30. #define ERROR_RET_IF(cond, txt) do {if(cond){tlog1 << BOOST_CURRENT_FUNCTION << ": " << txt << std::endl; return;}} while(0)
  31. #define ERROR_RET_VAL_IF(cond, txt, retVal) do {if(cond){tlog1 << BOOST_CURRENT_FUNCTION << ": " << txt << std::endl; return retVal;}} while(0)
  32. extern boost::rand48 ran;
  33. CGameState * CPrivilagedInfoCallback::gameState ()
  34. {
  35. return gs;
  36. }
  37. int CGameInfoCallback::getOwner(int heroID) const
  38. {
  39. const CGObjectInstance *obj = getObj(heroID);
  40. ERROR_RET_VAL_IF(!obj, "No such object!", -1);
  41. return gs->map->objects[heroID]->tempOwner;
  42. }
  43. int CGameInfoCallback::getResource(int Player, int which) const
  44. {
  45. const PlayerState *p = getPlayer(Player);
  46. ERROR_RET_VAL_IF(!p, "No player info!", -1);
  47. ERROR_RET_VAL_IF(p->resources.size() <= which || which < 0, "No such resource!", -1);
  48. return p->resources[which];
  49. }
  50. const CGHeroInstance* CGameInfoCallback::getSelectedHero( int Player ) const
  51. {
  52. const PlayerState *p = getPlayer(Player);
  53. ERROR_RET_VAL_IF(!p, "No player info!", NULL);
  54. return getHero(p->currentSelection);
  55. }
  56. const CGHeroInstance* CGameInfoCallback::getSelectedHero() const
  57. {
  58. return getSelectedHero(gs->currentPlayer);
  59. }
  60. const PlayerSettings * CGameInfoCallback::getPlayerSettings(int color) const
  61. {
  62. return &gs->scenarioOps->getIthPlayersSettings(color);
  63. }
  64. void CPrivilagedInfoCallback::getTilesInRange( boost::unordered_set<int3, ShashInt3> &tiles, int3 pos, int radious, int player/*=-1*/, int mode/*=0*/ ) const
  65. {
  66. if(player >= GameConstants::PLAYER_LIMIT)
  67. {
  68. tlog1 << "Illegal call to getTilesInRange!\n";
  69. return;
  70. }
  71. if (radious == -1) //reveal entire map
  72. getAllTiles (tiles, player, -1, 0);
  73. else
  74. {
  75. const TeamState * team = gs->getPlayerTeam(player);
  76. for (int xd = std::max<int>(pos.x - radious , 0); xd <= std::min<int>(pos.x + radious, gs->map->width - 1); xd++)
  77. {
  78. for (int yd = std::max<int>(pos.y - radious, 0); yd <= std::min<int>(pos.y + radious, gs->map->height - 1); yd++)
  79. {
  80. double distance = pos.dist2d(int3(xd,yd,pos.z)) - 0.5;
  81. if(distance <= radious)
  82. {
  83. if(player < 0
  84. || (mode == 1 && team->fogOfWarMap[xd][yd][pos.z]==0)
  85. || (mode == -1 && team->fogOfWarMap[xd][yd][pos.z]==1)
  86. )
  87. tiles.insert(int3(xd,yd,pos.z));
  88. }
  89. }
  90. }
  91. }
  92. }
  93. void CPrivilagedInfoCallback::getAllTiles (boost::unordered_set<int3, ShashInt3> &tiles, int Player/*=-1*/, int level, int surface ) const
  94. {
  95. if(Player >= GameConstants::PLAYER_LIMIT)
  96. {
  97. tlog1 << "Illegal call to getAllTiles !\n";
  98. return;
  99. }
  100. bool water = surface == 0 || surface == 2,
  101. land = surface == 0 || surface == 1;
  102. std::vector<int> floors;
  103. if(level == -1)
  104. {
  105. for(int b = 0; b < (gs->map->twoLevel ? 2 : 1); ++b)
  106. {
  107. floors.push_back(b);
  108. }
  109. }
  110. else
  111. floors.push_back(level);
  112. for (std::vector<int>::const_iterator i = floors.begin(); i!= floors.end(); i++)
  113. {
  114. register int zd = *i;
  115. for (int xd = 0; xd < gs->map->width; xd++)
  116. {
  117. for (int yd = 0; yd < gs->map->height; yd++)
  118. {
  119. if ((getTile (int3 (xd,yd,zd))->terType == 8 && water)
  120. || (getTile (int3 (xd,yd,zd))->terType != 8 && land))
  121. tiles.insert(int3(xd,yd,zd));
  122. }
  123. }
  124. }
  125. }
  126. void CPrivilagedInfoCallback::getFreeTiles (std::vector<int3> &tiles) const
  127. {
  128. std::vector<int> floors;
  129. for (int b = 0; b < (gs->map->twoLevel ? 2 : 1); ++b)
  130. {
  131. floors.push_back(b);
  132. }
  133. const TerrainTile *tinfo;
  134. for (std::vector<int>::const_iterator i = floors.begin(); i!= floors.end(); i++)
  135. {
  136. register int zd = *i;
  137. for (int xd = 0; xd < gs->map->width; xd++)
  138. {
  139. for (int yd = 0; yd < gs->map->height; yd++)
  140. {
  141. tinfo = getTile(int3 (xd,yd,zd));
  142. if (tinfo->terType != 8 && !tinfo->blocked) //land and free
  143. tiles.push_back (int3 (xd,yd,zd));
  144. }
  145. }
  146. }
  147. }
  148. bool CGameInfoCallback::isAllowed( int type, int id )
  149. {
  150. switch(type)
  151. {
  152. case 0:
  153. return gs->map->allowedSpell[id];
  154. case 1:
  155. return gs->map->allowedArtifact[id];
  156. case 2:
  157. return gs->map->allowedAbilities[id];
  158. default:
  159. ERROR_RET_VAL_IF(1, "Wrong type!", false);
  160. }
  161. }
  162. void CPrivilagedInfoCallback::pickAllowedArtsSet(std::vector<const CArtifact*> &out)
  163. {
  164. for (int j = 0; j < 3 ; j++)
  165. out.push_back(VLC->arth->artifacts[getRandomArt(CArtifact::ART_TREASURE)]);
  166. for (int j = 0; j < 3 ; j++)
  167. out.push_back(VLC->arth->artifacts[getRandomArt(CArtifact::ART_MINOR)]);
  168. out.push_back(VLC->arth->artifacts[getRandomArt(CArtifact::ART_MAJOR)]);
  169. }
  170. ui16 CPrivilagedInfoCallback::getRandomArt (int flags)
  171. {
  172. return VLC->arth->getRandomArt(flags);
  173. }
  174. ui16 CPrivilagedInfoCallback::getArtSync (ui32 rand, int flags)
  175. {
  176. return VLC->arth->getArtSync (rand, flags);
  177. }
  178. void CPrivilagedInfoCallback::erasePickedArt (si32 id)
  179. {
  180. VLC->arth->erasePickedArt(id);
  181. }
  182. void CPrivilagedInfoCallback::getAllowedSpells(std::vector<ui16> &out, ui16 level)
  183. {
  184. CSpell *spell;
  185. for (ui32 i = 0; i < gs->map->allowedSpell.size(); i++) //spellh size appears to be greater (?)
  186. {
  187. spell = VLC->spellh->spells[i];
  188. if (isAllowed (0, spell->id) && spell->level == level)
  189. {
  190. out.push_back(spell->id);
  191. }
  192. }
  193. }
  194. inline TerrainTile * CNonConstInfoCallback::getTile( int3 pos )
  195. {
  196. if(!gs->map->isInTheMap(pos))
  197. return NULL;
  198. return &gs->map->getTile(pos);
  199. }
  200. const PlayerState * CGameInfoCallback::getPlayer(int color, bool verbose) const
  201. {
  202. ERROR_VERBOSE_OR_NOT_RET_VAL_IF(!hasAccess(color), verbose, "Cannot access player " << color << "info!", NULL);
  203. ERROR_VERBOSE_OR_NOT_RET_VAL_IF(!vstd::contains(gs->players,color), verbose, "Cannot find player " << color << "info!", NULL);
  204. return &gs->players[color];
  205. }
  206. const CTown * CGameInfoCallback::getNativeTown(int color) const
  207. {
  208. const PlayerSettings *ps = getPlayerSettings(color);
  209. ERROR_RET_VAL_IF(!ps, "There is no such player!", NULL);
  210. return &VLC->townh->towns[ps->castle];
  211. }
  212. const CGObjectInstance * CGameInfoCallback::getObjByQuestIdentifier(int identifier) const
  213. {
  214. ERROR_RET_VAL_IF(!vstd::contains(gs->map->questIdentifierToId, identifier), "There is no object with such quest identifier!", NULL);
  215. return getObj(gs->map->questIdentifierToId[identifier]);
  216. }
  217. /************************************************************************/
  218. /* */
  219. /************************************************************************/
  220. const CGObjectInstance* CGameInfoCallback::getObj(int objid, bool verbose) const
  221. {
  222. if(objid < 0 || objid >= gs->map->objects.size())
  223. {
  224. if(verbose)
  225. tlog1 << "Cannot get object with id " << objid << std::endl;
  226. return NULL;
  227. }
  228. const CGObjectInstance *ret = gs->map->objects[objid];
  229. if(!ret)
  230. {
  231. if(verbose)
  232. tlog1 << "Cannot get object with id " << objid << ". Object was removed.\n";
  233. return NULL;
  234. }
  235. if(!isVisible(ret, player))
  236. {
  237. if(verbose)
  238. tlog1 << "Cannot get object with id " << objid << ". Object is not visible.\n";
  239. return NULL;
  240. }
  241. return ret;
  242. }
  243. const CGHeroInstance* CGameInfoCallback::getHero(int objid) const
  244. {
  245. const CGObjectInstance *obj = getObj(objid, false);
  246. if(obj)
  247. return dynamic_cast<const CGHeroInstance*>(obj);
  248. else
  249. return NULL;
  250. }
  251. const CGTownInstance* CGameInfoCallback::getTown(int objid) const
  252. {
  253. const CGObjectInstance *obj = getObj(objid, false);
  254. if(obj)
  255. return dynamic_cast<const CGTownInstance*>(gs->map->objects[objid].get());
  256. else
  257. return NULL;
  258. }
  259. void CGameInfoCallback::getUpgradeInfo(const CArmedInstance *obj, int stackPos, UpgradeInfo &out) const
  260. {
  261. //boost::shared_lock<boost::shared_mutex> lock(*gs->mx);
  262. ERROR_RET_IF(!canGetFullInfo(obj), "Cannot get info about not owned object!");
  263. ERROR_RET_IF(!obj->hasStackAtSlot(stackPos), "There is no such stack!");
  264. out = gs->getUpgradeInfo(obj->getStack(stackPos));
  265. //return gs->getUpgradeInfo(obj->getStack(stackPos));
  266. }
  267. const StartInfo * CGameInfoCallback::getStartInfo(bool beforeRandomization /*= false*/) const
  268. {
  269. //boost::shared_lock<boost::shared_mutex> lock(*gs->mx);
  270. if(beforeRandomization)
  271. return gs->initialOpts;
  272. else
  273. return gs->scenarioOps;
  274. }
  275. int CGameInfoCallback::getSpellCost(const CSpell * sp, const CGHeroInstance * caster) const
  276. {
  277. //boost::shared_lock<boost::shared_mutex> lock(*gs->mx);
  278. ERROR_RET_VAL_IF(!canGetFullInfo(caster), "Cannot get info about caster!", -1);
  279. //if there is a battle
  280. if(gs->curB)
  281. return gs->curB->battleGetSpellCost(sp, caster);
  282. //if there is no battle
  283. return caster->getSpellCost(sp);
  284. }
  285. int CGameInfoCallback::estimateSpellDamage(const CSpell * sp, const CGHeroInstance * hero) const
  286. {
  287. //boost::shared_lock<boost::shared_mutex> lock(*gs->mx);
  288. ERROR_RET_VAL_IF(hero && !canGetFullInfo(hero), "Cannot get info about caster!", -1);
  289. if(!gs->curB) //no battle
  290. {
  291. if (hero) //but we see hero's spellbook
  292. return gs->curB->calculateSpellDmg(sp, hero, NULL, hero->getSpellSchoolLevel(sp), hero->getPrimSkillLevel(2));
  293. else
  294. return 0; //mage guild
  295. }
  296. //gs->getHero(gs->currentPlayer)
  297. //const CGHeroInstance * ourHero = gs->curB->heroes[0]->tempOwner == player ? gs->curB->heroes[0] : gs->curB->heroes[1];
  298. const CGHeroInstance * ourHero = hero;
  299. return gs->curB->calculateSpellDmg(sp, ourHero, NULL, ourHero->getSpellSchoolLevel(sp), ourHero->getPrimSkillLevel(2));
  300. }
  301. void CGameInfoCallback::getThievesGuildInfo(SThievesGuildInfo & thi, const CGObjectInstance * obj)
  302. {
  303. //boost::shared_lock<boost::shared_mutex> lock(*gs->mx);
  304. ERROR_RET_IF(!obj, "No guild object!");
  305. ERROR_RET_IF(obj->ID == Obj::TOWN && !canGetFullInfo(obj), "Cannot get info about town guild object!");
  306. //TODO: advmap object -> check if they're visited by our hero
  307. if(obj->ID == Obj::TOWN || obj->ID == Obj::TAVERN)
  308. {
  309. gs->obtainPlayersStats(thi, gs->players[obj->tempOwner].towns.size());
  310. }
  311. else if(obj->ID == Obj::DEN_OF_THIEVES)
  312. {
  313. gs->obtainPlayersStats(thi, 20);
  314. }
  315. }
  316. int CGameInfoCallback::howManyTowns(int Player) const
  317. {
  318. ERROR_RET_VAL_IF(!hasAccess(Player), "Access forbidden!", -1);
  319. return gs->players[Player].towns.size();
  320. }
  321. bool CGameInfoCallback::getTownInfo( const CGObjectInstance *town, InfoAboutTown &dest ) const
  322. {
  323. ERROR_RET_VAL_IF(!isVisible(town, player), "Town is not visible!", false); //it's not a town or it's not visible for layer
  324. bool detailed = hasAccess(town->tempOwner);
  325. //TODO vision support
  326. if(town->ID == Obj::TOWN)
  327. dest.initFromTown(static_cast<const CGTownInstance *>(town), detailed);
  328. else if(town->ID == Obj::GARRISON || town->ID == Obj::GARRISON2)
  329. dest.initFromArmy(static_cast<const CArmedInstance *>(town), detailed);
  330. else
  331. return false;
  332. return true;
  333. }
  334. int3 CGameInfoCallback::guardingCreaturePosition (int3 pos) const
  335. {
  336. ERROR_RET_VAL_IF(!isVisible(pos), "Tile is not visible!", int3(-1,-1,-1));
  337. return gs->guardingCreaturePosition(pos);
  338. }
  339. std::vector<const CGObjectInstance*> CGameInfoCallback::getGuardingCreatures (int3 pos) const
  340. {
  341. ERROR_RET_VAL_IF(!isVisible(pos), "Tile is not visible!", std::vector<const CGObjectInstance*>());
  342. std::vector<const CGObjectInstance*> ret;
  343. BOOST_FOREACH(auto cr, gs->guardingCreatures(pos))
  344. {
  345. ret.push_back(cr);
  346. }
  347. return ret;
  348. }
  349. bool CGameInfoCallback::getHeroInfo( const CGObjectInstance *hero, InfoAboutHero &dest ) const
  350. {
  351. const CGHeroInstance *h = dynamic_cast<const CGHeroInstance *>(hero);
  352. ERROR_RET_VAL_IF(!h, "That's not a hero!", false);
  353. ERROR_RET_VAL_IF(!isVisible(h->getPosition(false)), "That hero is not visible!", false);
  354. //TODO vision support
  355. dest.initFromHero(h, hasAccess(h->tempOwner));
  356. return true;
  357. }
  358. int CGameInfoCallback::getDate(Date::EDateType mode) const
  359. {
  360. //boost::shared_lock<boost::shared_mutex> lock(*gs->mx);
  361. return gs->getDate(mode);
  362. }
  363. std::vector < std::string > CGameInfoCallback::getObjDescriptions(int3 pos) const
  364. {
  365. //boost::shared_lock<boost::shared_mutex> lock(*gs->mx);
  366. std::vector<std::string> ret;
  367. const TerrainTile *t = getTile(pos);
  368. ERROR_RET_VAL_IF(!t, "Not a valid tile given!", ret);
  369. BOOST_FOREACH(const CGObjectInstance * obj, t->blockingObjects)
  370. ret.push_back(obj->getHoverText());
  371. return ret;
  372. }
  373. bool CGameInfoCallback::verifyPath(CPath * path, bool blockSea) const
  374. {
  375. for (size_t i=0; i < path->nodes.size(); ++i)
  376. {
  377. const TerrainTile *t = getTile(path->nodes[i].coord); //current tile
  378. ERROR_RET_VAL_IF(!t, "Path contains not visible tile: " << path->nodes[i].coord << "!", false);
  379. if (t->blocked && !t->visitable)
  380. return false; //path is wrong - one of the tiles is blocked
  381. if (blockSea)
  382. {
  383. if (i==0)
  384. continue;
  385. const TerrainTile *prev = getTile(path->nodes[i-1].coord); //tile of previous node on the path
  386. if (( t->terType == ETerrainType::WATER && prev->terType != ETerrainType::WATER)
  387. || (t->terType != ETerrainType::WATER && prev->terType == ETerrainType::WATER)
  388. || prev->terType == ETerrainType::ROCK
  389. )
  390. return false;
  391. }
  392. }
  393. return true;
  394. }
  395. bool CGameInfoCallback::isVisible(int3 pos, int Player) const
  396. {
  397. //boost::shared_lock<boost::shared_mutex> lock(*gs->mx);
  398. return gs->map->isInTheMap(pos) && (Player == -1 || gs->isVisible(pos, Player));
  399. }
  400. bool CGameInfoCallback::isVisible(int3 pos) const
  401. {
  402. return isVisible(pos,player);
  403. }
  404. bool CGameInfoCallback::isVisible( const CGObjectInstance *obj, int Player ) const
  405. {
  406. return gs->isVisible(obj, Player);
  407. }
  408. bool CGameInfoCallback::isVisible(const CGObjectInstance *obj) const
  409. {
  410. return isVisible(obj, player);
  411. }
  412. // const CCreatureSet* CInfoCallback::getGarrison(const CGObjectInstance *obj) const
  413. // {
  414. // //boost::shared_lock<boost::shared_mutex> lock(*gs->mx);
  415. // if()
  416. // const CArmedInstance *armi = dynamic_cast<const CArmedInstance*>(obj);
  417. // if(!armi)
  418. // return NULL;
  419. // else
  420. // return armi;
  421. // }
  422. std::vector < const CGObjectInstance * > CGameInfoCallback::getBlockingObjs( int3 pos ) const
  423. {
  424. std::vector<const CGObjectInstance *> ret;
  425. const TerrainTile *t = getTile(pos);
  426. ERROR_RET_VAL_IF(!t, "Not a valid tile requested!", ret);
  427. BOOST_FOREACH(const CGObjectInstance * obj, t->blockingObjects)
  428. ret.push_back(obj);
  429. return ret;
  430. }
  431. std::vector <const CGObjectInstance * > CGameInfoCallback::getVisitableObjs(int3 pos, bool verbose /*= true*/) const
  432. {
  433. std::vector<const CGObjectInstance *> ret;
  434. const TerrainTile *t = getTile(pos, verbose);
  435. ERROR_VERBOSE_OR_NOT_RET_VAL_IF(!t, verbose, pos << " is not visible!", ret);
  436. BOOST_FOREACH(const CGObjectInstance * obj, t->visitableObjects)
  437. {
  438. if(player < 0 || obj->ID != Obj::EVENT) //hide events from players
  439. ret.push_back(obj);
  440. }
  441. return ret;
  442. }
  443. std::vector < const CGObjectInstance * > CGameInfoCallback::getFlaggableObjects(int3 pos) const
  444. {
  445. std::vector<const CGObjectInstance *> ret;
  446. const TerrainTile *t = getTile(pos);
  447. ERROR_RET_VAL_IF(!t, "Not a valid tile requested!", ret);
  448. BOOST_FOREACH(const CGObjectInstance *obj, t->blockingObjects)
  449. if(obj->tempOwner != 254)
  450. ret.push_back(obj);
  451. // const std::vector < std::pair<const CGObjectInstance*,SDL_Rect> > & objs = CGI->mh->ttiles[pos.x][pos.y][pos.z].objects;
  452. // for(size_t b=0; b<objs.size(); ++b)
  453. // {
  454. // if(objs[b].first->tempOwner!=254 && !((objs[b].first->defInfo->blockMap[pos.y - objs[b].first->pos.y + 5] >> (objs[b].first->pos.x - pos.x)) & 1))
  455. // ret.push_back(CGI->mh->ttiles[pos.x][pos.y][pos.z].objects[b].first);
  456. // }
  457. return ret;
  458. }
  459. int3 CGameInfoCallback::getMapSize() const
  460. {
  461. return int3(gs->map->width, gs->map->height, gs->map->twoLevel ? 2 : 1);
  462. }
  463. std::vector<const CGHeroInstance *> CGameInfoCallback::getAvailableHeroes(const CGObjectInstance * townOrTavern) const
  464. {
  465. std::vector<const CGHeroInstance *> ret;
  466. //ERROR_RET_VAL_IF(!isOwnedOrVisited(townOrTavern), "Town or tavern must be owned or visited!", ret);
  467. //TODO: town needs to be owned, advmap tavern needs to be visited; to be reimplemented when visit tracking is done
  468. ret.resize(gs->players[player].availableHeroes.size());
  469. std::copy(gs->players[player].availableHeroes.begin(),gs->players[player].availableHeroes.end(),ret.begin());
  470. return ret;
  471. }
  472. const TerrainTile * CGameInfoCallback::getTile( int3 tile, bool verbose) const
  473. {
  474. ERROR_VERBOSE_OR_NOT_RET_VAL_IF(!isVisible(tile), verbose, tile << " is not visible!", NULL);
  475. //boost::shared_lock<boost::shared_mutex> lock(*gs->mx);
  476. return &gs->map->getTile(tile);
  477. }
  478. int CGameInfoCallback::canBuildStructure( const CGTownInstance *t, int ID )
  479. {
  480. ERROR_RET_VAL_IF(!canGetFullInfo(t), "Town is not owned!", -1);
  481. CBuilding * pom = t->town->buildings[ID];
  482. if(!pom)
  483. return EBuildingState::BUILDING_ERROR;
  484. if(t->hasBuilt(ID)) //already built
  485. return EBuildingState::ALREADY_PRESENT;
  486. //can we build it?
  487. if(t->forbiddenBuildings.find(ID)!=t->forbiddenBuildings.end())
  488. return EBuildingState::FORBIDDEN; //forbidden
  489. //checking for requirements
  490. std::set<int> reqs = getBuildingRequiments(t, ID);//getting all requirements
  491. bool notAllBuilt = false;
  492. for( std::set<int>::iterator ri = reqs.begin(); ri != reqs.end(); ri++ )
  493. {
  494. if(!t->hasBuilt(*ri)) //lack of requirements - cannot build
  495. {
  496. if(vstd::contains(t->forbiddenBuildings, *ri)) // not built requirement forbidden - same goes to this build
  497. return EBuildingState::FORBIDDEN;
  498. else
  499. notAllBuilt = true; // no return here - we need to check if any required builds are forbidden
  500. }
  501. }
  502. if(t->builded >= VLC->modh->settings.MAX_BUILDING_PER_TURN)
  503. return EBuildingState::CANT_BUILD_TODAY; //building limit
  504. if (notAllBuilt)
  505. return EBuildingState::PREREQUIRES;
  506. if(ID == 13) //capitol
  507. {
  508. const PlayerState *ps = getPlayer(t->tempOwner);
  509. if(ps)
  510. {
  511. BOOST_FOREACH(const CGTownInstance *t, ps->towns)
  512. {
  513. if(t->hasBuilt(EBuilding::CAPITOL))
  514. {
  515. return EBuildingState::HAVE_CAPITAL; //no more than one capitol
  516. }
  517. }
  518. }
  519. }
  520. else if(ID == 6) //shipyard
  521. {
  522. const TerrainTile *tile = getTile(t->bestLocation(), false);
  523. if(!tile || tile->terType != ETerrainType::WATER)
  524. return EBuildingState::NO_WATER; //lack of water
  525. }
  526. //checking resources
  527. if(!pom->resources.canBeAfforded(getPlayer(t->tempOwner)->resources))
  528. return EBuildingState::NO_RESOURCES; //lack of res
  529. return EBuildingState::ALLOWED;
  530. }
  531. std::set<int> CGameInfoCallback::getBuildingRequiments( const CGTownInstance *t, int ID )
  532. {
  533. ERROR_RET_VAL_IF(!canGetFullInfo(t), "Town is not owned!", std::set<int>());
  534. std::set<int> used;
  535. used.insert(ID);
  536. auto reqs = t->town->buildings[ID]->requirements;
  537. bool found;
  538. do
  539. {
  540. found = false;
  541. for(auto i=reqs.begin();i!=reqs.end();i++)
  542. {
  543. if(used.find(*i)==used.end()) //we haven't added requirements for this building
  544. {
  545. found = true;
  546. auto & requires = t->town->buildings[*i]->requirements;
  547. used.insert(*i);
  548. for(auto j = requires.begin(); j!= requires.end(); j++)
  549. reqs.insert(*j);//creating full list of requirements
  550. }
  551. }
  552. }
  553. while (found);
  554. return reqs;
  555. }
  556. const CMapHeader * CGameInfoCallback::getMapHeader() const
  557. {
  558. return gs->map;
  559. }
  560. bool CGameInfoCallback::hasAccess(int playerId) const
  561. {
  562. return player < 0 || gs->getPlayerRelations( playerId, player );
  563. }
  564. int CGameInfoCallback::getPlayerStatus(int player) const
  565. {
  566. const PlayerState *ps = gs->getPlayer(player, false);
  567. if(!ps)
  568. return -1;
  569. return ps->status;
  570. }
  571. std::string CGameInfoCallback::getTavernGossip(const CGObjectInstance * townOrTavern) const
  572. {
  573. return "GOSSIP TEST";
  574. }
  575. int CGameInfoCallback::getPlayerRelations( ui8 color1, ui8 color2 ) const
  576. {
  577. return gs->getPlayerRelations(color1, color2);
  578. }
  579. bool CGameInfoCallback::canGetFullInfo(const CGObjectInstance *obj) const
  580. {
  581. return !obj || hasAccess(obj->tempOwner);
  582. }
  583. int CGameInfoCallback::getHeroCount( int player, bool includeGarrisoned ) const
  584. {
  585. int ret = 0;
  586. const PlayerState *p = gs->getPlayer(player);
  587. ERROR_RET_VAL_IF(!p, "No such player!", -1);
  588. if(includeGarrisoned)
  589. return p->heroes.size();
  590. else
  591. for(ui32 i = 0; i < p->heroes.size(); i++)
  592. if(!p->heroes[i]->inTownGarrison)
  593. ret++;
  594. return ret;
  595. }
  596. bool CGameInfoCallback::isOwnedOrVisited(const CGObjectInstance *obj) const
  597. {
  598. if(canGetFullInfo(obj))
  599. return true;
  600. const TerrainTile *t = getTile(obj->visitablePos()); //get entrance tile
  601. const CGObjectInstance *visitor = t->visitableObjects.back(); //visitong hero if present or the obejct itself at last
  602. return visitor->ID == Obj::HERO && canGetFullInfo(visitor); //owned or allied hero is a visitor
  603. }
  604. int CGameInfoCallback::getCurrentPlayer() const
  605. {
  606. return gs->currentPlayer;
  607. }
  608. CGameInfoCallback::CGameInfoCallback()
  609. {
  610. }
  611. CGameInfoCallback::CGameInfoCallback(CGameState *GS, int Player)
  612. {
  613. gs = GS;
  614. player = Player;
  615. }
  616. const std::vector< std::vector< std::vector<ui8> > > & CPlayerSpecificInfoCallback::getVisibilityMap() const
  617. {
  618. //boost::shared_lock<boost::shared_mutex> lock(*gs->mx);
  619. return gs->getPlayerTeam(player)->fogOfWarMap;
  620. }
  621. int CPlayerSpecificInfoCallback::howManyTowns() const
  622. {
  623. //boost::shared_lock<boost::shared_mutex> lock(*gs->mx);
  624. ERROR_RET_VAL_IF(player == -1, "Applicable only for player callbacks", -1);
  625. return CGameInfoCallback::howManyTowns(player);
  626. }
  627. std::vector < const CGTownInstance *> CPlayerSpecificInfoCallback::getTownsInfo(bool onlyOur) const
  628. {
  629. //boost::shared_lock<boost::shared_mutex> lock(*gs->mx);
  630. std::vector < const CGTownInstance *> ret = std::vector < const CGTownInstance *>();
  631. for ( auto i=gs->players.begin() ; i!=gs->players.end();i++)
  632. {
  633. for (size_t j = 0; j < (*i).second.towns.size(); ++j)
  634. {
  635. if ((*i).first==player
  636. || (isVisible((*i).second.towns[j],player) && !onlyOur))
  637. {
  638. ret.push_back((*i).second.towns[j]);
  639. }
  640. }
  641. } // for ( std::map<int, PlayerState>::iterator i=gs->players.begin() ; i!=gs->players.end();i++)
  642. return ret;
  643. }
  644. std::vector < const CGHeroInstance *> CPlayerSpecificInfoCallback::getHeroesInfo(bool onlyOur) const
  645. {
  646. //boost::shared_lock<boost::shared_mutex> lock(*gs->mx);
  647. std::vector < const CGHeroInstance *> ret;
  648. for(size_t i=0;i<gs->map->heroes.size();i++)
  649. {
  650. if( (gs->map->heroes[i]->tempOwner==player) ||
  651. (isVisible(gs->map->heroes[i]->getPosition(false),player) && !onlyOur) )
  652. {
  653. ret.push_back(gs->map->heroes[i]);
  654. }
  655. }
  656. return ret;
  657. }
  658. int CPlayerSpecificInfoCallback::getMyColor() const
  659. {
  660. return player;
  661. }
  662. int CPlayerSpecificInfoCallback::getHeroSerial(const CGHeroInstance * hero, bool includeGarrisoned) const
  663. {
  664. if (hero->inTownGarrison && !includeGarrisoned)
  665. return -1;
  666. size_t index = 0;
  667. auto & heroes = gs->players[player].heroes;
  668. for (auto curHero= heroes.begin(); curHero!=heroes.end(); curHero++)
  669. {
  670. if (includeGarrisoned || !(*curHero)->inTownGarrison)
  671. index++;
  672. if (*curHero == hero)
  673. return index;
  674. }
  675. return -1;
  676. }
  677. int3 CPlayerSpecificInfoCallback::getGrailPos( double &outKnownRatio )
  678. {
  679. if (CGObelisk::obeliskCount == 0)
  680. {
  681. outKnownRatio = 0.0;
  682. }
  683. else
  684. {
  685. outKnownRatio = static_cast<double>(CGObelisk::visited[gs->getPlayerTeam(player)->id]) / CGObelisk::obeliskCount;
  686. }
  687. return gs->map->grailPos;
  688. }
  689. std::vector < const CGObjectInstance * > CPlayerSpecificInfoCallback::getMyObjects() const
  690. {
  691. std::vector < const CGObjectInstance * > ret;
  692. BOOST_FOREACH(const CGObjectInstance * obj, gs->map->objects)
  693. {
  694. if(obj && obj->tempOwner == player)
  695. ret.push_back(obj);
  696. }
  697. return ret;
  698. }
  699. std::vector < const CGDwelling * > CPlayerSpecificInfoCallback::getMyDwellings() const
  700. {
  701. std::vector < const CGDwelling * > ret;
  702. BOOST_FOREACH(CGDwelling * dw, gs->getPlayer(player)->dwellings)
  703. {
  704. ret.push_back(dw);
  705. }
  706. return ret;
  707. }
  708. std::vector <QuestInfo> CPlayerSpecificInfoCallback::getMyQuests() const
  709. {
  710. std::vector <QuestInfo> ret;
  711. BOOST_FOREACH (auto quest, gs->getPlayer(player)->quests)
  712. {
  713. ret.push_back (quest);
  714. }
  715. return ret;
  716. }
  717. int CPlayerSpecificInfoCallback::howManyHeroes(bool includeGarrisoned) const
  718. {
  719. //boost::shared_lock<boost::shared_mutex> lock(*gs->mx);
  720. ERROR_RET_VAL_IF(player == -1, "Applicable only for player callbacks", -1);
  721. return getHeroCount(player,includeGarrisoned);
  722. }
  723. const CGHeroInstance* CPlayerSpecificInfoCallback::getHeroBySerial(int serialId, bool includeGarrisoned) const
  724. {
  725. const PlayerState *p = getPlayer(player);
  726. ERROR_RET_VAL_IF(!p, "No player info", NULL);
  727. if (!includeGarrisoned)
  728. {
  729. for(ui32 i = 0; i < p->heroes.size() && i<=serialId; i++)
  730. if(p->heroes[i]->inTownGarrison)
  731. serialId++;
  732. }
  733. ERROR_RET_VAL_IF(serialId < 0 || serialId >= p->heroes.size(), "No player info", NULL);
  734. return p->heroes[serialId];
  735. }
  736. const CGTownInstance* CPlayerSpecificInfoCallback::getTownBySerial(int serialId) const
  737. {
  738. const PlayerState *p = getPlayer(player);
  739. ERROR_RET_VAL_IF(!p, "No player info", NULL);
  740. ERROR_RET_VAL_IF(serialId < 0 || serialId >= p->towns.size(), "No player info", NULL);
  741. return p->towns[serialId];
  742. }
  743. int CPlayerSpecificInfoCallback::getResourceAmount(int type) const
  744. {
  745. //boost::shared_lock<boost::shared_mutex> lock(*gs->mx);
  746. ERROR_RET_VAL_IF(player == -1, "Applicable only for player callbacks", -1);
  747. return getResource(player, type);
  748. }
  749. TResources CPlayerSpecificInfoCallback::getResourceAmount() const
  750. {
  751. //boost::shared_lock<boost::shared_mutex> lock(*gs->mx);
  752. ERROR_RET_VAL_IF(player == -1, "Applicable only for player callbacks", TResources());
  753. return gs->players[player].resources;
  754. }
  755. CGHeroInstance *CNonConstInfoCallback::getHero(int objid)
  756. {
  757. return const_cast<CGHeroInstance*>(CGameInfoCallback::getHero(objid));
  758. }
  759. CGTownInstance *CNonConstInfoCallback::getTown(int objid)
  760. {
  761. return const_cast<CGTownInstance*>(CGameInfoCallback::getTown(objid));
  762. }
  763. TeamState *CNonConstInfoCallback::getTeam(ui8 teamID)
  764. {
  765. return const_cast<TeamState*>(CGameInfoCallback::getTeam(teamID));
  766. }
  767. TeamState *CNonConstInfoCallback::getPlayerTeam(ui8 color)
  768. {
  769. return const_cast<TeamState*>(CGameInfoCallback::getPlayerTeam(color));
  770. }
  771. PlayerState * CNonConstInfoCallback::getPlayer( ui8 color, bool verbose )
  772. {
  773. return const_cast<PlayerState*>(CGameInfoCallback::getPlayer(color, verbose));
  774. }
  775. const TeamState * CGameInfoCallback::getTeam( ui8 teamID ) const
  776. {
  777. ERROR_RET_VAL_IF(!vstd::contains(gs->teams, teamID), "Cannot find info for team " << int(teamID), NULL);
  778. const TeamState *ret = &gs->teams[teamID];
  779. ERROR_RET_VAL_IF(player != -1 && !vstd::contains(ret->players, player), "Illegal attempt to access team data!", NULL);
  780. return ret;
  781. }
  782. const TeamState * CGameInfoCallback::getPlayerTeam( ui8 teamID ) const
  783. {
  784. const PlayerState * ps = getPlayer(teamID);
  785. if (ps)
  786. return getTeam(ps->team);
  787. return NULL;
  788. }
  789. const CGHeroInstance* CGameInfoCallback::getHeroWithSubid( int subid ) const
  790. {
  791. BOOST_FOREACH(const CGHeroInstance *h, gs->map->heroes)
  792. if(h->subID == subid)
  793. return h;
  794. return NULL;
  795. }
  796. int CGameInfoCallback::getLocalPlayer() const
  797. {
  798. return getCurrentPlayer();
  799. }
  800. bool CGameInfoCallback::isInTheMap(const int3 &pos) const
  801. {
  802. return gs->map->isInTheMap(pos);
  803. }
  804. void IGameEventRealizer::showInfoDialog( InfoWindow *iw )
  805. {
  806. commitPackage(iw);
  807. }
  808. void IGameEventRealizer::showInfoDialog(const std::string &msg, int player)
  809. {
  810. InfoWindow iw;
  811. iw.player = player;
  812. iw.text << msg;
  813. showInfoDialog(&iw);
  814. }
  815. void IGameEventRealizer::setObjProperty(int objid, int prop, si64 val)
  816. {
  817. SetObjectProperty sob;
  818. sob.id = objid;
  819. sob.what = prop;
  820. sob.val = static_cast<ui32>(val);
  821. commitPackage(&sob);
  822. }
  823. const CGObjectInstance * IGameCallback::putNewObject(int ID, int subID, int3 pos)
  824. {
  825. NewObject no;
  826. no.ID = ID; //creature
  827. no.subID= subID;
  828. no.pos = pos;
  829. commitPackage(&no);
  830. return getObj(no.id); //id field will be filled during applying on gs
  831. }
  832. const CGCreature * IGameCallback::putNewMonster(int creID, int count, int3 pos)
  833. {
  834. const CGObjectInstance *m = putNewObject(54, creID, pos);
  835. setObjProperty(m->id, ObjProperty::MONSTER_COUNT, count);
  836. setObjProperty(m->id, ObjProperty::MONSTER_POWER, (si64)1000*count);
  837. return dynamic_cast<const CGCreature*>(m);
  838. }