dfhack/plugins/tubefill.cpp

121 lines
3.9 KiB
C++

// Adamantine tube filler. Replaces mined out tiles but leaves hollow tubes alone (to prevent problems)
2011-03-23 21:06:14 -06:00
2011-08-02 08:25:10 -06:00
#include <stdint.h>
2011-03-23 21:06:14 -06:00
#include <iostream>
#include <map>
2011-08-02 08:25:10 -06:00
#include <vector>
2011-12-31 04:48:42 -07:00
#include "Core.h"
#include "Console.h"
#include "Export.h"
#include "PluginManager.h"
#include "modules/Maps.h"
#include "modules/World.h"
#include "modules/Gui.h"
#include "TileTypes.h"
2014-06-27 12:01:46 -06:00
#include "df/deep_vein_hollow.h"
2011-08-02 08:25:10 -06:00
using namespace DFHack;
using namespace df::enums;
using df::global::world;
2011-08-02 08:25:10 -06:00
bool isDesignatedHollow(df::coord pos)
{
for (size_t i = 0; i < world->deep_vein_hollows.size(); i++)
{
auto *vein = world->deep_vein_hollows[i];
for (size_t j = 0; j < vein->tiles.x.size(); j++)
if (pos == df::coord(vein->tiles.x[j], vein->tiles.y[j], vein->tiles.z[j]))
return true;
}
return false;
}
command_result tubefill(color_ostream &out, std::vector<std::string> & params);
2011-08-02 08:25:10 -06:00
2012-02-21 10:19:17 -07:00
DFHACK_PLUGIN("tubefill");
2011-08-02 08:25:10 -06:00
DFhackCExport command_result plugin_init ( color_ostream &out, std::vector <PluginCommand> &commands)
2011-08-02 08:25:10 -06:00
{
commands.push_back(PluginCommand("tubefill","Fill in all the adamantine tubes again.",tubefill, false,
"Replenishes mined out adamantine but does not fill hollow adamantine tubes.\n"
"Specify 'hollow' to fill hollow tubes, but beware glitchy HFS spawns.\n"));
2011-08-02 08:25:10 -06:00
return CR_OK;
}
2011-03-23 21:06:14 -06:00
DFhackCExport command_result plugin_shutdown ( color_ostream &out )
2011-08-02 08:25:10 -06:00
{
return CR_OK;
}
2011-03-23 21:06:14 -06:00
command_result tubefill(color_ostream &out, std::vector<std::string> & params)
2011-03-23 21:06:14 -06:00
{
uint64_t count = 0;
bool hollow = false;
2011-03-23 21:06:14 -06:00
for(size_t i = 0; i < params.size();i++)
{
if(params[i] == "help" || params[i] == "?")
return CR_WRONG_USAGE;
if (params[i] == "hollow")
hollow = true;
}
CoreSuspender suspend;
2012-01-19 20:44:17 -07:00
if (!Maps::IsValid())
2011-03-23 21:06:14 -06:00
{
out.printerr("Map is not available!\n");
2011-08-02 08:25:10 -06:00
return CR_FAILURE;
2011-03-23 21:06:14 -06:00
}
// walk the map
for (size_t i = 0; i < world->map.map_blocks.size(); i++)
2011-03-23 21:06:14 -06:00
{
df::map_block *block = world->map.map_blocks[i];
df::map_block *above = Maps::getTileBlock(block->map_pos + df::coord(0,0,1));
DFHack::t_feature feature;
if (!Maps::ReadFeatures(block, &feature, NULL))
continue;
if (feature.type != feature_type::deep_special_tube)
continue;
for (uint32_t x = 0; x < 16; x++)
2011-03-23 21:06:14 -06:00
{
for (uint32_t y = 0; y < 16; y++)
2011-03-23 21:06:14 -06:00
{
if (!block->designation[x][y].bits.feature_local)
continue;
if (!hollow && isDesignatedHollow(block->map_pos + df::coord(x,y,0)))
continue;
// Is the tile already a wall?
if (tileShape(block->tiletype[x][y]) == tiletype_shape::WALL)
continue;
2011-03-23 21:06:14 -06:00
// Does the tile contain liquid?
if (block->designation[x][y].bits.flow_size)
continue;
// Check the tile above this one, in case we need to add a floor
if (above)
{
if ((tileShape(above->tiletype[x][y]) == tiletype_shape::EMPTY) || (tileShape(above->tiletype[x][y]) == tiletype_shape::RAMP_TOP))
{
// if this tile isn't a local feature, it's likely the tile underneath was originally just a floor
// it's also possible there was just solid non-feature stone above, but we don't care enough to check
if (!above->designation[x][y].bits.feature_local)
continue;
above->tiletype[x][y] = findRandomVariant(tiletype::FeatureFloor1);
}
2011-03-23 21:06:14 -06:00
}
block->tiletype[x][y] = tiletype::FeatureWall;
world->reindex_pathfinding = true;
++count;
2011-03-23 21:06:14 -06:00
}
}
}
out.print("Found and changed %d tiles.\n", count);
2011-08-02 08:25:10 -06:00
return CR_OK;
2011-03-23 21:06:14 -06:00
}