change signature

This commit is contained in:
ohyzha
2024-10-07 17:46:34 +03:00
parent a4e716006a
commit aaa8974006
6 changed files with 21 additions and 14 deletions

View File

@@ -61,9 +61,15 @@ namespace OpenVulkano::Image
return std::make_unique<Image>(std::move(result));
}
bool IImageLoader::GetDimensionsInternal(const std::string& filename, int& width, int& height)
Math::Vector2i IImageLoader::GetDimensionsInternal(const std::string& filename)
{
Math::Vector2i res = {};
int channels;
return stbi_info(filename.c_str(), &width, &height, &channels);
if (!stbi_info(filename.c_str(), &res.x, &res.y, &channels))
{
Logger::FILESYS->error("Failed to read image header of file {}", filename);
return Math::Vector2i{ -1, -1 };
}
return res;
}
}

View File

@@ -21,8 +21,8 @@ namespace OpenVulkano::Image
static std::unique_ptr<Image> loadData(const uint8_t* data, int size, int desiredChannels = 0);
virtual std::unique_ptr<Image> loadFromFile(const std::string& filePath) = 0;
virtual std::unique_ptr<Image> loadFromMemory(const std::vector<uint8_t>& buffer) = 0;
virtual bool GetImageDimensions(const std::string& filename, int& width, int& height) = 0;
virtual Math::Vector2i GetImageDimensions(const std::string& filename) = 0;
protected:
static bool GetDimensionsInternal(const std::string& filename, int& width, int& height);
static Math::Vector2i GetDimensionsInternal(const std::string& filename);
};
}

View File

@@ -31,7 +31,7 @@ namespace OpenVulkano::Image
return loadJpeg(buffer.data(), buffer.size());
}
bool ImageLoaderJpeg::GetImageDimensions(const std::string& filename, int& width, int& height)
Math::Vector2i ImageLoaderJpeg::GetImageDimensions(const std::string& filename)
{
#if __has_include("turbojpeg.h")
auto image = Utils::ReadFile(filename);
@@ -39,21 +39,22 @@ namespace OpenVulkano::Image
if (!jpegDecompressor)
{
Logger::FILESYS->error("Failed to read jpeg header. Error: {}", tjGetErrorStr());
return false;
return Math::Vector2i { -1, -1 };
}
Math::Vector2i res = {};
int size = 0, jpegSubsamp = 0;
int status = tjDecompressHeader2(jpegDecompressor, reinterpret_cast<unsigned char*>(image.Data()), image.Size(),
&width, &height, &jpegSubsamp);
&res.x, &res.y, &jpegSubsamp);
if (status != 0)
{
Logger::FILESYS->error("Failed to read jpeg header. Error: {}", tjGetErrorStr());
tjDestroy(jpegDecompressor);
return false;
return Math::Vector2i { -1, -1 };
}
tjDestroy(jpegDecompressor);
return true;
return res;
#else
return IImageLoader::GetDimensionsInternal(filename, width, height);
return IImageLoader::GetDimensionsInternal(filename);
#endif
}

View File

@@ -15,7 +15,7 @@ namespace OpenVulkano::Image
public:
std::unique_ptr<Image> loadFromFile(const std::string& filePath) override;
std::unique_ptr<Image> loadFromMemory(const std::vector<uint8_t>& buffer) override;
bool GetImageDimensions(const std::string& filename, int& width, int& height) override;
Math::Vector2i GetImageDimensions(const std::string& filename) override;
private:
std::unique_ptr<Image> loadJpeg(const uint8_t* data, size_t size);
};

View File

@@ -23,8 +23,8 @@ namespace OpenVulkano::Image
return loadData(buffer.data(), static_cast<int>(buffer.size()));
}
bool ImageLoaderPng::GetImageDimensions(const std::string& filename, int& width, int& height)
Math::Vector2i ImageLoaderPng::GetImageDimensions(const std::string& filename)
{
return GetDimensionsInternal(filename, width, height);
return GetDimensionsInternal(filename);
}
}

View File

@@ -15,6 +15,6 @@ namespace OpenVulkano::Image
public:
std::unique_ptr<Image> loadFromFile(const std::string& filePath) override;
std::unique_ptr<Image> loadFromMemory(const std::vector<uint8_t>& buffer) override;
bool GetImageDimensions(const std::string& filename, int& width, int& height) override;
Math::Vector2i GetImageDimensions(const std::string& filename) override;
};
}