merged input injection patch by belal
Petr Mrázek
parent
2a2a60723a
commit
337d42e15a
File diff suppressed because it is too large
Load Diff
@ -0,0 +1,312 @@
|
||||
// Creature dump
|
||||
|
||||
#include <iostream>
|
||||
#include <sstream>
|
||||
#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;
|
||||
}
|
||||
}
|
||||
vector <string> buildingtypes;
|
||||
map<string, vector<string> > names;
|
||||
uint32_t numCreatures;
|
||||
vector<t_matgloss> creaturestypes;
|
||||
void printDwarves(DFHack::API & DF)
|
||||
{
|
||||
int dwarfCounter = 0;
|
||||
for(uint32_t i = 0; i < numCreatures; i++)
|
||||
{
|
||||
t_creature temp;
|
||||
DF.ReadCreature(i, temp);
|
||||
string type = creaturestypes[temp.type].id;
|
||||
if(type == "DWARF" && !temp.flags1.bits.dead && !temp.flags2.bits.killed){
|
||||
cout << i << ":";
|
||||
if(temp.nick_name[0])
|
||||
{
|
||||
cout << temp.nick_name;
|
||||
}
|
||||
else{
|
||||
cout << temp.first_name;
|
||||
}
|
||||
string transName = DF.TranslateName(temp.last_name,names,creaturestypes[temp.type].id);
|
||||
cout << " " << temp.custom_profession; //transName;
|
||||
if(dwarfCounter%3 != 2){
|
||||
cout << '\t';
|
||||
}
|
||||
else{
|
||||
cout << endl;
|
||||
}
|
||||
dwarfCounter++;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
bool getDwarfSelection(DFHack::API & DF, t_creature & toChange,string & changeString, string & commandString,int & eraseAmount,int &dwarfNum,bool &isName)
|
||||
{
|
||||
DF.ForceResume();
|
||||
static string lastText;
|
||||
bool dwarfSuccess = false;
|
||||
|
||||
while(!dwarfSuccess)
|
||||
{
|
||||
string input;
|
||||
cout << "\nSelect Dwarf to Change or q to Quit" << endl;
|
||||
getline (cin, input);
|
||||
if(input == "q")
|
||||
{
|
||||
return false;
|
||||
}
|
||||
else if(input == "r")
|
||||
{
|
||||
DF.Suspend();
|
||||
printDwarves(DF);
|
||||
DF.Resume();
|
||||
}
|
||||
else if(!input.empty())
|
||||
{
|
||||
int num;
|
||||
stringstream(input) >> num;//= atol(input.c_str());
|
||||
DF.Suspend();
|
||||
dwarfSuccess = DF.ReadCreature(num,toChange);
|
||||
DF.Resume();
|
||||
string type = creaturestypes[toChange.type].id;
|
||||
if(type != "DWARF"){
|
||||
dwarfSuccess = false;
|
||||
}
|
||||
else
|
||||
{
|
||||
dwarfNum = num;
|
||||
}
|
||||
}
|
||||
}
|
||||
bool changeType = false;
|
||||
while(!changeType)
|
||||
{
|
||||
string input;
|
||||
cout << "\n(n)ickname or (p)rofession?" << endl;
|
||||
getline(cin, input);
|
||||
if(input == "q"){
|
||||
return false;
|
||||
}
|
||||
if(input == "n"){
|
||||
commandString = "pzyn";
|
||||
eraseAmount = string(toChange.nick_name).length();
|
||||
changeType = true;
|
||||
isName = true;
|
||||
}
|
||||
else if(input == "p"){
|
||||
commandString = "pzyp";
|
||||
eraseAmount = string(toChange.custom_profession).length();
|
||||
changeType = true;
|
||||
isName = false;
|
||||
}
|
||||
}
|
||||
bool changeValue = false;
|
||||
while(!changeValue)
|
||||
{
|
||||
string input;
|
||||
cout << "value to change to?" << endl;
|
||||
getline(cin, input);
|
||||
if(input == "q"){
|
||||
return false;
|
||||
}
|
||||
if(!lastText.empty() && input.empty()){
|
||||
changeValue = true;
|
||||
}
|
||||
else if( !input.empty()){
|
||||
lastText = input;
|
||||
changeValue = true;
|
||||
}
|
||||
}
|
||||
changeString = lastText;
|
||||
return true;
|
||||
}
|
||||
bool waitTillChanged(DFHack::API &DF, int creatureToCheck, string changeValue, bool isName)
|
||||
{
|
||||
DF.Suspend();
|
||||
t_creature testCre;
|
||||
DF.ReadCreature(creatureToCheck,testCre);
|
||||
int tryCount = 0;
|
||||
if(isName)
|
||||
{
|
||||
while(testCre.nick_name != changeValue && tryCount <50)
|
||||
{
|
||||
DF.TypeSpecial(WAIT,1,100);
|
||||
DF.Suspend();
|
||||
DF.ReadCreature(creatureToCheck,testCre);
|
||||
tryCount++;
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
while(testCre.custom_profession != changeValue && tryCount < 50)
|
||||
{
|
||||
DF.TypeSpecial(WAIT,1,100);
|
||||
DF.Suspend();
|
||||
DF.ReadCreature(creatureToCheck,testCre);
|
||||
tryCount++;
|
||||
}
|
||||
}
|
||||
if(tryCount >= 50){
|
||||
cerr << "Something went wrong, make sure that DF is at the correct screen";
|
||||
return false;
|
||||
}
|
||||
return true;
|
||||
}
|
||||
bool waitTillScreenState(DFHack::API &DF, string screenState)
|
||||
{
|
||||
t_viewscreen current;
|
||||
DF.Suspend();
|
||||
DF.ReadViewScreen(current);
|
||||
int tryCount = 0;
|
||||
while(buildingtypes[current.type] != screenState && tryCount < 50){
|
||||
DF.TypeSpecial(WAIT,1,100);
|
||||
DF.Suspend();
|
||||
DF.ReadViewScreen(current);
|
||||
tryCount++;
|
||||
}
|
||||
if(tryCount >= 50){
|
||||
cerr << "Something went wrong, DF at " << buildingtypes[current.type] << endl;
|
||||
return false;
|
||||
}
|
||||
return true;
|
||||
}
|
||||
|
||||
bool waitTillCursorState(DFHack::API &DF, bool On)
|
||||
{
|
||||
int32_t x,y,z;
|
||||
int tryCount = 0;
|
||||
DF.Suspend();
|
||||
bool cursorResult = DF.getCursorCoords(x,y,z);
|
||||
while(tryCount < 50 && On && !cursorResult || !On && cursorResult)
|
||||
{
|
||||
DF.TypeSpecial(WAIT,1,100);
|
||||
tryCount++;
|
||||
DF.Suspend();
|
||||
cursorResult = DF.getCursorCoords(x,y,z);
|
||||
}
|
||||
if(tryCount >= 50)
|
||||
{
|
||||
cerr << "Something went wrong, cursor at x: " << x << " y: " << y << " z: " << z << endl;
|
||||
return false;
|
||||
}
|
||||
return true;
|
||||
}
|
||||
int main (void)
|
||||
{
|
||||
DFHack::API DF("Memory.xml");
|
||||
if(!DF.Attach())
|
||||
{
|
||||
cerr << "DF not found" << endl;
|
||||
return 1;
|
||||
}
|
||||
//Need buildingtypes for the viewscreen vtables, this really should not be just buildings, as viewscreen and items both use the same interface
|
||||
uint32_t numBuildings = DF.InitReadBuildings(buildingtypes);
|
||||
|
||||
DFHack::memory_info mem = DF.getMemoryInfo();
|
||||
// get stone matgloss mapping
|
||||
if(!DF.ReadCreatureMatgloss(creaturestypes))
|
||||
{
|
||||
cerr << "Can't get the creature types." << endl;
|
||||
return 1;
|
||||
}
|
||||
|
||||
DF.InitReadNameTables(names);
|
||||
numCreatures = DF.InitReadCreatures();
|
||||
DF.InitViewAndCursor();
|
||||
DF.Suspend();
|
||||
printDwarves(DF);
|
||||
t_creature toChange;
|
||||
string changeString,commandString;
|
||||
int eraseAmount;
|
||||
int toChangeNum;
|
||||
bool isName;
|
||||
while(getDwarfSelection(DF,toChange,changeString,commandString,eraseAmount,toChangeNum,isName))
|
||||
{
|
||||
bool completed = false;
|
||||
int32_t x,y,z;
|
||||
DF.Suspend();
|
||||
DF.getCursorCoords(x,y,z);
|
||||
if(x == -30000)// cursor not displayed
|
||||
{
|
||||
DF.TypeStr("v");
|
||||
}
|
||||
if(waitTillCursorState(DF,true))
|
||||
{
|
||||
DF.Suspend();
|
||||
DF.setCursorCoords(toChange.x, toChange.y,toChange.z);
|
||||
vector<uint32_t> underCursor;
|
||||
while(!DF.getCurrentCursorCreatures(underCursor))
|
||||
{
|
||||
DF.TypeSpecial(WAIT,1,100);
|
||||
DF.Suspend();
|
||||
DF.setCursorCoords(toChange.x, toChange.y,toChange.z);
|
||||
}
|
||||
//CurrentCursorCreatures gives the creatures in the order that you see them with the 'k' cursor.
|
||||
//The 'v' cursor displays them in the order of last, then first,second,third and so on
|
||||
//Pretty weird, but it works
|
||||
//The only place that seems to display which creature is currently selected is on the stack, whose location is likely not static, so not usable
|
||||
if(underCursor[underCursor.size()-1] != toChange.origin)
|
||||
{
|
||||
for(int i = 0;i<underCursor.size()-1;i++)
|
||||
{
|
||||
DF.TypeStr("v",100);
|
||||
if(underCursor[i] == toChange.origin)
|
||||
{
|
||||
break;
|
||||
}
|
||||
}
|
||||
}
|
||||
DF.Resume();
|
||||
DF.TypeStr(commandString.c_str());
|
||||
if(waitTillScreenState(DF,"viewscreen_customize_unit"))
|
||||
{
|
||||
DF.TypeSpecial(BACK_SPACE,eraseAmount);
|
||||
if(waitTillChanged(DF,toChangeNum,"",isName))
|
||||
{
|
||||
DF.TypeStr(changeString.c_str());
|
||||
if(waitTillChanged(DF,toChangeNum,changeString,isName))
|
||||
{
|
||||
DF.TypeSpecial(ENTER);
|
||||
DF.TypeSpecial(SPACE); // should take you to unit screen if everything worked
|
||||
if(waitTillScreenState(DF,"viewscreen_unit"))
|
||||
{
|
||||
DF.TypeSpecial(SPACE);
|
||||
if(waitTillScreenState(DF,"viewscreen_dwarfmode"))
|
||||
{
|
||||
DF.TypeSpecial(SPACE);
|
||||
if(waitTillCursorState(DF,false))
|
||||
{
|
||||
completed = true;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
if(!completed){
|
||||
cerr << "Something went wrong, please reset DF to its original state, then press any key to continue" << endl;
|
||||
string line;
|
||||
DF.Resume();
|
||||
getline(cin, line);
|
||||
}
|
||||
DF.Suspend();
|
||||
printDwarves(DF);
|
||||
}
|
||||
return 0;
|
||||
}
|
||||
Loading…
Reference in New Issue