dfhack/plugins/weather.cpp

138 lines
3.4 KiB
C++

2011-12-31 04:48:42 -07:00
#include "Core.h"
#include "Console.h"
#include "Export.h"
#include "PluginManager.h"
#include <vector>
#include <string>
#include "modules/World.h"
using std::vector;
using std::string;
using namespace DFHack;
bool locked = false;
unsigned char locked_data[25];
DFhackCExport command_result weather (Core * c, vector <string> & parameters);
DFhackCExport const char * plugin_name ( void )
{
return "weather";
}
DFhackCExport command_result plugin_init ( Core * c, std::vector <PluginCommand> &commands)
{
commands.clear();
commands.push_back(PluginCommand(
"weather", "Print the weather map or change weather.",
weather, false,
" Prints the current weather map by default.\n"
"Options:\n"
" snow - make it snow everywhere.\n"
" rain - make it rain.\n"
" clear - clear the sky.\n"
));
return CR_OK;
}
DFhackCExport command_result plugin_shutdown ( Core * c )
{
return CR_OK;
}
DFhackCExport command_result weather (Core * c, vector <string> & parameters)
{
2011-07-13 20:05:27 -06:00
Console & con = c->con;
bool lock = false;
bool unlock = false;
bool snow = false;
bool rain = false;
bool clear = false;
for(size_t i = 0; i < parameters.size();i++)
{
if(parameters[i] == "rain")
rain = true;
else if(parameters[i] == "snow")
snow = true;
else if(parameters[i] == "clear")
clear = true;
else if(parameters[i] == "lock")
lock = true;
else if(parameters[i] == "unlock")
unlock = true;
else
return CR_WRONG_USAGE;
}
if(lock && unlock)
{
2011-07-13 20:05:27 -06:00
con << "Lock or unlock? DECIDE!" << std::endl;
return CR_FAILURE;
}
int cnt = 0;
cnt += rain;
cnt += snow;
cnt += clear;
if(cnt > 1)
{
2011-07-13 20:05:27 -06:00
con << "Rain, snow or clear sky? DECIDE!" << std::endl;
return CR_FAILURE;
}
bool something = lock || unlock || rain || snow || clear;
CoreSuspender suspend(c);
DFHack::World * w = c->getWorld();
if(!w->wmap)
{
2011-07-13 20:05:27 -06:00
con << "Weather support seems broken :(" << std::endl;
return CR_FAILURE;
}
if(!something)
{
// paint weather map
2011-07-13 20:05:27 -06:00
con << "Weather map (C = clear, R = rain, S = snow):" << std::endl;
for(int y = 0; y<5;y++)
{
for(int x = 0; x<5;x++)
{
switch((*w->wmap)[x][y])
{
case CLEAR:
con << "C ";
break;
case RAINING:
con << "R ";
break;
case SNOWING:
con << "S ";
break;
default:
con << (int) (*w->wmap)[x][y] << " ";
break;
}
}
2011-07-13 20:05:27 -06:00
con << std::endl;
}
}
else
{
// weather changing action!
if(rain)
{
2011-07-13 20:05:27 -06:00
con << "Here comes the rain." << std::endl;
w->SetCurrentWeather(RAINING);
}
if(snow)
{
2011-07-13 20:05:27 -06:00
con << "Snow everywhere!" << std::endl;
w->SetCurrentWeather(SNOWING);
}
if(clear)
{
2011-07-13 20:05:27 -06:00
con << "Suddenly, sunny weather!" << std::endl;
w->SetCurrentWeather(CLEAR);
}
// FIXME: weather lock needs map ID to work reliably... needs to be implemented.
}
return CR_OK;
}