hardware.hpp

#ifndef TZ_GPU_HARDWARE_HPP
#define TZ_GPU_HARDWARE_HPP
#include "tz/core/handle.hpp"
#include "tz/core/error.hpp"
#include <string>
#include <span>
#include <cstddef>
 
namespace tz::gpu
{
    enum class hardware_type
    {
        gpu,
        integrated_gpu,
        cpu,
        unknown
    };
 
    enum class hardware_capabilities
    {
        graphics_compute,
        graphics_only,
        compute_only,
        neither
    };
 
    enum class hardware_feature_coverage
    {
        ideal,
        insufficient,
        poor,
        _count
    };
 
    using hardware_handle = tz::handle<hardware_type>;
 
    struct hardware
    {
        std::string name;
        std::uint64_t vram_size_mib;
        hardware_type type;
        hardware_feature_coverage features;
        hardware_capabilities caps;
 
        struct internals_t
        {
            hardware_handle i0;
            std::uint32_t i1;
        } internals;
 
        bool operator==(const hardware& rhs) const
        {
            return this->internals.i0 == rhs.internals.i0;
        }
 
        bool operator!=(const hardware& rhs) const
        {
            return this->internals.i0 != rhs.internals.i0;
        }
    };
 
    error_code iterate_hardware(std::span<hardware> hardwares, std::size_t* hardware_count = nullptr);
    hardware find_best_hardware();
    error_code use_hardware(hardware hw);
    hardware get_used_hardware();
    bool hardware_in_use();
}
 
#endif // TZ_GPU_HARDWARE_HPP