summaryrefslogtreecommitdiff
path: root/plugins/probe.cpp
diff options
context:
space:
mode:
authorPetr Mrázek2011-07-25 10:14:58 +0200
committerPetr Mrázek2011-07-25 10:14:58 +0200
commitd65f5596cbe5b2b20dc932cf6d841f29cb9d77bb (patch)
treee75e8064aff2b85eff071fc59d71fad60f8becd4 /plugins/probe.cpp
parentbd9643c8cca04117167c3e708fc23e35b1a42b45 (diff)
downloaddfhack-d65f5596cbe5b2b20dc932cf6d841f29cb9d77bb.tar.gz
dfhack-d65f5596cbe5b2b20dc932cf6d841f29cb9d77bb.tar.bz2
dfhack-d65f5596cbe5b2b20dc932cf6d841f29cb9d77bb.tar.xz
Ported mode, probe. Some changes required for that.
Diffstat (limited to 'plugins/probe.cpp')
-rw-r--r--plugins/probe.cpp330
1 files changed, 330 insertions, 0 deletions
diff --git a/plugins/probe.cpp b/plugins/probe.cpp
new file mode 100644
index 00000000..dc1812eb
--- /dev/null
+++ b/plugins/probe.cpp
@@ -0,0 +1,330 @@
+// Just show some position data
+
+#include <iostream>
+#include <iomanip>
+#include <climits>
+#include <vector>
+#include <sstream>
+#include <ctime>
+#include <cstdio>
+using namespace std;
+
+#include <dfhack/Core.h>
+#include <dfhack/Console.h>
+#include <dfhack/Export.h>
+#include <dfhack/PluginManager.h>
+#include <vector>
+#include <string>
+#include <dfhack/modules/Maps.h>
+#include <dfhack/modules/Gui.h>
+#include <dfhack/modules/Materials.h>
+#include <dfhack/extra/MapExtras.h>
+#include <dfhack/MiscUtils.h>
+#include <xgetopt.h>
+
+using std::vector;
+using std::string;
+using namespace DFHack;
+
+DFhackCExport command_result df_probe (Core * c, vector <string> & parameters);
+
+DFhackCExport const char * plugin_name ( void )
+{
+ return "probe";
+}
+
+DFhackCExport command_result plugin_init ( Core * c, std::vector <PluginCommand> &commands)
+{
+ commands.clear();
+ commands.push_back(PluginCommand("probe",
+ "A tile probe",
+ df_probe));
+ return CR_OK;
+}
+
+DFhackCExport command_result plugin_shutdown ( Core * c )
+{
+ return CR_OK;
+}
+/*
+bool parseOptions(vector<string> &params, bool &showBlock, bool &showDesig,
+ bool &showOccup, bool &showTile, bool &showMisc)
+{
+ // With no options set, show everything.
+ showBlock = true;
+ showDesig = true;
+ showOccup = true;
+ showTile = true;
+ showMisc = true;
+
+ bool _showBlock = false;
+ bool _showDesig = false;
+ bool _showOccup = false;
+ bool _showTile = false;
+ bool _showMisc = false;
+
+ char c;
+ xgetopt opt(params, "bdotm");
+ opt.opterr = 0;
+ while ((c = opt()) != -1)
+ {
+ switch (c)
+ {
+ case 'b':
+ _showBlock = true;
+ break;
+ case 'd':
+ _showDesig = true;
+ break;
+ case 'o':
+ _showOccup = true;
+ break;
+ case 't':
+ _showTile = true;
+ break;
+ case 'm':
+ _showMisc = 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;
+ }
+ }
+
+ // If any options set, show only those requested via options.
+ if(_showBlock || _showDesig || _showOccup || _showTile || _showMisc)
+ {
+ showBlock = false;
+ showDesig = false;
+ showOccup = false;
+ showTile = false;
+ showMisc = false;
+
+ showBlock = _showBlock;
+ showDesig = _showDesig;
+ showOccup = _showOccup;
+ showTile = _showTile;
+ showMisc = _showMisc;
+ }
+
+ return true;
+}
+
+*/
+using namespace DFHack;
+DFhackCExport command_result df_probe (Core * c, vector <string> & parameters)
+{
+ //bool showBlock, showDesig, showOccup, showTile, showMisc;
+ Console & con = c->con;
+ /*
+ if (!parseOptions(parameters, showBlock, showDesig, showOccup,
+ showTile, showMisc))
+ {
+ con.printerr("Unknown parameters!\n");
+ return CR_FAILURE;
+ }
+ */
+
+ BEGIN_PROBE:
+ c->Suspend();
+
+ DFHack::Gui *Gui = c->getGui();
+ DFHack::Materials *Materials = c->getMaterials();
+ DFHack::VersionInfo* mem = c->vinfo;
+ DFHack::Maps *Maps = c->getMaps();
+ bool hasmats = Materials->ReadInorganicMaterials();
+
+ if(!Maps->Start())
+ {
+ con.printerr("Unable to access map data!\n");
+ }
+ else
+ {
+ MapExtras::MapCache mc (Maps);
+
+ int32_t regionX, regionY, regionZ;
+ Maps->getPosition(regionX,regionY,regionZ);
+
+ bool have_features = Maps->StartFeatures();
+
+ int32_t cursorX, cursorY, cursorZ;
+ Gui->getCursorCoords(cursorX,cursorY,cursorZ);
+ if(cursorX == -30000)
+ {
+ con.printerr("No cursor; place cursor over tile to probe.\n");
+ }
+ else
+ {
+ DFCoord cursor (cursorX,cursorY,cursorZ);
+
+ uint32_t blockX = cursorX / 16;
+ uint32_t tileX = cursorX % 16;
+ uint32_t blockY = cursorY / 16;
+ uint32_t tileY = cursorY % 16;
+
+ MapExtras::Block * b = mc.BlockAt(cursor/16);
+ mapblock40d & block = b->raw;
+ if(b)
+ {
+ con.print("block addr: 0x%x\n\n", block.origin);
+/*
+ if (showBlock)
+ {
+ con.print("block flags:\n");
+ print_bits<uint32_t>(block.blockflags.whole,con);
+ con.print("\n\n");
+ }
+*/
+ int16_t tiletype = mc.tiletypeAt(cursor);
+ naked_designation &des = block.designation[tileX][tileY].bits;
+/*
+ if(showDesig)
+ {
+ con.print("designation\n");
+ print_bits<uint32_t>(block.designation[tileX][tileY].whole,
+ con);
+ con.print("\n\n");
+ }
+
+ if(showOccup)
+ {
+ con.print("occupancy\n");
+ print_bits<uint32_t>(block.occupancy[tileX][tileY].whole,
+ con);
+ con.print("\n\n");
+ }
+*/
+
+ // tiletype
+ con.print("tiletype: %d", tiletype);
+ if(tileName(tiletype))
+ con.print(" = %s",tileName(tiletype));
+ con.print("\n");
+
+ DFHack::TileShape shape = tileShape(tiletype);
+ DFHack::TileMaterial material = tileMaterial(tiletype);
+ DFHack::TileSpecial special = tileSpecial(tiletype);
+ con.print("%-10s: %4d %s\n","Class" ,shape,
+ TileShapeString[ shape ]);
+ con.print("%-10s: %4d %s\n","Material" ,
+ material,TileMaterialString[ material ]);
+ con.print("%-10s: %4d %s\n","Special" ,
+ special, TileSpecialString[ special ]);
+ con.print("%-10s: %4d\n" ,"Variant" ,
+ tileVariant(tiletype));
+ con.print("%-10s: %s\n" ,"Direction",
+ tileDirection(tiletype).getStr());
+ con.print("\n");
+
+ con.print("temperature1: %d U\n",mc.temperature1At(cursor));
+ con.print("temperature2: %d U\n",mc.temperature2At(cursor));
+
+ // biome, geolayer
+ con << "biome: " << des.biome << std::endl;
+ con << "geolayer: " << des.geolayer_index
+ << std::endl;
+ int16_t base_rock = mc.baseMaterialAt(cursor);
+ if(base_rock != -1)
+ {
+ con << "Layer material: " << dec << base_rock;
+ if(hasmats)
+ con << " / " << Materials->inorganic[base_rock].id
+ << " / "
+ << Materials->inorganic[base_rock].name
+ << endl;
+ else
+ con << endl;
+ }
+ int16_t vein_rock = mc.veinMaterialAt(cursor);
+ if(vein_rock != -1)
+ {
+ con << "Vein material (final): " << dec << vein_rock;
+ if(hasmats)
+ con << " / " << Materials->inorganic[vein_rock].id
+ << " / "
+ << Materials->inorganic[vein_rock].name
+ << endl;
+ else
+ con << endl;
+ }
+ // liquids
+ if(des.flow_size)
+ {
+ if(des.liquid_type == DFHack::liquid_magma)
+ con <<"magma: ";
+ else con <<"water: ";
+ con << des.flow_size << std::endl;
+ }
+ if(des.flow_forbid)
+ con << "flow forbid" << std::endl;
+ if(des.pile)
+ con << "stockpile?" << std::endl;
+ if(des.rained)
+ con << "rained?" << std::endl;
+ if(des.smooth)
+ con << "smooth?" << std::endl;
+
+ #define PRINT_FLAG( X ) con.print("%-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);
+
+ if(have_features)
+ {
+ t_feature * local = 0;
+ t_feature * global = 0;
+ Maps->ReadFeatures(&(b->raw),&local,&global);
+ PRINT_FLAG( feature_local );
+ if(local)
+ {
+ con.print("%-16s", "");
+ con.print(" %4d", block.local_feature);
+ con.print(" (%2d)", local->type);
+ con.print(" addr 0x%X ", local->origin);
+ con.print(" %s\n", sa_feature(local->type));
+ }
+ PRINT_FLAG( feature_global );
+ if(global)
+ {
+ con.print("%-16s", "");
+ con.print(" %4d", block.global_feature);
+ con.print(" (%2d)", global->type);
+ con.print(" %s\n", sa_feature(global->type));
+ }
+ }
+ else
+ {
+ PRINT_FLAG( feature_local );
+ PRINT_FLAG( feature_global );
+ }
+ #undef PRINT_FLAG
+ con << "local feature idx: " << block.local_feature
+ << endl;
+ con << "global feature idx: " << block.global_feature
+ << endl;
+ con << "mystery: " << block.mystery << endl;
+ con << std::endl;
+ }
+ }
+ }
+ c->Resume();
+ return CR_OK;
+}