88 lines
2.6 KiB
C++
88 lines
2.6 KiB
C++
#include "Atlas.h"
|
|
|
|
#include <stdexcept>
|
|
#include "../../maths/LMPacker.h"
|
|
#include "Texture.h"
|
|
#include "ImageData.h"
|
|
|
|
Atlas::Atlas(ImageData* image, std::unordered_map<std::string, UVRegion> regions)
|
|
: texture(Texture::from(image)),
|
|
image(image),
|
|
regions(regions) {
|
|
}
|
|
|
|
Atlas::~Atlas() {
|
|
}
|
|
|
|
bool Atlas::has(const std::string& name) const {
|
|
return regions.find(name) != regions.end();
|
|
}
|
|
|
|
const UVRegion& Atlas::get(const std::string& name) const {
|
|
return regions.at(name);
|
|
}
|
|
|
|
Texture* Atlas::getTexture() const {
|
|
return texture.get();
|
|
}
|
|
|
|
ImageData* Atlas::getImage() const {
|
|
return image.get();
|
|
}
|
|
|
|
void AtlasBuilder::add(std::string name, ImageData* image) {
|
|
entries.push_back(atlasentry{name, std::shared_ptr<ImageData>(image)});
|
|
names.insert(name);
|
|
}
|
|
|
|
bool AtlasBuilder::has(const std::string& name) const {
|
|
return names.find(name) != names.end();
|
|
}
|
|
|
|
Atlas* AtlasBuilder::build(uint extrusion, uint maxResolution) {
|
|
auto sizes = std::make_unique<uint[]>(entries.size() * 2);
|
|
uint index = 0;
|
|
for (auto& entry : entries) {
|
|
auto image = entry.image;
|
|
sizes[index++] = image->getWidth();
|
|
sizes[index++] = image->getHeight();
|
|
}
|
|
LMPacker packer(sizes.get(), entries.size()*2);
|
|
sizes.reset(nullptr);
|
|
|
|
uint width = 32;
|
|
uint height = 32;
|
|
while (!packer.buildCompact(width, height, extrusion)) {
|
|
if (width > height) {
|
|
height *= 2;
|
|
} else {
|
|
width *= 2;
|
|
}
|
|
if (width > maxResolution || height > maxResolution) {
|
|
throw std::runtime_error("max atlas resolution "+
|
|
std::to_string(maxResolution)+" exceeded");
|
|
}
|
|
}
|
|
|
|
auto canvas = std::make_unique<ImageData>(ImageFormat::rgba8888, width, height);
|
|
std::unordered_map<std::string, UVRegion> regions;
|
|
std::vector<rectangle> rects = packer.getResult();
|
|
for (uint i = 0; i < entries.size(); i++) {
|
|
const rectangle& rect = rects[i];
|
|
const atlasentry& entry = entries[rect.idx];
|
|
uint x = rect.x;
|
|
uint y = rect.y;
|
|
uint w = rect.width;
|
|
uint h = rect.height;
|
|
canvas->blit(entry.image.get(), rect.x, rect.y);
|
|
for (uint j = 0; j < extrusion; j++) {
|
|
canvas->extrude(x - j, y - j, w + j*2, h + j*2);
|
|
}
|
|
float unitX = 1.0f / width;
|
|
float unitY = 1.0f / height;
|
|
regions[entry.name] = UVRegion(unitX * x, unitY * y,
|
|
unitX * (x + w), unitY * (y + h));
|
|
}
|
|
return new Atlas(canvas.release(), regions);
|
|
}
|