File DeviceConfig.h

File List > Intern > rayx-core > src > Tracer > DeviceConfig.h

Go to the documentation of this file

#pragma once

#include <string>
#include <vector>

#include "Core.h"

namespace rayx {

struct RAYX_API DeviceConfig {
    enum RAYX_API DeviceType {
        None        = 0,
        CpuSerial   = 1 << 0,
        CpuParallel = 1 << 1,
        Cpu         = CpuSerial | CpuParallel,
        GpuCuda     = 1 << 2,
        Gpu         = GpuCuda,
        All         = Cpu | Gpu,
    };

    struct RAYX_API Device {
        using Index = size_t;
        using Score = size_t;

        DeviceType type;
        std::string name;
        Index index;
        Score score;
        bool enable;
    };

    DeviceConfig(DeviceType fetchedDeviceType = DeviceType::All);
    DeviceConfig(const DeviceConfig&) = default;
    DeviceConfig(DeviceConfig&&)      = default;

    DeviceConfig& operator=(const DeviceConfig&) = default;
    DeviceConfig& operator=(DeviceConfig&&)      = default;

    void dumpDevices() const;
    size_t enabledDevicesCount() const;

    DeviceConfig& disableAllDevices(DeviceType deviceType = DeviceType::All);
    DeviceConfig& enableAllDevices(DeviceType deviceType = DeviceType::All);

    DeviceConfig& disableDeviceByIndex(const Device::Index deviceIndex);
    DeviceConfig& enableDeviceByIndex(const Device::Index deviceIndex);

    DeviceConfig& enableBestDevice(DeviceType deviceType = DeviceType::All);

    static DeviceType availableDeviceTypes();

    std::vector<Device> devices;

  private:
    DeviceType m_fetchedDeviceType;
};

}  // namespace rayx