Ported autodump tool
Petr Mrázek
parent
ff4d545ae7
commit
a8543f5ef0
@ -0,0 +1,215 @@
|
|||||||
|
// Quick Dumper : Moves items marked as "dump" to cursor
|
||||||
|
// FIXME: local item cache in map blocks needs to be fixed after teleporting items
|
||||||
|
#include <iostream>
|
||||||
|
#include <iomanip>
|
||||||
|
#include <sstream>
|
||||||
|
#include <climits>
|
||||||
|
#include <vector>
|
||||||
|
#include <set>
|
||||||
|
using namespace std;
|
||||||
|
|
||||||
|
#include <dfhack/Core.h>
|
||||||
|
#include <dfhack/Console.h>
|
||||||
|
#include <dfhack/Export.h>
|
||||||
|
#include <dfhack/PluginManager.h>
|
||||||
|
#include <vector>
|
||||||
|
#include <string>
|
||||||
|
#include <dfhack/modules/Maps.h>
|
||||||
|
#include <dfhack/modules/Gui.h>
|
||||||
|
#include <dfhack/modules/Items.h>
|
||||||
|
#include <dfhack/modules/Materials.h>
|
||||||
|
#include <dfhack/extra/MapExtras.h>
|
||||||
|
|
||||||
|
using namespace DFHack;
|
||||||
|
using MapExtras::Block;
|
||||||
|
using MapExtras::MapCache;
|
||||||
|
|
||||||
|
DFhackCExport command_result df_autodump (Core * c, vector <string> & parameters);
|
||||||
|
|
||||||
|
DFhackCExport const char * plugin_name ( void )
|
||||||
|
{
|
||||||
|
return "autodump";
|
||||||
|
}
|
||||||
|
|
||||||
|
DFhackCExport command_result plugin_init ( Core * c, std::vector <PluginCommand> &commands)
|
||||||
|
{
|
||||||
|
commands.clear();
|
||||||
|
commands.push_back(PluginCommand("autodump",
|
||||||
|
"Teleport items marked for dumping to the cursor.",
|
||||||
|
df_autodump));
|
||||||
|
return CR_OK;
|
||||||
|
}
|
||||||
|
|
||||||
|
DFhackCExport command_result plugin_shutdown ( Core * c )
|
||||||
|
{
|
||||||
|
return CR_OK;
|
||||||
|
}
|
||||||
|
|
||||||
|
typedef std::map <DFCoord, uint32_t> coordmap;
|
||||||
|
|
||||||
|
DFhackCExport command_result df_autodump (Core * c, vector <string> & parameters)
|
||||||
|
{
|
||||||
|
// Command line options
|
||||||
|
bool destroy = false;
|
||||||
|
if(parameters.size() > 0)
|
||||||
|
{
|
||||||
|
string & p = parameters[0];
|
||||||
|
if(p == "destroy")
|
||||||
|
destroy = true;
|
||||||
|
else if(p == "?" || p == "help")
|
||||||
|
{
|
||||||
|
c->con.print(
|
||||||
|
"This utility lets you quickly move all items designated to be dumped.\n"
|
||||||
|
"Items are instantly moved to the cursor position, the dump flag is unset,\n"
|
||||||
|
"and the forbid flag is set, as if it had been dumped normally.\n"
|
||||||
|
"Be aware that any active dump item tasks still point at the item.\n\n"
|
||||||
|
);
|
||||||
|
return CR_OK;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
c->Suspend();
|
||||||
|
DFHack::occupancies40d * occupancies;
|
||||||
|
DFHack::VersionInfo *mem = c->vinfo;
|
||||||
|
DFHack::Gui * Gui = c->getGui();
|
||||||
|
DFHack::Items * Items = c->getItems();
|
||||||
|
DFHack::Maps *Maps = c->getMaps();
|
||||||
|
|
||||||
|
vector <t_item*> p_items;
|
||||||
|
if(!Items->readItemVector(p_items))
|
||||||
|
{
|
||||||
|
c->con.printerr("Can't access the item vector.\n");
|
||||||
|
c->Resume();
|
||||||
|
return CR_FAILURE;
|
||||||
|
}
|
||||||
|
std::size_t numItems = p_items.size();
|
||||||
|
|
||||||
|
// init the map
|
||||||
|
if(!Maps->Start())
|
||||||
|
{
|
||||||
|
c->con.printerr("Can't initialize map.\n");
|
||||||
|
c->Resume();
|
||||||
|
return CR_FAILURE;
|
||||||
|
}
|
||||||
|
MapCache MC (Maps);
|
||||||
|
int i = 0;
|
||||||
|
int dumped_total = 0;
|
||||||
|
|
||||||
|
int cx, cy, cz;
|
||||||
|
DFCoord pos_cursor;
|
||||||
|
if(!destroy)
|
||||||
|
{
|
||||||
|
if (!Gui->getCursorCoords(cx,cy,cz))
|
||||||
|
{
|
||||||
|
c->con.printerr("Cursor position not found. Please enabled the cursor.\n");
|
||||||
|
c->Resume();
|
||||||
|
return CR_FAILURE;
|
||||||
|
}
|
||||||
|
pos_cursor = DFCoord(cx,cy,cz);
|
||||||
|
{
|
||||||
|
Block * b = MC.BlockAt(pos_cursor / 16);
|
||||||
|
if(!b)
|
||||||
|
{
|
||||||
|
c->con.printerr("Cursor is in an invalid/uninitialized area. Place it over a floor.\n");
|
||||||
|
c->Resume();
|
||||||
|
return CR_FAILURE;
|
||||||
|
}
|
||||||
|
uint16_t ttype = MC.tiletypeAt(pos_cursor);
|
||||||
|
if(!DFHack::isFloorTerrain(ttype))
|
||||||
|
{
|
||||||
|
c->con.printerr("Cursor should be placed over a floor.\n");
|
||||||
|
c->Resume();
|
||||||
|
return CR_FAILURE;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
coordmap counts;
|
||||||
|
// proceed with the dumpification operation
|
||||||
|
for(std::size_t i=0; i< numItems; i++)
|
||||||
|
{
|
||||||
|
t_item * itm = p_items[i];
|
||||||
|
DFCoord pos_item(itm->x, itm->y, itm->z);
|
||||||
|
|
||||||
|
// keep track how many items are at places. all items.
|
||||||
|
coordmap::iterator it = counts.find(pos_item);
|
||||||
|
if(it == counts.end())
|
||||||
|
{
|
||||||
|
std::pair< coordmap::iterator, bool > inserted = counts.insert(std::make_pair(pos_item,1));
|
||||||
|
it = inserted.first;
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
it->second ++;
|
||||||
|
}
|
||||||
|
// iterator is valid here, we use it later to decrement the pile counter if the item is moved
|
||||||
|
|
||||||
|
// only dump the stuff marked for dumping and laying on the ground
|
||||||
|
if ( !itm->flags.dump
|
||||||
|
|| !itm->flags.on_ground
|
||||||
|
|| itm->flags.construction
|
||||||
|
|| itm->flags.hidden
|
||||||
|
|| itm->flags.in_building
|
||||||
|
|| itm->flags.in_chest
|
||||||
|
|| itm->flags.in_inventory
|
||||||
|
|| itm->flags.construction
|
||||||
|
)
|
||||||
|
continue;
|
||||||
|
|
||||||
|
if(!destroy) // move to cursor
|
||||||
|
{
|
||||||
|
// Change flags to indicate the dump was completed, as if by super-dwarfs
|
||||||
|
itm->flags.dump = false;
|
||||||
|
itm->flags.forbid = true;
|
||||||
|
|
||||||
|
// Don't move items if they're already at the cursor
|
||||||
|
if (pos_cursor == pos_item)
|
||||||
|
continue;
|
||||||
|
|
||||||
|
// Move the item
|
||||||
|
itm->x = pos_cursor.x;
|
||||||
|
itm->y = pos_cursor.y;
|
||||||
|
itm->z = pos_cursor.z;
|
||||||
|
}
|
||||||
|
else // destroy
|
||||||
|
{
|
||||||
|
itm->flags.garbage_colect = true;
|
||||||
|
}
|
||||||
|
// keeping track of item pile sizes ;)
|
||||||
|
it->second --;
|
||||||
|
dumped_total++;
|
||||||
|
}
|
||||||
|
if(!destroy) // TODO: do we have to do any of this when destroying items?
|
||||||
|
{
|
||||||
|
// for each item pile, see if it reached zero. if so, unset item flag on the tile it's on
|
||||||
|
coordmap::iterator it = counts.begin();
|
||||||
|
coordmap::iterator end = counts.end();
|
||||||
|
while(it != end)
|
||||||
|
{
|
||||||
|
if(it->second == 0)
|
||||||
|
{
|
||||||
|
t_occupancy occ = MC.occupancyAt(it->first);
|
||||||
|
occ.bits.item = false;
|
||||||
|
MC.setOccupancyAt(it->first, occ);
|
||||||
|
}
|
||||||
|
it++;
|
||||||
|
}
|
||||||
|
// Set "item here" flag on target tile, if we moved any items to the target tile.
|
||||||
|
if (dumped_total > 0)
|
||||||
|
{
|
||||||
|
// assume there is a possibility the cursor points at some weird location with missing block data
|
||||||
|
Block * b = MC.BlockAt(pos_cursor / 16);
|
||||||
|
if(b)
|
||||||
|
{
|
||||||
|
t_occupancy occ = MC.occupancyAt(pos_cursor);
|
||||||
|
occ.bits.item = 1;
|
||||||
|
MC.setOccupancyAt(pos_cursor,occ);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
// write map changes back to DF.
|
||||||
|
MC.WriteAll();
|
||||||
|
// Is this necessary? Is "forbid" a dirtyable attribute like "dig" is?
|
||||||
|
Maps->WriteDirtyBit(cx/16, cy/16, cz, true);
|
||||||
|
}
|
||||||
|
c->Resume();
|
||||||
|
c->con.print("Done. %d items %s.\n", dumped_total, destroy ? "marked for desctruction" : "quickdumped");
|
||||||
|
return CR_OK;
|
||||||
|
}
|
||||||
@ -1,215 +0,0 @@
|
|||||||
// Quick Dumper : Moves items marked as "dump" to cursor
|
|
||||||
|
|
||||||
#include <iostream>
|
|
||||||
#include <iomanip>
|
|
||||||
#include <sstream>
|
|
||||||
#include <climits>
|
|
||||||
#include <vector>
|
|
||||||
#include <set>
|
|
||||||
using namespace std;
|
|
||||||
|
|
||||||
#include <DFHack.h>
|
|
||||||
#include <dfhack/DFVector.h>
|
|
||||||
#include <dfhack/extra/MapExtras.h> // map cache for the win.
|
|
||||||
#include <dfhack/extra/termutil.h>
|
|
||||||
using namespace DFHack;
|
|
||||||
using MapExtras::Block;
|
|
||||||
using MapExtras::MapCache;
|
|
||||||
|
|
||||||
typedef std::map <DFCoord, uint32_t> coordmap;
|
|
||||||
|
|
||||||
int main (int argc, char * argv[])
|
|
||||||
{
|
|
||||||
bool temporary_terminal = TemporaryTerminal();
|
|
||||||
// Command line options
|
|
||||||
bool destroy = false;
|
|
||||||
if(argc > 1 && strcmp(argv[1],"-d") == 0)
|
|
||||||
destroy = true;
|
|
||||||
|
|
||||||
DFHack::ContextManager CM ("Memory.xml");
|
|
||||||
DFHack::Context * DF;
|
|
||||||
DFHack::VersionInfo *mem;
|
|
||||||
DFHack::Gui * Gui;
|
|
||||||
DFHack::Items * Items;
|
|
||||||
DFHack::Maps *Maps;
|
|
||||||
DFHack::occupancies40d * occupancies;
|
|
||||||
|
|
||||||
cout << "This utility lets you quickly move all items designated to be dumped." << endl
|
|
||||||
<< "Items are instantly moved to the cursor position, the dump flag is unset," << endl
|
|
||||||
<< "and the forbid flag is set, as if it had been dumped normally." << endl
|
|
||||||
<< "Be aware that any active dump item tasks still point at the item." << endl << endl;
|
|
||||||
|
|
||||||
uint32_t vector_addr = 0;
|
|
||||||
|
|
||||||
try
|
|
||||||
{
|
|
||||||
DF = CM.getSingleContext();
|
|
||||||
DF->Attach();
|
|
||||||
mem = DF->getMemoryInfo();
|
|
||||||
Gui = DF->getGui();
|
|
||||||
Items = DF->getItems();
|
|
||||||
Maps = DF->getMaps();
|
|
||||||
DFHack::OffsetGroup* itemGroup = mem->getGroup("Items");
|
|
||||||
vector_addr = itemGroup->getAddress("items_vector");
|
|
||||||
}
|
|
||||||
catch (exception& e)
|
|
||||||
{
|
|
||||||
cerr << e.what() << endl;
|
|
||||||
if(temporary_terminal)
|
|
||||||
cin.ignore();
|
|
||||||
return 1;
|
|
||||||
}
|
|
||||||
|
|
||||||
DFHack::Process * p = DF->getProcess();
|
|
||||||
DFHack::DfVector <uint32_t> p_items (p, vector_addr);
|
|
||||||
uint32_t numItems = p_items.size();
|
|
||||||
|
|
||||||
// init the map
|
|
||||||
if(!Maps->Start())
|
|
||||||
{
|
|
||||||
cerr << "Can't initialize map." << endl;
|
|
||||||
if(temporary_terminal)
|
|
||||||
cin.ignore();
|
|
||||||
return 1;
|
|
||||||
}
|
|
||||||
MapCache MC (Maps);
|
|
||||||
|
|
||||||
int i = 0;
|
|
||||||
int dumped_total = 0;
|
|
||||||
|
|
||||||
int cx, cy, cz;
|
|
||||||
DFCoord pos_cursor;
|
|
||||||
if(!destroy)
|
|
||||||
{
|
|
||||||
if (!Gui->getCursorCoords(cx,cy,cz))
|
|
||||||
{
|
|
||||||
cerr << "Cursor position not found. Please enabled the cursor." << endl;
|
|
||||||
if(temporary_terminal)
|
|
||||||
cin.ignore();
|
|
||||||
return 1;
|
|
||||||
}
|
|
||||||
pos_cursor = DFCoord(cx,cy,cz);
|
|
||||||
{
|
|
||||||
Block * b = MC.BlockAt(pos_cursor / 16);
|
|
||||||
if(!b)
|
|
||||||
{
|
|
||||||
cerr << "Cursor is in an invalid area. Place it over something save first." << endl;
|
|
||||||
if(temporary_terminal)
|
|
||||||
cin.ignore();
|
|
||||||
return 1;
|
|
||||||
}
|
|
||||||
uint16_t ttype = MC.tiletypeAt(pos_cursor);
|
|
||||||
if(!DFHack::isFloorTerrain(ttype))
|
|
||||||
{
|
|
||||||
cerr << "Cursor should be placed over a floor." << endl;
|
|
||||||
if(temporary_terminal)
|
|
||||||
cin.ignore();
|
|
||||||
return 1;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
coordmap counts;
|
|
||||||
// proceed with the dumpification operation
|
|
||||||
for(uint32_t i=0; i< numItems; i++)
|
|
||||||
{
|
|
||||||
DFHack::dfh_item temp;
|
|
||||||
Items->readItem(p_items[i],temp);
|
|
||||||
DFCoord pos_item(temp.base.x, temp.base.y, temp.base.z);
|
|
||||||
|
|
||||||
// keep track how many items are at places. all items.
|
|
||||||
coordmap::iterator it = counts.find(pos_item);
|
|
||||||
if(it == counts.end())
|
|
||||||
{
|
|
||||||
std::pair< coordmap::iterator, bool > inserted = counts.insert(std::make_pair(pos_item,1));
|
|
||||||
it = inserted.first;
|
|
||||||
}
|
|
||||||
else
|
|
||||||
{
|
|
||||||
it->second ++;
|
|
||||||
}
|
|
||||||
// iterator is valid here, we use it later to decrement the pile counter if the item is moved
|
|
||||||
|
|
||||||
// only dump the stuff marked for dumping and laying on the ground
|
|
||||||
if ( !temp.base.flags.dump
|
|
||||||
|| !temp.base.flags.on_ground
|
|
||||||
|| temp.base.flags.construction
|
|
||||||
|| temp.base.flags.hidden
|
|
||||||
|| temp.base.flags.in_building
|
|
||||||
|| temp.base.flags.in_chest
|
|
||||||
|| temp.base.flags.in_inventory
|
|
||||||
|| temp.base.flags.construction
|
|
||||||
)
|
|
||||||
continue;
|
|
||||||
|
|
||||||
if(!destroy) // move to cursor
|
|
||||||
{
|
|
||||||
// Change flags to indicate the dump was completed, as if by super-dwarfs
|
|
||||||
temp.base.flags.dump = false;
|
|
||||||
temp.base.flags.forbid = true;
|
|
||||||
|
|
||||||
// Don't move items if they're already at the cursor
|
|
||||||
if (pos_cursor == pos_item)
|
|
||||||
continue;
|
|
||||||
|
|
||||||
// Move the item
|
|
||||||
temp.base.x = pos_cursor.x;
|
|
||||||
temp.base.y = pos_cursor.y;
|
|
||||||
temp.base.z = pos_cursor.z;
|
|
||||||
}
|
|
||||||
else // destroy
|
|
||||||
{
|
|
||||||
temp.base.flags.garbage_colect = true;
|
|
||||||
}
|
|
||||||
Items->writeItem(temp);
|
|
||||||
// keeping track of item pile sizes ;)
|
|
||||||
it->second --;
|
|
||||||
dumped_total++;
|
|
||||||
}
|
|
||||||
if(!destroy) // TODO: do we have to do any of this when destroying items?
|
|
||||||
{
|
|
||||||
// for each item pile, see if it reached zero. if so, unset item flag on the tile it's on
|
|
||||||
coordmap::iterator it = counts.begin();
|
|
||||||
coordmap::iterator end = counts.end();
|
|
||||||
while(it != end)
|
|
||||||
{
|
|
||||||
if(it->second == 0)
|
|
||||||
{
|
|
||||||
t_occupancy occ = MC.occupancyAt(it->first);
|
|
||||||
occ.bits.item = false;
|
|
||||||
MC.setOccupancyAt(it->first, occ);
|
|
||||||
}
|
|
||||||
it++;
|
|
||||||
}
|
|
||||||
// Set "item here" flag on target tile, if we moved any items to the target tile.
|
|
||||||
if (dumped_total > 0)
|
|
||||||
{
|
|
||||||
// assume there is a possibility the cursor points at some weird location with missing block data
|
|
||||||
Block * b = MC.BlockAt(pos_cursor / 16);
|
|
||||||
if(b)
|
|
||||||
{
|
|
||||||
t_occupancy occ = MC.occupancyAt(pos_cursor);
|
|
||||||
occ.bits.item = 1;
|
|
||||||
MC.setOccupancyAt(pos_cursor,occ);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
// write map changes back to DF.
|
|
||||||
MC.WriteAll();
|
|
||||||
// Is this necessary? Is "forbid" a dirtyable attribute like "dig" is?
|
|
||||||
Maps->WriteDirtyBit(cx/16, cy/16, cz, true);
|
|
||||||
}
|
|
||||||
DF->Detach();
|
|
||||||
if(!destroy)
|
|
||||||
{
|
|
||||||
cout << "Done. " << dumped_total << " items quickdumped." << endl;
|
|
||||||
}
|
|
||||||
else
|
|
||||||
{
|
|
||||||
cout << "Done. " << dumped_total << " items marked for destruction." << endl;
|
|
||||||
}
|
|
||||||
if(temporary_terminal)
|
|
||||||
{
|
|
||||||
cout << "Press any key to continue" << endl;
|
|
||||||
cin.ignore();
|
|
||||||
}
|
|
||||||
return 0;
|
|
||||||
}
|
|
||||||
Loading…
Reference in New Issue