Occupancy support in MapCache. Added the item dumper tool mentioned on irc (autodump).
Petr Mrázek
parent
fc1cdac01d
commit
923c8cae9c
@ -0,0 +1,178 @@
|
||||
// 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.
|
||||
using namespace DFHack;
|
||||
using MapExtras::Block;
|
||||
using MapExtras::MapCache;
|
||||
|
||||
typedef std::map <DFCoord, uint32_t> coordmap;
|
||||
|
||||
int main ()
|
||||
{
|
||||
|
||||
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;
|
||||
try
|
||||
{
|
||||
DF = CM.getSingleContext();
|
||||
DF->Attach();
|
||||
mem = DF->getMemoryInfo();
|
||||
Gui = DF->getGui();
|
||||
Items = DF->getItems();
|
||||
Maps = DF->getMaps();
|
||||
}
|
||||
catch (exception& e)
|
||||
{
|
||||
cerr << e.what() << endl;
|
||||
#ifndef LINUX_BUILD
|
||||
cin.ignore();
|
||||
#endif
|
||||
return 1;
|
||||
}
|
||||
|
||||
DFHack::Process * p = DF->getProcess();
|
||||
|
||||
// FIXME: these can fail and should be wrapped in a try-catch
|
||||
DFHack::OffsetGroup* itemGroup = mem->getGroup("Items");
|
||||
unsigned vector_addr = itemGroup->getAddress("items_vector");
|
||||
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;
|
||||
#ifndef LINUX_BUILD
|
||||
cin.ignore();
|
||||
#endif
|
||||
return 1;
|
||||
}
|
||||
MapCache MC (Maps);
|
||||
|
||||
int i = 0;
|
||||
int dumped_total = 0;
|
||||
|
||||
int cx, cy, cz;
|
||||
|
||||
if (!Gui->getCursorCoords(cx,cy,cz))
|
||||
{
|
||||
cerr << "Cursor position not found. Please enabled the cursor." << endl;
|
||||
#ifndef LINUX_BUILD
|
||||
cin.ignore();
|
||||
#endif
|
||||
return 1;
|
||||
}
|
||||
DFCoord pos_cursor(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;
|
||||
#ifndef LINUX_BUILD
|
||||
cin.ignore();
|
||||
#endif
|
||||
return 1;
|
||||
}
|
||||
// TODO: check if the target is floor? maybe?
|
||||
}
|
||||
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)
|
||||
continue;
|
||||
|
||||
// Change flags to indicate the dump was completed, as if by super-dwarfs
|
||||
temp.base.flags.dump = 0;
|
||||
temp.base.flags.forbid = 1;
|
||||
|
||||
// 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;
|
||||
Items->writeItem(temp);
|
||||
// keeping track of item pile sizes ;)
|
||||
it->second --;
|
||||
dumped_total++;
|
||||
}
|
||||
// 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();
|
||||
cout << "Done. " << dumped_total << " items quickdumped." << endl;
|
||||
#ifndef LINUX_BUILD
|
||||
cout << "Press any key to continue" << endl;
|
||||
cin.ignore();
|
||||
#endif
|
||||
return 0;
|
||||
}
|
||||
Loading…
Reference in New Issue