CDrawRoadsOperation.h 1.4 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758
  1. /*
  2. * CDrawRoadsOperation.h, 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. #pragma once
  11. #include "../CRandomGenerator.h"
  12. #include "CMapEditManager.h"
  13. struct TerrainTile;
  14. class CDrawRoadsOperation : public CMapOperation
  15. {
  16. public:
  17. CDrawRoadsOperation(CMap * map, const CTerrainSelection & terrainSel, ERoadType::ERoadType roadType, CRandomGenerator * gen);
  18. void execute() override;
  19. void undo() override;
  20. void redo() override;
  21. std::string getLabel() const override;
  22. private:
  23. struct RoadPattern
  24. {
  25. std::string data[9];
  26. std::pair<int, int> roadMapping, riverMapping;
  27. bool hasHFlip, hasVFlip;
  28. };
  29. struct ValidationResult
  30. {
  31. ValidationResult(bool result): result(result), flip(0){};
  32. bool result;
  33. int flip;
  34. };
  35. static const std::vector<RoadPattern> patterns;
  36. void flipPattern(RoadPattern & pattern, int flip) const;
  37. void updateTiles(std::set<int3> & invalidated);
  38. ValidationResult validateTile(const RoadPattern & pattern, const int3 & pos);
  39. void updateTile(TerrainTile & tile, const RoadPattern & pattern, const int flip);
  40. bool canApplyPattern(const RoadPattern & pattern) const;
  41. bool needUpdateTile(const TerrainTile & tile) const;
  42. bool tileHasSomething(const int3 & pos) const;
  43. CTerrainSelection terrainSel;
  44. ERoadType::ERoadType roadType;
  45. CRandomGenerator * gen;
  46. };