64 lines
1.7 KiB
C++
64 lines
1.7 KiB
C++
|
#include <stdint.h>
|
||
|
|
#include <iostream>
|
||
|
|
#include <map>
|
||
|
|
#include <vector>
|
||
|
|
#include "Core.h"
|
||
|
|
#include "Console.h"
|
||
|
|
#include "Export.h"
|
||
|
|
#include "PluginManager.h"
|
||
|
|
#include "modules/Maps.h"
|
||
|
|
#include "modules/World.h"
|
||
|
|
#include "modules/MapCache.h"
|
||
|
|
#include "modules/Gui.h"
|
||
|
|
using namespace DFHack;
|
||
|
|
using namespace df::enums;
|
||
|
|
using df::global::world;
|
||
|
|
|
||
|
|
DFHACK_PLUGIN("lair");
|
||
|
|
|
||
|
|
enum state
|
||
|
|
{
|
||
|
|
LAIR_RESET,
|
||
|
|
LAIR_SET,
|
||
|
|
};
|
||
|
|
|
||
|
|
command_result lair(color_ostream &out, std::vector<std::string> & params)
|
||
|
|
{
|
||
|
|
state do_what = LAIR_SET;
|
||
|
|
for(auto iter = params.begin(); iter != params.end(); iter++)
|
||
|
|
{
|
||
|
|
if(*iter == "reset")
|
||
|
|
do_what = LAIR_RESET;
|
||
|
|
}
|
||
|
|
CoreSuspender lock;
|
||
|
|
if (!Maps::IsValid())
|
||
|
|
{
|
||
|
|
out.printerr("Map is not available!\n");
|
||
|
|
return CR_FAILURE;
|
||
|
|
}
|
||
|
|
uint32_t x_max,y_max,z_max;
|
||
|
|
Maps::getSize(x_max,y_max,z_max);
|
||
|
|
for (size_t i = 0; i < world->map.map_blocks.size(); i++)
|
||
|
|
{
|
||
|
|
df::map_block *block = world->map.map_blocks[i];
|
||
|
|
DFHack::occupancies40d & 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, std::vector <PluginCommand> &commands)
|
||
|
|
{
|
||
|
|
commands.push_back(PluginCommand("lair","Mark the map as a monster lair, preventing item scatter ('lair reset' reverts that).",lair));
|
||
|
|
return CR_OK;
|
||
|
|
}
|