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