CMapEditManager.cpp 23 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904
  1. #include "StdInc.h"
  2. #include "CMapEditManager.h"
  3. #include "../JsonNode.h"
  4. #include "../filesystem/CResourceLoader.h"
  5. #include "../CDefObjInfoHandler.h"
  6. MapRect::MapRect() : x(0), y(0), z(0), width(0), height(0)
  7. {
  8. }
  9. MapRect::MapRect(int3 pos, si32 width, si32 height) : x(pos.x), y(pos.y), z(pos.z), width(width), height(height)
  10. {
  11. }
  12. MapRect MapRect::operator&(const MapRect & rect) const
  13. {
  14. bool intersect = right() > rect.left() && rect.right() > left() &&
  15. bottom() > rect.top() && rect.bottom() > top() &&
  16. z == rect.z;
  17. if(intersect)
  18. {
  19. MapRect ret;
  20. ret.x = std::max(left(), rect.left());
  21. ret.y = std::max(top(), rect.top());
  22. ret.z = rect.z;
  23. ret.width = std::min(right(), rect.right()) - ret.x;
  24. ret.height = std::min(bottom(), rect.bottom()) - ret.y;
  25. return ret;
  26. }
  27. else
  28. {
  29. return MapRect();
  30. }
  31. }
  32. si32 MapRect::left() const
  33. {
  34. return x;
  35. }
  36. si32 MapRect::right() const
  37. {
  38. return x + width;
  39. }
  40. si32 MapRect::top() const
  41. {
  42. return y;
  43. }
  44. si32 MapRect::bottom() const
  45. {
  46. return y + height;
  47. }
  48. int3 MapRect::topLeft() const
  49. {
  50. return int3(x, y, z);
  51. }
  52. int3 MapRect::topRight() const
  53. {
  54. return int3(right(), y, z);
  55. }
  56. int3 MapRect::bottomLeft() const
  57. {
  58. return int3(x, bottom(), z);
  59. }
  60. int3 MapRect::bottomRight() const
  61. {
  62. return int3(right(), bottom(), z);
  63. }
  64. CTerrainSelection::CTerrainSelection(CMap * map) : CMapSelection(map)
  65. {
  66. }
  67. void CTerrainSelection::selectRange(const MapRect & rect)
  68. {
  69. rect.forEach([this](const int3 pos)
  70. {
  71. this->select(pos);
  72. });
  73. }
  74. void CTerrainSelection::deselectRange(const MapRect & rect)
  75. {
  76. rect.forEach([this](const int3 pos)
  77. {
  78. this->deselect(pos);
  79. });
  80. }
  81. void CTerrainSelection::selectAll()
  82. {
  83. selectRange(MapRect(int3(0, 0, 0), getMap()->width, getMap()->height));
  84. selectRange(MapRect(int3(0, 0, 1), getMap()->width, getMap()->height));
  85. }
  86. void CTerrainSelection::clearSelection()
  87. {
  88. deselectRange(MapRect(int3(0, 0, 0), getMap()->width, getMap()->height));
  89. deselectRange(MapRect(int3(0, 0, 1), getMap()->width, getMap()->height));
  90. }
  91. CObjectSelection::CObjectSelection(CMap * map) : CMapSelection(map)
  92. {
  93. }
  94. CMapOperation::CMapOperation(CMap * map) : map(map)
  95. {
  96. }
  97. std::string CMapOperation::getLabel() const
  98. {
  99. return "";
  100. }
  101. CMapUndoManager::CMapUndoManager() : undoRedoLimit(10)
  102. {
  103. }
  104. void CMapUndoManager::undo()
  105. {
  106. doOperation(undoStack, redoStack, true);
  107. }
  108. void CMapUndoManager::redo()
  109. {
  110. doOperation(redoStack, undoStack, false);
  111. }
  112. void CMapUndoManager::clearAll()
  113. {
  114. undoStack.clear();
  115. redoStack.clear();
  116. }
  117. int CMapUndoManager::getUndoRedoLimit() const
  118. {
  119. return undoRedoLimit;
  120. }
  121. void CMapUndoManager::setUndoRedoLimit(int value)
  122. {
  123. assert(value >= 0);
  124. undoStack.resize(std::min(undoStack.size(), static_cast<TStack::size_type>(value)));
  125. redoStack.resize(std::min(redoStack.size(), static_cast<TStack::size_type>(value)));
  126. }
  127. const CMapOperation * CMapUndoManager::peekRedo() const
  128. {
  129. return peek(redoStack);
  130. }
  131. const CMapOperation * CMapUndoManager::peekUndo() const
  132. {
  133. return peek(undoStack);
  134. }
  135. void CMapUndoManager::addOperation(unique_ptr<CMapOperation> && operation)
  136. {
  137. undoStack.push_front(std::move(operation));
  138. if(undoStack.size() > undoRedoLimit) undoStack.pop_back();
  139. redoStack.clear();
  140. }
  141. void CMapUndoManager::doOperation(TStack & fromStack, TStack & toStack, bool doUndo)
  142. {
  143. if(fromStack.empty()) return;
  144. auto & operation = fromStack.front();
  145. if(doUndo)
  146. {
  147. operation->undo();
  148. }
  149. else
  150. {
  151. operation->redo();
  152. }
  153. toStack.push_front(std::move(operation));
  154. fromStack.pop_front();
  155. }
  156. const CMapOperation * CMapUndoManager::peek(const TStack & stack) const
  157. {
  158. if(stack.empty()) return nullptr;
  159. return stack.front().get();
  160. }
  161. CMapEditManager::CMapEditManager(CMap * map)
  162. : map(map), terrainSel(map), objectSel(map)
  163. {
  164. }
  165. CMap * CMapEditManager::getMap()
  166. {
  167. return map;
  168. }
  169. void CMapEditManager::clearTerrain(CRandomGenerator * gen/* = nullptr*/)
  170. {
  171. execute(make_unique<CClearTerrainOperation>(map, gen ? gen : &(this->gen)));
  172. }
  173. void CMapEditManager::drawTerrain(ETerrainType terType, CRandomGenerator * gen/* = nullptr*/)
  174. {
  175. execute(make_unique<CDrawTerrainOperation>(map, terrainSel, terType, gen ? gen : &(this->gen)));
  176. terrainSel.clearSelection();
  177. }
  178. void CMapEditManager::insertObject(CGObjectInstance * obj, const int3 & pos)
  179. {
  180. execute(make_unique<CInsertObjectOperation>(map, obj, pos));
  181. }
  182. void CMapEditManager::execute(unique_ptr<CMapOperation> && operation)
  183. {
  184. operation->execute();
  185. undoManager.addOperation(std::move(operation));
  186. }
  187. CTerrainSelection & CMapEditManager::getTerrainSelection()
  188. {
  189. return terrainSel;
  190. }
  191. CObjectSelection & CMapEditManager::getObjectSelection()
  192. {
  193. return objectSel;
  194. }
  195. CMapUndoManager & CMapEditManager::getUndoManager()
  196. {
  197. return undoManager;
  198. }
  199. CComposedOperation::CComposedOperation(CMap * map) : CMapOperation(map)
  200. {
  201. }
  202. void CComposedOperation::execute()
  203. {
  204. BOOST_FOREACH(auto & operation, operations)
  205. {
  206. operation->execute();
  207. }
  208. }
  209. void CComposedOperation::undo()
  210. {
  211. BOOST_FOREACH(auto & operation, operations)
  212. {
  213. operation->undo();
  214. }
  215. }
  216. void CComposedOperation::redo()
  217. {
  218. BOOST_FOREACH(auto & operation, operations)
  219. {
  220. operation->redo();
  221. }
  222. }
  223. void CComposedOperation::addOperation(unique_ptr<CMapOperation> && operation)
  224. {
  225. operations.push_back(std::move(operation));
  226. }
  227. const std::string TerrainViewPattern::FLIP_MODE_DIFF_IMAGES = "D";
  228. const std::string TerrainViewPattern::RULE_DIRT = "D";
  229. const std::string TerrainViewPattern::RULE_SAND = "S";
  230. const std::string TerrainViewPattern::RULE_TRANSITION = "T";
  231. const std::string TerrainViewPattern::RULE_NATIVE = "N";
  232. const std::string TerrainViewPattern::RULE_ANY = "?";
  233. TerrainViewPattern::TerrainViewPattern() : diffImages(false), rotationTypesCount(0), minPoints(0), terGroup(ETerrainGroup::NORMAL)
  234. {
  235. maxPoints = std::numeric_limits<int>::max();
  236. }
  237. TerrainViewPattern::WeightedRule::WeightedRule() : points(0)
  238. {
  239. }
  240. bool TerrainViewPattern::WeightedRule::isStandardRule() const
  241. {
  242. return TerrainViewPattern::RULE_ANY == name || TerrainViewPattern::RULE_DIRT == name
  243. || TerrainViewPattern::RULE_NATIVE == name || TerrainViewPattern::RULE_SAND == name
  244. || TerrainViewPattern::RULE_TRANSITION == name;
  245. }
  246. boost::mutex CTerrainViewPatternConfig::smx;
  247. CTerrainViewPatternConfig & CTerrainViewPatternConfig::get()
  248. {
  249. TLockGuard _(smx);
  250. static CTerrainViewPatternConfig instance;
  251. return instance;
  252. }
  253. CTerrainViewPatternConfig::CTerrainViewPatternConfig()
  254. {
  255. const JsonNode config(ResourceID("config/terrainViewPatterns.json"));
  256. const auto & patternsVec = config.Vector();
  257. BOOST_FOREACH(const auto & ptrnNode, patternsVec)
  258. {
  259. TerrainViewPattern pattern;
  260. // Read pattern data
  261. const JsonVector & data = ptrnNode["data"].Vector();
  262. assert(data.size() == 9);
  263. for(int i = 0; i < data.size(); ++i)
  264. {
  265. std::string cell = data[i].String();
  266. boost::algorithm::erase_all(cell, " ");
  267. std::vector<std::string> rules;
  268. boost::split(rules, cell, boost::is_any_of(","));
  269. BOOST_FOREACH(std::string ruleStr, rules)
  270. {
  271. std::vector<std::string> ruleParts;
  272. boost::split(ruleParts, ruleStr, boost::is_any_of("-"));
  273. TerrainViewPattern::WeightedRule rule;
  274. rule.name = ruleParts[0];
  275. assert(!rule.name.empty());
  276. if(ruleParts.size() > 1)
  277. {
  278. rule.points = boost::lexical_cast<int>(ruleParts[1]);
  279. }
  280. pattern.data[i].push_back(rule);
  281. }
  282. }
  283. // Read various properties
  284. pattern.id = ptrnNode["id"].String();
  285. assert(!pattern.id.empty());
  286. pattern.minPoints = static_cast<int>(ptrnNode["minPoints"].Float());
  287. pattern.maxPoints = static_cast<int>(ptrnNode["maxPoints"].Float());
  288. if(pattern.maxPoints == 0) pattern.maxPoints = std::numeric_limits<int>::max();
  289. // Read mapping
  290. const auto & mappingStruct = ptrnNode["mapping"].Struct();
  291. BOOST_FOREACH(const auto & mappingPair, mappingStruct)
  292. {
  293. TerrainViewPattern terGroupPattern = pattern;
  294. auto mappingStr = mappingPair.second.String();
  295. boost::algorithm::erase_all(mappingStr, " ");
  296. auto colonIndex = mappingStr.find_first_of(":");
  297. const auto & flipMode = mappingStr.substr(0, colonIndex);
  298. terGroupPattern.diffImages = TerrainViewPattern::FLIP_MODE_DIFF_IMAGES == &(flipMode[flipMode.length() - 1]);
  299. if(terGroupPattern.diffImages)
  300. {
  301. terGroupPattern.rotationTypesCount = boost::lexical_cast<int>(flipMode.substr(0, flipMode.length() - 1));
  302. assert(terGroupPattern.rotationTypesCount == 2 || terGroupPattern.rotationTypesCount == 4);
  303. }
  304. mappingStr = mappingStr.substr(colonIndex + 1);
  305. std::vector<std::string> mappings;
  306. boost::split(mappings, mappingStr, boost::is_any_of(","));
  307. BOOST_FOREACH(std::string mapping, mappings)
  308. {
  309. std::vector<std::string> range;
  310. boost::split(range, mapping, boost::is_any_of("-"));
  311. terGroupPattern.mapping.push_back(std::make_pair(boost::lexical_cast<int>(range[0]),
  312. boost::lexical_cast<int>(range.size() > 1 ? range[1] : range[0])));
  313. }
  314. const auto & terGroup = getTerrainGroup(mappingPair.first);
  315. terGroupPattern.terGroup = terGroup;
  316. patterns[terGroup].push_back(terGroupPattern);
  317. }
  318. }
  319. }
  320. CTerrainViewPatternConfig::~CTerrainViewPatternConfig()
  321. {
  322. }
  323. ETerrainGroup::ETerrainGroup CTerrainViewPatternConfig::getTerrainGroup(const std::string & terGroup) const
  324. {
  325. static const std::map<std::string, ETerrainGroup::ETerrainGroup> terGroups
  326. = boost::assign::map_list_of("normal", ETerrainGroup::NORMAL)("dirt", ETerrainGroup::DIRT)
  327. ("sand", ETerrainGroup::SAND)("water", ETerrainGroup::WATER)("rock", ETerrainGroup::ROCK);
  328. auto it = terGroups.find(terGroup);
  329. if(it == terGroups.end()) throw std::runtime_error(boost::str(boost::format("Terrain group '%s' does not exist.") % terGroup));
  330. return it->second;
  331. }
  332. const std::vector<TerrainViewPattern> & CTerrainViewPatternConfig::getPatternsForGroup(ETerrainGroup::ETerrainGroup terGroup) const
  333. {
  334. return patterns.find(terGroup)->second;
  335. }
  336. boost::optional<const TerrainViewPattern &> CTerrainViewPatternConfig::getPatternById(ETerrainGroup::ETerrainGroup terGroup, const std::string & id) const
  337. {
  338. const std::vector<TerrainViewPattern> & groupPatterns = getPatternsForGroup(terGroup);
  339. BOOST_FOREACH(const TerrainViewPattern & pattern, groupPatterns)
  340. {
  341. if(id == pattern.id)
  342. {
  343. return boost::optional<const TerrainViewPattern &>(pattern);
  344. }
  345. }
  346. return boost::optional<const TerrainViewPattern &>();
  347. }
  348. CDrawTerrainOperation::CDrawTerrainOperation(CMap * map, const CTerrainSelection & terrainSel, ETerrainType terType, CRandomGenerator * gen)
  349. : CMapOperation(map), terrainSel(terrainSel), terType(terType), gen(gen)
  350. {
  351. }
  352. void CDrawTerrainOperation::execute()
  353. {
  354. BOOST_FOREACH(const auto & pos, terrainSel.getSelectedItems())
  355. {
  356. auto & tile = map->getTile(pos);
  357. tile.terType = terType;
  358. invalidateTerrainViews(pos);
  359. }
  360. updateTerrainTypes();
  361. updateTerrainViews();
  362. //TODO add coastal bit to extTileFlags appropriately
  363. }
  364. void CDrawTerrainOperation::undo()
  365. {
  366. //TODO
  367. }
  368. void CDrawTerrainOperation::redo()
  369. {
  370. //TODO
  371. }
  372. std::string CDrawTerrainOperation::getLabel() const
  373. {
  374. return "Draw Terrain";
  375. }
  376. void CDrawTerrainOperation::updateTerrainTypes()
  377. {
  378. auto positions = terrainSel.getSelectedItems();
  379. while(!positions.empty())
  380. {
  381. const auto & centerPos = *(positions.begin());
  382. auto centerTile = map->getTile(centerPos);
  383. auto tiles = getInvalidTiles(centerPos);
  384. auto updateTerrainType = [&](const int3 & pos)
  385. {
  386. map->getTile(pos).terType = centerTile.terType;
  387. invalidateTerrainViews(pos);
  388. logGlobal->debugStream() << boost::format("Update terrain tile at '%s' to type '%i'.") % pos % centerTile.terType;
  389. };
  390. // Fill foreign invalid tiles
  391. BOOST_FOREACH(const auto & tile, tiles.foreignTiles)
  392. {
  393. updateTerrainType(tile);
  394. }
  395. if(tiles.nativeTiles.find(centerPos) != tiles.nativeTiles.end())
  396. {
  397. // Blow up
  398. auto rect = extendTileAroundSafely(centerPos);
  399. std::set<int3> suitableTiles;
  400. int invalidForeignTilesCnt, invalidNativeTilesCnt;
  401. invalidForeignTilesCnt = invalidNativeTilesCnt = std::numeric_limits<int>::max();
  402. rect.forEach([&](const int3 & posToTest)
  403. {
  404. auto & terrainTile = map->getTile(posToTest);
  405. if(centerTile.terType != terrainTile.terType)
  406. {
  407. auto formerTerType = terrainTile.terType;
  408. terrainTile.terType = centerTile.terType;
  409. auto testTile = getInvalidTiles(posToTest);
  410. auto addToSuitableTiles = [&](const int3 & pos)
  411. {
  412. suitableTiles.insert(pos);
  413. logGlobal->debugStream() << boost::format("Found suitable tile '%s' for main tile '%s'.") % pos % centerPos;
  414. };
  415. if(testTile.nativeTiles.size() < invalidNativeTilesCnt ||
  416. (testTile.nativeTiles.size() == invalidNativeTilesCnt && testTile.foreignTiles.size() < invalidForeignTilesCnt))
  417. {
  418. suitableTiles.clear();
  419. addToSuitableTiles(posToTest);
  420. }
  421. else if(testTile.nativeTiles.size() == invalidNativeTilesCnt &&
  422. testTile.foreignTiles.size() == invalidForeignTilesCnt)
  423. {
  424. addToSuitableTiles(posToTest);
  425. }
  426. terrainTile.terType = formerTerType;
  427. }
  428. });
  429. if(suitableTiles.size() == 1)
  430. {
  431. updateTerrainType(*suitableTiles.begin());
  432. }
  433. else
  434. {
  435. const int3 directions[] = { int3(0, -1, 0), int3(-1, 0, 0), int3(0, 1, 0), int3(1, 0, 0),
  436. int3(-1, -1, 0), int3(-1, 1, 0), int3(1, 1, 0), int3(1, -1, 0)};
  437. for(int i = 0; i < ARRAY_COUNT(directions); ++i)
  438. {
  439. auto it = suitableTiles.find(centerPos + directions[i]);
  440. if(it != suitableTiles.end())
  441. {
  442. updateTerrainType(*it);
  443. break;
  444. }
  445. }
  446. }
  447. }
  448. else
  449. {
  450. positions.erase(centerPos);
  451. }
  452. }
  453. }
  454. void CDrawTerrainOperation::updateTerrainViews()
  455. {
  456. BOOST_FOREACH(const auto & pos, invalidatedTerViews)
  457. {
  458. const auto & patterns =
  459. CTerrainViewPatternConfig::get().getPatternsForGroup(getTerrainGroup(map->getTile(pos).terType));
  460. // Detect a pattern which fits best
  461. int bestPattern = -1;
  462. ValidationResult valRslt(false);
  463. for(int k = 0; k < patterns.size(); ++k)
  464. {
  465. const auto & pattern = patterns[k];
  466. valRslt = validateTerrainView(pos, pattern);
  467. if(valRslt.result)
  468. {
  469. /*logGlobal->debugStream() << boost::format("Pattern detected at pos '%s': Pattern '%s', Flip '%i', Repl. '%s'.") %
  470. pos % pattern.id % valRslt.flip % valRslt.transitionReplacement;*/
  471. bestPattern = k;
  472. break;
  473. }
  474. }
  475. assert(bestPattern != -1);
  476. if(bestPattern == -1)
  477. {
  478. // This shouldn't be the case
  479. logGlobal->warnStream() << boost::format("No pattern detected at pos '%s'.") % pos;
  480. continue;
  481. }
  482. // Get mapping
  483. const TerrainViewPattern & pattern = patterns[bestPattern];
  484. std::pair<int, int> mapping;
  485. if(valRslt.transitionReplacement.empty())
  486. {
  487. mapping = pattern.mapping[0];
  488. }
  489. else
  490. {
  491. mapping = valRslt.transitionReplacement == TerrainViewPattern::RULE_DIRT ? pattern.mapping[0] : pattern.mapping[1];
  492. }
  493. // Set terrain view
  494. auto & tile = map->getTile(pos);
  495. if(!pattern.diffImages)
  496. {
  497. tile.terView = gen->getInteger(mapping.first, mapping.second);
  498. tile.extTileFlags = valRslt.flip;
  499. }
  500. else
  501. {
  502. const int framesPerRot = (mapping.second - mapping.first + 1) / pattern.rotationTypesCount;
  503. int flip = (pattern.rotationTypesCount == 2 && valRslt.flip == 2) ? 1 : valRslt.flip;
  504. int firstFrame = mapping.first + flip * framesPerRot;
  505. tile.terView = gen->getInteger(firstFrame, firstFrame + framesPerRot - 1);
  506. tile.extTileFlags = 0;
  507. }
  508. }
  509. }
  510. ETerrainGroup::ETerrainGroup CDrawTerrainOperation::getTerrainGroup(ETerrainType terType) const
  511. {
  512. switch(terType)
  513. {
  514. case ETerrainType::DIRT:
  515. return ETerrainGroup::DIRT;
  516. case ETerrainType::SAND:
  517. return ETerrainGroup::SAND;
  518. case ETerrainType::WATER:
  519. return ETerrainGroup::WATER;
  520. case ETerrainType::ROCK:
  521. return ETerrainGroup::ROCK;
  522. default:
  523. return ETerrainGroup::NORMAL;
  524. }
  525. }
  526. CDrawTerrainOperation::ValidationResult CDrawTerrainOperation::validateTerrainView(const int3 & pos, const TerrainViewPattern & pattern, int recDepth /*= 0*/) const
  527. {
  528. for(int flip = 0; flip < 4; ++flip)
  529. {
  530. auto valRslt = validateTerrainViewInner(pos, flip > 0 ? getFlippedPattern(pattern, flip) : pattern, recDepth);
  531. if(valRslt.result)
  532. {
  533. valRslt.flip = flip;
  534. return valRslt;
  535. }
  536. }
  537. return ValidationResult(false);
  538. }
  539. CDrawTerrainOperation::ValidationResult CDrawTerrainOperation::validateTerrainViewInner(const int3 & pos, const TerrainViewPattern & pattern, int recDepth /*= 0*/) const
  540. {
  541. ETerrainType centerTerType = map->getTile(pos).terType;
  542. int totalPoints = 0;
  543. std::string transitionReplacement;
  544. for(int i = 0; i < 9; ++i)
  545. {
  546. // The center, middle cell can be skipped
  547. if(i == 4)
  548. {
  549. continue;
  550. }
  551. // Get terrain group of the current cell
  552. int cx = pos.x + (i % 3) - 1;
  553. int cy = pos.y + (i / 3) - 1;
  554. int3 currentPos(cx, cy, pos.z);
  555. bool isAlien = false;
  556. ETerrainType terType;
  557. if(!map->isInTheMap(currentPos))
  558. {
  559. terType = centerTerType;
  560. }
  561. else
  562. {
  563. terType = map->getTile(currentPos).terType;
  564. if(terType != centerTerType)
  565. {
  566. isAlien = true;
  567. }
  568. }
  569. // Validate all rules per cell
  570. int topPoints = -1;
  571. for(int j = 0; j < pattern.data[i].size(); ++j)
  572. {
  573. TerrainViewPattern::WeightedRule rule = pattern.data[i][j];
  574. if(!rule.isStandardRule())
  575. {
  576. if(recDepth == 0 && map->isInTheMap(currentPos))
  577. {
  578. const auto & patternForRule = CTerrainViewPatternConfig::get().getPatternById(getTerrainGroup(terType), rule.name);
  579. if(patternForRule)
  580. {
  581. auto rslt = validateTerrainView(currentPos, *patternForRule, 1);
  582. if(rslt.result) topPoints = std::max(topPoints, rule.points);
  583. }
  584. continue;
  585. }
  586. else
  587. {
  588. rule.name = TerrainViewPattern::RULE_NATIVE;
  589. }
  590. }
  591. auto applyValidationRslt = [&](bool rslt)
  592. {
  593. if(rslt)
  594. {
  595. topPoints = std::max(topPoints, rule.points);
  596. }
  597. };
  598. // Validate cell with the ruleset of the pattern
  599. if(pattern.terGroup == ETerrainGroup::NORMAL)
  600. {
  601. bool dirtTestOk = (rule.name == TerrainViewPattern::RULE_DIRT || rule.name == TerrainViewPattern::RULE_TRANSITION)
  602. && isAlien && !isSandType(terType);
  603. bool sandTestOk = (rule.name == TerrainViewPattern::RULE_SAND || rule.name == TerrainViewPattern::RULE_TRANSITION)
  604. && isSandType(terType);
  605. if(transitionReplacement.empty() && rule.name == TerrainViewPattern::RULE_TRANSITION
  606. && (dirtTestOk || sandTestOk))
  607. {
  608. transitionReplacement = dirtTestOk ? TerrainViewPattern::RULE_DIRT : TerrainViewPattern::RULE_SAND;
  609. }
  610. if(rule.name == TerrainViewPattern::RULE_TRANSITION)
  611. {
  612. applyValidationRslt((dirtTestOk && transitionReplacement != TerrainViewPattern::RULE_SAND) ||
  613. (sandTestOk && transitionReplacement != TerrainViewPattern::RULE_DIRT));
  614. }
  615. else
  616. {
  617. bool nativeTestOk = rule.name == TerrainViewPattern::RULE_NATIVE && !isAlien;
  618. applyValidationRslt(rule.name == TerrainViewPattern::RULE_ANY || dirtTestOk || sandTestOk || nativeTestOk);
  619. }
  620. }
  621. else if(pattern.terGroup == ETerrainGroup::DIRT)
  622. {
  623. bool nativeTestOk = rule.name == TerrainViewPattern::RULE_NATIVE && !isSandType(terType);
  624. bool sandTestOk = (rule.name == TerrainViewPattern::RULE_SAND || rule.name == TerrainViewPattern::RULE_TRANSITION)
  625. && isSandType(terType);
  626. applyValidationRslt(rule.name == TerrainViewPattern::RULE_ANY || sandTestOk || nativeTestOk);
  627. }
  628. else if(pattern.terGroup == ETerrainGroup::SAND)
  629. {
  630. applyValidationRslt(true);
  631. }
  632. else if(pattern.terGroup == ETerrainGroup::WATER || pattern.terGroup == ETerrainGroup::ROCK)
  633. {
  634. bool nativeTestOk = rule.name == TerrainViewPattern::RULE_NATIVE && !isAlien;
  635. bool sandTestOk = (rule.name == TerrainViewPattern::RULE_SAND || rule.name == TerrainViewPattern::RULE_TRANSITION)
  636. && isAlien;
  637. applyValidationRslt(rule.name == TerrainViewPattern::RULE_ANY || sandTestOk || nativeTestOk);
  638. }
  639. }
  640. if(topPoints == -1)
  641. {
  642. return ValidationResult(false);
  643. }
  644. else
  645. {
  646. totalPoints += topPoints;
  647. }
  648. }
  649. if(totalPoints >= pattern.minPoints && totalPoints <= pattern.maxPoints)
  650. {
  651. return ValidationResult(true, transitionReplacement);
  652. }
  653. else
  654. {
  655. return ValidationResult(false);
  656. }
  657. }
  658. bool CDrawTerrainOperation::isSandType(ETerrainType terType) const
  659. {
  660. switch(terType)
  661. {
  662. case ETerrainType::WATER:
  663. case ETerrainType::SAND:
  664. case ETerrainType::ROCK:
  665. return true;
  666. default:
  667. return false;
  668. }
  669. }
  670. TerrainViewPattern CDrawTerrainOperation::getFlippedPattern(const TerrainViewPattern & pattern, int flip) const
  671. {
  672. if(flip == 0)
  673. {
  674. return pattern;
  675. }
  676. TerrainViewPattern ret = pattern;
  677. if(flip == FLIP_PATTERN_HORIZONTAL || flip == FLIP_PATTERN_BOTH)
  678. {
  679. for(int i = 0; i < 3; ++i)
  680. {
  681. int y = i * 3;
  682. std::swap(ret.data[y], ret.data[y + 2]);
  683. }
  684. }
  685. if(flip == FLIP_PATTERN_VERTICAL || flip == FLIP_PATTERN_BOTH)
  686. {
  687. for(int i = 0; i < 3; ++i)
  688. {
  689. std::swap(ret.data[i], ret.data[6 + i]);
  690. }
  691. }
  692. return ret;
  693. }
  694. void CDrawTerrainOperation::invalidateTerrainViews(const int3 & centerPos)
  695. {
  696. auto rect = extendTileAroundSafely(centerPos);
  697. rect.forEach([&](const int3 & pos)
  698. {
  699. invalidatedTerViews.insert(pos);
  700. });
  701. }
  702. CDrawTerrainOperation::InvalidTiles CDrawTerrainOperation::getInvalidTiles(const int3 & centerPos) const
  703. {
  704. InvalidTiles tiles;
  705. auto centerTerType = map->getTile(centerPos).terType;
  706. auto rect = extendTileAround(centerPos);
  707. rect.forEach([&](const int3 & pos)
  708. {
  709. if(map->isInTheMap(pos))
  710. {
  711. auto terType = map->getTile(pos).terType;
  712. // Pattern 2x2
  713. const int3 translations[4] = { int3(-1, -1, 0), int3(0, -1, 0), int3(-1, 0, 0), int3(0, 0, 0) };
  714. bool valid = true;
  715. for(int i = 0; i < ARRAY_COUNT(translations); ++i)
  716. {
  717. valid = true;
  718. MapRect square(int3(pos.x + translations[i].x, pos.y + translations[i].y, pos.z), 2, 2);
  719. square.forEach([&](const int3 & pos)
  720. {
  721. if(map->isInTheMap(pos) && map->getTile(pos).terType != terType) valid = false;
  722. });
  723. if(valid) break;
  724. }
  725. if(!valid)
  726. {
  727. if(terType == centerTerType) tiles.nativeTiles.insert(pos);
  728. else tiles.foreignTiles.insert(pos);
  729. }
  730. }
  731. });
  732. return tiles;
  733. }
  734. MapRect CDrawTerrainOperation::extendTileAround(const int3 & centerPos) const
  735. {
  736. return MapRect(int3(centerPos.x - 1, centerPos.y - 1, centerPos.z), 3, 3);
  737. }
  738. MapRect CDrawTerrainOperation::extendTileAroundSafely(const int3 & centerPos) const
  739. {
  740. return extendTileAround(centerPos) & MapRect(int3(0, 0, centerPos.z), map->width, map->height);
  741. }
  742. CDrawTerrainOperation::ValidationResult::ValidationResult(bool result, const std::string & transitionReplacement /*= ""*/)
  743. : result(result), transitionReplacement(transitionReplacement)
  744. {
  745. }
  746. CClearTerrainOperation::CClearTerrainOperation(CMap * map, CRandomGenerator * gen) : CComposedOperation(map)
  747. {
  748. CTerrainSelection terrainSel(map);
  749. terrainSel.selectRange(MapRect(int3(0, 0, 0), map->width, map->height));
  750. addOperation(make_unique<CDrawTerrainOperation>(map, terrainSel, ETerrainType::WATER, gen));
  751. terrainSel.clearSelection();
  752. terrainSel.selectRange(MapRect(int3(0, 0, 1), map->width, map->height));
  753. addOperation(make_unique<CDrawTerrainOperation>(map, terrainSel, ETerrainType::ROCK, gen));
  754. }
  755. std::string CClearTerrainOperation::getLabel() const
  756. {
  757. return "Clear Terrain";
  758. }
  759. CInsertObjectOperation::CInsertObjectOperation(CMap * map, CGObjectInstance * obj, const int3 & pos)
  760. : CMapOperation(map), pos(pos), obj(obj)
  761. {
  762. }
  763. void CInsertObjectOperation::execute()
  764. {
  765. obj->pos = pos;
  766. obj->id = ObjectInstanceID(map->objects.size());
  767. map->objects.push_back(obj);
  768. if(obj->ID == Obj::TOWN)
  769. {
  770. map->towns.push_back(static_cast<CGTownInstance *>(obj));
  771. }
  772. if(obj->ID == Obj::HERO)
  773. {
  774. map->heroes.push_back(static_cast<CGHeroInstance*>(obj));
  775. }
  776. map->addBlockVisTiles(obj);
  777. }
  778. void CInsertObjectOperation::undo()
  779. {
  780. //TODO
  781. }
  782. void CInsertObjectOperation::redo()
  783. {
  784. execute();
  785. }
  786. std::string CInsertObjectOperation::getLabel() const
  787. {
  788. return "Insert Object";
  789. }