/*
 * Copyright (C) 2019 Intel Corporation
 *
 * SPDX-License-Identifier: MIT
 *
 */

#include "runtime/device/root_device.h"

#include "runtime/device/sub_device.h"
#include "runtime/os_interface/debug_settings_manager.h"

namespace NEO {

RootDevice::~RootDevice() = default;

uint32_t RootDevice::getNumSubDevices() const {
    return static_cast<uint32_t>(subdevices.size());
}

RootDevice::RootDevice(ExecutionEnvironment *executionEnvironment, uint32_t deviceIndex) : Device(executionEnvironment, deviceIndex) {}
bool RootDevice::createDeviceImpl() {
    auto status = Device::createDeviceImpl();
    if (!status) {
        return status;
    }
    auto numSubDevices = DebugManager.flags.CreateMultipleSubDevices.get();
    subdevices.reserve(numSubDevices);
    for (int i = 0; i < numSubDevices; i++) {

        auto subDevice = Device::create<SubDevice>(executionEnvironment, deviceIndex + i + 1, i, *this);
        if (!subDevice) {
            return false;
        }
        subdevices.push_back(std::unique_ptr<SubDevice>(subDevice));
    }
    return true;
}

/* We hide the retain and release function of BaseObject. */
void RootDevice::retain() {
    DEBUG_BREAK_IF(!isValid());
}

unique_ptr_if_unused<Device> RootDevice::release() {
    DEBUG_BREAK_IF(!isValid());
    return unique_ptr_if_unused<Device>(this, false);
}
DeviceBitfield RootDevice::getDeviceBitfieldForOsContext() const {
    DeviceBitfield deviceBitfield;
    deviceBitfield.set(deviceIndex);
    return deviceBitfield;
}
} // namespace NEO