minetestmapper/TileGenerator.cpp

209 lines
4.0 KiB
C++
Raw Normal View History

2012-08-23 12:46:22 +02:00
/*
* =====================================================================
* Version: 1.0
* Created: 23.08.2012 12:35:53
* Author: Miroslav Bendík
* Company: LinuxOS.sk
* =====================================================================
*/
2012-08-23 14:21:34 +02:00
#include <cstdio>
#include <cstdlib>
#include <fstream>
#include <iostream>
2012-08-23 12:46:22 +02:00
#include "TileGenerator.h"
using namespace std;
2012-08-23 12:46:22 +02:00
TileGenerator::TileGenerator():
m_bgColor("#ffffff"),
m_scaleColor("#000000"),
m_originColor("#ff0000"),
m_playerColor("#ff0000"),
m_drawOrigin(false),
m_drawPlayers(false),
m_drawScale(false),
2012-08-23 13:32:22 +02:00
m_drawUnderground(false),
2012-08-23 14:06:16 +02:00
m_db(0),
2012-08-23 14:21:34 +02:00
m_image(0),
2012-08-23 14:06:16 +02:00
m_xMin(0),
m_xMax(0),
m_zMin(0),
m_zMax(0)
2012-08-23 12:46:22 +02:00
{
}
TileGenerator::~TileGenerator()
{
2012-08-23 13:32:22 +02:00
if (m_db != 0) {
sqlite3_close(m_db);
m_db = 0;
}
2012-08-23 12:46:22 +02:00
}
void TileGenerator::setBgColor(const std::string &bgColor)
{
m_bgColor = bgColor;
}
void TileGenerator::setScaleColor(const std::string &scaleColor)
{
m_scaleColor = scaleColor;
}
void TileGenerator::setOriginColor(const std::string &originColor)
{
m_originColor = originColor;
}
void TileGenerator::setPlayerColor(const std::string &playerColor)
{
m_playerColor = playerColor;
}
void TileGenerator::setDrawOrigin(bool drawOrigin)
{
m_drawOrigin = drawOrigin;
}
void TileGenerator::setDrawPlayers(bool drawPlayers)
{
m_drawPlayers = drawPlayers;
}
void TileGenerator::setDrawScale(bool drawScale)
{
m_drawScale = drawScale;
}
void TileGenerator::setDrawUnderground(bool drawUnderground)
{
m_drawUnderground = drawUnderground;
}
void TileGenerator::parseColorsFile(const std::string &fileName)
{
ifstream in;
in.open(fileName.c_str(), ifstream::in);
if (!in.is_open()) {
std::cerr << "File colors.txt does not exist" << std::endl;
exit(-2);
}
while (in.good()) {
string name;
Color color;
in >> name;
if (name[0] == '#') {
in.ignore(65536, '\n');
in >> name;
}
while (name == "\n" && in.good()) {
in >> name;
}
int r, g, b;
in >> r;
in >> g;
in >> b;
if (in.good()) {
m_colors[name] = color;
color.r = r;
color.g = g;
color.b = b;
}
}
}
2012-08-23 14:21:34 +02:00
void TileGenerator::generate(const std::string &input, const std::string &output)
2012-08-23 13:32:22 +02:00
{
openDb(input);
loadBlocks();
2012-08-23 14:21:34 +02:00
createImage();
writeImage(output);
2012-08-23 13:32:22 +02:00
}
void TileGenerator::openDb(const std::string &input)
{
string db_name = input + "map.sqlite";
if (sqlite3_open(db_name.c_str(), &m_db) != SQLITE_OK) {
throw DbError();
}
}
void TileGenerator::loadBlocks()
{
sqlite3_stmt *statement;
string sql = "SELECT pos FROM blocks";
if (sqlite3_prepare_v2(m_db, sql.c_str(), sql.length(), &statement, 0) == SQLITE_OK) {
int result = 0;
while (true) {
result = sqlite3_step(statement);
if(result == SQLITE_ROW) {
sqlite3_int64 blocknum = sqlite3_column_int64(statement, 0);
2012-08-23 14:06:16 +02:00
BlockPos pos = decodeBlockPos(blocknum);
if (pos.x > SectorXMax || pos.x < SectorXMin || pos.z > SectorZMax || pos.z < SectorZMin) {
continue;
}
if (pos.x < m_xMin) {
m_xMin = pos.x;
}
if (pos.x > m_xMax) {
m_xMax = pos.x;
}
if (pos.z < m_zMin) {
m_zMin = pos.z;
}
if (pos.z > m_zMax) {
m_zMax = pos.z;
}
2012-08-23 13:32:22 +02:00
}
else {
break;
}
}
}
else {
throw DbError();
}
}
inline BlockPos TileGenerator::decodeBlockPos(sqlite3_int64 blockId)
{
BlockPos pos;
pos.x = unsignedToSigned(blockId % 4096, 2048);
2012-08-23 14:06:16 +02:00
blockId = (blockId - pos.x) / 4096;
2012-08-23 13:32:22 +02:00
pos.y = unsignedToSigned(blockId % 4096, 2048);
2012-08-23 14:06:16 +02:00
blockId = (blockId - pos.y) / 4096;
2012-08-23 13:32:22 +02:00
pos.z = unsignedToSigned(blockId % 4096, 2048);
return pos;
}
inline int TileGenerator::unsignedToSigned(long i, long max_positive)
{
if (i < max_positive) {
return i;
}
else {
return i - 2l * max_positive;
}
}
2012-08-23 14:21:34 +02:00
void TileGenerator::createImage()
{
m_imgWidth = (m_xMax - m_xMin) * 16;
m_imgHeight = (m_zMax - m_zMin) * 16;
m_image = gdImageCreate(m_imgWidth, m_imgHeight);
// Background
gdImageColorAllocate(m_image, 255, 255, 255);
}
void TileGenerator::writeImage(const std::string &output)
{
FILE *out;
out = fopen(output.c_str(), "w");
gdImagePng(m_image, out);
fclose(out);
gdImageDestroy(m_image);
}