#include "PluginManager.h"

#include "modules/Maps.h"

#include "df/global_objects.h"

#include "df/map_block.h"

#include "df/world.h"

using namespace DFHack;

using std::string;

using std::vector;

DFHACK_PLUGIN("lair");

REQUIRE_GLOBAL(world);

enum state

{

LAIR_RESET,

LAIR_SET,

};

command_result lair(color_ostream &out, vector<string> & params) {

state do_what = LAIR_SET;

for (auto & param : params) {

if (param == "reset")

do_what = LAIR_RESET;

}

if (!Maps::IsValid()) {

out.printerr("Map is not available!\n");

return CR_FAILURE;

}

for (auto block : world->map.map_blocks) {

auto & occupancies = block->occupancy;

// for each tile in block

for (uint32_t x = 0; x < 16; x++) for (uint32_t y = 0; y < 16; y++) {

// set to revealed

occupancies[x][y].bits.monster_lair = (do_what == LAIR_SET);

}

}

if (do_what == LAIR_SET)

out.print("Map marked as lair.\n");

else

out.print("Map no longer marked as lair.\n");

return CR_OK;

}

DFhackCExport command_result plugin_init(color_ostream &out, vector<PluginCommand> &commands) {

commands.push_back(PluginCommand(

"lair",

"Prevent item scatter when a site is reclaimed or visited.",

lair));

return CR_OK;

}