Move IndirectHeap to core and improve unit tests

Change-Id: If7e329fc12fe934eb0f6b1762d3b0d86aab287e6
Signed-off-by: Dunajski, Bartosz <bartosz.dunajski@intel.com>

core/unit_tests/indirect_heap/CMakeLists.txt

@@ -0,0 +1,13 @@
+#
+# Copyright (C) 2019 Intel Corporation
+#
+# SPDX-License-Identifier: MIT
+#
+
+set(NEO_CORE_INDIRECT_HEAP_TESTS
+  ${CMAKE_CURRENT_SOURCE_DIR}/CMakeLists.txt
+  ${CMAKE_CURRENT_SOURCE_DIR}/indirect_heap_tests.cpp
+)
+
+set_property(GLOBAL PROPERTY NEO_CORE_INDIRECT_HEAP_TESTS ${NEO_CORE_INDIRECT_HEAP_TESTS})
+add_subdirectories()

core/unit_tests/indirect_heap/indirect_heap_tests.cpp

@@ -0,0 +1,149 @@
+/*
+ * Copyright (C) 2017-2019 Intel Corporation
+ *
+ * SPDX-License-Identifier: MIT
+ *
+ */
+
+#include "core/indirect_heap/indirect_heap.h"
+#include "core/memory_manager/graphics_allocation.h"
+#include "test.h"
+
+using namespace NEO;
+
+struct IndirectHeapTest : public ::testing::Test {
+    class MyMockGraphicsAllocation : public GraphicsAllocation {
+      public:
+        MyMockGraphicsAllocation(void *ptr, size_t size)
+            : GraphicsAllocation(0, AllocationType::UNKNOWN, ptr, castToUint64(ptr), 0, size, MemoryPool::System4KBPages) {}
+    };
+
+    uint8_t buffer[256];
+    MyMockGraphicsAllocation gfxAllocation = {(void *)buffer, sizeof(buffer)};
+    IndirectHeap indirectHeap = {&gfxAllocation};
+};
+
+TEST_F(IndirectHeapTest, getSpaceTestSizeZero) {
+    EXPECT_NE(nullptr, indirectHeap.getSpace(0));
+}
+
+TEST_F(IndirectHeapTest, getSpaceTestSizeNonZero) {
+    EXPECT_NE(nullptr, indirectHeap.getSpace(sizeof(uint32_t)));
+}
+
+TEST_F(IndirectHeapTest, getSpaceTestReturnsWritablePointer) {
+    uint32_t cmd = 0xbaddf00d;
+    auto pCmd = indirectHeap.getSpace(sizeof(cmd));
+    ASSERT_NE(nullptr, pCmd);
+    *(uint32_t *)pCmd = cmd;
+}
+
+TEST_F(IndirectHeapTest, getSpaceReturnsDifferentPointersForEachRequest) {
+    auto pCmd = indirectHeap.getSpace(sizeof(uint32_t));
+    ASSERT_NE(nullptr, pCmd);
+    auto pCmd2 = indirectHeap.getSpace(sizeof(uint32_t));
+    ASSERT_NE(pCmd2, pCmd);
+}
+
+TEST_F(IndirectHeapTest, getMaxAvailableSpace) {
+    ASSERT_EQ(indirectHeap.getMaxAvailableSpace(), indirectHeap.getAvailableSpace());
+}
+
+TEST_F(IndirectHeapTest, getAvailableSpaceShouldBeNonZeroAfterInit) {
+    EXPECT_NE(0u, indirectHeap.getAvailableSpace());
+}
+
+TEST_F(IndirectHeapTest, getSpaceReducesAvailableSpace) {
+    auto originalAvailable = indirectHeap.getAvailableSpace();
+    indirectHeap.getSpace(sizeof(uint32_t));
+
+    EXPECT_LT(indirectHeap.getAvailableSpace(), originalAvailable);
+}
+
+TEST_F(IndirectHeapTest, align) {
+    size_t alignment = 64 * sizeof(uint8_t);
+    indirectHeap.align(alignment);
+
+    size_t size = 1;
+    auto ptr = indirectHeap.getSpace(size);
+    EXPECT_NE(nullptr, ptr);
+
+    uintptr_t address = (uintptr_t)ptr;
+    EXPECT_EQ(0u, address % alignment);
+}
+
+TEST_F(IndirectHeapTest, alignShouldAlignUpward) {
+    size_t size = 1;
+    auto ptr1 = indirectHeap.getSpace(size);
+    ASSERT_NE(nullptr, ptr1);
+
+    size_t alignment = 64 * sizeof(uint8_t);
+    indirectHeap.align(alignment);
+
+    auto ptr2 = indirectHeap.getSpace(size);
+    ASSERT_NE(nullptr, ptr2);
+
+    // Should align up
+    EXPECT_GT(ptr2, ptr1);
+}
+
+TEST_F(IndirectHeapTest, givenIndirectHeapWhenGetCpuBaseIsCalledThenCpuAddressIsReturned) {
+    auto base = indirectHeap.getCpuBase();
+    EXPECT_EQ(base, buffer);
+}
+
+using IndirectHeapWith4GbAllocatorTest = IndirectHeapTest;
+
+TEST_F(IndirectHeapWith4GbAllocatorTest, givenIndirectHeapNotSupporting4GbModeWhenAskedForHeapGpuStartOffsetThenZeroIsReturned) {
+    auto cpuBaseAddress = reinterpret_cast<void *>(0x3000);
+    MyMockGraphicsAllocation graphicsAllocation(cpuBaseAddress, 4096u);
+    graphicsAllocation.setGpuBaseAddress(4096u);
+    IndirectHeap indirectHeap(&graphicsAllocation, false);
+
+    EXPECT_EQ(0u, indirectHeap.getHeapGpuStartOffset());
+}
+
+TEST_F(IndirectHeapWith4GbAllocatorTest, givenIndirectHeapSupporting4GbModeWhenAskedForHeapGpuStartOffsetThenZeroIsReturned) {
+    auto cpuBaseAddress = reinterpret_cast<void *>(0x3000);
+    MyMockGraphicsAllocation graphicsAllocation(cpuBaseAddress, 4096u);
+    graphicsAllocation.setGpuBaseAddress(4096u);
+    IndirectHeap indirectHeap(&graphicsAllocation, true);
+
+    EXPECT_EQ(8192u, indirectHeap.getHeapGpuStartOffset());
+}
+
+TEST_F(IndirectHeapWith4GbAllocatorTest, givenIndirectHeapSupporting4GbModeWhenAskedForHeapBaseThenGpuBaseIsReturned) {
+    auto cpuBaseAddress = reinterpret_cast<void *>(0x2000);
+    MyMockGraphicsAllocation graphicsAllocation(cpuBaseAddress, 4096u);
+    graphicsAllocation.setGpuBaseAddress(4096u);
+    IndirectHeap indirectHeap(&graphicsAllocation, true);
+
+    EXPECT_EQ(4096u, indirectHeap.getHeapGpuBase());
+}
+
+TEST_F(IndirectHeapWith4GbAllocatorTest, givenIndirectHeapNotSupporting4GbModeWhenAskedForHeapBaseThenGpuAddressIsReturned) {
+    auto cpuBaseAddress = reinterpret_cast<void *>(0x2000);
+    MyMockGraphicsAllocation graphicsAllocation(cpuBaseAddress, 4096u);
+    graphicsAllocation.setGpuBaseAddress(4096u);
+    IndirectHeap indirectHeap(&graphicsAllocation, false);
+
+    EXPECT_EQ(8192u, indirectHeap.getHeapGpuBase());
+}
+
+TEST_F(IndirectHeapWith4GbAllocatorTest, givenIndirectHeapNotSupporting4GbModeWhenAskedForHeapSizeThenGraphicsAllocationSizeInPagesIsReturned) {
+    auto cpuBaseAddress = reinterpret_cast<void *>(0x2000);
+    MyMockGraphicsAllocation graphicsAllocation(cpuBaseAddress, 4096u);
+    graphicsAllocation.setGpuBaseAddress(4096u);
+    IndirectHeap indirectHeap(&graphicsAllocation, false);
+
+    EXPECT_EQ(1u, indirectHeap.getHeapSizeInPages());
+}
+
+TEST_F(IndirectHeapWith4GbAllocatorTest, givenIndirectHeapSupporting4GbModeWhenAskedForHeapSizeThen4GbSizeInPagesIsReturned) {
+    auto cpuBaseAddress = reinterpret_cast<void *>(0x2000);
+    MyMockGraphicsAllocation graphicsAllocation(cpuBaseAddress, 4096u);
+    graphicsAllocation.setGpuBaseAddress(4096u);
+    IndirectHeap indirectHeap(&graphicsAllocation, true);
+
+    EXPECT_EQ(MemoryConstants::sizeOf4GBinPageEntities, indirectHeap.getHeapSizeInPages());
+}