273 lines
7.6 KiB
C++
273 lines
7.6 KiB
C++
#include <algorithm>
|
|
#include <cstring>
|
|
#include <graphics/texture_atlas.h>
|
|
#include <utils/logger.h>
|
|
|
|
|
|
namespace extra2d {
|
|
|
|
// ============================================================================
|
|
// TextureAtlasPage 实现
|
|
// ============================================================================
|
|
|
|
TextureAtlasPage::TextureAtlasPage(int width, int height)
|
|
: width_(width), height_(height), isFull_(false), usedArea_(0) {
|
|
// 创建空白纹理
|
|
std::vector<uint8_t> emptyData(width * height * 4, 0);
|
|
texture_ = shared<GLTexture>(width, height, emptyData.data(), 4);
|
|
|
|
// 初始化矩形打包根节点
|
|
root_ = std::make_unique<PackNode>(0, 0, width, height);
|
|
|
|
E2D_LOG_INFO("Created texture atlas page: {}x{}", width, height);
|
|
}
|
|
|
|
TextureAtlasPage::~TextureAtlasPage() = default;
|
|
|
|
bool TextureAtlasPage::tryAddTexture(const std::string &name, int texWidth,
|
|
int texHeight, const uint8_t *pixels,
|
|
Rect &outUvRect) {
|
|
if (isFull_) {
|
|
return false;
|
|
}
|
|
|
|
// 添加边距
|
|
int paddedWidth = texWidth + 2 * PADDING;
|
|
int paddedHeight = texHeight + 2 * PADDING;
|
|
|
|
// 如果纹理太大，无法放入
|
|
if (paddedWidth > width_ || paddedHeight > height_) {
|
|
return false;
|
|
}
|
|
|
|
// 尝试插入
|
|
PackNode *node = insert(root_.get(), paddedWidth, paddedHeight);
|
|
if (node == nullptr) {
|
|
// 无法放入，标记为满
|
|
isFull_ = true;
|
|
return false;
|
|
}
|
|
|
|
// 写入像素数据（跳过边距区域）
|
|
writePixels(node->x + PADDING, node->y + PADDING, texWidth, texHeight,
|
|
pixels);
|
|
|
|
// 创建条目
|
|
AtlasEntry entry;
|
|
entry.name = name;
|
|
entry.originalSize =
|
|
Vec2(static_cast<float>(texWidth), static_cast<float>(texHeight));
|
|
entry.padding = PADDING;
|
|
|
|
// 计算 UV 坐标（考虑边距）
|
|
float u1 = static_cast<float>(node->x + PADDING) / width_;
|
|
float v1 = static_cast<float>(node->y + PADDING) / height_;
|
|
float u2 = static_cast<float>(node->x + PADDING + texWidth) / width_;
|
|
float v2 = static_cast<float>(node->y + PADDING + texHeight) / height_;
|
|
|
|
entry.uvRect = Rect(u1, v1, u2 - u1, v2 - v1);
|
|
outUvRect = entry.uvRect;
|
|
|
|
entries_[name] = std::move(entry);
|
|
usedArea_ += paddedWidth * paddedHeight;
|
|
|
|
E2D_LOG_DEBUG("Added texture '{}' to atlas: {}x{} at ({}, {})", name,
|
|
texWidth, texHeight, node->x, node->y);
|
|
|
|
return true;
|
|
}
|
|
|
|
TextureAtlasPage::PackNode *TextureAtlasPage::insert(PackNode *node, int width,
|
|
int height) {
|
|
if (node == nullptr) {
|
|
return nullptr;
|
|
}
|
|
|
|
// 如果节点已被使用，尝试子节点
|
|
if (node->used) {
|
|
PackNode *result = insert(node->left.get(), width, height);
|
|
if (result != nullptr) {
|
|
return result;
|
|
}
|
|
return insert(node->right.get(), width, height);
|
|
}
|
|
|
|
// 检查是否适合
|
|
if (width > node->width || height > node->height) {
|
|
return nullptr;
|
|
}
|
|
|
|
// 如果刚好合适，使用此节点
|
|
if (width == node->width && height == node->height) {
|
|
node->used = true;
|
|
return node;
|
|
}
|
|
|
|
// 需要分割节点
|
|
int dw = node->width - width;
|
|
int dh = node->height - height;
|
|
|
|
if (dw > dh) {
|
|
// 水平分割
|
|
node->left =
|
|
std::make_unique<PackNode>(node->x, node->y, width, node->height);
|
|
node->right =
|
|
std::make_unique<PackNode>(node->x + width, node->y, dw, node->height);
|
|
} else {
|
|
// 垂直分割
|
|
node->left =
|
|
std::make_unique<PackNode>(node->x, node->y, node->width, height);
|
|
node->right =
|
|
std::make_unique<PackNode>(node->x, node->y + height, node->width, dh);
|
|
}
|
|
|
|
// 递归插入到左子节点
|
|
return insert(node->left.get(), width, height);
|
|
}
|
|
|
|
void TextureAtlasPage::writePixels(int x, int y, int w, int h,
|
|
const uint8_t *pixels) {
|
|
if (texture_ == nullptr || pixels == nullptr) {
|
|
return;
|
|
}
|
|
|
|
// 使用 glTexSubImage2D 更新纹理数据
|
|
GLuint texID = static_cast<GLuint>(
|
|
reinterpret_cast<uintptr_t>(texture_->getNativeHandle()));
|
|
|
|
glBindTexture(GL_TEXTURE_2D, texID);
|
|
glTexSubImage2D(GL_TEXTURE_2D, 0, x, y, w, h, GL_RGBA, GL_UNSIGNED_BYTE,
|
|
pixels);
|
|
glBindTexture(GL_TEXTURE_2D, 0);
|
|
}
|
|
|
|
const AtlasEntry *TextureAtlasPage::getEntry(const std::string &name) const {
|
|
auto it = entries_.find(name);
|
|
if (it != entries_.end()) {
|
|
return &it->second;
|
|
}
|
|
return nullptr;
|
|
}
|
|
|
|
float TextureAtlasPage::getUsageRatio() const {
|
|
return static_cast<float>(usedArea_) / (width_ * height_);
|
|
}
|
|
|
|
// ============================================================================
|
|
// TextureAtlas 实现
|
|
// ============================================================================
|
|
|
|
TextureAtlas::TextureAtlas()
|
|
: pageSize_(TextureAtlasPage::DEFAULT_SIZE), sizeThreshold_(256),
|
|
enabled_(true), initialized_(false) {}
|
|
|
|
TextureAtlas::~TextureAtlas() = default;
|
|
|
|
void TextureAtlas::init(int pageSize) {
|
|
pageSize_ = pageSize;
|
|
initialized_ = true;
|
|
E2D_LOG_INFO("TextureAtlas initialized with page size: {}", pageSize);
|
|
}
|
|
|
|
bool TextureAtlas::addTexture(const std::string &name, int width, int height,
|
|
const uint8_t *pixels) {
|
|
if (!enabled_ || !initialized_) {
|
|
return false;
|
|
}
|
|
|
|
// 检查是否已存在
|
|
if (contains(name)) {
|
|
return true;
|
|
}
|
|
|
|
// 检查纹理大小
|
|
if (width > sizeThreshold_ || height > sizeThreshold_) {
|
|
E2D_LOG_DEBUG("Texture '{}' too large for atlas ({}x{} > {}), skipping",
|
|
name, width, height, sizeThreshold_);
|
|
return false;
|
|
}
|
|
|
|
// 尝试添加到现有页面
|
|
Rect uvRect;
|
|
for (auto &page : pages_) {
|
|
if (page->tryAddTexture(name, width, height, pixels, uvRect)) {
|
|
entryToPage_[name] = page.get();
|
|
return true;
|
|
}
|
|
}
|
|
|
|
// 创建新页面
|
|
auto newPage = std::make_unique<TextureAtlasPage>(pageSize_, pageSize_);
|
|
if (newPage->tryAddTexture(name, width, height, pixels, uvRect)) {
|
|
entryToPage_[name] = newPage.get();
|
|
pages_.push_back(std::move(newPage));
|
|
return true;
|
|
}
|
|
|
|
E2D_LOG_WARN("Failed to add texture '{}' to atlas", name);
|
|
return false;
|
|
}
|
|
|
|
bool TextureAtlas::contains(const std::string &name) const {
|
|
return entryToPage_.find(name) != entryToPage_.end();
|
|
}
|
|
|
|
const Texture *TextureAtlas::getAtlasTexture(const std::string &name) const {
|
|
auto it = entryToPage_.find(name);
|
|
if (it != entryToPage_.end()) {
|
|
return it->second->getTexture().get();
|
|
}
|
|
return nullptr;
|
|
}
|
|
|
|
Rect TextureAtlas::getUVRect(const std::string &name) const {
|
|
auto it = entryToPage_.find(name);
|
|
if (it != entryToPage_.end()) {
|
|
const AtlasEntry *entry = it->second->getEntry(name);
|
|
if (entry != nullptr) {
|
|
return entry->uvRect;
|
|
}
|
|
}
|
|
return Rect(0, 0, 1, 1); // 默认 UV
|
|
}
|
|
|
|
Vec2 TextureAtlas::getOriginalSize(const std::string &name) const {
|
|
auto it = entryToPage_.find(name);
|
|
if (it != entryToPage_.end()) {
|
|
const AtlasEntry *entry = it->second->getEntry(name);
|
|
if (entry != nullptr) {
|
|
return entry->originalSize;
|
|
}
|
|
}
|
|
return Vec2(0, 0);
|
|
}
|
|
|
|
float TextureAtlas::getTotalUsageRatio() const {
|
|
if (pages_.empty()) {
|
|
return 0.0f;
|
|
}
|
|
|
|
float total = 0.0f;
|
|
for (const auto &page : pages_) {
|
|
total += page->getUsageRatio();
|
|
}
|
|
return total / pages_.size();
|
|
}
|
|
|
|
void TextureAtlas::clear() {
|
|
pages_.clear();
|
|
entryToPage_.clear();
|
|
E2D_LOG_INFO("TextureAtlas cleared");
|
|
}
|
|
|
|
// ============================================================================
|
|
// TextureAtlasManager 单例实现
|
|
// ============================================================================
|
|
|
|
TextureAtlasManager &TextureAtlasManager::getInstance() {
|
|
static TextureAtlasManager instance;
|
|
return instance;
|
|
}
|
|
|
|
} // namespace extra2d
|