316 lines
11 KiB
C++
316 lines
11 KiB
C++
//
|
|
|
|
#include <iostream>
|
|
#include <vector>
|
|
#include <map>
|
|
#include <cstdlib>
|
|
#include <limits>
|
|
using namespace std;
|
|
|
|
#include <conio.h>
|
|
|
|
|
|
#include <DFHack.h>
|
|
#include <dfhack/DFTileTypes.h>
|
|
|
|
//Avoid including Windows.h because it causes name clashes
|
|
extern "C" __declspec(dllimport) void __stdcall Sleep(unsigned long milliseconds);
|
|
|
|
//Trim
|
|
#define WHITESPACE " \t\r\n"
|
|
inline string trimr(const string & s, const string & t = WHITESPACE)
|
|
{
|
|
string d (s);
|
|
string::size_type i (d.find_last_not_of (t));
|
|
if (i == string::npos)
|
|
return "";
|
|
else
|
|
return d.erase (d.find_last_not_of (t) + 1) ;
|
|
}
|
|
inline string triml(const string & s, const string & t = WHITESPACE)
|
|
{
|
|
string d (s);
|
|
return d.erase (0, s.find_first_not_of (t)) ;
|
|
}
|
|
inline string trim(const string & s, const string & t = WHITESPACE)
|
|
{
|
|
string d (s);
|
|
return triml(trimr(d, t), t);
|
|
}
|
|
|
|
void printtiletype( int i ){
|
|
printf("%s\n%4i ; %-13s ; %-11s ; %c ; %-12s ; %s\n",
|
|
( DFHack::tileTypeTable[i].name ? DFHack::tileTypeTable[i].name : "[invalid tile]" ),
|
|
i,
|
|
( DFHack::tileTypeTable[i].name ? DFHack::TileShapeString[ DFHack::tileTypeTable[i].shape ] : "" ),
|
|
( DFHack::tileTypeTable[i].name ? DFHack::TileMaterialString[ DFHack::tileTypeTable[i].material ] : "" ),
|
|
( DFHack::tileTypeTable[i].variant ? '0'+ DFHack::tileTypeTable[i].variant : ' ' ),
|
|
( DFHack::tileTypeTable[i].special ? DFHack::TileSpecialString[ DFHack::tileTypeTable[i].special ] : "" ),
|
|
( DFHack::tileTypeTable[i].direction.whole ? DFHack::tileTypeTable[i].direction.getStr() : "" ),
|
|
0
|
|
);
|
|
}
|
|
|
|
|
|
int main (void)
|
|
{
|
|
int32_t x,y,z,tx,ty;
|
|
//DFHack::designations40d designations;
|
|
DFHack::tiletypes40d tiles;
|
|
//DFHack::t_temperatures temp1,temp2;
|
|
uint32_t x_max,y_max,z_max;
|
|
int32_t oldT, newT;
|
|
int count, dirty;
|
|
|
|
//Brush defaults
|
|
DFHack::TileShape BrushClass = DFHack::WALL;
|
|
DFHack::TileMaterial BrushMat = DFHack::tilematerial_invalid;
|
|
int BrushType = -1;
|
|
|
|
DFHack::ContextManager DFMgr("Memory.xml");
|
|
DFHack::Context *DF;
|
|
DFHack::Maps * Maps;
|
|
DFHack::Gui * Gui;
|
|
try
|
|
{
|
|
DF=DFMgr.getSingleContext();
|
|
DF->Attach();
|
|
Maps = DF->getMaps();
|
|
Maps->Start();
|
|
Maps->getSize(x_max,y_max,z_max);
|
|
Gui = DF->getGui();
|
|
}
|
|
catch (exception& e)
|
|
{
|
|
cerr << e.what() << endl;
|
|
#ifndef LINUX_BUILD
|
|
cin.ignore();
|
|
#endif
|
|
return 1;
|
|
}
|
|
bool end = false;
|
|
cout << "Welcome to the Tile Drawing tool.\nType 'help' or ? for a list of available commands, 'q' to quit" << endl;
|
|
string mode = "wall";
|
|
string command = "";
|
|
|
|
while(!end)
|
|
{
|
|
DF->Resume();
|
|
|
|
cout << endl << ":";
|
|
getline(cin, command);
|
|
int ch = command[0];
|
|
if(command.length()<=0) ch=0;
|
|
if( ((int)command.find("help")) >=0 ) ch='?'; //under windows, find was casting unsigned!
|
|
switch(ch)
|
|
{
|
|
case '?':
|
|
cout << "Modes:" << endl
|
|
<< "O - draw Open Space" << endl
|
|
<< "M - draw material only (shape unchanged)" << endl
|
|
<< "m number - use Material value entered" << endl
|
|
<< "r - use Rock/stone material" << endl
|
|
<< "l - use Soil material" << endl
|
|
<< "v - use Vein material" << endl
|
|
<< "H - draw tile shape only (material unchanged)" << endl
|
|
<< "h number - draw Tile Shape value entered" << endl
|
|
<< "w - draw Wall tiles" << endl
|
|
<< "f - draw Floor tiles" << endl
|
|
<< "t number - draw exact tile type entered" << endl
|
|
<< "Commands:" << endl
|
|
<< "p - print tile shapes and materials, and current brush" << endl
|
|
<< "P - print all tile types" << endl
|
|
<< "q - quit" << endl
|
|
<< "help OR ? - print this list of commands" << endl
|
|
<< "d - being drawing" << endl
|
|
<< endl
|
|
<< "Usage:\nChoose a mode (default is walls), then enter 'd' to being drawing.\nMove the cursor in DF wherever you want to draw.\nPress any key to pause drawing." << endl;
|
|
break;
|
|
case 'p':
|
|
//Classes
|
|
printf("\nTile Type Classes:\n");
|
|
for(int i=0;i<DFHack::tileshape_count;++i)
|
|
{
|
|
printf("%4i ; %s\n", i, DFHack::TileShapeString[i] ,0 );
|
|
}
|
|
//Materials
|
|
printf("\nTile Type Materials:\n");
|
|
for(int i=0;i<DFHack::tilematerial_count;++i)
|
|
{
|
|
printf("%4i ; %s\n", i, DFHack::TileMaterialString[i] ,0 );
|
|
}
|
|
//fall through...
|
|
case 10:
|
|
case 13:
|
|
case 0:
|
|
//Print current cursor & brush settings.
|
|
cout << "\nCurrent Brush:\n";
|
|
cout << "tile = ";
|
|
if(BrushClass<0) cout<<"(not drawing)"; else cout<<DFHack::TileShapeString[BrushClass]; cout << endl;
|
|
cout << "mat = ";
|
|
if(BrushMat<0) cout<<"(not drawing)"; else cout<<DFHack::TileMaterialString[BrushMat]; cout << endl;
|
|
cout << "type = ";
|
|
if(BrushType<0){
|
|
cout<<"(not drawing)";
|
|
}else{
|
|
printtiletype(BrushType);
|
|
}
|
|
break;
|
|
case 'P':
|
|
cout << "\nAll Valid Tile Types:\n";
|
|
for(int i=0;i<TILE_TYPE_ARRAY_LENGTH;++i)
|
|
{
|
|
if( DFHack::tileTypeTable[i].name )
|
|
printtiletype(i);
|
|
}
|
|
case 'w':
|
|
BrushType=-1;
|
|
BrushClass = DFHack::WALL;
|
|
cout << "Tile brush shape set to Wall." << endl;
|
|
break;
|
|
case 'f':
|
|
BrushType=-1;
|
|
BrushClass = DFHack::FLOOR;
|
|
cout << "Tile brush shape set to Floor." << endl;
|
|
break;
|
|
case 'h':
|
|
BrushType=-1;
|
|
BrushClass = (DFHack::TileShape)atol( command.c_str()+1 );
|
|
cout << "Tile brush shape set to " << BrushClass << endl;
|
|
break;
|
|
case 'M':
|
|
BrushClass = DFHack::tileshape_invalid;
|
|
cout << "Tile brush will not draw tile shape." << endl;
|
|
break;
|
|
case 'r':
|
|
BrushType=-1;
|
|
BrushMat = DFHack::STONE;
|
|
cout << "Tile brush material set to Rock." << endl;
|
|
break;
|
|
case 'l':
|
|
BrushType=-1;
|
|
BrushMat = DFHack::SOIL;
|
|
cout << "Tile brush material set to Soil." << endl;
|
|
break;
|
|
case 'v':
|
|
BrushType=-1;
|
|
BrushMat = DFHack::VEIN;
|
|
cout << "Tile brush material set to Vein." << endl;
|
|
break;
|
|
case 'm':
|
|
BrushType=-1;
|
|
BrushMat = (DFHack::TileMaterial)atol( command.c_str()+1 );
|
|
cout << "Tile brush material set to " << BrushMat << endl;
|
|
break;
|
|
case 'H':
|
|
BrushMat = DFHack::tilematerial_invalid;
|
|
cout << "Tile brush will not draw material." << endl;
|
|
break;
|
|
case 'O':
|
|
BrushType=-1;
|
|
BrushClass = DFHack::EMPTY;
|
|
BrushMat = DFHack::AIR;
|
|
cout << "Tile brush will draw Open Space." << endl;
|
|
break;
|
|
case 't':
|
|
BrushClass = DFHack::tileshape_invalid ;
|
|
BrushMat = DFHack::tilematerial_invalid;
|
|
BrushType = atol( command.c_str()+1 );
|
|
cout << "Tile brush type set to:" << endl;
|
|
printtiletype(BrushType);
|
|
break;
|
|
case 'q':
|
|
end = true;
|
|
cout << "Bye!" << endl;
|
|
break;
|
|
case 'd':
|
|
{
|
|
count=0;
|
|
cout << "Beginning to draw at cursor." << endl << "Press any key to stop drawing." << endl;
|
|
//DF->Suspend();
|
|
kbhit(); //throw away, just to be sure.
|
|
for(;;)
|
|
{
|
|
if(!Maps->Start())
|
|
{
|
|
cout << "Can't see any DF map loaded." << endl;
|
|
break;
|
|
}
|
|
if(!Gui->getCursorCoords(x,y,z))
|
|
{
|
|
cout << "Can't get cursor coords! Make sure you have a cursor active in DF." << endl;
|
|
break;
|
|
}
|
|
//cout << "cursor coords: " << x << "/" << y << "/" << z << endl;
|
|
tx=x%16; ty=y%16;
|
|
|
|
if(!Maps->isValidBlock(x/16,y/16,z))
|
|
{
|
|
cout << "Invalid block." << endl;
|
|
break;
|
|
}
|
|
|
|
//Read the tiles.
|
|
dirty=0;
|
|
Maps->ReadTileTypes((x/16),(y/16),z, &tiles);
|
|
oldT = tiles[tx][ty];
|
|
|
|
newT = -1;
|
|
if( 0<BrushType ){
|
|
//Explicit tile type set. Trust the user.
|
|
newT = BrushType;
|
|
}else if( 0==BrushMat && 0==BrushClass ){
|
|
//Special case, Empty Air.
|
|
newT = 32;
|
|
}else if( BrushMat>=0 && BrushClass>=0 && ( BrushClass != DFHack::tileTypeTable[oldT].shape || BrushMat != DFHack::tileTypeTable[oldT].material) ){
|
|
//Set tile material and class
|
|
newT = DFHack::findTileType(BrushClass,BrushMat, DFHack::tileTypeTable[oldT].variant , DFHack::tileTypeTable[oldT].special , DFHack::tileTypeTable[oldT].direction );
|
|
if(newT<0) newT = DFHack::findTileType(BrushClass,BrushMat, DFHack::tilevariant_invalid, DFHack::tileTypeTable[oldT].special , DFHack::tileTypeTable[oldT].direction );
|
|
if(newT<0) newT = DFHack::findTileType(BrushClass,BrushMat, DFHack::tilevariant_invalid , DFHack::tileTypeTable[oldT].special , (uint32_t)0 );
|
|
}else if( BrushMat<0 && BrushClass>=0 && BrushClass != DFHack::tileTypeTable[oldT].shape ){
|
|
//Set current tile class only, as accurately as can be expected
|
|
newT = DFHack::findSimilarTileType(oldT,BrushClass);
|
|
}else if( BrushClass<0 && BrushMat>=0 && BrushMat != DFHack::tileTypeTable[oldT].material ){
|
|
//Set current tile material only
|
|
newT = DFHack::findTileType(DFHack::tileTypeTable[oldT].shape,BrushMat, DFHack::tileTypeTable[oldT].variant , DFHack::tileTypeTable[oldT].special , DFHack::tileTypeTable[oldT].direction );
|
|
if(newT<0) newT = DFHack::findTileType(DFHack::tileTypeTable[oldT].shape,BrushMat, DFHack::tilevariant_invalid , DFHack::tileTypeTable[oldT].special , DFHack::tileTypeTable[oldT].direction );
|
|
if(newT<0) newT = DFHack::findTileType(DFHack::tileTypeTable[oldT].shape,BrushMat, DFHack::tilevariant_invalid , DFHack::tileTypeTable[oldT].special , (uint32_t)0 );
|
|
}
|
|
//If no change, skip it (couldn't find a good tile type, or already what we want)
|
|
if ( newT > 0 && oldT != newT ){
|
|
//Set new tile type
|
|
tiles[tx][ty] = newT;
|
|
dirty=-1;
|
|
}
|
|
//If anything was changed, write it all.
|
|
if (dirty)
|
|
{
|
|
//Maps->WriteDesignations(x/16,y/16,z/16, &designations);
|
|
Maps->WriteTileTypes(x/16,y/16,z, &tiles);
|
|
printf("(%4d,%4d,%4d)",x,y,z);
|
|
++count;
|
|
}
|
|
|
|
Maps->Finish();
|
|
|
|
Sleep(10);
|
|
if( kbhit() ) break;
|
|
}
|
|
cin.clear();
|
|
cout << endl << count << " tiles were drawn." << endl << "Drawing halted. Entering command mode." << endl;
|
|
}
|
|
continue;
|
|
break;
|
|
default:
|
|
cout << "Unknown command: " << command << endl;
|
|
}
|
|
|
|
}
|
|
DF->Detach();
|
|
#ifndef LINUX_BUILD
|
|
cout << "Done. Press any key to continue" << endl;
|
|
cin.ignore();
|
|
#endif
|
|
return 0;
|
|
}
|