evert/osb
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix ImageMetadataDatabase::calculateImageSize calling Image::readPngMetadata on non-PNG images.

Browse Source
This commit is contained in:
floydinator-git 2024-09-10 23:04:09 -04:00
parent 8457c2e954
commit 1f5e8a4629
3 changed files with 12 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
source/core/StarImage.cpp
View File

@ -17,6 +17,12 @@ void readPngData(png_structp pngPtr, png_bytep data, png_size_t length) {
((IODevice*)png_get_io_ptr(pngPtr))->readFull((char*)data, length);
};
bool Image::isPng(IODevicePtr device) {
png_byte header[8];
device->readAbsolute(0, (char*)header, sizeof(header));
return !png_sig_cmp(header, 0, sizeof(header));
}
Image Image::readPng(IODevicePtr device) {
png_byte header[8];
device->readFull((char*)header, sizeof(header));

1
source/core/StarImage.hpp
View File

@ -27,6 +27,7 @@ STAR_CLASS(Image);
class Image {
public:
static Image readPng(IODevicePtr device);
static bool isPng(IODevicePtr device);
// Returns the size and pixel format that would be constructed from the given
// png file, without actually loading it.
static tuple<Vec2U, PixelFormat> readPngMetadata(IODevicePtr device);

6
source/game/StarImageMetadataDatabase.cpp
View File

@ -174,7 +174,11 @@ Vec2U ImageMetadataDatabase::calculateImageSize(AssetPath const& path) const {
imageSize = *size;
} else {
locker.unlock();
imageSize = get<0>(Image::readPngMetadata(assets->openFile(path.basePath)));
auto file = assets->openFile(path.basePath);
if (Image::isPng(file))
imageSize = get<0>(Image::readPngMetadata(file));
else
imageSize = fallback();
locker.lock();
m_sizeCache[path.basePath] = imageSize;
}