NetPacksLib.cpp 60 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829183018311832183318341835183618371838183918401841184218431844184518461847184818491850185118521853185418551856185718581859186018611862186318641865186618671868186918701871187218731874187518761877187818791880188118821883188418851886188718881889189018911892189318941895189618971898189919001901190219031904190519061907190819091910191119121913191419151916191719181919192019211922192319241925192619271928192919301931193219331934193519361937193819391940194119421943194419451946194719481949195019511952195319541955195619571958195919601961196219631964196519661967196819691970197119721973197419751976197719781979198019811982198319841985198619871988198919901991199219931994199519961997199819992000200120022003200420052006200720082009201020112012201320142015201620172018201920202021202220232024202520262027202820292030203120322033203420352036203720382039204020412042204320442045204620472048204920502051205220532054205520562057205820592060206120622063206420652066206720682069207020712072207320742075207620772078207920802081208220832084208520862087208820892090209120922093209420952096209720982099210021012102210321042105210621072108210921102111211221132114211521162117211821192120212121222123212421252126212721282129213021312132213321342135213621372138213921402141214221432144214521462147214821492150215121522153215421552156215721582159216021612162216321642165216621672168216921702171217221732174217521762177217821792180218121822183218421852186218721882189219021912192219321942195219621972198219922002201220222032204220522062207220822092210221122122213221422152216221722182219222022212222222322242225222622272228222922302231223222332234223522362237223822392240224122422243224422452246224722482249225022512252225322542255225622572258225922602261226222632264226522662267226822692270227122722273227422752276227722782279228022812282228322842285228622872288228922902291229222932294229522962297229822992300230123022303230423052306230723082309231023112312231323142315231623172318231923202321232223232324232523262327232823292330233123322333233423352336233723382339234023412342234323442345234623472348234923502351235223532354235523562357235823592360236123622363236423652366236723682369237023712372237323742375237623772378237923802381238223832384238523862387238823892390239123922393239423952396239723982399240024012402240324042405240624072408240924102411241224132414241524162417241824192420242124222423242424252426242724282429243024312432243324342435243624372438243924402441244224432444244524462447244824492450245124522453245424552456245724582459246024612462246324642465246624672468246924702471247224732474
  1. /*
  2. * NetPacksLib.cpp, part of VCMI engine
  3. *
  4. * Authors: listed in file AUTHORS in main folder
  5. *
  6. * License: GNU General Public License v2.0 or later
  7. * Full text of license available in license.txt file, in main folder
  8. *
  9. */
  10. #include "StdInc.h"
  11. #include "ArtifactUtils.h"
  12. #include "PacksForClient.h"
  13. #include "PacksForClientBattle.h"
  14. #include "PacksForServer.h"
  15. #include "SaveLocalState.h"
  16. #include "SetRewardableConfiguration.h"
  17. #include "StackLocation.h"
  18. #include "PacksForLobby.h"
  19. #include "SetStackEffect.h"
  20. #include "NetPackVisitor.h"
  21. #include "texts/CGeneralTextHandler.h"
  22. #include "CArtHandler.h"
  23. #include "GameLibrary.h"
  24. #include "mapping/CMap.h"
  25. #include "spells/CSpellHandler.h"
  26. #include "CCreatureHandler.h"
  27. #include "gameState/CGameState.h"
  28. #include "gameState/TavernHeroesPool.h"
  29. #include "CStack.h"
  30. #include "battle/BattleInfo.h"
  31. #include "mapping/CMapInfo.h"
  32. #include "StartInfo.h"
  33. #include "CPlayerState.h"
  34. #include "TerrainHandler.h"
  35. #include "entities/building/CBuilding.h"
  36. #include "entities/building/TownFortifications.h"
  37. #include "mapObjects/CGCreature.h"
  38. #include "mapObjects/CGMarket.h"
  39. #include "mapObjects/TownBuildingInstance.h"
  40. #include "mapObjects/CGTownInstance.h"
  41. #include "mapObjects/CQuest.h"
  42. #include "mapObjects/MiscObjects.h"
  43. #include "mapObjectConstructors/AObjectTypeHandler.h"
  44. #include "mapObjectConstructors/CObjectClassesHandler.h"
  45. #include "campaign/CampaignState.h"
  46. #include "IGameSettings.h"
  47. #include "mapObjects/FlaggableMapObject.h"
  48. VCMI_LIB_NAMESPACE_BEGIN
  49. void CPack::visit(ICPackVisitor & visitor)
  50. {
  51. visitBasic(visitor);
  52. // visitBasic may destroy this and in such cases we do not want to call visitTyped
  53. if(visitor.callTyped())
  54. {
  55. visitTyped(visitor);
  56. }
  57. }
  58. void CPack::visitBasic(ICPackVisitor & visitor)
  59. {
  60. }
  61. void CPack::visitTyped(ICPackVisitor & visitor)
  62. {
  63. }
  64. void CPackForClient::visitBasic(ICPackVisitor & visitor)
  65. {
  66. visitor.visitForClient(*this);
  67. }
  68. void CPackForServer::visitBasic(ICPackVisitor & visitor)
  69. {
  70. visitor.visitForServer(*this);
  71. }
  72. void CPackForLobby::visitBasic(ICPackVisitor & visitor)
  73. {
  74. visitor.visitForLobby(*this);
  75. }
  76. bool CPackForLobby::isForServer() const
  77. {
  78. return false;
  79. }
  80. bool CLobbyPackToServer::isForServer() const
  81. {
  82. return true;
  83. }
  84. void SaveLocalState::visitTyped(ICPackVisitor & visitor)
  85. {
  86. visitor.visitSaveLocalState(*this);
  87. }
  88. void PackageApplied::visitTyped(ICPackVisitor & visitor)
  89. {
  90. visitor.visitPackageApplied(*this);
  91. }
  92. void SystemMessage::visitTyped(ICPackVisitor & visitor)
  93. {
  94. visitor.visitSystemMessage(*this);
  95. }
  96. void PlayerBlocked::visitTyped(ICPackVisitor & visitor)
  97. {
  98. visitor.visitPlayerBlocked(*this);
  99. }
  100. void PlayerCheated::visitTyped(ICPackVisitor & visitor)
  101. {
  102. visitor.visitPlayerCheated(*this);
  103. }
  104. void PlayerStartsTurn::visitTyped(ICPackVisitor & visitor)
  105. {
  106. visitor.visitPlayerStartsTurn(*this);
  107. }
  108. void DaysWithoutTown::visitTyped(ICPackVisitor & visitor)
  109. {
  110. visitor.visitDaysWithoutTown(*this);
  111. }
  112. void EntitiesChanged::visitTyped(ICPackVisitor & visitor)
  113. {
  114. visitor.visitEntitiesChanged(*this);
  115. }
  116. void SetRewardableConfiguration::visitTyped(ICPackVisitor & visitor)
  117. {
  118. visitor.visitSetRewardableConfiguration(*this);
  119. }
  120. void SetResources::visitTyped(ICPackVisitor & visitor)
  121. {
  122. visitor.visitSetResources(*this);
  123. }
  124. void SetPrimSkill::visitTyped(ICPackVisitor & visitor)
  125. {
  126. visitor.visitSetPrimSkill(*this);
  127. }
  128. void SetSecSkill::visitTyped(ICPackVisitor & visitor)
  129. {
  130. visitor.visitSetSecSkill(*this);
  131. }
  132. void HeroVisitCastle::visitTyped(ICPackVisitor & visitor)
  133. {
  134. visitor.visitHeroVisitCastle(*this);
  135. }
  136. void ChangeSpells::visitTyped(ICPackVisitor & visitor)
  137. {
  138. visitor.visitChangeSpells(*this);
  139. }
  140. void SetResearchedSpells::visitTyped(ICPackVisitor & visitor)
  141. {
  142. visitor.visitSetResearchedSpells(*this);
  143. }
  144. void SetMana::visitTyped(ICPackVisitor & visitor)
  145. {
  146. visitor.visitSetMana(*this);
  147. }
  148. void SetMovePoints::visitTyped(ICPackVisitor & visitor)
  149. {
  150. visitor.visitSetMovePoints(*this);
  151. }
  152. void FoWChange::visitTyped(ICPackVisitor & visitor)
  153. {
  154. visitor.visitFoWChange(*this);
  155. }
  156. void SetAvailableHero::visitTyped(ICPackVisitor & visitor)
  157. {
  158. visitor.visitSetAvailableHeroes(*this);
  159. }
  160. void GiveBonus::visitTyped(ICPackVisitor & visitor)
  161. {
  162. visitor.visitGiveBonus(*this);
  163. }
  164. void ChangeObjPos::visitTyped(ICPackVisitor & visitor)
  165. {
  166. visitor.visitChangeObjPos(*this);
  167. }
  168. void PlayerEndsTurn::visitTyped(ICPackVisitor & visitor)
  169. {
  170. visitor.visitPlayerEndsTurn(*this);
  171. }
  172. void PlayerEndsGame::visitTyped(ICPackVisitor & visitor)
  173. {
  174. visitor.visitPlayerEndsGame(*this);
  175. }
  176. void PlayerReinitInterface::visitTyped(ICPackVisitor & visitor)
  177. {
  178. visitor.visitPlayerReinitInterface(*this);
  179. }
  180. void RemoveBonus::visitTyped(ICPackVisitor & visitor)
  181. {
  182. visitor.visitRemoveBonus(*this);
  183. }
  184. void SetCommanderProperty::visitTyped(ICPackVisitor & visitor)
  185. {
  186. visitor.visitSetCommanderProperty(*this);
  187. }
  188. void AddQuest::visitTyped(ICPackVisitor & visitor)
  189. {
  190. visitor.visitAddQuest(*this);
  191. }
  192. void UpdateArtHandlerLists::visitTyped(ICPackVisitor & visitor)
  193. {
  194. visitor.visitUpdateArtHandlerLists(*this);
  195. }
  196. void ChangeFormation::visitTyped(ICPackVisitor & visitor)
  197. {
  198. visitor.visitChangeFormation(*this);
  199. }
  200. void RemoveObject::visitTyped(ICPackVisitor & visitor)
  201. {
  202. visitor.visitRemoveObject(*this);
  203. }
  204. void TryMoveHero::visitTyped(ICPackVisitor & visitor)
  205. {
  206. visitor.visitTryMoveHero(*this);
  207. }
  208. void NewStructures::visitTyped(ICPackVisitor & visitor)
  209. {
  210. visitor.visitNewStructures(*this);
  211. }
  212. void RazeStructures::visitTyped(ICPackVisitor & visitor)
  213. {
  214. visitor.visitRazeStructures(*this);
  215. }
  216. void SetAvailableCreatures::visitTyped(ICPackVisitor & visitor)
  217. {
  218. visitor.visitSetAvailableCreatures(*this);
  219. }
  220. void SetHeroesInTown::visitTyped(ICPackVisitor & visitor)
  221. {
  222. visitor.visitSetHeroesInTown(*this);
  223. }
  224. void HeroRecruited::visitTyped(ICPackVisitor & visitor)
  225. {
  226. visitor.visitHeroRecruited(*this);
  227. }
  228. void GiveHero::visitTyped(ICPackVisitor & visitor)
  229. {
  230. visitor.visitGiveHero(*this);
  231. }
  232. void OpenWindow::visitTyped(ICPackVisitor & visitor)
  233. {
  234. visitor.visitOpenWindow(*this);
  235. }
  236. void NewObject::visitTyped(ICPackVisitor & visitor)
  237. {
  238. visitor.visitNewObject(*this);
  239. }
  240. void SetAvailableArtifacts::visitTyped(ICPackVisitor & visitor)
  241. {
  242. visitor.visitSetAvailableArtifacts(*this);
  243. }
  244. void NewArtifact::visitTyped(ICPackVisitor & visitor)
  245. {
  246. visitor.visitNewArtifact(*this);
  247. }
  248. void ChangeStackCount::visitTyped(ICPackVisitor & visitor)
  249. {
  250. visitor.visitChangeStackCount(*this);
  251. }
  252. void SetStackType::visitTyped(ICPackVisitor & visitor)
  253. {
  254. visitor.visitSetStackType(*this);
  255. }
  256. void EraseStack::visitTyped(ICPackVisitor & visitor)
  257. {
  258. visitor.visitEraseStack(*this);
  259. }
  260. void SwapStacks::visitTyped(ICPackVisitor & visitor)
  261. {
  262. visitor.visitSwapStacks(*this);
  263. }
  264. void InsertNewStack::visitTyped(ICPackVisitor & visitor)
  265. {
  266. visitor.visitInsertNewStack(*this);
  267. }
  268. void RebalanceStacks::visitTyped(ICPackVisitor & visitor)
  269. {
  270. visitor.visitRebalanceStacks(*this);
  271. }
  272. void BulkRebalanceStacks::visitTyped(ICPackVisitor & visitor)
  273. {
  274. visitor.visitBulkRebalanceStacks(*this);
  275. }
  276. void BulkSmartRebalanceStacks::visitTyped(ICPackVisitor & visitor)
  277. {
  278. visitor.visitBulkSmartRebalanceStacks(*this);
  279. }
  280. void PutArtifact::visitTyped(ICPackVisitor & visitor)
  281. {
  282. visitor.visitPutArtifact(*this);
  283. }
  284. void BulkEraseArtifacts::visitTyped(ICPackVisitor & visitor)
  285. {
  286. visitor.visitEraseArtifact(*this);
  287. }
  288. void BulkMoveArtifacts::visitTyped(ICPackVisitor & visitor)
  289. {
  290. visitor.visitBulkMoveArtifacts(*this);
  291. }
  292. void AssembledArtifact::visitTyped(ICPackVisitor & visitor)
  293. {
  294. visitor.visitAssembledArtifact(*this);
  295. }
  296. void DisassembledArtifact::visitTyped(ICPackVisitor & visitor)
  297. {
  298. visitor.visitDisassembledArtifact(*this);
  299. }
  300. void HeroVisit::visitTyped(ICPackVisitor & visitor)
  301. {
  302. visitor.visitHeroVisit(*this);
  303. }
  304. void NewTurn::visitTyped(ICPackVisitor & visitor)
  305. {
  306. visitor.visitNewTurn(*this);
  307. }
  308. void InfoWindow::visitTyped(ICPackVisitor & visitor)
  309. {
  310. visitor.visitInfoWindow(*this);
  311. }
  312. void SetObjectProperty::visitTyped(ICPackVisitor & visitor)
  313. {
  314. visitor.visitSetObjectProperty(*this);
  315. }
  316. void ChangeObjectVisitors::visitTyped(ICPackVisitor & visitor)
  317. {
  318. visitor.visitChangeObjectVisitors(*this);
  319. }
  320. void ChangeArtifactsCostume::visitTyped(ICPackVisitor & visitor)
  321. {
  322. visitor.visitChangeArtifactsCostume(*this);
  323. }
  324. void HeroLevelUp::visitTyped(ICPackVisitor & visitor)
  325. {
  326. visitor.visitHeroLevelUp(*this);
  327. }
  328. void CommanderLevelUp::visitTyped(ICPackVisitor & visitor)
  329. {
  330. visitor.visitCommanderLevelUp(*this);
  331. }
  332. void BlockingDialog::visitTyped(ICPackVisitor & visitor)
  333. {
  334. visitor.visitBlockingDialog(*this);
  335. }
  336. void GarrisonDialog::visitTyped(ICPackVisitor & visitor)
  337. {
  338. visitor.visitGarrisonDialog(*this);
  339. }
  340. void ExchangeDialog::visitTyped(ICPackVisitor & visitor)
  341. {
  342. visitor.visitExchangeDialog(*this);
  343. }
  344. void TeleportDialog::visitTyped(ICPackVisitor & visitor)
  345. {
  346. visitor.visitTeleportDialog(*this);
  347. }
  348. void MapObjectSelectDialog::visitTyped(ICPackVisitor & visitor)
  349. {
  350. visitor.visitMapObjectSelectDialog(*this);
  351. }
  352. void BattleStart::visitTyped(ICPackVisitor & visitor)
  353. {
  354. visitor.visitBattleStart(*this);
  355. }
  356. void BattleNextRound::visitTyped(ICPackVisitor & visitor)
  357. {
  358. visitor.visitBattleNextRound(*this);
  359. }
  360. void BattleSetActiveStack::visitTyped(ICPackVisitor & visitor)
  361. {
  362. visitor.visitBattleSetActiveStack(*this);
  363. }
  364. void BattleResult::visitTyped(ICPackVisitor & visitor)
  365. {
  366. visitor.visitBattleResult(*this);
  367. }
  368. void BattleLogMessage::visitTyped(ICPackVisitor & visitor)
  369. {
  370. visitor.visitBattleLogMessage(*this);
  371. }
  372. void BattleStackMoved::visitTyped(ICPackVisitor & visitor)
  373. {
  374. visitor.visitBattleStackMoved(*this);
  375. }
  376. void BattleUnitsChanged::visitTyped(ICPackVisitor & visitor)
  377. {
  378. visitor.visitBattleUnitsChanged(*this);
  379. }
  380. void BattleAttack::visitTyped(ICPackVisitor & visitor)
  381. {
  382. visitor.visitBattleAttack(*this);
  383. }
  384. void StartAction::visitTyped(ICPackVisitor & visitor)
  385. {
  386. visitor.visitStartAction(*this);
  387. }
  388. void EndAction::visitTyped(ICPackVisitor & visitor)
  389. {
  390. visitor.visitEndAction(*this);
  391. }
  392. void BattleSpellCast::visitTyped(ICPackVisitor & visitor)
  393. {
  394. visitor.visitBattleSpellCast(*this);
  395. }
  396. void SetStackEffect::visitTyped(ICPackVisitor & visitor)
  397. {
  398. visitor.visitSetStackEffect(*this);
  399. }
  400. void StacksInjured::visitTyped(ICPackVisitor & visitor)
  401. {
  402. visitor.visitStacksInjured(*this);
  403. }
  404. void BattleResultsApplied::visitTyped(ICPackVisitor & visitor)
  405. {
  406. visitor.visitBattleResultsApplied(*this);
  407. }
  408. void BattleObstaclesChanged::visitTyped(ICPackVisitor & visitor)
  409. {
  410. visitor.visitBattleObstaclesChanged(*this);
  411. }
  412. void BattleSetStackProperty::visitTyped(ICPackVisitor & visitor)
  413. {
  414. visitor.visitBattleSetStackProperty(*this);
  415. }
  416. void BattleTriggerEffect::visitTyped(ICPackVisitor & visitor)
  417. {
  418. visitor.visitBattleTriggerEffect(*this);
  419. }
  420. void BattleUpdateGateState::visitTyped(ICPackVisitor & visitor)
  421. {
  422. visitor.visitBattleUpdateGateState(*this);
  423. }
  424. void AdvmapSpellCast::visitTyped(ICPackVisitor & visitor)
  425. {
  426. visitor.visitAdvmapSpellCast(*this);
  427. }
  428. void ShowWorldViewEx::visitTyped(ICPackVisitor & visitor)
  429. {
  430. visitor.visitShowWorldViewEx(*this);
  431. }
  432. void EndTurn::visitTyped(ICPackVisitor & visitor)
  433. {
  434. visitor.visitEndTurn(*this);
  435. }
  436. void GamePause::visitTyped(ICPackVisitor & visitor)
  437. {
  438. visitor.visitGamePause(*this);
  439. }
  440. void DismissHero::visitTyped(ICPackVisitor & visitor)
  441. {
  442. visitor.visitDismissHero(*this);
  443. }
  444. void MoveHero::visitTyped(ICPackVisitor & visitor)
  445. {
  446. visitor.visitMoveHero(*this);
  447. }
  448. void CastleTeleportHero::visitTyped(ICPackVisitor & visitor)
  449. {
  450. visitor.visitCastleTeleportHero(*this);
  451. }
  452. void ArrangeStacks::visitTyped(ICPackVisitor & visitor)
  453. {
  454. visitor.visitArrangeStacks(*this);
  455. }
  456. void BulkMoveArmy::visitTyped(ICPackVisitor & visitor)
  457. {
  458. visitor.visitBulkMoveArmy(*this);
  459. }
  460. void BulkSplitStack::visitTyped(ICPackVisitor & visitor)
  461. {
  462. visitor.visitBulkSplitStack(*this);
  463. }
  464. void BulkMergeStacks::visitTyped(ICPackVisitor & visitor)
  465. {
  466. visitor.visitBulkMergeStacks(*this);
  467. }
  468. void BulkSmartSplitStack::visitTyped(ICPackVisitor & visitor)
  469. {
  470. visitor.visitBulkSmartSplitStack(*this);
  471. }
  472. void DisbandCreature::visitTyped(ICPackVisitor & visitor)
  473. {
  474. visitor.visitDisbandCreature(*this);
  475. }
  476. void BuildStructure::visitTyped(ICPackVisitor & visitor)
  477. {
  478. visitor.visitBuildStructure(*this);
  479. }
  480. void VisitTownBuilding::visitTyped(ICPackVisitor & visitor)
  481. {
  482. visitor.visitVisitTownBuilding(*this);
  483. }
  484. void RazeStructure::visitTyped(ICPackVisitor & visitor)
  485. {
  486. visitor.visitRazeStructure(*this);
  487. }
  488. void SpellResearch::visitTyped(ICPackVisitor & visitor)
  489. {
  490. visitor.visitSpellResearch(*this);
  491. }
  492. void RecruitCreatures::visitTyped(ICPackVisitor & visitor)
  493. {
  494. visitor.visitRecruitCreatures(*this);
  495. }
  496. void UpgradeCreature::visitTyped(ICPackVisitor & visitor)
  497. {
  498. visitor.visitUpgradeCreature(*this);
  499. }
  500. void GarrisonHeroSwap::visitTyped(ICPackVisitor & visitor)
  501. {
  502. visitor.visitGarrisonHeroSwap(*this);
  503. }
  504. void ExchangeArtifacts::visitTyped(ICPackVisitor & visitor)
  505. {
  506. visitor.visitExchangeArtifacts(*this);
  507. }
  508. void BulkExchangeArtifacts::visitTyped(ICPackVisitor & visitor)
  509. {
  510. visitor.visitBulkExchangeArtifacts(*this);
  511. }
  512. void ManageBackpackArtifacts::visitTyped(ICPackVisitor & visitor)
  513. {
  514. visitor.visitManageBackpackArtifacts(*this);
  515. }
  516. void ManageEquippedArtifacts::visitTyped(ICPackVisitor & visitor)
  517. {
  518. visitor.visitManageEquippedArtifacts(*this);
  519. }
  520. void AssembleArtifacts::visitTyped(ICPackVisitor & visitor)
  521. {
  522. visitor.visitAssembleArtifacts(*this);
  523. }
  524. void EraseArtifactByClient::visitTyped(ICPackVisitor & visitor)
  525. {
  526. visitor.visitEraseArtifactByClient(*this);
  527. }
  528. void BuyArtifact::visitTyped(ICPackVisitor & visitor)
  529. {
  530. visitor.visitBuyArtifact(*this);
  531. }
  532. void TradeOnMarketplace::visitTyped(ICPackVisitor & visitor)
  533. {
  534. visitor.visitTradeOnMarketplace(*this);
  535. }
  536. void SetFormation::visitTyped(ICPackVisitor & visitor)
  537. {
  538. visitor.visitSetFormation(*this);
  539. }
  540. void HireHero::visitTyped(ICPackVisitor & visitor)
  541. {
  542. visitor.visitHireHero(*this);
  543. }
  544. void BuildBoat::visitTyped(ICPackVisitor & visitor)
  545. {
  546. visitor.visitBuildBoat(*this);
  547. }
  548. void QueryReply::visitTyped(ICPackVisitor & visitor)
  549. {
  550. visitor.visitQueryReply(*this);
  551. }
  552. void MakeAction::visitTyped(ICPackVisitor & visitor)
  553. {
  554. visitor.visitMakeAction(*this);
  555. }
  556. void DigWithHero::visitTyped(ICPackVisitor & visitor)
  557. {
  558. visitor.visitDigWithHero(*this);
  559. }
  560. void CastAdvSpell::visitTyped(ICPackVisitor & visitor)
  561. {
  562. visitor.visitCastAdvSpell(*this);
  563. }
  564. void SaveGame::visitTyped(ICPackVisitor & visitor)
  565. {
  566. visitor.visitSaveGame(*this);
  567. }
  568. void PlayerMessage::visitTyped(ICPackVisitor & visitor)
  569. {
  570. visitor.visitPlayerMessage(*this);
  571. }
  572. void PlayerMessageClient::visitTyped(ICPackVisitor & visitor)
  573. {
  574. visitor.visitPlayerMessageClient(*this);
  575. }
  576. void CenterView::visitTyped(ICPackVisitor & visitor)
  577. {
  578. visitor.visitCenterView(*this);
  579. }
  580. void LobbyClientConnected::visitTyped(ICPackVisitor & visitor)
  581. {
  582. visitor.visitLobbyClientConnected(*this);
  583. }
  584. void LobbyClientDisconnected::visitTyped(ICPackVisitor & visitor)
  585. {
  586. visitor.visitLobbyClientDisconnected(*this);
  587. }
  588. void LobbyChatMessage::visitTyped(ICPackVisitor & visitor)
  589. {
  590. visitor.visitLobbyChatMessage(*this);
  591. }
  592. void LobbyGuiAction::visitTyped(ICPackVisitor & visitor)
  593. {
  594. visitor.visitLobbyGuiAction(*this);
  595. }
  596. void LobbyLoadProgress::visitTyped(ICPackVisitor & visitor)
  597. {
  598. visitor.visitLobbyLoadProgress(*this);
  599. }
  600. void LobbyRestartGame::visitTyped(ICPackVisitor & visitor)
  601. {
  602. visitor.visitLobbyRestartGame(*this);
  603. }
  604. void LobbyStartGame::visitTyped(ICPackVisitor & visitor)
  605. {
  606. visitor.visitLobbyStartGame(*this);
  607. }
  608. void LobbyPrepareStartGame::visitTyped(ICPackVisitor & visitor)
  609. {
  610. visitor.visitLobbyPrepareStartGame(*this);
  611. }
  612. void LobbyChangeHost::visitTyped(ICPackVisitor & visitor)
  613. {
  614. visitor.visitLobbyChangeHost(*this);
  615. }
  616. void LobbyUpdateState::visitTyped(ICPackVisitor & visitor)
  617. {
  618. visitor.visitLobbyUpdateState(*this);
  619. }
  620. void LobbySetMap::visitTyped(ICPackVisitor & visitor)
  621. {
  622. visitor.visitLobbySetMap(*this);
  623. }
  624. void LobbySetCampaign::visitTyped(ICPackVisitor & visitor)
  625. {
  626. visitor.visitLobbySetCampaign(*this);
  627. }
  628. void LobbySetCampaignMap::visitTyped(ICPackVisitor & visitor)
  629. {
  630. visitor.visitLobbySetCampaignMap(*this);
  631. }
  632. void LobbySetCampaignBonus::visitTyped(ICPackVisitor & visitor)
  633. {
  634. visitor.visitLobbySetCampaignBonus(*this);
  635. }
  636. void LobbyChangePlayerOption::visitTyped(ICPackVisitor & visitor)
  637. {
  638. visitor.visitLobbyChangePlayerOption(*this);
  639. }
  640. void LobbySetPlayer::visitTyped(ICPackVisitor & visitor)
  641. {
  642. visitor.visitLobbySetPlayer(*this);
  643. }
  644. void LobbySetPlayerName::visitTyped(ICPackVisitor & visitor)
  645. {
  646. visitor.visitLobbySetPlayerName(*this);
  647. }
  648. void LobbySetPlayerHandicap::visitTyped(ICPackVisitor & visitor)
  649. {
  650. visitor.visitLobbySetPlayerHandicap(*this);
  651. }
  652. void LobbySetSimturns::visitTyped(ICPackVisitor & visitor)
  653. {
  654. visitor.visitLobbySetSimturns(*this);
  655. }
  656. void LobbySetTurnTime::visitTyped(ICPackVisitor & visitor)
  657. {
  658. visitor.visitLobbySetTurnTime(*this);
  659. }
  660. void LobbySetExtraOptions::visitTyped(ICPackVisitor & visitor)
  661. {
  662. visitor.visitLobbySetExtraOptions(*this);
  663. }
  664. void LobbySetDifficulty::visitTyped(ICPackVisitor & visitor)
  665. {
  666. visitor.visitLobbySetDifficulty(*this);
  667. }
  668. void LobbyForceSetPlayer::visitTyped(ICPackVisitor & visitor)
  669. {
  670. visitor.visitLobbyForceSetPlayer(*this);
  671. }
  672. void LobbyShowMessage::visitTyped(ICPackVisitor & visitor)
  673. {
  674. visitor.visitLobbyShowMessage(*this);
  675. }
  676. void LobbyPvPAction::visitTyped(ICPackVisitor & visitor)
  677. {
  678. visitor.visitLobbyPvPAction(*this);
  679. }
  680. void LobbyDelete::visitTyped(ICPackVisitor & visitor)
  681. {
  682. visitor.visitLobbyDelete(*this);
  683. }
  684. void SetResources::applyGs(CGameState *gs)
  685. {
  686. assert(player.isValidPlayer());
  687. if(abs)
  688. gs->getPlayerState(player)->resources = res;
  689. else
  690. gs->getPlayerState(player)->resources += res;
  691. gs->getPlayerState(player)->resources.amin(GameConstants::PLAYER_RESOURCES_CAP);
  692. //just ensure that player resources are not negative
  693. //server is responsible to check if player can afford deal
  694. //but events on server side are allowed to take more than player have
  695. gs->getPlayerState(player)->resources.positive();
  696. }
  697. void SetPrimSkill::applyGs(CGameState *gs)
  698. {
  699. CGHeroInstance * hero = gs->getHero(id);
  700. assert(hero);
  701. hero->setPrimarySkill(which, val, abs);
  702. }
  703. void SetSecSkill::applyGs(CGameState *gs)
  704. {
  705. CGHeroInstance *hero = gs->getHero(id);
  706. hero->setSecSkillLevel(which, val, abs);
  707. }
  708. void SetCommanderProperty::applyGs(CGameState *gs)
  709. {
  710. const auto & commander = gs->getHero(heroid)->getCommander();
  711. assert (commander);
  712. switch (which)
  713. {
  714. case BONUS:
  715. commander->accumulateBonus (std::make_shared<Bonus>(accumulatedBonus));
  716. break;
  717. case SPECIAL_SKILL:
  718. commander->accumulateBonus (std::make_shared<Bonus>(accumulatedBonus));
  719. commander->specialSkills.insert (additionalInfo);
  720. break;
  721. case SECONDARY_SKILL:
  722. commander->secondarySkills[additionalInfo] = static_cast<ui8>(amount);
  723. break;
  724. case ALIVE:
  725. if (amount)
  726. commander->setAlive(true);
  727. else
  728. commander->setAlive(false);
  729. break;
  730. case EXPERIENCE:
  731. commander->giveStackExp(amount); //TODO: allow setting exp for stacks via netpacks
  732. commander->nodeHasChanged();
  733. break;
  734. }
  735. }
  736. void AddQuest::applyGs(CGameState *gs)
  737. {
  738. assert (vstd::contains(gs->players, player));
  739. auto * vec = &gs->players.at(player).quests;
  740. if (!vstd::contains(*vec, quest))
  741. vec->push_back (quest);
  742. else
  743. logNetwork->warn("Warning! Attempt to add duplicated quest");
  744. }
  745. void UpdateArtHandlerLists::applyGs(CGameState *gs)
  746. {
  747. gs->allocatedArtifacts = allocatedArtifacts;
  748. }
  749. void ChangeFormation::applyGs(CGameState *gs)
  750. {
  751. gs->getHero(hid)->setFormation(formation);
  752. }
  753. void HeroVisitCastle::applyGs(CGameState *gs)
  754. {
  755. CGHeroInstance *h = gs->getHero(hid);
  756. CGTownInstance *t = gs->getTown(tid);
  757. assert(h);
  758. assert(t);
  759. if(start())
  760. t->setVisitingHero(h);
  761. else
  762. t->setVisitingHero(nullptr);
  763. }
  764. void ChangeSpells::applyGs(CGameState *gs)
  765. {
  766. CGHeroInstance *hero = gs->getHero(hid);
  767. if(learn)
  768. for(const auto & sid : spells)
  769. hero->addSpellToSpellbook(sid);
  770. else
  771. for(const auto & sid : spells)
  772. hero->removeSpellFromSpellbook(sid);
  773. }
  774. void SetResearchedSpells::applyGs(CGameState *gs)
  775. {
  776. CGTownInstance *town = gs->getTown(tid);
  777. town->spells[level] = spells;
  778. town->spellResearchCounterDay++;
  779. if(accepted)
  780. town->spellResearchAcceptedCounter++;
  781. }
  782. void SetMana::applyGs(CGameState *gs)
  783. {
  784. CGHeroInstance * hero = gs->getHero(hid);
  785. assert(hero);
  786. if(absolute)
  787. hero->mana = val;
  788. else
  789. hero->mana += val;
  790. vstd::amax(hero->mana, 0); //not less than 0
  791. }
  792. void SetMovePoints::applyGs(CGameState *gs)
  793. {
  794. CGHeroInstance *hero = gs->getHero(hid);
  795. assert(hero);
  796. if(absolute)
  797. hero->setMovementPoints(val);
  798. else
  799. hero->setMovementPoints(hero->movementPointsRemaining() + val);
  800. }
  801. void FoWChange::applyGs(CGameState *gs)
  802. {
  803. TeamState * team = gs->getPlayerTeam(player);
  804. auto & fogOfWarMap = team->fogOfWarMap;
  805. for(const int3 & t : tiles)
  806. fogOfWarMap[t.z][t.x][t.y] = mode != ETileVisibility::HIDDEN;
  807. if (mode == ETileVisibility::HIDDEN) //do not hide too much
  808. {
  809. std::unordered_set<int3> tilesRevealed;
  810. for (auto & o : gs->getMap().getObjects())
  811. {
  812. if (o->asOwnable())
  813. {
  814. if(vstd::contains(team->players, o->getOwner())) //check owned observators
  815. gs->getTilesInRange(tilesRevealed, o->getSightCenter(), o->getSightRadius(), ETileVisibility::HIDDEN, o->tempOwner);
  816. }
  817. }
  818. for(const int3 & t : tilesRevealed) //probably not the most optimal solution ever
  819. fogOfWarMap[t.z][t.x][t.y] = 1;
  820. }
  821. }
  822. void SetAvailableHero::applyGs(CGameState *gs)
  823. {
  824. gs->heroesPool->setHeroForPlayer(player, slotID, hid, army, roleID, replenishPoints);
  825. }
  826. void GiveBonus::applyGs(CGameState *gs)
  827. {
  828. CBonusSystemNode *cbsn = nullptr;
  829. switch(who)
  830. {
  831. case ETarget::OBJECT:
  832. cbsn = dynamic_cast<CBonusSystemNode*>(gs->getObjInstance(id.as<ObjectInstanceID>()));
  833. break;
  834. case ETarget::PLAYER:
  835. cbsn = gs->getPlayerState(id.as<PlayerColor>());
  836. break;
  837. case ETarget::BATTLE:
  838. assert(Bonus::OneBattle(&bonus));
  839. cbsn = dynamic_cast<CBonusSystemNode*>(gs->getBattle(id.as<BattleID>()));
  840. break;
  841. }
  842. assert(cbsn);
  843. if(Bonus::OneWeek(&bonus))
  844. bonus.turnsRemain = 8 - gs->getDate(Date::DAY_OF_WEEK); // set correct number of days before adding bonus
  845. auto b = std::make_shared<Bonus>(bonus);
  846. cbsn->addNewBonus(b);
  847. }
  848. void ChangeObjPos::applyGs(CGameState *gs)
  849. {
  850. CGObjectInstance *obj = gs->getObjInstance(objid);
  851. if(!obj)
  852. {
  853. logNetwork->error("Wrong ChangeObjPos: object %d doesn't exist!", objid.getNum());
  854. return;
  855. }
  856. gs->getMap().moveObject(objid, nPos + obj->getVisitableOffset());
  857. }
  858. void ChangeObjectVisitors::applyGs(CGameState *gs)
  859. {
  860. switch (mode) {
  861. case VISITOR_ADD_HERO:
  862. gs->getPlayerTeam(gs->getHero(hero)->tempOwner)->scoutedObjects.insert(object);
  863. gs->getHero(hero)->visitedObjects.insert(object);
  864. gs->getPlayerState(gs->getHero(hero)->tempOwner)->visitedObjects.insert(object);
  865. break;
  866. case VISITOR_ADD_PLAYER:
  867. gs->getPlayerTeam(gs->getHero(hero)->tempOwner)->scoutedObjects.insert(object);
  868. for(const auto & color : gs->getPlayerTeam(gs->getHero(hero)->tempOwner)->players)
  869. gs->getPlayerState(color)->visitedObjects.insert(object);
  870. break;
  871. case VISITOR_CLEAR:
  872. // remove visit info from all heroes, including those that are not present on map
  873. for (auto heroID : gs->getMap().getHeroesOnMap())
  874. gs->getHero(heroID)->visitedObjects.erase(object);
  875. for (auto heroID : gs->getMap().getHeroesInPool())
  876. gs->getMap().tryGetFromHeroPool(heroID)->visitedObjects.erase(object);
  877. for(auto &elem : gs->players)
  878. elem.second.visitedObjects.erase(object);
  879. for(auto &elem : gs->teams)
  880. elem.second.scoutedObjects.erase(object);
  881. break;
  882. case VISITOR_SCOUTED:
  883. gs->getPlayerTeam(gs->getHero(hero)->tempOwner)->scoutedObjects.insert(object);
  884. break;
  885. case VISITOR_GLOBAL:
  886. {
  887. CGObjectInstance * objectPtr = gs->getObjInstance(object);
  888. gs->getPlayerState(gs->getHero(hero)->tempOwner)->visitedObjectsGlobal.insert({objectPtr->ID, objectPtr->subID});
  889. break;
  890. }
  891. }
  892. }
  893. void ChangeArtifactsCostume::applyGs(CGameState *gs)
  894. {
  895. auto & allCostumes = gs->getPlayerState(player)->costumesArtifacts;
  896. if(const auto & costume = allCostumes.find(costumeIdx); costume != allCostumes.end())
  897. costume->second = costumeSet;
  898. else
  899. allCostumes.try_emplace(costumeIdx, costumeSet);
  900. }
  901. void PlayerEndsGame::applyGs(CGameState *gs)
  902. {
  903. PlayerState *p = gs->getPlayerState(player);
  904. if(victoryLossCheckResult.victory())
  905. {
  906. p->status = EPlayerStatus::WINNER;
  907. // TODO: Campaign-specific code might as well go somewhere else
  908. // keep all heroes from the winning player
  909. if(p->human && gs->getStartInfo()->campState)
  910. {
  911. std::vector<CGHeroInstance *> crossoverHeroes;
  912. for (auto hero : p->getHeroes())
  913. if (hero->tempOwner == player)
  914. crossoverHeroes.push_back(hero);
  915. gs->getStartInfo()->campState->setCurrentMapAsConquered(crossoverHeroes);
  916. }
  917. }
  918. else
  919. {
  920. p->status = EPlayerStatus::LOSER;
  921. }
  922. // defeated player may be making turn right now
  923. gs->actingPlayers.erase(player);
  924. }
  925. void PlayerReinitInterface::applyGs(CGameState *gs)
  926. {
  927. if(!gs || !gs->getStartInfo())
  928. return;
  929. //TODO: what does mean if more that one player connected?
  930. if(playerConnectionId == PlayerSettings::PLAYER_AI)
  931. {
  932. for(const auto & player : players)
  933. gs->getStartInfo()->getIthPlayersSettings(player).connectedPlayerIDs.clear();
  934. }
  935. }
  936. void RemoveBonus::applyGs(CGameState *gs)
  937. {
  938. CBonusSystemNode *node = nullptr;
  939. switch(who)
  940. {
  941. case GiveBonus::ETarget::OBJECT:
  942. node = dynamic_cast<CBonusSystemNode*>(gs->getObjInstance(whoID.as<ObjectInstanceID>()));
  943. break;
  944. case GiveBonus::ETarget::PLAYER:
  945. node = gs->getPlayerState(whoID.as<PlayerColor>());
  946. break;
  947. case GiveBonus::ETarget::BATTLE:
  948. assert(Bonus::OneBattle(&bonus));
  949. node = dynamic_cast<CBonusSystemNode*>(gs->getBattle(whoID.as<BattleID>()));
  950. break;
  951. }
  952. BonusList &bonuses = node->getExportedBonusList();
  953. for(const auto & b : bonuses)
  954. {
  955. if(b->source == source && b->sid == id)
  956. {
  957. bonus = *b; //backup bonus (to show to interfaces later)
  958. node->removeBonus(b);
  959. break;
  960. }
  961. }
  962. }
  963. void RemoveObject::applyGs(CGameState *gs)
  964. {
  965. CGObjectInstance *obj = gs->getObjInstance(objectID);
  966. logGlobal->debug("removing object id=%d; address=%x; name=%s", objectID, (intptr_t)obj, obj->getObjectName());
  967. if (initiator.isValidPlayer())
  968. gs->getPlayerState(initiator)->destroyedObjects.insert(objectID);
  969. if(obj->getOwner().isValidPlayer())
  970. {
  971. gs->getPlayerState(obj->getOwner())->removeOwnedObject(obj); //object removed via map event or hero got beaten
  972. FlaggableMapObject* flaggableObject = dynamic_cast<FlaggableMapObject*>(obj);
  973. if(flaggableObject)
  974. {
  975. flaggableObject->markAsDeleted();
  976. }
  977. }
  978. if(obj->ID == Obj::HERO) //remove beaten hero
  979. {
  980. auto beatenHero = dynamic_cast<CGHeroInstance*>(obj);
  981. assert(beatenHero);
  982. auto * siegeNode = beatenHero->whereShouldBeAttachedOnSiege(gs);
  983. vstd::erase_if(beatenHero->artifactsInBackpack, [](const ArtSlotInfo& asi)
  984. {
  985. return asi.getArt()->getTypeId() == ArtifactID::GRAIL;
  986. });
  987. if(beatenHero->getVisitedTown())
  988. {
  989. if(beatenHero->getVisitedTown()->getGarrisonHero() == beatenHero)
  990. beatenHero->getVisitedTown()->setGarrisonedHero(nullptr);
  991. else
  992. beatenHero->getVisitedTown()->setVisitingHero(nullptr);
  993. beatenHero->setVisitedTown(nullptr, false);
  994. }
  995. beatenHero->detachFromBonusSystem(*gs);
  996. beatenHero->tempOwner = PlayerColor::NEUTRAL; //no one owns beaten hero
  997. // FIXME: workaround:
  998. // hero should be attached to siegeNode after battle
  999. // however this code might also be called on dismissing hero while in town
  1000. if (siegeNode && vstd::contains(beatenHero->getParentNodes(), siegeNode))
  1001. beatenHero->detachFrom(*siegeNode);
  1002. //If hero on Boat is removed, the Boat disappears
  1003. if(beatenHero->inBoat())
  1004. {
  1005. auto boat = beatenHero->getBoat();
  1006. beatenHero->setBoat(nullptr);
  1007. gs->getMap().eraseObject(boat->id);
  1008. }
  1009. auto beatenObject = gs->getMap().eraseObject(obj->id);
  1010. //return hero to the pool, so he may reappear in tavern
  1011. gs->heroesPool->addHeroToPool(beatenHero->getHeroTypeID());
  1012. gs->getMap().addToHeroPool(std::dynamic_pointer_cast<CGHeroInstance>(beatenObject));
  1013. return;
  1014. }
  1015. const auto * quest = dynamic_cast<const IQuestObject *>(obj);
  1016. if (quest)
  1017. {
  1018. for (auto &player : gs->players)
  1019. {
  1020. vstd::erase_if(player.second.quests, [obj](const QuestInfo & q){
  1021. return q.obj == obj->id;
  1022. });
  1023. }
  1024. }
  1025. gs->getMap().eraseObject(objectID);
  1026. gs->getMap().calculateGuardingGreaturePositions();//FIXME: excessive, update only affected tiles
  1027. }
  1028. static int getDir(const int3 & src, const int3 & dst)
  1029. {
  1030. int ret = -1;
  1031. if(dst.x+1 == src.x && dst.y+1 == src.y) //tl
  1032. {
  1033. ret = 1;
  1034. }
  1035. else if(dst.x == src.x && dst.y+1 == src.y) //t
  1036. {
  1037. ret = 2;
  1038. }
  1039. else if(dst.x-1 == src.x && dst.y+1 == src.y) //tr
  1040. {
  1041. ret = 3;
  1042. }
  1043. else if(dst.x-1 == src.x && dst.y == src.y) //r
  1044. {
  1045. ret = 4;
  1046. }
  1047. else if(dst.x-1 == src.x && dst.y-1 == src.y) //br
  1048. {
  1049. ret = 5;
  1050. }
  1051. else if(dst.x == src.x && dst.y-1 == src.y) //b
  1052. {
  1053. ret = 6;
  1054. }
  1055. else if(dst.x+1 == src.x && dst.y-1 == src.y) //bl
  1056. {
  1057. ret = 7;
  1058. }
  1059. else if(dst.x+1 == src.x && dst.y == src.y) //l
  1060. {
  1061. ret = 8;
  1062. }
  1063. return ret;
  1064. }
  1065. void TryMoveHero::applyGs(CGameState *gs)
  1066. {
  1067. CGHeroInstance *h = gs->getHero(id);
  1068. if (!h)
  1069. {
  1070. logGlobal->error("Attempt ot move unavailable hero %d", id.getNum());
  1071. return;
  1072. }
  1073. const TerrainTile & fromTile = gs->getMap().getTile(h->convertToVisitablePos(start));
  1074. const TerrainTile & destTile = gs->getMap().getTile(h->convertToVisitablePos(end));
  1075. h->setMovementPoints(movePoints);
  1076. if((result == SUCCESS || result == BLOCKING_VISIT || result == EMBARK || result == DISEMBARK) && start != end)
  1077. {
  1078. auto dir = getDir(start,end);
  1079. if(dir > 0 && dir <= 8)
  1080. h->moveDir = dir;
  1081. //else don`t change move direction - hero might have traversed the subterranean gate, direction should be kept
  1082. }
  1083. if(result == EMBARK) //hero enters boat at destination tile
  1084. {
  1085. const TerrainTile &tt = gs->getMap().getTile(h->convertToVisitablePos(end));
  1086. ObjectInstanceID topObjectID = tt.visitableObjects.back();
  1087. CGObjectInstance * topObject = gs->getObjInstance(topObjectID);
  1088. assert(tt.visitableObjects.size() >= 1 && topObject->ID == Obj::BOAT); //the only visitable object at destination is Boat
  1089. auto * boat = dynamic_cast<CGBoat *>(topObject);
  1090. assert(boat);
  1091. gs->getMap().hideObject(boat); //hero blockvis mask will be used, we don't need to duplicate it with boat
  1092. h->setBoat(boat);
  1093. }
  1094. else if(result == DISEMBARK) //hero leaves boat to destination tile
  1095. {
  1096. auto * b = h->getBoat();
  1097. b->direction = h->moveDir;
  1098. b->pos = start;
  1099. gs->getMap().showObject(b);
  1100. h->setBoat(nullptr);
  1101. }
  1102. if(start!=end && (result == SUCCESS || result == TELEPORTATION || result == EMBARK || result == DISEMBARK))
  1103. {
  1104. gs->getMap().hideObject(h);
  1105. h->setAnchorPos(end);
  1106. if(auto * b = h->getBoat())
  1107. b->setAnchorPos(end);
  1108. gs->getMap().showObject(h);
  1109. }
  1110. auto & fogOfWarMap = gs->getPlayerTeam(h->getOwner())->fogOfWarMap;
  1111. for(const int3 & t : fowRevealed)
  1112. fogOfWarMap[t.z][t.x][t.y] = 1;
  1113. if (fromTile.getTerrainID() != destTile.getTerrainID())
  1114. h->nodeHasChanged(); // update bonuses with terrain limiter
  1115. }
  1116. void NewStructures::applyGs(CGameState *gs)
  1117. {
  1118. CGTownInstance *t = gs->getTown(tid);
  1119. for(const auto & id : bid)
  1120. {
  1121. assert(t->getTown()->buildings.at(id) != nullptr);
  1122. t->addBuilding(id);
  1123. }
  1124. t->updateAppearance();
  1125. t->built = built;
  1126. t->recreateBuildingsBonuses();
  1127. }
  1128. void RazeStructures::applyGs(CGameState *gs)
  1129. {
  1130. CGTownInstance *t = gs->getTown(tid);
  1131. for(const auto & id : bid)
  1132. {
  1133. t->removeBuilding(id);
  1134. t->updateAppearance();
  1135. }
  1136. t->destroyed = destroyed; //yeaha
  1137. t->recreateBuildingsBonuses();
  1138. }
  1139. void SetAvailableCreatures::applyGs(CGameState *gs)
  1140. {
  1141. auto * dw = dynamic_cast<CGDwelling *>(gs->getObjInstance(tid));
  1142. assert(dw);
  1143. dw->creatures = creatures;
  1144. }
  1145. void SetHeroesInTown::applyGs(CGameState *gs)
  1146. {
  1147. CGTownInstance *t = gs->getTown(tid);
  1148. CGHeroInstance * v = gs->getHero(visiting);
  1149. CGHeroInstance * g = gs->getHero(garrison);
  1150. bool newVisitorComesFromGarrison = v && v == t->getGarrisonHero();
  1151. bool newGarrisonComesFromVisiting = g && g == t->getVisitingHero();
  1152. if(newVisitorComesFromGarrison)
  1153. t->setGarrisonedHero(nullptr);
  1154. if(newGarrisonComesFromVisiting)
  1155. t->setVisitingHero(nullptr);
  1156. if(!newGarrisonComesFromVisiting || v)
  1157. t->setVisitingHero(v);
  1158. if(!newVisitorComesFromGarrison || g)
  1159. t->setGarrisonedHero(g);
  1160. if(v)
  1161. gs->getMap().showObject(v);
  1162. if(g)
  1163. gs->getMap().hideObject(g);
  1164. }
  1165. void HeroRecruited::applyGs(CGameState *gs)
  1166. {
  1167. auto h = gs->heroesPool->takeHeroFromPool(hid);
  1168. CGTownInstance *t = gs->getTown(tid);
  1169. PlayerState *p = gs->getPlayerState(player);
  1170. if (boatId != ObjectInstanceID::NONE)
  1171. {
  1172. CGObjectInstance *obj = gs->getObjInstance(boatId);
  1173. auto * boat = dynamic_cast<CGBoat *>(obj);
  1174. if (boat)
  1175. {
  1176. gs->getMap().hideObject(boat);
  1177. h->setBoat(boat);
  1178. }
  1179. }
  1180. h->setOwner(player);
  1181. h->pos = tile;
  1182. h->updateAppearance();
  1183. assert(h->id.hasValue());
  1184. gs->getMap().addNewObject(h);
  1185. p->addOwnedObject(h.get());
  1186. h->attachToBonusSystem(*gs);
  1187. if(t)
  1188. t->setVisitingHero(h.get());
  1189. }
  1190. void GiveHero::applyGs(CGameState *gs)
  1191. {
  1192. CGHeroInstance *h = gs->getHero(id);
  1193. if (boatId != ObjectInstanceID::NONE)
  1194. {
  1195. CGObjectInstance *obj = gs->getObjInstance(boatId);
  1196. auto * boat = dynamic_cast<CGBoat *>(obj);
  1197. if (boat)
  1198. {
  1199. gs->getMap().hideObject(boat);
  1200. h->setBoat(boat);
  1201. }
  1202. }
  1203. //bonus system
  1204. h->detachFrom(gs->globalEffects);
  1205. h->attachTo(*gs->getPlayerState(player));
  1206. auto oldVisitablePos = h->visitablePos();
  1207. gs->getMap().hideObject(h);
  1208. h->updateAppearance();
  1209. h->setOwner(player);
  1210. h->setMovementPoints(h->movementPointsLimit(true));
  1211. h->setAnchorPos(h->convertFromVisitablePos(oldVisitablePos));
  1212. gs->getMap().heroAddedToMap(h);
  1213. gs->getPlayerState(h->getOwner())->addOwnedObject(h);
  1214. gs->getMap().showObject(h);
  1215. h->setVisitedTown(nullptr, false);
  1216. }
  1217. void NewObject::applyGs(CGameState *gs)
  1218. {
  1219. gs->getMap().addNewObject(newObject);
  1220. gs->getMap().calculateGuardingGreaturePositions();
  1221. // attach newly spawned wandering monster to global bonus system node
  1222. auto newArmy = std::dynamic_pointer_cast<CArmedInstance>(newObject);
  1223. if (newArmy)
  1224. newArmy->attachToBonusSystem(*gs);
  1225. logGlobal->debug("Added object id=%d; name=%s", newObject->id, newObject->getObjectName());
  1226. }
  1227. void NewArtifact::applyGs(CGameState *gs)
  1228. {
  1229. auto art = gs->createArtifact(artId, spellId);
  1230. PutArtifact pa(art->getId(), ArtifactLocation(artHolder, pos), false);
  1231. pa.applyGs(gs);
  1232. }
  1233. void ChangeStackCount::applyGs(CGameState *gs)
  1234. {
  1235. auto * srcObj = gs->getArmyInstance(army);
  1236. if(!srcObj)
  1237. throw std::runtime_error("ChangeStackCount: invalid army object " + std::to_string(army.getNum()) + ", possible game state corruption.");
  1238. if(absoluteValue)
  1239. srcObj->setStackCount(slot, count);
  1240. else
  1241. srcObj->changeStackCount(slot, count);
  1242. }
  1243. void SetStackType::applyGs(CGameState *gs)
  1244. {
  1245. auto * srcObj = gs->getArmyInstance(army);
  1246. if(!srcObj)
  1247. throw std::runtime_error("SetStackType: invalid army object " + std::to_string(army.getNum()) + ", possible game state corruption.");
  1248. srcObj->setStackType(slot, type);
  1249. }
  1250. void EraseStack::applyGs(CGameState *gs)
  1251. {
  1252. auto * srcObj = gs->getArmyInstance(army);
  1253. if(!srcObj)
  1254. throw std::runtime_error("EraseStack: invalid army object " + std::to_string(army.getNum()) + ", possible game state corruption.");
  1255. srcObj->eraseStack(slot);
  1256. }
  1257. void SwapStacks::applyGs(CGameState *gs)
  1258. {
  1259. auto * srcObj = gs->getArmyInstance(srcArmy);
  1260. if(!srcObj)
  1261. throw std::runtime_error("SwapStacks: invalid army object " + std::to_string(srcArmy.getNum()) + ", possible game state corruption.");
  1262. auto * dstObj = gs->getArmyInstance(dstArmy);
  1263. if(!dstObj)
  1264. throw std::runtime_error("SwapStacks: invalid army object " + std::to_string(dstArmy.getNum()) + ", possible game state corruption.");
  1265. auto s1 = srcObj->detachStack(srcSlot);
  1266. auto s2 = dstObj->detachStack(dstSlot);
  1267. srcObj->putStack(srcSlot, std::move(s2));
  1268. dstObj->putStack(dstSlot, std::move(s1));
  1269. }
  1270. void InsertNewStack::applyGs(CGameState *gs)
  1271. {
  1272. if(auto * obj = gs->getArmyInstance(army))
  1273. obj->putStack(slot, std::make_unique<CStackInstance>(gs->cb, type, count));
  1274. else
  1275. throw std::runtime_error("InsertNewStack: invalid army object " + std::to_string(army.getNum()) + ", possible game state corruption.");
  1276. }
  1277. void RebalanceStacks::applyGs(CGameState *gs)
  1278. {
  1279. auto * srcObj = gs->getArmyInstance(srcArmy);
  1280. if(!srcObj)
  1281. throw std::runtime_error("RebalanceStacks: invalid army object " + std::to_string(srcArmy.getNum()) + ", possible game state corruption.");
  1282. auto * dstObj = gs->getArmyInstance(dstArmy);
  1283. if(!dstObj)
  1284. throw std::runtime_error("RebalanceStacks: invalid army object " + std::to_string(dstArmy.getNum()) + ", possible game state corruption.");
  1285. StackLocation src(srcObj->id, srcSlot);
  1286. StackLocation dst(dstObj->id, dstSlot);
  1287. const CCreature * srcType = srcObj->getCreature(src.slot);
  1288. TQuantity srcCount = srcObj->getStackCount(src.slot);
  1289. bool stackExp = gs->getSettings().getBoolean(EGameSettings::MODULE_STACK_EXPERIENCE);
  1290. if(srcCount == count) //moving whole stack
  1291. {
  1292. const auto c = dstObj->getCreature(dst.slot);
  1293. if(c) //stack at dest -> merge
  1294. {
  1295. assert(c == srcType);
  1296. const auto srcHero = dynamic_cast<CGHeroInstance*>(srcObj);
  1297. const auto dstHero = dynamic_cast<CGHeroInstance*>(dstObj);
  1298. auto srcStack = const_cast<CStackInstance*>(srcObj->getStackPtr(src.slot));
  1299. auto dstStack = const_cast<CStackInstance*>(dstObj->getStackPtr(dst.slot));
  1300. if(srcStack->getArt(ArtifactPosition::CREATURE_SLOT))
  1301. {
  1302. if(auto dstArt = dstStack->getArt(ArtifactPosition::CREATURE_SLOT))
  1303. {
  1304. bool artifactIsLost = true;
  1305. if(srcHero)
  1306. {
  1307. auto dstSlot = ArtifactUtils::getArtBackpackPosition(srcHero, dstArt->getTypeId());
  1308. if (dstSlot != ArtifactPosition::PRE_FIRST)
  1309. {
  1310. gs->getMap().moveArtifactInstance(*dstStack, ArtifactPosition::CREATURE_SLOT, *srcHero, dstSlot);
  1311. artifactIsLost = false;
  1312. }
  1313. }
  1314. if (artifactIsLost)
  1315. {
  1316. BulkEraseArtifacts ea;
  1317. ea.artHolder = dstHero->id;
  1318. ea.posPack.emplace_back(ArtifactPosition::CREATURE_SLOT);
  1319. ea.creature = dst.slot;
  1320. ea.applyGs(gs);
  1321. logNetwork->warn("Cannot move artifact! No free slots");
  1322. }
  1323. gs->getMap().moveArtifactInstance(*srcStack, ArtifactPosition::CREATURE_SLOT, *dstStack, ArtifactPosition::CREATURE_SLOT);
  1324. //TODO: choose from dialog
  1325. }
  1326. else //just move to the other slot before stack gets erased
  1327. {
  1328. gs->getMap().moveArtifactInstance(*srcStack, ArtifactPosition::CREATURE_SLOT, *dstStack, ArtifactPosition::CREATURE_SLOT);
  1329. }
  1330. }
  1331. if (stackExp)
  1332. {
  1333. ui64 totalExp = srcCount * srcObj->getStackExperience(src.slot) + dstObj->getStackCount(dst.slot) * dstObj->getStackExperience(dst.slot);
  1334. srcObj->eraseStack(src.slot);
  1335. dstObj->changeStackCount(dst.slot, count);
  1336. dstObj->setStackExp(dst.slot, totalExp /(dstObj->getStackCount(dst.slot))); //mean
  1337. }
  1338. else
  1339. {
  1340. srcObj->eraseStack(src.slot);
  1341. dstObj->changeStackCount(dst.slot, count);
  1342. }
  1343. }
  1344. else //move stack to an empty slot, no exp change needed
  1345. {
  1346. auto stackDetached = srcObj->detachStack(src.slot);
  1347. dstObj->putStack(dst.slot, std::move(stackDetached));
  1348. }
  1349. }
  1350. else
  1351. {
  1352. [[maybe_unused]] const CCreature *c = dstObj->getCreature(dst.slot);
  1353. if(c) //stack at dest -> rebalance
  1354. {
  1355. assert(c == srcType);
  1356. if (stackExp)
  1357. {
  1358. ui64 totalExp = srcCount * srcObj->getStackExperience(src.slot) + dstObj->getStackCount(dst.slot) * dstObj->getStackExperience(dst.slot);
  1359. srcObj->changeStackCount(src.slot, -count);
  1360. dstObj->changeStackCount(dst.slot, count);
  1361. dstObj->setStackExp(dst.slot, totalExp /(srcObj->getStackCount(src.slot) + dstObj->getStackCount(dst.slot))); //mean
  1362. }
  1363. else
  1364. {
  1365. srcObj->changeStackCount(src.slot, -count);
  1366. dstObj->changeStackCount(dst.slot, count);
  1367. }
  1368. }
  1369. else //split stack to an empty slot
  1370. {
  1371. srcObj->changeStackCount(src.slot, -count);
  1372. dstObj->addToSlot(dst.slot, srcType->getId(), count, false);
  1373. if (stackExp)
  1374. dstObj->setStackExp(dst.slot, srcObj->getStackExperience(src.slot));
  1375. }
  1376. }
  1377. srcObj->nodeHasChanged();
  1378. if (srcObj != dstObj)
  1379. dstObj->nodeHasChanged();
  1380. }
  1381. void BulkRebalanceStacks::applyGs(CGameState *gs)
  1382. {
  1383. for(auto & move : moves)
  1384. move.applyGs(gs);
  1385. }
  1386. void BulkSmartRebalanceStacks::applyGs(CGameState *gs)
  1387. {
  1388. for(auto & move : moves)
  1389. move.applyGs(gs);
  1390. for(auto & change : changes)
  1391. change.applyGs(gs);
  1392. }
  1393. void PutArtifact::applyGs(CGameState *gs)
  1394. {
  1395. auto art = gs->getArtInstance(id);
  1396. assert(!art->getParentNodes().empty());
  1397. auto hero = gs->getHero(al.artHolder);
  1398. assert(hero);
  1399. assert(art && art->canBePutAt(hero, al.slot));
  1400. assert(ArtifactUtils::checkIfSlotValid(*hero, al.slot));
  1401. gs->getMap().putArtifactInstance(*hero, art->getId(), al.slot);
  1402. }
  1403. void BulkEraseArtifacts::applyGs(CGameState *gs)
  1404. {
  1405. const auto artSet = gs->getArtSet(artHolder);
  1406. assert(artSet);
  1407. std::sort(posPack.begin(), posPack.end(), [](const ArtifactPosition & slot0, const ArtifactPosition & slot1) -> bool
  1408. {
  1409. return slot0.num > slot1.num;
  1410. });
  1411. for(const auto & slot : posPack)
  1412. {
  1413. const auto slotInfo = artSet->getSlot(slot);
  1414. const ArtifactInstanceID artifactID = slotInfo->artifactID;
  1415. const CArtifactInstance * artifact = gs->getArtInstance(artifactID);
  1416. if(slotInfo->locked)
  1417. {
  1418. logGlobal->debug("Erasing locked artifact: %s", artifact->getType()->getNameTranslated());
  1419. DisassembledArtifact dis;
  1420. dis.al.artHolder = artHolder;
  1421. for(auto & slotInfoWorn : artSet->artifactsWorn)
  1422. {
  1423. auto art = slotInfoWorn.second.getArt();
  1424. if(art->isCombined() && art->isPart(artifact))
  1425. {
  1426. dis.al.slot = artSet->getArtPos(art);
  1427. break;
  1428. }
  1429. }
  1430. assert((dis.al.slot != ArtifactPosition::PRE_FIRST) && "Failed to determine the assembly this locked artifact belongs to");
  1431. logGlobal->debug("Found the corresponding assembly: %s", artSet->getArt(dis.al.slot)->getType()->getNameTranslated());
  1432. dis.applyGs(gs);
  1433. }
  1434. else
  1435. {
  1436. logGlobal->debug("Erasing artifact %s", artifact->getType()->getNameTranslated());
  1437. }
  1438. gs->getMap().removeArtifactInstance(*artSet, slot);
  1439. }
  1440. }
  1441. void BulkMoveArtifacts::applyGs(CGameState *gs)
  1442. {
  1443. const auto bulkArtsRemove = [gs](std::vector<MoveArtifactInfo> & artsPack, CArtifactSet & artSet)
  1444. {
  1445. std::vector<ArtifactPosition> packToRemove;
  1446. for(const auto & slotsPair : artsPack)
  1447. packToRemove.push_back(slotsPair.srcPos);
  1448. std::sort(packToRemove.begin(), packToRemove.end(), [](const ArtifactPosition & slot0, const ArtifactPosition & slot1) -> bool
  1449. {
  1450. return slot0.num > slot1.num;
  1451. });
  1452. for(const auto & slot : packToRemove)
  1453. gs->getMap().removeArtifactInstance(artSet, slot);
  1454. };
  1455. const auto bulkArtsPut = [gs](std::vector<MoveArtifactInfo> & artsPack, CArtifactSet & initArtSet, CArtifactSet & dstArtSet)
  1456. {
  1457. for(const auto & slotsPair : artsPack)
  1458. {
  1459. auto * art = initArtSet.getArt(slotsPair.srcPos);
  1460. assert(art);
  1461. gs->getMap().putArtifactInstance(dstArtSet, art->getId(), slotsPair.dstPos);
  1462. }
  1463. };
  1464. auto * leftSet = gs->getArtSet(ArtifactLocation(srcArtHolder, srcCreature));
  1465. assert(leftSet);
  1466. auto * rightSet = gs->getArtSet(ArtifactLocation(dstArtHolder, dstCreature));
  1467. assert(rightSet);
  1468. CArtifactFittingSet artInitialSetLeft(*leftSet);
  1469. bulkArtsRemove(artsPack0, *leftSet);
  1470. if(!artsPack1.empty())
  1471. {
  1472. CArtifactFittingSet artInitialSetRight(*rightSet);
  1473. bulkArtsRemove(artsPack1, *rightSet);
  1474. bulkArtsPut(artsPack1, artInitialSetRight, *leftSet);
  1475. }
  1476. bulkArtsPut(artsPack0, artInitialSetLeft, *rightSet);
  1477. }
  1478. void AssembledArtifact::applyGs(CGameState *gs)
  1479. {
  1480. auto artSet = gs->getArtSet(al.artHolder);
  1481. assert(artSet);
  1482. const auto transformedArt = artSet->getArt(al.slot);
  1483. assert(transformedArt);
  1484. const auto builtArt = artId.toArtifact();
  1485. assert(vstd::contains_if(ArtifactUtils::assemblyPossibilities(artSet, transformedArt->getTypeId()), [=](const CArtifact * art)->bool
  1486. {
  1487. return art->getId() == builtArt->getId();
  1488. }));
  1489. auto * combinedArt = gs->getMap().createArtifactComponent(artId);
  1490. // Find slots for all involved artifacts
  1491. std::set<ArtifactPosition, std::greater<>> slotsInvolved = { al.slot };
  1492. CArtifactFittingSet fittingSet(*artSet);
  1493. auto parts = builtArt->getConstituents();
  1494. parts.erase(std::find(parts.begin(), parts.end(), transformedArt->getType()));
  1495. for(const auto constituent : parts)
  1496. {
  1497. const auto slot = fittingSet.getArtPos(constituent->getId(), false, false);
  1498. fittingSet.lockSlot(slot);
  1499. assert(slot != ArtifactPosition::PRE_FIRST);
  1500. slotsInvolved.insert(slot);
  1501. }
  1502. // Find a slot for combined artifact
  1503. if(ArtifactUtils::isSlotEquipment(al.slot) && ArtifactUtils::isSlotBackpack(*slotsInvolved.begin()))
  1504. {
  1505. al.slot = ArtifactPosition::BACKPACK_START;
  1506. }
  1507. else if(ArtifactUtils::isSlotBackpack(al.slot))
  1508. {
  1509. for(const auto & slot : slotsInvolved)
  1510. if(ArtifactUtils::isSlotBackpack(slot))
  1511. al.slot = slot;
  1512. }
  1513. else
  1514. {
  1515. for(const auto & slot : slotsInvolved)
  1516. if(!vstd::contains(builtArt->getPossibleSlots().at(artSet->bearerType()), al.slot)
  1517. && vstd::contains(builtArt->getPossibleSlots().at(artSet->bearerType()), slot))
  1518. {
  1519. al.slot = slot;
  1520. break;
  1521. }
  1522. }
  1523. // Delete parts from hero
  1524. for(const auto & slot : slotsInvolved)
  1525. {
  1526. const auto constituentInstance = artSet->getArt(slot);
  1527. gs->getMap().removeArtifactInstance(*artSet, slot);
  1528. if(!combinedArt->getType()->isFused())
  1529. {
  1530. if(ArtifactUtils::isSlotEquipment(al.slot) && slot != al.slot)
  1531. combinedArt->addPart(constituentInstance, slot);
  1532. else
  1533. combinedArt->addPart(constituentInstance, ArtifactPosition::PRE_FIRST);
  1534. }
  1535. }
  1536. // Put new combined artifacts
  1537. gs->getMap().putArtifactInstance(*artSet, combinedArt->getId(), al.slot);
  1538. }
  1539. void DisassembledArtifact::applyGs(CGameState *gs)
  1540. {
  1541. auto hero = gs->getHero(al.artHolder);
  1542. assert(hero);
  1543. auto disassembledArtID = hero->getArtID(al.slot);
  1544. auto disassembledArt = gs->getArtInstance(disassembledArtID);
  1545. assert(disassembledArt);
  1546. const auto parts = disassembledArt->getPartsInfo();
  1547. gs->getMap().removeArtifactInstance(*hero, al.slot);
  1548. for(auto & part : parts)
  1549. {
  1550. // ArtifactPosition::PRE_FIRST is value of main part slot -> it'll replace combined artifact in its pos
  1551. auto slot = (ArtifactUtils::isSlotEquipment(part.slot) ? part.slot : al.slot);
  1552. disassembledArt->detachFromSource(*part.getArtifact());
  1553. gs->getMap().putArtifactInstance(*hero, part.getArtifact()->getId(), slot);
  1554. }
  1555. gs->getMap().eraseArtifactInstance(disassembledArt->getId());
  1556. }
  1557. void HeroVisit::applyGs(CGameState *gs)
  1558. {
  1559. }
  1560. void SetAvailableArtifacts::applyGs(CGameState *gs)
  1561. {
  1562. if(id != ObjectInstanceID::NONE)
  1563. {
  1564. if(auto * bm = dynamic_cast<CGBlackMarket *>(gs->getObjInstance(id)))
  1565. {
  1566. bm->artifacts = arts;
  1567. }
  1568. else
  1569. {
  1570. logNetwork->error("Wrong black market id!");
  1571. }
  1572. }
  1573. else
  1574. {
  1575. gs->getMap().townMerchantArtifacts = arts;
  1576. }
  1577. }
  1578. void NewTurn::applyGs(CGameState *gs)
  1579. {
  1580. gs->day = day;
  1581. // Update bonuses before doing anything else so hero don't get more MP than needed
  1582. gs->globalEffects.removeBonusesRecursive(Bonus::OneDay); //works for children -> all game objs
  1583. gs->globalEffects.reduceBonusDurations(Bonus::NDays);
  1584. gs->globalEffects.reduceBonusDurations(Bonus::OneWeek);
  1585. //TODO not really a single root hierarchy, what about bonuses placed elsewhere? [not an issue with H3 mechanics but in the future...]
  1586. for(auto & manaPack : heroesMana)
  1587. manaPack.applyGs(gs);
  1588. for(auto & movePack : heroesMovement)
  1589. movePack.applyGs(gs);
  1590. gs->heroesPool->onNewDay();
  1591. for(auto & entry : playerIncome)
  1592. {
  1593. gs->getPlayerState(entry.first)->resources += entry.second;
  1594. gs->getPlayerState(entry.first)->resources.amin(GameConstants::PLAYER_RESOURCES_CAP);
  1595. }
  1596. for(auto & creatureSet : availableCreatures) //set available creatures in towns
  1597. creatureSet.applyGs(gs);
  1598. for (const auto & townID : gs->getMap().getAllTowns())
  1599. {
  1600. auto t = gs->getTown(townID);
  1601. t->built = 0;
  1602. t->spellResearchCounterDay = 0;
  1603. }
  1604. if(newRumor)
  1605. gs->currentRumor = *newRumor;
  1606. }
  1607. void SetObjectProperty::applyGs(CGameState *gs)
  1608. {
  1609. CGObjectInstance *obj = gs->getObjInstance(id);
  1610. if(!obj)
  1611. {
  1612. logNetwork->error("Wrong object ID - property cannot be set!");
  1613. return;
  1614. }
  1615. auto * cai = dynamic_cast<CArmedInstance *>(obj);
  1616. if(what == ObjProperty::OWNER && obj->asOwnable())
  1617. {
  1618. PlayerColor oldOwner = obj->getOwner();
  1619. PlayerColor newOwner = identifier.as<PlayerColor>();
  1620. if(oldOwner.isValidPlayer())
  1621. gs->getPlayerState(oldOwner)->removeOwnedObject(obj);
  1622. if(newOwner.isValidPlayer())
  1623. gs->getPlayerState(newOwner)->addOwnedObject(obj);
  1624. }
  1625. if(what == ObjProperty::OWNER && cai)
  1626. {
  1627. if(obj->ID == Obj::TOWN)
  1628. {
  1629. auto * t = dynamic_cast<CGTownInstance *>(obj);
  1630. assert(t);
  1631. PlayerColor oldOwner = t->tempOwner;
  1632. if(oldOwner.isValidPlayer())
  1633. {
  1634. auto * state = gs->getPlayerState(oldOwner);
  1635. if(state->getTowns().empty())
  1636. state->daysWithoutCastle = 0;
  1637. }
  1638. if(identifier.as<PlayerColor>().isValidPlayer())
  1639. {
  1640. //reset counter before NewTurn to avoid no town message if game loaded at turn when one already captured
  1641. PlayerState * p = gs->getPlayerState(identifier.as<PlayerColor>());
  1642. if(p->daysWithoutCastle)
  1643. p->daysWithoutCastle = std::nullopt;
  1644. }
  1645. }
  1646. cai->detachFromBonusSystem(*gs);
  1647. obj->setProperty(what, identifier);
  1648. cai->attachToBonusSystem(*gs);
  1649. }
  1650. else //not an armed instance
  1651. {
  1652. obj->setProperty(what, identifier);
  1653. }
  1654. }
  1655. void HeroLevelUp::applyGs(CGameState *gs)
  1656. {
  1657. auto * hero = gs->getHero(heroId);
  1658. assert(hero);
  1659. hero->levelUp(skills);
  1660. }
  1661. void CommanderLevelUp::applyGs(CGameState *gs)
  1662. {
  1663. auto * hero = gs->getHero(heroId);
  1664. assert(hero);
  1665. const auto & commander = hero->getCommander();
  1666. assert(commander);
  1667. commander->levelUp();
  1668. }
  1669. void BattleStart::applyGs(CGameState *gs)
  1670. {
  1671. assert(battleID == gs->nextBattleID);
  1672. info->battleID = gs->nextBattleID;
  1673. info->localInit();
  1674. gs->currentBattles.push_back(std::move(info));
  1675. gs->nextBattleID = BattleID(gs->nextBattleID.getNum() + 1);
  1676. }
  1677. void BattleNextRound::applyGs(CGameState *gs)
  1678. {
  1679. gs->getBattle(battleID)->nextRound();
  1680. }
  1681. void BattleSetActiveStack::applyGs(CGameState *gs)
  1682. {
  1683. gs->getBattle(battleID)->nextTurn(stack);
  1684. }
  1685. void BattleTriggerEffect::applyGs(CGameState *gs)
  1686. {
  1687. CStack * st = gs->getBattle(battleID)->getStack(stackID);
  1688. assert(st);
  1689. switch(static_cast<BonusType>(effect))
  1690. {
  1691. case BonusType::HP_REGENERATION:
  1692. {
  1693. int64_t toHeal = val;
  1694. st->heal(toHeal, EHealLevel::HEAL, EHealPower::PERMANENT);
  1695. break;
  1696. }
  1697. case BonusType::MANA_DRAIN:
  1698. {
  1699. CGHeroInstance * h = gs->getHero(ObjectInstanceID(additionalInfo));
  1700. st->drainedMana = true;
  1701. h->mana -= val;
  1702. vstd::amax(h->mana, 0);
  1703. break;
  1704. }
  1705. case BonusType::POISON:
  1706. {
  1707. auto b = st->getLocalBonus(Selector::source(BonusSource::SPELL_EFFECT, SpellID(SpellID::POISON))
  1708. .And(Selector::type()(BonusType::STACK_HEALTH)));
  1709. if (b)
  1710. b->val = val;
  1711. break;
  1712. }
  1713. case BonusType::ENCHANTER:
  1714. case BonusType::MORALE:
  1715. break;
  1716. case BonusType::FEAR:
  1717. st->fear = true;
  1718. break;
  1719. default:
  1720. logNetwork->error("Unrecognized trigger effect type %d", effect);
  1721. }
  1722. }
  1723. void BattleUpdateGateState::applyGs(CGameState *gs)
  1724. {
  1725. if(gs->getBattle(battleID))
  1726. gs->getBattle(battleID)->si.gateState = state;
  1727. }
  1728. void BattleCancelled::applyGs(CGameState *gs)
  1729. {
  1730. auto currentBattle = boost::range::find_if(gs->currentBattles, [&](const auto & battle)
  1731. {
  1732. return battle->battleID == battleID;
  1733. });
  1734. assert(currentBattle != gs->currentBattles.end());
  1735. gs->currentBattles.erase(currentBattle);
  1736. }
  1737. void BattleResultAccepted::applyGs(CGameState *gs)
  1738. {
  1739. // Remove any "until next battle" bonuses
  1740. if(const auto attackerHero = gs->getHero(heroResult[BattleSide::ATTACKER].heroID))
  1741. attackerHero->removeBonusesRecursive(Bonus::OneBattle);
  1742. if(const auto defenderHero = gs->getHero(heroResult[BattleSide::DEFENDER].heroID))
  1743. defenderHero->removeBonusesRecursive(Bonus::OneBattle);
  1744. if(winnerSide != BattleSide::NONE)
  1745. {
  1746. // Grow up growing artifacts
  1747. if(const auto winnerHero = gs->getHero(heroResult[winnerSide].heroID))
  1748. {
  1749. if(winnerHero->getCommander() && winnerHero->getCommander()->alive)
  1750. {
  1751. for(auto & art : winnerHero->getCommander()->artifactsWorn)
  1752. gs->getArtInstance(art.second.getID())->growingUp();
  1753. }
  1754. for(auto & art : winnerHero->artifactsWorn)
  1755. gs->getArtInstance(art.second.getID())->growingUp();
  1756. }
  1757. }
  1758. if(gs->getSettings().getBoolean(EGameSettings::MODULE_STACK_EXPERIENCE))
  1759. {
  1760. if(const auto attackerArmy = gs->getArmyInstance(heroResult[BattleSide::ATTACKER].armyID))
  1761. {
  1762. attackerArmy->giveStackExp(heroResult[BattleSide::ATTACKER].exp);
  1763. attackerArmy->nodeHasChanged();
  1764. }
  1765. if(const auto defenderArmy = gs->getArmyInstance(heroResult[BattleSide::DEFENDER].armyID))
  1766. {
  1767. defenderArmy->giveStackExp(heroResult[BattleSide::DEFENDER].exp);
  1768. defenderArmy->nodeHasChanged();
  1769. }
  1770. }
  1771. }
  1772. void BattleLogMessage::applyGs(CGameState *gs)
  1773. {
  1774. //nothing
  1775. }
  1776. void BattleLogMessage::applyBattle(IBattleState * battleState)
  1777. {
  1778. //nothing
  1779. }
  1780. void BattleStackMoved::applyGs(CGameState *gs)
  1781. {
  1782. applyBattle(gs->getBattle(battleID));
  1783. }
  1784. void BattleStackMoved::applyBattle(IBattleState * battleState)
  1785. {
  1786. battleState->moveUnit(stack, tilesToMove.back());
  1787. }
  1788. void BattleStackAttacked::applyGs(CGameState *gs)
  1789. {
  1790. applyBattle(gs->getBattle(battleID));
  1791. }
  1792. void BattleStackAttacked::applyBattle(IBattleState * battleState)
  1793. {
  1794. battleState->setUnitState(newState.id, newState.data, newState.healthDelta);
  1795. }
  1796. void BattleAttack::applyGs(CGameState *gs)
  1797. {
  1798. CStack * attacker = gs->getBattle(battleID)->getStack(stackAttacking);
  1799. assert(attacker);
  1800. attackerChanges.applyGs(gs);
  1801. for(BattleStackAttacked & stackAttacked : bsa)
  1802. stackAttacked.applyGs(gs);
  1803. attacker->removeBonusesRecursive(Bonus::UntilAttack);
  1804. if(!this->counter())
  1805. attacker->removeBonusesRecursive(Bonus::UntilOwnAttack);
  1806. }
  1807. void StartAction::applyGs(CGameState *gs)
  1808. {
  1809. CStack *st = gs->getBattle(battleID)->getStack(ba.stackNumber);
  1810. if(ba.actionType == EActionType::END_TACTIC_PHASE)
  1811. {
  1812. gs->getBattle(battleID)->tacticDistance = 0;
  1813. return;
  1814. }
  1815. if(gs->getBattle(battleID)->tacticDistance)
  1816. {
  1817. // moves in tactics phase do not affect creature status
  1818. // (tactics stack queue is managed by client)
  1819. return;
  1820. }
  1821. if (ba.isUnitAction())
  1822. {
  1823. assert(st); // stack must exists for all non-hero actions
  1824. switch(ba.actionType)
  1825. {
  1826. case EActionType::DEFEND:
  1827. st->waiting = false;
  1828. st->defending = true;
  1829. st->defendingAnim = true;
  1830. break;
  1831. case EActionType::WAIT:
  1832. st->defendingAnim = false;
  1833. st->waiting = true;
  1834. st->waitedThisTurn = true;
  1835. break;
  1836. case EActionType::HERO_SPELL: //no change in current stack state
  1837. break;
  1838. default: //any active stack action - attack, catapult, heal, spell...
  1839. st->waiting = false;
  1840. st->defendingAnim = false;
  1841. st->movedThisRound = true;
  1842. st->castSpellThisTurn = ba.actionType == EActionType::MONSTER_SPELL;
  1843. break;
  1844. }
  1845. }
  1846. else
  1847. {
  1848. if(ba.actionType == EActionType::HERO_SPELL)
  1849. gs->getBattle(battleID)->getSide(ba.side).usedSpellsHistory.push_back(ba.spell);
  1850. }
  1851. }
  1852. void BattleSpellCast::applyGs(CGameState *gs)
  1853. {
  1854. if(castByHero && side != BattleSide::NONE)
  1855. gs->getBattle(battleID)->getSide(side).castSpellsCount++;
  1856. }
  1857. void SetStackEffect::applyGs(CGameState *gs)
  1858. {
  1859. applyBattle(gs->getBattle(battleID));
  1860. }
  1861. void SetStackEffect::applyBattle(IBattleState * battleState)
  1862. {
  1863. for(const auto & stackData : toRemove)
  1864. battleState->removeUnitBonus(stackData.first, stackData.second);
  1865. for(const auto & stackData : toUpdate)
  1866. battleState->updateUnitBonus(stackData.first, stackData.second);
  1867. for(const auto & stackData : toAdd)
  1868. battleState->addUnitBonus(stackData.first, stackData.second);
  1869. }
  1870. void StacksInjured::applyGs(CGameState *gs)
  1871. {
  1872. applyBattle(gs->getBattle(battleID));
  1873. }
  1874. void StacksInjured::applyBattle(IBattleState * battleState)
  1875. {
  1876. for(BattleStackAttacked stackAttacked : stacks)
  1877. stackAttacked.applyBattle(battleState);
  1878. }
  1879. void BattleUnitsChanged::applyGs(CGameState *gs)
  1880. {
  1881. applyBattle(gs->getBattle(battleID));
  1882. }
  1883. void BattleUnitsChanged::applyBattle(IBattleState * battleState)
  1884. {
  1885. for(auto & elem : changedStacks)
  1886. {
  1887. switch(elem.operation)
  1888. {
  1889. case BattleChanges::EOperation::RESET_STATE:
  1890. battleState->setUnitState(elem.id, elem.data, elem.healthDelta);
  1891. break;
  1892. case BattleChanges::EOperation::REMOVE:
  1893. battleState->removeUnit(elem.id);
  1894. break;
  1895. case BattleChanges::EOperation::ADD:
  1896. battleState->addUnit(elem.id, elem.data);
  1897. break;
  1898. case BattleChanges::EOperation::UPDATE:
  1899. battleState->updateUnit(elem.id, elem.data);
  1900. break;
  1901. default:
  1902. logNetwork->error("Unknown unit operation %d", static_cast<int>(elem.operation));
  1903. break;
  1904. }
  1905. }
  1906. }
  1907. void BattleResultsApplied::applyGs(CGameState * gs)
  1908. {
  1909. learnedSpells.applyGs(gs);
  1910. for(auto & artPack : artifacts)
  1911. artPack.applyGs(gs);
  1912. const auto currentBattle = std::find_if(gs->currentBattles.begin(), gs->currentBattles.end(),
  1913. [this](const auto & battle)
  1914. {
  1915. return battle->battleID == battleID;
  1916. });
  1917. assert(currentBattle != gs->currentBattles.end());
  1918. gs->currentBattles.erase(currentBattle);
  1919. }
  1920. void BattleObstaclesChanged::applyGs(CGameState *gs)
  1921. {
  1922. applyBattle(gs->getBattle(battleID));
  1923. }
  1924. void BattleObstaclesChanged::applyBattle(IBattleState * battleState)
  1925. {
  1926. for(const auto & change : changes)
  1927. {
  1928. switch(change.operation)
  1929. {
  1930. case BattleChanges::EOperation::REMOVE:
  1931. battleState->removeObstacle(change.id);
  1932. break;
  1933. case BattleChanges::EOperation::ADD:
  1934. battleState->addObstacle(change);
  1935. break;
  1936. case BattleChanges::EOperation::UPDATE:
  1937. battleState->updateObstacle(change);
  1938. break;
  1939. default:
  1940. logNetwork->error("Unknown obstacle operation %d", static_cast<int>(change.operation));
  1941. break;
  1942. }
  1943. }
  1944. }
  1945. CatapultAttack::CatapultAttack() = default;
  1946. CatapultAttack::~CatapultAttack() = default;
  1947. void CatapultAttack::applyGs(CGameState *gs)
  1948. {
  1949. applyBattle(gs->getBattle(battleID));
  1950. }
  1951. void CatapultAttack::visitTyped(ICPackVisitor & visitor)
  1952. {
  1953. visitor.visitCatapultAttack(*this);
  1954. }
  1955. void CatapultAttack::applyBattle(IBattleState * battleState)
  1956. {
  1957. const auto * town = battleState->getDefendedTown();
  1958. if(!town)
  1959. return;
  1960. if(town->fortificationsLevel().wallsHealth == 0)
  1961. return;
  1962. for(const auto & part : attackedParts)
  1963. {
  1964. auto newWallState = SiegeInfo::applyDamage(battleState->getWallState(part.attackedPart), part.damageDealt);
  1965. battleState->setWallState(part.attackedPart, newWallState);
  1966. }
  1967. }
  1968. void BattleSetStackProperty::applyGs(CGameState *gs)
  1969. {
  1970. CStack * stack = gs->getBattle(battleID)->getStack(stackID, false);
  1971. switch(which)
  1972. {
  1973. case CASTS:
  1974. {
  1975. if(absolute)
  1976. logNetwork->error("Can not change casts in absolute mode");
  1977. else
  1978. stack->casts.use(-val);
  1979. break;
  1980. }
  1981. case ENCHANTER_COUNTER:
  1982. {
  1983. auto & counter = gs->getBattle(battleID)->getSide(gs->getBattle(battleID)->whatSide(stack->unitOwner())).enchanterCounter;
  1984. if(absolute)
  1985. counter = val;
  1986. else
  1987. counter += val;
  1988. vstd::amax(counter, 0);
  1989. break;
  1990. }
  1991. case UNBIND:
  1992. {
  1993. stack->removeBonusesRecursive(Selector::type()(BonusType::BIND_EFFECT));
  1994. break;
  1995. }
  1996. case CLONED:
  1997. {
  1998. stack->cloned = true;
  1999. break;
  2000. }
  2001. case HAS_CLONE:
  2002. {
  2003. stack->cloneID = val;
  2004. break;
  2005. }
  2006. }
  2007. }
  2008. void PlayerCheated::applyGs(CGameState *gs)
  2009. {
  2010. if(!player.isValidPlayer())
  2011. return;
  2012. gs->getPlayerState(player)->enteredLosingCheatCode = losingCheatCode;
  2013. gs->getPlayerState(player)->enteredWinningCheatCode = winningCheatCode;
  2014. gs->getPlayerState(player)->cheated = true;
  2015. }
  2016. void PlayerStartsTurn::applyGs(CGameState *gs)
  2017. {
  2018. //assert(gs->actingPlayers.count(player) == 0);//Legal - may happen after loading of deserialized map state
  2019. gs->actingPlayers.insert(player);
  2020. }
  2021. void PlayerEndsTurn::applyGs(CGameState *gs)
  2022. {
  2023. assert(gs->actingPlayers.count(player) == 1);
  2024. gs->actingPlayers.erase(player);
  2025. }
  2026. void DaysWithoutTown::applyGs(CGameState *gs)
  2027. {
  2028. auto & playerState = gs->players.at(player);
  2029. playerState.daysWithoutCastle = daysWithoutCastle;
  2030. }
  2031. void TurnTimeUpdate::applyGs(CGameState *gs)
  2032. {
  2033. auto & playerState = gs->players.at(player);
  2034. playerState.turnTimer = turnTimer;
  2035. }
  2036. void EntitiesChanged::applyGs(CGameState *gs)
  2037. {
  2038. for(const auto & change : changes)
  2039. gs->updateEntity(change.metatype, change.entityIndex, change.data);
  2040. }
  2041. void SetRewardableConfiguration::applyGs(CGameState *gs)
  2042. {
  2043. auto * objectPtr = gs->getObjInstance(objectID);
  2044. if (!buildingID.hasValue())
  2045. {
  2046. auto * rewardablePtr = dynamic_cast<CRewardableObject *>(objectPtr);
  2047. assert(rewardablePtr);
  2048. rewardablePtr->configuration = configuration;
  2049. rewardablePtr->initializeGuards();
  2050. }
  2051. else
  2052. {
  2053. auto * townPtr = dynamic_cast<CGTownInstance*>(objectPtr);
  2054. TownBuildingInstance * buildingPtr = nullptr;
  2055. for (auto & building : townPtr->rewardableBuildings)
  2056. if (building.second->getBuildingType() == buildingID)
  2057. buildingPtr = building.second.get();
  2058. auto * rewardablePtr = dynamic_cast<TownRewardableBuildingInstance *>(buildingPtr);
  2059. assert(rewardablePtr);
  2060. rewardablePtr->configuration = configuration;
  2061. }
  2062. }
  2063. ArtSlotInfo::ArtSlotInfo(IGameCallback * cb)
  2064. : GameCallbackHolder(cb)
  2065. {
  2066. }
  2067. ArtSlotInfo::ArtSlotInfo(const CArtifactInstance * artifact, bool locked)
  2068. : GameCallbackHolder(artifact->cb)
  2069. , artifactID(artifact->getId())
  2070. , locked(locked)
  2071. {
  2072. }
  2073. const CArtifactInstance * ArtSlotInfo::getArt() const
  2074. {
  2075. if(!artifactID.hasValue())
  2076. return nullptr;
  2077. return cb->getArtInstance(artifactID);
  2078. }
  2079. ArtifactInstanceID ArtSlotInfo::getID() const
  2080. {
  2081. return artifactID;
  2082. }
  2083. VCMI_LIB_NAMESPACE_END