heightmap.cpp

Go to the documentation of this file.

/*
 * This file is part of OpenTTD.
 * OpenTTD is free software; you can redistribute it and/or modify it under the terms of the GNU General Public License as published by the Free Software Foundation, version 2.
 * OpenTTD is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
 * See the GNU General Public License for more details. You should have received a copy of the GNU General Public License along with OpenTTD. If not, see <http://www.gnu.org/licenses/>.
 */
 
#include "stdafx.h"
#include "heightmap.h"
#include "clear_map.h"
#include "void_map.h"
#include "error.h"
#include "saveload/saveload.h"
#include "bmp.h"
#include "gfx_func.h"
#include "fios.h"
#include "fileio_func.h"
 
#include "table/strings.h"
 
#include "safeguards.h"
 
static const uint MAX_HEIGHTMAP_SIDE_LENGTH_IN_PIXELS = 2 * MAX_MAP_SIZE;
 
/*
 * Maximum size in pixels of the heightmap image.
 */
static const uint MAX_HEIGHTMAP_SIZE_PIXELS = 256 << 20; // ~256 million
/*
 * When loading a PNG or BMP the 24 bpp variant requires at least 4 bytes per pixel
 * of memory to load the data. Make sure the "reasonable" limit is well within the
 * maximum amount of memory allocatable on 32 bit platforms.
 */
static_assert(MAX_HEIGHTMAP_SIZE_PIXELS < UINT32_MAX / 8);
 
static inline bool IsValidHeightmapDimension(size_t width, size_t height)
{
  return (uint64_t)width * height <= MAX_HEIGHTMAP_SIZE_PIXELS &&
    width > 0 && width <= MAX_HEIGHTMAP_SIDE_LENGTH_IN_PIXELS &&
    height > 0 && height <= MAX_HEIGHTMAP_SIDE_LENGTH_IN_PIXELS;
}
 
static inline uint8_t RGBToGrayscale(uint8_t red, uint8_t green, uint8_t blue)
{
  /* To avoid doubles and stuff, multiply it with a total of 65536 (16bits), then
   *  divide by it to normalize the value to a byte again. */
  return ((red * 19595) + (green * 38470) + (blue * 7471)) / 65536;
}
 
 
#ifdef WITH_PNG
 
#include <png.h>
 
static void ReadHeightmapPNGImageData(std::span<uint8_t> map, png_structp png_ptr, png_infop info_ptr)
{
  uint x, y;
  uint8_t gray_palette[256];
  png_bytep *row_pointers = nullptr;
  bool has_palette = png_get_color_type(png_ptr, info_ptr) == PNG_COLOR_TYPE_PALETTE;
  uint channels = png_get_channels(png_ptr, info_ptr);
 
  /* Get palette and convert it to grayscale */
  if (has_palette) {
    int i;
    int palette_size;
    png_color *palette;
    bool all_gray = true;
 
    png_get_PLTE(png_ptr, info_ptr, &palette, &palette_size);
    for (i = 0; i < palette_size && (palette_size != 16 || all_gray); i++) {
      all_gray &= palette[i].red == palette[i].green && palette[i].red == palette[i].blue;
      gray_palette[i] = RGBToGrayscale(palette[i].red, palette[i].green, palette[i].blue);
    }
 
    if (palette_size == 16 && !all_gray) {
      for (i = 0; i < palette_size; i++) {
        gray_palette[i] = 256 * i / palette_size;
      }
    }
  }
 
  row_pointers = png_get_rows(png_ptr, info_ptr);
 
  /* Read the raw image data and convert in 8-bit grayscale */
  for (x = 0; x < png_get_image_width(png_ptr, info_ptr); x++) {
    for (y = 0; y < png_get_image_height(png_ptr, info_ptr); y++) {
      uint8_t *pixel = &map[y * png_get_image_width(png_ptr, info_ptr) + x];
      uint x_offset = x * channels;
 
      if (has_palette) {
        *pixel = gray_palette[row_pointers[y][x_offset]];
      } else if (channels == 3) {
        *pixel = RGBToGrayscale(row_pointers[y][x_offset + 0],
            row_pointers[y][x_offset + 1], row_pointers[y][x_offset + 2]);
      } else {
        *pixel = row_pointers[y][x_offset];
      }
    }
  }
}
 
static bool ReadHeightmapPNG(const char *filename, uint *x, uint *y, std::vector<uint8_t> *map)
{
  png_structp png_ptr = nullptr;
  png_infop info_ptr  = nullptr;
 
  auto fp = FioFOpenFile(filename, "rb", HEIGHTMAP_DIR);
  if (!fp.has_value()) {
    ShowErrorMessage(STR_ERROR_PNGMAP, STR_ERROR_PNGMAP_FILE_NOT_FOUND, WL_ERROR);
    return false;
  }
 
  png_ptr = png_create_read_struct(PNG_LIBPNG_VER_STRING, nullptr, nullptr, nullptr);
  if (png_ptr == nullptr) {
    ShowErrorMessage(STR_ERROR_PNGMAP, STR_ERROR_PNGMAP_MISC, WL_ERROR);
    return false;
  }
 
  info_ptr = png_create_info_struct(png_ptr);
  if (info_ptr == nullptr || setjmp(png_jmpbuf(png_ptr))) {
    ShowErrorMessage(STR_ERROR_PNGMAP, STR_ERROR_PNGMAP_MISC, WL_ERROR);
    png_destroy_read_struct(&png_ptr, &info_ptr, nullptr);
    return false;
  }
 
  png_init_io(png_ptr, *fp);
 
  /* Allocate memory and read image, without alpha or 16-bit samples
   * (result is either 8-bit indexed/grayscale or 24-bit RGB) */
  png_set_packing(png_ptr);
  png_read_png(png_ptr, info_ptr, PNG_TRANSFORM_PACKING | PNG_TRANSFORM_STRIP_ALPHA | PNG_TRANSFORM_STRIP_16, nullptr);
 
  /* Maps of wrong colour-depth are not used.
   * (this should have been taken care of by stripping alpha and 16-bit samples on load) */
  if ((png_get_channels(png_ptr, info_ptr) != 1) && (png_get_channels(png_ptr, info_ptr) != 3) && (png_get_bit_depth(png_ptr, info_ptr) != 8)) {
    ShowErrorMessage(STR_ERROR_PNGMAP, STR_ERROR_PNGMAP_IMAGE_TYPE, WL_ERROR);
    png_destroy_read_struct(&png_ptr, &info_ptr, nullptr);
    return false;
  }
 
  uint width = png_get_image_width(png_ptr, info_ptr);
  uint height = png_get_image_height(png_ptr, info_ptr);
 
  if (!IsValidHeightmapDimension(width, height)) {
    ShowErrorMessage(STR_ERROR_PNGMAP, STR_ERROR_HEIGHTMAP_TOO_LARGE, WL_ERROR);
    png_destroy_read_struct(&png_ptr, &info_ptr, nullptr);
    return false;
  }
 
  if (map != nullptr) {
    map->resize(static_cast<size_t>(width) * height);
    ReadHeightmapPNGImageData(*map, png_ptr, info_ptr);
  }
 
  *x = width;
  *y = height;
 
  png_destroy_read_struct(&png_ptr, &info_ptr, nullptr);
  return true;
}
 
#endif /* WITH_PNG */
 
 
static void ReadHeightmapBMPImageData(std::span<uint8_t> map, const BmpInfo &info, const BmpData &data)
{
  uint8_t gray_palette[256];
 
  if (!data.palette.empty()) {
    bool all_gray = true;
 
    if (info.palette_size != 2) {
      for (uint i = 0; i < info.palette_size && (info.palette_size != 16 || all_gray); i++) {
        all_gray &= data.palette[i].r == data.palette[i].g && data.palette[i].r == data.palette[i].b;
        gray_palette[i] = RGBToGrayscale(data.palette[i].r, data.palette[i].g, data.palette[i].b);
      }
 
      if (info.palette_size == 16 && !all_gray) {
        for (uint i = 0; i < info.palette_size; i++) {
          gray_palette[i] = 256 * i / info.palette_size;
        }
      }
    } else {
      gray_palette[0] = 0;
      gray_palette[1] = 16;
    }
  }
 
  /* Read the raw image data and convert in 8-bit grayscale */
  for (uint y = 0; y < info.height; y++) {
    uint8_t *pixel = &map[y * static_cast<size_t>(info.width)];
    const uint8_t *bitmap = &data.bitmap[y * static_cast<size_t>(info.width) * (info.bpp == 24 ? 3 : 1)];
 
    for (uint x = 0; x < info.width; x++) {
      if (info.bpp != 24) {
        *pixel++ = gray_palette[*bitmap++];
      } else {
        *pixel++ = RGBToGrayscale(*bitmap, *(bitmap + 1), *(bitmap + 2));
        bitmap += 3;
      }
    }
  }
}
 
static bool ReadHeightmapBMP(const char *filename, uint *x, uint *y, std::vector<uint8_t> *map)
{
  auto f = FioFOpenFile(filename, "rb", HEIGHTMAP_DIR);
  if (!f.has_value()) {
    ShowErrorMessage(STR_ERROR_BMPMAP, STR_ERROR_PNGMAP_FILE_NOT_FOUND, WL_ERROR);
    return false;
  }
 
  RandomAccessFile file(filename, HEIGHTMAP_DIR);
  BmpInfo info{};
  BmpData data{};
 
  if (!BmpReadHeader(file, info, data)) {
    ShowErrorMessage(STR_ERROR_BMPMAP, STR_ERROR_BMPMAP_IMAGE_TYPE, WL_ERROR);
    return false;
  }
 
  if (!IsValidHeightmapDimension(info.width, info.height)) {
    ShowErrorMessage(STR_ERROR_BMPMAP, STR_ERROR_HEIGHTMAP_TOO_LARGE, WL_ERROR);
    return false;
  }
 
  if (map != nullptr) {
    if (!BmpReadBitmap(file, info, data)) {
      ShowErrorMessage(STR_ERROR_BMPMAP, STR_ERROR_BMPMAP_IMAGE_TYPE, WL_ERROR);
      return false;
    }
 
    map->resize(static_cast<size_t>(info.width) * info.height);
    ReadHeightmapBMPImageData(*map, info, data);
  }
 
  *x = info.width;
  *y = info.height;
 
  return true;
}
 
static void GrayscaleToMapHeights(uint img_width, uint img_height, std::span<const uint8_t> map)
{
  /* Defines the detail of the aspect ratio (to avoid doubles) */
  const uint num_div = 16384;
  /* Ensure multiplication with num_div does not cause overflows. */
  static_assert(num_div <= std::numeric_limits<uint>::max() / MAX_HEIGHTMAP_SIDE_LENGTH_IN_PIXELS);
 
  uint width, height;
  uint row, col;
  uint row_pad = 0, col_pad = 0;
  uint img_scale;
  uint img_row, img_col;
  TileIndex tile;
 
  /* Get map size and calculate scale and padding values */
  switch (_settings_game.game_creation.heightmap_rotation) {
    default: NOT_REACHED();
    case HM_COUNTER_CLOCKWISE:
      width   = Map::SizeX();
      height  = Map::SizeY();
      break;
    case HM_CLOCKWISE:
      width   = Map::SizeY();
      height  = Map::SizeX();
      break;
  }
 
  if ((img_width * num_div) / img_height > ((width * num_div) / height)) {
    /* Image is wider than map - center vertically */
    img_scale = (width * num_div) / img_width;
    row_pad = (1 + height - ((img_height * img_scale) / num_div)) / 2;
  } else {
    /* Image is taller than map - center horizontally */
    img_scale = (height * num_div) / img_height;
    col_pad = (1 + width - ((img_width * img_scale) / num_div)) / 2;
  }
 
  if (_settings_game.construction.freeform_edges) {
    for (uint x = 0; x < Map::SizeX(); x++) MakeVoid(TileXY(x, 0));
    for (uint y = 0; y < Map::SizeY(); y++) MakeVoid(TileXY(0, y));
  }
 
  /* Form the landscape */
  for (row = 0; row < height; row++) {
    for (col = 0; col < width; col++) {
      switch (_settings_game.game_creation.heightmap_rotation) {
        default: NOT_REACHED();
        case HM_COUNTER_CLOCKWISE: tile = TileXY(col, row); break;
        case HM_CLOCKWISE:         tile = TileXY(row, col); break;
      }
 
      /* Check if current tile is within the 1-pixel map edge or padding regions */
      if ((!_settings_game.construction.freeform_edges && DistanceFromEdge(tile) <= 1) ||
          (row < row_pad) || (row >= (height - row_pad - (_settings_game.construction.freeform_edges ? 0 : 1))) ||
          (col < col_pad) || (col >= (width  - col_pad - (_settings_game.construction.freeform_edges ? 0 : 1)))) {
        SetTileHeight(tile, 0);
      } else {
        /* Use nearest neighbour resizing to scale map data.
         *  We rotate the map 45 degrees (counter)clockwise */
        img_row = (((row - row_pad) * num_div) / img_scale);
        switch (_settings_game.game_creation.heightmap_rotation) {
          default: NOT_REACHED();
          case HM_COUNTER_CLOCKWISE:
            img_col = (((width - 1 - col - col_pad) * num_div) / img_scale);
            break;
          case HM_CLOCKWISE:
            img_col = (((col - col_pad) * num_div) / img_scale);
            break;
        }
 
        assert(img_row < img_height);
        assert(img_col < img_width);
 
        uint heightmap_height = map[img_row * img_width + img_col];
 
        if (heightmap_height > 0) {
          /* 0 is sea level.
           * Other grey scales are scaled evenly to the available height levels > 0.
           * (The coastline is independent from the number of height levels) */
          heightmap_height = 1 + (heightmap_height - 1) * _settings_game.game_creation.heightmap_height / 255;
        }
 
        SetTileHeight(tile, heightmap_height);
      }
      /* Only clear the tiles within the map area. */
      if (IsInnerTile(tile)) {
        MakeClear(tile, CLEAR_GRASS, 3);
      }
    }
  }
}
 
void FixSlopes()
{
  uint width, height;
  int row, col;
  uint8_t current_tile;
 
  /* Adjust height difference to maximum one horizontal/vertical change. */
  width   = Map::SizeX();
  height  = Map::SizeY();
 
  /* Top and left edge */
  for (row = 0; (uint)row < height; row++) {
    for (col = 0; (uint)col < width; col++) {
      current_tile = MAX_TILE_HEIGHT;
      if (col != 0) {
        /* Find lowest tile; either the top or left one */
        current_tile = TileHeight(TileXY(col - 1, row)); // top edge
      }
      if (row != 0) {
        if (TileHeight(TileXY(col, row - 1)) < current_tile) {
          current_tile = TileHeight(TileXY(col, row - 1)); // left edge
        }
      }
 
      /* Does the height differ more than one? */
      if (TileHeight(TileXY(col, row)) >= (uint)current_tile + 2) {
        /* Then change the height to be no more than one */
        SetTileHeight(TileXY(col, row), current_tile + 1);
      }
    }
  }
 
  /* Bottom and right edge */
  for (row = height - 1; row >= 0; row--) {
    for (col = width - 1; col >= 0; col--) {
      current_tile = MAX_TILE_HEIGHT;
      if ((uint)col != width - 1) {
        /* Find lowest tile; either the bottom and right one */
        current_tile = TileHeight(TileXY(col + 1, row)); // bottom edge
      }
 
      if ((uint)row != height - 1) {
        if (TileHeight(TileXY(col, row + 1)) < current_tile) {
          current_tile = TileHeight(TileXY(col, row + 1)); // right edge
        }
      }
 
      /* Does the height differ more than one? */
      if (TileHeight(TileXY(col, row)) >= (uint)current_tile + 2) {
        /* Then change the height to be no more than one */
        SetTileHeight(TileXY(col, row), current_tile + 1);
      }
    }
  }
}
 
static bool ReadHeightMap(DetailedFileType dft, const char *filename, uint *x, uint *y, std::vector<uint8_t> *map)
{
  switch (dft) {
    default:
      NOT_REACHED();
 
#ifdef WITH_PNG
    case DFT_HEIGHTMAP_PNG:
      return ReadHeightmapPNG(filename, x, y, map);
#endif /* WITH_PNG */
 
    case DFT_HEIGHTMAP_BMP:
      return ReadHeightmapBMP(filename, x, y, map);
  }
}
 
bool GetHeightmapDimensions(DetailedFileType dft, const char *filename, uint *x, uint *y)
{
  return ReadHeightMap(dft, filename, x, y, nullptr);
}
 
bool LoadHeightmap(DetailedFileType dft, const char *filename)
{
  uint x, y;
  std::vector<uint8_t> map;
 
  if (!ReadHeightMap(dft, filename, &x, &y, &map)) {
    return false;
  }
 
  GrayscaleToMapHeights(x, y, map);
 
  FixSlopes();
  MarkWholeScreenDirty();
 
  return true;
}
 
void FlatEmptyWorld(uint8_t tile_height)
{
  int edge_distance = _settings_game.construction.freeform_edges ? 0 : 2;
  for (uint row = edge_distance; row < Map::SizeY() - edge_distance; row++) {
    for (uint col = edge_distance; col < Map::SizeX() - edge_distance; col++) {
      SetTileHeight(TileXY(col, row), tile_height);
    }
  }
 
  FixSlopes();
  MarkWholeScreenDirty();
}