109 lines
3.3 KiB
C++
109 lines
3.3 KiB
C++
#include <mbgl/renderer/pattern_atlas.hpp>
|
|
#include <mbgl/gfx/upload_pass.hpp>
|
|
#include <mbgl/gfx/context.hpp>
|
|
#include <mbgl/gfx/texture2d.hpp>
|
|
|
|
namespace mbgl {
|
|
|
|
namespace {
|
|
|
|
// When copied into the atlas texture, image data is padded by one pixel on each
|
|
// side. Pattern images are padded with a copy of the image data wrapped from
|
|
// the opposite side. This ensures the correct behavior of GL_LINEAR texture
|
|
// sampling mode.
|
|
const uint16_t padding = 1;
|
|
|
|
mapbox::ShelfPack::ShelfPackOptions shelfPackOptions() {
|
|
mapbox::ShelfPack::ShelfPackOptions options;
|
|
options.autoResize = true;
|
|
return options;
|
|
}
|
|
|
|
} // namespace
|
|
|
|
PatternAtlas::PatternAtlas()
|
|
: shelfPack(64, 64, shelfPackOptions()) {}
|
|
|
|
PatternAtlas::~PatternAtlas() = default;
|
|
|
|
std::optional<ImagePosition> PatternAtlas::getPattern(const std::string& id) const {
|
|
auto it = patterns.find(id);
|
|
if (it != patterns.end()) {
|
|
return it->second.position;
|
|
}
|
|
return std::nullopt;
|
|
}
|
|
|
|
std::optional<ImagePosition> PatternAtlas::addPattern(const style::Image::Impl& image) {
|
|
if (patterns.contains(image.id)) {
|
|
return std::nullopt;
|
|
}
|
|
const uint16_t width = image.image.size.width + padding * 2;
|
|
const uint16_t height = image.image.size.height + padding * 2;
|
|
|
|
mapbox::Bin* bin = shelfPack.packOne(-1, width, height);
|
|
if (!bin) {
|
|
return std::nullopt;
|
|
}
|
|
|
|
atlasImage.resize(getPixelSize());
|
|
|
|
const PremultipliedImage& src = image.image;
|
|
|
|
const uint32_t x = bin->x + padding;
|
|
const uint32_t y = bin->y + padding;
|
|
const uint32_t w = src.size.width;
|
|
const uint32_t h = src.size.height;
|
|
|
|
PremultipliedImage::copy(src, atlasImage, {0, 0}, {x, y}, {w, h});
|
|
|
|
// Add 1 pixel wrapped padding on each side of the image.
|
|
PremultipliedImage::copy(src, atlasImage, {0, h - 1}, {x, y - 1}, {w, 1}); // T
|
|
PremultipliedImage::copy(src, atlasImage, {0, 0}, {x, y + h}, {w, 1}); // B
|
|
PremultipliedImage::copy(src, atlasImage, {w - 1, 0}, {x - 1, y}, {1, h}); // L
|
|
PremultipliedImage::copy(src, atlasImage, {0, 0}, {x + w, y}, {1, h}); // R
|
|
|
|
dirty = true;
|
|
|
|
return patterns
|
|
.emplace(image.id, Pattern{.bin = bin, .position = {Rect<uint16_t>(bin->x, bin->y, bin->w, bin->h), image}})
|
|
.first->second.position;
|
|
}
|
|
|
|
void PatternAtlas::removePattern(const std::string& id) {
|
|
auto it = patterns.find(id);
|
|
if (it != patterns.end()) {
|
|
// Clear pattern from the atlas image.
|
|
const uint32_t x = it->second.bin->x;
|
|
const uint32_t y = it->second.bin->y;
|
|
const uint32_t w = it->second.bin->w;
|
|
const uint32_t h = it->second.bin->h;
|
|
PremultipliedImage::clear(atlasImage, {x, y}, {w, h});
|
|
|
|
shelfPack.unref(*it->second.bin);
|
|
patterns.erase(it);
|
|
}
|
|
}
|
|
|
|
Size PatternAtlas::getPixelSize() const {
|
|
return {static_cast<uint32_t>(shelfPack.width()), static_cast<uint32_t>(shelfPack.height())};
|
|
}
|
|
|
|
void PatternAtlas::upload([[maybe_unused]] gfx::UploadPass& uploadPass) {
|
|
if (!atlasTexture2D) {
|
|
atlasTexture2D = uploadPass.getContext().createTexture2D();
|
|
if (atlasTexture2D) {
|
|
atlasTexture2D->upload(atlasImage);
|
|
}
|
|
} else if (dirty) {
|
|
atlasTexture2D->upload(atlasImage);
|
|
}
|
|
dirty = false;
|
|
}
|
|
|
|
const std::shared_ptr<gfx::Texture2D>& PatternAtlas::texture() const {
|
|
return atlasTexture2D;
|
|
}
|
|
|
|
} // namespace mbgl
|