Atlas minor refactor

This commit is contained in:
MihailRis 2024-03-18 20:06:58 +03:00
parent bca4de179d
commit ba762584a7
2 changed files with 22 additions and 31 deletions

View File

@ -5,51 +5,42 @@
#include "Texture.h"
#include "ImageData.h"
using std::vector;
using std::string;
using std::unique_ptr;
using std::shared_ptr;
using std::unordered_map;
Atlas::Atlas(ImageData* image,
unordered_map<string, UVRegion> regions)
: texture(Texture::from(image)),
image(image),
regions(regions) {
Atlas::Atlas(ImageData* image, std::unordered_map<std::string, UVRegion> regions)
: texture(Texture::from(image)),
image(image),
regions(regions) {
}
Atlas::~Atlas() {
delete image;
delete texture;
}
bool Atlas::has(string name) const {
bool Atlas::has(const std::string& name) const {
return regions.find(name) != regions.end();
}
const UVRegion& Atlas::get(string name) const {
const UVRegion& Atlas::get(const std::string& name) const {
return regions.at(name);
}
Texture* Atlas::getTexture() const {
return texture;
return texture.get();
}
ImageData* Atlas::getImage() const {
return image;
return image.get();
}
void AtlasBuilder::add(string name, ImageData* image) {
entries.push_back(atlasentry{name, shared_ptr<ImageData>(image)});
void AtlasBuilder::add(std::string name, ImageData* image) {
entries.push_back(atlasentry{name, std::shared_ptr<ImageData>(image)});
names.insert(name);
}
bool AtlasBuilder::has(string name) const {
bool AtlasBuilder::has(const std::string& name) const {
return names.find(name) != names.end();
}
Atlas* AtlasBuilder::build(uint extrusion, uint maxResolution) {
unique_ptr<uint[]> sizes (new uint[entries.size() * 2]);
auto sizes = std::make_unique<uint[]>(entries.size() * 2);
uint index = 0;
for (auto& entry : entries) {
auto image = entry.image;
@ -73,9 +64,9 @@ Atlas* AtlasBuilder::build(uint extrusion, uint maxResolution) {
}
}
unordered_map<string, UVRegion> regions;
unique_ptr<ImageData> canvas (new ImageData(ImageFormat::rgba8888, width, height));
vector<rectangle> rects = packer.getResult();
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];
@ -93,4 +84,4 @@ Atlas* AtlasBuilder::build(uint extrusion, uint maxResolution) {
unitX * (x + w), unitY * (y + h));
}
return new Atlas(canvas.release(), regions);
}
}

View File

@ -13,15 +13,15 @@ class ImageData;
class Texture;
class Atlas {
Texture* texture;
ImageData* image;
std::unique_ptr<Texture> texture;
std::unique_ptr<ImageData> image;
std::unordered_map<std::string, UVRegion> regions;
public:
Atlas(ImageData* image, std::unordered_map<std::string, UVRegion> regions);
~Atlas();
bool has(std::string name) const;
const UVRegion& get(std::string name) const;
bool has(const std::string& name) const;
const UVRegion& get(const std::string& name) const;
Texture* getTexture() const;
ImageData* getImage() const;
@ -39,10 +39,10 @@ class AtlasBuilder {
public:
AtlasBuilder() {}
void add(std::string name, ImageData* image);
bool has(std::string name) const;
bool has(const std::string& name) const;
const std::set<std::string>& getNames() { return names; };
Atlas* build(uint extrusion, uint maxResolution=8192);
};
#endif // GRAPHICS_ATLAS_H_
#endif // GRAPHICS_ATLAS_H_