openspades/Sources/Core/BitmapAtlasGenerator.cpp
Tomoaki Kawada ebd79c0099 Change #include style
All `#include`s to other subsystem now use angle brackets, e.g., `#include <Core/Math.h>`.
2016-12-03 18:49:07 +09:00

140 lines
3.5 KiB
C++

/*
Copyright (c) 2013 yvt
This file is part of OpenSpades.
OpenSpades 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, either version 3 of the License, or
(at your option) any later version.
OpenSpades 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 OpenSpades. If not, see <http://www.gnu.org/licenses/>.
*/
#include <cstring>
#include <binpack2d/binpack2d.hpp>
#include "Bitmap.h"
#include "BitmapAtlasGenerator.h"
#include "Debug.h"
#include "Exception.h"
namespace spades {
BitmapAtlasGenerator::BitmapAtlasGenerator() {}
BitmapAtlasGenerator::~BitmapAtlasGenerator() {}
void BitmapAtlasGenerator::AddBitmap(spades::Bitmap *b) { bmps.push_back(b); }
int BitmapAtlasGenerator::MinRectSize() {
int area = 0;
for (size_t i = 0; i < bmps.size(); i++) {
Bitmap *bmp = bmps[i];
area += bmp->GetWidth() * bmp->GetHeight();
}
for (int b = 1; b < 16; b++) {
int s = 1 << b;
if (s * s >= area)
return s;
}
SPAssert(false);
return 0;
}
int BitmapAtlasGenerator::MinWidth() {
int w = 0;
for (size_t i = 0; i < bmps.size(); i++) {
Bitmap *bmp = bmps[i];
if (bmp->GetWidth() > w)
w = bmp->GetWidth();
}
for (int b = 1; b < 16; b++) {
int s = 1 << b;
if (s >= w)
return s;
}
SPAssert(false);
return 0;
}
int BitmapAtlasGenerator::MinHeight() {
int w = 0;
for (size_t i = 0; i < bmps.size(); i++) {
Bitmap *bmp = bmps[i];
if (bmp->GetHeight() > w)
w = bmp->GetWidth();
}
for (int b = 1; b < 16; b++) {
int s = 1 << b;
if (s >= w)
return s;
}
SPAssert(false);
return 0;
}
BitmapAtlasGenerator::Result BitmapAtlasGenerator::Pack() {
BinPack2D::ContentAccumulator<Item> items;
int minSize = MinRectSize();
int minW = MinWidth(), minH = MinHeight();
minSize = std::max(minSize, std::min(minW, minH));
int mw = minSize, mh = minSize;
for (size_t i = 0; i < bmps.size(); i++) {
Bitmap *b = bmps[i];
Item itm = {b, 0, 0, b->GetWidth(), b->GetHeight()};
items += BinPack2D::Content<Item>(itm, BinPack2D::Coord(),
BinPack2D::Size(itm.w, itm.h), false);
}
while (true) {
int ww = std::max(mw, minW), hh = std::max(mh, minH);
BinPack2D::CanvasArray<Item> canvasArray =
BinPack2D::UniformCanvasArrayBuilder<Item>(ww, hh, 1).Build();
BinPack2D::ContentAccumulator<Item> remainder;
if (canvasArray.Place(items, remainder) && remainder.Get().empty()) {
BinPack2D::ContentAccumulator<Item> output;
canvasArray.CollectContent(output);
Bitmap *outbmp = new Bitmap(ww, hh);
uint32_t *outPixels = outbmp->GetPixels();
Result result;
for (BinPack2D::Content<Item>::Vector::iterator it = output.Get().begin();
it != output.Get().end(); it++) {
const BinPack2D::Content<Item> &c = *it;
Item itm = c.content;
itm.x = c.coord.x;
itm.y = c.coord.y;
result.items.push_back(itm);
Bitmap *b = itm.bitmap;
uint32_t *inPixels = b->GetPixels();
int w = itm.w, h = itm.h;
for (int i = 0; i < h; i++) {
memcpy(outPixels + itm.x + (itm.y + i) * ww, inPixels + i * w, w * 4);
}
}
result.bitmap = outbmp;
return result;
}
mw <<= 1;
mh <<= 1;
}
}
}