//
// Copyright © 2017 Arm Ltd. All rights reserved.
// SPDX-License-Identifier: MIT
//

#pragma once

#include <armnn/Tensor.hpp>
#include <armnn/utility/Assert.hpp>

#include <cstddef>
#include <vector>

template <typename T, std::size_t n>
struct LayerTestResult
{
    LayerTestResult(const armnn::TensorInfo& outputInfo)
        : m_Supported(true)
        , m_CompareBoolean(false)
    {
        m_ActualData.reserve(outputInfo.GetNumElements());
        m_ExpectedData.reserve(outputInfo.GetNumElements());
        m_ActualShape = outputInfo.GetShape();
        m_ExpectedShape = outputInfo.GetShape();
    }

    LayerTestResult(const std::vector<T>& actualData,
                    const std::vector<T>& expectedData,
                    const armnn::TensorShape& actualShape,
                    const armnn::TensorShape& expectedShape)
        : m_ActualData(actualData)
        , m_ExpectedData(expectedData)
        , m_ActualShape(actualShape)
        , m_ExpectedShape(expectedShape)
        , m_Supported(true)
        , m_CompareBoolean(false)
    {}

    LayerTestResult(const std::vector<T>& actualData,
                    const std::vector<T>& expectedData,
                    const armnn::TensorShape& actualShape,
                    const armnn::TensorShape& expectedShape,
                    const bool compareBoolean)
        : m_ActualData(actualData)
        , m_ExpectedData(expectedData)
        , m_ActualShape(actualShape)
        , m_ExpectedShape(expectedShape)
        , m_Supported(true)
        , m_CompareBoolean(compareBoolean)
    {}

    std::vector<T> m_ActualData;
    std::vector<T> m_ExpectedData;
    armnn::TensorShape m_ActualShape;
    armnn::TensorShape m_ExpectedShape;

    bool m_Supported;
    bool m_CompareBoolean;
};