// Copyright (C) 2018-2020 Intel Corporation
// SPDX-License-Identifier: Apache-2.0
//

#include <cstdint>
#include <vector>
#include <thread>
#include <memory>

#include <gtest/gtest.h>
#include <memory/gna_allocator.hpp>
#include "gna_device.hpp"

// dummy definitions to work around issue with Linux userspace library
typedef unsigned long long time_tsc;
typedef struct {
    time_tsc            start;      // time value on profiler start
    time_tsc            stop;       // time value on profiler stop
    time_tsc            passed;     // time passed between start and stop
} intel_gna_profiler_tsc;

void profilerTscStop(intel_gna_profiler_tsc* p) {
    if (NULL == p) return;
    p->passed = 0;
    p->stop = 0;
    p->start = 0;
}

void profilerTscStartAccumulate(intel_gna_profiler_tsc* p) {
    if (NULL == p) return;
    p->stop = 0;
    p->start = 0;
}

void profilerTscStopAccumulate(intel_gna_profiler_tsc* p) {
    if (NULL == p) return;
    p->stop = 0;
    p->passed += p->stop - p->start;
}

class GNAAllocatorTest : public ::testing::Test {
protected:
    std::shared_ptr<GNADeviceHelper> gnadevice;
    void SetUp() override  {
       // gnadevice.reset(new GNADeviceHelper());
    }
};

TEST_F(GNAAllocatorTest, canAllocateStdMemory) {
    auto sp = GNAPluginNS::memory::make_polymorph<std::allocator<uint8_t>>();
    uint8_t *x = nullptr;
    ASSERT_NO_THROW(x = sp.allocate(100));
    ASSERT_NE(x, nullptr);
    ASSERT_NO_THROW(sp.deallocate(x, 100));
}

TEST_F(GNAAllocatorTest, canAllocateGNAMemory) {
    // GNA device can be opened one per process for now
    gnadevice.reset(new GNADeviceHelper());
    auto sp = GNAPluginNS::memory::make_polymorph<GNAPluginNS::memory::GNAAllocator>(gnadevice);
    uint8_t *x = nullptr;
    ASSERT_NO_THROW(x = sp.allocate(100));
    ASSERT_NE(x, nullptr);
    ASSERT_NO_THROW(sp.deallocate(x, 100));
}

TEST_F(GNAAllocatorTest, canOpenDevice) {
    std::thread th([]() {
        GNADeviceHelper h1;
    });

    th.join();

    std::thread th2([]() {
       GNADeviceHelper h1;
    });

    th2.join();
}
