498 lines
15 KiB
C++
498 lines
15 KiB
C++
// Creature dump
|
|
|
|
#include <iostream>
|
|
#include <climits>
|
|
#include <integers.h>
|
|
#include <vector>
|
|
using namespace std;
|
|
|
|
#include <DFTypes.h>
|
|
#include <DFHackAPI.h>
|
|
#include <DFMemInfo.h>
|
|
|
|
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;
|
|
}
|
|
}
|
|
struct matGlosses
|
|
{
|
|
vector<DFHack::t_matglossPlant> plantMat;
|
|
vector<DFHack::t_matgloss> woodMat;
|
|
vector<DFHack::t_matgloss> stoneMat;
|
|
vector<DFHack::t_matgloss> metalMat;
|
|
vector<DFHack::t_matgloss> creatureMat;
|
|
};
|
|
enum likeType
|
|
{
|
|
FAIL = 0,
|
|
MATERIAL = 1,
|
|
ITEM = 2,
|
|
FOOD = 3
|
|
};
|
|
|
|
vector<DFHack::t_matgloss> creaturestypes;
|
|
matGlosses mat;
|
|
vector< vector <DFHack::t_itemType> > itemTypes;
|
|
map<string, vector<string> > names;
|
|
DFHack::memory_info *mem;
|
|
|
|
likeType printLike(DFHack::t_like like, const matGlosses & mat,const vector< vector <DFHack::t_itemType> > & itemTypes)
|
|
{ // The function in DF which prints out the likes is a monster, it is a huge switch statement with tons of options and calls a ton of other functions as well,
|
|
//so I am not going to try and put all the possibilites here, only the low hanging fruit, with stones and metals, as well as items,
|
|
//you can easily find good canidates for military duty for instance
|
|
//The ideal thing to do would be to call the df function directly with the desired likes, the df function modifies a string, so it should be possible to do...
|
|
if(like.active){
|
|
if(like.type ==0){
|
|
switch (like.material.type)
|
|
{
|
|
case 0:
|
|
cout << mat.woodMat[like.material.index].name;
|
|
return(MATERIAL);
|
|
case 1:
|
|
cout << mat.stoneMat[like.material.index].name;
|
|
return(MATERIAL);
|
|
case 2:
|
|
cout << mat.metalMat[like.material.index].name;
|
|
return(MATERIAL);
|
|
case 12: // don't ask me why this has such a large jump, maybe this is not actually the matType for plants, but they all have this set to 12
|
|
cout << mat.plantMat[like.material.index].name;
|
|
return(MATERIAL);
|
|
case 32:
|
|
cout << mat.plantMat[like.material.index].name;
|
|
return(MATERIAL);
|
|
case 121:
|
|
cout << mat.creatureMat[like.material.index].name;
|
|
return(MATERIAL);
|
|
default:
|
|
return(FAIL);
|
|
}
|
|
}
|
|
else if(like.type == 4 && like.itemIndex != -1){
|
|
switch(like.itemClass)
|
|
{
|
|
case 24:
|
|
cout << itemTypes[0][like.itemIndex].name;
|
|
return(ITEM);
|
|
case 25:
|
|
cout << itemTypes[4][like.itemIndex].name;
|
|
return(ITEM);
|
|
case 26:
|
|
cout << itemTypes[8][like.itemIndex].name;
|
|
return(ITEM);
|
|
case 27:
|
|
cout << itemTypes[9][like.itemIndex].name;
|
|
return(ITEM);
|
|
case 28:
|
|
cout << itemTypes[10][like.itemIndex].name;
|
|
return(ITEM);
|
|
case 29:
|
|
cout << itemTypes[7][like.itemIndex].name;
|
|
return(ITEM);
|
|
case 38:
|
|
cout << itemTypes[5][like.itemIndex].name;
|
|
return(ITEM);
|
|
case 63:
|
|
cout << itemTypes[11][like.itemIndex].name;
|
|
return(ITEM);
|
|
case 68:
|
|
case 69:
|
|
cout << itemTypes[6][like.itemIndex].name;
|
|
return(ITEM);
|
|
case 70:
|
|
cout << itemTypes[1][like.itemIndex].name;
|
|
return(ITEM);
|
|
default:
|
|
// cout << like.itemClass << ":" << like.itemIndex;
|
|
return(FAIL);
|
|
}
|
|
}
|
|
else if(like.material.type != -1){// && like.material.index == -1){
|
|
if(like.type == 2){
|
|
switch(like.itemClass)
|
|
{
|
|
case 52:
|
|
case 53:
|
|
case 58:
|
|
cout << mat.plantMat[like.material.type].name;
|
|
return(FOOD);
|
|
case 72:
|
|
if(like.material.type =! 10){ // 10 is for milk stuff, which I don't know how to do
|
|
cout << mat.plantMat[like.material.index].extract_name;
|
|
return(FOOD);
|
|
}
|
|
return(FAIL);
|
|
case 74:
|
|
cout << mat.plantMat[like.material.index].drink_name;
|
|
return(FOOD);
|
|
case 75:
|
|
cout << mat.plantMat[like.material.index].food_name;
|
|
return(FOOD);
|
|
case 47:
|
|
case 48:
|
|
cout << mat.creatureMat[like.material.type].name;
|
|
return(FOOD);
|
|
default:
|
|
return(FAIL);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
return(FAIL);
|
|
}
|
|
|
|
|
|
void printCreature(DFHack::API & DF, const DFHack::t_creature & creature)
|
|
{
|
|
if(string(creaturestypes[creature.type].id) == "DWARF")
|
|
{
|
|
cout << "address: " << creature.origin << " creature type: " << creaturestypes[creature.type].id << ", position: " << creature.x << "x " << creature.y << "y "<< creature.z << "z" << endl;
|
|
bool addendl = false;
|
|
if(creature.first_name[0])
|
|
{
|
|
cout << "first name: " << creature.first_name;
|
|
addendl = true;
|
|
}
|
|
if(creature.nick_name[0])
|
|
{
|
|
cout << ", nick name: " << creature.nick_name;
|
|
addendl = true;
|
|
}
|
|
string transName = DF.TranslateName(creature.last_name,names,creaturestypes[creature.type].id);
|
|
if(!transName.empty())
|
|
{
|
|
cout << ", trans name: " << transName;
|
|
addendl=true;
|
|
}
|
|
//cout << ", generic name: " << DF.TranslateName(creature.last_name,names,"GENERIC");
|
|
/*
|
|
if(!creature.trans_name.empty()){
|
|
cout << ", trans name: " << creature.trans_name;
|
|
addendl =true;
|
|
}
|
|
if(!creature.generic_name.empty()){
|
|
cout << ", generic name: " << creature.generic_name;
|
|
addendl=true;
|
|
}
|
|
*/
|
|
cout << ", likes: ";
|
|
for(uint32_t i = 0;i<creature.numLikes; i++)
|
|
{
|
|
if(printLike(creature.likes[i],mat,itemTypes))
|
|
{
|
|
cout << ", ";
|
|
}
|
|
}
|
|
if(addendl)
|
|
{
|
|
cout << endl;
|
|
addendl = false;
|
|
}
|
|
cout << "profession: " << mem->getProfession(creature.profession) << "(" << (int) creature.profession << ")";
|
|
if(creature.custom_profession[0])
|
|
{
|
|
cout << ", custom profession: " << creature.custom_profession;
|
|
}
|
|
if(creature.current_job.active)
|
|
{
|
|
cout << ", current job: " << mem->getJob(creature.current_job.jobId);
|
|
}
|
|
cout << endl;
|
|
cout << "happiness: " << creature.happiness << ", strength: " << creature.strength << ", agility: "
|
|
<< creature.agility << ", toughness: " << creature.toughness << ", money: " << creature.money << ", id: " << creature.id;
|
|
if(creature.squad_leader_id != -1)
|
|
{
|
|
cout << ", squad_leader_id: " << creature.squad_leader_id;
|
|
}
|
|
cout << ", sex: ";
|
|
if(creature.sex == 0)
|
|
{
|
|
cout << "Female";
|
|
}
|
|
else
|
|
{
|
|
cout <<"Male";
|
|
}
|
|
cout << endl;
|
|
/*
|
|
//skills
|
|
for(unsigned int i = 0; i < creature.skills.size();i++){
|
|
if(i > 0){
|
|
cout << ", ";
|
|
}
|
|
cout << creature.skills[i].name << ": " << creature.skills[i].rating;
|
|
}
|
|
*/
|
|
/*
|
|
* FLAGS 1
|
|
*/
|
|
cout << "flags1: ";
|
|
print_bits(creature.flags1.whole, cout);
|
|
cout << endl;
|
|
if(creature.flags1.bits.dead)
|
|
{
|
|
cout << "dead ";
|
|
}
|
|
if(creature.flags1.bits.on_ground)
|
|
{
|
|
cout << "on the ground, ";
|
|
}
|
|
if(creature.flags1.bits.skeleton)
|
|
{
|
|
cout << "skeletal ";
|
|
}
|
|
if(creature.flags1.bits.zombie)
|
|
{
|
|
cout << "zombie ";
|
|
}
|
|
if(creature.flags1.bits.tame)
|
|
{
|
|
cout << "tame ";
|
|
}
|
|
if(creature.flags1.bits.royal_guard)
|
|
{
|
|
cout << "royal_guard ";
|
|
}
|
|
if(creature.flags1.bits.fortress_guard)
|
|
{
|
|
cout << "fortress_guard ";
|
|
}
|
|
/*
|
|
* FLAGS 2
|
|
*/
|
|
cout << endl << "flags2: ";
|
|
print_bits(creature.flags2.whole, cout);
|
|
cout << endl;
|
|
if(creature.flags2.bits.killed)
|
|
{
|
|
cout << "killed by kill function, ";
|
|
}
|
|
if(creature.flags2.bits.resident)
|
|
{
|
|
cout << "resident, ";
|
|
}
|
|
if(creature.flags2.bits.gutted)
|
|
{
|
|
cout << "gutted, ";
|
|
}
|
|
if(creature.flags2.bits.slaughter)
|
|
{
|
|
cout << "marked for slaughter, ";
|
|
}
|
|
if(creature.flags2.bits.underworld)
|
|
{
|
|
cout << "from the underworld, ";
|
|
}
|
|
cout << endl << endl;
|
|
}
|
|
}
|
|
|
|
|
|
int main (void)
|
|
{
|
|
DFHack::API DF("Memory.xml");
|
|
if(!DF.Attach())
|
|
{
|
|
cerr << "DF not found" << endl;
|
|
return 1;
|
|
}
|
|
|
|
|
|
DF.ReadItemTypes(itemTypes);
|
|
|
|
|
|
DF.ReadPlantMatgloss(mat.plantMat);
|
|
DF.ReadWoodMatgloss(mat.woodMat);
|
|
DF.ReadStoneMatgloss(mat.stoneMat);
|
|
DF.ReadMetalMatgloss(mat.metalMat);
|
|
DF.ReadCreatureMatgloss(mat.creatureMat);
|
|
|
|
mem = DF.getMemoryInfo();
|
|
// get stone matgloss mapping
|
|
if(!DF.ReadCreatureMatgloss(creaturestypes))
|
|
{
|
|
cerr << "Can't get the creature types." << endl;
|
|
return 1;
|
|
}
|
|
|
|
vector< vector<string> > englishWords;
|
|
vector< vector<string> > foreignWords;
|
|
if(!DF.InitReadNameTables(englishWords,foreignWords))
|
|
{
|
|
cerr << "Can't get name tables" << endl;
|
|
return 1;
|
|
}
|
|
uint32_t numCreatures;
|
|
if(!DF.InitReadCreatures(numCreatures))
|
|
{
|
|
cerr << "Can't get creatures" << endl;
|
|
return 1;
|
|
}
|
|
DF.InitViewAndCursor();
|
|
for(uint32_t i = 0; i < numCreatures; i++)
|
|
{
|
|
DFHack::t_creature temp;
|
|
DF.ReadCreature(i, temp);
|
|
if(string(creaturestypes[temp.type].id) == "DWARF")
|
|
{
|
|
cout << "address: " << temp.origin << " creature type: " << creaturestypes[temp.type].id << ", position: " << temp.x << "x " << temp.y << "y "<< temp.z << "z" << endl;
|
|
bool addendl = false;
|
|
if(temp.name.first_name[0])
|
|
{
|
|
cout << "first name: " << temp.name.first_name;
|
|
addendl = true;
|
|
}
|
|
if(temp.name.nickname[0])
|
|
{
|
|
cout << ", nick name: " << temp.name.nickname;
|
|
addendl = true;
|
|
}
|
|
string transName = DF.TranslateName(temp.name,englishWords,foreignWords,false);
|
|
if(!transName.empty())
|
|
{
|
|
cout << ", trans name: " << transName;
|
|
addendl=true;
|
|
}
|
|
//cout << ", generic name: " << DF.TranslateName(temp.last_name,names,"GENERIC");
|
|
/*
|
|
if(!temp.trans_name.empty()){
|
|
cout << ", trans name: " << temp.trans_name;
|
|
addendl =true;
|
|
}
|
|
if(!temp.generic_name.empty()){
|
|
cout << ", generic name: " << temp.generic_name;
|
|
addendl=true;
|
|
}
|
|
*/
|
|
cout << ", likes: ";
|
|
for(uint32_t i = 0;i<temp.numLikes; i++)
|
|
{
|
|
if(printLike(temp.likes[i],mat,itemTypes))
|
|
{
|
|
cout << ", ";
|
|
}
|
|
}
|
|
if(addendl)
|
|
{
|
|
cout << endl;
|
|
addendl = false;
|
|
}
|
|
cout << "profession: " << mem->getProfession(temp.profession) << "(" << (int) temp.profession << ")";
|
|
if(temp.custom_profession[0])
|
|
{
|
|
cout << ", custom profession: " << temp.custom_profession;
|
|
}
|
|
if(temp.current_job.active)
|
|
{
|
|
cout << ", current job: " << mem->getJob(temp.current_job.jobId);
|
|
}
|
|
cout << endl;
|
|
cout << "happiness: " << temp.happiness << ", strength: " << temp.strength << ", agility: "
|
|
<< temp.agility << ", toughness: " << temp.toughness << ", money: " << temp.money << ", id: " << temp.id;
|
|
if(temp.squad_leader_id != -1)
|
|
{
|
|
cout << ", squad_leader_id: " << temp.squad_leader_id;
|
|
}
|
|
cout << ", sex: ";
|
|
if(temp.sex == 0)
|
|
{
|
|
cout << "Female";
|
|
}
|
|
else
|
|
{
|
|
cout <<"Male";
|
|
}
|
|
cout << endl;
|
|
/*
|
|
//skills
|
|
for(unsigned int i = 0; i < temp.skills.size();i++){
|
|
if(i > 0){
|
|
cout << ", ";
|
|
}
|
|
cout << temp.skills[i].name << ": " << temp.skills[i].rating;
|
|
}
|
|
*/
|
|
/*
|
|
* FLAGS 1
|
|
*/
|
|
cout << "flags1: ";
|
|
print_bits(temp.flags1.whole, cout);
|
|
cout << endl;
|
|
if(temp.flags1.bits.dead)
|
|
{
|
|
cout << "dead ";
|
|
}
|
|
if(temp.flags1.bits.on_ground)
|
|
{
|
|
cout << "on the ground, ";
|
|
}
|
|
if(temp.flags1.bits.skeleton)
|
|
{
|
|
cout << "skeletal ";
|
|
}
|
|
if(temp.flags1.bits.zombie)
|
|
{
|
|
cout << "zombie ";
|
|
}
|
|
if(temp.flags1.bits.tame)
|
|
{
|
|
cout << "tame ";
|
|
}
|
|
if(temp.flags1.bits.royal_guard)
|
|
{
|
|
cout << "royal_guard ";
|
|
}
|
|
if(temp.flags1.bits.fortress_guard)
|
|
{
|
|
cout << "fortress_guard ";
|
|
}
|
|
/*
|
|
* FLAGS 2
|
|
*/
|
|
cout << endl << "flags2: ";
|
|
print_bits(temp.flags2.whole, cout);
|
|
cout << endl;
|
|
if(temp.flags2.bits.killed)
|
|
{
|
|
cout << "killed by kill function, ";
|
|
}
|
|
if(temp.flags2.bits.resident)
|
|
{
|
|
cout << "resident, ";
|
|
}
|
|
if(temp.flags2.bits.gutted)
|
|
{
|
|
cout << "gutted, ";
|
|
}
|
|
if(temp.flags2.bits.slaughter)
|
|
{
|
|
cout << "marked for slaughter, ";
|
|
}
|
|
if(temp.flags2.bits.underworld)
|
|
{
|
|
cout << "from the underworld, ";
|
|
}
|
|
cout << endl << endl;
|
|
}
|
|
}
|
|
uint32_t currentIdx;
|
|
DFHack::t_creature currentCreature;
|
|
DF.getCurrentCursorCreature(currentIdx);
|
|
cout << "current creature" << endl;
|
|
|
|
DF.ReadCreature(currentIdx, currentCreature);
|
|
printCreature(DF,currentCreature);
|
|
DF.FinishReadCreatures();
|
|
DF.Detach();
|
|
#ifndef LINUX_BUILD
|
|
cout << "Done. Press any key to continue" << endl;
|
|
cin.ignore();
|
|
#endif
|
|
return 0;
|
|
}
|