mirror of
https://github.com/vcmi/vcmi.git
synced 2024-11-26 08:41:13 +02:00
73 lines
1.8 KiB
C++
73 lines
1.8 KiB
C++
/*
|
|
* 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::shared_ptr<const ObjectTemplate> templ;
|
|
ui32 value = 0;
|
|
ui16 probability = 0;
|
|
ui32 maxPerZone = 1;
|
|
//ui32 maxPerMap; //unused
|
|
std::function<CGObjectInstance *()> generateObject;
|
|
|
|
void setTemplate(si32 type, si32 subtype, TerrainId terrain);
|
|
|
|
bool operator==(const ObjectInfo& oi) const { return (templ == oi.templ); }
|
|
};
|
|
|
|
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<ObjectInfo*> prepareTreasurePile(const CTreasureInfo & treasureInfo);
|
|
rmg::Object constructTreasurePile(const std::vector<ObjectInfo*> & treasureInfos, bool densePlacement = false);
|
|
|
|
protected:
|
|
std::vector<ObjectInfo> possibleObjects;
|
|
int minGuardedValue = 0;
|
|
|
|
rmg::Area treasureArea;
|
|
rmg::Area treasureBlockArea;
|
|
rmg::Area guards;
|
|
|
|
size_t maxPrisons;
|
|
};
|
|
|
|
VCMI_LIB_NAMESPACE_END
|