Sean Keely
99 righe
3.7 KiB
C++
99 righe
3.7 KiB
C++
#ifndef AMD_HSA_EXT_IMAGE_IMAGE_MANAGER_H
|
|
#define AMD_HSA_EXT_IMAGE_IMAGE_MANAGER_H
|
|
|
|
#include <cstring>
|
|
#include "inc/hsa.h"
|
|
#include "inc/hsa_ext_image.h"
|
|
#include "resource.h"
|
|
#include "util.h"
|
|
|
|
namespace amd {
|
|
/// @brief Abstract class for creating AMD agent specific image / sampler
|
|
/// resources and data transfer.
|
|
class ImageManager {
|
|
public:
|
|
explicit ImageManager();
|
|
virtual ~ImageManager();
|
|
|
|
virtual hsa_status_t Initialize(hsa_agent_t agent_handle) = 0;
|
|
|
|
virtual void Cleanup() = 0;
|
|
|
|
/// @brief Retrieve device specific image property of a certain format
|
|
/// and geometry.
|
|
virtual ImageProperty GetImageProperty(
|
|
hsa_agent_t component, const hsa_ext_image_format_t& format,
|
|
hsa_ext_image_geometry_t geometry) const = 0;
|
|
|
|
/// @brief Retrieve device specific supported max width, height, depth,
|
|
/// and array size of an image geometry.
|
|
virtual void GetImageInfoMaxDimension(hsa_agent_t component,
|
|
hsa_ext_image_geometry_t geometry,
|
|
uint32_t& width, uint32_t& height,
|
|
uint32_t& depth,
|
|
uint32_t& array_size) const = 0;
|
|
|
|
/// @brief Calculate the size and alignment of the backing storage of an
|
|
/// image.
|
|
virtual hsa_status_t CalculateImageSizeAndAlignment(
|
|
hsa_agent_t component, const hsa_ext_image_descriptor_t& desc,
|
|
hsa_ext_image_data_layout_t image_data_layout,
|
|
size_t image_data_row_pitch,
|
|
size_t image_data_slice_pitch,
|
|
hsa_ext_image_data_info_t& image_info) const = 0;
|
|
|
|
/// @brief Fill image structure with device specific image object.
|
|
virtual hsa_status_t PopulateImageSrd(Image& image) const = 0;
|
|
|
|
/// @brief Fill image structure with device specific image object using the given format.
|
|
virtual hsa_status_t PopulateImageSrd(Image& image, const metadata_amd_t* desc) const = 0;
|
|
|
|
/// @brief Modify device specific image object according to the specified
|
|
/// new format.
|
|
virtual hsa_status_t ModifyImageSrd(
|
|
Image& image, hsa_ext_image_format_t& new_format) const = 0;
|
|
|
|
/// @brief Fill sampler structure with device specific sampler object.
|
|
virtual hsa_status_t PopulateSamplerSrd(Sampler& sampler) const = 0;
|
|
|
|
// @brief Copy the content of a linear memory to an image object.
|
|
virtual hsa_status_t CopyBufferToImage(
|
|
const void* src_memory, size_t src_row_pitch, size_t src_slice_pitch,
|
|
const Image& dst_image, const hsa_ext_image_region_t& image_region);
|
|
|
|
/// @brief Copy the content of an image object to a linear memory.
|
|
virtual hsa_status_t CopyImageToBuffer(
|
|
const Image& src_image, void* dst_memory, size_t dst_row_pitch,
|
|
size_t dst_slice_pitch, const hsa_ext_image_region_t& image_region);
|
|
|
|
/// @brief Transfer images backing storage.
|
|
virtual hsa_status_t CopyImage(const Image& dst_image, const Image& src_image,
|
|
const hsa_dim3_t& dst_origin,
|
|
const hsa_dim3_t& src_origin,
|
|
const hsa_dim3_t size);
|
|
|
|
/// @brief Fill image backing storage using host copy.
|
|
virtual hsa_status_t FillImage(const Image& image, const void* pattern,
|
|
const hsa_ext_image_region_t& region);
|
|
|
|
protected:
|
|
static uint16_t FloatToHalf(float in);
|
|
|
|
static inline float Normalize(uint8_t u_val);
|
|
|
|
static inline uint8_t Denormalize(float f_val);
|
|
|
|
static float StandardToLinearRGB(float s_val);
|
|
|
|
static float LinearToStandardRGB(float l_val);
|
|
|
|
static void FormatPattern(const hsa_ext_image_format_t& format,
|
|
const void* pattern_in, void* pattern_out);
|
|
|
|
private:
|
|
DISALLOW_COPY_AND_ASSIGN(ImageManager);
|
|
};
|
|
|
|
} // namespace
|
|
#endif // AMD_HSA_EXT_IMAGE_IMAGE_MANAGER_H
|