Compare commits

..

No commits in common. "master" and "20241111" have entirely different histories.

46 changed files with 794 additions and 1314 deletions

View File

@ -1,6 +1,6 @@
name: build name: build
# build on source or workflow changes # build on c/cpp changes or workflow changes
on: on:
push: push:
paths: paths:
@ -37,11 +37,8 @@ jobs:
- name: Test - name: Test
run: | run: |
./util/ci/test.sh source util/ci/script.sh
do_functional_test
- name: Test Install
run: |
make DESTDIR=/tmp/install install
clang: clang:
runs-on: ubuntu-22.04 runs-on: ubuntu-22.04
@ -61,25 +58,6 @@ jobs:
CXX: clang++ CXX: clang++
- name: Test - name: Test
run: |
./util/ci/test.sh
gcc_fedora:
runs-on: ubuntu-latest
container:
image: fedora:latest
steps:
- uses: actions/checkout@v4
- name: Install deps
run: | run: |
source util/ci/script.sh source util/ci/script.sh
install_linux_deps do_functional_test
- name: Build
run: |
source util/ci/script.sh
run_build
- name: Test
run: |
./util/ci/test.sh

8
.gitignore vendored
View File

@ -1,8 +1,8 @@
*~ *~
/minetestmapper minetestmapper
/minetestmapper.exe minetestmapper.exe
/colors.txt colors.txt
CMakeCache.txt CMakeCache.txt
CMakeFiles/ CMakeFiles/
@ -12,5 +12,3 @@ install_manifest.txt
Makefile Makefile
cmake_install.cmake cmake_install.cmake
cmake_config.h cmake_config.h
compile_commands.json
.vscode/

View File

@ -4,7 +4,6 @@
#include "BlockDecoder.h" #include "BlockDecoder.h"
#include "ZlibDecompressor.h" #include "ZlibDecompressor.h"
#include "log.h"
static inline uint16_t readU16(const unsigned char *data) static inline uint16_t readU16(const unsigned char *data)
{ {
@ -56,12 +55,13 @@ void BlockDecoder::decode(const ustring &datastr)
} }
m_version = version; m_version = version;
ustring datastr2;
if (version >= 29) { if (version >= 29) {
// decompress whole block at once // decompress whole block at once
m_zstd_decompressor.setData(data, length, 1); m_zstd_decompressor.setData(data, length, 1);
m_zstd_decompressor.decompress(m_scratch); datastr2 = m_zstd_decompressor.decompress();
data = m_scratch.c_str(); data = datastr2.c_str();
length = m_scratch.size(); length = datastr2.size();
} }
size_t dataOffset = 0; size_t dataOffset = 0;
@ -106,16 +106,16 @@ void BlockDecoder::decode(const ustring &datastr)
m_contentWidth = contentWidth; m_contentWidth = contentWidth;
if (version >= 29) { if (version >= 29) {
size_t mapDataSize = (contentWidth + paramsWidth) * 4096; m_mapData.resize((contentWidth + paramsWidth) * 4096);
m_mapData.assign(data + dataOffset, mapDataSize); m_mapData.assign(data + dataOffset, m_mapData.size());
return; // we have read everything we need and can return early return; // we have read everything we need and can return early
} }
// version < 29 // version < 29
ZlibDecompressor decompressor(data, length); ZlibDecompressor decompressor(data, length);
decompressor.setSeekPos(dataOffset); decompressor.setSeekPos(dataOffset);
decompressor.decompress(m_mapData); m_mapData = decompressor.decompress();
decompressor.decompress(m_scratch); // unused metadata decompressor.decompress(); // unused metadata
dataOffset = decompressor.seekPos(); dataOffset = decompressor.seekPos();
// Skip unused node timers // Skip unused node timers
@ -161,7 +161,7 @@ const std::string &BlockDecoder::getNode(u8 x, u8 y, u8 z) const
return empty; return empty;
NameMap::const_iterator it = m_nameMap.find(content); NameMap::const_iterator it = m_nameMap.find(content);
if (it == m_nameMap.end()) { if (it == m_nameMap.end()) {
errorstream << "Skipping node with invalid ID." << std::endl; std::cerr << "Skipping node with invalid ID." << std::endl;
return empty; return empty;
} }
return it->second; return it->second;

View File

@ -43,7 +43,7 @@ if(NOT CUSTOM_DOCDIR STREQUAL "")
message(STATUS "Using DOCDIR=${DOCDIR}") message(STATUS "Using DOCDIR=${DOCDIR}")
endif() endif()
list(APPEND CMAKE_MODULE_PATH "${PROJECT_SOURCE_DIR}/cmake") set(CMAKE_MODULE_PATH ${PROJECT_SOURCE_DIR}/cmake)
# Libraries: gd # Libraries: gd
@ -57,17 +57,7 @@ endif(NOT LIBGD_LIBRARY OR NOT LIBGD_INCLUDE_DIR)
# Libraries: zlib # Libraries: zlib
find_package(zlib-ng QUIET)
if(zlib-ng_FOUND)
set(ZLIB_INCLUDE_DIR zlib-ng::zlib)
set(ZLIB_LIBRARY zlib-ng::zlib)
set(USE_ZLIB_NG TRUE)
message(STATUS "Found zlib-ng, using it instead of zlib.")
else()
message(STATUS "zlib-ng not found, falling back to zlib.")
find_package(ZLIB REQUIRED) find_package(ZLIB REQUIRED)
set(USE_ZLIB_NG FALSE)
endif()
# Libraries: zstd # Libraries: zstd
@ -156,48 +146,9 @@ endif(ENABLE_REDIS)
# Compiling & Linking # Compiling & Linking
configure_file( include_directories(
"${CMAKE_CURRENT_SOURCE_DIR}/src/cmake_config.h.in" "${PROJECT_BINARY_DIR}"
"${CMAKE_CURRENT_BINARY_DIR}/cmake_config.h" "${CMAKE_CURRENT_SOURCE_DIR}/include"
)
if(CMAKE_CXX_COMPILER_ID MATCHES "^(GNU|Clang|AppleClang)$")
set(CMAKE_CXX_FLAGS_RELEASE "-O2")
set(CMAKE_CXX_FLAGS_DEBUG "-Og -g2")
add_compile_options(-Wall -pipe)
elseif(MSVC)
add_compile_options(/GR- /Zl)
endif()
if(CMAKE_BUILD_TYPE STREQUAL "Release")
add_definitions(-DNDEBUG)
endif()
add_executable(minetestmapper)
target_include_directories(minetestmapper PRIVATE
"${CMAKE_CURRENT_SOURCE_DIR}"
"${CMAKE_CURRENT_BINARY_DIR}"
)
target_sources(minetestmapper PRIVATE
src/BlockDecoder.cpp
src/PixelAttributes.cpp
src/PlayerAttributes.cpp
src/TileGenerator.cpp
src/ZlibDecompressor.cpp
src/ZstdDecompressor.cpp
src/Image.cpp
src/mapper.cpp
src/util.cpp
src/log.cpp
src/db-sqlite3.cpp
$<$<BOOL:${USE_POSTGRESQL}>:src/db-postgresql.cpp>
$<$<BOOL:${USE_LEVELDB}>:src/db-leveldb.cpp>
$<$<BOOL:${USE_REDIS}>:src/db-redis.cpp>
)
target_include_directories(minetestmapper PRIVATE
"${CMAKE_CURRENT_SOURCE_DIR}/src"
"${CMAKE_CURRENT_BINARY_DIR}" "${CMAKE_CURRENT_BINARY_DIR}"
${SQLITE3_INCLUDE_DIR} ${SQLITE3_INCLUDE_DIR}
${LIBGD_INCLUDE_DIR} ${LIBGD_INCLUDE_DIR}
@ -205,7 +156,39 @@ target_include_directories(minetestmapper PRIVATE
${ZSTD_INCLUDE_DIR} ${ZSTD_INCLUDE_DIR}
) )
target_link_libraries(minetestmapper configure_file(
"${PROJECT_SOURCE_DIR}/include/cmake_config.h.in"
"${PROJECT_BINARY_DIR}/cmake_config.h"
)
add_definitions(-DUSE_CMAKE_CONFIG_H)
if(CMAKE_CXX_COMPILER_ID MATCHES "^(GNU|Clang|AppleClang)$")
set(CMAKE_CXX_FLAGS_RELEASE "-O2")
set(CMAKE_CXX_FLAGS_DEBUG "-Og -g2")
add_compile_options(-Wall -pipe)
endif()
if(CMAKE_BUILD_TYPE STREQUAL "Release")
add_definitions(-DNDEBUG)
endif()
add_executable(minetestmapper
BlockDecoder.cpp
PixelAttributes.cpp
PlayerAttributes.cpp
TileGenerator.cpp
ZlibDecompressor.cpp
ZstdDecompressor.cpp
Image.cpp
mapper.cpp
util.cpp
db-sqlite3.cpp
$<$<BOOL:${USE_POSTGRESQL}>:db-postgresql.cpp>
$<$<BOOL:${USE_LEVELDB}>:db-leveldb.cpp>
$<$<BOOL:${USE_REDIS}>:db-redis.cpp>
)
target_link_libraries(
minetestmapper
${SQLITE3_LIBRARY} ${SQLITE3_LIBRARY}
${PostgreSQL_LIBRARIES} ${PostgreSQL_LIBRARIES}
${LEVELDB_LIBRARY} ${LEVELDB_LIBRARY}

131
PlayerAttributes.cpp Normal file
View File

@ -0,0 +1,131 @@
#include <fstream>
#include <sstream>
#include <stdexcept>
#include <dirent.h>
#include <unistd.h> // for usleep
#include <sqlite3.h>
#include "config.h"
#include "PlayerAttributes.h"
#include "util.h"
PlayerAttributes::PlayerAttributes(const std::string &worldDir)
{
std::ifstream ifs(worldDir + "world.mt");
if (!ifs.good())
throw std::runtime_error("Failed to read world.mt");
std::string backend = read_setting_default("player_backend", ifs, "files");
ifs.close();
if (backend == "files")
readFiles(worldDir + "players");
else if (backend == "sqlite3")
readSqlite(worldDir + "players.sqlite");
else
throw std::runtime_error(std::string("Unknown player backend: ") + backend);
}
void PlayerAttributes::readFiles(const std::string &playersPath)
{
DIR *dir;
dir = opendir (playersPath.c_str());
if (!dir)
return;
struct dirent *ent;
while ((ent = readdir (dir)) != NULL) {
if (ent->d_name[0] == '.')
continue;
std::ifstream in(playersPath + PATH_SEPARATOR + ent->d_name);
if (!in.good())
continue;
std::string name, position;
name = read_setting("name", in);
in.seekg(0);
position = read_setting("position", in);
Player player;
std::istringstream iss(position);
char tmp;
iss >> tmp; // '('
iss >> player.x;
iss >> tmp; // ','
iss >> player.y;
iss >> tmp; // ','
iss >> player.z;
iss >> tmp; // ')'
if (tmp != ')')
continue;
player.name = name;
player.x /= 10.0f;
player.y /= 10.0f;
player.z /= 10.0f;
m_players.push_back(player);
}
closedir(dir);
}
/**********/
#define SQLRES(f, good) \
result = (sqlite3_##f); \
if (result != good) { \
throw std::runtime_error(sqlite3_errmsg(db));\
}
#define SQLOK(f) SQLRES(f, SQLITE_OK)
void PlayerAttributes::readSqlite(const std::string &db_name)
{
int result;
sqlite3 *db;
sqlite3_stmt *stmt_get_player_pos;
SQLOK(open_v2(db_name.c_str(), &db, SQLITE_OPEN_READONLY |
SQLITE_OPEN_PRIVATECACHE, 0))
SQLOK(prepare_v2(db,
"SELECT name, posX, posY, posZ FROM player",
-1, &stmt_get_player_pos, NULL))
while ((result = sqlite3_step(stmt_get_player_pos)) != SQLITE_DONE) {
if (result == SQLITE_BUSY) { // Wait some time and try again
usleep(10000);
} else if (result != SQLITE_ROW) {
throw std::runtime_error(sqlite3_errmsg(db));
}
Player player;
const unsigned char *name_ = sqlite3_column_text(stmt_get_player_pos, 0);
player.name = reinterpret_cast<const char*>(name_);
player.x = sqlite3_column_double(stmt_get_player_pos, 1);
player.y = sqlite3_column_double(stmt_get_player_pos, 2);
player.z = sqlite3_column_double(stmt_get_player_pos, 3);
player.x /= 10.0f;
player.y /= 10.0f;
player.z /= 10.0f;
m_players.push_back(player);
}
sqlite3_finalize(stmt_get_player_pos);
sqlite3_close(db);
}
/**********/
PlayerAttributes::Players::const_iterator PlayerAttributes::begin() const
{
return m_players.cbegin();
}
PlayerAttributes::Players::const_iterator PlayerAttributes::end() const
{
return m_players.cend();
}

View File

@ -4,18 +4,16 @@ Minetest Mapper C++
.. image:: https://github.com/minetest/minetestmapper/workflows/build/badge.svg .. image:: https://github.com/minetest/minetestmapper/workflows/build/badge.svg
:target: https://github.com/minetest/minetestmapper/actions/workflows/build.yml :target: https://github.com/minetest/minetestmapper/actions/workflows/build.yml
Minetestmapper generates a top-down overview image from a Luanti map. Minetestmapper generates an overview image from a Luanti map.
A port of minetestmapper.py to C++ from `the obsolete Python script A port of minetestmapper.py to C++ from `the obsolete Python script
<https://github.com/minetest/minetest/tree/0.4.17/util>`_. <https://github.com/minetest/minetest/tree/0.4.17/util>`_.
This version is both faster and provides more features. This version is both faster and provides more features.
Minetestmapper ships with a colors.txt file suitable for Minetest Game, Minetestmapper ships with a colors.txt file for Minetest Game, if you use a different game or have
if you use a different game or have mods installed you should generate a many mods installed you should generate a matching colors.txt for better results.
matching colors.txt for better results (colors will be missing otherwise).
The `generate_colorstxt.py script The `generate_colorstxt.py script
<./util/generate_colorstxt.py>`_ in the util folder exists for this purpose, <./util/generate_colorstxt.py>`_ in the util folder exists for this purpose, detailed instructions can be found within.
detailed instructions can be found within.
Requirements Requirements
------------ ------------
@ -43,8 +41,7 @@ Minetestmapper for Windows can be downloaded `from the Releases section
<https://github.com/minetest/minetestmapper/releases>`_. <https://github.com/minetest/minetestmapper/releases>`_.
After extracting the archive, it can be invoked from cmd.exe or PowerShell: After extracting the archive, it can be invoked from cmd.exe or PowerShell:
::
.. code-block:: dos
cd C:\Users\yourname\Desktop\example\path cd C:\Users\yourname\Desktop\example\path
minetestmapper.exe --help minetestmapper.exe --help
@ -52,7 +49,7 @@ After extracting the archive, it can be invoked from cmd.exe or PowerShell:
Compilation Compilation
----------- -----------
.. code-block:: bash ::
cmake . -DENABLE_LEVELDB=1 cmake . -DENABLE_LEVELDB=1
make -j$(nproc) make -j$(nproc)
@ -60,8 +57,8 @@ Compilation
Usage Usage
----- -----
``minetestmapper`` has two mandatory paremeters, ``-i`` (input world path) `minetestmapper` has two mandatory paremeters, `-i` (input world path)
and ``-o`` (output image path). and `-o` (output image path).
:: ::
@ -93,7 +90,7 @@ draworigin:
Draw origin indicator, ``--draworigin`` Draw origin indicator, ``--draworigin``
drawalpha: drawalpha:
Allow nodes to be drawn with transparency (such as water), ``--drawalpha`` Allow nodes to be drawn with transparency (e.g. water), ``--drawalpha``
extent: extent:
Don't output any imagery, just print the extent of the full map, ``--extent`` Don't output any imagery, just print the extent of the full map, ``--extent``
@ -104,14 +101,11 @@ noshading:
noemptyimage: noemptyimage:
Don't output anything when the image would be empty, ``--noemptyimage`` Don't output anything when the image would be empty, ``--noemptyimage``
verbose:
Enable verbose log putput, ``--verbose``
min-y: min-y:
Don't draw nodes below this Y value, e.g. ``--min-y -25`` Don't draw nodes below this y value, e.g. ``--min-y -25``
max-y: max-y:
Don't draw nodes above this Y value, e.g. ``--max-y 75`` Don't draw nodes above this y value, e.g. ``--max-y 75``
backend: backend:
Override auto-detected map backend; supported: *sqlite3*, *leveldb*, *redis*, *postgresql*, e.g. ``--backend leveldb`` Override auto-detected map backend; supported: *sqlite3*, *leveldb*, *redis*, *postgresql*, e.g. ``--backend leveldb``
@ -119,10 +113,8 @@ backend:
geometry: geometry:
Limit area to specific geometry (*x:z+w+h* where x and z specify the lower left corner), e.g. ``--geometry -800:-800+1600+1600`` Limit area to specific geometry (*x:z+w+h* where x and z specify the lower left corner), e.g. ``--geometry -800:-800+1600+1600``
The coordinates are specified with the same axes as in-game. The Z axis becomes Y when projected on the image.
zoom: zoom:
Zoom the image by using more than one pixel per node, e.g. ``--zoom 4`` Apply zoom to drawn nodes by enlarging them to n*n squares, e.g. ``--zoom 4``
colors: colors:
Override auto-detected path to colors.txt, e.g. ``--colors ../world/mycolors.txt`` Override auto-detected path to colors.txt, e.g. ``--colors ../world/mycolors.txt``
@ -131,9 +123,6 @@ scales:
Draw scales on specified image edges (letters *t b l r* meaning top, bottom, left and right), e.g. ``--scales tbr`` Draw scales on specified image edges (letters *t b l r* meaning top, bottom, left and right), e.g. ``--scales tbr``
exhaustive: exhaustive:
Select if database should be traversed exhaustively or using range queries, available: *never*, *y*, *full*, *auto* | Select if database should be traversed exhaustively or using range queries, available: *never*, *y*, *full*, *auto*
| Defaults to *auto*. You shouldn't need to change this, but doing so can improve rendering times on large maps.
Defaults to *auto*. You shouldn't need to change this, as minetestmapper tries to automatically picks the best option. | For these optimizations to work it is important that you set ``min-y`` and ``max-y`` when you don't care about the world below e.g. -60 and above 1000 nodes.
dumpblock:
Instead of rendering anything try to load the block at the given position (*x,y,z*) and print its raw data as hexadecimal.

View File

@ -17,7 +17,6 @@
#include "BlockDecoder.h" #include "BlockDecoder.h"
#include "Image.h" #include "Image.h"
#include "util.h" #include "util.h"
#include "log.h"
#include "db-sqlite3.h" #include "db-sqlite3.h"
#if USE_POSTGRESQL #if USE_POSTGRESQL
@ -105,8 +104,8 @@ static Color parseColor(const std::string &color)
static Color mixColors(Color a, Color b) static Color mixColors(Color a, Color b)
{ {
Color result; Color result;
float a1 = a.a / 255.0f; double a1 = a.a / 255.0;
float a2 = b.a / 255.0f; double a2 = b.a / 255.0;
result.r = (int) (a1 * a.r + a2 * (1 - a1) * b.r); result.r = (int) (a1 * a.r + a2 * (1 - a1) * b.r);
result.g = (int) (a1 * a.g + a2 * (1 - a1) * b.g); result.g = (int) (a1 * a.g + a2 * (1 - a1) * b.g);
@ -154,8 +153,6 @@ TileGenerator::TileGenerator():
TileGenerator::~TileGenerator() TileGenerator::~TileGenerator()
{ {
closeDatabase(); closeDatabase();
delete m_image;
m_image = nullptr;
} }
void TileGenerator::setBgColor(const std::string &bgColor) void TileGenerator::setBgColor(const std::string &bgColor)
@ -258,7 +255,6 @@ void TileGenerator::parseColorsFile(const std::string &fileName)
std::ifstream in(fileName); std::ifstream in(fileName);
if (!in.good()) if (!in.good())
throw std::runtime_error("Specified colors file could not be found"); throw std::runtime_error("Specified colors file could not be found");
verbosestream << "Parsing colors.txt: " << fileName << std::endl;
parseColorsStream(in); parseColorsStream(in);
} }
@ -297,26 +293,19 @@ void TileGenerator::dumpBlock(const std::string &input_path, BlockPos pos)
void TileGenerator::generate(const std::string &input_path, const std::string &output) void TileGenerator::generate(const std::string &input_path, const std::string &output)
{ {
if (m_dontWriteEmpty) // FIXME: possible too, just needs to be done differently
setExhaustiveSearch(EXH_NEVER);
openDb(input_path); openDb(input_path);
loadBlocks(); loadBlocks();
// If we needed to load positions and there are none, that means the if (m_dontWriteEmpty && m_positions.empty())
// result will be empty. {
if (m_dontWriteEmpty && (m_exhaustiveSearch == EXH_NEVER || closeDatabase();
m_exhaustiveSearch == EXH_Y) && m_positions.empty()) {
verbosestream << "Result is empty (no positions)" << std::endl;
return; return;
} }
createImage(); createImage();
renderMap(); renderMap();
if (m_dontWriteEmpty && !m_renderedAny) {
verbosestream << "Result is empty (no pixels)" << std::endl;
printUnknown();
return;
}
closeDatabase(); closeDatabase();
if (m_drawScale) { if (m_drawScale) {
renderScale(); renderScale();
@ -350,7 +339,7 @@ void TileGenerator::parseColorsStream(std::istream &in)
unsigned int r, g, b, a = 255, t = 0; unsigned int r, g, b, a = 255, t = 0;
int items = sscanf(line, "%200s %u %u %u %u %u", name, &r, &g, &b, &a, &t); int items = sscanf(line, "%200s %u %u %u %u %u", name, &r, &g, &b, &a, &t);
if (items < 4) { if (items < 4) {
errorstream << "Failed to parse color entry '" << line << "'" << std::endl; std::cerr << "Failed to parse color entry '" << line << "'" << std::endl;
continue; continue;
} }
@ -376,59 +365,45 @@ std::set<std::string> TileGenerator::getSupportedBackends()
void TileGenerator::openDb(const std::string &input_path) void TileGenerator::openDb(const std::string &input_path)
{ {
if (dir_exists(input_path.c_str())) {
// ok
} else if (file_exists(input_path.c_str())) {
throw std::runtime_error("Input path is a file, it should point to the world folder instead");
} else {
throw std::runtime_error("Input path does not exist");
}
std::string input = input_path; std::string input = input_path;
if (input.back() != PATH_SEPARATOR) if (input.back() != PATH_SEPARATOR)
input += PATH_SEPARATOR; input += PATH_SEPARATOR;
std::ifstream ifs(input + "world.mt");
std::string backend = m_backend; std::string backend = m_backend;
if (backend.empty() && !ifs.good()) { if (backend.empty()) {
std::ifstream ifs(input + "world.mt");
if(!ifs.good())
throw std::runtime_error("Failed to open world.mt"); throw std::runtime_error("Failed to open world.mt");
} else if (backend.empty()) {
backend = read_setting_default("backend", ifs, "sqlite3"); backend = read_setting_default("backend", ifs, "sqlite3");
ifs.close();
} }
if (backend == "dummy") { if (backend == "sqlite3")
throw std::runtime_error("This map uses the dummy backend and contains no data");
} else if (backend == "sqlite3") {
m_db = new DBSQLite3(input); m_db = new DBSQLite3(input);
#if USE_POSTGRESQL #if USE_POSTGRESQL
} else if (backend == "postgresql") { else if (backend == "postgresql")
m_db = new DBPostgreSQL(input); m_db = new DBPostgreSQL(input);
#endif #endif
#if USE_LEVELDB #if USE_LEVELDB
} else if (backend == "leveldb") { else if (backend == "leveldb")
m_db = new DBLevelDB(input); m_db = new DBLevelDB(input);
#endif #endif
#if USE_REDIS #if USE_REDIS
} else if (backend == "redis") { else if (backend == "redis")
m_db = new DBRedis(input); m_db = new DBRedis(input);
#endif #endif
} else { else
throw std::runtime_error(std::string("Unknown map backend: ") + backend); throw std::runtime_error(std::string("Unknown map backend: ") + backend);
}
if (!read_setting_default("readonly_backend", ifs, "").empty()) {
errorstream << "Warning: Map with readonly_backend is not supported. "
"The result may be incomplete." << std::endl;
}
// Determine how we're going to traverse the database (heuristic) // Determine how we're going to traverse the database (heuristic)
if (m_exhaustiveSearch == EXH_AUTO) { if (m_exhaustiveSearch == EXH_AUTO) {
size_t y_range = (m_yMax / 16 + 1) - (m_yMin / 16); size_t y_range = (m_yMax / 16 + 1) - (m_yMin / 16);
size_t blocks = sat_mul<size_t>(m_geomX2 - m_geomX, y_range, m_geomY2 - m_geomY); size_t blocks = sat_mul<size_t>(m_geomX2 - m_geomX, y_range, m_geomY2 - m_geomY);
verbosestream << "Heuristic parameters:" #ifndef NDEBUG
std::cerr << "Heuristic parameters:"
<< " preferRangeQueries()=" << m_db->preferRangeQueries() << " preferRangeQueries()=" << m_db->preferRangeQueries()
<< " y_range=" << y_range << " blocks=" << blocks << std::endl; << " y_range=" << y_range << " blocks=" << blocks << std::endl;
#endif
if (m_db->preferRangeQueries()) if (m_db->preferRangeQueries())
m_exhaustiveSearch = EXH_NEVER; m_exhaustiveSearch = EXH_NEVER;
else if (blocks < 200000) else if (blocks < 200000)
@ -439,8 +414,8 @@ void TileGenerator::openDb(const std::string &input_path)
m_exhaustiveSearch = EXH_NEVER; m_exhaustiveSearch = EXH_NEVER;
} else if (m_exhaustiveSearch == EXH_FULL || m_exhaustiveSearch == EXH_Y) { } else if (m_exhaustiveSearch == EXH_FULL || m_exhaustiveSearch == EXH_Y) {
if (m_db->preferRangeQueries()) { if (m_db->preferRangeQueries()) {
errorstream << "Note: The current database backend supports efficient " std::cerr << "Note: The current database backend supports efficient "
"range queries, forcing exhaustive search will generally result " "range queries, forcing exhaustive search should always result "
" in worse performance." << std::endl; " in worse performance." << std::endl;
} }
} }
@ -466,20 +441,26 @@ void TileGenerator::loadBlocks()
const int16_t yMin = mod16(m_yMin); const int16_t yMin = mod16(m_yMin);
if (m_exhaustiveSearch == EXH_NEVER || m_exhaustiveSearch == EXH_Y) { if (m_exhaustiveSearch == EXH_NEVER || m_exhaustiveSearch == EXH_Y) {
std::vector<BlockPos> vec = m_db->getBlockPosXZ( std::vector<BlockPos> vec = m_db->getBlockPos(
BlockPos(m_geomX, yMin, m_geomY), BlockPos(m_geomX, yMin, m_geomY),
BlockPos(m_geomX2, yMax, m_geomY2) BlockPos(m_geomX2, yMax, m_geomY2)
); );
for (auto pos : vec) { for (auto pos : vec) {
assert(pos.x >= m_geomX && pos.x < m_geomX2); assert(pos.x >= m_geomX && pos.x < m_geomX2);
assert(pos.y >= yMin && pos.y < yMax);
assert(pos.z >= m_geomY && pos.z < m_geomY2); assert(pos.z >= m_geomY && pos.z < m_geomY2);
// Adjust minimum and maximum positions to the nearest block // Adjust minimum and maximum positions to the nearest block
m_xMin = mymin<int>(m_xMin, pos.x); if (pos.x < m_xMin)
m_xMax = mymax<int>(m_xMax, pos.x); m_xMin = pos.x;
m_zMin = mymin<int>(m_zMin, pos.z); if (pos.x > m_xMax)
m_zMax = mymax<int>(m_zMax, pos.z); m_xMax = pos.x;
if (pos.z < m_zMin)
m_zMin = pos.z;
if (pos.z > m_zMax)
m_zMax = pos.z;
m_positions[pos.z].emplace(pos.x); m_positions[pos.z].emplace(pos.x);
} }
@ -488,8 +469,10 @@ void TileGenerator::loadBlocks()
for (const auto &it : m_positions) for (const auto &it : m_positions)
count += it.second.size(); count += it.second.size();
m_progressMax = count; m_progressMax = count;
verbosestream << "Loaded " << count #ifndef NDEBUG
std::cerr << "Loaded " << count
<< " positions (across Z: " << m_positions.size() << ") for rendering" << std::endl; << " positions (across Z: " << m_positions.size() << ") for rendering" << std::endl;
#endif
} }
} }
@ -528,11 +511,8 @@ void TileGenerator::createImage()
image_height += (m_scales & SCALE_BOTTOM) ? scale_d : 0; image_height += (m_scales & SCALE_BOTTOM) ? scale_d : 0;
if(image_width > 4096 || image_height > 4096) { if(image_width > 4096 || image_height > 4096) {
errorstream << "Warning: The side length of the image to be created exceeds 4096 pixels!" std::cerr << "Warning: The width or height of the image to be created exceeds 4096 pixels!"
<< " (dimensions: " << image_width << "x" << image_height << ")" << " (Dimensions: " << image_width << "x" << image_height << ")"
<< std::endl;
} else {
verbosestream << "Creating image with size " << image_width << "x" << image_height
<< std::endl; << std::endl;
} }
m_image = new Image(image_width, image_height); m_image = new Image(image_width, image_height);
@ -597,8 +577,10 @@ void TileGenerator::renderMap()
postRenderRow(zPos); postRenderRow(zPos);
} }
} else if (m_exhaustiveSearch == EXH_Y) { } else if (m_exhaustiveSearch == EXH_Y) {
verbosestream << "Exhaustively searching height of " #ifndef NDEBUG
std::cerr << "Exhaustively searching height of "
<< (yMax - yMin) << " blocks" << std::endl; << (yMax - yMin) << " blocks" << std::endl;
#endif
std::vector<BlockPos> positions; std::vector<BlockPos> positions;
positions.reserve(yMax - yMin); positions.reserve(yMax - yMin);
for (auto it = m_positions.rbegin(); it != m_positions.rend(); ++it) { for (auto it = m_positions.rbegin(); it != m_positions.rend(); ++it) {
@ -622,9 +604,11 @@ void TileGenerator::renderMap()
} else if (m_exhaustiveSearch == EXH_FULL) { } else if (m_exhaustiveSearch == EXH_FULL) {
const size_t span_y = yMax - yMin; const size_t span_y = yMax - yMin;
m_progressMax = (m_geomX2 - m_geomX) * span_y * (m_geomY2 - m_geomY); m_progressMax = (m_geomX2 - m_geomX) * span_y * (m_geomY2 - m_geomY);
verbosestream << "Exhaustively searching " #ifndef NDEBUG
std::cerr << "Exhaustively searching "
<< (m_geomX2 - m_geomX) << "x" << span_y << "x" << (m_geomX2 - m_geomX) << "x" << span_y << "x"
<< (m_geomY2 - m_geomY) << " blocks" << std::endl; << (m_geomY2 - m_geomY) << " blocks" << std::endl;
#endif
std::vector<BlockPos> positions; std::vector<BlockPos> positions;
positions.reserve(span_y); positions.reserve(span_y);
@ -852,8 +836,8 @@ void TileGenerator::renderPlayers(const std::string &input_path)
PlayerAttributes players(input); PlayerAttributes players(input);
for (auto &player : players) { for (auto &player : players) {
if (player.x < m_xMin * 16 || player.x >= (m_xMax+1) * 16 || if (player.x < m_xMin * 16 || player.x > m_xMax * 16 ||
player.z < m_zMin * 16 || player.z >= (m_zMax+1) * 16) player.z < m_zMin * 16 || player.z > m_zMax * 16)
continue; continue;
if (player.y < m_yMin || player.y > m_yMax) if (player.y < m_yMin || player.y > m_yMax)
continue; continue;
@ -862,7 +846,6 @@ void TileGenerator::renderPlayers(const std::string &input_path)
m_image->drawFilledRect(imageX - 1, imageY, 3, 1, m_playerColor); m_image->drawFilledRect(imageX - 1, imageY, 3, 1, m_playerColor);
m_image->drawFilledRect(imageX, imageY - 1, 1, 3, m_playerColor); m_image->drawFilledRect(imageX, imageY - 1, 1, 3, m_playerColor);
assert(!player.name.empty());
m_image->drawText(imageX + 2, imageY, player.name, m_playerColor); m_image->drawText(imageX + 2, imageY, player.name, m_playerColor);
} }
} }
@ -878,15 +861,14 @@ void TileGenerator::printUnknown()
{ {
if (m_unknownNodes.empty()) if (m_unknownNodes.empty())
return; return;
errorstream << "Unknown nodes:\n"; std::cerr << "Unknown nodes:" << std::endl;
for (const auto &node : m_unknownNodes) for (const auto &node : m_unknownNodes)
errorstream << "\t" << node << '\n'; std::cerr << "\t" << node << std::endl;
if (!m_renderedAny) { if (!m_renderedAny) {
errorstream << "The map was read successfully and not empty, but none of the " std::cerr << "The map was read successfully and not empty, but none of the "
"encountered nodes had a color associated.\nCheck that you're using " "encountered nodes had a color associated.\nCheck that you're using "
"the right colors.txt. It should match the game you have installed.\n"; "the right colors.txt. It should match the game you have installed." << std::endl;
} }
errorstream << std::flush;
} }
void TileGenerator::reportProgress(size_t count) void TileGenerator::reportProgress(size_t count)

View File

@ -1,16 +1,6 @@
#include <cstdint>
#include "ZlibDecompressor.h"
#include "config.h"
// for convenient usage of both
#if USE_ZLIB_NG
#include <zlib-ng.h>
#define z_stream zng_stream
#define Z(x) zng_ ## x
#else
#include <zlib.h> #include <zlib.h>
#define Z(x) x #include <stdint.h>
#endif #include "ZlibDecompressor.h"
ZlibDecompressor::ZlibDecompressor(const u8 *data, size_t size): ZlibDecompressor::ZlibDecompressor(const u8 *data, size_t size):
m_data(data), m_data(data),
@ -28,13 +18,18 @@ void ZlibDecompressor::setSeekPos(size_t seekPos)
m_seekPos = seekPos; m_seekPos = seekPos;
} }
void ZlibDecompressor::decompress(ustring &buffer) size_t ZlibDecompressor::seekPos() const
{
return m_seekPos;
}
ustring ZlibDecompressor::decompress()
{ {
const unsigned char *data = m_data + m_seekPos; const unsigned char *data = m_data + m_seekPos;
const size_t size = m_size - m_seekPos; const size_t size = m_size - m_seekPos;
// output space is extended in chunks of this size ustring buffer;
constexpr size_t BUFSIZE = 8 * 1024; constexpr size_t BUFSIZE = 32 * 1024;
z_stream strm; z_stream strm;
strm.zalloc = Z_NULL; strm.zalloc = Z_NULL;
@ -43,22 +38,21 @@ void ZlibDecompressor::decompress(ustring &buffer)
strm.next_in = Z_NULL; strm.next_in = Z_NULL;
strm.avail_in = 0; strm.avail_in = 0;
if (Z(inflateInit)(&strm) != Z_OK) if (inflateInit(&strm) != Z_OK)
throw DecompressError(); throw DecompressError();
strm.next_in = const_cast<unsigned char *>(data); strm.next_in = const_cast<unsigned char *>(data);
strm.avail_in = size; strm.avail_in = size;
if (buffer.empty())
buffer.resize(BUFSIZE); buffer.resize(BUFSIZE);
strm.next_out = &buffer[0]; strm.next_out = &buffer[0];
strm.avail_out = buffer.size(); strm.avail_out = BUFSIZE;
int ret = 0; int ret = 0;
do { do {
ret = Z(inflate)(&strm, Z_NO_FLUSH); ret = inflate(&strm, Z_NO_FLUSH);
if (strm.avail_out == 0) { if (strm.avail_out == 0) {
const auto off = buffer.size(); const auto off = buffer.size();
buffer.resize(off + BUFSIZE); buffer.reserve(off + BUFSIZE);
strm.next_out = &buffer[off]; strm.next_out = &buffer[off];
strm.avail_out = BUFSIZE; strm.avail_out = BUFSIZE;
} }
@ -68,6 +62,8 @@ void ZlibDecompressor::decompress(ustring &buffer)
m_seekPos += strm.next_in - data; m_seekPos += strm.next_in - data;
buffer.resize(buffer.size() - strm.avail_out); buffer.resize(buffer.size() - strm.avail_out);
(void) Z(inflateEnd)(&strm); (void) inflateEnd(&strm);
return buffer;
} }

View File

@ -21,15 +21,19 @@ void ZstdDecompressor::setData(const u8 *data, size_t size, size_t seekPos)
m_size = size; m_size = size;
} }
void ZstdDecompressor::decompress(ustring &buffer) std::size_t ZstdDecompressor::seekPos() const
{
return m_seekPos;
}
ustring ZstdDecompressor::decompress()
{ {
ZSTD_DStream *stream = reinterpret_cast<ZSTD_DStream*>(m_stream); ZSTD_DStream *stream = reinterpret_cast<ZSTD_DStream*>(m_stream);
ZSTD_inBuffer inbuf = { m_data, m_size, m_seekPos }; ZSTD_inBuffer inbuf = { m_data, m_size, m_seekPos };
// output space is extended in chunks of this size ustring buffer;
constexpr size_t BUFSIZE = 8 * 1024; constexpr size_t BUFSIZE = 32 * 1024;
if (buffer.empty())
buffer.resize(BUFSIZE); buffer.resize(BUFSIZE);
ZSTD_outBuffer outbuf = { &buffer[0], buffer.size(), 0 }; ZSTD_outBuffer outbuf = { &buffer[0], buffer.size(), 0 };
@ -38,15 +42,17 @@ void ZstdDecompressor::decompress(ustring &buffer)
size_t ret; size_t ret;
do { do {
ret = ZSTD_decompressStream(stream, &outbuf, &inbuf); ret = ZSTD_decompressStream(stream, &outbuf, &inbuf);
if (ret && ZSTD_isError(ret))
throw DecompressError();
if (outbuf.size == outbuf.pos) { if (outbuf.size == outbuf.pos) {
outbuf.size += BUFSIZE; outbuf.size += BUFSIZE;
buffer.resize(outbuf.size); buffer.resize(outbuf.size);
outbuf.dst = &buffer[0]; outbuf.dst = &buffer[0];
} }
if (ret && ZSTD_isError(ret))
throw DecompressError();
} while (ret != 0); } while (ret != 0);
m_seekPos = inbuf.pos; m_seekPos = inbuf.pos;
buffer.resize(outbuf.pos); buffer.resize(outbuf.pos);
return buffer;
} }

View File

@ -1,6 +1,5 @@
#include <stdexcept> #include <stdexcept>
#include <sstream> #include <sstream>
#include <algorithm>
#include "db-leveldb.h" #include "db-leveldb.h"
#include "types.h" #include "types.h"
@ -19,12 +18,6 @@ static inline std::string i64tos(int64_t i)
return os.str(); return os.str();
} }
// finds the first position in the list where it.x >= x
#define lower_bound_x(container, find_x) \
std::lower_bound((container).begin(), (container).end(), (find_x), \
[] (const vec2 &left, int16_t right) { \
return left.x < right; \
})
DBLevelDB::DBLevelDB(const std::string &mapdir) DBLevelDB::DBLevelDB(const std::string &mapdir)
{ {
@ -32,7 +25,7 @@ DBLevelDB::DBLevelDB(const std::string &mapdir)
options.create_if_missing = false; options.create_if_missing = false;
leveldb::Status status = leveldb::DB::Open(options, mapdir + "map.db", &db); leveldb::Status status = leveldb::DB::Open(options, mapdir + "map.db", &db);
if (!status.ok()) { if (!status.ok()) {
throw std::runtime_error(std::string("Failed to open database: ") + status.ToString()); throw std::runtime_error(std::string("Failed to open Database: ") + status.ToString());
} }
/* LevelDB is a dumb key-value store, so the only optimization we can do /* LevelDB is a dumb key-value store, so the only optimization we can do
@ -48,24 +41,18 @@ DBLevelDB::~DBLevelDB()
} }
std::vector<BlockPos> DBLevelDB::getBlockPosXZ(BlockPos min, BlockPos max) std::vector<BlockPos> DBLevelDB::getBlockPos(BlockPos min, BlockPos max)
{ {
std::vector<BlockPos> res; std::vector<BlockPos> res;
for (const auto &it : posCache) { for (const auto &it : posCache) {
const int16_t zpos = it.first; if (it.first < min.z || it.first >= max.z)
if (zpos < min.z || zpos >= max.z)
continue; continue;
auto it2 = lower_bound_x(it.second, min.x); for (auto pos2 : it.second) {
for (; it2 != it.second.end(); it2++) { if (pos2.first < min.x || pos2.first >= max.x)
const auto &pos2 = *it2;
if (pos2.x >= max.x)
break; // went past
if (pos2.y < min.y || pos2.y >= max.y)
continue; continue;
// skip duplicates if (pos2.second < min.y || pos2.second >= max.y)
if (!res.empty() && res.back().x == pos2.x && res.back().z == zpos)
continue; continue;
res.emplace_back(pos2.x, pos2.y, zpos); res.emplace_back(pos2.first, pos2.second, it.first);
} }
} }
return res; return res;
@ -82,9 +69,6 @@ void DBLevelDB::loadPosCache()
posCache[pos.z].emplace_back(pos.x, pos.y); posCache[pos.z].emplace_back(pos.x, pos.y);
} }
delete it; delete it;
for (auto &it : posCache)
std::sort(it.second.begin(), it.second.end());
} }
@ -97,18 +81,13 @@ void DBLevelDB::getBlocksOnXZ(BlockList &blocks, int16_t x, int16_t z,
auto it = posCache.find(z); auto it = posCache.find(z);
if (it == posCache.cend()) if (it == posCache.cend())
return; return;
auto it2 = lower_bound_x(it->second, x); for (auto pos2 : it->second) {
if (it2 == it->second.end() || it2->x != x) if (pos2.first != x)
return; continue;
// it2 is now pointing to a contigous part where it2->x == x if (pos2.second < min_y || pos2.second >= max_y)
for (; it2 != it->second.end(); it2++) {
const auto &pos2 = *it2;
if (pos2.x != x)
break; // went past
if (pos2.y < min_y || pos2.y >= max_y)
continue; continue;
BlockPos pos(x, pos2.y, z); BlockPos pos(x, pos2.second, z);
status = db->Get(leveldb::ReadOptions(), i64tos(encodeBlockPos(pos)), &datastr); status = db->Get(leveldb::ReadOptions(), i64tos(encodeBlockPos(pos)), &datastr);
if (status.ok()) { if (status.ok()) {
blocks.emplace_back( blocks.emplace_back(

View File

@ -3,71 +3,11 @@
#include <fstream> #include <fstream>
#include <cstdlib> #include <cstdlib>
#include <arpa/inet.h> #include <arpa/inet.h>
#include "db-postgresql.h" #include "db-postgresql.h"
#include "util.h" #include "util.h"
#include "log.h"
#include "types.h" #include "types.h"
/* PostgreSQLBase */ #define ARRLEN(x) (sizeof(x) / sizeof((x)[0]))
PostgreSQLBase::~PostgreSQLBase()
{
if (db)
PQfinish(db);
}
void PostgreSQLBase::openDatabase(const char *connect_string)
{
if (db)
throw std::logic_error("Database already open");
db = PQconnectdb(connect_string);
if (PQstatus(db) != CONNECTION_OK) {
throw std::runtime_error(std::string("PostgreSQL database error: ") +
PQerrorMessage(db)
);
}
}
PGresult *PostgreSQLBase::checkResults(PGresult *res, bool clear)
{
ExecStatusType statusType = PQresultStatus(res);
switch (statusType) {
case PGRES_COMMAND_OK:
case PGRES_TUPLES_OK:
break;
case PGRES_FATAL_ERROR:
throw std::runtime_error(
std::string("PostgreSQL database error: ") +
PQresultErrorMessage(res)
);
default:
throw std::runtime_error(
std::string("Unhandled PostgreSQL result code ") +
std::to_string(statusType)
);
}
if (clear)
PQclear(res);
return res;
}
PGresult *PostgreSQLBase::execPrepared(
const char *stmtName, const int paramsNumber,
const void **params,
const int *paramsLengths, const int *paramsFormats,
bool clear)
{
return checkResults(PQexecPrepared(db, stmtName, paramsNumber,
(const char* const*) params, paramsLengths, paramsFormats,
1 /* binary output */), clear
);
}
/* DBPostgreSQL */
DBPostgreSQL::DBPostgreSQL(const std::string &mapdir) DBPostgreSQL::DBPostgreSQL(const std::string &mapdir)
{ {
@ -76,15 +16,21 @@ DBPostgreSQL::DBPostgreSQL(const std::string &mapdir)
throw std::runtime_error("Failed to read world.mt"); throw std::runtime_error("Failed to read world.mt");
std::string connect_string = read_setting("pgsql_connection", ifs); std::string connect_string = read_setting("pgsql_connection", ifs);
ifs.close(); ifs.close();
db = PQconnectdb(connect_string.c_str());
openDatabase(connect_string.c_str()); if (PQstatus(db) != CONNECTION_OK) {
throw std::runtime_error(std::string(
"PostgreSQL database error: ") +
PQerrorMessage(db)
);
}
prepareStatement( prepareStatement(
"get_block_pos", "get_block_pos",
"SELECT posX::int4, posZ::int4 FROM blocks WHERE" "SELECT posX::int4, posY::int4, posZ::int4 FROM blocks WHERE"
" (posX BETWEEN $1::int4 AND $2::int4) AND" " (posX BETWEEN $1::int4 AND $2::int4) AND"
" (posY BETWEEN $3::int4 AND $4::int4) AND" " (posY BETWEEN $3::int4 AND $4::int4) AND"
" (posZ BETWEEN $5::int4 AND $6::int4) GROUP BY posX, posZ" " (posZ BETWEEN $5::int4 AND $6::int4)"
); );
prepareStatement( prepareStatement(
"get_blocks", "get_blocks",
@ -108,12 +54,13 @@ DBPostgreSQL::~DBPostgreSQL()
try { try {
checkResults(PQexec(db, "COMMIT;")); checkResults(PQexec(db, "COMMIT;"));
} catch (const std::exception& caught) { } catch (const std::exception& caught) {
errorstream << "could not finalize: " << caught.what() << std::endl; std::cerr << "could not finalize: " << caught.what() << std::endl;
} }
PQfinish(db);
} }
std::vector<BlockPos> DBPostgreSQL::getBlockPosXZ(BlockPos min, BlockPos max) std::vector<BlockPos> DBPostgreSQL::getBlockPos(BlockPos min, BlockPos max)
{ {
int32_t const x1 = htonl(min.x); int32_t const x1 = htonl(min.x);
int32_t const x2 = htonl(max.x - 1); int32_t const x2 = htonl(max.x - 1);
@ -136,14 +83,11 @@ std::vector<BlockPos> DBPostgreSQL::getBlockPosXZ(BlockPos min, BlockPos max)
std::vector<BlockPos> positions; std::vector<BlockPos> positions;
positions.reserve(numrows); positions.reserve(numrows);
BlockPos pos; for (int row = 0; row < numrows; ++row)
for (int row = 0; row < numrows; ++row) { positions.emplace_back(pg_to_blockpos(results, row, 0));
pos.x = pg_binary_to_int(results, row, 0);
pos.z = pg_binary_to_int(results, row, 1);
positions.push_back(pos);
}
PQclear(results); PQclear(results);
return positions; return positions;
} }
@ -222,8 +166,61 @@ void DBPostgreSQL::getBlocksByPos(BlockList &blocks,
} }
} }
PGresult *DBPostgreSQL::checkResults(PGresult *res, bool clear)
{
ExecStatusType statusType = PQresultStatus(res);
switch (statusType) {
case PGRES_COMMAND_OK:
case PGRES_TUPLES_OK:
break;
case PGRES_FATAL_ERROR:
throw std::runtime_error(
std::string("PostgreSQL database error: ") +
PQresultErrorMessage(res)
);
default:
throw std::runtime_error(
"Unhandled PostgreSQL result code"
);
}
if (clear)
PQclear(res);
return res;
}
void DBPostgreSQL::prepareStatement(const std::string &name, const std::string &sql)
{
checkResults(PQprepare(db, name.c_str(), sql.c_str(), 0, NULL));
}
PGresult *DBPostgreSQL::execPrepared(
const char *stmtName, const int paramsNumber,
const void **params,
const int *paramsLengths, const int *paramsFormats,
bool clear
)
{
return checkResults(PQexecPrepared(db, stmtName, paramsNumber,
(const char* const*) params, paramsLengths, paramsFormats,
1 /* binary output */), clear
);
}
int DBPostgreSQL::pg_binary_to_int(PGresult *res, int row, int col) int DBPostgreSQL::pg_binary_to_int(PGresult *res, int row, int col)
{ {
int32_t* raw = reinterpret_cast<int32_t*>(PQgetvalue(res, row, col)); int32_t* raw = reinterpret_cast<int32_t*>(PQgetvalue(res, row, col));
return ntohl(*raw); return ntohl(*raw);
} }
BlockPos DBPostgreSQL::pg_to_blockpos(PGresult *res, int row, int col)
{
BlockPos result;
result.x = pg_binary_to_int(res, row, col);
result.y = pg_binary_to_int(res, row, col + 1);
result.z = pg_binary_to_int(res, row, col + 2);
return result;
}

View File

@ -68,7 +68,7 @@ DBRedis::~DBRedis()
} }
std::vector<BlockPos> DBRedis::getBlockPosXZ(BlockPos min, BlockPos max) std::vector<BlockPos> DBRedis::getBlockPos(BlockPos min, BlockPos max)
{ {
std::vector<BlockPos> res; std::vector<BlockPos> res;
for (const auto &it : posCache) { for (const auto &it : posCache) {

196
db-sqlite3.cpp Normal file
View File

@ -0,0 +1,196 @@
#include <stdexcept>
#include <unistd.h> // for usleep
#include <iostream>
#include <algorithm>
#include <time.h>
#include "db-sqlite3.h"
#include "types.h"
#define SQLRES(f, good) \
result = (sqlite3_##f);\
if (result != good) {\
throw std::runtime_error(sqlite3_errmsg(db));\
}
#define SQLOK(f) SQLRES(f, SQLITE_OK)
DBSQLite3::DBSQLite3(const std::string &mapdir)
{
int result;
std::string db_name = mapdir + "map.sqlite";
SQLOK(open_v2(db_name.c_str(), &db, SQLITE_OPEN_READONLY |
SQLITE_OPEN_PRIVATECACHE, 0))
SQLOK(prepare_v2(db,
"SELECT pos, data FROM blocks WHERE pos BETWEEN ? AND ?",
-1, &stmt_get_blocks_z, NULL))
SQLOK(prepare_v2(db,
"SELECT data FROM blocks WHERE pos = ?",
-1, &stmt_get_block_exact, NULL))
SQLOK(prepare_v2(db,
"SELECT pos FROM blocks",
-1, &stmt_get_block_pos, NULL))
SQLOK(prepare_v2(db,
"SELECT pos FROM blocks WHERE pos BETWEEN ? AND ?",
-1, &stmt_get_block_pos_z, NULL))
}
DBSQLite3::~DBSQLite3()
{
sqlite3_finalize(stmt_get_blocks_z);
sqlite3_finalize(stmt_get_block_pos);
sqlite3_finalize(stmt_get_block_pos_z);
sqlite3_finalize(stmt_get_block_exact);
if (sqlite3_close(db) != SQLITE_OK) {
std::cerr << "Error closing SQLite database." << std::endl;
};
}
inline void DBSQLite3::getPosRange(int64_t &min, int64_t &max, int16_t zPos,
int16_t zPos2) const
{
/* The range of block positions is [-2048, 2047], which turns into [0, 4095]
* when casted to unsigned. This didn't actually help me understand the
* numbers below, but I wanted to write it down.
*/
// Magic numbers!
min = encodeBlockPos(BlockPos(0, -2048, zPos));
max = encodeBlockPos(BlockPos(0, 2048, zPos2)) - 1;
}
std::vector<BlockPos> DBSQLite3::getBlockPos(BlockPos min, BlockPos max)
{
int result;
sqlite3_stmt *stmt;
if(min.z <= -2048 && max.z >= 2048) {
stmt = stmt_get_block_pos;
} else {
stmt = stmt_get_block_pos_z;
int64_t minPos, maxPos;
if (min.z < -2048)
min.z = -2048;
if (max.z > 2048)
max.z = 2048;
getPosRange(minPos, maxPos, min.z, max.z - 1);
SQLOK(bind_int64(stmt, 1, minPos))
SQLOK(bind_int64(stmt, 2, maxPos))
}
std::vector<BlockPos> positions;
while ((result = sqlite3_step(stmt)) != SQLITE_DONE) {
if (result == SQLITE_BUSY) { // Wait some time and try again
usleep(10000);
} else if (result != SQLITE_ROW) {
throw std::runtime_error(sqlite3_errmsg(db));
}
int64_t posHash = sqlite3_column_int64(stmt, 0);
BlockPos pos = decodeBlockPos(posHash);
if(pos.x >= min.x && pos.x < max.x && pos.y >= min.y && pos.y < max.y)
positions.emplace_back(pos);
}
SQLOK(reset(stmt));
return positions;
}
void DBSQLite3::loadBlockCache(int16_t zPos)
{
int result;
blockCache.clear();
int64_t minPos, maxPos;
getPosRange(minPos, maxPos, zPos, zPos);
SQLOK(bind_int64(stmt_get_blocks_z, 1, minPos));
SQLOK(bind_int64(stmt_get_blocks_z, 2, maxPos));
while ((result = sqlite3_step(stmt_get_blocks_z)) != SQLITE_DONE) {
if (result == SQLITE_BUSY) { // Wait some time and try again
usleep(10000);
} else if (result != SQLITE_ROW) {
throw std::runtime_error(sqlite3_errmsg(db));
}
int64_t posHash = sqlite3_column_int64(stmt_get_blocks_z, 0);
BlockPos pos = decodeBlockPos(posHash);
const unsigned char *data = reinterpret_cast<const unsigned char *>(
sqlite3_column_blob(stmt_get_blocks_z, 1));
size_t size = sqlite3_column_bytes(stmt_get_blocks_z, 1);
blockCache[pos.x].emplace_back(pos, ustring(data, size));
}
SQLOK(reset(stmt_get_blocks_z))
}
void DBSQLite3::getBlocksOnXZ(BlockList &blocks, int16_t x, int16_t z,
int16_t min_y, int16_t max_y)
{
/* Cache the blocks on the given Z coordinate between calls, this only
* works due to order in which the TileGenerator asks for blocks. */
if (z != blockCachedZ) {
loadBlockCache(z);
blockCachedZ = z;
}
auto it = blockCache.find(x);
if (it == blockCache.end())
return;
if (it->second.empty()) {
/* We have swapped this list before, this is not supposed to happen
* because it's bad for performance. But rather than silently breaking
* do the right thing and load the blocks again. */
#ifndef NDEBUG
std::cerr << "Warning: suboptimal access pattern for sqlite3 backend" << std::endl;
#endif
loadBlockCache(z);
}
// Swap lists to avoid copying contents
blocks.clear();
std::swap(blocks, it->second);
for (auto it = blocks.begin(); it != blocks.end(); ) {
if (it->first.y < min_y || it->first.y >= max_y)
it = blocks.erase(it);
else
it++;
}
}
void DBSQLite3::getBlocksByPos(BlockList &blocks,
const std::vector<BlockPos> &positions)
{
int result;
for (auto pos : positions) {
int64_t dbPos = encodeBlockPos(pos);
SQLOK(bind_int64(stmt_get_block_exact, 1, dbPos));
while ((result = sqlite3_step(stmt_get_block_exact)) == SQLITE_BUSY) {
usleep(10000); // Wait some time and try again
}
if (result == SQLITE_DONE) {
// no data
} else if (result != SQLITE_ROW) {
throw std::runtime_error(sqlite3_errmsg(db));
} else {
const unsigned char *data = reinterpret_cast<const unsigned char *>(
sqlite3_column_blob(stmt_get_block_exact, 0));
size_t size = sqlite3_column_bytes(stmt_get_block_exact, 0);
blocks.emplace_back(pos, ustring(data, size));
}
SQLOK(reset(stmt_get_block_exact))
}
}

View File

@ -23,7 +23,6 @@ private:
u8 m_version, m_contentWidth; u8 m_version, m_contentWidth;
ustring m_mapData; ustring m_mapData;
// cached allocations/instances for performance // one instance for performance
ZstdDecompressor m_zstd_decompressor; ZstdDecompressor m_zstd_decompressor;
ustring m_scratch;
}; };

View File

@ -19,5 +19,8 @@ public:
Players::const_iterator end() const; Players::const_iterator end() const;
private: private:
void readFiles(const std::string &playersPath);
void readSqlite(const std::string &db_name);
Players m_players; Players m_players;
}; };

View File

@ -11,10 +11,8 @@ public:
ZlibDecompressor(const u8 *data, size_t size); ZlibDecompressor(const u8 *data, size_t size);
~ZlibDecompressor(); ~ZlibDecompressor();
void setSeekPos(size_t seekPos); void setSeekPos(size_t seekPos);
size_t seekPos() const { return m_seekPos; } size_t seekPos() const;
// Decompress and return one zlib stream from the buffer ustring decompress();
// Advances seekPos as appropriate.
void decompress(ustring &dst);
private: private:
const u8 *m_data; const u8 *m_data;

View File

@ -11,10 +11,8 @@ public:
ZstdDecompressor(); ZstdDecompressor();
~ZstdDecompressor(); ~ZstdDecompressor();
void setData(const u8 *data, size_t size, size_t seekPos); void setData(const u8 *data, size_t size, size_t seekPos);
size_t seekPos() const { return m_seekPos; } size_t seekPos() const;
// Decompress and return one zstd stream from the buffer ustring decompress();
// Advances seekPos as appropriate.
void decompress(ustring &dst);
private: private:
void *m_stream; // ZSTD_DStream void *m_stream; // ZSTD_DStream

View File

@ -6,7 +6,6 @@
#cmakedefine01 USE_POSTGRESQL #cmakedefine01 USE_POSTGRESQL
#cmakedefine01 USE_LEVELDB #cmakedefine01 USE_LEVELDB
#cmakedefine01 USE_REDIS #cmakedefine01 USE_REDIS
#cmakedefine01 USE_ZLIB_NG
#define SHAREDIR "@SHAREDIR@" #define SHAREDIR "@SHAREDIR@"

11
include/config.h Normal file
View File

@ -0,0 +1,11 @@
#if MSDOS || __OS2__ || __NT__ || _WIN32
#define PATH_SEPARATOR '\\'
#else
#define PATH_SEPARATOR '/'
#endif
#ifdef USE_CMAKE_CONFIG_H
#include "cmake_config.h"
#else
#error missing config
#endif

View File

@ -8,7 +8,7 @@
class DBLevelDB : public DB { class DBLevelDB : public DB {
public: public:
DBLevelDB(const std::string &mapdir); DBLevelDB(const std::string &mapdir);
std::vector<BlockPos> getBlockPosXZ(BlockPos min, BlockPos max) override; std::vector<BlockPos> getBlockPos(BlockPos min, BlockPos max) override;
void getBlocksOnXZ(BlockList &blocks, int16_t x, int16_t z, void getBlocksOnXZ(BlockList &blocks, int16_t x, int16_t z,
int16_t min_y, int16_t max_y) override; int16_t min_y, int16_t max_y) override;
void getBlocksByPos(BlockList &blocks, void getBlocksByPos(BlockList &blocks,
@ -18,24 +18,11 @@ public:
bool preferRangeQueries() const override { return false; } bool preferRangeQueries() const override { return false; }
private: private:
struct vec2 { using pos2d = std::pair<int16_t, int16_t>;
int16_t x, y;
constexpr vec2() : x(0), y(0) {}
constexpr vec2(int16_t x, int16_t y) : x(x), y(y) {}
inline bool operator<(const vec2 &p) const
{
if (x < p.x)
return true;
if (x > p.x)
return false;
return y < p.y;
}
};
void loadPosCache(); void loadPosCache();
// indexed by Z, contains all (x,y) position pairs // indexed by Z, contains all (x,y) position pairs
std::unordered_map<int16_t, std::vector<vec2>> posCache; std::unordered_map<int16_t, std::vector<pos2d>> posCache;
leveldb::DB *db = NULL; leveldb::DB *db;
}; };

View File

@ -3,31 +3,10 @@
#include "db.h" #include "db.h"
#include <libpq-fe.h> #include <libpq-fe.h>
class PostgreSQLBase { class DBPostgreSQL : public DB {
public:
~PostgreSQLBase();
protected:
void openDatabase(const char *connect_string);
PGresult *checkResults(PGresult *res, bool clear = true);
void prepareStatement(const std::string &name, const std::string &sql) {
checkResults(PQprepare(db, name.c_str(), sql.c_str(), 0, NULL));
}
PGresult *execPrepared(
const char *stmtName, const int paramsNumber,
const void **params,
const int *paramsLengths = nullptr, const int *paramsFormats = nullptr,
bool clear = true
);
PGconn *db = NULL;
};
class DBPostgreSQL : public DB, PostgreSQLBase {
public: public:
DBPostgreSQL(const std::string &mapdir); DBPostgreSQL(const std::string &mapdir);
std::vector<BlockPos> getBlockPosXZ(BlockPos min, BlockPos max) override; std::vector<BlockPos> getBlockPos(BlockPos min, BlockPos max) override;
void getBlocksOnXZ(BlockList &blocks, int16_t x, int16_t z, void getBlocksOnXZ(BlockList &blocks, int16_t x, int16_t z,
int16_t min_y, int16_t max_y) override; int16_t min_y, int16_t max_y) override;
void getBlocksByPos(BlockList &blocks, void getBlocksByPos(BlockList &blocks,
@ -36,6 +15,18 @@ public:
bool preferRangeQueries() const override { return true; } bool preferRangeQueries() const override { return true; }
private: protected:
PGresult *checkResults(PGresult *res, bool clear = true);
void prepareStatement(const std::string &name, const std::string &sql);
PGresult *execPrepared(
const char *stmtName, const int paramsNumber,
const void **params,
const int *paramsLengths = nullptr, const int *paramsFormats = nullptr,
bool clear = true
);
int pg_binary_to_int(PGresult *res, int row, int col); int pg_binary_to_int(PGresult *res, int row, int col);
BlockPos pg_to_blockpos(PGresult *res, int row, int col);
private:
PGconn *db;
}; };

View File

@ -9,7 +9,7 @@
class DBRedis : public DB { class DBRedis : public DB {
public: public:
DBRedis(const std::string &mapdir); DBRedis(const std::string &mapdir);
std::vector<BlockPos> getBlockPosXZ(BlockPos min, BlockPos max) override; std::vector<BlockPos> getBlockPos(BlockPos min, BlockPos max) override;
void getBlocksOnXZ(BlockList &blocks, int16_t x, int16_t z, void getBlocksOnXZ(BlockList &blocks, int16_t x, int16_t z,
int16_t min_y, int16_t max_y) override; int16_t min_y, int16_t max_y) override;
void getBlocksByPos(BlockList &blocks, void getBlocksByPos(BlockList &blocks,

33
include/db-sqlite3.h Normal file
View File

@ -0,0 +1,33 @@
#pragma once
#include "db.h"
#include <unordered_map>
#include <sqlite3.h>
class DBSQLite3 : public DB {
public:
DBSQLite3(const std::string &mapdir);
std::vector<BlockPos> getBlockPos(BlockPos min, BlockPos max) override;
void getBlocksOnXZ(BlockList &blocks, int16_t x, int16_t z,
int16_t min_y, int16_t max_y) override;
void getBlocksByPos(BlockList &blocks,
const std::vector<BlockPos> &positions) override;
~DBSQLite3() override;
bool preferRangeQueries() const override { return false; }
private:
inline void getPosRange(int64_t &min, int64_t &max, int16_t zPos,
int16_t zPos2) const;
void loadBlockCache(int16_t zPos);
sqlite3 *db;
sqlite3_stmt *stmt_get_block_pos;
sqlite3_stmt *stmt_get_block_pos_z;
sqlite3_stmt *stmt_get_blocks_z;
sqlite3_stmt *stmt_get_block_exact;
int16_t blockCachedZ = -10000;
std::unordered_map<int16_t, BlockList> blockCache; // indexed by X
};

View File

@ -6,15 +6,18 @@
#include <utility> #include <utility>
#include "types.h" #include "types.h"
struct BlockPos {
int16_t x, y, z;
constexpr BlockPos() : x(0), y(0), z(0) {} struct BlockPos {
explicit constexpr BlockPos(int16_t v) : x(v), y(v), z(v) {} int16_t x;
constexpr BlockPos(int16_t x, int16_t y, int16_t z) : x(x), y(y), z(z) {} int16_t y;
int16_t z;
BlockPos() : x(0), y(0), z(0) {}
explicit BlockPos(int16_t v) : x(v), y(v), z(v) {}
BlockPos(int16_t x, int16_t y, int16_t z) : x(x), y(y), z(z) {}
// Implements the inverse ordering so that (2,2,2) < (1,1,1) // Implements the inverse ordering so that (2,2,2) < (1,1,1)
inline bool operator<(const BlockPos &p) const bool operator < (const BlockPos &p) const
{ {
if (z > p.z) if (z > p.z)
return true; return true;
@ -24,7 +27,11 @@ struct BlockPos {
return true; return true;
if (y < p.y) if (y < p.y)
return false; return false;
return x > p.x; if (x > p.x)
return true;
if (x < p.x)
return false;
return false;
} }
}; };
@ -36,32 +43,29 @@ typedef std::list<Block> BlockList;
class DB { class DB {
protected: protected:
// Helpers that implement the hashed positions used by most backends // Helpers that implement the hashed positions used by most backends
static inline int64_t encodeBlockPos(const BlockPos pos); inline int64_t encodeBlockPos(const BlockPos pos) const;
static inline BlockPos decodeBlockPos(int64_t hash); inline BlockPos decodeBlockPos(int64_t hash) const;
public: public:
/* Return all unique (X, Z) position pairs inside area given by min and max, /* Return all block positions inside the range given by min and max,
* so that min.x <= x < max.x && min.z <= z < max.z * so that min.x <= x < max.x, ...
* Note: duplicates are allowed, but results in wasted time.
*/ */
virtual std::vector<BlockPos> getBlockPosXZ(BlockPos min, BlockPos max) = 0; virtual std::vector<BlockPos> getBlockPos(BlockPos min, BlockPos max) = 0;
/* Read all blocks in column given by x and z /* Read all blocks in column given by x and z
* and inside the given Y range (min_y <= y < max_y) into list * and inside the given Y range (min_y <= y < max_y) into list
*/ */
virtual void getBlocksOnXZ(BlockList &blocks, int16_t x, int16_t z, virtual void getBlocksOnXZ(BlockList &blocks, int16_t x, int16_t z,
int16_t min_y, int16_t max_y) = 0; int16_t min_y, int16_t max_y) = 0;
/* Read blocks at given positions into list /* Read blocks at given positions into list
*/ */
virtual void getBlocksByPos(BlockList &blocks, virtual void getBlocksByPos(BlockList &blocks,
const std::vector<BlockPos> &positions) = 0; const std::vector<BlockPos> &positions) = 0;
/* Can this database efficiently do range queries? /* Can this database efficiently do range queries?
* (for large data sets, more efficient that brute force) * (for large data sets, more efficient that brute force)
*/ */
virtual bool preferRangeQueries() const = 0; virtual bool preferRangeQueries() const = 0;
virtual ~DB() {} virtual ~DB() {}
}; };
@ -94,7 +98,7 @@ static inline int64_t pythonmodulo(int64_t i, int64_t mod)
} }
inline int64_t DB::encodeBlockPos(const BlockPos pos) inline int64_t DB::encodeBlockPos(const BlockPos pos) const
{ {
return (uint64_t) pos.z * 0x1000000 + return (uint64_t) pos.z * 0x1000000 +
(uint64_t) pos.y * 0x1000 + (uint64_t) pos.y * 0x1000 +
@ -102,7 +106,7 @@ inline int64_t DB::encodeBlockPos(const BlockPos pos)
} }
inline BlockPos DB::decodeBlockPos(int64_t hash) inline BlockPos DB::decodeBlockPos(int64_t hash) const
{ {
BlockPos pos; BlockPos pos;
pos.x = unsigned_to_signed(pythonmodulo(hash, 4096), 2048); pos.x = unsigned_to_signed(pythonmodulo(hash, 4096), 2048);

5
include/types.h Normal file
View File

@ -0,0 +1,5 @@
#include <string>
typedef std::basic_string<unsigned char> ustring;
typedef unsigned int uint;
typedef unsigned char u8;

View File

@ -3,8 +3,6 @@
#include <string> #include <string>
#include <iostream> #include <iostream>
#define ARRLEN(x) (sizeof(x) / sizeof((x)[0]))
template<typename T> template<typename T>
static inline T mymax(T a, T b) static inline T mymax(T a, T b)
{ {
@ -21,7 +19,3 @@ std::string read_setting(const std::string &name, std::istream &is);
std::string read_setting_default(const std::string &name, std::istream &is, std::string read_setting_default(const std::string &name, std::istream &is,
const std::string &def); const std::string &def);
bool file_exists(const char *path);
bool dir_exists(const char *path);

View File

@ -10,14 +10,12 @@
#include <stdexcept> #include <stdexcept>
#include "config.h" #include "config.h"
#include "TileGenerator.h" #include "TileGenerator.h"
#include "util.h"
#include "log.h"
static void usage() static void usage()
{ {
static const std::pair<const char*, const char*> options[] = { const std::pair<const char*, const char*> options[] = {
{"-i/--input", "<path>"}, {"-i/--input", "<world_path>"},
{"-o/--output", "<path>"}, {"-o/--output", "<output_image.png>"},
{"--bgcolor", "<color>"}, {"--bgcolor", "<color>"},
{"--scalecolor", "<color>"}, {"--scalecolor", "<color>"},
{"--playercolor", "<color>"}, {"--playercolor", "<color>"},
@ -28,20 +26,19 @@ static void usage()
{"--drawalpha", ""}, {"--drawalpha", ""},
{"--noshading", ""}, {"--noshading", ""},
{"--noemptyimage", ""}, {"--noemptyimage", ""},
{"-v/--verbose", ""},
{"--min-y", "<y>"}, {"--min-y", "<y>"},
{"--max-y", "<y>"}, {"--max-y", "<y>"},
{"--backend", "<backend>"}, {"--backend", "<backend>"},
{"--geometry", "x:z+w+h"}, {"--geometry", "x:y+w+h"},
{"--extent", ""}, {"--extent", ""},
{"--zoom", "<factor>"}, {"--zoom", "<zoomlevel>"},
{"--colors", "<path>"}, {"--colors", "<colors.txt>"},
{"--scales", "[t][b][l][r]"}, {"--scales", "[t][b][l][r]"},
{"--exhaustive", "never|y|full|auto"}, {"--exhaustive", "never|y|full|auto"},
{"--dumpblock", "x,y,z"}, {"--dumpblock", "x,y,z"},
}; };
const char *top_text = const char *top_text =
"minetestmapper -i <world_path> -o <output_image> [options]\n" "minetestmapper -i <world_path> -o <output_image.png> [options]\n"
"Generate an overview image of a Luanti map.\n" "Generate an overview image of a Luanti map.\n"
"\n" "\n"
"Options:\n"; "Options:\n";
@ -58,16 +55,12 @@ static void usage()
for (auto s : backends) for (auto s : backends)
printf("%s ", s.c_str()); printf("%s ", s.c_str());
printf("\n"); printf("\n");
#ifdef _WIN32
printf("See also the full documentation in README.rst\n");
#else
printf("See also the full documentation in minetestmapper(6) or README.rst\n");
#endif
} }
static inline bool file_exists(const std::string &path) static inline bool file_exists(const std::string &path)
{ {
return file_exists(path.c_str()); std::ifstream ifs(path);
return ifs.is_open();
} }
static inline int stoi(const char *s) static inline int stoi(const char *s)
@ -85,24 +78,23 @@ static std::string search_colors(const std::string &worldpath)
#ifndef _WIN32 #ifndef _WIN32
char *home = std::getenv("HOME"); char *home = std::getenv("HOME");
if (home && home[0]) { if (home) {
std::string check = std::string(home) + "/.minetest/colors.txt"; std::string check = std::string(home) + "/.minetest/colors.txt";
if (file_exists(check)) if (file_exists(check))
return check; return check;
} }
#endif #endif
constexpr bool sharedir_valid = !(SHAREDIR[0] == '.' || !SHAREDIR[0]); constexpr bool sharedir_valid = !(SHAREDIR[0] == '.' || SHAREDIR[0] == '\0');
if (sharedir_valid && file_exists(SHAREDIR "/colors.txt")) if (sharedir_valid && file_exists(SHAREDIR "/colors.txt"))
return SHAREDIR "/colors.txt"; return SHAREDIR "/colors.txt";
errorstream << "Warning: Falling back to using colors.txt from current directory." << std::endl; std::cerr << "Warning: Falling back to using colors.txt from current directory." << std::endl;
return "./colors.txt"; return "colors.txt";
} }
int main(int argc, char *argv[]) int main(int argc, char *argv[])
{ {
const char *short_options = "hi:o:v";
const static struct option long_options[] = const static struct option long_options[] =
{ {
{"help", no_argument, 0, 'h'}, {"help", no_argument, 0, 'h'},
@ -128,12 +120,9 @@ int main(int argc, char *argv[])
{"noemptyimage", no_argument, 0, 'n'}, {"noemptyimage", no_argument, 0, 'n'},
{"exhaustive", required_argument, 0, 'j'}, {"exhaustive", required_argument, 0, 'j'},
{"dumpblock", required_argument, 0, 'k'}, {"dumpblock", required_argument, 0, 'k'},
{"verbose", no_argument, 0, 'v'},
{0, 0, 0, 0} {0, 0, 0, 0}
}; };
configure_log_streams(false);
std::string input; std::string input;
std::string output; std::string output;
std::string colors; std::string colors;
@ -143,7 +132,7 @@ int main(int argc, char *argv[])
TileGenerator generator; TileGenerator generator;
while (1) { while (1) {
int option_index; int option_index;
int c = getopt_long(argc, argv, short_options, long_options, &option_index); int c = getopt_long(argc, argv, "hi:o:", long_options, &option_index);
if (c == -1) if (c == -1)
break; // done break; // done
@ -203,7 +192,7 @@ int main(int argc, char *argv[])
geometry >> x >> c >> y >> w >> h; geometry >> x >> c >> y >> w >> h;
if (geometry.fail() || c != ':' || w < 1 || h < 1) { if (geometry.fail() || c != ':' || w < 1 || h < 1) {
usage(); usage();
return 1; exit(1);
} }
generator.setGeometry(x, y, w, h); generator.setGeometry(x, y, w, h);
} }
@ -231,7 +220,7 @@ int main(int argc, char *argv[])
generator.setDontWriteEmpty(true); generator.setDontWriteEmpty(true);
break; break;
case 'j': { case 'j': {
int mode = EXH_AUTO; int mode = EXH_AUTO;;
if (!strcmp(optarg, "never")) if (!strcmp(optarg, "never"))
mode = EXH_NEVER; mode = EXH_NEVER;
else if (!strcmp(optarg, "y")) else if (!strcmp(optarg, "y"))
@ -247,15 +236,12 @@ int main(int argc, char *argv[])
iss >> dumpblock.x >> c >> dumpblock.y >> c2 >> dumpblock.z; iss >> dumpblock.x >> c >> dumpblock.y >> c2 >> dumpblock.z;
if (iss.fail() || c != ',' || c2 != ',') { if (iss.fail() || c != ',' || c2 != ',') {
usage(); usage();
return 1; exit(1);
} }
break; break;
} }
case 'v':
configure_log_streams(true);
break;
default: default:
return 1; exit(1);
} }
} }
@ -266,6 +252,7 @@ int main(int argc, char *argv[])
} }
try { try {
if (onlyPrintExtent) { if (onlyPrintExtent) {
generator.printGeometry(input); generator.printGeometry(input);
return 0; return 0;
@ -280,7 +267,7 @@ int main(int argc, char *argv[])
generator.generate(input, output); generator.generate(input, output);
} catch (const std::exception &e) { } catch (const std::exception &e) {
errorstream << "Exception: " << e.what() << std::endl; std::cerr << "Exception: " << e.what() << std::endl;
return 1; return 1;
} }
return 0; return 0;

View File

@ -9,22 +9,16 @@ minetestmapper \- generate an overview image of a Luanti map
See additional optional parameters below. See additional optional parameters below.
.SH DESCRIPTION .SH DESCRIPTION
.B minetestmapper .B minetestmapper
generates a top-down overview image of a Luanti map. generates an overview image of a Luanti map. This is a port of
This is a port of the obsolete minetestmapper.py script to C++, the original minetestmapper.py to C++, that is both faster and
that is both faster and provides more features. provides more functionality than the obsolete Python script.
Minetestmapper ships with a colors.txt file suitable for Minetest Game,
if you use a different game or have mods installed you should generate a
matching colors.txt for better results (colors will be missing otherwise).
.SH MANDATORY PARAMETERS .SH MANDATORY PARAMETERS
.TP .TP
.BR \-i " " \fIworld_path\fR .BR \-i " " \fIworld_path\fR
Input world path Input world path.
.TP .TP
.BR \-o " " \fIoutput_image\fR .BR \-o " " \fIoutput_image\fR
Path to output image Path to output image. (only PNG supported currently)
.SH OPTIONAL PARAMETERS .SH OPTIONAL PARAMETERS
.TP .TP
.BR \-\-bgcolor " " \fIcolor\fR .BR \-\-bgcolor " " \fIcolor\fR
@ -32,7 +26,7 @@ Background color of image, e.g. "--bgcolor #ffffff"
.TP .TP
.BR \-\-scalecolor " " \fIcolor\fR .BR \-\-scalecolor " " \fIcolor\fR
Color of scale marks and text, e.g. "--scalecolor #000000" Color of scale, e.g. "--scalecolor #000000"
.TP .TP
.BR \-\-playercolor " " \fIcolor\fR .BR \-\-playercolor " " \fIcolor\fR
@ -44,11 +38,11 @@ Color of origin indicator, e.g. "--origincolor #ff0000"
.TP .TP
.BR \-\-drawscale .BR \-\-drawscale
Draw scale(s) with tick marks and numbers Draw tick marks
.TP .TP
.BR \-\-drawplayers .BR \-\-drawplayers
Draw player indicators with name Draw player indicators
.TP .TP
.BR \-\-draworigin .BR \-\-draworigin
@ -56,7 +50,7 @@ Draw origin indicator
.TP .TP
.BR \-\-drawalpha .BR \-\-drawalpha
Allow nodes to be drawn with transparency (such as water) Allow nodes to be drawn with transparency
.TP .TP
.BR \-\-noshading .BR \-\-noshading
@ -64,29 +58,23 @@ Don't draw shading on nodes
.TP .TP
.BR \-\-noemptyimage .BR \-\-noemptyimage
Don't output anything when the image would be empty Don't output anything when the image would be empty.
.TP
.BR \-\-verbose
Enable verbose log output.
.TP .TP
.BR \-\-min-y " " \fInumber\fR .BR \-\-min-y " " \fInumber\fR
Don't draw nodes below this Y value, e.g. "--min-y -25" Don't draw nodes below this y value, e.g. "--min-y -25"
.TP .TP
.BR \-\-max-y " " \fInumber\fR .BR \-\-max-y " " \fInumber\fR
Don't draw nodes above this Y value, e.g. "--max-y 75" Don't draw nodes above this y value, e.g. "--max-y 75"
.TP .TP
.BR \-\-backend " " \fIbackend\fR .BR \-\-backend " " \fIbackend\fR
Override auto-detected map backend; supported: \fIsqlite3\fP, \fIleveldb\fP, \fIredis\fP, \fIpostgresql\fP, e.g. "--backend leveldb" Use specific map backend; supported: \fIsqlite3\fP, \fIleveldb\fP, \fIredis\fP, \fIpostgresql\fP, e.g. "--backend leveldb"
.TP .TP
.BR \-\-geometry " " \fIgeometry\fR .BR \-\-geometry " " \fIgeometry\fR
Limit area to specific geometry (\fIx:z+w+h\fP where x and z specify the lower left corner), e.g. "--geometry -800:-800+1600+1600" Limit area to specific geometry (\fIx:y+w+h\fP where x and y specify the lower left corner), e.g. "--geometry -800:-800+1600+1600"
The coordinates are specified with the same axes as in-game. The Z axis becomes Y when projected on the image.
.TP .TP
.BR \-\-extent .BR \-\-extent
@ -98,7 +86,7 @@ Zoom the image by using more than one pixel per node, e.g. "--zoom 4"
.TP .TP
.BR \-\-colors " " \fIpath\fR .BR \-\-colors " " \fIpath\fR
Override auto-detected path to colors.txt, e.g. "--colors ../world/mycolors.txt" Forcefully set path to colors.txt file (autodetected otherwise), e.g. "--colors ../world/mycolors.txt"
.TP .TP
.BR \-\-scales " " \fIedges\fR .BR \-\-scales " " \fIedges\fR
@ -108,14 +96,19 @@ Draw scales on specified image edges (letters \fIt b l r\fP meaning top, bottom,
.BR \-\-exhaustive " " \fImode\fR .BR \-\-exhaustive " " \fImode\fR
Select if database should be traversed exhaustively or using range queries, available: \fInever\fP, \fIy\fP, \fIfull\fP, \fIauto\fP Select if database should be traversed exhaustively or using range queries, available: \fInever\fP, \fIy\fP, \fIfull\fP, \fIauto\fP
Defaults to \fIauto\fP. You shouldn't need to change this, as minetestmapper tries to automatically picks the best option. Defaults to \fIauto\fP. You shouldn't need to change this, but doing so can improve rendering times on large maps.
For these optimizations to work it is important that you set
.B min-y
and
.B max-y
when you don't care about the world below e.g. -60 and above 1000 nodes.
.TP .TP
.BR \-\-dumpblock " " \fIpos\fR .BR \-\-dumpblock " " \fIpos\fR
Instead of rendering anything try to load the block at the given position (\fIx,y,z\fR) and print its raw data as hexadecimal. Instead of rendering anything try to load the block at the given position (\fIx,y,z\fR) and print its raw data as hexadecimal.
.SH MORE INFORMATION .SH MORE INFORMATION
Website: https://github.com/luanti-org/minetestmapper Website: https://github.com/minetest/minetestmapper
.SH MAN PAGE AUTHOR .SH MAN PAGE AUTHOR
Daniel Moerner Daniel Moerner

View File

@ -1,159 +0,0 @@
#include <fstream>
#include <sstream>
#include <stdexcept>
#include <dirent.h>
#include <unistd.h> // usleep
#include "config.h"
#include "PlayerAttributes.h"
#include "util.h"
#include "log.h"
#include "db-sqlite3.h" // SQLite3Base
namespace {
bool parse_pos(std::string position, Player &dst)
{
if (position.empty())
return false;
if (position.front() == '(' && position.back() == ')')
position = position.substr(1, position.size() - 2);
std::istringstream iss(position);
if (!(iss >> dst.x))
return false;
if (iss.get() != ',')
return false;
if (!(iss >> dst.y))
return false;
if (iss.get() != ',')
return false;
if (!(iss >> dst.z))
return false;
return iss.eof();
}
// Helper classes per backend
class FilesReader {
std::string path;
DIR *dir = nullptr;
public:
FilesReader(const std::string &path) : path(path) {
dir = opendir(path.c_str());
}
~FilesReader() {
if (dir)
closedir(dir);
}
void read(PlayerAttributes::Players &dest);
};
class SQLiteReader : SQLite3Base {
sqlite3_stmt *stmt_get_player_pos = NULL;
public:
SQLiteReader(const std::string &database) {
openDatabase(database.c_str());
}
~SQLiteReader() {
sqlite3_finalize(stmt_get_player_pos);
}
void read(PlayerAttributes::Players &dest);
};
}
void FilesReader::read(PlayerAttributes::Players &dest)
{
if (!dir)
return;
struct dirent *ent;
std::string name, position;
while ((ent = readdir(dir)) != NULL) {
if (ent->d_name[0] == '.')
continue;
std::ifstream in(path + PATH_SEPARATOR + ent->d_name);
if (!in.good())
continue;
name = read_setting("name", in);
position = read_setting("position", in);
Player player;
player.name = name;
if (!parse_pos(position, player)) {
errorstream << "Failed to parse position '" << position << "' in "
<< ent->d_name << std::endl;
continue;
}
player.x /= 10.0f;
player.y /= 10.0f;
player.z /= 10.0f;
dest.push_back(std::move(player));
}
}
#define SQLRES(r, good) check_result(r, good)
#define SQLOK(r) SQLRES(r, SQLITE_OK)
void SQLiteReader::read(PlayerAttributes::Players &dest)
{
SQLOK(prepare(stmt_get_player_pos,
"SELECT name, posX, posY, posZ FROM player"));
int result;
while ((result = sqlite3_step(stmt_get_player_pos)) != SQLITE_DONE) {
if (result == SQLITE_BUSY) { // Wait some time and try again
usleep(10000);
} else if (result != SQLITE_ROW) {
throw std::runtime_error(sqlite3_errmsg(db));
}
Player player;
player.name = read_str(stmt_get_player_pos, 0);
player.x = sqlite3_column_double(stmt_get_player_pos, 1);
player.y = sqlite3_column_double(stmt_get_player_pos, 2);
player.z = sqlite3_column_double(stmt_get_player_pos, 3);
player.x /= 10.0f;
player.y /= 10.0f;
player.z /= 10.0f;
dest.push_back(std::move(player));
}
}
/**********/
PlayerAttributes::PlayerAttributes(const std::string &worldDir)
{
std::ifstream ifs(worldDir + "world.mt");
if (!ifs.good())
throw std::runtime_error("Failed to read world.mt");
std::string backend = read_setting_default("player_backend", ifs, "files");
ifs.close();
verbosestream << "Player backend: " << backend << std::endl;
if (backend == "files")
FilesReader(worldDir + "players").read(m_players);
else if (backend == "sqlite3")
SQLiteReader(worldDir + "players.sqlite").read(m_players);
else
throw std::runtime_error(std::string("Unknown player backend: ") + backend);
verbosestream << "Loaded " << m_players.size() << " players" << std::endl;
}
PlayerAttributes::Players::const_iterator PlayerAttributes::begin() const
{
return m_players.cbegin();
}
PlayerAttributes::Players::const_iterator PlayerAttributes::end() const
{
return m_players.cend();
}

View File

@ -1,7 +0,0 @@
#if defined(MSDOS) || defined(__OS2__) || defined(__NT__) || defined(_WIN32)
#define PATH_SEPARATOR '\\'
#else
#define PATH_SEPARATOR '/'
#endif
#include "cmake_config.h"

View File

@ -1,261 +0,0 @@
#include <stdexcept>
#include <unistd.h> // for usleep
#include <iostream>
#include <algorithm>
#include <cassert>
#include "db-sqlite3.h"
#include "log.h"
#include "types.h"
/* SQLite3Base */
#define SQLRES(r, good) check_result(r, good)
#define SQLOK(r) SQLRES(r, SQLITE_OK)
SQLite3Base::~SQLite3Base()
{
if (db && sqlite3_close(db) != SQLITE_OK) {
errorstream << "Error closing SQLite database: "
<< sqlite3_errmsg(db) << std::endl;
}
}
void SQLite3Base::openDatabase(const char *path, bool readonly)
{
if (db)
throw std::logic_error("Database already open");
int flags = 0;
if (readonly)
flags |= SQLITE_OPEN_READONLY | SQLITE_OPEN_PRIVATECACHE;
else
flags |= SQLITE_OPEN_READWRITE | SQLITE_OPEN_CREATE;
#ifdef SQLITE_OPEN_EXRESCODE
flags |= SQLITE_OPEN_EXRESCODE;
#endif
SQLOK(sqlite3_open_v2(path, &db, flags, 0));
}
/* DBSQLite3 */
// make sure a row is available. intended to be used outside a loop.
// compare result to SQLITE_ROW afterwards.
#define SQLROW1(stmt) \
while ((result = sqlite3_step(stmt)) == SQLITE_BUSY) \
usleep(10000); /* wait some time and try again */ \
if (result != SQLITE_ROW && result != SQLITE_DONE) { \
throw std::runtime_error(sqlite3_errmsg(db)); \
}
// make sure next row is available. intended to be used in a while(sqlite3_step) loop
#define SQLROW2() \
if (result == SQLITE_BUSY) { \
usleep(10000); /* wait some time and try again */ \
continue; \
} else if (result != SQLITE_ROW) { \
throw std::runtime_error(sqlite3_errmsg(db)); \
}
DBSQLite3::DBSQLite3(const std::string &mapdir)
{
std::string db_name = mapdir + "map.sqlite";
openDatabase(db_name.c_str());
// There's a simple, dumb way to check if we have a new or old database schema.
// If we prepare a statement that references columns that don't exist, it will
// error right there.
int result = prepare(stmt_get_block_pos, "SELECT x, y, z FROM blocks");
newFormat = result == SQLITE_OK;
verbosestream << "Detected " << (newFormat ? "new" : "old") << " SQLite schema" << std::endl;
if (newFormat) {
SQLOK(prepare(stmt_get_blocks_xz_range,
"SELECT y, data FROM blocks WHERE "
"x = ? AND z = ? AND y BETWEEN ? AND ?"));
SQLOK(prepare(stmt_get_block_exact,
"SELECT data FROM blocks WHERE x = ? AND y = ? AND z = ?"));
SQLOK(prepare(stmt_get_block_pos_range,
"SELECT x, z FROM blocks WHERE "
"x >= ? AND y >= ? AND z >= ? AND "
"x < ? AND y < ? AND z < ? GROUP BY x, z"));
} else {
SQLOK(prepare(stmt_get_blocks_z,
"SELECT pos, data FROM blocks WHERE pos BETWEEN ? AND ?"));
SQLOK(prepare(stmt_get_block_exact,
"SELECT data FROM blocks WHERE pos = ?"));
SQLOK(prepare(stmt_get_block_pos,
"SELECT pos FROM blocks"));
SQLOK(prepare(stmt_get_block_pos_range,
"SELECT pos FROM blocks WHERE pos BETWEEN ? AND ?"));
}
#undef RANGE
}
DBSQLite3::~DBSQLite3()
{
sqlite3_finalize(stmt_get_blocks_z);
sqlite3_finalize(stmt_get_blocks_xz_range);
sqlite3_finalize(stmt_get_block_pos);
sqlite3_finalize(stmt_get_block_pos_range);
sqlite3_finalize(stmt_get_block_exact);
}
inline void DBSQLite3::getPosRange(int64_t &min, int64_t &max,
int16_t zPos, int16_t zPos2)
{
// Magic numbers!
min = encodeBlockPos(BlockPos(0, -2048, zPos));
max = encodeBlockPos(BlockPos(0, 2048, zPos2)) - 1;
}
std::vector<BlockPos> DBSQLite3::getBlockPosXZ(BlockPos min, BlockPos max)
{
int result;
sqlite3_stmt *stmt;
if (newFormat) {
stmt = stmt_get_block_pos_range;
int col = bind_pos(stmt, 1, min);
bind_pos(stmt, col, max);
} else {
// can handle range query on Z axis via SQL
if (min.z <= -2048 && max.z >= 2048) {
stmt = stmt_get_block_pos;
} else {
stmt = stmt_get_block_pos_range;
int64_t minPos, maxPos;
if (min.z < -2048)
min.z = -2048;
if (max.z > 2048)
max.z = 2048;
getPosRange(minPos, maxPos, min.z, max.z - 1);
SQLOK(sqlite3_bind_int64(stmt, 1, minPos));
SQLOK(sqlite3_bind_int64(stmt, 2, maxPos));
}
}
std::vector<BlockPos> positions;
BlockPos pos;
while ((result = sqlite3_step(stmt)) != SQLITE_DONE) {
SQLROW2()
if (newFormat) {
pos.x = sqlite3_column_int(stmt, 0);
pos.z = sqlite3_column_int(stmt, 1);
} else {
pos = decodeBlockPos(sqlite3_column_int64(stmt, 0));
if (pos.x < min.x || pos.x >= max.x || pos.y < min.y || pos.y >= max.y)
continue;
// note that we can't try to deduplicate these because the order
// of the encoded pos (if sorted) is ZYX.
}
positions.emplace_back(pos);
}
SQLOK(sqlite3_reset(stmt));
return positions;
}
void DBSQLite3::loadBlockCache(int16_t zPos)
{
int result;
blockCache.clear();
assert(!newFormat);
int64_t minPos, maxPos;
getPosRange(minPos, maxPos, zPos, zPos);
SQLOK(sqlite3_bind_int64(stmt_get_blocks_z, 1, minPos));
SQLOK(sqlite3_bind_int64(stmt_get_blocks_z, 2, maxPos));
while ((result = sqlite3_step(stmt_get_blocks_z)) != SQLITE_DONE) {
SQLROW2()
int64_t posHash = sqlite3_column_int64(stmt_get_blocks_z, 0);
BlockPos pos = decodeBlockPos(posHash);
blockCache[pos.x].emplace_back(pos, read_blob(stmt_get_blocks_z, 1));
}
SQLOK(sqlite3_reset(stmt_get_blocks_z));
}
void DBSQLite3::getBlocksOnXZ(BlockList &blocks, int16_t x, int16_t z,
int16_t min_y, int16_t max_y)
{
// New format: use a real range query
if (newFormat) {
auto *stmt = stmt_get_blocks_xz_range;
SQLOK(sqlite3_bind_int(stmt, 1, x));
SQLOK(sqlite3_bind_int(stmt, 2, z));
SQLOK(sqlite3_bind_int(stmt, 3, min_y));
SQLOK(sqlite3_bind_int(stmt, 4, max_y - 1)); // BETWEEN is inclusive
int result;
while ((result = sqlite3_step(stmt)) != SQLITE_DONE) {
SQLROW2()
BlockPos pos(x, sqlite3_column_int(stmt, 0), z);
blocks.emplace_back(pos, read_blob(stmt, 1));
}
SQLOK(sqlite3_reset(stmt));
return;
}
/* Cache the blocks on the given Z coordinate between calls, this only
* works due to order in which the TileGenerator asks for blocks. */
if (z != blockCachedZ) {
loadBlockCache(z);
blockCachedZ = z;
}
auto it = blockCache.find(x);
if (it == blockCache.end())
return;
if (it->second.empty()) {
/* We have swapped this list before, this is not supposed to happen
* because it's bad for performance. But rather than silently breaking
* do the right thing and load the blocks again. */
verbosestream << "suboptimal access pattern for sqlite3 backend?!" << std::endl;
loadBlockCache(z);
}
// Swap lists to avoid copying contents
blocks.clear();
std::swap(blocks, it->second);
for (auto it = blocks.begin(); it != blocks.end(); ) {
if (it->first.y < min_y || it->first.y >= max_y)
it = blocks.erase(it);
else
it++;
}
}
void DBSQLite3::getBlocksByPos(BlockList &blocks,
const std::vector<BlockPos> &positions)
{
int result;
for (auto pos : positions) {
bind_pos(stmt_get_block_exact, 1, pos);
SQLROW1(stmt_get_block_exact)
if (result == SQLITE_ROW)
blocks.emplace_back(pos, read_blob(stmt_get_block_exact, 0));
SQLOK(sqlite3_reset(stmt_get_block_exact));
}
}

View File

@ -1,87 +0,0 @@
#pragma once
#include "db.h"
#include <unordered_map>
#include <sqlite3.h>
class SQLite3Base {
public:
~SQLite3Base();
protected:
void openDatabase(const char *path, bool readonly = true);
// check function result or throw error
inline void check_result(int result, int good = SQLITE_OK)
{
if (result != good)
throw std::runtime_error(sqlite3_errmsg(db));
}
// prepare a statement
inline int prepare(sqlite3_stmt *&stmt, const char *sql)
{
return sqlite3_prepare_v2(db, sql, -1, &stmt, NULL);
}
// read string from statement
static inline std::string read_str(sqlite3_stmt *stmt, int iCol)
{
auto *data = reinterpret_cast<const char*>(
sqlite3_column_text(stmt, iCol));
return std::string(data);
}
// read blob from statement
static inline ustring read_blob(sqlite3_stmt *stmt, int iCol)
{
auto *data = reinterpret_cast<const unsigned char *>(
sqlite3_column_blob(stmt, iCol));
size_t size = sqlite3_column_bytes(stmt, iCol);
return ustring(data, size);
}
sqlite3 *db = NULL;
};
class DBSQLite3 : public DB, SQLite3Base {
public:
DBSQLite3(const std::string &mapdir);
std::vector<BlockPos> getBlockPosXZ(BlockPos min, BlockPos max) override;
void getBlocksOnXZ(BlockList &blocks, int16_t x, int16_t z,
int16_t min_y, int16_t max_y) override;
void getBlocksByPos(BlockList &blocks,
const std::vector<BlockPos> &positions) override;
~DBSQLite3() override;
bool preferRangeQueries() const override { return newFormat; }
private:
static inline void getPosRange(int64_t &min, int64_t &max, int16_t zPos,
int16_t zPos2);
void loadBlockCache(int16_t zPos);
// bind pos to statement. returns index of next column.
inline int bind_pos(sqlite3_stmt *stmt, int iCol, BlockPos pos)
{
if (newFormat) {
sqlite3_bind_int(stmt, iCol, pos.x);
sqlite3_bind_int(stmt, iCol + 1, pos.y);
sqlite3_bind_int(stmt, iCol + 2, pos.z);
return iCol + 3;
} else {
sqlite3_bind_int64(stmt, iCol, encodeBlockPos(pos));
return iCol + 1;
}
}
sqlite3_stmt *stmt_get_block_pos = NULL;
sqlite3_stmt *stmt_get_block_pos_range = NULL;
sqlite3_stmt *stmt_get_blocks_z = NULL;
sqlite3_stmt *stmt_get_blocks_xz_range = NULL;
sqlite3_stmt *stmt_get_block_exact = NULL;
bool newFormat = false;
int16_t blockCachedZ = -10000;
std::unordered_map<int16_t, BlockList> blockCache; // indexed by X
};

View File

@ -1,16 +0,0 @@
#include <iostream>
#include "log.h"
StreamProxy errorstream(nullptr);
StreamProxy verbosestream(nullptr);
void configure_log_streams(bool verbose)
{
errorstream << std::flush;
verbosestream << std::flush;
errorstream = std::cerr.good() ? &std::cerr : nullptr;
// std::clog does not automatically flush
verbosestream = (verbose && std::clog.good()) ? &std::clog : nullptr;
}

View File

@ -1,40 +0,0 @@
#pragma once
#include <iostream>
#include <utility>
// Forwards to an ostream, optionally
class StreamProxy {
public:
StreamProxy(std::ostream *os) : m_os(os) {}
template<typename T>
StreamProxy &operator<<(T &&arg)
{
if (m_os)
*m_os << std::forward<T>(arg);
return *this;
}
StreamProxy &operator<<(std::ostream &(*func)(std::ostream&))
{
if (m_os)
*m_os << func;
return *this;
}
private:
std::ostream *m_os;
};
/// Error and warning output, forwards to std::cerr
extern StreamProxy errorstream;
/// Verbose output, might forward to std::cerr
extern StreamProxy verbosestream;
/**
* Configure log streams defined in this file.
* @param verbose enable verbose output
* @note not thread-safe!
*/
void configure_log_streams(bool verbose);

View File

@ -1,40 +0,0 @@
#pragma once
#include <string>
// Define custom char traits since std::char_traits<unsigend char> is not part of C++ standard
struct uchar_traits : std::char_traits<char>
{
using super = std::char_traits<char>;
using char_type = unsigned char;
static void assign(char_type& c1, const char_type& c2) noexcept {
c1 = c2;
}
static char_type* assign(char_type* ptr, std::size_t count, char_type c2) {
return reinterpret_cast<char_type*>(
super::assign(reinterpret_cast<char*>(ptr), count, static_cast<char>(c2)));
}
static char_type* move(char_type* dest, const char_type* src, std::size_t count) {
return reinterpret_cast<char_type*>(
super::move(reinterpret_cast<char*>(dest), reinterpret_cast<const char*>(src), count));
}
static char_type* copy(char_type* dest, const char_type* src, std::size_t count) {
return reinterpret_cast<char_type*>(
super::copy(reinterpret_cast<char*>(dest), reinterpret_cast<const char*>(src), count));
}
static int compare(const char_type* s1, const char_type* s2, std::size_t count) {
return super::compare(reinterpret_cast<const char*>(s1), reinterpret_cast<const char*>(s2), count);
}
static char_type to_char_type(int_type c) noexcept {
return static_cast<char_type>(c);
}
};
typedef std::basic_string<unsigned char, uchar_traits> ustring;
typedef unsigned int uint;
typedef unsigned char u8;

View File

@ -1,76 +0,0 @@
#include <stdexcept>
#include <iostream>
#include <sys/types.h>
#include <sys/stat.h>
#include "util.h"
static std::string trim(const std::string &s)
{
auto isspace = [] (char c) {
return c == ' ' || c == '\t' || c == '\r' || c == '\n';
};
size_t front = 0;
while (isspace(s[front]))
++front;
size_t back = s.size() - 1;
while (back > front && isspace(s[back]))
--back;
return s.substr(front, back - front + 1);
}
static bool read_setting(const std::string &name, std::istream &is, std::string &out)
{
char linebuf[512];
is.seekg(0);
while (is.good()) {
is.getline(linebuf, sizeof(linebuf));
std::string line(linebuf);
auto pos = line.find('#');
if (pos != std::string::npos)
line.erase(pos); // remove comments
pos = line.find('=');
if (pos == std::string::npos)
continue;
auto key = trim(line.substr(0, pos));
if (key != name)
continue;
out = trim(line.substr(pos+1));
return true;
}
return false;
}
std::string read_setting(const std::string &name, std::istream &is)
{
std::string ret;
if (!read_setting(name, is, ret))
throw std::runtime_error(std::string("Setting not found: ") + name);
return ret;
}
std::string read_setting_default(const std::string &name, std::istream &is,
const std::string &def)
{
std::string ret;
if (!read_setting(name, is, ret))
return def;
return ret;
}
bool file_exists(const char *path)
{
struct stat s{};
// check for !dir to allow symlinks or such
return stat(path, &s) == 0 && (s.st_mode & S_IFDIR) != S_IFDIR;
}
bool dir_exists(const char *path)
{
struct stat s{};
return stat(path, &s) == 0 && (s.st_mode & S_IFDIR) == S_IFDIR;
}

55
util.cpp Normal file
View File

@ -0,0 +1,55 @@
#include <stdexcept>
#include <sstream>
#include "util.h"
static std::string trim(const std::string &s)
{
auto isspace = [] (char c) -> bool { return c == ' ' || c == '\t' || c == '\r' || c == '\n'; };
size_t front = 0;
while (isspace(s[front]))
++front;
size_t back = s.size() - 1;
while (back > front && isspace(s[back]))
--back;
return s.substr(front, back - front + 1);
}
std::string read_setting(const std::string &name, std::istream &is)
{
char linebuf[512];
while (is.good()) {
is.getline(linebuf, sizeof(linebuf));
for (char *p = linebuf; *p; p++) {
if(*p != '#')
continue;
*p = '\0'; // Cut off at the first #
break;
}
std::string line(linebuf);
auto pos = line.find('=');
if (pos == std::string::npos)
continue;
auto key = trim(line.substr(0, pos));
if (key != name)
continue;
return trim(line.substr(pos+1));
}
std::ostringstream oss;
oss << "Setting '" << name << "' not found";
throw std::runtime_error(oss.str());
}
std::string read_setting_default(const std::string &name, std::istream &is,
const std::string &def)
{
try {
return read_setting(name, is);
} catch(const std::runtime_error &e) {
return def;
}
}

View File

@ -1,27 +1,26 @@
#!/bin/bash -e #!/bin/bash -e
[ -z "$CXX" ] && exit 255
export CC=false # don't need it actually
variant=win32
[[ "$(basename "$CXX")" == "x86_64-"* ]] && variant=win64
####### #######
# this expects unpacked libraries and a toolchain file like Luanti's buildbot uses # this expects unpacked libraries similar to what Luanti's buildbot uses
# $extradlls will typically contain the compiler-specific DLLs and libpng # $extradlls will typically point to the DLLs for libgcc, libstdc++ and libpng
toolchain_file=
libgd_dir= libgd_dir=
zlib_dir= zlib_dir=
zstd_dir= zstd_dir=
sqlite_dir= sqlite_dir=
leveldb_dir= leveldb_dir=
extradlls=( extradlls=()
)
####### #######
[ -f "$toolchain_file" ] || exit 1 [ -f ./CMakeLists.txt ] || exit 1
variant=win32
grep -q 'CX?X?_COMPILER.*x86_64-' $toolchain_file && variant=win64
echo "Detected target $variant"
[ -f ./CMakeLists.txt ] || { echo "run from root folder" >&2; exit 1; }
cmake -S . -B build \ cmake -S . -B build \
-DCMAKE_TOOLCHAIN_FILE="$toolchain_file" \ -DCMAKE_SYSTEM_NAME=Windows \
-DCMAKE_EXE_LINKER_FLAGS="-s" \ -DCMAKE_EXE_LINKER_FLAGS="-s" \
\ \
-DENABLE_LEVELDB=1 \ -DENABLE_LEVELDB=1 \
@ -35,7 +34,7 @@ cmake -S . -B build \
-DZLIB_INCLUDE_DIR=$zlib_dir/include \ -DZLIB_INCLUDE_DIR=$zlib_dir/include \
-DZLIB_LIBRARY=$zlib_dir/lib/libz.dll.a \ -DZLIB_LIBRARY=$zlib_dir/lib/libz.dll.a \
-DZSTD_INCLUDE_DIR=$zstd_dir/include \ -DZSTD_INCLUDE_DIR=$zstd_dir/include \
-DZSTD_LIBRARY=$zstd_dir/lib/libzstd.dll.a -DZSTD_LIBRARY=$zstd_dir/lib/libzstd.dll.a \
make -C build -j4 make -C build -j4

View File

@ -1,29 +1,32 @@
#!/bin/bash -e #!/bin/bash -e
install_linux_deps() { install_linux_deps() {
local upkgs=( local pkgs=(cmake libgd-dev libsqlite3-dev libleveldb-dev libpq-dev libhiredis-dev libzstd-dev)
cmake libgd-dev libsqlite3-dev libleveldb-dev libpq-dev
libhiredis-dev libzstd-dev
)
local fpkgs=(
cmake gcc-g++ gd-devel sqlite-devel libzstd-devel zlib-ng-devel
)
if command -v dnf; then
sudo dnf install --setopt=install_weak_deps=False -y "${fpkgs[@]}"
else
sudo apt-get update sudo apt-get update
sudo apt-get install -y --no-install-recommends "${upkgs[@]}" sudo apt-get remove -y 'libgd3' nginx || : # ????
fi sudo apt-get install -y --no-install-recommends "${pkgs[@]}" "$@"
} }
run_build() { run_build() {
local args=( local args=(
-DCMAKE_BUILD_TYPE=Debug -DCMAKE_BUILD_TYPE=Debug
-DENABLE_LEVELDB=ON -DENABLE_POSTGRESQL=ON -DENABLE_REDIS=ON -DENABLE_LEVELDB=1 -DENABLE_POSTGRESQL=1 -DENABLE_REDIS=1
) )
[[ "$CXX" == clang* ]] && args+=(-DCMAKE_CXX_FLAGS="-fsanitize=address") [[ "$CXX" == clang* ]] && args+=(-DCMAKE_CXX_FLAGS="-fsanitize=address")
cmake . "${args[@]}" cmake . "${args[@]}"
make -j2 make -j2
} }
do_functional_test() {
mkdir testmap
echo "backend = sqlite3" >testmap/world.mt
sqlite3 testmap/map.sqlite <<END
CREATE TABLE blocks(pos INT,data BLOB);
INSERT INTO blocks(pos, data) VALUES(0, x'$(cat util/ci/test_block)');
END
./minetestmapper --noemptyimage -i ./testmap -o map.png
file map.png
}

View File

@ -1,116 +0,0 @@
#!/bin/bash
set -eo pipefail
mapdir=./testmap
msg () {
echo
echo "==== $1"
echo
}
# encodes a block position by X, Y, Z (positive numbers only!)
encodepos () {
echo "$(($1 + 0x1000 * $2 + 0x1000000 * $3))"
}
# create map file with sql statements
writemap () {
rm -rf $mapdir
mkdir $mapdir
echo "backend = sqlite3" >$mapdir/world.mt
echo "default:stone 10 10 10" >$mapdir/colors.txt
printf '%s\n' \
"CREATE TABLE d(d BLOB);" \
"INSERT INTO d VALUES (x'$(cat util/ci/test_block)');" \
"$1" \
"DROP TABLE d;" | sqlite3 $mapdir/map.sqlite
}
# check that a non-empty ($1=1) or empty map ($1=0) was written with the args ($2 ...)
checkmap () {
local c=$1
shift
rm -f map.png
./minetestmapper --noemptyimage -v -i ./testmap -o map.png "$@"
if [[ $c -eq 1 && ! -f map.png ]]; then
echo "Output not generated!"
exit 1
elif [[ $c -eq 0 && -f map.png ]]; then
echo "Output was generated, none expected!"
exit 1
fi
echo "Passed."
}
# this is missing the indices and primary keys but that doesn't matter
schema_old="CREATE TABLE blocks(pos INT, data BLOB);"
schema_new="CREATE TABLE blocks(x INT, y INT, z INT, data BLOB);"
msg "old schema"
writemap "
$schema_old
INSERT INTO blocks SELECT $(encodepos 0 1 0), d FROM d;
"
checkmap 1
msg "old schema: Y limit"
# Note: test data contains a plane at y = 17 an a single node at y = 18
checkmap 1 --max-y 17
checkmap 0 --max-y 16
checkmap 1 --min-y 18
checkmap 0 --min-y 19
# do this for every strategy
for exh in never y full; do
msg "old schema: all limits ($exh)"
# fill the map with more blocks and then request just a single one to be rendered
# this will run through internal consistency asserts.
writemap "
$schema_old
INSERT INTO blocks SELECT $(encodepos 2 2 2), d FROM d;
INSERT INTO blocks SELECT $(encodepos 1 2 2), d FROM d;
INSERT INTO blocks SELECT $(encodepos 2 1 2), d FROM d;
INSERT INTO blocks SELECT $(encodepos 2 2 1), d FROM d;
INSERT INTO blocks SELECT $(encodepos 3 2 2), d FROM d;
INSERT INTO blocks SELECT $(encodepos 2 3 2), d FROM d;
INSERT INTO blocks SELECT $(encodepos 2 2 3), d FROM d;
"
checkmap 1 --geometry 32:32+16+16 --min-y 32 --max-y $((32+16-1)) --exhaustive $exh
done
msg "new schema"
writemap "
$schema_new
INSERT INTO blocks SELECT 0, 1, 0, d FROM d;
"
checkmap 1
# same as above
for exh in never y full; do
msg "new schema: all limits ($exh)"
writemap "
$schema_new
INSERT INTO blocks SELECT 2, 2, 2, d FROM d;
INSERT INTO blocks SELECT 1, 2, 2, d FROM d;
INSERT INTO blocks SELECT 2, 1, 2, d FROM d;
INSERT INTO blocks SELECT 2, 2, 1, d FROM d;
INSERT INTO blocks SELECT 3, 2, 2, d FROM d;
INSERT INTO blocks SELECT 2, 3, 2, d FROM d;
INSERT INTO blocks SELECT 2, 2, 3, d FROM d;
"
checkmap 1 --geometry 32:32+16+16 --min-y 32 --max-y $((32+16-1)) --exhaustive $exh
done
msg "new schema: empty map"
writemap "$schema_new"
checkmap 0
msg "drawplayers"
writemap "
$schema_new
INSERT INTO blocks SELECT 0, 0, 0, d FROM d;
"
mkdir $mapdir/players
printf '%s\n' "name = cat" "position = (80,0,80)" >$mapdir/players/cat
# we can't check that it actually worked, however
checkmap 1 --drawplayers --zoom 4

View File

@ -2,20 +2,9 @@ local function get_tile(tiles, n)
local tile = tiles[n] local tile = tiles[n]
if type(tile) == 'table' then if type(tile) == 'table' then
return tile.name or tile.image return tile.name or tile.image
elseif type(tile) == 'string' then end
return tile return tile
end end
end
local function strip_texture(tex)
tex = (tex .. '^'):match('%(*(.-)%)*^') -- strip modifiers
if tex:find("[combine", 1, true) then
tex = tex:match('.-=([^:]-)') -- extract first texture
elseif tex:find("[png", 1, true) then
return nil -- can't
end
return tex
end
local function pairs_s(dict) local function pairs_s(dict)
local keys = {} local keys = {}
@ -31,7 +20,7 @@ core.register_chatcommand("dumpnodes", {
func = function() func = function()
local ntbl = {} local ntbl = {}
for _, nn in pairs_s(minetest.registered_nodes) do for _, nn in pairs_s(minetest.registered_nodes) do
local prefix, name = nn:match('(.-):(.*)') local prefix, name = nn:match('(.*):(.*)')
if prefix == nil or name == nil then if prefix == nil or name == nil then
print("ignored(1): " .. nn) print("ignored(1): " .. nn)
else else
@ -56,15 +45,14 @@ core.register_chatcommand("dumpnodes", {
print("ignored(2): " .. nn) print("ignored(2): " .. nn)
else else
local tex = get_tile(tiles, 1) local tex = get_tile(tiles, 1)
tex = tex and strip_texture(tex) tex = (tex .. '^'):match('%(*(.-)%)*^') -- strip modifiers
if not tex then if tex:find("[combine", 1, true) then
print("ignored(3): " .. nn) tex = tex:match('.-=([^:]-)') -- extract first texture
else end
out:write(nn .. ' ' .. tex .. '\n') out:write(nn .. ' ' .. tex .. '\n')
n = n + 1 n = n + 1
end end
end end
end
out:write('\n') out:write('\n')
end end
out:close() out:close()