Merge branch 'master', remote branch 'peterix/master'
Erik Youngren
commit
e9a7026342
File diff suppressed because it is too large
Load Diff
@ -1,778 +0,0 @@
|
|||||||
/*
|
|
||||||
www.sourceforge.net/projects/dfhack
|
|
||||||
Copyright (c) 2009 Petr Mrázek (peterix), Kenneth Ferland (Impaler[WrG]), dorf
|
|
||||||
|
|
||||||
This software is provided 'as-is', without any express or implied
|
|
||||||
warranty. In no event will the authors be held liable for any
|
|
||||||
damages arising from the use of this software.
|
|
||||||
|
|
||||||
Permission is granted to anyone to use this software for any
|
|
||||||
purpose, including commercial applications, and to alter it and
|
|
||||||
redistribute it freely, subject to the following restrictions:
|
|
||||||
|
|
||||||
1. The origin of this software must not be misrepresented; you must
|
|
||||||
not claim that you wrote the original software. If you use this
|
|
||||||
software in a product, an acknowledgment in the product documentation
|
|
||||||
would be appreciated but is not required.
|
|
||||||
|
|
||||||
2. Altered source versions must be plainly marked as such, and
|
|
||||||
must not be misrepresented as being the original software.
|
|
||||||
|
|
||||||
3. This notice may not be removed or altered from any source
|
|
||||||
distribution.
|
|
||||||
*/
|
|
||||||
|
|
||||||
#include "Internal.h"
|
|
||||||
#include "dfhack/DFMemInfo.h"
|
|
||||||
#include "dfhack/DFError.h"
|
|
||||||
#include "dfhack/DFProcess.h"
|
|
||||||
|
|
||||||
//Inital amount of space in levels vector (since we usually know the number, efficent!)
|
|
||||||
#define NUM_RESERVE_LVLS 20
|
|
||||||
#define NUM_RESERVE_MOODS 6
|
|
||||||
using namespace DFHack;
|
|
||||||
|
|
||||||
/*
|
|
||||||
* Common data types
|
|
||||||
*/
|
|
||||||
namespace DFHack
|
|
||||||
{
|
|
||||||
struct t_type
|
|
||||||
{
|
|
||||||
t_type(uint32_t assign, uint32_t type, std::string classname)
|
|
||||||
:classname(classname),assign(assign),type(type){};
|
|
||||||
std::string classname;
|
|
||||||
uint32_t assign;
|
|
||||||
uint32_t type;
|
|
||||||
};
|
|
||||||
|
|
||||||
struct t_class
|
|
||||||
{
|
|
||||||
t_class(const t_class &old)
|
|
||||||
{
|
|
||||||
classname = old.classname;
|
|
||||||
vtable = old.vtable;
|
|
||||||
assign = old.assign;
|
|
||||||
type_offset = old.type_offset;
|
|
||||||
for(uint32_t i = 0; i < old.subs.size();i++)
|
|
||||||
{
|
|
||||||
t_type * t = new t_type (*old.subs[i]);
|
|
||||||
subs.push_back(t);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
t_class ()
|
|
||||||
{
|
|
||||||
vtable = 0;
|
|
||||||
assign = 0;
|
|
||||||
type_offset = 0;
|
|
||||||
}
|
|
||||||
~t_class()
|
|
||||||
{
|
|
||||||
for(uint32_t i = 0; i < subs.size();i++)
|
|
||||||
{
|
|
||||||
delete subs[i];
|
|
||||||
}
|
|
||||||
subs.clear();
|
|
||||||
}
|
|
||||||
std::string classname;
|
|
||||||
uint32_t vtable;
|
|
||||||
uint32_t assign;// index to typeclass array if multiclass. return value if not.
|
|
||||||
uint32_t type_offset; // offset of type data for multiclass
|
|
||||||
std::vector<t_type *> subs;
|
|
||||||
};
|
|
||||||
}
|
|
||||||
/*
|
|
||||||
* Private data
|
|
||||||
*/
|
|
||||||
class memory_info::Private
|
|
||||||
{
|
|
||||||
public:
|
|
||||||
map <string, uint32_t> addresses;
|
|
||||||
map <string, int32_t> offsets;
|
|
||||||
map <string, uint32_t> hexvals;
|
|
||||||
map <string, string> strings;
|
|
||||||
|
|
||||||
vector<string> professions;
|
|
||||||
vector<string> jobs;
|
|
||||||
vector<string> skills;
|
|
||||||
vector<DFHack::t_level> levels;
|
|
||||||
vector< vector<string> > traits;
|
|
||||||
vector<string> moods;
|
|
||||||
map <uint32_t, string> labors;
|
|
||||||
|
|
||||||
// storage for class and multiclass
|
|
||||||
vector<t_class *> classes;
|
|
||||||
|
|
||||||
// cache for faster name lookup, indexed by classID
|
|
||||||
vector<string> classnames;
|
|
||||||
// map between vptr and class id, needs further type id lookup for multi-classes, not inherited
|
|
||||||
map<uint32_t, t_class *> classIDs;
|
|
||||||
|
|
||||||
// index for the next added class
|
|
||||||
uint32_t classindex;
|
|
||||||
|
|
||||||
int32_t base;
|
|
||||||
Process * p; // the process this belongs to
|
|
||||||
|
|
||||||
string version;
|
|
||||||
OSType OS;
|
|
||||||
};
|
|
||||||
|
|
||||||
// normal constructor
|
|
||||||
memory_info::memory_info()
|
|
||||||
:d(new Private)
|
|
||||||
{
|
|
||||||
d->base = 0;
|
|
||||||
d->p = 0;
|
|
||||||
d->classindex = 0;
|
|
||||||
d->levels.reserve(NUM_RESERVE_LVLS);
|
|
||||||
d->moods.reserve(NUM_RESERVE_MOODS);
|
|
||||||
}
|
|
||||||
|
|
||||||
// copy constructor
|
|
||||||
memory_info::memory_info(const memory_info &old)
|
|
||||||
:d(new Private)
|
|
||||||
{
|
|
||||||
d->version = old.d->version;
|
|
||||||
d->OS = old.d->OS;
|
|
||||||
d->addresses = old.d->addresses;
|
|
||||||
d->offsets = old.d->offsets;
|
|
||||||
d->hexvals = old.d->hexvals;
|
|
||||||
d->strings = old.d->strings;
|
|
||||||
d->base = old.d->base;
|
|
||||||
//d->classes = old.d->classes;
|
|
||||||
for(uint32_t i = 0; i < old.d->classes.size(); i++)
|
|
||||||
{
|
|
||||||
t_class * copy = new t_class(*old.d->classes[i]);
|
|
||||||
d->classes.push_back(copy);
|
|
||||||
}
|
|
||||||
d->classnames = old.d->classnames;
|
|
||||||
d->classindex = old.d->classindex;
|
|
||||||
d->professions = old.d->professions;
|
|
||||||
d->jobs = old.d->jobs;
|
|
||||||
d->skills = old.d->skills;
|
|
||||||
d->traits = old.d->traits;
|
|
||||||
d->labors = old.d->labors;
|
|
||||||
d->levels = old.d->levels;
|
|
||||||
d->moods = old.d->moods;
|
|
||||||
}
|
|
||||||
void memory_info::setParentProcess(Process * _p)
|
|
||||||
{
|
|
||||||
d->p = _p;
|
|
||||||
}
|
|
||||||
|
|
||||||
// destructor
|
|
||||||
memory_info::~memory_info()
|
|
||||||
{
|
|
||||||
// delete the vtables
|
|
||||||
for(uint32_t i = 0; i < d->classes.size();i++)
|
|
||||||
{
|
|
||||||
delete d->classes[i];
|
|
||||||
}
|
|
||||||
// delete our data
|
|
||||||
delete d;
|
|
||||||
}
|
|
||||||
|
|
||||||
void memory_info::setVersion(const char * v)
|
|
||||||
{
|
|
||||||
d->version = v;
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
void memory_info::setVersion(const string &v)
|
|
||||||
{
|
|
||||||
d->version = v;
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
string memory_info::getVersion()
|
|
||||||
{
|
|
||||||
return d->version;
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
void memory_info::setOS(const char *os)
|
|
||||||
{
|
|
||||||
string oss = os;
|
|
||||||
if(oss == "windows")
|
|
||||||
d->OS = OS_WINDOWS;
|
|
||||||
else if(oss == "linux")
|
|
||||||
d->OS = OS_LINUX;
|
|
||||||
else
|
|
||||||
d->OS = OS_BAD;
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
void memory_info::setOS(const string &os)
|
|
||||||
{
|
|
||||||
if(os == "windows")
|
|
||||||
d->OS = OS_WINDOWS;
|
|
||||||
else if(os == "linux")
|
|
||||||
d->OS = OS_LINUX;
|
|
||||||
else
|
|
||||||
d->OS = OS_BAD;
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
void memory_info::setOS(OSType os)
|
|
||||||
{
|
|
||||||
if(os >= OS_WINDOWS && os < OS_BAD)
|
|
||||||
{
|
|
||||||
d->OS = os;
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
d->OS = OS_BAD;
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
memory_info::OSType memory_info::getOS() const
|
|
||||||
{
|
|
||||||
return d->OS;
|
|
||||||
}
|
|
||||||
|
|
||||||
uint32_t memory_info::getBase () const
|
|
||||||
{
|
|
||||||
return d->base;
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
void memory_info::setBase (const string &s)
|
|
||||||
{
|
|
||||||
d->base = strtol(s.c_str(), NULL, 16);
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
void memory_info::setBase (const uint32_t b)
|
|
||||||
{
|
|
||||||
d->base = b;
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
void memory_info::setOffset (const string & key, const string & value)
|
|
||||||
{
|
|
||||||
int32_t offset = strtol(value.c_str(), NULL, 16);
|
|
||||||
d->offsets[key] = offset;
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
void memory_info::setAddress (const string & key, const string & value)
|
|
||||||
{
|
|
||||||
uint32_t address = strtol(value.c_str(), NULL, 16);
|
|
||||||
d->addresses[key] = address;
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
void memory_info::setHexValue (const string & key, const string & value)
|
|
||||||
{
|
|
||||||
uint32_t hexval = strtol(value.c_str(), NULL, 16);
|
|
||||||
d->hexvals[key] = hexval;
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
void memory_info::setString (const string & key, const string & value)
|
|
||||||
{
|
|
||||||
d->strings[key] = value;
|
|
||||||
}
|
|
||||||
|
|
||||||
void memory_info::setLabor(const string & key, const string & value)
|
|
||||||
{
|
|
||||||
uint32_t keyInt = strtol(key.c_str(), NULL, 10);
|
|
||||||
d->labors[keyInt] = value;
|
|
||||||
}
|
|
||||||
|
|
||||||
void memory_info::setProfession (const string & key, const string & value)
|
|
||||||
{
|
|
||||||
uint32_t keyInt = strtol(key.c_str(), NULL, 10);
|
|
||||||
if(d->professions.size() <= keyInt)
|
|
||||||
{
|
|
||||||
d->professions.resize(keyInt+1,"");
|
|
||||||
}
|
|
||||||
d->professions[keyInt] = value;
|
|
||||||
}
|
|
||||||
|
|
||||||
void memory_info::setJob (const string & key, const string & value)
|
|
||||||
{
|
|
||||||
uint32_t keyInt = strtol(key.c_str(), NULL, 10);
|
|
||||||
if(d->jobs.size() <= keyInt)
|
|
||||||
{
|
|
||||||
d->jobs.resize(keyInt+1);
|
|
||||||
}
|
|
||||||
d->jobs[keyInt] = value;
|
|
||||||
}
|
|
||||||
|
|
||||||
void memory_info::setSkill (const string & key, const string & value)
|
|
||||||
{
|
|
||||||
uint32_t keyInt = strtol(key.c_str(), NULL, 10);
|
|
||||||
if(d->skills.size() <= keyInt){
|
|
||||||
d->skills.resize(keyInt+1);
|
|
||||||
}
|
|
||||||
d->skills[keyInt] = value;
|
|
||||||
}
|
|
||||||
|
|
||||||
void memory_info::setLevel(const std::string &nLevel,
|
|
||||||
const std::string &nName,
|
|
||||||
const std::string &nXp)
|
|
||||||
{
|
|
||||||
uint32_t keyInt = strtol(nLevel.c_str(), NULL, 10);
|
|
||||||
|
|
||||||
if(d->levels.size() <= keyInt)
|
|
||||||
d->levels.resize(keyInt+1);
|
|
||||||
|
|
||||||
d->levels[keyInt].level = keyInt;
|
|
||||||
d->levels[keyInt].name = nName;
|
|
||||||
d->levels[keyInt].xpNxtLvl = strtol(nXp.c_str(), NULL, 10);
|
|
||||||
}
|
|
||||||
|
|
||||||
void memory_info::setMood(const std::string &id, const std::string &mood)
|
|
||||||
{
|
|
||||||
uint32_t keyInt = strtol(id.c_str(), NULL, 10);
|
|
||||||
|
|
||||||
if(d->moods.size() <= keyInt)
|
|
||||||
d->moods.resize(keyInt+1);
|
|
||||||
|
|
||||||
d->moods[keyInt] = mood;
|
|
||||||
}
|
|
||||||
|
|
||||||
void memory_info::setTrait(const string & key,
|
|
||||||
const string & value,
|
|
||||||
const string & zero,
|
|
||||||
const string & one,
|
|
||||||
const string & two,
|
|
||||||
const string & three,
|
|
||||||
const string & four,
|
|
||||||
const string & five)
|
|
||||||
{
|
|
||||||
uint32_t keyInt = strtol(key.c_str(), NULL, 10);
|
|
||||||
if(d->traits.size() <= keyInt)
|
|
||||||
{
|
|
||||||
d->traits.resize(keyInt+1);
|
|
||||||
}
|
|
||||||
d->traits[keyInt].push_back(zero);
|
|
||||||
d->traits[keyInt].push_back(one);
|
|
||||||
d->traits[keyInt].push_back(two);
|
|
||||||
d->traits[keyInt].push_back(three);
|
|
||||||
d->traits[keyInt].push_back(four);
|
|
||||||
d->traits[keyInt].push_back(five);
|
|
||||||
d->traits[keyInt].push_back(value);
|
|
||||||
}
|
|
||||||
|
|
||||||
// FIXME: next three methods should use some kind of custom container so it doesn't have to search so much.
|
|
||||||
t_class * memory_info::setClass (const char * name, uint32_t vtable, uint32_t typeoffset)
|
|
||||||
{
|
|
||||||
if(name == 0)
|
|
||||||
return 0;
|
|
||||||
for (uint32_t i=0; i<d->classes.size(); i++)
|
|
||||||
{
|
|
||||||
if(d->classes[i]->classname == name)
|
|
||||||
{
|
|
||||||
if(vtable != 0)
|
|
||||||
d->classes[i]->vtable = vtable;
|
|
||||||
if(typeoffset != 0)
|
|
||||||
d->classes[i]->type_offset = typeoffset;
|
|
||||||
return d->classes[i];
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
t_class *cls = new t_class();
|
|
||||||
// get an unique ID and add ourselves to the index
|
|
||||||
cls->assign = d->classindex;
|
|
||||||
cls->classname = name;
|
|
||||||
d->classnames.push_back(name);
|
|
||||||
|
|
||||||
// vtables no longer a requirement
|
|
||||||
cls->vtable = vtable;
|
|
||||||
|
|
||||||
// multi class yes/no
|
|
||||||
cls->type_offset = typeoffset;
|
|
||||||
|
|
||||||
d->classes.push_back(cls);
|
|
||||||
d->classindex++;
|
|
||||||
return cls;
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
void memory_info::setClassChild (t_class * parent, const char * name, const char * type)
|
|
||||||
{
|
|
||||||
vector <t_type *>& vec = parent->subs;
|
|
||||||
for (uint32_t i=0; i<vec.size(); i++)
|
|
||||||
{
|
|
||||||
if(vec[i]->classname == name)
|
|
||||||
{
|
|
||||||
vec[i]->type = strtol(type, NULL, 16);
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
// new multiclass child
|
|
||||||
t_type *mcc = new t_type(d->classindex,strtol(type, NULL, 16),name);
|
|
||||||
d->classnames.push_back(name);
|
|
||||||
vec.push_back(mcc);
|
|
||||||
d->classindex++;
|
|
||||||
|
|
||||||
//cout << " classtype " << name << ", assign " << mcc->assign << ", vtable " << mcc->type << endl;
|
|
||||||
}
|
|
||||||
|
|
||||||
// FIXME: stupid. we need a better container
|
|
||||||
bool memory_info::resolveObjectToClassID(const uint32_t address, int32_t & classid)
|
|
||||||
{
|
|
||||||
uint32_t vtable = d->p->readDWord(address);
|
|
||||||
// try to find the vtable in our cache
|
|
||||||
map<uint32_t, t_class *>::iterator it;
|
|
||||||
it = d->classIDs.find(vtable);
|
|
||||||
t_class * cl;
|
|
||||||
|
|
||||||
// class found in cache?
|
|
||||||
if(it != d->classIDs.end())
|
|
||||||
{
|
|
||||||
cl = (*it).second;
|
|
||||||
}
|
|
||||||
else// couldn't find?
|
|
||||||
{
|
|
||||||
// we set up the class for the first time
|
|
||||||
string classname = d->p->readClassName(vtable);
|
|
||||||
d->classIDs[vtable] = cl = setClass(classname.c_str(),vtable);
|
|
||||||
}
|
|
||||||
// and isn't a multi-class
|
|
||||||
if(!cl->type_offset)
|
|
||||||
{
|
|
||||||
// return
|
|
||||||
classid = cl->assign;
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
// and is a multiclass
|
|
||||||
else
|
|
||||||
{
|
|
||||||
// find the type
|
|
||||||
vector <t_type*>& vec = cl->subs;
|
|
||||||
uint32_t type = d->p->readWord(address + cl->type_offset);
|
|
||||||
// return typed building if successful
|
|
||||||
//FIXME: the vector should map directly to the type IDs here, so we don't have to mess with O(n) search
|
|
||||||
for (uint32_t k = 0; k < vec.size();k++)
|
|
||||||
{
|
|
||||||
if(vec[k]->type == type)
|
|
||||||
{
|
|
||||||
//cout << " multi " << address + classes[i].type_offset << " " << vec[k].classname << endl;
|
|
||||||
classid = vec[k]->assign;
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
// couldn't find the type... now what do we do here? throw?
|
|
||||||
// this is a case where it might be a good idea, because it uncovers some deeper problems
|
|
||||||
// we return the parent class instead, it doesn't change the API semantics and sort-of makes sense
|
|
||||||
classid = cl->assign;
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
//ALERT: doesn't care about multiclasses
|
|
||||||
bool memory_info::resolveClassnameToVPtr(const string classname, uint32_t & vptr)
|
|
||||||
{
|
|
||||||
// FIXME: another stupid search.
|
|
||||||
for(uint32_t i = 0;i< d->classes.size();i++)
|
|
||||||
{
|
|
||||||
//if(classes[i].)
|
|
||||||
if(d->classes[i]->classname == classname) // got class
|
|
||||||
{
|
|
||||||
vptr = d->classes[i]->vtable;
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
// we failed to find anything that would match
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
|
|
||||||
bool memory_info::resolveClassnameToClassID (const string classname, int32_t & classID)
|
|
||||||
{
|
|
||||||
// FIXME: another stupid search.
|
|
||||||
classID = -1;
|
|
||||||
for(uint32_t i = 0;i< d->classnames.size();i++)
|
|
||||||
{
|
|
||||||
if(d->classnames[i] == classname)
|
|
||||||
{
|
|
||||||
classID = i;
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
// we failed to find anything that would match
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
|
|
||||||
bool memory_info::resolveClassIDToClassname (const int32_t classID, string & classname)
|
|
||||||
{
|
|
||||||
if (classID >=0 && (uint32_t)classID < d->classnames.size())
|
|
||||||
{
|
|
||||||
classname = d->classnames[classID];
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
// return pointer to our internal classID -> className mapping
|
|
||||||
const vector<string> * memory_info::getClassIDMapping()
|
|
||||||
{
|
|
||||||
return &d->classnames;
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
// change base of all addresses
|
|
||||||
void memory_info::RebaseAddresses(const int32_t new_base)
|
|
||||||
{
|
|
||||||
map<string, uint32_t>::iterator iter;
|
|
||||||
int32_t rebase = - (int32_t)d->base + new_base;
|
|
||||||
for(iter = d->addresses.begin(); iter != d->addresses.end(); iter++)
|
|
||||||
{
|
|
||||||
d->addresses[iter->first] = iter->second + rebase;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
// change base of all addresses *and* vtable entries
|
|
||||||
void memory_info::RebaseAll(int32_t new_base)
|
|
||||||
{
|
|
||||||
map<string, uint32_t>::iterator iter;
|
|
||||||
int32_t rebase = - (int32_t)d->base + new_base;
|
|
||||||
for(iter = d->addresses.begin(); iter != d->addresses.end(); iter++)
|
|
||||||
{
|
|
||||||
d->addresses[iter->first] = iter->second + rebase;
|
|
||||||
}
|
|
||||||
RebaseVTable(rebase);
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
// change all vtable entries by offset
|
|
||||||
void memory_info::RebaseVTable(int32_t offset)
|
|
||||||
{
|
|
||||||
vector<t_class *>::iterator iter;
|
|
||||||
for(iter = d->classes.begin(); iter != d->classes.end(); iter++)
|
|
||||||
{
|
|
||||||
(*iter)->vtable += offset;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// Get named address
|
|
||||||
uint32_t memory_info::getAddress (const char *key)
|
|
||||||
{
|
|
||||||
map <string, uint32_t>::iterator iter = d->addresses.find(key);
|
|
||||||
|
|
||||||
if(iter != d->addresses.end())
|
|
||||||
{
|
|
||||||
return (*iter).second;
|
|
||||||
}
|
|
||||||
throw Error::MissingMemoryDefinition("address", key);
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
// Get named offset
|
|
||||||
int32_t memory_info::getOffset (const char *key)
|
|
||||||
{
|
|
||||||
map <string, int32_t>::iterator iter = d->offsets.find(key);
|
|
||||||
if(iter != d->offsets.end())
|
|
||||||
{
|
|
||||||
return (*iter).second;
|
|
||||||
}
|
|
||||||
throw Error::MissingMemoryDefinition("offset", key);
|
|
||||||
}
|
|
||||||
|
|
||||||
// Get named numerical value
|
|
||||||
uint32_t memory_info::getHexValue (const char *key)
|
|
||||||
{
|
|
||||||
map <string, uint32_t>::iterator iter = d->hexvals.find(key);
|
|
||||||
if(iter != d->hexvals.end())
|
|
||||||
{
|
|
||||||
return (*iter).second;
|
|
||||||
}
|
|
||||||
throw Error::MissingMemoryDefinition("hexvalue", key);
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
// Get named address
|
|
||||||
uint32_t memory_info::getAddress (const string &key)
|
|
||||||
{
|
|
||||||
map <string, uint32_t>::iterator iter = d->addresses.find(key);
|
|
||||||
|
|
||||||
if(iter != d->addresses.end())
|
|
||||||
{
|
|
||||||
return (*iter).second;
|
|
||||||
}
|
|
||||||
throw Error::MissingMemoryDefinition("address", key.c_str());
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
// Get named offset
|
|
||||||
int32_t memory_info::getOffset (const string &key)
|
|
||||||
{
|
|
||||||
map <string, int32_t>::iterator iter = d->offsets.find(key);
|
|
||||||
if(iter != d->offsets.end())
|
|
||||||
{
|
|
||||||
return (*iter).second;
|
|
||||||
}
|
|
||||||
throw Error::MissingMemoryDefinition("offset", key.c_str());
|
|
||||||
}
|
|
||||||
|
|
||||||
// Get named numerical value
|
|
||||||
uint32_t memory_info::getHexValue (const string &key)
|
|
||||||
{
|
|
||||||
map <string, uint32_t>::iterator iter = d->hexvals.find(key);
|
|
||||||
if(iter != d->hexvals.end())
|
|
||||||
{
|
|
||||||
return (*iter).second;
|
|
||||||
}
|
|
||||||
throw Error::MissingMemoryDefinition("hexvalue", key.c_str());
|
|
||||||
}
|
|
||||||
|
|
||||||
// Get named string
|
|
||||||
std::string memory_info::getString (const string &key)
|
|
||||||
{
|
|
||||||
map <string, string>::iterator iter = d->strings.find(key);
|
|
||||||
if(iter != d->strings.end())
|
|
||||||
{
|
|
||||||
return (*iter).second;
|
|
||||||
}
|
|
||||||
throw Error::MissingMemoryDefinition("string", key.c_str());
|
|
||||||
}
|
|
||||||
|
|
||||||
// Get Profession
|
|
||||||
string memory_info::getProfession (const uint32_t key) const
|
|
||||||
{
|
|
||||||
if(d->professions.size() > key)
|
|
||||||
{
|
|
||||||
return d->professions[key];
|
|
||||||
}
|
|
||||||
throw Error::MissingMemoryDefinition("profession", key);
|
|
||||||
}
|
|
||||||
|
|
||||||
// Get Job
|
|
||||||
string memory_info::getJob (const uint32_t key) const
|
|
||||||
{
|
|
||||||
if(d->jobs.size() > key)
|
|
||||||
|
|
||||||
{
|
|
||||||
return d->jobs[key];
|
|
||||||
}
|
|
||||||
throw Error::MissingMemoryDefinition("job", key);
|
|
||||||
}
|
|
||||||
|
|
||||||
string memory_info::getSkill (const uint32_t key) const
|
|
||||||
{
|
|
||||||
if(d->skills.size() > key)
|
|
||||||
{
|
|
||||||
return d->skills[key];
|
|
||||||
}
|
|
||||||
throw Error::MissingMemoryDefinition("skill", key);
|
|
||||||
}
|
|
||||||
|
|
||||||
DFHack::t_level memory_info::getLevelInfo(const uint32_t level) const
|
|
||||||
{
|
|
||||||
if(d->levels.size() > level)
|
|
||||||
{
|
|
||||||
return d->levels[level];
|
|
||||||
}
|
|
||||||
throw Error::MissingMemoryDefinition("Level", level);
|
|
||||||
}
|
|
||||||
|
|
||||||
// FIXME: ugly hack that needs to die
|
|
||||||
int absolute (int number)
|
|
||||||
{
|
|
||||||
if (number < 0)
|
|
||||||
return -number;
|
|
||||||
return number;
|
|
||||||
}
|
|
||||||
|
|
||||||
string memory_info::getTrait (const uint32_t traitIdx, const uint32_t traitValue) const
|
|
||||||
{
|
|
||||||
if(d->traits.size() > traitIdx)
|
|
||||||
{
|
|
||||||
int diff = absolute(traitValue-50);
|
|
||||||
if(diff < 10)
|
|
||||||
{
|
|
||||||
return string("");
|
|
||||||
}
|
|
||||||
if (traitValue >= 91)
|
|
||||||
return d->traits[traitIdx][5];
|
|
||||||
else if (traitValue >= 76)
|
|
||||||
return d->traits[traitIdx][4];
|
|
||||||
else if (traitValue >= 61)
|
|
||||||
return d->traits[traitIdx][3];
|
|
||||||
else if (traitValue >= 25)
|
|
||||||
return d->traits[traitIdx][2];
|
|
||||||
else if (traitValue >= 10)
|
|
||||||
return d->traits[traitIdx][1];
|
|
||||||
else
|
|
||||||
return d->traits[traitIdx][0];
|
|
||||||
}
|
|
||||||
throw Error::MissingMemoryDefinition("trait", traitIdx);
|
|
||||||
}
|
|
||||||
|
|
||||||
string memory_info::getTraitName(const uint32_t traitIdx) const
|
|
||||||
{
|
|
||||||
if(d->traits.size() > traitIdx)
|
|
||||||
{
|
|
||||||
return d->traits[traitIdx][d->traits[traitIdx].size()-1];
|
|
||||||
}
|
|
||||||
throw Error::MissingMemoryDefinition("traitname", traitIdx);
|
|
||||||
}
|
|
||||||
|
|
||||||
std::vector< std::vector<std::string> > const& memory_info::getAllTraits()
|
|
||||||
{
|
|
||||||
return d->traits;
|
|
||||||
}
|
|
||||||
|
|
||||||
string memory_info::getLabor (const uint32_t laborIdx)
|
|
||||||
{
|
|
||||||
if(d->labors.count(laborIdx))
|
|
||||||
{
|
|
||||||
return d->labors[laborIdx];
|
|
||||||
}
|
|
||||||
throw Error::MissingMemoryDefinition("labor", laborIdx);
|
|
||||||
}
|
|
||||||
|
|
||||||
std::string memory_info::getMood(const uint32_t moodID)
|
|
||||||
{
|
|
||||||
if(d->moods.size() > moodID)
|
|
||||||
{
|
|
||||||
return d->moods[moodID];
|
|
||||||
}
|
|
||||||
throw Error::MissingMemoryDefinition("Mood", moodID);
|
|
||||||
}
|
|
||||||
|
|
||||||
std::string memory_info::PrintOffsets()
|
|
||||||
{
|
|
||||||
ostringstream ss;
|
|
||||||
ss << "<Version name=\"" << getVersion() << "\">" << endl;
|
|
||||||
switch (getOS())
|
|
||||||
{
|
|
||||||
case OS_LINUX:
|
|
||||||
ss << "<MD5 value=\"" << getString("md5") << "\" />" << endl;
|
|
||||||
break;
|
|
||||||
case OS_WINDOWS:
|
|
||||||
ss << "<PETimeStamp value=\"" << hex << "0x" << getHexValue("pe_timestamp") << "\" />" << endl;
|
|
||||||
ss << "<MD5 value=\"" << getString("md5") << "\" />" << endl;
|
|
||||||
break;
|
|
||||||
default:
|
|
||||||
ss << " UNKNOWN" << endl;
|
|
||||||
}
|
|
||||||
ss << "<Offsets>" << endl;
|
|
||||||
map<string,uint32_t>::const_iterator iter;
|
|
||||||
for(iter = d->addresses.begin(); iter != d->addresses.end(); iter++)
|
|
||||||
{
|
|
||||||
ss << " <Address name=\"" << (*iter).first << "\" value=\"" << hex << "0x" << (*iter).second << "\" />" << endl;
|
|
||||||
}
|
|
||||||
map<string,int32_t>::const_iterator iter2;
|
|
||||||
for(iter2 = d->offsets.begin(); iter2 != d->offsets.end(); iter2++)
|
|
||||||
{
|
|
||||||
ss << " <Offset name=\"" << (*iter2).first << "\" value=\"" << hex << "0x" << (*iter2).second <<"\" />" << endl;
|
|
||||||
}
|
|
||||||
for(iter = d->hexvals.begin(); iter != d->hexvals.end(); iter++)
|
|
||||||
{
|
|
||||||
ss << " <HexValue name=\"" << (*iter).first << "\" value=\"" << hex << "0x" << (*iter).second <<"\" />" << endl;
|
|
||||||
}
|
|
||||||
map<string,string>::const_iterator iter3;
|
|
||||||
for(iter3 = d->strings.begin(); iter3 != d->strings.end(); iter3++)
|
|
||||||
{
|
|
||||||
ss << " <String name=\"" << (*iter3).first << "\" value=\"" << (*iter3).second <<"\" />" << endl;
|
|
||||||
}
|
|
||||||
ss << "</Offsets>" << endl;
|
|
||||||
ss << "</Version>" << endl;
|
|
||||||
return ss.str();
|
|
||||||
}
|
|
||||||
@ -1,297 +0,0 @@
|
|||||||
/*
|
|
||||||
www.sourceforge.net/projects/dfhack
|
|
||||||
Copyright (c) 2009 Petr Mrázek (peterix), Kenneth Ferland (Impaler[WrG]), dorf
|
|
||||||
|
|
||||||
This software is provided 'as-is', without any express or implied
|
|
||||||
warranty. In no event will the authors be held liable for any
|
|
||||||
damages arising from the use of this software.
|
|
||||||
|
|
||||||
Permission is granted to anyone to use this software for any
|
|
||||||
purpose, including commercial applications, and to alter it and
|
|
||||||
redistribute it freely, subject to the following restrictions:
|
|
||||||
|
|
||||||
1. The origin of this software must not be misrepresented; you must
|
|
||||||
not claim that you wrote the original software. If you use this
|
|
||||||
software in a product, an acknowledgment in the product documentation
|
|
||||||
would be appreciated but is not required.
|
|
||||||
|
|
||||||
2. Altered source versions must be plainly marked as such, and
|
|
||||||
must not be misrepresented as being the original software.
|
|
||||||
|
|
||||||
3. This notice may not be removed or altered from any source
|
|
||||||
distribution.
|
|
||||||
*/
|
|
||||||
|
|
||||||
#include "Internal.h"
|
|
||||||
#include "DFMemInfoManager.h"
|
|
||||||
|
|
||||||
#include "dfhack/DFMemInfo.h"
|
|
||||||
#include "dfhack/DFError.h"
|
|
||||||
|
|
||||||
using namespace DFHack;
|
|
||||||
|
|
||||||
MemInfoManager::~MemInfoManager()
|
|
||||||
{
|
|
||||||
// for each in std::vector<memory_info*> meminfo;, delete
|
|
||||||
for(uint32_t i = 0; i < meminfo.size();i++)
|
|
||||||
{
|
|
||||||
delete meminfo[i];
|
|
||||||
}
|
|
||||||
meminfo.clear();
|
|
||||||
}
|
|
||||||
|
|
||||||
void MemInfoManager::ParseVTable(TiXmlElement* vtable, memory_info* mem)
|
|
||||||
{
|
|
||||||
TiXmlElement* pClassEntry;
|
|
||||||
TiXmlElement* pClassSubEntry;
|
|
||||||
// check for rebase, do rebase if check positive
|
|
||||||
const char * rebase = vtable->Attribute("rebase");
|
|
||||||
if(rebase)
|
|
||||||
{
|
|
||||||
int32_t rebase_offset = strtol(rebase, NULL, 16);
|
|
||||||
mem->RebaseVTable(rebase_offset);
|
|
||||||
}
|
|
||||||
// parse vtable entries
|
|
||||||
pClassEntry = vtable->FirstChildElement();
|
|
||||||
for(;pClassEntry;pClassEntry=pClassEntry->NextSiblingElement())
|
|
||||||
{
|
|
||||||
string type = pClassEntry->Value();
|
|
||||||
const char *cstr_name = pClassEntry->Attribute("name");
|
|
||||||
const char *cstr_vtable = pClassEntry->Attribute("vtable");
|
|
||||||
uint32_t vtable = 0;
|
|
||||||
if(cstr_vtable)
|
|
||||||
vtable = strtol(cstr_vtable, NULL, 16);
|
|
||||||
// it's a simple class
|
|
||||||
if(type== "class")
|
|
||||||
{
|
|
||||||
mem->setClass(cstr_name, vtable);
|
|
||||||
}
|
|
||||||
// it's a multi-type class
|
|
||||||
else if (type == "multiclass")
|
|
||||||
{
|
|
||||||
// get offset of the type variable
|
|
||||||
const char *cstr_typeoffset = pClassEntry->Attribute("typeoffset");
|
|
||||||
uint32_t typeoffset = 0;
|
|
||||||
if(cstr_typeoffset)
|
|
||||||
typeoffset = strtol(cstr_typeoffset, NULL, 16);
|
|
||||||
t_class * mclass = mem->setClass(cstr_name, vtable, typeoffset);
|
|
||||||
// parse class sub-entries
|
|
||||||
pClassSubEntry = pClassEntry->FirstChildElement();
|
|
||||||
for(;pClassSubEntry;pClassSubEntry=pClassSubEntry->NextSiblingElement())
|
|
||||||
{
|
|
||||||
type = pClassSubEntry->Value();
|
|
||||||
if(type== "class")
|
|
||||||
{
|
|
||||||
// type is a value loaded from type offset
|
|
||||||
cstr_name = pClassSubEntry->Attribute("name");
|
|
||||||
const char *cstr_value = pClassSubEntry->Attribute("type");
|
|
||||||
mem->setClassChild(mclass,cstr_name,cstr_value);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
void MemInfoManager::ParseEntry (TiXmlElement* entry, memory_info* mem, map <string ,TiXmlElement *>& knownEntries)
|
|
||||||
{
|
|
||||||
TiXmlElement* pMemEntry;
|
|
||||||
const char *cstr_version = entry->Attribute("version");
|
|
||||||
const char *cstr_os = entry->Attribute("os");
|
|
||||||
const char *cstr_base = entry->Attribute("base");
|
|
||||||
const char *cstr_rebase = entry->Attribute("rebase");
|
|
||||||
if(cstr_base)
|
|
||||||
{
|
|
||||||
string base = cstr_base;
|
|
||||||
ParseEntry(knownEntries[base], mem, knownEntries);
|
|
||||||
}
|
|
||||||
|
|
||||||
if (!cstr_version)
|
|
||||||
throw Error::MemoryXmlBadAttribute("version");
|
|
||||||
if (!cstr_os)
|
|
||||||
throw Error::MemoryXmlBadAttribute("os");
|
|
||||||
|
|
||||||
string os = cstr_os;
|
|
||||||
mem->setVersion(cstr_version);
|
|
||||||
mem->setOS(cstr_os);
|
|
||||||
|
|
||||||
// offset inherited addresses by 'rebase'.
|
|
||||||
int32_t rebase = 0;
|
|
||||||
if(cstr_rebase)
|
|
||||||
{
|
|
||||||
rebase = mem->getBase() + strtol(cstr_rebase, NULL, 16);
|
|
||||||
mem->RebaseAddresses(rebase);
|
|
||||||
}
|
|
||||||
|
|
||||||
//set base to default, we're overwriting this because the previous rebase could cause havoc on Vista/7
|
|
||||||
if(os == "windows")
|
|
||||||
{
|
|
||||||
// set default image base. this is fixed for base relocation later
|
|
||||||
mem->setBase(0x400000);
|
|
||||||
}
|
|
||||||
else if(os == "linux")
|
|
||||||
{
|
|
||||||
// this is wrong... I'm not going to do base image relocation on linux though.
|
|
||||||
// users are free to use a sane kernel that doesn't do this kind of **** by default
|
|
||||||
mem->setBase(0x0);
|
|
||||||
}
|
|
||||||
else if ( os == "all")
|
|
||||||
{
|
|
||||||
// yay
|
|
||||||
}
|
|
||||||
else
|
|
||||||
{
|
|
||||||
throw Error::MemoryXmlBadAttribute("os");
|
|
||||||
}
|
|
||||||
|
|
||||||
// process additional entries
|
|
||||||
//cout << "Entry " << cstr_version << " " << cstr_os << endl;
|
|
||||||
pMemEntry = entry->FirstChildElement()->ToElement();
|
|
||||||
for(;pMemEntry;pMemEntry=pMemEntry->NextSiblingElement())
|
|
||||||
{
|
|
||||||
// only elements get processed
|
|
||||||
const char *cstr_type = pMemEntry->Value();
|
|
||||||
const char *cstr_name = pMemEntry->Attribute("name");
|
|
||||||
const char *cstr_value = pMemEntry->GetText();
|
|
||||||
|
|
||||||
if(!cstr_value)
|
|
||||||
cstr_value = pMemEntry->Attribute("id");
|
|
||||||
|
|
||||||
// check for missing parts
|
|
||||||
string type, name, value;
|
|
||||||
type = cstr_type;
|
|
||||||
if(type == "VTable")
|
|
||||||
{
|
|
||||||
ParseVTable(pMemEntry, mem);
|
|
||||||
continue;
|
|
||||||
}
|
|
||||||
if(!(cstr_name && cstr_value))
|
|
||||||
{
|
|
||||||
throw Error::MemoryXmlUnderspecifiedEntry(cstr_version);
|
|
||||||
}
|
|
||||||
name = cstr_name;
|
|
||||||
value = cstr_value;
|
|
||||||
if (type == "HexValue")
|
|
||||||
{
|
|
||||||
mem->setHexValue(name, value);
|
|
||||||
}
|
|
||||||
else if (type == "Address")
|
|
||||||
{
|
|
||||||
mem->setAddress(name, value);
|
|
||||||
}
|
|
||||||
else if (type == "Offset")
|
|
||||||
{
|
|
||||||
mem->setOffset(name, value);
|
|
||||||
}
|
|
||||||
else if (type == "String")
|
|
||||||
{
|
|
||||||
mem->setString(name, value);
|
|
||||||
}
|
|
||||||
else if (type == "Profession")
|
|
||||||
{
|
|
||||||
mem->setProfession(value,name);
|
|
||||||
}
|
|
||||||
else if (type == "Job")
|
|
||||||
{
|
|
||||||
mem->setJob(value,name);
|
|
||||||
}
|
|
||||||
else if (type == "Skill")
|
|
||||||
{
|
|
||||||
mem->setSkill(value,name);
|
|
||||||
}
|
|
||||||
else if (type == "Trait")
|
|
||||||
{
|
|
||||||
mem->setTrait(value, name,pMemEntry->Attribute("level_0"),pMemEntry->Attribute("level_1"),pMemEntry->Attribute("level_2"),pMemEntry->Attribute("level_3"),pMemEntry->Attribute("level_4"),pMemEntry->Attribute("level_5"));
|
|
||||||
}
|
|
||||||
else if (type == "Labor")
|
|
||||||
{
|
|
||||||
mem->setLabor(value,name);
|
|
||||||
}
|
|
||||||
else if (type == "Level")
|
|
||||||
{
|
|
||||||
mem->setLevel(value, name, pMemEntry->Attribute("xpNxtLvl"));
|
|
||||||
}
|
|
||||||
else if (type == "Mood")
|
|
||||||
{
|
|
||||||
mem->setMood(value, name);
|
|
||||||
}
|
|
||||||
else
|
|
||||||
{
|
|
||||||
throw Error::MemoryXmlUnknownType(type.c_str());
|
|
||||||
}
|
|
||||||
} // for
|
|
||||||
} // method
|
|
||||||
|
|
||||||
MemInfoManager::MemInfoManager(string path_to_xml)
|
|
||||||
{
|
|
||||||
error = false;
|
|
||||||
loadFile(path_to_xml);
|
|
||||||
}
|
|
||||||
|
|
||||||
// load the XML file with offsets
|
|
||||||
bool MemInfoManager::loadFile(string path_to_xml)
|
|
||||||
{
|
|
||||||
TiXmlDocument doc( path_to_xml.c_str() );
|
|
||||||
//bool loadOkay = doc.LoadFile();
|
|
||||||
if (!doc.LoadFile())
|
|
||||||
{
|
|
||||||
error = true;
|
|
||||||
throw Error::MemoryXmlParse(doc.ErrorDesc(), doc.ErrorId(), doc.ErrorRow(), doc.ErrorCol());
|
|
||||||
}
|
|
||||||
TiXmlHandle hDoc(&doc);
|
|
||||||
TiXmlElement* pElem;
|
|
||||||
TiXmlHandle hRoot(0);
|
|
||||||
memory_info mem;
|
|
||||||
|
|
||||||
// block: name
|
|
||||||
{
|
|
||||||
pElem=hDoc.FirstChildElement().Element();
|
|
||||||
// should always have a valid root but handle gracefully if it does
|
|
||||||
if (!pElem)
|
|
||||||
{
|
|
||||||
error = true;
|
|
||||||
throw Error::MemoryXmlNoRoot();
|
|
||||||
}
|
|
||||||
string m_name=pElem->Value();
|
|
||||||
if(m_name != "DFExtractor")
|
|
||||||
{
|
|
||||||
error = true;
|
|
||||||
throw Error::MemoryXmlNoDFExtractor(m_name.c_str());
|
|
||||||
}
|
|
||||||
// save this for later
|
|
||||||
hRoot=TiXmlHandle(pElem);
|
|
||||||
}
|
|
||||||
// transform elements
|
|
||||||
{
|
|
||||||
// trash existing list
|
|
||||||
for(uint32_t i = 0; i < meminfo.size(); i++)
|
|
||||||
{
|
|
||||||
delete meminfo[i];
|
|
||||||
}
|
|
||||||
meminfo.clear();
|
|
||||||
TiXmlElement* pMemInfo=hRoot.FirstChild( "MemoryDescriptors" ).FirstChild( "Entry" ).Element();
|
|
||||||
map <string ,TiXmlElement *> map_pNamedEntries;
|
|
||||||
vector <TiXmlElement *> v_pEntries;
|
|
||||||
for( ; pMemInfo; pMemInfo=pMemInfo->NextSiblingElement("Entry"))
|
|
||||||
{
|
|
||||||
v_pEntries.push_back(pMemInfo);
|
|
||||||
const char *id = pMemInfo->Attribute("id");
|
|
||||||
if(id)
|
|
||||||
{
|
|
||||||
string str_id = id;
|
|
||||||
map_pNamedEntries[str_id] = pMemInfo;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
for(uint32_t i = 0; i< v_pEntries.size();i++)
|
|
||||||
{
|
|
||||||
memory_info *mem = new memory_info();
|
|
||||||
//FIXME: add a set of entries processed in a step of this cycle, use it to check for infinite loops
|
|
||||||
/* recursive */ParseEntry( v_pEntries[i] , mem , map_pNamedEntries);
|
|
||||||
meminfo.push_back(mem);
|
|
||||||
}
|
|
||||||
// process found things here
|
|
||||||
}
|
|
||||||
error = false;
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
File diff suppressed because it is too large
Load Diff
@ -0,0 +1,630 @@
|
|||||||
|
/*
|
||||||
|
www.sourceforge.net/projects/dfhack
|
||||||
|
Copyright (c) 2009 Petr Mrázek (peterix), Kenneth Ferland (Impaler[WrG]), dorf
|
||||||
|
|
||||||
|
This software is provided 'as-is', without any express or implied
|
||||||
|
warranty. In no event will the authors be held liable for any
|
||||||
|
damages arising from the use of this software.
|
||||||
|
|
||||||
|
Permission is granted to anyone to use this software for any
|
||||||
|
purpose, including commercial applications, and to alter it and
|
||||||
|
redistribute it freely, subject to the following restrictions:
|
||||||
|
|
||||||
|
1. The origin of this software must not be misrepresented; you must
|
||||||
|
not claim that you wrote the original software. If you use this
|
||||||
|
software in a product, an acknowledgment in the product documentation
|
||||||
|
would be appreciated but is not required.
|
||||||
|
|
||||||
|
2. Altered source versions must be plainly marked as such, and
|
||||||
|
must not be misrepresented as being the original software.
|
||||||
|
|
||||||
|
3. This notice may not be removed or altered from any source
|
||||||
|
distribution.
|
||||||
|
*/
|
||||||
|
|
||||||
|
#include "Internal.h"
|
||||||
|
|
||||||
|
#include "dfhack/VersionInfoFactory.h"
|
||||||
|
#include "dfhack/VersionInfo.h"
|
||||||
|
#include "dfhack/DFError.h"
|
||||||
|
#include <algorithm>
|
||||||
|
|
||||||
|
using namespace DFHack;
|
||||||
|
|
||||||
|
VersionInfoFactory::~VersionInfoFactory()
|
||||||
|
{
|
||||||
|
// for each stored version, delete
|
||||||
|
for(uint32_t i = 0; i < versions.size();i++)
|
||||||
|
{
|
||||||
|
delete versions[i];
|
||||||
|
}
|
||||||
|
versions.clear();
|
||||||
|
}
|
||||||
|
|
||||||
|
void VersionInfoFactory::ParseVTable(TiXmlElement* vtable, VersionInfo* mem)
|
||||||
|
{
|
||||||
|
TiXmlElement* pClassEntry;
|
||||||
|
TiXmlElement* pClassSubEntry;
|
||||||
|
/*
|
||||||
|
// check for rebase, do rebase if check positive
|
||||||
|
const char * rebase = vtable->Attribute("rebase");
|
||||||
|
if(rebase)
|
||||||
|
{
|
||||||
|
int32_t rebase_offset = strtol(rebase, NULL, 16);
|
||||||
|
mem->RebaseVTable(rebase_offset);
|
||||||
|
}
|
||||||
|
*/
|
||||||
|
// parse vtable entries
|
||||||
|
pClassEntry = vtable->FirstChildElement();
|
||||||
|
for(;pClassEntry;pClassEntry=pClassEntry->NextSiblingElement())
|
||||||
|
{
|
||||||
|
string type = pClassEntry->Value();
|
||||||
|
const char *cstr_name = pClassEntry->Attribute("name");
|
||||||
|
const char *cstr_vtable = pClassEntry->Attribute("vtable");
|
||||||
|
uint32_t vtable = 0;
|
||||||
|
if(cstr_vtable)
|
||||||
|
vtable = strtol(cstr_vtable, NULL, 16);
|
||||||
|
// it's a simple class
|
||||||
|
if(type== "class")
|
||||||
|
{
|
||||||
|
mem->setClass(cstr_name, vtable);
|
||||||
|
}
|
||||||
|
// it's a multi-type class
|
||||||
|
else if (type == "multiclass")
|
||||||
|
{
|
||||||
|
// get offset of the type variable
|
||||||
|
const char *cstr_typeoffset = pClassEntry->Attribute("typeoffset");
|
||||||
|
uint32_t typeoffset = 0;
|
||||||
|
if(cstr_typeoffset)
|
||||||
|
typeoffset = strtol(cstr_typeoffset, NULL, 16);
|
||||||
|
t_class * mclass = mem->setClass(cstr_name, vtable, typeoffset);
|
||||||
|
// parse class sub-entries
|
||||||
|
pClassSubEntry = pClassEntry->FirstChildElement();
|
||||||
|
for(;pClassSubEntry;pClassSubEntry=pClassSubEntry->NextSiblingElement())
|
||||||
|
{
|
||||||
|
type = pClassSubEntry->Value();
|
||||||
|
if(type== "class")
|
||||||
|
{
|
||||||
|
// type is a value loaded from type offset
|
||||||
|
cstr_name = pClassSubEntry->Attribute("name");
|
||||||
|
const char *cstr_value = pClassSubEntry->Attribute("type");
|
||||||
|
mem->setClassChild(mclass,cstr_name,cstr_value);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// FIXME: this is ripe for replacement with a more generic approach
|
||||||
|
void VersionInfoFactory::ParseOffsets(TiXmlElement * parent, VersionInfo* target, bool initial)
|
||||||
|
{
|
||||||
|
// we parse the groups iteratively instead of recursively
|
||||||
|
// breadcrubs acts like a makeshift stack
|
||||||
|
// first pair entry stores the current element of that level
|
||||||
|
// second pair entry the group object from OffsetGroup
|
||||||
|
typedef pair < TiXmlElement *, OffsetGroup * > groupPair;
|
||||||
|
vector< groupPair > breadcrumbs;
|
||||||
|
{
|
||||||
|
TiXmlElement* pEntry;
|
||||||
|
// we get the <Offsets>, look at the children
|
||||||
|
pEntry = parent->FirstChildElement();
|
||||||
|
if(!pEntry)
|
||||||
|
return;
|
||||||
|
|
||||||
|
OffsetGroup * currentGroup = reinterpret_cast<OffsetGroup *> (target);
|
||||||
|
breadcrumbs.push_back(groupPair(pEntry,currentGroup));
|
||||||
|
}
|
||||||
|
|
||||||
|
// work variables
|
||||||
|
OffsetGroup * currentGroup = 0;
|
||||||
|
TiXmlElement * currentElem = 0;
|
||||||
|
//cerr << "<Offsets>"<< endl;
|
||||||
|
while(1)
|
||||||
|
{
|
||||||
|
// get current work variables
|
||||||
|
currentElem = breadcrumbs.back().first;
|
||||||
|
currentGroup = breadcrumbs.back().second;
|
||||||
|
|
||||||
|
// we reached the end of the current group?
|
||||||
|
if(!currentElem)
|
||||||
|
{
|
||||||
|
// go one level up
|
||||||
|
breadcrumbs.pop_back();
|
||||||
|
// exit if no more work
|
||||||
|
if(breadcrumbs.empty())
|
||||||
|
{
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
//cerr << "</group>" << endl;
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if(!currentGroup)
|
||||||
|
{
|
||||||
|
groupPair & gp = breadcrumbs.back();
|
||||||
|
gp.first = gp.first->NextSiblingElement();
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
|
||||||
|
// skip non-elements
|
||||||
|
if (currentElem->Type() != TiXmlNode::ELEMENT)
|
||||||
|
{
|
||||||
|
groupPair & gp = breadcrumbs.back();
|
||||||
|
gp.first = gp.first->NextSiblingElement();
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
|
||||||
|
// we have a valid current element and current group
|
||||||
|
// get properties
|
||||||
|
string type = currentElem->Value();
|
||||||
|
std::transform(type.begin(), type.end(), type.begin(), ::tolower);
|
||||||
|
const char *cstr_name = currentElem->Attribute("name");
|
||||||
|
if(!cstr_name)
|
||||||
|
{
|
||||||
|
// ERROR, missing attribute
|
||||||
|
}
|
||||||
|
|
||||||
|
// evaluate elements
|
||||||
|
const char *cstr_value = currentElem->Attribute("value");
|
||||||
|
if(type == "group")
|
||||||
|
{
|
||||||
|
// FIXME: possibly use setGroup always, with the initial flag as parameter?
|
||||||
|
// create or get group
|
||||||
|
OffsetGroup * og;
|
||||||
|
if(initial)
|
||||||
|
og = currentGroup->createGroup(cstr_name);
|
||||||
|
else
|
||||||
|
og = currentGroup->getGroup(cstr_name);
|
||||||
|
//cerr << "<group name=\"" << cstr_name << "\">" << endl;
|
||||||
|
// advance this level to the next element
|
||||||
|
groupPair & gp = breadcrumbs.back();
|
||||||
|
gp.first = currentElem->NextSiblingElement();
|
||||||
|
|
||||||
|
if(!og)
|
||||||
|
{
|
||||||
|
string fullname = currentGroup->getFullName() + cstr_name;
|
||||||
|
throw Error::MissingMemoryDefinition("group", fullname);
|
||||||
|
}
|
||||||
|
|
||||||
|
// add a new level that will be processed next
|
||||||
|
breadcrumbs.push_back(groupPair(currentElem->FirstChildElement(), og));
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
else if(type == "address")
|
||||||
|
{
|
||||||
|
if(initial)
|
||||||
|
{
|
||||||
|
currentGroup->createAddress(cstr_name);
|
||||||
|
}
|
||||||
|
else if(cstr_value)
|
||||||
|
{
|
||||||
|
currentGroup->setAddress(cstr_name, cstr_value);
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
// ERROR, missing attribute
|
||||||
|
}
|
||||||
|
}
|
||||||
|
else if(type == "offset")
|
||||||
|
{
|
||||||
|
if(initial)
|
||||||
|
{
|
||||||
|
currentGroup->createOffset(cstr_name);
|
||||||
|
}
|
||||||
|
else if(cstr_value)
|
||||||
|
{
|
||||||
|
currentGroup->setOffset(cstr_name, cstr_value);
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
// ERROR, missing attribute
|
||||||
|
}
|
||||||
|
}
|
||||||
|
else if(type == "string")
|
||||||
|
{
|
||||||
|
if(initial)
|
||||||
|
{
|
||||||
|
currentGroup->createString(cstr_name);
|
||||||
|
}
|
||||||
|
else if(cstr_value)
|
||||||
|
{
|
||||||
|
currentGroup->setString(cstr_name, cstr_value);
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
// ERROR, missing attribute
|
||||||
|
}
|
||||||
|
}
|
||||||
|
else if(type == "hexvalue")
|
||||||
|
{
|
||||||
|
if(initial)
|
||||||
|
{
|
||||||
|
currentGroup->createHexValue(cstr_name);
|
||||||
|
}
|
||||||
|
else if(cstr_value)
|
||||||
|
{
|
||||||
|
currentGroup->setHexValue(cstr_name, cstr_value);
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
// ERROR, missing attribute
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// advance to next element
|
||||||
|
groupPair & gp = breadcrumbs.back();
|
||||||
|
gp.first = currentElem->NextSiblingElement();
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
//cerr << "</Offsets>"<< endl;
|
||||||
|
}
|
||||||
|
|
||||||
|
void VersionInfoFactory::ParseBase (TiXmlElement* entry, VersionInfo* mem)
|
||||||
|
{
|
||||||
|
TiXmlElement* pElement;
|
||||||
|
TiXmlElement* pElement2nd;
|
||||||
|
const char *cstr_version = entry->Attribute("name");
|
||||||
|
|
||||||
|
if (!cstr_version)
|
||||||
|
throw Error::MemoryXmlBadAttribute("name");
|
||||||
|
|
||||||
|
mem->setVersion(cstr_version);
|
||||||
|
mem->setOS(VersionInfo::OS_BAD);
|
||||||
|
|
||||||
|
// process additional entries
|
||||||
|
pElement = entry->FirstChildElement()->ToElement();
|
||||||
|
for(;pElement;pElement=pElement->NextSiblingElement())
|
||||||
|
{
|
||||||
|
// only elements get processed
|
||||||
|
const char *cstr_type = pElement->Value();
|
||||||
|
std::string type = cstr_type;
|
||||||
|
if(type == "VTable")
|
||||||
|
{
|
||||||
|
ParseVTable(pElement, mem);
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
else if(type == "Offsets")
|
||||||
|
{
|
||||||
|
// we don't care about the descriptions here, do nothing
|
||||||
|
ParseOffsets(pElement, mem, true);
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
else if (type == "Professions")
|
||||||
|
{
|
||||||
|
pElement2nd = pElement->FirstChildElement("Profession");
|
||||||
|
for(;pElement2nd;pElement2nd=pElement2nd->NextSiblingElement("Profession"))
|
||||||
|
{
|
||||||
|
const char * id = pElement2nd->Attribute("id");
|
||||||
|
const char * name = pElement2nd->Attribute("name");
|
||||||
|
// FIXME: missing some attributes here
|
||||||
|
if(id && name)
|
||||||
|
{
|
||||||
|
mem->setProfession(id,name);
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
// FIXME: this is crap, doesn't tell anything about the error
|
||||||
|
throw Error::MemoryXmlUnderspecifiedEntry(name);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
else if (type == "Jobs")
|
||||||
|
{
|
||||||
|
pElement2nd = pElement->FirstChildElement("Job");
|
||||||
|
for(;pElement2nd;pElement2nd=pElement2nd->NextSiblingElement("Job"))
|
||||||
|
{
|
||||||
|
const char * id = pElement2nd->Attribute("id");
|
||||||
|
const char * name = pElement2nd->Attribute("name");
|
||||||
|
if(id && name)
|
||||||
|
{
|
||||||
|
mem->setJob(id,name);
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
// FIXME: this is crap, doesn't tell anything about the error
|
||||||
|
throw Error::MemoryXmlUnderspecifiedEntry(name);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
else if (type == "Skills")
|
||||||
|
{
|
||||||
|
pElement2nd = pElement->FirstChildElement("Skill");
|
||||||
|
for(;pElement2nd;pElement2nd=pElement2nd->NextSiblingElement("Skill"))
|
||||||
|
{
|
||||||
|
const char * id = pElement2nd->Attribute("id");
|
||||||
|
const char * name = pElement2nd->Attribute("name");
|
||||||
|
if(id && name)
|
||||||
|
{
|
||||||
|
mem->setSkill(id,name);
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
// FIXME: this is crap, doesn't tell anything about the error
|
||||||
|
throw Error::MemoryXmlUnderspecifiedEntry(name);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
else if (type == "Traits")
|
||||||
|
{
|
||||||
|
pElement2nd = pElement->FirstChildElement("Trait");
|
||||||
|
for(;pElement2nd;pElement2nd=pElement2nd->NextSiblingElement("Trait"))
|
||||||
|
{
|
||||||
|
const char * id = pElement2nd->Attribute("id");
|
||||||
|
const char * name = pElement2nd->Attribute("name");
|
||||||
|
const char * lvl0 = pElement2nd->Attribute("level_0");
|
||||||
|
const char * lvl1 = pElement2nd->Attribute("level_1");
|
||||||
|
const char * lvl2 = pElement2nd->Attribute("level_2");
|
||||||
|
const char * lvl3 = pElement2nd->Attribute("level_3");
|
||||||
|
const char * lvl4 = pElement2nd->Attribute("level_4");
|
||||||
|
const char * lvl5 = pElement2nd->Attribute("level_5");
|
||||||
|
if(id && name && lvl0 && lvl1 && lvl2 && lvl3 && lvl4 && lvl5)
|
||||||
|
{
|
||||||
|
mem->setTrait(id, name, lvl0, lvl1, lvl2, lvl3, lvl4, lvl5);
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
// FIXME: this is crap, doesn't tell anything about the error
|
||||||
|
throw Error::MemoryXmlUnderspecifiedEntry(name);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
else if (type == "Labors")
|
||||||
|
{
|
||||||
|
pElement2nd = pElement->FirstChildElement("Labor");
|
||||||
|
for(;pElement2nd;pElement2nd=pElement2nd->NextSiblingElement("Labor"))
|
||||||
|
{
|
||||||
|
const char * id = pElement2nd->Attribute("id");
|
||||||
|
const char * name = pElement2nd->Attribute("name");
|
||||||
|
if(id && name)
|
||||||
|
{
|
||||||
|
mem->setLabor(id,name);
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
// FIXME: this is crap, doesn't tell anything about the error
|
||||||
|
throw Error::MemoryXmlUnderspecifiedEntry(name);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
else if (type == "Levels")
|
||||||
|
{
|
||||||
|
pElement2nd = pElement->FirstChildElement("Level");
|
||||||
|
for(;pElement2nd;pElement2nd=pElement2nd->NextSiblingElement("Level"))
|
||||||
|
{
|
||||||
|
const char * id = pElement2nd->Attribute("id");
|
||||||
|
const char * name = pElement2nd->Attribute("name");
|
||||||
|
const char * nextlvl = pElement2nd->Attribute("xpNxtLvl");
|
||||||
|
if(id && name && nextlvl)
|
||||||
|
{
|
||||||
|
mem->setLevel(id, name, nextlvl);
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
// FIXME: this is crap, doesn't tell anything about the error
|
||||||
|
throw Error::MemoryXmlUnderspecifiedEntry(name);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
else if (type == "Moods")
|
||||||
|
{
|
||||||
|
pElement2nd = pElement->FirstChildElement("Mood");
|
||||||
|
for(;pElement2nd;pElement2nd=pElement2nd->NextSiblingElement("Mood"))
|
||||||
|
{
|
||||||
|
const char * id = pElement2nd->Attribute("id");
|
||||||
|
const char * name = pElement2nd->Attribute("name");
|
||||||
|
if(id && name)
|
||||||
|
{
|
||||||
|
mem->setMood(id, name);
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
// FIXME: this is crap, doesn't tell anything about the error
|
||||||
|
throw Error::MemoryXmlUnderspecifiedEntry(name);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
//FIXME: only log, not hard error
|
||||||
|
//throw Error::MemoryXmlUnknownType(type.c_str());
|
||||||
|
}
|
||||||
|
} // for
|
||||||
|
} // method
|
||||||
|
|
||||||
|
void VersionInfoFactory::EvalVersion(string base, VersionInfo * mem)
|
||||||
|
{
|
||||||
|
if(knownVersions.find(base) != knownVersions.end())
|
||||||
|
{
|
||||||
|
v_descr & desc = knownVersions[base];
|
||||||
|
if (!desc.second)
|
||||||
|
{
|
||||||
|
VersionInfo * newmem = new VersionInfo();
|
||||||
|
ParseVersion(desc.first, newmem);
|
||||||
|
desc.second = newmem;
|
||||||
|
}
|
||||||
|
mem->copy(desc.second);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
void VersionInfoFactory::ParseVersion (TiXmlElement* entry, VersionInfo* mem)
|
||||||
|
{
|
||||||
|
TiXmlElement* pMemEntry;
|
||||||
|
const char *cstr_name = entry->Attribute("name");
|
||||||
|
const char *cstr_os = entry->Attribute("os");
|
||||||
|
const char *cstr_base = entry->Attribute("base");
|
||||||
|
const char *cstr_rebase = entry->Attribute("rebase");
|
||||||
|
if(cstr_base)
|
||||||
|
{
|
||||||
|
string base = cstr_base;
|
||||||
|
EvalVersion(base, mem);
|
||||||
|
}
|
||||||
|
|
||||||
|
if (!cstr_name)
|
||||||
|
throw Error::MemoryXmlBadAttribute("name");
|
||||||
|
if (!cstr_os)
|
||||||
|
throw Error::MemoryXmlBadAttribute("os");
|
||||||
|
|
||||||
|
string os = cstr_os;
|
||||||
|
mem->setVersion(cstr_name);
|
||||||
|
mem->setOS(cstr_os);
|
||||||
|
|
||||||
|
// offset inherited addresses by 'rebase'.
|
||||||
|
int32_t rebase = 0;
|
||||||
|
if(cstr_rebase)
|
||||||
|
{
|
||||||
|
rebase = mem->getBase() + strtol(cstr_rebase, NULL, 16);
|
||||||
|
mem->RebaseAddresses(rebase);
|
||||||
|
}
|
||||||
|
|
||||||
|
//set base to default, we're overwriting this because the previous rebase could cause havoc on Vista/7
|
||||||
|
if(os == "windows")
|
||||||
|
{
|
||||||
|
// set default image base. this is fixed for base relocation later
|
||||||
|
mem->setBase(0x400000);
|
||||||
|
}
|
||||||
|
else if(os == "linux")
|
||||||
|
{
|
||||||
|
// this is wrong... I'm not going to do base image relocation on linux though.
|
||||||
|
// users are free to use a sane kernel that doesn't do this kind of **** by default
|
||||||
|
mem->setBase(0x0);
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
throw Error::MemoryXmlBadAttribute("os");
|
||||||
|
}
|
||||||
|
|
||||||
|
// process additional entries
|
||||||
|
//cout << "Entry " << cstr_version << " " << cstr_os << endl;
|
||||||
|
pMemEntry = entry->FirstChildElement()->ToElement();
|
||||||
|
for(;pMemEntry;pMemEntry=pMemEntry->NextSiblingElement())
|
||||||
|
{
|
||||||
|
string type, name, value;
|
||||||
|
const char *cstr_type = pMemEntry->Value();
|
||||||
|
type = cstr_type;
|
||||||
|
// check for missing parts
|
||||||
|
if(type == "VTable")
|
||||||
|
{
|
||||||
|
ParseVTable(pMemEntry, mem);
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
else if(type == "Offsets")
|
||||||
|
{
|
||||||
|
ParseOffsets(pMemEntry, mem);
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
else if (type == "MD5")
|
||||||
|
{
|
||||||
|
const char *cstr_value = pMemEntry->Attribute("value");
|
||||||
|
if(!cstr_value)
|
||||||
|
throw Error::MemoryXmlUnderspecifiedEntry(cstr_name);
|
||||||
|
mem->setMD5(cstr_value);
|
||||||
|
}
|
||||||
|
else if (type == "PETimeStamp")
|
||||||
|
{
|
||||||
|
const char *cstr_value = pMemEntry->Attribute("value");
|
||||||
|
if(!cstr_value)
|
||||||
|
throw Error::MemoryXmlUnderspecifiedEntry(cstr_name);
|
||||||
|
mem->setPE(strtol(cstr_value, 0, 16));
|
||||||
|
}
|
||||||
|
} // for
|
||||||
|
} // method
|
||||||
|
|
||||||
|
VersionInfoFactory::VersionInfoFactory(string path_to_xml)
|
||||||
|
{
|
||||||
|
error = false;
|
||||||
|
loadFile(path_to_xml);
|
||||||
|
}
|
||||||
|
|
||||||
|
// load the XML file with offsets
|
||||||
|
bool VersionInfoFactory::loadFile(string path_to_xml)
|
||||||
|
{
|
||||||
|
TiXmlDocument doc( path_to_xml.c_str() );
|
||||||
|
//bool loadOkay = doc.LoadFile();
|
||||||
|
if (!doc.LoadFile())
|
||||||
|
{
|
||||||
|
error = true;
|
||||||
|
throw Error::MemoryXmlParse(doc.ErrorDesc(), doc.ErrorId(), doc.ErrorRow(), doc.ErrorCol());
|
||||||
|
}
|
||||||
|
TiXmlHandle hDoc(&doc);
|
||||||
|
TiXmlElement* pElem;
|
||||||
|
TiXmlHandle hRoot(0);
|
||||||
|
VersionInfo *mem;
|
||||||
|
|
||||||
|
// block: name
|
||||||
|
{
|
||||||
|
pElem=hDoc.FirstChildElement().Element();
|
||||||
|
// should always have a valid root but handle gracefully if it does
|
||||||
|
if (!pElem)
|
||||||
|
{
|
||||||
|
error = true;
|
||||||
|
throw Error::MemoryXmlNoRoot();
|
||||||
|
}
|
||||||
|
string m_name=pElem->Value();
|
||||||
|
if(m_name != "DFHack")
|
||||||
|
{
|
||||||
|
error = true;
|
||||||
|
throw Error::MemoryXmlNoRoot();
|
||||||
|
}
|
||||||
|
// save this for later
|
||||||
|
hRoot=TiXmlHandle(pElem);
|
||||||
|
}
|
||||||
|
// transform elements
|
||||||
|
{
|
||||||
|
// trash existing list
|
||||||
|
for(uint32_t i = 0; i < versions.size(); i++)
|
||||||
|
{
|
||||||
|
delete versions[i];
|
||||||
|
}
|
||||||
|
versions.clear();
|
||||||
|
|
||||||
|
// For each base version
|
||||||
|
TiXmlElement* pMemInfo=hRoot.FirstChild( "Base" ).Element();
|
||||||
|
map <string ,TiXmlElement *> map_pNamedEntries;
|
||||||
|
vector <string> v_sEntries;
|
||||||
|
for( ; pMemInfo; pMemInfo=pMemInfo->NextSiblingElement("Base"))
|
||||||
|
{
|
||||||
|
const char *name = pMemInfo->Attribute("name");
|
||||||
|
if(name)
|
||||||
|
{
|
||||||
|
string str_name = name;
|
||||||
|
VersionInfo *base = new VersionInfo();
|
||||||
|
mem = new VersionInfo();
|
||||||
|
ParseBase( pMemInfo , mem );
|
||||||
|
knownVersions[str_name] = v_descr (pMemInfo, mem);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// For each derivative version
|
||||||
|
pMemInfo=hRoot.FirstChild( "Version" ).Element();
|
||||||
|
for( ; pMemInfo; pMemInfo=pMemInfo->NextSiblingElement("Version"))
|
||||||
|
{
|
||||||
|
const char *name = pMemInfo->Attribute("name");
|
||||||
|
if(name)
|
||||||
|
{
|
||||||
|
string str_name = name;
|
||||||
|
knownVersions[str_name] = v_descr (pMemInfo, NULL);
|
||||||
|
v_sEntries.push_back(str_name);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
// Parse the versions
|
||||||
|
for(uint32_t i = 0; i< v_sEntries.size();i++)
|
||||||
|
{
|
||||||
|
//FIXME: add a set of entries processed in a step of this cycle, use it to check for infinite loops
|
||||||
|
string & name = v_sEntries[i];
|
||||||
|
v_descr & desc = knownVersions[name];
|
||||||
|
if(!desc.second)
|
||||||
|
{
|
||||||
|
VersionInfo *version = new VersionInfo();
|
||||||
|
ParseVersion( desc.first , version );
|
||||||
|
versions.push_back(version);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// process found things here
|
||||||
|
}
|
||||||
|
error = false;
|
||||||
|
return true;
|
||||||
|
}
|
||||||
@ -0,0 +1,449 @@
|
|||||||
|
// Console version of DF copy paste, proof of concept
|
||||||
|
// By belal
|
||||||
|
|
||||||
|
#include <iostream>
|
||||||
|
#include <iomanip>
|
||||||
|
#include <climits>
|
||||||
|
#include <vector>
|
||||||
|
#include <sstream>
|
||||||
|
#include <ctime>
|
||||||
|
#include <cstdio>
|
||||||
|
#include <fstream>
|
||||||
|
|
||||||
|
#define DFHACK_WANT_MISCUTILS
|
||||||
|
#define DFHACK_WANT_TILETYPES
|
||||||
|
#include <DFHack.h>
|
||||||
|
#include "dfhack/modules/WindowIO.h"
|
||||||
|
|
||||||
|
using namespace DFHack;
|
||||||
|
//bool waitTillCursorState(DFHack::Context *DF, bool On);
|
||||||
|
//bool waitTillCursorPositionState(DFHack::Context *DF, int32_t x,int32_t y, int32_t z);
|
||||||
|
|
||||||
|
//change this if you are having problems getting correct results, lower if you would like to go faster
|
||||||
|
//const int WAIT_AMT = 25;
|
||||||
|
|
||||||
|
void sort(uint32_t &a,uint32_t &b)
|
||||||
|
{
|
||||||
|
if(a > b){
|
||||||
|
uint32_t c = b;
|
||||||
|
b = a;
|
||||||
|
a = c;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
void sort(int32_t &a,int32_t &b)
|
||||||
|
{
|
||||||
|
if(a > b){
|
||||||
|
int16_t c = b;
|
||||||
|
b = a;
|
||||||
|
a = c;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
void printVecOfVec(ostream &out, vector<vector<vector<string> > >vec,char sep){
|
||||||
|
for(int k=0;k<vec.size();k++){
|
||||||
|
for(int i =0;i<vec[k].size();i++){
|
||||||
|
for(int j=0;j<vec[k][i].size();j++){
|
||||||
|
out << vec[k][i][j];
|
||||||
|
if(j==vec[k][i].size()-1)
|
||||||
|
{
|
||||||
|
out << "\n";
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
out << sep;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
out << "#<\n";
|
||||||
|
}
|
||||||
|
}
|
||||||
|
int main (int numargs, const char ** args)
|
||||||
|
{
|
||||||
|
map<string, string> buildCommands;
|
||||||
|
buildCommands["building_stockpilest"]="";
|
||||||
|
buildCommands["building_zonest"]="";
|
||||||
|
buildCommands["building_construction_blueprintst"]="";
|
||||||
|
buildCommands["building_wagonst"]="";
|
||||||
|
buildCommands["building_armor_standst"]="a";
|
||||||
|
buildCommands["building_bedst"]="b";
|
||||||
|
buildCommands["building_seatst"]="c";
|
||||||
|
buildCommands["building_burial_receptaclest"]="n";
|
||||||
|
buildCommands["building_doorst"]="d";
|
||||||
|
buildCommands["building_floodgatest"]="x";
|
||||||
|
buildCommands["building_floor_hatchst"]="H";
|
||||||
|
buildCommands["building_wall_gratest"]="W";
|
||||||
|
buildCommands["building_floor_gratest"]="G";
|
||||||
|
buildCommands["building_vertical_barsst"]="B";
|
||||||
|
buildCommands["building_floor_barsst"]="alt-b";
|
||||||
|
buildCommands["building_cabinetst"]="f";
|
||||||
|
buildCommands["building_containerst"]="h";
|
||||||
|
buildCommands["building_shopst"]="";
|
||||||
|
buildCommands["building_workshopst"]="";
|
||||||
|
buildCommands["building_alchemists_laboratoryst"]="wa";
|
||||||
|
buildCommands["building_carpenters_workshopst"]="wc";
|
||||||
|
buildCommands["building_farmers_workshopst"]="ww";
|
||||||
|
buildCommands["building_masons_workshopst"]="wm";
|
||||||
|
buildCommands["building_craftdwarfs_workshopst"]="wr";
|
||||||
|
buildCommands["building_jewelers_workshopst"]="wj";
|
||||||
|
buildCommands["building_metalsmiths_workshopst"]="wf";
|
||||||
|
buildCommands["building_magma_forgest"]="";
|
||||||
|
buildCommands["building_bowyers_workshopst"]="wb";
|
||||||
|
buildCommands["building_mechanics_workshopst"]="wt";
|
||||||
|
buildCommands["building_siege_workshopst"]="ws";
|
||||||
|
buildCommands["building_butchers_shopst"]="wU";
|
||||||
|
buildCommands["building_leather_worksst"]="we";
|
||||||
|
buildCommands["building_tanners_shopst"]="wn";
|
||||||
|
buildCommands["building_clothiers_shopst"]="wk";
|
||||||
|
buildCommands["building_fisheryst"]="wh";
|
||||||
|
buildCommands["building_stillst"]="wl";
|
||||||
|
buildCommands["building_loomst"]="wo";
|
||||||
|
buildCommands["building_quernst"]="wq";
|
||||||
|
buildCommands["building_kennelsst"]="k";
|
||||||
|
buildCommands["building_kitchenst"]="wz";
|
||||||
|
buildCommands["building_asheryst"]="wy";
|
||||||
|
buildCommands["building_dyers_shopst"]="wd";
|
||||||
|
buildCommands["building_millstonest"]="wM";
|
||||||
|
buildCommands["building_farm_plotst"]="p";
|
||||||
|
buildCommands["building_weapon_rackst"]="r";
|
||||||
|
buildCommands["building_statuest"]="s";
|
||||||
|
buildCommands["building_tablest"]="t";
|
||||||
|
buildCommands["building_paved_roadst"]="o";
|
||||||
|
buildCommands["building_bridgest"]="g";
|
||||||
|
buildCommands["building_wellst"]="l";
|
||||||
|
buildCommands["building_siege enginest"]="i";
|
||||||
|
buildCommands["building_catapultst"]="ic";
|
||||||
|
buildCommands["building_ballistast"]="ib";
|
||||||
|
buildCommands["building_furnacest"]="";
|
||||||
|
buildCommands["building_wood_furnacest"]="ew";
|
||||||
|
buildCommands["building_smelterst"]="es";
|
||||||
|
buildCommands["building_glass_furnacest"]="ek";
|
||||||
|
buildCommands["building_kilnst"]="ek";
|
||||||
|
buildCommands["building_magma_smelterst"]="es";
|
||||||
|
buildCommands["building_magma_glass_furnacest"]="ek";
|
||||||
|
buildCommands["building_magma_kilnst"]="ek";
|
||||||
|
buildCommands["building_glass_windowst"]="y";
|
||||||
|
buildCommands["building_gem_windowst"]="Y";
|
||||||
|
buildCommands["building_tradedepotst"]="D";
|
||||||
|
buildCommands["building_mechanismst"]="";
|
||||||
|
buildCommands["building_leverst"]="Tl";
|
||||||
|
buildCommands["building_pressure_platest"]="Tp";
|
||||||
|
buildCommands["building_cage_trapst"]="Tc";
|
||||||
|
buildCommands["building_stonefall_trapst"]="Ts";
|
||||||
|
buildCommands["building_weapon_trapst"]="Tw";
|
||||||
|
buildCommands["building_spikest"]="";
|
||||||
|
buildCommands["building_animal_trapst"]="m";
|
||||||
|
buildCommands["building_screw_pumpst"]="Ms";
|
||||||
|
buildCommands["building_water_wheelst"]="Mw";
|
||||||
|
buildCommands["building_windmillst"]="Mm";
|
||||||
|
buildCommands["building_gear_assemblyst"]="Mg";
|
||||||
|
buildCommands["building_horizontal_axlest"]="Mh";
|
||||||
|
buildCommands["building_vertical_axlest"]="Mv";
|
||||||
|
buildCommands["building_supportst"]="S";
|
||||||
|
buildCommands["building_cagest"]="j";
|
||||||
|
buildCommands["building_archery_targetst"]="A";
|
||||||
|
buildCommands["building_restraintst"]="v";
|
||||||
|
|
||||||
|
DFHack::ContextManager DFMgr("Memory.xml");
|
||||||
|
DFHack::Context *DF = DFMgr.getSingleContext();
|
||||||
|
|
||||||
|
try
|
||||||
|
{
|
||||||
|
DF->Attach();
|
||||||
|
}
|
||||||
|
catch (std::exception& e)
|
||||||
|
{
|
||||||
|
std::cerr << e.what() << std::endl;
|
||||||
|
#ifndef LINUX_BUILD
|
||||||
|
cin.ignore();
|
||||||
|
#endif
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
DFHack::Position *Pos = DF->getPosition();
|
||||||
|
DFHack::VersionInfo* mem = DF->getMemoryInfo();
|
||||||
|
DFHack::Process * p = DF->getProcess();
|
||||||
|
OffsetGroup * OG_Maps = mem->getGroup("Maps");
|
||||||
|
OffsetGroup * OG_MapBlock = OG_Maps->getGroup("block");
|
||||||
|
OffsetGroup * OG_LocalFt = OG_Maps->getGroup("features")->getGroup("local");
|
||||||
|
uint32_t designations = OG_MapBlock->getOffset("designation");
|
||||||
|
uint32_t block_feature1 = OG_MapBlock->getOffset("feature_local");
|
||||||
|
uint32_t block_feature2 = OG_MapBlock->getOffset("feature_global");
|
||||||
|
uint32_t region_x_offset = OG_Maps->getAddress("region_x");
|
||||||
|
uint32_t region_y_offset = OG_Maps->getAddress("region_y");
|
||||||
|
uint32_t region_z_offset = OG_Maps->getAddress("region_z");
|
||||||
|
uint32_t feature1_start_ptr = OG_LocalFt->getAddress("start_ptr");
|
||||||
|
int32_t regionX, regionY, regionZ;
|
||||||
|
|
||||||
|
// read position of the region inside DF world
|
||||||
|
p->readDWord (region_x_offset, (uint32_t &)regionX);
|
||||||
|
p->readDWord (region_y_offset, (uint32_t &)regionY);
|
||||||
|
p->readDWord (region_z_offset, (uint32_t &)regionZ);
|
||||||
|
while(1){
|
||||||
|
int32_t cx1,cy1,cz1;
|
||||||
|
cx1 = -30000;
|
||||||
|
while(cx1 == -30000){
|
||||||
|
DF->ForceResume();
|
||||||
|
cout << "Set cursor at first position, then press any key";
|
||||||
|
cin.ignore();
|
||||||
|
DF->Suspend();
|
||||||
|
Pos->getCursorCoords(cx1,cy1,cz1);
|
||||||
|
}
|
||||||
|
|
||||||
|
uint32_t tx1,ty1,tz1;
|
||||||
|
tx1 = cx1/16;
|
||||||
|
ty1 = cy1/16;
|
||||||
|
tz1 = cz1;
|
||||||
|
|
||||||
|
int32_t cx2,cy2,cz2;
|
||||||
|
cx2 = -30000;
|
||||||
|
while(cx2 == -30000){
|
||||||
|
DF->Resume();
|
||||||
|
cout << "Set cursor at second position, then press any key";
|
||||||
|
cin.ignore();
|
||||||
|
DF->Suspend();
|
||||||
|
Pos->getCursorCoords(cx2,cy2,cz2);
|
||||||
|
}
|
||||||
|
uint32_t tx2,ty2,tz2;
|
||||||
|
tx2 = cx2/16;
|
||||||
|
ty2 = cy2/16;
|
||||||
|
tz2 = cz2;
|
||||||
|
sort(tx1,tx2);
|
||||||
|
sort(ty1,ty2);
|
||||||
|
sort(tz1,tz2);
|
||||||
|
sort(cx1,cx2);
|
||||||
|
sort(cy1,cy2);
|
||||||
|
sort(cz1,cz2);
|
||||||
|
|
||||||
|
vector <vector<vector<string> > >dig(cz2-cz1+1,vector<vector<string> >(cy2-cy1+1,vector<string>(cx2-cx1+1)));
|
||||||
|
vector <vector<vector<string> > >build(cz2-cz1+1,vector<vector<string> >(cy2-cy1+1,vector<string>(cx2-cx1+1)));
|
||||||
|
mapblock40d block;
|
||||||
|
DFHack::Maps *Maps = DF->getMaps();
|
||||||
|
Maps->Start();
|
||||||
|
for(uint32_t y = ty1;y<=ty2;y++)
|
||||||
|
{
|
||||||
|
for(uint32_t x = tx1;x<=tx2;x++)
|
||||||
|
{
|
||||||
|
for(uint32_t z = tz1;z<=tz2;z++)
|
||||||
|
{
|
||||||
|
if(Maps->isValidBlock(x,y,z))
|
||||||
|
{
|
||||||
|
if(Maps->ReadBlock40d(x,y,z,&block))
|
||||||
|
{
|
||||||
|
int ystart,yend,xstart,xend;
|
||||||
|
ystart=xstart=0;
|
||||||
|
yend=xend=15;
|
||||||
|
if(y == ty2)
|
||||||
|
{
|
||||||
|
yend = cy2 % 16;
|
||||||
|
}
|
||||||
|
if(y == ty1)
|
||||||
|
{
|
||||||
|
ystart = cy1 % 16;
|
||||||
|
}
|
||||||
|
if(x == tx2)
|
||||||
|
{
|
||||||
|
xend = cx2 % 16;
|
||||||
|
}
|
||||||
|
if(x == tx1)
|
||||||
|
{
|
||||||
|
xstart = cx1 % 16;
|
||||||
|
}
|
||||||
|
int zidx = z-tz1;
|
||||||
|
for(int yy = ystart; yy <= yend;yy++)
|
||||||
|
{
|
||||||
|
int yidx = yy+(16*(y-ty1)-(cy1%16));
|
||||||
|
for(int xx = xstart; xx <= xend;xx++)
|
||||||
|
{
|
||||||
|
int xidx = xx+(16*(x-tx1)-(cx1%16));
|
||||||
|
if(DFHack::isOpenTerrain(block.tiletypes[xx][yy]) || DFHack::isFloorTerrain(block.tiletypes[xx][yy])) {dig[zidx][yidx][xidx] = "d";}
|
||||||
|
else if(DFHack::STAIR_DOWN == DFHack::tileTypeTable[block.tiletypes[xx][yy]].c){ dig [zidx][yidx][xidx] = "j"; build [zidx][yidx][xidx] = "Cd";}
|
||||||
|
else if(DFHack::STAIR_UP == DFHack::tileTypeTable[block.tiletypes[xx][yy]].c){ dig [zidx][yidx][xidx] = "u"; build [zidx][yidx][xidx] = "Cu";}
|
||||||
|
else if(DFHack::STAIR_UPDOWN == DFHack::tileTypeTable[block.tiletypes[xx][yy]].c){ dig [zidx][yidx][xidx] = "i"; build [zidx][yidx][xidx] = "Cx";}
|
||||||
|
else if(DFHack::isRampTerrain(block.tiletypes[xx][yy])){dig [zidx][yidx][xidx] = "r";build [zidx][yidx][xidx] = "Cr";}
|
||||||
|
else if(DFHack::isWallTerrain(block.tiletypes[xx][yy])){build [zidx][yidx][xidx] = "Cw";}
|
||||||
|
}
|
||||||
|
yidx++;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
DFHack::Buildings * Bld = DF->getBuildings();
|
||||||
|
std::map <uint32_t, std::string> custom_workshop_types;
|
||||||
|
uint32_t numBuildings;
|
||||||
|
if(Bld->Start(numBuildings))
|
||||||
|
{
|
||||||
|
Bld->ReadCustomWorkshopTypes(custom_workshop_types);
|
||||||
|
for(uint32_t i = 0; i < numBuildings; i++)
|
||||||
|
{
|
||||||
|
DFHack::t_building temp;
|
||||||
|
Bld->Read(i, temp);
|
||||||
|
if(temp.type != 0xFFFFFFFF) // check if type isn't invalid
|
||||||
|
{
|
||||||
|
std::string typestr;
|
||||||
|
mem->resolveClassIDToClassname(temp.type, typestr);
|
||||||
|
if(temp.z == cz1 && cx1 <= temp.x1 && cx2 >= temp.x2 && cy1 <= temp.y1 && cy2 >= temp.y2)
|
||||||
|
{
|
||||||
|
string currStr = build[temp.z-cz1][temp.y1-cy1][temp.x1-cx1];
|
||||||
|
stringstream stream;
|
||||||
|
string newStr = buildCommands[typestr];
|
||||||
|
if(temp.x1 != temp.x2)
|
||||||
|
{
|
||||||
|
stream << "(" << temp.x2-temp.x1+1 << "x" << temp.y2-temp.y1+1 << ")";
|
||||||
|
newStr += stream.str();
|
||||||
|
}
|
||||||
|
build[temp.z-cz1][temp.y1-cy1][temp.x1-cx1] = newStr + currStr;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
// for testing purposes
|
||||||
|
//ofstream outfile("test.txt");
|
||||||
|
// printVecOfVec(outfile, dig,'\t');
|
||||||
|
// outfile << endl;
|
||||||
|
// printVecOfVec(outfile, build,'\t');
|
||||||
|
// outfile << endl;
|
||||||
|
// outfile.close();
|
||||||
|
|
||||||
|
int32_t cx3,cy3,cz3,cx4,cy4,cz4;
|
||||||
|
uint32_t tx3,ty3,tz3,tx4,ty4,tz4;
|
||||||
|
char result;
|
||||||
|
while(1){
|
||||||
|
cx3 = -30000;
|
||||||
|
while(cx3 == -30000){
|
||||||
|
DF->Resume();
|
||||||
|
cout << "Set cursor at new position, then press any key:";
|
||||||
|
result = cin.get();
|
||||||
|
DF->Suspend();
|
||||||
|
Pos->getCursorCoords(cx3,cy3,cz3);
|
||||||
|
}
|
||||||
|
if(result == 'q'){
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
cx4 = cx3+cx2-cx1;
|
||||||
|
cy4 = cy3+cy2-cy1;
|
||||||
|
cz4 = cz3+cz2-cz1;
|
||||||
|
tx3=cx3/16;
|
||||||
|
ty3=cy3/16;
|
||||||
|
tz3=cz3;
|
||||||
|
tx4=cx4/16;
|
||||||
|
ty4=cy4/16;
|
||||||
|
tz4=cz4;
|
||||||
|
DFHack::WindowIO * Win = DF->getWindowIO();
|
||||||
|
designations40d designationBlock;
|
||||||
|
for(uint32_t y = ty3;y<=ty4;y++)
|
||||||
|
{
|
||||||
|
for(uint32_t x = tx3;x<=tx4;x++)
|
||||||
|
{
|
||||||
|
for(uint32_t z = tz3;z<=tz4;z++)
|
||||||
|
{
|
||||||
|
Maps->Start();
|
||||||
|
Maps->ReadBlock40d(x,y,z,&block);
|
||||||
|
Maps->ReadDesignations(x,y,z,&designationBlock);
|
||||||
|
int ystart,yend,xstart,xend;
|
||||||
|
ystart=xstart=0;
|
||||||
|
yend=xend=15;
|
||||||
|
if(y == ty4){
|
||||||
|
yend = cy4 % 16;
|
||||||
|
}
|
||||||
|
if(y == ty3){
|
||||||
|
ystart = cy3 % 16;
|
||||||
|
}
|
||||||
|
if(x == tx4){
|
||||||
|
xend = cx4 % 16;
|
||||||
|
}
|
||||||
|
if(x == tx3){
|
||||||
|
xstart = cx3 % 16;
|
||||||
|
}
|
||||||
|
int zidx = z-tz3;
|
||||||
|
for(int yy = ystart; yy <= yend;yy++){
|
||||||
|
int yidx = yy+(16*(y-ty3)-(cy3%16));
|
||||||
|
for(int xx = xstart; xx <= xend;xx++){
|
||||||
|
int xidx = xx+(16*(x-tx3)-(cx3%16));
|
||||||
|
if(dig[zidx][yidx][xidx] != ""){
|
||||||
|
char test = dig[zidx][yidx][xidx].c_str()[0];
|
||||||
|
switch (test){
|
||||||
|
case 'd':
|
||||||
|
designationBlock[xx][yy].bits.dig = DFHack::designation_default;
|
||||||
|
break;
|
||||||
|
case 'i':
|
||||||
|
designationBlock[xx][yy].bits.dig = DFHack::designation_ud_stair;
|
||||||
|
break;
|
||||||
|
case 'u':
|
||||||
|
designationBlock[xx][yy].bits.dig = DFHack::designation_u_stair;
|
||||||
|
break;
|
||||||
|
case 'j':
|
||||||
|
designationBlock[xx][yy].bits.dig = DFHack::designation_d_stair;
|
||||||
|
break;
|
||||||
|
case 'r':
|
||||||
|
designationBlock[xx][yy].bits.dig = DFHack::designation_ramp;
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
}
|
||||||
|
yidx++;
|
||||||
|
}
|
||||||
|
Maps->Start();
|
||||||
|
Maps->WriteDesignations(x,y,z,&designationBlock);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
DF->Detach();
|
||||||
|
#ifndef LINUX_BUILD
|
||||||
|
std::cout << "Done. Press any key to continue" << std::endl;
|
||||||
|
cin.ignore();
|
||||||
|
#endif
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
/*
|
||||||
|
bool waitTillCursorState(DFHack::Context *DF, bool On)
|
||||||
|
{
|
||||||
|
DFHack::WindowIO * w = DF->getWindowIO();
|
||||||
|
DFHack::Position * p = DF->getPosition();
|
||||||
|
int32_t x,y,z;
|
||||||
|
int tryCount = 0;
|
||||||
|
DF->Suspend();
|
||||||
|
bool cursorResult = p->getCursorCoords(x,y,z);
|
||||||
|
while(tryCount < 50 && On && !cursorResult || !On && cursorResult)
|
||||||
|
{
|
||||||
|
DF->Resume();
|
||||||
|
w->TypeSpecial(DFHack::WAIT,1,WAIT_AMT);
|
||||||
|
tryCount++;
|
||||||
|
DF->Suspend();
|
||||||
|
cursorResult = p->getCursorCoords(x,y,z);
|
||||||
|
}
|
||||||
|
if(tryCount >= 50)
|
||||||
|
{
|
||||||
|
cerr << "Something went wrong, cursor at x: " << x << " y: " << y << " z: " << z << endl;
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
DF->Resume();
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
bool waitTillCursorPositionState(DFHack::Context *DF, int32_t x,int32_t y, int32_t z)
|
||||||
|
{
|
||||||
|
DFHack::WindowIO * w = DF->getWindowIO();
|
||||||
|
DFHack::Position * p = DF->getPosition();
|
||||||
|
int32_t x2,y2,z2;
|
||||||
|
int tryCount = 0;
|
||||||
|
DF->Suspend();
|
||||||
|
bool cursorResult = p->getCursorCoords(x2,y2,z2);
|
||||||
|
while(tryCount < 50 && (x != x2 || y != y2 || z != z2))
|
||||||
|
{
|
||||||
|
DF->Resume();
|
||||||
|
w->TypeSpecial(DFHack::WAIT,1,WAIT_AMT);
|
||||||
|
tryCount++;
|
||||||
|
DF->Suspend();
|
||||||
|
cursorResult = p->getCursorCoords(x2,y2,z2);
|
||||||
|
}
|
||||||
|
if(tryCount >= 50)
|
||||||
|
{
|
||||||
|
cerr << "Something went wrong, cursor at x: " << x2 << " y: " << y2 << " z: " << z2 << endl;
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
DF->Resume();
|
||||||
|
return true;
|
||||||
|
}*/
|
||||||
@ -0,0 +1,98 @@
|
|||||||
|
// Just show some position data
|
||||||
|
|
||||||
|
#include <iostream>
|
||||||
|
#include <iomanip>
|
||||||
|
#include <climits>
|
||||||
|
#include <vector>
|
||||||
|
#include <sstream>
|
||||||
|
#include <ctime>
|
||||||
|
#include <cstdio>
|
||||||
|
|
||||||
|
#define DFHACK_WANT_MISCUTILS
|
||||||
|
#define DFHACK_WANT_TILETYPES
|
||||||
|
#include <DFHack.h>
|
||||||
|
|
||||||
|
using namespace DFHack;
|
||||||
|
|
||||||
|
void printWeather(DFHack::WeatherType current)
|
||||||
|
{
|
||||||
|
switch (current)
|
||||||
|
{
|
||||||
|
case CLEAR:
|
||||||
|
cout << "The sky is clear." << endl;
|
||||||
|
cout << "Options:" << endl;
|
||||||
|
cout << "'r' to make it rain." << endl;
|
||||||
|
cout << "'s' to make it snow." << endl;
|
||||||
|
break;
|
||||||
|
case RAINING:
|
||||||
|
cout << "It is raining." << endl;
|
||||||
|
cout << "Options:" << endl;
|
||||||
|
cout << "'c' to clear the sky." << endl;
|
||||||
|
cout << "'s' to make it snow." << endl;
|
||||||
|
break;
|
||||||
|
case SNOWING:
|
||||||
|
cout << "It is snowing." << endl;
|
||||||
|
cout << "Options:" << endl;
|
||||||
|
cout << "'c' to clear the sky." << endl;
|
||||||
|
cout << "'r' to make it rain." << endl;
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
cout << "'q' to quit." << endl;
|
||||||
|
cout << "anything else to refresh" << endl;
|
||||||
|
cout << ">";
|
||||||
|
}
|
||||||
|
|
||||||
|
using namespace DFHack;
|
||||||
|
int main (int numargs, const char ** args)
|
||||||
|
{
|
||||||
|
DFHack::ContextManager DFMgr("Memory.xml");
|
||||||
|
DFHack::Context *DF = DFMgr.getSingleContext();
|
||||||
|
|
||||||
|
try
|
||||||
|
{
|
||||||
|
DF->Attach();
|
||||||
|
}
|
||||||
|
catch (std::exception& e)
|
||||||
|
{
|
||||||
|
std::cerr << e.what() << std::endl;
|
||||||
|
#ifndef LINUX_BUILD
|
||||||
|
cin.ignore();
|
||||||
|
#endif
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
World *W = DF->getWorld();
|
||||||
|
W->Start();
|
||||||
|
bool end = false;
|
||||||
|
while(!end)
|
||||||
|
{
|
||||||
|
WeatherType current = (WeatherType) W->ReadCurrentWeather();
|
||||||
|
DF->Resume();
|
||||||
|
string command = "";
|
||||||
|
printWeather(current);
|
||||||
|
getline(cin, command);
|
||||||
|
DF->Suspend();
|
||||||
|
if(command == "c")
|
||||||
|
{
|
||||||
|
W->SetCurrentWeather(CLEAR);
|
||||||
|
}
|
||||||
|
else if(command == "r")
|
||||||
|
{
|
||||||
|
W->SetCurrentWeather(RAINING);
|
||||||
|
}
|
||||||
|
else if(command == "s")
|
||||||
|
{
|
||||||
|
W->SetCurrentWeather(SNOWING);
|
||||||
|
}
|
||||||
|
else if(command == "q")
|
||||||
|
{
|
||||||
|
end = true;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
#ifndef LINUX_BUILD
|
||||||
|
std::cout << "Done. Press any key to continue" << std::endl;
|
||||||
|
cin.ignore();
|
||||||
|
#endif
|
||||||
|
DF->Detach();
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
Loading…
Reference in New Issue