dfhack/tools/playground/dfstatus.cpp

/*
 * 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;
#include <DFHack.h>
#include <dfhack/DFVector.h>
#include <dfhack/extra/stopwatch.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;
    uint64_t start_time = 0;
    uint64_t end_time = 0;
    uint64_t total_time = 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);
    total_time += end_time - start_time;
    mvwprintw(local_win,12,2,"Time used:   %d ms, Time total:  %d ms", end_time - start_time, total_time);

    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;
    DFHack::Materials * Materials;
    try{ //is DF running?
        DF = DFMgr.getSingleContext();
        DF->Attach();
        Materials = DF->getMaterials();
        Materials->ReadAllMaterials();
        DF->Resume();
    }
    catch (exception& e){
        cerr << e.what() << endl;
        return 1;
    }
    //init and Attach
    ofstream file("dfstatus_errors.txt");
    streambuf* strm_buffer = cerr.rdbuf(); // save cerr's output buffer
    cerr.rdbuf (file.rdbuf()); // redirect output into the file

    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");
        start_time = GetTimeMs64();
        if(!DF->Suspend())
        {
            break;
        }

        //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->Resume();
        end_time = GetTimeMs64();
        //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        */
    cerr.rdbuf (strm_buffer); // restore old output buffer
    file.close();

    return 0;
}