146 lines
3.8 KiB
C++
146 lines
3.8 KiB
C++
#include <dfhack/Core.h>
|
|
#include <dfhack/Console.h>
|
|
#include <dfhack/Export.h>
|
|
#include <dfhack/PluginManager.h>
|
|
#include <vector>
|
|
#include <string>
|
|
#include "dfhack/extra/stopwatch.h"
|
|
|
|
using std::vector;
|
|
using std::string;
|
|
using namespace DFHack;
|
|
//FIXME: possible race conditions with calling kittens from the IO thread and shutdown from Core.
|
|
bool shutdown_flag = false;
|
|
bool final_flag = true;
|
|
bool timering = false;
|
|
uint64_t timeLast = 0;
|
|
|
|
DFhackCExport command_result kittens (Core * c, vector <string> & parameters);
|
|
DFhackCExport command_result ktimer (Core * c, vector <string> & parameters);
|
|
|
|
DFhackCExport const char * plugin_name ( void )
|
|
{
|
|
return "kittens";
|
|
}
|
|
|
|
DFhackCExport command_result plugin_init ( Core * c, std::vector <PluginCommand> &commands)
|
|
{
|
|
commands.clear();
|
|
commands.push_back(PluginCommand("nyan","NYAN CAT INVASION!",kittens));
|
|
commands.push_back(PluginCommand("ktimer","Time events...",ktimer));
|
|
return CR_OK;
|
|
}
|
|
|
|
DFhackCExport command_result plugin_shutdown ( Core * c )
|
|
{
|
|
shutdown_flag = true;
|
|
while(!final_flag)
|
|
{
|
|
c->con.msleep(60);
|
|
}
|
|
return CR_OK;
|
|
}
|
|
|
|
|
|
|
|
DFhackCExport command_result plugin_onupdate ( Core * c )
|
|
{
|
|
if(timering == true)
|
|
{
|
|
uint64_t time2 = GetTimeMs64();
|
|
// harmless potential data race here...
|
|
uint64_t delta = time2-timeLast;
|
|
// harmless potential data race here...
|
|
timeLast = time2;
|
|
c->con.print("Time delta = %d ms\n", delta);
|
|
}
|
|
return CR_OK;
|
|
}
|
|
|
|
DFhackCExport command_result ktimer (Core * c, vector <string> & parameters)
|
|
{
|
|
if(timering)
|
|
{
|
|
timering = false;
|
|
return CR_OK;
|
|
}
|
|
uint64_t timestart = GetTimeMs64();
|
|
c->Suspend();
|
|
c->Resume();
|
|
uint64_t timeend = GetTimeMs64();
|
|
c->con.print("Time to suspend = %d ms\n",timeend - timestart);
|
|
// harmless potential data race here...
|
|
timeLast = timeend;
|
|
timering = true;
|
|
return CR_OK;
|
|
}
|
|
|
|
DFhackCExport command_result kittens (Core * c, vector <string> & parameters)
|
|
{
|
|
final_flag = false;
|
|
Console & con = c->con;
|
|
// http://evilzone.org/creative-arts/nyan-cat-ascii/
|
|
const char * nyan []=
|
|
{
|
|
"NYAN NYAN NYAN NYAN NYAN NYAN NYAN",
|
|
"+ o + o ",
|
|
" + o + +",
|
|
"o +",
|
|
" o + + +",
|
|
"+ o o + o",
|
|
"-_-_-_-_-_-_-_,------, o ",
|
|
"_-_-_-_-_-_-_-| /\\_/\\ ",
|
|
"-_-_-_-_-_-_-~|__( ^ .^) + + ",
|
|
"_-_-_-_-_-_-_-\"\" \"\" ",
|
|
"+ o o + o",
|
|
" + +",
|
|
"o o o o +",
|
|
" o +",
|
|
"+ + o o + ",
|
|
"NYAN NYAN NYAN NYAN NYAN NYAN NYAN",
|
|
0
|
|
};
|
|
const char * kittenz1 []=
|
|
{
|
|
" ____",
|
|
" (. \\",
|
|
" \\ | ",
|
|
" \\ |___(\\--/)",
|
|
" __/ ( . . )",
|
|
" \"'._. '-.O.'",
|
|
" '-. \\ \"|\\",
|
|
" '.,,/'.,,mrf",
|
|
0
|
|
};
|
|
con.cursor(false);
|
|
con.clear();
|
|
Console::color_value color = Console::COLOR_BLUE;
|
|
while(1)
|
|
{
|
|
if(shutdown_flag)
|
|
{
|
|
final_flag = true;
|
|
con.reset_color();
|
|
con << std::endl << "NYAN!" << std::endl << std::flush;
|
|
return CR_OK;
|
|
}
|
|
con.color(color);
|
|
int index = 0;
|
|
const char * kit = nyan[index];
|
|
con.gotoxy(1,1);
|
|
//con << "Your DF is now full of kittens!" << std::endl;
|
|
while (kit != 0)
|
|
{
|
|
con.gotoxy(1,1+index);
|
|
con << kit << std::endl;
|
|
index++;
|
|
kit = nyan[index];
|
|
}
|
|
con.flush();
|
|
con.msleep(60);
|
|
((int&)color) ++;
|
|
if(color > Console::COLOR_MAX)
|
|
color = Console::COLOR_BLUE;
|
|
}
|
|
}
|