CMapGenerator.cpp 6.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195
  1. #include "StdInc.h"
  2. #include "CMapGenerator.h"
  3. #include "../mapping/CMap.h"
  4. #include "../VCMI_Lib.h"
  5. #include "../CGeneralTextHandler.h"
  6. #include "../mapping/CMapEditManager.h"
  7. #include "../CObjectHandler.h"
  8. #include "../CDefObjInfoHandler.h"
  9. #include "../CTownHandler.h"
  10. #include "../StringConstants.h"
  11. #include "../filesystem/Filesystem.h"
  12. #include "CRmgTemplate.h"
  13. #include "CRmgTemplateZone.h"
  14. CMapGenerator::CMapGenerator(shared_ptr<CMapGenOptions> mapGenOptions, int randomSeed /*= std::time(nullptr)*/) :
  15. mapGenOptions(mapGenOptions), randomSeed(randomSeed)
  16. {
  17. rand.setSeed(randomSeed);
  18. }
  19. CMapGenerator::~CMapGenerator()
  20. {
  21. }
  22. std::unique_ptr<CMap> CMapGenerator::generate()
  23. {
  24. mapGenOptions->finalize(rand);
  25. map = make_unique<CMap>();
  26. editManager = map->getEditManager();
  27. try
  28. {
  29. editManager->getUndoManager().setUndoRedoLimit(0);
  30. addHeaderInfo();
  31. genZones();
  32. map->calculateGuardingGreaturePositions(); //clear map so that all tiles are unguarded
  33. fillZones();
  34. }
  35. catch (rmgException &e)
  36. {
  37. logGlobal->errorStream() << "Random map generation received exception: " << e.what();
  38. }
  39. return std::move(map);
  40. }
  41. std::string CMapGenerator::getMapDescription() const
  42. {
  43. const std::string waterContentStr[3] = { "none", "normal", "islands" };
  44. const std::string monsterStrengthStr[3] = { "weak", "normal", "strong" };
  45. std::stringstream ss;
  46. ss << boost::str(boost::format(std::string("Map created by the Random Map Generator.\nTemplate was %s, Random seed was %d, size %dx%d") +
  47. ", levels %s, humans %d, computers %d, water %s, monster %s, second expansion map") % mapGenOptions->getMapTemplate()->getName() %
  48. randomSeed % map->width % map->height % (map->twoLevel ? "2" : "1") % static_cast<int>(mapGenOptions->getPlayerCount()) %
  49. static_cast<int>(mapGenOptions->getCompOnlyPlayerCount()) % waterContentStr[mapGenOptions->getWaterContent()] %
  50. monsterStrengthStr[mapGenOptions->getMonsterStrength()]);
  51. for(const auto & pair : mapGenOptions->getPlayersSettings())
  52. {
  53. const auto & pSettings = pair.second;
  54. if(pSettings.getPlayerType() == EPlayerType::HUMAN)
  55. {
  56. ss << ", " << GameConstants::PLAYER_COLOR_NAMES[pSettings.getColor().getNum()] << " is human";
  57. }
  58. if(pSettings.getStartingTown() != CMapGenOptions::CPlayerSettings::RANDOM_TOWN)
  59. {
  60. ss << ", " << GameConstants::PLAYER_COLOR_NAMES[pSettings.getColor().getNum()]
  61. << " town choice is " << VLC->townh->factions[pSettings.getStartingTown()]->name;
  62. }
  63. }
  64. return ss.str();
  65. }
  66. void CMapGenerator::addPlayerInfo()
  67. {
  68. // Calculate which team numbers exist
  69. std::array<std::list<int>, 2> teamNumbers; // 0= cpu/human, 1= cpu only
  70. int teamOffset = 0;
  71. for(int i = 0; i < 2; ++i)
  72. {
  73. int playerCount = i == 0 ? mapGenOptions->getPlayerCount() : mapGenOptions->getCompOnlyPlayerCount();
  74. int teamCount = i == 0 ? mapGenOptions->getTeamCount() : mapGenOptions->getCompOnlyTeamCount();
  75. if(playerCount == 0)
  76. {
  77. continue;
  78. }
  79. int playersPerTeam = playerCount /
  80. (teamCount == 0 ? playerCount : teamCount);
  81. int teamCountNorm = teamCount;
  82. if(teamCountNorm == 0)
  83. {
  84. teamCountNorm = playerCount;
  85. }
  86. for(int j = 0; j < teamCountNorm; ++j)
  87. {
  88. for(int k = 0; k < playersPerTeam; ++k)
  89. {
  90. teamNumbers[i].push_back(j + teamOffset);
  91. }
  92. }
  93. for(int j = 0; j < playerCount - teamCountNorm * playersPerTeam; ++j)
  94. {
  95. teamNumbers[i].push_back(j + teamOffset);
  96. }
  97. teamOffset += teamCountNorm;
  98. }
  99. // Team numbers are assigned randomly to every player
  100. for(const auto & pair : mapGenOptions->getPlayersSettings())
  101. {
  102. const auto & pSettings = pair.second;
  103. PlayerInfo player;
  104. player.canComputerPlay = true;
  105. int j = pSettings.getPlayerType() == EPlayerType::COMP_ONLY ? 1 : 0;
  106. if(j == 0)
  107. {
  108. player.canHumanPlay = true;
  109. }
  110. auto itTeam = std::next(teamNumbers[j].begin(), rand.nextInt (teamNumbers[j].size()));
  111. player.team = TeamID(*itTeam);
  112. teamNumbers[j].erase(itTeam);
  113. map->players[pSettings.getColor().getNum()] = player;
  114. }
  115. map->howManyTeams = (mapGenOptions->getTeamCount() == 0 ? mapGenOptions->getPlayerCount() : mapGenOptions->getTeamCount())
  116. + (mapGenOptions->getCompOnlyTeamCount() == 0 ? mapGenOptions->getCompOnlyPlayerCount() : mapGenOptions->getCompOnlyTeamCount());
  117. }
  118. void CMapGenerator::genZones()
  119. {
  120. map->initTerrain();
  121. editManager->clearTerrain(&rand);
  122. editManager->getTerrainSelection().selectRange(MapRect(int3(0, 0, 0), mapGenOptions->getWidth(), mapGenOptions->getHeight()));
  123. editManager->drawTerrain(ETerrainType::GRASS, &rand);
  124. auto pcnt = mapGenOptions->getPlayerCount();
  125. auto w = mapGenOptions->getWidth();
  126. auto h = mapGenOptions->getHeight();
  127. auto tmpl = mapGenOptions->getMapTemplate();
  128. auto zones = tmpl->getZones();
  129. int player_per_side = zones.size() > 4 ? 3 : 2;
  130. int zones_cnt = zones.size() > 4 ? 9 : 4;
  131. logGlobal->infoStream() << boost::format("Map size %d %d, players per side %d") % w % h % player_per_side;
  132. int i = 0;
  133. int part_w = w/player_per_side;
  134. int part_h = h/player_per_side;
  135. for(auto const it : zones)
  136. {
  137. CRmgTemplateZone * zone = it.second;
  138. std::vector<int3> shape;
  139. int left = part_w*(i%player_per_side);
  140. int top = part_h*(i/player_per_side);
  141. shape.push_back(int3(left, top, 0));
  142. shape.push_back(int3(left + part_w, top, 0));
  143. shape.push_back(int3(left + part_w, top + part_h, 0));
  144. shape.push_back(int3(left, top + part_h, 0));
  145. zone->setShape(shape);
  146. zone->setType(i < pcnt ? ETemplateZoneType::PLAYER_START : ETemplateZoneType::TREASURE);
  147. this->zones[it.first] = zone;
  148. ++i;
  149. }
  150. logGlobal->infoStream() << "Zones generated successfully";
  151. }
  152. void CMapGenerator::fillZones()
  153. {
  154. logGlobal->infoStream() << "Started filling zones";
  155. for(auto it : zones)
  156. {
  157. it.second->fill(this);
  158. }
  159. logGlobal->infoStream() << "Zones filled successfully";
  160. }
  161. void CMapGenerator::addHeaderInfo()
  162. {
  163. map->version = EMapFormat::SOD;
  164. map->width = mapGenOptions->getWidth();
  165. map->height = mapGenOptions->getHeight();
  166. map->twoLevel = mapGenOptions->getHasTwoLevels();
  167. map->name = VLC->generaltexth->allTexts[740];
  168. map->description = getMapDescription();
  169. map->difficulty = 1;
  170. addPlayerInfo();
  171. }