/* * TreasurePlacer.h, part of VCMI engine * * Authors: listed in file AUTHORS in main folder * * License: GNU General Public License v2.0 or later * Full text of license available in license.txt file, in main folder * */ #pragma once #include "../Zone.h" #include "../../mapObjects/ObjectTemplate.h" VCMI_LIB_NAMESPACE_BEGIN class CGObjectInstance; class ObjectManager; class RmgMap; class CMapGenerator; class CRandomGenerator; struct ObjectInfo { std::vector> templates; ui32 value = 0; ui16 probability = 0; ui32 maxPerZone = 1; //ui32 maxPerMap; //unused std::function generateObject; void setTemplates(si32 type, si32 subtype, TerrainId terrain); }; class TreasurePlacer: public Modificator { public: MODIFICATOR(TreasurePlacer); void process() override; void init() override; char dump(const int3 &) override; void createTreasures(ObjectManager & manager); void addObjectToRandomPool(const ObjectInfo& oi); void addAllPossibleObjects(); //add objects, including zone-specific, to possibleObjects size_t getPossibleObjectsSize() const; void setMaxPrisons(size_t count); size_t getMaxPrisons() const; protected: bool isGuardNeededForTreasure(int value); ObjectInfo * getRandomObject(ui32 desiredValue, ui32 currentValue, bool allowLargeObjects); std::vector prepareTreasurePile(const CTreasureInfo & treasureInfo); rmg::Object constructTreasurePile(const std::vector & treasureInfos, bool densePlacement = false); protected: std::vector possibleObjects; int minGuardedValue = 0; rmg::Area treasureArea; rmg::Area treasureBlockArea; rmg::Area guards; size_t maxPrisons; }; VCMI_LIB_NAMESPACE_END