CMapGenOptions.cpp 21 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821
  1. /*
  2. * CMapGenOptions.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 "CMapGenOptions.h"
  12. #include "../mapping/CMapHeader.h"
  13. #include "CRmgTemplateStorage.h"
  14. #include "CRmgTemplate.h"
  15. #include "CRandomGenerator.h"
  16. #include "../VCMI_Lib.h"
  17. #include "../CTownHandler.h"
  18. VCMI_LIB_NAMESPACE_BEGIN
  19. CMapGenOptions::CMapGenOptions()
  20. : width(CMapHeader::MAP_SIZE_MIDDLE), height(CMapHeader::MAP_SIZE_MIDDLE), hasTwoLevels(true),
  21. humanOrCpuPlayerCount(RANDOM_SIZE), teamCount(RANDOM_SIZE), compOnlyPlayerCount(RANDOM_SIZE), compOnlyTeamCount(RANDOM_SIZE),
  22. waterContent(EWaterContent::RANDOM), monsterStrength(EMonsterStrength::RANDOM), mapTemplate(nullptr),
  23. customizedPlayers(false)
  24. {
  25. initPlayersMap();
  26. setRoadEnabled(RoadId(Road::DIRT_ROAD), true);
  27. setRoadEnabled(RoadId(Road::GRAVEL_ROAD), true);
  28. setRoadEnabled(RoadId(Road::COBBLESTONE_ROAD), true);
  29. }
  30. si32 CMapGenOptions::getWidth() const
  31. {
  32. return width;
  33. }
  34. void CMapGenOptions::setWidth(si32 value)
  35. {
  36. assert(value >= 1);
  37. width = value;
  38. }
  39. si32 CMapGenOptions::getHeight() const
  40. {
  41. return height;
  42. }
  43. void CMapGenOptions::setHeight(si32 value)
  44. {
  45. assert(value >= 1);
  46. height = value;
  47. }
  48. bool CMapGenOptions::getHasTwoLevels() const
  49. {
  50. return hasTwoLevels;
  51. }
  52. void CMapGenOptions::setHasTwoLevels(bool value)
  53. {
  54. hasTwoLevels = value;
  55. }
  56. si8 CMapGenOptions::getHumanOrCpuPlayerCount() const
  57. {
  58. return humanOrCpuPlayerCount;
  59. }
  60. void CMapGenOptions::setHumanOrCpuPlayerCount(si8 value)
  61. {
  62. assert((value >= 1 && value <= PlayerColor::PLAYER_LIMIT_I) || value == RANDOM_SIZE);
  63. humanOrCpuPlayerCount = value;
  64. // Use template player limit, if any
  65. auto playerLimit = getPlayerLimit();
  66. auto possibleCompPlayersCount = playerLimit - std::max<si8>(0, humanOrCpuPlayerCount);
  67. if (compOnlyPlayerCount > possibleCompPlayersCount)
  68. {
  69. setCompOnlyPlayerCount(possibleCompPlayersCount);
  70. }
  71. resetPlayersMap();
  72. }
  73. si8 CMapGenOptions::getMinPlayersCount(bool withTemplateLimit) const
  74. {
  75. auto totalPlayers = 0;
  76. si8 humans = getHumanOrCpuPlayerCount();
  77. si8 cpus = getCompOnlyPlayerCount();
  78. if (humans == RANDOM_SIZE && cpus == RANDOM_SIZE)
  79. {
  80. totalPlayers = 2;
  81. }
  82. else if (humans == RANDOM_SIZE)
  83. {
  84. totalPlayers = cpus + 1; // Must add at least 1 player
  85. }
  86. else if (cpus == RANDOM_SIZE)
  87. {
  88. totalPlayers = humans;
  89. }
  90. else
  91. {
  92. totalPlayers = humans + cpus;
  93. }
  94. if (withTemplateLimit && mapTemplate)
  95. {
  96. auto playersRange = mapTemplate->getPlayers();
  97. //New template can also impose higher limit than current settings
  98. vstd::amax(totalPlayers, playersRange.minValue());
  99. }
  100. // Can't play without at least 2 players
  101. vstd::amax(totalPlayers, 2);
  102. return totalPlayers;
  103. }
  104. si8 CMapGenOptions::getMaxPlayersCount(bool withTemplateLimit) const
  105. {
  106. // Max number of players possible with current settings
  107. auto totalPlayers = 0;
  108. si8 humans = getHumanOrCpuPlayerCount();
  109. si8 cpus = getCompOnlyPlayerCount();
  110. if (humans == RANDOM_SIZE || cpus == RANDOM_SIZE)
  111. {
  112. totalPlayers = PlayerColor::PLAYER_LIMIT_I;
  113. }
  114. else
  115. {
  116. totalPlayers = humans + cpus;
  117. }
  118. if (withTemplateLimit && mapTemplate)
  119. {
  120. auto playersRange = mapTemplate->getPlayers();
  121. //New template can also impose higher limit than current settings
  122. vstd::amin(totalPlayers, playersRange.maxValue());
  123. }
  124. assert (totalPlayers <= PlayerColor::PLAYER_LIMIT_I);
  125. assert (totalPlayers >= 2);
  126. return totalPlayers;
  127. }
  128. si8 CMapGenOptions::getTeamCount() const
  129. {
  130. return teamCount;
  131. }
  132. void CMapGenOptions::setTeamCount(si8 value)
  133. {
  134. assert(getHumanOrCpuPlayerCount() == RANDOM_SIZE || (value >= 0 && value < getHumanOrCpuPlayerCount()) || value == RANDOM_SIZE);
  135. teamCount = value;
  136. }
  137. si8 CMapGenOptions::getCompOnlyPlayerCount() const
  138. {
  139. return compOnlyPlayerCount;
  140. }
  141. si8 CMapGenOptions::getPlayerLimit() const
  142. {
  143. //How many players could we set with current template, ignoring other settings
  144. si8 playerLimit = PlayerColor::PLAYER_LIMIT_I;
  145. if (auto temp = getMapTemplate())
  146. {
  147. playerLimit = static_cast<si8>(temp->getPlayers().maxValue());
  148. }
  149. return playerLimit;
  150. }
  151. void CMapGenOptions::setCompOnlyPlayerCount(si8 value)
  152. {
  153. auto playerLimit = getPlayerLimit();
  154. assert(value == RANDOM_SIZE || (getHumanOrCpuPlayerCount() == RANDOM_SIZE || (value >= 0 && value <= playerLimit - getHumanOrCpuPlayerCount())));
  155. compOnlyPlayerCount = value;
  156. resetPlayersMap();
  157. }
  158. si8 CMapGenOptions::getCompOnlyTeamCount() const
  159. {
  160. return compOnlyTeamCount;
  161. }
  162. void CMapGenOptions::setCompOnlyTeamCount(si8 value)
  163. {
  164. assert(value == RANDOM_SIZE || compOnlyPlayerCount == RANDOM_SIZE || (value >= 0 && value <= std::max(compOnlyPlayerCount - 1, 0)));
  165. compOnlyTeamCount = value;
  166. }
  167. EWaterContent::EWaterContent CMapGenOptions::getWaterContent() const
  168. {
  169. return waterContent;
  170. }
  171. void CMapGenOptions::setWaterContent(EWaterContent::EWaterContent value)
  172. {
  173. waterContent = value;
  174. }
  175. EMonsterStrength::EMonsterStrength CMapGenOptions::getMonsterStrength() const
  176. {
  177. return monsterStrength;
  178. }
  179. void CMapGenOptions::setMonsterStrength(EMonsterStrength::EMonsterStrength value)
  180. {
  181. monsterStrength = value;
  182. }
  183. void CMapGenOptions::initPlayersMap()
  184. {
  185. players.clear();
  186. int realPlayersCnt = getHumanOrCpuPlayerCount();
  187. // Initialize settings for all color even if not present
  188. for(int color = 0; color < getPlayerLimit(); ++color)
  189. {
  190. CPlayerSettings player;
  191. auto pc = PlayerColor(color);
  192. player.setColor(pc);
  193. auto playerType = EPlayerType::AI;
  194. // Color doesn't have to be continuous. Player colors can later be changed manually
  195. if (getHumanOrCpuPlayerCount() != RANDOM_SIZE && color < realPlayersCnt)
  196. {
  197. playerType = EPlayerType::HUMAN;
  198. }
  199. else if((getHumanOrCpuPlayerCount() != RANDOM_SIZE && color >= realPlayersCnt)
  200. || (compOnlyPlayerCount != RANDOM_SIZE && color >= (PlayerColor::PLAYER_LIMIT_I - compOnlyPlayerCount)))
  201. {
  202. playerType = EPlayerType::COMP_ONLY;
  203. }
  204. player.setPlayerType(playerType);
  205. players[pc] = player;
  206. }
  207. savePlayersMap();
  208. }
  209. void CMapGenOptions::resetPlayersMap()
  210. {
  211. // Called when number of player changes
  212. // But do not update info about already made selections
  213. savePlayersMap();
  214. int realPlayersCnt = getMaxPlayersCount();
  215. //Trim the number of AI players, then CPU-only players, finally human players
  216. auto eraseLastPlayer = [this](EPlayerType playerType) -> bool
  217. {
  218. for (auto it = players.rbegin(); it != players.rend(); ++it)
  219. {
  220. if (it->second.getPlayerType() == playerType)
  221. {
  222. players.erase(it->first);
  223. return true;
  224. }
  225. }
  226. return false; //Can't earse any player of this type
  227. };
  228. while (players.size() > realPlayersCnt)
  229. {
  230. while (eraseLastPlayer(EPlayerType::AI));
  231. while (eraseLastPlayer(EPlayerType::COMP_ONLY));
  232. while (eraseLastPlayer(EPlayerType::HUMAN));
  233. }
  234. //First colors from the list are assigned to human players, then to CPU players
  235. std::vector<PlayerColor> availableColors;
  236. for (ui8 color = 0; color < PlayerColor::PLAYER_LIMIT_I; color++)
  237. {
  238. availableColors.push_back(PlayerColor(color));
  239. }
  240. auto removeUsedColors = [this, &availableColors](EPlayerType playerType)
  241. {
  242. for (auto& player : players)
  243. {
  244. if (player.second.getPlayerType() == playerType)
  245. {
  246. vstd::erase(availableColors, player.second.getColor());
  247. }
  248. }
  249. };
  250. removeUsedColors(EPlayerType::HUMAN);
  251. removeUsedColors(EPlayerType::COMP_ONLY);
  252. //removeUsedColors(EPlayerType::AI);
  253. //Assign unused colors to remaining AI players
  254. while (players.size() < realPlayersCnt && !availableColors.empty())
  255. {
  256. auto color = availableColors.front();
  257. setPlayerTypeForStandardPlayer(color, EPlayerType::AI);
  258. players[color].setColor(color);
  259. availableColors.erase(availableColors.begin());
  260. if (vstd::contains(savedPlayerSettings, color))
  261. {
  262. setPlayerTeam(color, savedPlayerSettings.at(color).getTeam());
  263. // TODO: setter
  264. players[color].setStartingTown(savedPlayerSettings.at(color).getStartingTown());
  265. }
  266. else
  267. {
  268. logGlobal->warn("Adding settings for player %s", color.encode(color));
  269. // Usually, all players should be initialized in initPlayersMap()
  270. CPlayerSettings settings;
  271. players[color] = settings;
  272. }
  273. }
  274. std::set<TeamID> occupiedTeams;
  275. for(auto & player : players)
  276. {
  277. auto team = player.second.getTeam();
  278. if (team != TeamID::NO_TEAM)
  279. {
  280. occupiedTeams.insert(team);
  281. }
  282. }
  283. // TODO: Handle situation when we remove a player and remaining players belong to only one team
  284. for(auto & player : players)
  285. {
  286. if (player.second.getTeam() == TeamID::NO_TEAM)
  287. {
  288. //Find first unused team
  289. for(int i = 0; i < PlayerColor::PLAYER_LIMIT_I; ++i)
  290. {
  291. TeamID team(i);
  292. if(!occupiedTeams.count(team))
  293. {
  294. player.second.setTeam(team);
  295. occupiedTeams.insert(team);
  296. break;
  297. }
  298. }
  299. }
  300. }
  301. /*
  302. for(int color = 0; color < totalPlayersLimit; ++color)
  303. {
  304. CPlayerSettings player;
  305. auto pc = PlayerColor(color);
  306. player.setColor(pc);
  307. auto playerType = EPlayerType::AI;
  308. if (getPlayerCount() != RANDOM_SIZE && color < realPlayersCnt)
  309. {
  310. playerType = EPlayerType::HUMAN;
  311. }
  312. else if((getPlayerCount() != RANDOM_SIZE && color >= realPlayersCnt)
  313. || (compOnlyPlayerCount != RANDOM_SIZE && color >= (PlayerColor::PLAYER_LIMIT_I-compOnlyPlayerCount)))
  314. {
  315. //FIXME: Allow humans to choose any color, even from the end of the list
  316. playerType = EPlayerType::COMP_ONLY;
  317. }
  318. player.setPlayerType(playerType);
  319. player.setTeam(rememberTeam[pc]);
  320. players[pc] = player;
  321. if (vstd::contains(rememberTownTypes, pc))
  322. players[pc].setStartingTown(rememberTownTypes[pc]);
  323. }
  324. */
  325. }
  326. void CMapGenOptions::savePlayersMap()
  327. {
  328. //Only save already configured players
  329. for (const auto& player : players)
  330. {
  331. savedPlayerSettings[player.first] = player.second;
  332. }
  333. }
  334. const std::map<PlayerColor, CMapGenOptions::CPlayerSettings> & CMapGenOptions::getSavedPlayersMap() const
  335. {
  336. return savedPlayerSettings;
  337. }
  338. const std::map<PlayerColor, CMapGenOptions::CPlayerSettings> & CMapGenOptions::getPlayersSettings() const
  339. {
  340. return players;
  341. }
  342. void CMapGenOptions::setStartingTownForPlayer(const PlayerColor & color, si32 town)
  343. {
  344. auto it = players.find(color);
  345. assert(it != players.end());
  346. it->second.setStartingTown(town);
  347. }
  348. void CMapGenOptions::setPlayerTypeForStandardPlayer(const PlayerColor & color, EPlayerType playerType)
  349. {
  350. // FIXME: Why actually not set it to COMP_ONLY? Ie. when swapping human to another color?
  351. assert(playerType != EPlayerType::COMP_ONLY);
  352. auto it = players.find(color);
  353. assert(it != players.end());
  354. it->second.setPlayerType(playerType);
  355. customizedPlayers = true;
  356. }
  357. const CRmgTemplate * CMapGenOptions::getMapTemplate() const
  358. {
  359. return mapTemplate;
  360. }
  361. void CMapGenOptions::setMapTemplate(const CRmgTemplate * value)
  362. {
  363. if (mapTemplate == value)
  364. {
  365. //Does not trigger during deserialization
  366. return;
  367. }
  368. mapTemplate = value;
  369. //validate & adapt options according to template
  370. if(mapTemplate)
  371. {
  372. if(!mapTemplate->matchesSize(int3(getWidth(), getHeight(), 1 + getHasTwoLevels())))
  373. {
  374. auto sizes = mapTemplate->getMapSizes();
  375. setWidth(sizes.first.x);
  376. setHeight(sizes.first.y);
  377. setHasTwoLevels(sizes.first.z - 1);
  378. }
  379. si8 maxPlayerCount = getMaxPlayersCount(false);
  380. si8 minPlayerCount = getMinPlayersCount(false);
  381. // Neither setting can fit within the template range
  382. if(!mapTemplate->getPlayers().isInRange(minPlayerCount) &&
  383. !mapTemplate->getPlayers().isInRange(maxPlayerCount))
  384. {
  385. setHumanOrCpuPlayerCount(RANDOM_SIZE);
  386. setCompOnlyPlayerCount(RANDOM_SIZE);
  387. }
  388. if(!mapTemplate->getWaterContentAllowed().count(getWaterContent()))
  389. setWaterContent(EWaterContent::RANDOM);
  390. resetPlayersMap(); // Update teams and player count
  391. }
  392. }
  393. void CMapGenOptions::setMapTemplate(const std::string & name)
  394. {
  395. if(!name.empty())
  396. setMapTemplate(VLC->tplh->getTemplate(name));
  397. }
  398. void CMapGenOptions::setRoadEnabled(const RoadId & roadType, bool enable)
  399. {
  400. if (enable)
  401. {
  402. enabledRoads.insert(roadType);
  403. }
  404. else
  405. {
  406. enabledRoads.erase(roadType);
  407. }
  408. }
  409. bool CMapGenOptions::isRoadEnabled(const RoadId & roadType) const
  410. {
  411. return enabledRoads.count(roadType);
  412. }
  413. bool CMapGenOptions::isRoadEnabled() const
  414. {
  415. return !enabledRoads.empty();
  416. }
  417. void CMapGenOptions::setPlayerTeam(const PlayerColor & color, const TeamID & team)
  418. {
  419. auto it = players.find(color);
  420. assert(it != players.end());
  421. it->second.setTeam(team);
  422. customizedPlayers = true;
  423. }
  424. void CMapGenOptions::finalize(CRandomGenerator & rand)
  425. {
  426. logGlobal->info("RMG map: %dx%d, %s underground", getWidth(), getHeight(), getHasTwoLevels() ? "WITH" : "NO");
  427. logGlobal->info("RMG settings: players %d, teams %d, computer players %d, computer teams %d, water %d, monsters %d",
  428. static_cast<int>(getHumanOrCpuPlayerCount()), static_cast<int>(getTeamCount()), static_cast<int>(getCompOnlyPlayerCount()),
  429. static_cast<int>(getCompOnlyTeamCount()), static_cast<int>(getWaterContent()), static_cast<int>(getMonsterStrength()));
  430. if(!mapTemplate)
  431. {
  432. mapTemplate = getPossibleTemplate(rand);
  433. }
  434. assert(mapTemplate);
  435. logGlobal->info("RMG template name: %s", mapTemplate->getName());
  436. auto maxPlayers = getMaxPlayersCount();
  437. if (getHumanOrCpuPlayerCount() == RANDOM_SIZE)
  438. {
  439. auto possiblePlayers = mapTemplate->getPlayers().getNumbers();
  440. //ignore all non-randomized players, make sure these players will not be missing after roll
  441. possiblePlayers.erase(possiblePlayers.begin(), possiblePlayers.lower_bound(countHumanPlayers() + countCompOnlyPlayers()));
  442. vstd::erase_if(possiblePlayers, [maxPlayers](int i)
  443. {
  444. return i > maxPlayers;
  445. });
  446. assert(!possiblePlayers.empty());
  447. setHumanOrCpuPlayerCount (*RandomGeneratorUtil::nextItem(possiblePlayers, rand));
  448. updatePlayers();
  449. }
  450. if(teamCount == RANDOM_SIZE)
  451. {
  452. teamCount = rand.nextInt(getHumanOrCpuPlayerCount() - 1);
  453. if (teamCount == 1)
  454. teamCount = 0;
  455. }
  456. if(compOnlyPlayerCount == RANDOM_SIZE)
  457. {
  458. // Use remaining range
  459. auto presentPlayers = getHumanOrCpuPlayerCount();
  460. auto possiblePlayers = mapTemplate->getPlayers().getNumbers();
  461. vstd::erase_if(possiblePlayers, [maxPlayers, presentPlayers](int i)
  462. {
  463. return i > (maxPlayers - presentPlayers);
  464. });
  465. if (possiblePlayers.empty())
  466. {
  467. compOnlyPlayerCount = 0;
  468. }
  469. else
  470. {
  471. compOnlyPlayerCount = *RandomGeneratorUtil::nextItem(possiblePlayers, rand);
  472. }
  473. updateCompOnlyPlayers();
  474. }
  475. if(compOnlyTeamCount == RANDOM_SIZE)
  476. {
  477. compOnlyTeamCount = rand.nextInt(std::max(compOnlyPlayerCount - 1, 0));
  478. }
  479. if(waterContent == EWaterContent::RANDOM)
  480. {
  481. auto allowedContent = mapTemplate->getWaterContentAllowed();
  482. if(!allowedContent.empty())
  483. {
  484. waterContent = *RandomGeneratorUtil::nextItem(mapTemplate->getWaterContentAllowed(), rand);
  485. }
  486. else
  487. {
  488. waterContent = EWaterContent::NONE;
  489. }
  490. }
  491. if(monsterStrength == EMonsterStrength::RANDOM)
  492. {
  493. monsterStrength = static_cast<EMonsterStrength::EMonsterStrength>(rand.nextInt(EMonsterStrength::GLOBAL_WEAK, EMonsterStrength::GLOBAL_STRONG));
  494. }
  495. assert (vstd::iswithin(waterContent, EWaterContent::NONE, EWaterContent::ISLANDS));
  496. assert (vstd::iswithin(monsterStrength, EMonsterStrength::GLOBAL_WEAK, EMonsterStrength::GLOBAL_STRONG));
  497. logGlobal->trace("Player config:");
  498. int cpuOnlyPlayers = 0;
  499. for(const auto & player : players)
  500. {
  501. std::string playerType;
  502. switch (player.second.getPlayerType())
  503. {
  504. case EPlayerType::AI:
  505. playerType = "AI";
  506. break;
  507. case EPlayerType::COMP_ONLY:
  508. playerType = "computer only";
  509. cpuOnlyPlayers++;
  510. break;
  511. case EPlayerType::HUMAN:
  512. playerType = "human only";
  513. break;
  514. default:
  515. assert(false);
  516. }
  517. logGlobal->trace("Player %d: %s", player.second.getColor(), playerType);
  518. }
  519. logGlobal->info("Final player config: %d total, %d cpu-only", players.size(), static_cast<int>(getCompOnlyPlayerCount()));
  520. }
  521. void CMapGenOptions::updatePlayers()
  522. {
  523. // Remove non-human players only from the end of the players map if necessary
  524. for(auto itrev = players.end(); itrev != players.begin();)
  525. {
  526. auto it = itrev;
  527. --it;
  528. if (players.size() == getHumanOrCpuPlayerCount()) break;
  529. if(it->second.getPlayerType() != EPlayerType::HUMAN)
  530. {
  531. players.erase(it);
  532. }
  533. else
  534. {
  535. --itrev;
  536. }
  537. }
  538. }
  539. void CMapGenOptions::updateCompOnlyPlayers()
  540. {
  541. // Remove comp only players only from the end of the players map if necessary
  542. for(auto itrev = players.end(); itrev != players.begin();)
  543. {
  544. auto it = itrev;
  545. --it;
  546. if (players.size() <= getHumanOrCpuPlayerCount()) break;
  547. if(it->second.getPlayerType() == EPlayerType::COMP_ONLY)
  548. {
  549. players.erase(it);
  550. }
  551. else
  552. {
  553. --itrev;
  554. }
  555. }
  556. // Add some comp only players if necessary
  557. int compOnlyPlayersToAdd = static_cast<int>(getHumanOrCpuPlayerCount() - players.size());
  558. if (compOnlyPlayersToAdd < 0)
  559. {
  560. logGlobal->error("Incorrect number of players to add. Requested players %d, current players %d", humanOrCpuPlayerCount, players.size());
  561. assert (compOnlyPlayersToAdd < 0);
  562. }
  563. for(int i = 0; i < compOnlyPlayersToAdd; ++i)
  564. {
  565. CPlayerSettings pSettings;
  566. pSettings.setPlayerType(EPlayerType::COMP_ONLY);
  567. pSettings.setColor(getNextPlayerColor());
  568. players[pSettings.getColor()] = pSettings;
  569. }
  570. }
  571. int CMapGenOptions::countHumanPlayers() const
  572. {
  573. return static_cast<int>(boost::count_if(players, [](const std::pair<PlayerColor, CPlayerSettings> & pair)
  574. {
  575. return pair.second.getPlayerType() == EPlayerType::HUMAN;
  576. }));
  577. }
  578. int CMapGenOptions::countCompOnlyPlayers() const
  579. {
  580. return static_cast<int>(boost::count_if(players, [](const std::pair<PlayerColor, CPlayerSettings> & pair)
  581. {
  582. return pair.second.getPlayerType() == EPlayerType::COMP_ONLY;
  583. }));
  584. }
  585. PlayerColor CMapGenOptions::getNextPlayerColor() const
  586. {
  587. for(PlayerColor i = PlayerColor(0); i < PlayerColor::PLAYER_LIMIT; i.advance(1))
  588. {
  589. if(!players.count(i))
  590. {
  591. return i;
  592. }
  593. }
  594. logGlobal->error("Failed to get next player color");
  595. assert(false);
  596. return PlayerColor(0);
  597. }
  598. bool CMapGenOptions::checkOptions() const
  599. {
  600. if(mapTemplate)
  601. {
  602. return true;
  603. }
  604. else
  605. {
  606. CRandomGenerator gen;
  607. return getPossibleTemplate(gen) != nullptr;
  608. }
  609. }
  610. bool CMapGenOptions::arePlayersCustomized() const
  611. {
  612. return customizedPlayers;
  613. }
  614. std::vector<const CRmgTemplate *> CMapGenOptions::getPossibleTemplates() const
  615. {
  616. int3 tplSize(width, height, (hasTwoLevels ? 2 : 1));
  617. auto humanPlayers = countHumanPlayers();
  618. auto templates = VLC->tplh->getTemplates();
  619. vstd::erase_if(templates, [this, &tplSize, humanPlayers](const CRmgTemplate * tmpl)
  620. {
  621. if(!tmpl->matchesSize(tplSize))
  622. return true;
  623. if(!tmpl->isWaterContentAllowed(getWaterContent()))
  624. return true;
  625. auto humanOrCpuPlayerCount = getHumanOrCpuPlayerCount();
  626. auto compOnlyPlayerCount = getCompOnlyPlayerCount();
  627. // Check if total number of players fall inside given range
  628. if(humanOrCpuPlayerCount != CMapGenOptions::RANDOM_SIZE && compOnlyPlayerCount != CMapGenOptions::RANDOM_SIZE)
  629. {
  630. if (!tmpl->getPlayers().isInRange(humanOrCpuPlayerCount + compOnlyPlayerCount))
  631. return true;
  632. }
  633. else if(humanOrCpuPlayerCount != CMapGenOptions::RANDOM_SIZE)
  634. {
  635. // We can always add any number CPU players, but not subtract
  636. if (!(humanOrCpuPlayerCount <= tmpl->getPlayers().maxValue()))
  637. return true;
  638. }
  639. else if(compOnlyPlayerCount != CMapGenOptions::RANDOM_SIZE)
  640. {
  641. //We must fit at least one more human player, but can add any number
  642. if (!(compOnlyPlayerCount < tmpl->getPlayers().maxValue()))
  643. return true;
  644. }
  645. else
  646. {
  647. // Human players shouldn't be banned when playing with random player count
  648. if(humanPlayers > tmpl->getPlayers().minValue())
  649. return true;
  650. }
  651. if(compOnlyPlayerCount != CMapGenOptions::RANDOM_SIZE)
  652. {
  653. if (!tmpl->getHumanPlayers().isInRange(compOnlyPlayerCount))
  654. return true;
  655. }
  656. return false;
  657. });
  658. return templates;
  659. }
  660. const CRmgTemplate * CMapGenOptions::getPossibleTemplate(CRandomGenerator & rand) const
  661. {
  662. auto templates = getPossibleTemplates();
  663. if(templates.empty())
  664. return nullptr;
  665. return *RandomGeneratorUtil::nextItem(templates, rand);
  666. }
  667. CMapGenOptions::CPlayerSettings::CPlayerSettings() : color(0), startingTown(FactionID::RANDOM), playerType(EPlayerType::AI), team(TeamID::NO_TEAM)
  668. {
  669. }
  670. PlayerColor CMapGenOptions::CPlayerSettings::getColor() const
  671. {
  672. return color;
  673. }
  674. void CMapGenOptions::CPlayerSettings::setColor(const PlayerColor & value)
  675. {
  676. assert(value >= PlayerColor(0) && value < PlayerColor::PLAYER_LIMIT);
  677. color = value;
  678. }
  679. si32 CMapGenOptions::CPlayerSettings::getStartingTown() const
  680. {
  681. return startingTown;
  682. }
  683. void CMapGenOptions::CPlayerSettings::setStartingTown(si32 value)
  684. {
  685. assert(value >= -1);
  686. if(value >= 0)
  687. {
  688. assert(value < static_cast<int>(VLC->townh->size()));
  689. assert((*VLC->townh)[value]->town != nullptr);
  690. }
  691. startingTown = value;
  692. }
  693. EPlayerType CMapGenOptions::CPlayerSettings::getPlayerType() const
  694. {
  695. return playerType;
  696. }
  697. void CMapGenOptions::CPlayerSettings::setPlayerType(EPlayerType value)
  698. {
  699. playerType = value;
  700. }
  701. TeamID CMapGenOptions::CPlayerSettings::getTeam() const
  702. {
  703. return team;
  704. }
  705. void CMapGenOptions::CPlayerSettings::setTeam(const TeamID & value)
  706. {
  707. team = value;
  708. }
  709. VCMI_LIB_NAMESPACE_END