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 beatenObject = gs->getMap().eraseObject(obj->id);
  981. auto beatenHero = std::dynamic_pointer_cast<CGHeroInstance>(beatenObject);
  982. assert(beatenHero);
  983. auto * siegeNode = beatenHero->whereShouldBeAttachedOnSiege(gs);
  984. // FIXME: workaround:
  985. // hero should be attached to siegeNode after battle
  986. // however this code might also be called on dismissing hero while in town
  987. if (siegeNode && vstd::contains(beatenHero->getParentNodes(), siegeNode))
  988. beatenHero->detachFrom(*siegeNode);
  989. beatenHero->tempOwner = PlayerColor::NEUTRAL; //no one owns beaten hero
  990. vstd::erase_if(beatenHero->artifactsInBackpack, [](const ArtSlotInfo& asi)
  991. {
  992. return asi.getArt()->getTypeId() == ArtifactID::GRAIL;
  993. });
  994. if(beatenHero->getVisitedTown())
  995. {
  996. if(beatenHero->getVisitedTown()->getGarrisonHero() == beatenHero.get())
  997. beatenHero->getVisitedTown()->setGarrisonedHero(nullptr);
  998. else
  999. beatenHero->getVisitedTown()->setVisitingHero(nullptr);
  1000. beatenHero->setVisitedTown(nullptr, false);
  1001. }
  1002. //return hero to the pool, so he may reappear in tavern
  1003. gs->heroesPool->addHeroToPool(beatenHero->getHeroTypeID());
  1004. gs->getMap().addToHeroPool(beatenHero);
  1005. //If hero on Boat is removed, the Boat disappears
  1006. if(beatenHero->inBoat())
  1007. {
  1008. auto boat = beatenHero->getBoat();
  1009. beatenHero->setBoat(nullptr);
  1010. gs->getMap().eraseObject(boat->id);
  1011. }
  1012. return;
  1013. }
  1014. const auto * quest = dynamic_cast<const IQuestObject *>(obj);
  1015. if (quest)
  1016. {
  1017. for (auto &player : gs->players)
  1018. {
  1019. vstd::erase_if(player.second.quests, [obj](const QuestInfo & q){
  1020. return q.obj == obj->id;
  1021. });
  1022. }
  1023. }
  1024. gs->getMap().eraseObject(objectID);
  1025. gs->getMap().calculateGuardingGreaturePositions();//FIXME: excessive, update only affected tiles
  1026. }
  1027. static int getDir(const int3 & src, const int3 & dst)
  1028. {
  1029. int ret = -1;
  1030. if(dst.x+1 == src.x && dst.y+1 == src.y) //tl
  1031. {
  1032. ret = 1;
  1033. }
  1034. else if(dst.x == src.x && dst.y+1 == src.y) //t
  1035. {
  1036. ret = 2;
  1037. }
  1038. else if(dst.x-1 == src.x && dst.y+1 == src.y) //tr
  1039. {
  1040. ret = 3;
  1041. }
  1042. else if(dst.x-1 == src.x && dst.y == src.y) //r
  1043. {
  1044. ret = 4;
  1045. }
  1046. else if(dst.x-1 == src.x && dst.y-1 == src.y) //br
  1047. {
  1048. ret = 5;
  1049. }
  1050. else if(dst.x == src.x && dst.y-1 == src.y) //b
  1051. {
  1052. ret = 6;
  1053. }
  1054. else if(dst.x+1 == src.x && dst.y-1 == src.y) //bl
  1055. {
  1056. ret = 7;
  1057. }
  1058. else if(dst.x+1 == src.x && dst.y == src.y) //l
  1059. {
  1060. ret = 8;
  1061. }
  1062. return ret;
  1063. }
  1064. void TryMoveHero::applyGs(CGameState *gs)
  1065. {
  1066. CGHeroInstance *h = gs->getHero(id);
  1067. if (!h)
  1068. {
  1069. logGlobal->error("Attempt ot move unavailable hero %d", id.getNum());
  1070. return;
  1071. }
  1072. const TerrainTile & fromTile = gs->getMap().getTile(h->convertToVisitablePos(start));
  1073. const TerrainTile & destTile = gs->getMap().getTile(h->convertToVisitablePos(end));
  1074. h->setMovementPoints(movePoints);
  1075. if((result == SUCCESS || result == BLOCKING_VISIT || result == EMBARK || result == DISEMBARK) && start != end)
  1076. {
  1077. auto dir = getDir(start,end);
  1078. if(dir > 0 && dir <= 8)
  1079. h->moveDir = dir;
  1080. //else don`t change move direction - hero might have traversed the subterranean gate, direction should be kept
  1081. }
  1082. if(result == EMBARK) //hero enters boat at destination tile
  1083. {
  1084. const TerrainTile &tt = gs->getMap().getTile(h->convertToVisitablePos(end));
  1085. ObjectInstanceID topObjectID = tt.visitableObjects.back();
  1086. CGObjectInstance * topObject = gs->getObjInstance(topObjectID);
  1087. assert(tt.visitableObjects.size() >= 1 && topObject->ID == Obj::BOAT); //the only visitable object at destination is Boat
  1088. auto * boat = dynamic_cast<CGBoat *>(topObject);
  1089. assert(boat);
  1090. gs->getMap().hideObject(boat); //hero blockvis mask will be used, we don't need to duplicate it with boat
  1091. h->setBoat(boat);
  1092. }
  1093. else if(result == DISEMBARK) //hero leaves boat to destination tile
  1094. {
  1095. auto * b = h->getBoat();
  1096. b->direction = h->moveDir;
  1097. b->pos = start;
  1098. gs->getMap().showObject(b);
  1099. h->setBoat(nullptr);
  1100. }
  1101. if(start!=end && (result == SUCCESS || result == TELEPORTATION || result == EMBARK || result == DISEMBARK))
  1102. {
  1103. gs->getMap().hideObject(h);
  1104. h->setAnchorPos(end);
  1105. if(auto * b = h->getBoat())
  1106. b->setAnchorPos(end);
  1107. gs->getMap().showObject(h);
  1108. }
  1109. auto & fogOfWarMap = gs->getPlayerTeam(h->getOwner())->fogOfWarMap;
  1110. for(const int3 & t : fowRevealed)
  1111. fogOfWarMap[t.z][t.x][t.y] = 1;
  1112. if (fromTile.getTerrainID() != destTile.getTerrainID())
  1113. h->nodeHasChanged(); // update bonuses with terrain limiter
  1114. }
  1115. void NewStructures::applyGs(CGameState *gs)
  1116. {
  1117. CGTownInstance *t = gs->getTown(tid);
  1118. for(const auto & id : bid)
  1119. {
  1120. assert(t->getTown()->buildings.at(id) != nullptr);
  1121. t->addBuilding(id);
  1122. }
  1123. t->updateAppearance();
  1124. t->built = built;
  1125. t->recreateBuildingsBonuses();
  1126. }
  1127. void RazeStructures::applyGs(CGameState *gs)
  1128. {
  1129. CGTownInstance *t = gs->getTown(tid);
  1130. for(const auto & id : bid)
  1131. {
  1132. t->removeBuilding(id);
  1133. t->updateAppearance();
  1134. }
  1135. t->destroyed = destroyed; //yeaha
  1136. t->recreateBuildingsBonuses();
  1137. }
  1138. void SetAvailableCreatures::applyGs(CGameState *gs)
  1139. {
  1140. auto * dw = dynamic_cast<CGDwelling *>(gs->getObjInstance(tid));
  1141. assert(dw);
  1142. dw->creatures = creatures;
  1143. }
  1144. void SetHeroesInTown::applyGs(CGameState *gs)
  1145. {
  1146. CGTownInstance *t = gs->getTown(tid);
  1147. CGHeroInstance * v = gs->getHero(visiting);
  1148. CGHeroInstance * g = gs->getHero(garrison);
  1149. bool newVisitorComesFromGarrison = v && v == t->getGarrisonHero();
  1150. bool newGarrisonComesFromVisiting = g && g == t->getVisitingHero();
  1151. if(newVisitorComesFromGarrison)
  1152. t->setGarrisonedHero(nullptr);
  1153. if(newGarrisonComesFromVisiting)
  1154. t->setVisitingHero(nullptr);
  1155. if(!newGarrisonComesFromVisiting || v)
  1156. t->setVisitingHero(v);
  1157. if(!newVisitorComesFromGarrison || g)
  1158. t->setGarrisonedHero(g);
  1159. if(v)
  1160. gs->getMap().showObject(v);
  1161. if(g)
  1162. gs->getMap().hideObject(g);
  1163. }
  1164. void HeroRecruited::applyGs(CGameState *gs)
  1165. {
  1166. auto h = gs->heroesPool->takeHeroFromPool(hid);
  1167. CGTownInstance *t = gs->getTown(tid);
  1168. PlayerState *p = gs->getPlayerState(player);
  1169. if (boatId != ObjectInstanceID::NONE)
  1170. {
  1171. CGObjectInstance *obj = gs->getObjInstance(boatId);
  1172. auto * boat = dynamic_cast<CGBoat *>(obj);
  1173. if (boat)
  1174. {
  1175. gs->getMap().hideObject(boat);
  1176. h->setBoat(boat);
  1177. }
  1178. }
  1179. h->setOwner(player);
  1180. h->pos = tile;
  1181. h->updateAppearance();
  1182. assert(h->id.hasValue());
  1183. gs->getMap().addNewObject(h);
  1184. p->addOwnedObject(h.get());
  1185. h->attachToBonusSystem(gs);
  1186. if(t)
  1187. t->setVisitingHero(h.get());
  1188. }
  1189. void GiveHero::applyGs(CGameState *gs)
  1190. {
  1191. CGHeroInstance *h = gs->getHero(id);
  1192. if (boatId != ObjectInstanceID::NONE)
  1193. {
  1194. CGObjectInstance *obj = gs->getObjInstance(boatId);
  1195. auto * boat = dynamic_cast<CGBoat *>(obj);
  1196. if (boat)
  1197. {
  1198. gs->getMap().hideObject(boat);
  1199. h->setBoat(boat);
  1200. }
  1201. }
  1202. //bonus system
  1203. h->detachFrom(gs->globalEffects);
  1204. h->attachTo(*gs->getPlayerState(player));
  1205. auto oldVisitablePos = h->visitablePos();
  1206. gs->getMap().hideObject(h);
  1207. h->updateAppearance();
  1208. h->setOwner(player);
  1209. h->setMovementPoints(h->movementPointsLimit(true));
  1210. h->setAnchorPos(h->convertFromVisitablePos(oldVisitablePos));
  1211. gs->getMap().heroAddedToMap(h);
  1212. gs->getPlayerState(h->getOwner())->addOwnedObject(h);
  1213. gs->getMap().showObject(h);
  1214. h->setVisitedTown(nullptr, false);
  1215. }
  1216. void NewObject::applyGs(CGameState *gs)
  1217. {
  1218. gs->getMap().addNewObject(newObject);
  1219. gs->getMap().calculateGuardingGreaturePositions();
  1220. // attach newly spawned wandering monster to global bonus system node
  1221. auto newArmy = std::dynamic_pointer_cast<CArmedInstance>(newObject);
  1222. if (newArmy)
  1223. newArmy->attachToBonusSystem(gs);
  1224. logGlobal->debug("Added object id=%d; name=%s", newObject->id, newObject->getObjectName());
  1225. }
  1226. void NewArtifact::applyGs(CGameState *gs)
  1227. {
  1228. auto art = gs->createArtifact(artId, spellId);
  1229. PutArtifact pa(art->getId(), ArtifactLocation(artHolder, pos), false);
  1230. pa.applyGs(gs);
  1231. }
  1232. void ChangeStackCount::applyGs(CGameState *gs)
  1233. {
  1234. auto * srcObj = gs->getArmyInstance(army);
  1235. if(!srcObj)
  1236. throw std::runtime_error("ChangeStackCount: invalid army object " + std::to_string(army.getNum()) + ", possible game state corruption.");
  1237. if(absoluteValue)
  1238. srcObj->setStackCount(slot, count);
  1239. else
  1240. srcObj->changeStackCount(slot, count);
  1241. }
  1242. void SetStackType::applyGs(CGameState *gs)
  1243. {
  1244. auto * srcObj = gs->getArmyInstance(army);
  1245. if(!srcObj)
  1246. throw std::runtime_error("SetStackType: invalid army object " + std::to_string(army.getNum()) + ", possible game state corruption.");
  1247. srcObj->setStackType(slot, type);
  1248. }
  1249. void EraseStack::applyGs(CGameState *gs)
  1250. {
  1251. auto * srcObj = gs->getArmyInstance(army);
  1252. if(!srcObj)
  1253. throw std::runtime_error("EraseStack: invalid army object " + std::to_string(army.getNum()) + ", possible game state corruption.");
  1254. srcObj->eraseStack(slot);
  1255. }
  1256. void SwapStacks::applyGs(CGameState *gs)
  1257. {
  1258. auto * srcObj = gs->getArmyInstance(srcArmy);
  1259. if(!srcObj)
  1260. throw std::runtime_error("SwapStacks: invalid army object " + std::to_string(srcArmy.getNum()) + ", possible game state corruption.");
  1261. auto * dstObj = gs->getArmyInstance(dstArmy);
  1262. if(!dstObj)
  1263. throw std::runtime_error("SwapStacks: invalid army object " + std::to_string(dstArmy.getNum()) + ", possible game state corruption.");
  1264. auto s1 = srcObj->detachStack(srcSlot);
  1265. auto s2 = dstObj->detachStack(dstSlot);
  1266. srcObj->putStack(srcSlot, std::move(s2));
  1267. dstObj->putStack(dstSlot, std::move(s1));
  1268. }
  1269. void InsertNewStack::applyGs(CGameState *gs)
  1270. {
  1271. if(auto * obj = gs->getArmyInstance(army))
  1272. obj->putStack(slot, std::make_unique<CStackInstance>(gs->cb, type, count));
  1273. else
  1274. throw std::runtime_error("InsertNewStack: invalid army object " + std::to_string(army.getNum()) + ", possible game state corruption.");
  1275. }
  1276. void RebalanceStacks::applyGs(CGameState *gs)
  1277. {
  1278. auto * srcObj = gs->getArmyInstance(srcArmy);
  1279. if(!srcObj)
  1280. throw std::runtime_error("RebalanceStacks: invalid army object " + std::to_string(srcArmy.getNum()) + ", possible game state corruption.");
  1281. auto * dstObj = gs->getArmyInstance(dstArmy);
  1282. if(!dstObj)
  1283. throw std::runtime_error("RebalanceStacks: invalid army object " + std::to_string(dstArmy.getNum()) + ", possible game state corruption.");
  1284. StackLocation src(srcObj->id, srcSlot);
  1285. StackLocation dst(dstObj->id, dstSlot);
  1286. const CCreature * srcType = srcObj->getCreature(src.slot);
  1287. TQuantity srcCount = srcObj->getStackCount(src.slot);
  1288. bool stackExp = gs->getSettings().getBoolean(EGameSettings::MODULE_STACK_EXPERIENCE);
  1289. if(srcCount == count) //moving whole stack
  1290. {
  1291. const auto c = dstObj->getCreature(dst.slot);
  1292. if(c) //stack at dest -> merge
  1293. {
  1294. assert(c == srcType);
  1295. const auto srcHero = dynamic_cast<CGHeroInstance*>(srcObj);
  1296. const auto dstHero = dynamic_cast<CGHeroInstance*>(dstObj);
  1297. auto srcStack = const_cast<CStackInstance*>(srcObj->getStackPtr(src.slot));
  1298. auto dstStack = const_cast<CStackInstance*>(dstObj->getStackPtr(dst.slot));
  1299. if(srcStack->getArt(ArtifactPosition::CREATURE_SLOT))
  1300. {
  1301. if(auto dstArt = dstStack->getArt(ArtifactPosition::CREATURE_SLOT))
  1302. {
  1303. bool artifactIsLost = true;
  1304. if(srcHero)
  1305. {
  1306. auto dstSlot = ArtifactUtils::getArtBackpackPosition(srcHero, dstArt->getTypeId());
  1307. if (dstSlot != ArtifactPosition::PRE_FIRST)
  1308. {
  1309. gs->getMap().moveArtifactInstance(*dstStack, ArtifactPosition::CREATURE_SLOT, *srcHero, dstSlot);
  1310. artifactIsLost = false;
  1311. }
  1312. }
  1313. if (artifactIsLost)
  1314. {
  1315. BulkEraseArtifacts ea;
  1316. ea.artHolder = dstHero->id;
  1317. ea.posPack.emplace_back(ArtifactPosition::CREATURE_SLOT);
  1318. ea.creature = dst.slot;
  1319. ea.applyGs(gs);
  1320. logNetwork->warn("Cannot move artifact! No free slots");
  1321. }
  1322. gs->getMap().moveArtifactInstance(*srcStack, ArtifactPosition::CREATURE_SLOT, *dstStack, ArtifactPosition::CREATURE_SLOT);
  1323. //TODO: choose from dialog
  1324. }
  1325. else //just move to the other slot before stack gets erased
  1326. {
  1327. gs->getMap().moveArtifactInstance(*srcStack, ArtifactPosition::CREATURE_SLOT, *dstStack, ArtifactPosition::CREATURE_SLOT);
  1328. }
  1329. }
  1330. if (stackExp)
  1331. {
  1332. ui64 totalExp = srcCount * srcObj->getStackExperience(src.slot) + dstObj->getStackCount(dst.slot) * dstObj->getStackExperience(dst.slot);
  1333. srcObj->eraseStack(src.slot);
  1334. dstObj->changeStackCount(dst.slot, count);
  1335. dstObj->setStackExp(dst.slot, totalExp /(dstObj->getStackCount(dst.slot))); //mean
  1336. }
  1337. else
  1338. {
  1339. srcObj->eraseStack(src.slot);
  1340. dstObj->changeStackCount(dst.slot, count);
  1341. }
  1342. }
  1343. else //move stack to an empty slot, no exp change needed
  1344. {
  1345. auto stackDetached = srcObj->detachStack(src.slot);
  1346. dstObj->putStack(dst.slot, std::move(stackDetached));
  1347. }
  1348. }
  1349. else
  1350. {
  1351. [[maybe_unused]] const CCreature *c = dstObj->getCreature(dst.slot);
  1352. if(c) //stack at dest -> rebalance
  1353. {
  1354. assert(c == srcType);
  1355. if (stackExp)
  1356. {
  1357. ui64 totalExp = srcCount * srcObj->getStackExperience(src.slot) + dstObj->getStackCount(dst.slot) * dstObj->getStackExperience(dst.slot);
  1358. srcObj->changeStackCount(src.slot, -count);
  1359. dstObj->changeStackCount(dst.slot, count);
  1360. dstObj->setStackExp(dst.slot, totalExp /(srcObj->getStackCount(src.slot) + dstObj->getStackCount(dst.slot))); //mean
  1361. }
  1362. else
  1363. {
  1364. srcObj->changeStackCount(src.slot, -count);
  1365. dstObj->changeStackCount(dst.slot, count);
  1366. }
  1367. }
  1368. else //split stack to an empty slot
  1369. {
  1370. srcObj->changeStackCount(src.slot, -count);
  1371. dstObj->addToSlot(dst.slot, srcType->getId(), count, false);
  1372. if (stackExp)
  1373. dstObj->setStackExp(dst.slot, srcObj->getStackExperience(src.slot));
  1374. }
  1375. }
  1376. srcObj->nodeHasChanged();
  1377. if (srcObj != dstObj)
  1378. dstObj->nodeHasChanged();
  1379. }
  1380. void BulkRebalanceStacks::applyGs(CGameState *gs)
  1381. {
  1382. for(auto & move : moves)
  1383. move.applyGs(gs);
  1384. }
  1385. void BulkSmartRebalanceStacks::applyGs(CGameState *gs)
  1386. {
  1387. for(auto & move : moves)
  1388. move.applyGs(gs);
  1389. for(auto & change : changes)
  1390. change.applyGs(gs);
  1391. }
  1392. void PutArtifact::applyGs(CGameState *gs)
  1393. {
  1394. auto art = gs->getArtInstance(id);
  1395. assert(!art->getParentNodes().empty());
  1396. auto hero = gs->getHero(al.artHolder);
  1397. assert(hero);
  1398. assert(art && art->canBePutAt(hero, al.slot));
  1399. assert(ArtifactUtils::checkIfSlotValid(*hero, al.slot));
  1400. gs->getMap().putArtifactInstance(*hero, art->getId(), al.slot);
  1401. }
  1402. void BulkEraseArtifacts::applyGs(CGameState *gs)
  1403. {
  1404. const auto artSet = gs->getArtSet(artHolder);
  1405. assert(artSet);
  1406. std::sort(posPack.begin(), posPack.end(), [](const ArtifactPosition & slot0, const ArtifactPosition & slot1) -> bool
  1407. {
  1408. return slot0.num > slot1.num;
  1409. });
  1410. for(const auto & slot : posPack)
  1411. {
  1412. const auto slotInfo = artSet->getSlot(slot);
  1413. const ArtifactInstanceID artifactID = slotInfo->artifactID;
  1414. const CArtifactInstance * artifact = gs->getArtInstance(artifactID);
  1415. if(slotInfo->locked)
  1416. {
  1417. logGlobal->debug("Erasing locked artifact: %s", artifact->getType()->getNameTranslated());
  1418. DisassembledArtifact dis;
  1419. dis.al.artHolder = artHolder;
  1420. for(auto & slotInfoWorn : artSet->artifactsWorn)
  1421. {
  1422. auto art = slotInfoWorn.second.getArt();
  1423. if(art->isCombined() && art->isPart(artifact))
  1424. {
  1425. dis.al.slot = artSet->getArtPos(art);
  1426. break;
  1427. }
  1428. }
  1429. assert((dis.al.slot != ArtifactPosition::PRE_FIRST) && "Failed to determine the assembly this locked artifact belongs to");
  1430. logGlobal->debug("Found the corresponding assembly: %s", artSet->getArt(dis.al.slot)->getType()->getNameTranslated());
  1431. dis.applyGs(gs);
  1432. }
  1433. else
  1434. {
  1435. logGlobal->debug("Erasing artifact %s", artifact->getType()->getNameTranslated());
  1436. }
  1437. gs->getMap().removeArtifactInstance(*artSet, slot);
  1438. }
  1439. }
  1440. void BulkMoveArtifacts::applyGs(CGameState *gs)
  1441. {
  1442. const auto bulkArtsRemove = [gs](std::vector<MoveArtifactInfo> & artsPack, CArtifactSet & artSet)
  1443. {
  1444. std::vector<ArtifactPosition> packToRemove;
  1445. for(const auto & slotsPair : artsPack)
  1446. packToRemove.push_back(slotsPair.srcPos);
  1447. std::sort(packToRemove.begin(), packToRemove.end(), [](const ArtifactPosition & slot0, const ArtifactPosition & slot1) -> bool
  1448. {
  1449. return slot0.num > slot1.num;
  1450. });
  1451. for(const auto & slot : packToRemove)
  1452. gs->getMap().removeArtifactInstance(artSet, slot);
  1453. };
  1454. const auto bulkArtsPut = [gs](std::vector<MoveArtifactInfo> & artsPack, CArtifactSet & initArtSet, CArtifactSet & dstArtSet)
  1455. {
  1456. for(const auto & slotsPair : artsPack)
  1457. {
  1458. auto * art = initArtSet.getArt(slotsPair.srcPos);
  1459. assert(art);
  1460. gs->getMap().putArtifactInstance(dstArtSet, art->getId(), slotsPair.dstPos);
  1461. }
  1462. };
  1463. auto * leftSet = gs->getArtSet(ArtifactLocation(srcArtHolder, srcCreature));
  1464. assert(leftSet);
  1465. auto * rightSet = gs->getArtSet(ArtifactLocation(dstArtHolder, dstCreature));
  1466. assert(rightSet);
  1467. CArtifactFittingSet artInitialSetLeft(*leftSet);
  1468. bulkArtsRemove(artsPack0, *leftSet);
  1469. if(!artsPack1.empty())
  1470. {
  1471. CArtifactFittingSet artInitialSetRight(*rightSet);
  1472. bulkArtsRemove(artsPack1, *rightSet);
  1473. bulkArtsPut(artsPack1, artInitialSetRight, *leftSet);
  1474. }
  1475. bulkArtsPut(artsPack0, artInitialSetLeft, *rightSet);
  1476. }
  1477. void AssembledArtifact::applyGs(CGameState *gs)
  1478. {
  1479. auto artSet = gs->getArtSet(al.artHolder);
  1480. assert(artSet);
  1481. const auto transformedArt = artSet->getArt(al.slot);
  1482. assert(transformedArt);
  1483. const auto builtArt = artId.toArtifact();
  1484. assert(vstd::contains_if(ArtifactUtils::assemblyPossibilities(artSet, transformedArt->getTypeId()), [=](const CArtifact * art)->bool
  1485. {
  1486. return art->getId() == builtArt->getId();
  1487. }));
  1488. auto * combinedArt = gs->getMap().createSingleArtifact(artId);
  1489. // Find slots for all involved artifacts
  1490. std::set<ArtifactPosition, std::greater<>> slotsInvolved = { al.slot };
  1491. CArtifactFittingSet fittingSet(*artSet);
  1492. auto parts = builtArt->getConstituents();
  1493. parts.erase(std::find(parts.begin(), parts.end(), transformedArt->getType()));
  1494. for(const auto constituent : parts)
  1495. {
  1496. const auto slot = fittingSet.getArtPos(constituent->getId(), false, false);
  1497. fittingSet.lockSlot(slot);
  1498. assert(slot != ArtifactPosition::PRE_FIRST);
  1499. slotsInvolved.insert(slot);
  1500. }
  1501. // Find a slot for combined artifact
  1502. if(ArtifactUtils::isSlotEquipment(al.slot) && ArtifactUtils::isSlotBackpack(*slotsInvolved.begin()))
  1503. {
  1504. al.slot = ArtifactPosition::BACKPACK_START;
  1505. }
  1506. else if(ArtifactUtils::isSlotBackpack(al.slot))
  1507. {
  1508. for(const auto & slot : slotsInvolved)
  1509. if(ArtifactUtils::isSlotBackpack(slot))
  1510. al.slot = slot;
  1511. }
  1512. else
  1513. {
  1514. for(const auto & slot : slotsInvolved)
  1515. if(!vstd::contains(builtArt->getPossibleSlots().at(artSet->bearerType()), al.slot)
  1516. && vstd::contains(builtArt->getPossibleSlots().at(artSet->bearerType()), slot))
  1517. {
  1518. al.slot = slot;
  1519. break;
  1520. }
  1521. }
  1522. // Delete parts from hero
  1523. for(const auto & slot : slotsInvolved)
  1524. {
  1525. const auto constituentInstance = artSet->getArt(slot);
  1526. gs->getMap().removeArtifactInstance(*artSet, slot);
  1527. if(!combinedArt->getType()->isFused())
  1528. {
  1529. if(ArtifactUtils::isSlotEquipment(al.slot) && slot != al.slot)
  1530. combinedArt->addPart(constituentInstance, slot);
  1531. else
  1532. combinedArt->addPart(constituentInstance, ArtifactPosition::PRE_FIRST);
  1533. }
  1534. }
  1535. // Put new combined artifacts
  1536. gs->getMap().putArtifactInstance(*artSet, combinedArt->getId(), al.slot);
  1537. }
  1538. void DisassembledArtifact::applyGs(CGameState *gs)
  1539. {
  1540. auto hero = gs->getHero(al.artHolder);
  1541. assert(hero);
  1542. auto disassembledArtID = hero->getArtID(al.slot);
  1543. auto disassembledArt = gs->getArtInstance(disassembledArtID);
  1544. assert(disassembledArt);
  1545. const auto parts = disassembledArt->getPartsInfo();
  1546. gs->getMap().removeArtifactInstance(*hero, al.slot);
  1547. for(auto & part : parts)
  1548. {
  1549. // ArtifactPosition::PRE_FIRST is value of main part slot -> it'll replace combined artifact in its pos
  1550. auto slot = (ArtifactUtils::isSlotEquipment(part.slot) ? part.slot : al.slot);
  1551. disassembledArt->detachFromSource(*part.getArtifact());
  1552. gs->getMap().putArtifactInstance(*hero, part.getArtifact()->getId(), slot);
  1553. }
  1554. gs->getMap().eraseArtifactInstance(disassembledArt->getId());
  1555. }
  1556. void HeroVisit::applyGs(CGameState *gs)
  1557. {
  1558. }
  1559. void SetAvailableArtifacts::applyGs(CGameState *gs)
  1560. {
  1561. if(id != ObjectInstanceID::NONE)
  1562. {
  1563. if(auto * bm = dynamic_cast<CGBlackMarket *>(gs->getObjInstance(id)))
  1564. {
  1565. bm->artifacts = arts;
  1566. }
  1567. else
  1568. {
  1569. logNetwork->error("Wrong black market id!");
  1570. }
  1571. }
  1572. else
  1573. {
  1574. gs->getMap().townMerchantArtifacts = arts;
  1575. }
  1576. }
  1577. void NewTurn::applyGs(CGameState *gs)
  1578. {
  1579. gs->day = day;
  1580. // Update bonuses before doing anything else so hero don't get more MP than needed
  1581. gs->globalEffects.removeBonusesRecursive(Bonus::OneDay); //works for children -> all game objs
  1582. gs->globalEffects.reduceBonusDurations(Bonus::NDays);
  1583. gs->globalEffects.reduceBonusDurations(Bonus::OneWeek);
  1584. //TODO not really a single root hierarchy, what about bonuses placed elsewhere? [not an issue with H3 mechanics but in the future...]
  1585. for(auto & manaPack : heroesMana)
  1586. manaPack.applyGs(gs);
  1587. for(auto & movePack : heroesMovement)
  1588. movePack.applyGs(gs);
  1589. gs->heroesPool->onNewDay();
  1590. for(auto & entry : playerIncome)
  1591. {
  1592. gs->getPlayerState(entry.first)->resources += entry.second;
  1593. gs->getPlayerState(entry.first)->resources.amin(GameConstants::PLAYER_RESOURCES_CAP);
  1594. }
  1595. for(auto & creatureSet : availableCreatures) //set available creatures in towns
  1596. creatureSet.applyGs(gs);
  1597. for (const auto & townID : gs->getMap().getAllTowns())
  1598. {
  1599. auto t = gs->getTown(townID);
  1600. t->built = 0;
  1601. t->spellResearchCounterDay = 0;
  1602. }
  1603. if(newRumor)
  1604. gs->currentRumor = *newRumor;
  1605. }
  1606. void SetObjectProperty::applyGs(CGameState *gs)
  1607. {
  1608. CGObjectInstance *obj = gs->getObjInstance(id);
  1609. if(!obj)
  1610. {
  1611. logNetwork->error("Wrong object ID - property cannot be set!");
  1612. return;
  1613. }
  1614. auto * cai = dynamic_cast<CArmedInstance *>(obj);
  1615. if(what == ObjProperty::OWNER && obj->asOwnable())
  1616. {
  1617. PlayerColor oldOwner = obj->getOwner();
  1618. PlayerColor newOwner = identifier.as<PlayerColor>();
  1619. if(oldOwner.isValidPlayer())
  1620. gs->getPlayerState(oldOwner)->removeOwnedObject(obj);
  1621. if(newOwner.isValidPlayer())
  1622. gs->getPlayerState(newOwner)->addOwnedObject(obj);
  1623. }
  1624. if(what == ObjProperty::OWNER && cai)
  1625. {
  1626. if(obj->ID == Obj::TOWN)
  1627. {
  1628. auto * t = dynamic_cast<CGTownInstance *>(obj);
  1629. assert(t);
  1630. PlayerColor oldOwner = t->tempOwner;
  1631. if(oldOwner.isValidPlayer())
  1632. {
  1633. auto * state = gs->getPlayerState(oldOwner);
  1634. if(state->getTowns().empty())
  1635. state->daysWithoutCastle = 0;
  1636. }
  1637. if(identifier.as<PlayerColor>().isValidPlayer())
  1638. {
  1639. //reset counter before NewTurn to avoid no town message if game loaded at turn when one already captured
  1640. PlayerState * p = gs->getPlayerState(identifier.as<PlayerColor>());
  1641. if(p->daysWithoutCastle)
  1642. p->daysWithoutCastle = std::nullopt;
  1643. }
  1644. }
  1645. cai->detachFromBonusSystem(gs);
  1646. obj->setProperty(what, identifier);
  1647. cai->attachToBonusSystem(gs);
  1648. }
  1649. else //not an armed instance
  1650. {
  1651. obj->setProperty(what, identifier);
  1652. }
  1653. }
  1654. void HeroLevelUp::applyGs(CGameState *gs)
  1655. {
  1656. auto * hero = gs->getHero(heroId);
  1657. assert(hero);
  1658. hero->levelUp(skills);
  1659. }
  1660. void CommanderLevelUp::applyGs(CGameState *gs)
  1661. {
  1662. auto * hero = gs->getHero(heroId);
  1663. assert(hero);
  1664. const auto & commander = hero->getCommander();
  1665. assert(commander);
  1666. commander->levelUp();
  1667. }
  1668. void BattleStart::applyGs(CGameState *gs)
  1669. {
  1670. assert(battleID == gs->nextBattleID);
  1671. info->battleID = gs->nextBattleID;
  1672. info->localInit();
  1673. gs->currentBattles.push_back(std::move(info));
  1674. gs->nextBattleID = BattleID(gs->nextBattleID.getNum() + 1);
  1675. }
  1676. void BattleNextRound::applyGs(CGameState *gs)
  1677. {
  1678. gs->getBattle(battleID)->nextRound();
  1679. }
  1680. void BattleSetActiveStack::applyGs(CGameState *gs)
  1681. {
  1682. gs->getBattle(battleID)->nextTurn(stack);
  1683. }
  1684. void BattleTriggerEffect::applyGs(CGameState *gs)
  1685. {
  1686. CStack * st = gs->getBattle(battleID)->getStack(stackID);
  1687. assert(st);
  1688. switch(static_cast<BonusType>(effect))
  1689. {
  1690. case BonusType::HP_REGENERATION:
  1691. {
  1692. int64_t toHeal = val;
  1693. st->heal(toHeal, EHealLevel::HEAL, EHealPower::PERMANENT);
  1694. break;
  1695. }
  1696. case BonusType::MANA_DRAIN:
  1697. {
  1698. CGHeroInstance * h = gs->getHero(ObjectInstanceID(additionalInfo));
  1699. st->drainedMana = true;
  1700. h->mana -= val;
  1701. vstd::amax(h->mana, 0);
  1702. break;
  1703. }
  1704. case BonusType::POISON:
  1705. {
  1706. auto b = st->getLocalBonus(Selector::source(BonusSource::SPELL_EFFECT, SpellID(SpellID::POISON))
  1707. .And(Selector::type()(BonusType::STACK_HEALTH)));
  1708. if (b)
  1709. b->val = val;
  1710. break;
  1711. }
  1712. case BonusType::ENCHANTER:
  1713. case BonusType::MORALE:
  1714. break;
  1715. case BonusType::FEAR:
  1716. st->fear = true;
  1717. break;
  1718. default:
  1719. logNetwork->error("Unrecognized trigger effect type %d", effect);
  1720. }
  1721. }
  1722. void BattleUpdateGateState::applyGs(CGameState *gs)
  1723. {
  1724. if(gs->getBattle(battleID))
  1725. gs->getBattle(battleID)->si.gateState = state;
  1726. }
  1727. void BattleCancelled::applyGs(CGameState *gs)
  1728. {
  1729. auto currentBattle = boost::range::find_if(gs->currentBattles, [&](const auto & battle)
  1730. {
  1731. return battle->battleID == battleID;
  1732. });
  1733. assert(currentBattle != gs->currentBattles.end());
  1734. gs->currentBattles.erase(currentBattle);
  1735. }
  1736. void BattleResultAccepted::applyGs(CGameState *gs)
  1737. {
  1738. // Remove any "until next battle" bonuses
  1739. if(const auto attackerHero = gs->getHero(heroResult[BattleSide::ATTACKER].heroID))
  1740. attackerHero->removeBonusesRecursive(Bonus::OneBattle);
  1741. if(const auto defenderHero = gs->getHero(heroResult[BattleSide::DEFENDER].heroID))
  1742. defenderHero->removeBonusesRecursive(Bonus::OneBattle);
  1743. if(winnerSide != BattleSide::NONE)
  1744. {
  1745. // Grow up growing artifacts
  1746. if(const auto winnerHero = gs->getHero(heroResult[winnerSide].heroID))
  1747. {
  1748. if(winnerHero->getCommander() && winnerHero->getCommander()->alive)
  1749. {
  1750. for(auto & art : winnerHero->getCommander()->artifactsWorn)
  1751. gs->getArtInstance(art.second.getID())->growingUp();
  1752. }
  1753. for(auto & art : winnerHero->artifactsWorn)
  1754. gs->getArtInstance(art.second.getID())->growingUp();
  1755. }
  1756. }
  1757. if(gs->getSettings().getBoolean(EGameSettings::MODULE_STACK_EXPERIENCE))
  1758. {
  1759. if(const auto attackerArmy = gs->getArmyInstance(heroResult[BattleSide::ATTACKER].armyID))
  1760. {
  1761. attackerArmy->giveStackExp(heroResult[BattleSide::ATTACKER].exp);
  1762. attackerArmy->nodeHasChanged();
  1763. }
  1764. if(const auto defenderArmy = gs->getArmyInstance(heroResult[BattleSide::DEFENDER].armyID))
  1765. {
  1766. defenderArmy->giveStackExp(heroResult[BattleSide::DEFENDER].exp);
  1767. defenderArmy->nodeHasChanged();
  1768. }
  1769. }
  1770. }
  1771. void BattleLogMessage::applyGs(CGameState *gs)
  1772. {
  1773. //nothing
  1774. }
  1775. void BattleLogMessage::applyBattle(IBattleState * battleState)
  1776. {
  1777. //nothing
  1778. }
  1779. void BattleStackMoved::applyGs(CGameState *gs)
  1780. {
  1781. applyBattle(gs->getBattle(battleID));
  1782. }
  1783. void BattleStackMoved::applyBattle(IBattleState * battleState)
  1784. {
  1785. battleState->moveUnit(stack, tilesToMove.back());
  1786. }
  1787. void BattleStackAttacked::applyGs(CGameState *gs)
  1788. {
  1789. applyBattle(gs->getBattle(battleID));
  1790. }
  1791. void BattleStackAttacked::applyBattle(IBattleState * battleState)
  1792. {
  1793. battleState->setUnitState(newState.id, newState.data, newState.healthDelta);
  1794. }
  1795. void BattleAttack::applyGs(CGameState *gs)
  1796. {
  1797. CStack * attacker = gs->getBattle(battleID)->getStack(stackAttacking);
  1798. assert(attacker);
  1799. attackerChanges.applyGs(gs);
  1800. for(BattleStackAttacked & stackAttacked : bsa)
  1801. stackAttacked.applyGs(gs);
  1802. attacker->removeBonusesRecursive(Bonus::UntilAttack);
  1803. if(!this->counter())
  1804. attacker->removeBonusesRecursive(Bonus::UntilOwnAttack);
  1805. }
  1806. void StartAction::applyGs(CGameState *gs)
  1807. {
  1808. CStack *st = gs->getBattle(battleID)->getStack(ba.stackNumber);
  1809. if(ba.actionType == EActionType::END_TACTIC_PHASE)
  1810. {
  1811. gs->getBattle(battleID)->tacticDistance = 0;
  1812. return;
  1813. }
  1814. if(gs->getBattle(battleID)->tacticDistance)
  1815. {
  1816. // moves in tactics phase do not affect creature status
  1817. // (tactics stack queue is managed by client)
  1818. return;
  1819. }
  1820. if (ba.isUnitAction())
  1821. {
  1822. assert(st); // stack must exists for all non-hero actions
  1823. switch(ba.actionType)
  1824. {
  1825. case EActionType::DEFEND:
  1826. st->waiting = false;
  1827. st->defending = true;
  1828. st->defendingAnim = true;
  1829. break;
  1830. case EActionType::WAIT:
  1831. st->defendingAnim = false;
  1832. st->waiting = true;
  1833. st->waitedThisTurn = true;
  1834. break;
  1835. case EActionType::HERO_SPELL: //no change in current stack state
  1836. break;
  1837. default: //any active stack action - attack, catapult, heal, spell...
  1838. st->waiting = false;
  1839. st->defendingAnim = false;
  1840. st->movedThisRound = true;
  1841. st->castSpellThisTurn = ba.actionType == EActionType::MONSTER_SPELL;
  1842. break;
  1843. }
  1844. }
  1845. else
  1846. {
  1847. if(ba.actionType == EActionType::HERO_SPELL)
  1848. gs->getBattle(battleID)->getSide(ba.side).usedSpellsHistory.push_back(ba.spell);
  1849. }
  1850. }
  1851. void BattleSpellCast::applyGs(CGameState *gs)
  1852. {
  1853. if(castByHero && side != BattleSide::NONE)
  1854. gs->getBattle(battleID)->getSide(side).castSpellsCount++;
  1855. }
  1856. void SetStackEffect::applyGs(CGameState *gs)
  1857. {
  1858. applyBattle(gs->getBattle(battleID));
  1859. }
  1860. void SetStackEffect::applyBattle(IBattleState * battleState)
  1861. {
  1862. for(const auto & stackData : toRemove)
  1863. battleState->removeUnitBonus(stackData.first, stackData.second);
  1864. for(const auto & stackData : toUpdate)
  1865. battleState->updateUnitBonus(stackData.first, stackData.second);
  1866. for(const auto & stackData : toAdd)
  1867. battleState->addUnitBonus(stackData.first, stackData.second);
  1868. }
  1869. void StacksInjured::applyGs(CGameState *gs)
  1870. {
  1871. applyBattle(gs->getBattle(battleID));
  1872. }
  1873. void StacksInjured::applyBattle(IBattleState * battleState)
  1874. {
  1875. for(BattleStackAttacked stackAttacked : stacks)
  1876. stackAttacked.applyBattle(battleState);
  1877. }
  1878. void BattleUnitsChanged::applyGs(CGameState *gs)
  1879. {
  1880. applyBattle(gs->getBattle(battleID));
  1881. }
  1882. void BattleUnitsChanged::applyBattle(IBattleState * battleState)
  1883. {
  1884. for(auto & elem : changedStacks)
  1885. {
  1886. switch(elem.operation)
  1887. {
  1888. case BattleChanges::EOperation::RESET_STATE:
  1889. battleState->setUnitState(elem.id, elem.data, elem.healthDelta);
  1890. break;
  1891. case BattleChanges::EOperation::REMOVE:
  1892. battleState->removeUnit(elem.id);
  1893. break;
  1894. case BattleChanges::EOperation::ADD:
  1895. battleState->addUnit(elem.id, elem.data);
  1896. break;
  1897. case BattleChanges::EOperation::UPDATE:
  1898. battleState->updateUnit(elem.id, elem.data);
  1899. break;
  1900. default:
  1901. logNetwork->error("Unknown unit operation %d", static_cast<int>(elem.operation));
  1902. break;
  1903. }
  1904. }
  1905. }
  1906. void BattleResultsApplied::applyGs(CGameState * gs)
  1907. {
  1908. learnedSpells.applyGs(gs);
  1909. for(auto & artPack : artifacts)
  1910. artPack.applyGs(gs);
  1911. const auto currentBattle = std::find_if(gs->currentBattles.begin(), gs->currentBattles.end(),
  1912. [this](const auto & battle)
  1913. {
  1914. return battle->battleID == battleID;
  1915. });
  1916. assert(currentBattle != gs->currentBattles.end());
  1917. gs->currentBattles.erase(currentBattle);
  1918. }
  1919. void BattleObstaclesChanged::applyGs(CGameState *gs)
  1920. {
  1921. applyBattle(gs->getBattle(battleID));
  1922. }
  1923. void BattleObstaclesChanged::applyBattle(IBattleState * battleState)
  1924. {
  1925. for(const auto & change : changes)
  1926. {
  1927. switch(change.operation)
  1928. {
  1929. case BattleChanges::EOperation::REMOVE:
  1930. battleState->removeObstacle(change.id);
  1931. break;
  1932. case BattleChanges::EOperation::ADD:
  1933. battleState->addObstacle(change);
  1934. break;
  1935. case BattleChanges::EOperation::UPDATE:
  1936. battleState->updateObstacle(change);
  1937. break;
  1938. default:
  1939. logNetwork->error("Unknown obstacle operation %d", static_cast<int>(change.operation));
  1940. break;
  1941. }
  1942. }
  1943. }
  1944. CatapultAttack::CatapultAttack() = default;
  1945. CatapultAttack::~CatapultAttack() = default;
  1946. void CatapultAttack::applyGs(CGameState *gs)
  1947. {
  1948. applyBattle(gs->getBattle(battleID));
  1949. }
  1950. void CatapultAttack::visitTyped(ICPackVisitor & visitor)
  1951. {
  1952. visitor.visitCatapultAttack(*this);
  1953. }
  1954. void CatapultAttack::applyBattle(IBattleState * battleState)
  1955. {
  1956. const auto * town = battleState->getDefendedTown();
  1957. if(!town)
  1958. return;
  1959. if(town->fortificationsLevel().wallsHealth == 0)
  1960. return;
  1961. for(const auto & part : attackedParts)
  1962. {
  1963. auto newWallState = SiegeInfo::applyDamage(battleState->getWallState(part.attackedPart), part.damageDealt);
  1964. battleState->setWallState(part.attackedPart, newWallState);
  1965. }
  1966. }
  1967. void BattleSetStackProperty::applyGs(CGameState *gs)
  1968. {
  1969. CStack * stack = gs->getBattle(battleID)->getStack(stackID, false);
  1970. switch(which)
  1971. {
  1972. case CASTS:
  1973. {
  1974. if(absolute)
  1975. logNetwork->error("Can not change casts in absolute mode");
  1976. else
  1977. stack->casts.use(-val);
  1978. break;
  1979. }
  1980. case ENCHANTER_COUNTER:
  1981. {
  1982. auto & counter = gs->getBattle(battleID)->getSide(gs->getBattle(battleID)->whatSide(stack->unitOwner())).enchanterCounter;
  1983. if(absolute)
  1984. counter = val;
  1985. else
  1986. counter += val;
  1987. vstd::amax(counter, 0);
  1988. break;
  1989. }
  1990. case UNBIND:
  1991. {
  1992. stack->removeBonusesRecursive(Selector::type()(BonusType::BIND_EFFECT));
  1993. break;
  1994. }
  1995. case CLONED:
  1996. {
  1997. stack->cloned = true;
  1998. break;
  1999. }
  2000. case HAS_CLONE:
  2001. {
  2002. stack->cloneID = val;
  2003. break;
  2004. }
  2005. }
  2006. }
  2007. void PlayerCheated::applyGs(CGameState *gs)
  2008. {
  2009. if(!player.isValidPlayer())
  2010. return;
  2011. gs->getPlayerState(player)->enteredLosingCheatCode = losingCheatCode;
  2012. gs->getPlayerState(player)->enteredWinningCheatCode = winningCheatCode;
  2013. gs->getPlayerState(player)->cheated = true;
  2014. }
  2015. void PlayerStartsTurn::applyGs(CGameState *gs)
  2016. {
  2017. //assert(gs->actingPlayers.count(player) == 0);//Legal - may happen after loading of deserialized map state
  2018. gs->actingPlayers.insert(player);
  2019. }
  2020. void PlayerEndsTurn::applyGs(CGameState *gs)
  2021. {
  2022. assert(gs->actingPlayers.count(player) == 1);
  2023. gs->actingPlayers.erase(player);
  2024. }
  2025. void DaysWithoutTown::applyGs(CGameState *gs)
  2026. {
  2027. auto & playerState = gs->players.at(player);
  2028. playerState.daysWithoutCastle = daysWithoutCastle;
  2029. }
  2030. void TurnTimeUpdate::applyGs(CGameState *gs)
  2031. {
  2032. auto & playerState = gs->players.at(player);
  2033. playerState.turnTimer = turnTimer;
  2034. }
  2035. void EntitiesChanged::applyGs(CGameState *gs)
  2036. {
  2037. for(const auto & change : changes)
  2038. gs->updateEntity(change.metatype, change.entityIndex, change.data);
  2039. }
  2040. void SetRewardableConfiguration::applyGs(CGameState *gs)
  2041. {
  2042. auto * objectPtr = gs->getObjInstance(objectID);
  2043. if (!buildingID.hasValue())
  2044. {
  2045. auto * rewardablePtr = dynamic_cast<CRewardableObject *>(objectPtr);
  2046. assert(rewardablePtr);
  2047. rewardablePtr->configuration = configuration;
  2048. rewardablePtr->initializeGuards();
  2049. }
  2050. else
  2051. {
  2052. auto * townPtr = dynamic_cast<CGTownInstance*>(objectPtr);
  2053. TownBuildingInstance * buildingPtr = nullptr;
  2054. for (auto & building : townPtr->rewardableBuildings)
  2055. if (building.second->getBuildingType() == buildingID)
  2056. buildingPtr = building.second.get();
  2057. auto * rewardablePtr = dynamic_cast<TownRewardableBuildingInstance *>(buildingPtr);
  2058. assert(rewardablePtr);
  2059. rewardablePtr->configuration = configuration;
  2060. }
  2061. }
  2062. ArtSlotInfo::ArtSlotInfo(IGameCallback * cb)
  2063. : GameCallbackHolder(cb)
  2064. {
  2065. }
  2066. ArtSlotInfo::ArtSlotInfo(const CArtifactInstance * artifact, bool locked)
  2067. : GameCallbackHolder(artifact->cb)
  2068. , artifactID(artifact->getId())
  2069. , locked(locked)
  2070. {
  2071. }
  2072. const CArtifactInstance * ArtSlotInfo::getArt() const
  2073. {
  2074. if(locked || !artifactID.hasValue())
  2075. return nullptr;
  2076. return cb->getArtInstance(artifactID);
  2077. }
  2078. ArtifactInstanceID ArtSlotInfo::getID() const
  2079. {
  2080. if(locked)
  2081. return {};
  2082. return artifactID;
  2083. }
  2084. VCMI_LIB_NAMESPACE_END