2011-08-01 18:21:25 -06:00
|
|
|
#include <iostream>
|
|
|
|
#include <iomanip>
|
|
|
|
#include <map>
|
|
|
|
#include <algorithm>
|
|
|
|
#include <vector>
|
|
|
|
#include <string>
|
|
|
|
|
2011-12-31 04:48:42 -07:00
|
|
|
#include "Core.h"
|
2012-01-15 13:54:14 -07:00
|
|
|
#include "Console.h"
|
|
|
|
#include "Export.h"
|
|
|
|
#include "PluginManager.h"
|
|
|
|
#include "modules/Vegetation.h"
|
|
|
|
#include "modules/Maps.h"
|
|
|
|
#include "modules/Gui.h"
|
|
|
|
#include "TileTypes.h"
|
|
|
|
#include "modules/MapCache.h"
|
2011-08-01 18:21:25 -06:00
|
|
|
|
|
|
|
using std::vector;
|
|
|
|
using std::string;
|
|
|
|
using namespace DFHack;
|
2012-01-21 16:54:57 -07:00
|
|
|
using df::global::world;
|
|
|
|
|
2011-08-01 18:21:25 -06:00
|
|
|
DFhackCExport command_result df_grow (Core * c, vector <string> & parameters);
|
|
|
|
DFhackCExport command_result df_immolate (Core * c, vector <string> & parameters);
|
|
|
|
DFhackCExport command_result df_extirpate (Core * c, vector <string> & parameters);
|
|
|
|
|
|
|
|
DFhackCExport const char * plugin_name ( void )
|
|
|
|
{
|
|
|
|
return "plants";
|
|
|
|
}
|
|
|
|
|
|
|
|
DFhackCExport command_result plugin_init ( Core * c, std::vector <PluginCommand> &commands)
|
|
|
|
{
|
|
|
|
commands.clear();
|
|
|
|
commands.push_back(PluginCommand("grow", "Grows saplings into trees (with active cursor, only the targetted one).", df_grow));
|
|
|
|
commands.push_back(PluginCommand("immolate", "Set plants on fire (under cursor, 'shrubs', 'trees' or 'all').", df_immolate));
|
|
|
|
commands.push_back(PluginCommand("extirpate", "Kill plants (same mechanics as immolate).", df_extirpate));
|
|
|
|
return CR_OK;
|
|
|
|
}
|
|
|
|
|
|
|
|
DFhackCExport command_result plugin_shutdown ( Core * c )
|
|
|
|
{
|
|
|
|
return CR_OK;
|
|
|
|
}
|
|
|
|
|
|
|
|
enum do_what
|
|
|
|
{
|
|
|
|
do_immolate,
|
|
|
|
do_extirpate
|
|
|
|
};
|
|
|
|
|
2011-08-08 17:50:22 -06:00
|
|
|
static bool getoptions( vector <string> & parameters, bool & shrubs, bool & trees, bool & help)
|
2011-08-01 18:21:25 -06:00
|
|
|
{
|
2012-01-31 09:55:38 -07:00
|
|
|
for(size_t i = 0;i < parameters.size();i++)
|
2011-08-01 18:21:25 -06:00
|
|
|
{
|
|
|
|
if(parameters[i] == "shrubs")
|
|
|
|
{
|
|
|
|
shrubs = true;
|
|
|
|
}
|
|
|
|
else if(parameters[i] == "trees")
|
|
|
|
{
|
|
|
|
trees = true;
|
|
|
|
}
|
|
|
|
else if(parameters[i] == "all")
|
|
|
|
{
|
|
|
|
trees = true;
|
|
|
|
shrubs = true;
|
|
|
|
}
|
2011-08-08 17:50:22 -06:00
|
|
|
else if(parameters[i] == "help" || parameters[i] == "?")
|
|
|
|
{
|
|
|
|
help = true;
|
|
|
|
}
|
2011-08-01 18:21:25 -06:00
|
|
|
else
|
|
|
|
{
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
|
|
|
/**
|
|
|
|
* Book of Immolations, chapter 1, verse 35:
|
|
|
|
* Armok emerged from the hellish depths and beheld the sunny realms for the first time.
|
|
|
|
* And he cursed the plants and trees for their bloodless wood, turning them into ash and smoldering ruin.
|
|
|
|
* Armok was pleased and great temples were built by the dwarves, for they shared his hatred for trees and plants.
|
|
|
|
*/
|
2011-08-08 17:50:22 -06:00
|
|
|
static command_result immolations (Core * c, do_what what, bool shrubs, bool trees, bool help)
|
2011-08-01 18:21:25 -06:00
|
|
|
{
|
2011-08-08 17:50:22 -06:00
|
|
|
static const char * what1 = "destroys";
|
|
|
|
static const char * what2 = "burns";
|
|
|
|
if(help)
|
|
|
|
{
|
|
|
|
c->con.print("Without any options, this command %s a plant under the cursor.\n"
|
|
|
|
"Options:\n"
|
|
|
|
"shrubs - affect all shrubs\n"
|
|
|
|
"trees - affect all trees\n"
|
|
|
|
"all - affect all plants\n",
|
|
|
|
what == do_immolate? what2 : what1
|
|
|
|
);
|
|
|
|
return CR_OK;
|
|
|
|
}
|
2012-01-21 17:31:15 -07:00
|
|
|
CoreSuspender suspend(c);
|
2012-01-19 20:44:17 -07:00
|
|
|
if (!Maps::IsValid())
|
2011-08-01 18:21:25 -06:00
|
|
|
{
|
2012-01-19 20:44:17 -07:00
|
|
|
c->con.printerr("Map is not available!\n");
|
2011-08-01 18:21:25 -06:00
|
|
|
return CR_FAILURE;
|
|
|
|
}
|
2012-02-13 15:56:33 -07:00
|
|
|
Gui * Gui = c->getGui();
|
2011-08-01 18:21:25 -06:00
|
|
|
uint32_t x_max, y_max, z_max;
|
2012-01-19 20:44:17 -07:00
|
|
|
Maps::getSize(x_max, y_max, z_max);
|
|
|
|
MapExtras::MapCache map;
|
2011-08-01 18:21:25 -06:00
|
|
|
if(shrubs || trees)
|
|
|
|
{
|
|
|
|
int destroyed = 0;
|
2012-01-21 16:54:57 -07:00
|
|
|
for(size_t i = 0 ; i < world->plants.all.size(); i++)
|
2011-08-01 18:21:25 -06:00
|
|
|
{
|
2012-01-21 16:54:57 -07:00
|
|
|
df::plant *p = world->plants.all[i];
|
|
|
|
if(shrubs && p->flags.bits.is_shrub || trees && !p->flags.bits.is_shrub)
|
2011-08-01 18:21:25 -06:00
|
|
|
{
|
|
|
|
if (what == do_immolate)
|
|
|
|
p->is_burning = true;
|
|
|
|
p->hitpoints = 0;
|
|
|
|
destroyed ++;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
c->con.print("Praise Armok!\n");
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
int32_t x,y,z;
|
|
|
|
if(Gui->getCursorCoords(x,y,z))
|
|
|
|
{
|
2012-01-21 16:54:57 -07:00
|
|
|
vector<df::plant *> * alltrees;
|
2012-01-19 20:44:17 -07:00
|
|
|
if(Maps::ReadVegetation(x/16,y/16,z,alltrees))
|
2011-08-01 18:21:25 -06:00
|
|
|
{
|
|
|
|
bool didit = false;
|
|
|
|
for(size_t i = 0 ; i < alltrees->size(); i++)
|
|
|
|
{
|
2012-01-21 16:54:57 -07:00
|
|
|
df::plant * tree = alltrees->at(i);
|
|
|
|
if(tree->pos.x == x && tree->pos.y == y && tree->pos.z == z)
|
2011-08-01 18:21:25 -06:00
|
|
|
{
|
|
|
|
if(what == do_immolate)
|
|
|
|
tree->is_burning = true;
|
|
|
|
tree->hitpoints = 0;
|
|
|
|
didit = true;
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
/*
|
|
|
|
if(!didit)
|
|
|
|
{
|
|
|
|
cout << "----==== There's NOTHING there! ====----" << endl;
|
|
|
|
}
|
|
|
|
*/
|
|
|
|
}
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
c->con.printerr("No mass destruction and no cursor...\n" );
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return CR_OK;
|
|
|
|
}
|
|
|
|
|
|
|
|
DFhackCExport command_result df_immolate (Core * c, vector <string> & parameters)
|
|
|
|
{
|
2011-08-08 17:50:22 -06:00
|
|
|
bool shrubs = false, trees = false, help = false;
|
|
|
|
if(getoptions(parameters,shrubs,trees,help))
|
2011-08-01 18:21:25 -06:00
|
|
|
{
|
2011-08-08 17:50:22 -06:00
|
|
|
return immolations(c,do_immolate,shrubs,trees, help);
|
2011-08-01 18:21:25 -06:00
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
c->con.printerr("Invalid parameter!\n");
|
|
|
|
return CR_FAILURE;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
DFhackCExport command_result df_extirpate (Core * c, vector <string> & parameters)
|
|
|
|
{
|
2011-08-08 17:50:22 -06:00
|
|
|
bool shrubs = false, trees = false, help = false;
|
|
|
|
if(getoptions(parameters,shrubs,trees, help))
|
2011-08-01 18:21:25 -06:00
|
|
|
{
|
2011-08-08 17:50:22 -06:00
|
|
|
return immolations(c,do_extirpate,shrubs,trees, help);
|
2011-08-01 18:21:25 -06:00
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
c->con.printerr("Invalid parameter!\n");
|
|
|
|
return CR_FAILURE;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
DFhackCExport command_result df_grow (Core * c, vector <string> & parameters)
|
|
|
|
{
|
2012-01-31 09:55:38 -07:00
|
|
|
for(size_t i = 0; i < parameters.size();i++)
|
2011-08-08 17:50:22 -06:00
|
|
|
{
|
|
|
|
if(parameters[i] == "help" || parameters[i] == "?")
|
|
|
|
{
|
|
|
|
c->con.print("This command turns all living saplings into full-grown trees.\n");
|
|
|
|
return CR_OK;
|
|
|
|
}
|
|
|
|
}
|
2012-01-21 17:31:15 -07:00
|
|
|
CoreSuspender suspend(c);
|
2011-08-01 18:21:25 -06:00
|
|
|
Console & con = c->con;
|
2012-01-19 20:44:17 -07:00
|
|
|
if (!Maps::IsValid())
|
2011-08-01 18:21:25 -06:00
|
|
|
{
|
2012-01-19 20:44:17 -07:00
|
|
|
c->con.printerr("Map is not available!\n");
|
2011-08-01 18:21:25 -06:00
|
|
|
return CR_FAILURE;
|
|
|
|
}
|
2012-01-19 20:44:17 -07:00
|
|
|
MapExtras::MapCache map;
|
2012-02-13 15:56:33 -07:00
|
|
|
Gui *Gui = c->getGui();
|
2011-08-01 18:21:25 -06:00
|
|
|
int32_t x,y,z;
|
|
|
|
if(Gui->getCursorCoords(x,y,z))
|
|
|
|
{
|
2012-01-21 16:54:57 -07:00
|
|
|
vector<df::plant *> * alltrees;
|
2012-01-19 20:44:17 -07:00
|
|
|
if(Maps::ReadVegetation(x/16,y/16,z,alltrees))
|
2011-08-01 18:21:25 -06:00
|
|
|
{
|
|
|
|
for(size_t i = 0 ; i < alltrees->size(); i++)
|
|
|
|
{
|
2012-01-21 16:54:57 -07:00
|
|
|
df::plant * tree = alltrees->at(i);
|
|
|
|
if(tree->pos.x == x && tree->pos.y == y && tree->pos.z == z)
|
2011-08-01 18:21:25 -06:00
|
|
|
{
|
2012-02-13 15:56:33 -07:00
|
|
|
if(tileShape(map.tiletypeAt(DFCoord(x,y,z))) == tiletype_shape::SAPLING &&
|
|
|
|
tileSpecial(map.tiletypeAt(DFCoord(x,y,z))) != tiletype_special::DEAD)
|
2011-08-01 18:21:25 -06:00
|
|
|
{
|
2012-01-24 11:02:12 -07:00
|
|
|
tree->grow_counter = Vegetation::sapling_to_tree_threshold;
|
2011-08-01 18:21:25 -06:00
|
|
|
}
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
int grown = 0;
|
2012-01-21 16:54:57 -07:00
|
|
|
for(size_t i = 0 ; i < world->plants.all.size(); i++)
|
2011-08-01 18:21:25 -06:00
|
|
|
{
|
2012-01-21 16:54:57 -07:00
|
|
|
df::plant *p = world->plants.all[i];
|
2012-02-13 15:56:33 -07:00
|
|
|
df::tiletype ttype = map.tiletypeAt(df::coord(p->pos.x,p->pos.y,p->pos.z));
|
|
|
|
if(!p->flags.bits.is_shrub && tileShape(ttype) == tiletype_shape::SAPLING && tileSpecial(ttype) != tiletype_special::DEAD)
|
2011-08-01 18:21:25 -06:00
|
|
|
{
|
2012-01-24 11:02:12 -07:00
|
|
|
p->grow_counter = Vegetation::sapling_to_tree_threshold;
|
2011-08-01 18:21:25 -06:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
return CR_OK;
|
|
|
|
}
|
|
|
|
|