/*
 * Copyright (c) 2019-2020 Arm Limited. All rights reserved.
 *
 * SPDX-License-Identifier: Apache-2.0
 *
 * Licensed under the Apache License, Version 2.0 (the License); you may
 * not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an AS IS BASIS, WITHOUT
 * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#if !defined ETHOSU_COMMON_H
#define ETHOSU_COMMON_H

#include "ethosu55_interface.h"

#if !defined(LOG_ENABLED)
#define LOG_INFO(format, ...)
#define LOG_ERR(format, ...)
#else
#define LOG_INFO(format, ...) fprintf(stdout, format, ##__VA_ARGS__)
#define LOG_ERR(format, ...)                                                                                           \
    fprintf(stderr, format, ##__VA_ARGS__);                                                                            \
    fflush(stderr)

#endif

#if defined(ASSERT_DISABLE)
#define ASSERT(args)
#else
#define ASSERT(args) assert(args)
#endif

#if defined(CPU_CORTEX_M55)
#define NPU_BASE ((uint32_t)0x41700000)
#else
#define NPU_BASE ((uint32_t)0x41105000)
#endif
#define UNUSED(x) ((void)x)

#define VER_STR(X) VNUM_STR(X)
#define VNUM_STR(X) #X

static const __attribute__((section("npu_driver_version"))) char driver_version_str[] = VER_STR(
    ETHOSU_DRIVER_VERSION_MAJOR) "." VER_STR(ETHOSU_DRIVER_VERSION_MINOR) "." VER_STR(ETHOSU_DRIVER_VERSION_PATCH);

static const __attribute__((section("npu_driver_arch_version"))) char driver_arch_version_str[] =
    VER_STR(NNX_ARCH_VERSION_MAJOR) "." VER_STR(NNX_ARCH_VERSION_MINOR) "." VER_STR(NNX_ARCH_VERSION_PATCH);

static inline uint32_t read_reg(uint32_t address)
{
    volatile uint32_t *reg = (uint32_t *)(uintptr_t)(NPU_BASE + address);
    return *reg;
}

static inline void write_reg(uint32_t address, uint32_t value)
{
    volatile uint32_t *reg = (uint32_t *)(uintptr_t)(NPU_BASE + address);
    *reg                   = value;
}

#endif // ETHOSU_COMMON_H