153 lines
3.4 KiB
C++
153 lines
3.4 KiB
C++
/* Copyright (c) 2018-2025, Arm Limited and Contributors
|
|
*
|
|
* SPDX-License-Identifier: Apache-2.0
|
|
*
|
|
* Licensed under the Apache License, Version 2.0 the "License";
|
|
* you may not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
*/
|
|
|
|
#pragma once
|
|
|
|
#include <memory>
|
|
#include <string>
|
|
#include <typeinfo>
|
|
#include <vector>
|
|
|
|
#include <volk.h>
|
|
|
|
#include "core/image.h"
|
|
#include "core/image_view.h"
|
|
#include "scene_graph/component.h"
|
|
|
|
namespace vkb
|
|
{
|
|
namespace sg
|
|
{
|
|
/**
|
|
* @param format Vulkan format
|
|
* @return Whether the vulkan format is ASTC
|
|
*/
|
|
bool is_astc(VkFormat format);
|
|
|
|
/**
|
|
* @brief Mipmap information
|
|
*/
|
|
struct Mipmap
|
|
{
|
|
/// Mipmap level
|
|
uint32_t level = 0;
|
|
|
|
/// Byte offset used for uploading
|
|
uint32_t offset = 0;
|
|
|
|
/// Width depth and height of the mipmap
|
|
VkExtent3D extent = {0, 0, 0};
|
|
};
|
|
|
|
class Image : public Component
|
|
{
|
|
public:
|
|
/**
|
|
* @brief Type of content held in image.
|
|
* This helps to steer the image loaders when deciding what the format should be.
|
|
* Some image containers don't know whether the data they contain is sRGB or not.
|
|
* Since most applications save color images in sRGB, knowing that an image
|
|
* contains color data helps us to better guess its format when unknown.
|
|
*/
|
|
enum ContentType
|
|
{
|
|
Unknown,
|
|
Color,
|
|
Other
|
|
};
|
|
|
|
Image(const std::string &name, std::vector<uint8_t> &&data = {}, std::vector<Mipmap> &&mipmaps = {{}});
|
|
|
|
static std::unique_ptr<Image> load(const std::string &name, const std::string &uri, ContentType content_type);
|
|
|
|
virtual ~Image() = default;
|
|
|
|
virtual std::type_index get_type() override;
|
|
|
|
const std::vector<uint8_t> &get_data() const;
|
|
|
|
size_t get_data_hash() const;
|
|
|
|
void clear_data();
|
|
|
|
VkFormat get_format() const;
|
|
|
|
const VkExtent3D &get_extent() const;
|
|
|
|
const uint32_t get_layers() const;
|
|
|
|
const std::vector<Mipmap> &get_mipmaps() const;
|
|
|
|
const std::vector<std::vector<VkDeviceSize>> &get_offsets() const;
|
|
|
|
void generate_mipmaps();
|
|
|
|
void create_vk_image(vkb::core::DeviceC &device, VkImageViewType image_view_type = VK_IMAGE_VIEW_TYPE_2D, VkImageCreateFlags flags = 0);
|
|
|
|
const core::Image &get_vk_image() const;
|
|
|
|
const core::ImageView &get_vk_image_view() const;
|
|
|
|
void coerce_format_to_srgb();
|
|
|
|
protected:
|
|
std::vector<uint8_t> &get_mut_data();
|
|
|
|
void set_data(const uint8_t *raw_data, size_t size);
|
|
|
|
void set_format(VkFormat format);
|
|
|
|
void set_width(uint32_t width);
|
|
|
|
void set_height(uint32_t height);
|
|
|
|
void set_depth(uint32_t depth);
|
|
|
|
void set_layers(uint32_t layers);
|
|
|
|
void set_offsets(const std::vector<std::vector<VkDeviceSize>> &offsets);
|
|
|
|
Mipmap &get_mipmap(size_t index);
|
|
|
|
std::vector<Mipmap> &get_mut_mipmaps();
|
|
|
|
void update_hash();
|
|
|
|
void update_hash(size_t data_hash);
|
|
|
|
private:
|
|
std::vector<uint8_t> data;
|
|
|
|
size_t data_hash{0};
|
|
|
|
VkFormat format{VK_FORMAT_UNDEFINED};
|
|
|
|
uint32_t layers{1};
|
|
|
|
std::vector<Mipmap> mipmaps{{}};
|
|
|
|
// Offsets stored like offsets[array_layer][mipmap_layer]
|
|
std::vector<std::vector<VkDeviceSize>> offsets;
|
|
|
|
std::unique_ptr<core::Image> vk_image;
|
|
|
|
std::unique_ptr<core::ImageView> vk_image_view;
|
|
};
|
|
|
|
} // namespace sg
|
|
} // namespace vkb
|