169 lines
4.5 KiB
C++
169 lines
4.5 KiB
C++
#include <iostream>
|
|
#include <iomanip>
|
|
#include <map>
|
|
#include <algorithm>
|
|
#include <vector>
|
|
|
|
using namespace std;
|
|
#include <DFHack.h>
|
|
#include <dfhack/extra/MapExtras.h>
|
|
#include <dfhack/extra/termutil.h>
|
|
#include <xgetopt.h>
|
|
#include <time.h>
|
|
#include <stdlib.h>
|
|
|
|
|
|
bool parseOptions(int argc, char **argv,
|
|
bool &trees, bool &shrubs, bool &immolate)
|
|
{
|
|
char c;
|
|
xgetopt opt(argc, argv, "sti");
|
|
opt.opterr = 0;
|
|
while ((c = opt()) != -1)
|
|
{
|
|
switch (c)
|
|
{
|
|
case 's':
|
|
shrubs = true;
|
|
break;
|
|
case 't':
|
|
trees = true;
|
|
break;
|
|
case 'i':
|
|
immolate = true;
|
|
break;
|
|
case '?':
|
|
switch (opt.optopt)
|
|
{
|
|
// For when we take arguments
|
|
default:
|
|
if (isprint(opt.optopt))
|
|
std::cerr << "Unknown option -" << opt.optopt << "!"
|
|
<< std::endl;
|
|
else
|
|
std::cerr << "Unknown option character " << (int) opt.optopt << "!"
|
|
<< std::endl;
|
|
}
|
|
default:
|
|
// Um.....
|
|
return false;
|
|
}
|
|
}
|
|
return true;
|
|
}
|
|
|
|
int main(int argc, char *argv[])
|
|
{
|
|
bool temporary_terminal = TemporaryTerminal();
|
|
bool all_trees = false;
|
|
bool all_shrubs = false;
|
|
bool immolate = false;
|
|
srand(time(0));
|
|
if (!parseOptions(argc, argv, all_trees, all_shrubs, immolate))
|
|
{
|
|
return -1;
|
|
}
|
|
|
|
uint32_t x_max = 0, y_max = 0, z_max = 0;
|
|
DFHack::ContextManager manager("Memory.xml");
|
|
|
|
DFHack::Context *context = manager.getSingleContext();
|
|
if (!context->Attach())
|
|
{
|
|
std::cerr << "Unable to attach to DF!" << std::endl;
|
|
if(temporary_terminal)
|
|
std::cin.ignore();
|
|
return 1;
|
|
}
|
|
|
|
DFHack::Maps *maps = context->getMaps();
|
|
if (!maps->Start())
|
|
{
|
|
std::cerr << "Cannot get map info!" << std::endl;
|
|
context->Detach();
|
|
if(temporary_terminal)
|
|
std::cin.ignore();
|
|
return 1;
|
|
}
|
|
DFHack::Gui * Gui = context->getGui();
|
|
maps->getSize(x_max, y_max, z_max);
|
|
MapExtras::MapCache map(maps);
|
|
uint32_t vegCount = 0;
|
|
DFHack::Vegetation *veg = context->getVegetation();
|
|
if (!veg->Start(vegCount))
|
|
{
|
|
std::cerr << "Unable to read vegetation!" << std::endl;
|
|
if(temporary_terminal)
|
|
cin.ignore();
|
|
return 1;
|
|
}
|
|
if(all_shrubs || all_trees)
|
|
{
|
|
int destroyed = 0;
|
|
for(size_t i = 0 ; i < vegCount; i++)
|
|
{
|
|
DFHack::dfh_plant p;
|
|
veg->Read(i,p);
|
|
if(all_shrubs && p.sdata.is_shrub || all_trees && !p.sdata.is_shrub)
|
|
{
|
|
if (immolate)
|
|
p.sdata.is_burning = true;
|
|
p.sdata.hitpoints = 0;
|
|
veg->Write(p);
|
|
destroyed ++;
|
|
}
|
|
}
|
|
cout << "Sacrificed " << destroyed;
|
|
if(all_shrubs)
|
|
cout << " shrubs to Armok." << endl;
|
|
if(all_trees)
|
|
cout << " trees to Armok." << endl;
|
|
cout << "----==== Praise Armok! ====----" << endl;
|
|
}
|
|
else
|
|
{
|
|
int32_t x,y,z;
|
|
if(Gui->getCursorCoords(x,y,z))
|
|
{
|
|
vector<DFHack::dfh_plant> alltrees;
|
|
if(maps->ReadVegetation(x/16,y/16,z,&alltrees))
|
|
{
|
|
bool didit = false;
|
|
for(size_t i = 0 ; i < alltrees.size(); i++)
|
|
{
|
|
DFHack::dfh_plant & tree = alltrees[i];
|
|
if(tree.sdata.x == x && tree.sdata.y == y && tree.sdata.z == z)
|
|
{
|
|
cout << "----==== Praise Armok! ====----" << endl;
|
|
if(immolate)
|
|
tree.sdata.is_burning = true;
|
|
tree.sdata.hitpoints = 0;
|
|
veg->Write(tree);
|
|
didit = true;
|
|
break;
|
|
}
|
|
}
|
|
if(!didit)
|
|
{
|
|
cout << "----==== There's NOTHING there! ====----" << endl;
|
|
}
|
|
}
|
|
}
|
|
else
|
|
{
|
|
cout << "No mass destruction and no cursor." << endl;
|
|
cout << "----==== Armok is not pleased! ====----" << endl;
|
|
}
|
|
}
|
|
// Cleanup
|
|
veg->Finish();
|
|
maps->Finish();
|
|
context->Detach();
|
|
if(temporary_terminal)
|
|
{
|
|
std::cout << " Press any key to finish.";
|
|
std::cin.ignore();
|
|
}
|
|
return 0;
|
|
}
|