// Creature dump

#include <iostream>

#include <iomanip>

#include <sstream>

#include <climits>

#include <integers.h>

#include <vector>

using namespace std;

#include <DFTypes.h>

#include <DFHackAPI.h>

#include <DFMemInfo.h>

/*

oh. fsck it. I'll do the hexdump now and add the things I found in a research/ folder

groups of four bytes, 4 per line

1 space between bytes

2 between groups

offset from the object start on the left

and length set from command line

default 256

*/

/*

address = absolute address of dump start

length = length in lines. 1 line = 16 bytes

*/

void hexdump (DFHack::API& DF, uint32_t address, uint32_t length)

{

char *buf = new char[length * 16];

DF.ReadRaw(address, length * 16, (uint8_t *) buf);

for (int i = 0; i < length; i++)

{

// leading offset

cout << "0x" << hex << setw(4) << i*16 << " ";

// groups

for(int j = 0; j < 4; j++)

{

// bytes

for(int k = 0; k < 4; k++)

{

int idx = i * 16 + j * 4 + k;

cout << hex << setw(2) << int(static_cast<unsigned char>(buf[idx])) << " ";

}

cout << " ";

}

cout << endl;

}

delete buf;

}

void interleave_hex (DFHack::API& DF, vector < uint32_t > & addresses, uint32_t length)

{

vector <char * > bufs;

for(int counter = 0; counter < addresses.size(); counter ++)

{

char * buf = new char[length * 16];

DF.ReadRaw(addresses[counter], length * 16, (uint8_t *) buf);

bufs.push_back(buf);

}

cout << setfill('0');

// output a header

cout << "line offset ";

for (int obj = 0; obj < addresses.size(); obj++)

{

cout << "0x" << hex << setw(9) << addresses[obj] << " ";

}

cout << endl;

for(int offs = 0 ; offs < length * 16; offs += 4)

{

if((!(offs % 16)) && offs != 0)

{

cout << endl;

}

cout << setfill(' ');

cout << dec << setw(4) << offs/4 << " ";

cout << setfill('0');

cout << "0x" << hex << setw(4) << offs << " ";

for (int object = 0; object < bufs.size(); object++)

{

// bytes

for(int k = 0; k < 4; k++)

{

uint8_t data = bufs[object][offs + k];

cout << hex << setw(2) << int(static_cast<unsigned char>(data)) << " ";

}

cout << " ";

}

cout << endl;

}

for(int counter = 0; counter < addresses.size(); counter ++)

{

delete bufs[counter];

}

}

template <typename T>

void print_bits ( T val, std::ostream& out )

{

T n_bits = sizeof ( val ) * CHAR_BIT;

for ( unsigned i = 0; i < n_bits; ++i ) {

out<< !!( val & 1 ) << " ";

val >>= 1;

}

}

int main (int argc,const char* argv[])

{

if (argc < 2 || argc > 3)

{

cout << "usage:" << endl;

cout << argv[0] << " object_name [number of lines]" << endl;

return 0;

}

int lines = 16;

if(argc == 3)

{

string s = argv[2]; //blah. I don't care

istringstream ins; // Declare an input string stream.

ins.str(s); // Specify string to read.

ins >> lines; // Reads the integers from the string.

}

vector<DFHack::t_matgloss> creaturestypes;

DFHack::API DF ("Memory.xml");

if(!DF.Attach())

{

cerr << "DF not found" << endl;

return 1;

}

DFHack::memory_info * mem = DF.getMemoryInfo();

uint32_t numBuildings;

if(DF.InitReadBuildings(numBuildings))

{

cout << numBuildings << endl;

vector < uint32_t > addresses;

for(uint32_t i = 0; i < numBuildings; i++)

{

DFHack::t_building temp;

DF.ReadBuilding(i, temp);

if(temp.type != 0xFFFFFFFF) // check if type isn't invalid

{

string typestr;

mem->resolveClassIDToClassname(temp.type, typestr);

if(typestr == argv[1])

{

//cout << buildingtypes[temp.type] << " 0x" << hex << temp.origin << endl;

//hexdump(DF, temp.origin, 16);

addresses.push_back(temp.origin);

}

}

else

{

// couldn't translate type, print out the vtable

cout << "unknown vtable: " << temp.vtable << endl;

}

}

interleave_hex(DF,addresses,lines / 4);

DF.FinishReadBuildings();

}

else

{

cerr << "buildings not supported for this DF version" << endl;

}

DF.Detach();

#ifndef LINUX_BUILD

cout << "Done. Press any key to continue" << endl;

cin.ignore();

#endif

return 0;

}