CMapGenOptions.cpp 20 KB

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