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

#include "runtime/helpers/aligned_memory.h"
#include "runtime/memory_manager/graphics_allocation.h"

namespace OCLRT {
bool GraphicsAllocation::isL3Capable() {
    auto ptr = ptrOffset(cpuPtr, static_cast<size_t>(this->allocationOffset));
    if (alignUp(ptr, MemoryConstants::cacheLineSize) == ptr && alignUp(this->size, MemoryConstants::cacheLineSize) == this->size) {
        return true;
    }
    return false;
}
GraphicsAllocation::GraphicsAllocation(void *cpuPtrIn, uint64_t gpuAddress, uint64_t baseAddress, size_t sizeIn) : gpuBaseAddress(baseAddress),
                                                                                                                   size(sizeIn),
                                                                                                                   cpuPtr(cpuPtrIn),
                                                                                                                   gpuAddress(gpuAddress),
                                                                                                                   taskCounts(maxOsContextCount) {
    initTaskCounts();
}

GraphicsAllocation::GraphicsAllocation(void *cpuPtrIn, size_t sizeIn, osHandle sharedHandleIn) : size(sizeIn),
                                                                                                 cpuPtr(cpuPtrIn),
                                                                                                 gpuAddress(castToUint64(cpuPtrIn)),
                                                                                                 sharedHandle(sharedHandleIn),
                                                                                                 taskCounts(maxOsContextCount) {
    initTaskCounts();
}
GraphicsAllocation::~GraphicsAllocation() = default;

bool GraphicsAllocation::peekWasUsed() const { return registeredContextsNum > 0; }

void GraphicsAllocation::updateTaskCount(uint32_t newTaskCount, uint32_t contextId) {
    UNRECOVERABLE_IF(contextId >= taskCounts.size());
    if (taskCounts[contextId] == ObjectNotUsed) {
        registeredContextsNum++;
    }
    if (newTaskCount == ObjectNotUsed) {
        registeredContextsNum--;
    }
    taskCounts[contextId] = newTaskCount;
}

uint32_t GraphicsAllocation::getTaskCount(uint32_t contextId) const {
    UNRECOVERABLE_IF(contextId >= taskCounts.size());
    return taskCounts[contextId];
}
void GraphicsAllocation::initTaskCounts() {
    for (auto i = 0u; i < taskCounts.size(); i++) {
        taskCounts[i] = ObjectNotUsed;
    }
}
} // namespace OCLRT