94 lines
2.7 KiB
C++
94 lines
2.7 KiB
C++
// Copyright (C) 2013 Jérôme Leclercq
|
|
// This file is part of the "Nazara Engine - Utility module"
|
|
// For conditions of distribution and use, see copyright notice in Config.hpp
|
|
|
|
#include <Nazara/Utility/Loaders/STB.hpp>
|
|
#include <stb_image/stb_image.h>
|
|
#include <Nazara/Core/Endianness.hpp>
|
|
#include <Nazara/Core/Error.hpp>
|
|
#include <Nazara/Core/File.hpp>
|
|
#include <Nazara/Core/InputStream.hpp>
|
|
#include <Nazara/Core/MemoryStream.hpp>
|
|
#include <Nazara/Utility/Image.hpp>
|
|
#include <Nazara/Utility/Debug.hpp>
|
|
|
|
namespace
|
|
{
|
|
int Read(void* userdata, char* data, int size)
|
|
{
|
|
NzInputStream* stream = reinterpret_cast<NzInputStream*>(userdata);
|
|
return static_cast<int>(stream->Read(data, size));
|
|
}
|
|
|
|
void Skip(void* userdata, unsigned int size)
|
|
{
|
|
NzInputStream* stream = reinterpret_cast<NzInputStream*>(userdata);
|
|
stream->Read(nullptr, size);
|
|
}
|
|
|
|
int Eof(void* userdata)
|
|
{
|
|
NzInputStream* stream = reinterpret_cast<NzInputStream*>(userdata);
|
|
return stream->GetCursorPos() >= stream->GetSize();
|
|
}
|
|
|
|
static stbi_io_callbacks callbacks = {Read, Skip, Eof};
|
|
|
|
bool IsSupported(const NzString& extension)
|
|
{
|
|
static std::set<NzString> supportedExtensions = {"bmp", "gif", "hdr", "jpg", "jpeg", "pic", "png", "psd", "tga"};
|
|
return supportedExtensions.find(extension) != supportedExtensions.end();
|
|
}
|
|
|
|
nzTernary Check(NzInputStream& stream, const NzImageParams& parameters)
|
|
{
|
|
NazaraUnused(parameters);
|
|
|
|
int width, height, bpp;
|
|
if (stbi_info_from_callbacks(&callbacks, &stream, &width, &height, &bpp))
|
|
return nzTernary_True;
|
|
else
|
|
return nzTernary_False;
|
|
}
|
|
|
|
bool Load(NzImage* image, NzInputStream& stream, const NzImageParams& parameters)
|
|
{
|
|
// Je charge tout en RGBA8 et je converti ensuite via la méthode Convert
|
|
// Ceci à cause d'un bug de STB lorsqu'il s'agit de charger certaines images (ex: JPG) en "default"
|
|
|
|
int width, height, bpp;
|
|
nzUInt8* ptr = stbi_load_from_callbacks(&callbacks, &stream, &width, &height, &bpp, STBI_rgb_alpha);
|
|
if (!ptr)
|
|
{
|
|
NazaraError("Failed to load image: " + NzString(stbi_failure_reason()));
|
|
return false;
|
|
}
|
|
|
|
if (!image->Create(nzImageType_2D, nzPixelFormat_RGBA8, width, height, 1, (parameters.levelCount > 0) ? parameters.levelCount : 1))
|
|
{
|
|
NazaraError("Failed to create image");
|
|
stbi_image_free(ptr);
|
|
|
|
return false;
|
|
}
|
|
|
|
image->Update(ptr);
|
|
stbi_image_free(ptr);
|
|
|
|
if (parameters.loadFormat != nzPixelFormat_Undefined)
|
|
image->Convert(parameters.loadFormat);
|
|
|
|
return true;
|
|
}
|
|
}
|
|
|
|
void NzLoaders_STB_Register()
|
|
{
|
|
NzImageLoader::RegisterLoader(IsSupported, Check, Load);
|
|
}
|
|
|
|
void NzLoaders_STB_Unregister()
|
|
{
|
|
NzImageLoader::UnregisterLoader(IsSupported, Check, Load);
|
|
}
|