dfhack/tools/supported/probe.cpp

216 lines
8.1 KiB
C++

2011-02-08 06:33:58 -07:00
// Just show some position data
#include <iostream>
#include <iomanip>
#include <climits>
#include <vector>
#include <sstream>
#include <ctime>
#include <cstdio>
using namespace std;
2011-02-08 06:33:58 -07:00
#define DFHACK_WANT_MISCUTILS 1
#define DFHACK_WANT_TILETYPES 1
#include <DFHack.h>
2011-04-09 22:17:11 -06:00
#include <dfhack/extra/MapExtras.h>
#include <dfhack/extra/termutil.h>
2011-02-08 06:33:58 -07:00
using namespace DFHack;
int main (int numargs, const char ** args)
{
bool temporary_terminal = TemporaryTerminal();
2011-02-08 06:33:58 -07:00
DFHack::ContextManager DFMgr("Memory.xml");
DFHack::Context *DF = DFMgr.getSingleContext();
BEGIN_PROBE:
2011-02-08 06:33:58 -07:00
try
{
DF->Attach();
}
catch (std::exception& e)
{
std::cerr << e.what() << std::endl;
if(temporary_terminal)
2011-02-08 06:33:58 -07:00
cin.ignore();
return 1;
}
2011-03-18 04:38:37 -06:00
DFHack::Gui *Gui = DF->getGui();
DFHack::Materials *Materials = DF->getMaterials();
2011-02-08 06:33:58 -07:00
DFHack::VersionInfo* mem = DF->getMemoryInfo();
DFHack::Maps *Maps = DF->getMaps();
DFHack::Process * p = DF->getProcess();
bool hasmats = Materials->ReadInorganicMaterials();
2011-02-08 06:33:58 -07:00
2011-04-09 22:17:11 -06:00
if(!Maps->Start())
{
std::cerr << "Unable to access map data!" << std::endl;
2011-04-09 22:17:11 -06:00
}
else
{
MapExtras::MapCache mc (Maps);
2011-02-08 06:33:58 -07:00
2011-04-09 22:17:11 -06:00
OffsetGroup *mapsg = mem->getGroup("Maps");
OffsetGroup *mapblockg = mapsg->getGroup("block");
2011-02-08 06:33:58 -07:00
2011-04-09 22:17:11 -06:00
uint32_t region_x_offset = mapsg->getAddress("region_x");
uint32_t region_y_offset = mapsg->getAddress("region_y");
uint32_t region_z_offset = mapsg->getAddress("region_z");
2011-02-08 06:33:58 -07:00
2011-04-09 22:17:11 -06:00
uint32_t designatus = mapblockg->getOffset("designation");
uint32_t occup = mapblockg->getOffset("occupancy");
uint32_t biomus = mapblockg->getOffset("biome_stuffs");
2011-02-08 06:33:58 -07:00
2011-04-09 22:17:11 -06:00
int32_t regionX, regionY, regionZ;
2011-03-06 21:01:36 -07:00
2011-04-09 22:17:11 -06:00
// read position of the region inside DF world
p->readDWord (region_x_offset, (uint32_t &)regionX);
p->readDWord (region_y_offset, (uint32_t &)regionY);
p->readDWord (region_z_offset, (uint32_t &)regionZ);
2011-03-06 21:01:36 -07:00
2011-05-06 19:44:10 -06:00
bool have_features = Maps->StartFeatures();
2011-04-09 22:17:11 -06:00
int32_t cursorX, cursorY, cursorZ;
Gui->getCursorCoords(cursorX,cursorY,cursorZ);
if(cursorX != -30000)
2011-02-08 06:33:58 -07:00
{
2011-04-09 22:17:11 -06:00
DFCoord cursor (cursorX,cursorY,cursorZ);
2011-03-06 21:01:36 -07:00
2011-04-09 22:17:11 -06:00
uint32_t blockX = cursorX / 16;
uint32_t tileX = cursorX % 16;
uint32_t blockY = cursorY / 16;
uint32_t tileY = cursorY % 16;
2011-03-06 21:01:36 -07:00
2011-04-09 22:17:11 -06:00
MapExtras::Block * b = mc.BlockAt(cursor/16);
mapblock40d & block = b->raw;
if(b)
2011-03-06 21:01:36 -07:00
{
2011-04-09 22:17:11 -06:00
printf("block addr: 0x%x\n", block.origin);
int16_t tiletype = mc.tiletypeAt(cursor);
naked_designation &des = block.designation[tileX][tileY].bits;
uint32_t designato = block.origin + designatus + (tileX * 16 + tileY) * sizeof(t_designation);
uint32_t occupr = block.origin + occup + (tileX * 16 + tileY) * sizeof(t_occupancy);
printf("designation offset: 0x%x\n", designato);
print_bits<uint32_t>(block.designation[tileX][tileY].whole,std::cout);
std::cout << endl;
printf("occupancy offset: 0x%x\n", occupr);
print_bits<uint32_t>(block.occupancy[tileX][tileY].whole,std::cout);
std::cout << endl;
// tiletype
std::cout <<"tiletype: " << tiletype;
if(tileName(tiletype))
std::cout << " = " << tileName(tiletype) << std::endl;
DFHack::TileShape shape = tileShape(tiletype);
DFHack::TileMaterial material = tileMaterial(tiletype);
DFHack::TileSpecial special = tileSpecial(tiletype);
printf("%-10s: %4d %s\n","Class" ,shape, TileShapeString[ shape ]);
printf("%-10s: %4d %s\n","Material" ,material,TileMaterialString[ material ]);
printf("%-10s: %4d %s\n","Special" ,special, TileSpecialString[ special ]);
printf("%-10s: %4d\n" ,"Variant" ,tileVariant(tiletype));
printf("%-10s: %s\n" ,"Direction",tileDirection(tiletype).getStr());
2011-04-09 22:17:11 -06:00
std::cout << std::endl;
std::cout <<"temperature1: " << mc.temperature1At(cursor) << " U" << std::endl;
std::cout <<"temperature2: " << mc.temperature2At(cursor) << " U" << std::endl;
// biome, geolayer
std::cout << "biome: " << des.biome << std::endl;
std::cout << "geolayer: " << des.geolayer_index << std::endl;
int16_t base_rock = mc.baseMaterialAt(cursor);
if(base_rock != -1)
{
cout << "Layer material: " << dec << base_rock;
if(hasmats)
cout << " / " << Materials->inorganic[base_rock].id << " / " << Materials->inorganic[base_rock].name << endl;
else
cout << endl;
}
int16_t vein_rock = mc.veinMaterialAt(cursor);
if(vein_rock != -1)
{
cout << "Vein material (final): " << dec << vein_rock;
if(hasmats)
cout << " / " << Materials->inorganic[vein_rock].id << " / " << Materials->inorganic[vein_rock].name << endl;
else
cout << endl;
}
2011-04-09 22:17:11 -06:00
// liquids
if(des.flow_size)
{
if(des.liquid_type == DFHack::liquid_magma)
std::cout <<"magma: ";
else std::cout <<"water: ";
std::cout << des.flow_size << std::endl;
}
if(des.flow_forbid)
std::cout << "flow forbid" << std::endl;
if(des.pile)
std::cout << "stockpile?" << std::endl;
if(des.rained)
std::cout << "rained?" << std::endl;
if(des.smooth)
std::cout << "smooth?" << std::endl;
printf("biomestuffs: 0x%x\n", block.origin + biomus);
#define PRINT_FLAG( X ) printf("%-16s= %c\n", #X , ( des.X ? 'Y' : ' ' ) )
PRINT_FLAG( hidden );
PRINT_FLAG( light );
PRINT_FLAG( skyview );
PRINT_FLAG( subterranean );
PRINT_FLAG( water_table );
PRINT_FLAG( rained );
DFCoord pc(blockX, blockY);
2011-05-06 19:44:10 -06:00
if(have_features)
2011-04-09 22:17:11 -06:00
{
2011-05-06 19:44:10 -06:00
t_feature * local = 0;
t_feature * global = 0;
Maps->ReadFeatures(&(b->raw),&local,&global);
PRINT_FLAG( feature_local );
if(local)
{
printf("%-16s", "");
printf(" %4d", block.local_feature);
printf(" (%2d)", local->type);
2011-05-07 01:07:28 -06:00
printf(" addr 0x%X ", local->origin);
2011-05-06 19:44:10 -06:00
printf(" %s\n", sa_feature(local->type));
}
PRINT_FLAG( feature_global );
if(global)
{
printf("%-16s", "");
printf(" %4d", block.global_feature);
printf(" (%2d)", global->type);
printf(" %s\n", sa_feature(global->type));
}
2011-04-09 22:17:11 -06:00
}
2011-05-06 19:44:10 -06:00
else
2011-04-09 22:17:11 -06:00
{
2011-05-06 19:44:10 -06:00
PRINT_FLAG( feature_local );
PRINT_FLAG( feature_global );
2011-04-09 22:17:11 -06:00
}
#undef PRINT_FLAG
cout << "local feature idx: " << block.local_feature << endl;
cout << "global feature idx: " << block.global_feature << endl;
cout << "mystery: " << block.mystery << endl;
std::cout << std::endl;
2011-02-08 06:33:58 -07:00
}
}
}
DF->Detach();
if(temporary_terminal)
{
2011-02-08 06:33:58 -07:00
std::cout << "Press any key to refresh..." << std::endl;
cin.ignore();
goto BEGIN_PROBE;
}
2011-02-08 06:33:58 -07:00
return 0;
}