Missing dfstatus.cpp
Petr Mrázek
parent
bb99924b24
commit
3fae382402
@ -0,0 +1,184 @@
|
||||
/*
|
||||
* dfstatus.cpp
|
||||
*/
|
||||
|
||||
#include <curses.h>
|
||||
|
||||
#ifndef LINUX_BUILD
|
||||
#include <windows.h>
|
||||
#endif
|
||||
|
||||
#include <time.h>
|
||||
#include <cstdio>
|
||||
#include <iostream>
|
||||
#include <fstream>
|
||||
#include <iomanip>
|
||||
#include <sstream>
|
||||
#include <climits>
|
||||
#include <vector>
|
||||
#include <cstring>
|
||||
#include <string>
|
||||
//#include <conio.h> //to break on keyboard input
|
||||
#include <stdio.h>
|
||||
#include <stdlib.h>
|
||||
#include <time.h>
|
||||
using namespace std;
|
||||
#define DFHACK_WANT_MISCUTILS
|
||||
#include <DFHack.h>
|
||||
#include <dfhack/DFVector.h>
|
||||
|
||||
WINDOW *create_newwin(int height, int width, int starty, int startx);
|
||||
|
||||
int32_t drinkCount = 0;
|
||||
int32_t plantCount = 0;
|
||||
int32_t fishCount = 0;
|
||||
int32_t meatCount = 0;
|
||||
int32_t logsCount = 0;
|
||||
int32_t barCount = 0;
|
||||
int32_t clothCount = 0;
|
||||
int32_t ironBars = 0;
|
||||
int32_t goldBars = 0;
|
||||
int32_t silverBars = 0;
|
||||
int32_t copperBars = 0;
|
||||
int32_t steelBars = 0;
|
||||
int32_t fuel = 0;
|
||||
|
||||
WINDOW *create_newwin(int height, int width, int starty, int startx){
|
||||
WINDOW *local_win;
|
||||
|
||||
local_win = newwin(height, width, starty, startx);
|
||||
box(local_win, 0, 0); /* 0, 0 gives default characters
|
||||
* for the vertical and horizontal
|
||||
* lines */
|
||||
//first row
|
||||
mvwprintw(local_win,2,2,"Drinks: %d", drinkCount);
|
||||
mvwprintw(local_win,4,2,"Plants: %d", plantCount);
|
||||
mvwprintw(local_win,5,4,"Fish: %d", fishCount);
|
||||
mvwprintw(local_win,6,4,"Meat: %d", meatCount);
|
||||
mvwprintw(local_win,8,2,"Logs: %d", logsCount);
|
||||
mvwprintw(local_win,10,2,"Cloth: %d", clothCount);
|
||||
//second row
|
||||
mvwprintw(local_win,2,22,"Iron Bars: %d", ironBars);
|
||||
mvwprintw(local_win,3,22,"Gold Bars: %d", goldBars);
|
||||
mvwprintw(local_win,4,22,"Silver Bars: %d", silverBars);
|
||||
mvwprintw(local_win,5,22,"Copper Bars: %d", copperBars);
|
||||
mvwprintw(local_win,6,22,"Steel Bars: %d", steelBars);
|
||||
mvwprintw(local_win,8,22,"Fuel: %d", fuel);
|
||||
|
||||
|
||||
wrefresh(local_win); // paint the screen and all components.
|
||||
|
||||
return local_win;
|
||||
}
|
||||
|
||||
int main()
|
||||
{
|
||||
WINDOW *my_win;
|
||||
int startx, starty, width, height;
|
||||
|
||||
DFHack::Process * p;
|
||||
DFHack::ContextManager DFMgr("Memory.xml");
|
||||
DFHack::Context * DF;
|
||||
try{ //is DF running?
|
||||
DF = DFMgr.getSingleContext();
|
||||
DF->Detach();
|
||||
}
|
||||
catch (exception& e){
|
||||
cerr << e.what() << endl;
|
||||
return 1;
|
||||
}
|
||||
//init and Attach
|
||||
|
||||
initscr(); //start curses.
|
||||
nonl();
|
||||
intrflush(stdscr, FALSE);
|
||||
keypad(stdscr, TRUE);
|
||||
|
||||
do{
|
||||
drinkCount = 0;
|
||||
plantCount = 0;
|
||||
fishCount = 0;
|
||||
meatCount = 0;
|
||||
logsCount = 0;
|
||||
barCount = 0;
|
||||
clothCount = 0;
|
||||
ironBars = 0;
|
||||
goldBars = 0;
|
||||
silverBars = 0;
|
||||
copperBars = 0;
|
||||
steelBars = 0;
|
||||
fuel = 0;
|
||||
|
||||
//FILE * pFile;
|
||||
//pFile = fopen("dump.txt","w");
|
||||
|
||||
DF->Attach();
|
||||
DFHack::Materials * Materials = DF->getMaterials();
|
||||
Materials->ReadAllMaterials();
|
||||
|
||||
//DFHack::Gui * Gui = DF->getGui();
|
||||
|
||||
DFHack::Items * Items = DF->getItems();
|
||||
Items->Start();
|
||||
|
||||
DFHack::VersionInfo * mem = DF->getMemoryInfo();
|
||||
p = DF->getProcess();
|
||||
|
||||
DFHack::OffsetGroup* itemGroup = mem->getGroup("Items");
|
||||
DFHack::DfVector <uint32_t> p_items (p, itemGroup->getAddress("items_vector"));
|
||||
uint32_t size = p_items.size();
|
||||
|
||||
DFHack::dfh_item itm; //declare itm
|
||||
//memset(&itm, 0, sizeof(DFHack::dfh_item)); //seems to set every value in itm to 0
|
||||
|
||||
for(unsigned int idx = 0; idx < size; idx++) //fill our item variables with this loop
|
||||
{
|
||||
Items->readItem(p_items[idx], itm);
|
||||
|
||||
if (itm.base.flags.owned) //only count what we actually own.
|
||||
continue;
|
||||
|
||||
string s0 = Items->getItemClass(itm.matdesc.itemType).c_str();
|
||||
string s1 = Items->getItemDescription(itm, Materials).c_str();
|
||||
|
||||
if( s0 == "drink" ) {drinkCount += itm.quantity;}
|
||||
else if(s0 == "plant"){plantCount += itm.quantity;}
|
||||
else if(s0 == "fish"){fishCount += itm.quantity;}
|
||||
else if(s0 == "meat"){meatCount += itm.quantity;}
|
||||
else if(s0 == "wood"){logsCount += itm.quantity;}
|
||||
else if(s0 == "cloth"){clothCount += itm.quantity;}
|
||||
else if(s0 == "bar") //need to break it down by ItemDescription to get the different types of bars.
|
||||
{
|
||||
barCount = barCount + itm.quantity;
|
||||
if(s1.find("IRON")!=string::npos){ironBars++;}
|
||||
else if(s1.find("GOLD")!=string::npos){goldBars++;}
|
||||
else if(s1.find("SILVER")!=string::npos){silverBars++;}
|
||||
else if(s1.find("COPPER")!=string::npos){copperBars++;}
|
||||
else if(s1.find("STEEL")!=string::npos){steelBars++;}
|
||||
else if(s1.find("COAL")!=string::npos){fuel++;}
|
||||
}
|
||||
/*if(s0 != "boulder" && s0 != "thread"){
|
||||
fprintf(pFile,"%5d: %12s - %64s - [%d]\n", idx, Items->getItemClass(itm.matdesc.itemType).c_str(), Items->getItemDescription(itm, Materials).c_str(), itm.quantity);
|
||||
}*/
|
||||
}
|
||||
DF->Detach();
|
||||
//printf("%d - %d\n", (clock()/CLOCKS_PER_SEC),(clock()/CLOCKS_PER_SEC)%60);
|
||||
height = LINES;
|
||||
width = COLS;
|
||||
starty = (LINES - height) / 2;
|
||||
startx = (COLS - width) / 2;
|
||||
|
||||
my_win = create_newwin(height, width, starty, startx);
|
||||
|
||||
#ifdef LINUX_BUILD
|
||||
sleep(10);
|
||||
#else
|
||||
Sleep(10000);
|
||||
#endif
|
||||
|
||||
} while(true);
|
||||
|
||||
endwin(); /* End curses mode */
|
||||
|
||||
return 0;
|
||||
}
|
||||
Loading…
Reference in New Issue