/*
 * Copyright (c) 2023 Nordic Semiconductor ASA
 *
 * SPDX-License-Identifier: Apache-2.0
 *
 */

#include "nrfx.h"
#if defined(NRF52833_XXAA)
#include "hal/nrf_ccm.h"
#include "hal/nrf_rtc.h"
#include "hal/nrf_gpio.h"
#include "hal/nrf_gpiote.h"
#include "hal/nrf_temp.h"
#include "hal/nrf_radio.h"
#include "hal/nrf_egu.h"
#include "hal/nrf_timer.h"
#include "hal/nrf_clock.h"
#include "hal/nrf_rng.h"
#include "hal/nrf_aar.h"
#include "hal/nrf_ppi.h"
#include "hal/nrf_ecb.h"
#include "hal/nrf_uart.h"
#include "hal/nrf_uarte.h"
#elif defined(NRF5340_XXAA_NETWORK)
#include "hal/nrf_aar.h"
#include "hal/nrf_ccm.h"
#include "hal/nrf_ecb.h"
#include "hal/nrf_ipc.h"
#include "hal/nrf_radio.h"
#include "hal/nrf_rng.h"
#include "hal/nrf_temp.h"

#include "hal/nrf_clock.h"
#include "hal/nrf_dppi.h"
#include "hal/nrf_egu.h"
#include "hal/nrf_gpiote.h"
#include "hal/nrf_ipc.h"
#include "hal/nrf_rtc.h"
#include "hal/nrf_timer.h"
#include "hal/nrf_uarte.h"
#elif defined(NRF5340_XXAA_APPLICATION)
#include "hal/nrf_clock.h"
#include "hal/nrf_dppi.h"
#include "hal/nrf_egu.h"
#include "hal/nrf_gpiote.h"
#include "hal/nrf_ipc.h"
#include "hal/nrf_rtc.h"
#include "hal/nrf_timer.h"
#include "hal/nrf_uarte.h"
#elif defined(NRF54L15_XXAA)
#include "hal/nrf_aar.h"
#include "hal/nrf_ccm.h"
#include "hal/nrf_clock.h"
#include "hal/nrf_ecb.h"
#include "hal/nrf_egu.h"
#include "hal/nrf_dppi.h"
#include "hal/nrf_gpiote.h"
#include "hal/nrf_grtc.h"
//#include "hal/nrf_pdm.h" //Pending support in real HAL
#include "hal/nrf_ppib.h"
#include "hal/nrf_pwm.h"
#include "hal/nrf_radio.h"
#include "hal/nrf_timer.h"
#include "hal/nrf_temp.h"
#include "hal/nrf_uarte.h"
#endif

#include "bs_tracing.h"

#define PERIPHERAL_REG_BASE(per, nbr, post) \
    (void*)NRF_##per##nbr##post##_BASE

#define IS_PERIPHERAL_REG(p, per, nbr, post) \
    (p >= PERIPHERAL_REG_BASE(per, nbr, post)) && \
    ((uintptr_t)p < (uintptr_t)PERIPHERAL_REG_BASE(per, nbr, post) + sizeof(NRF_##per##_Type))

typedef void (*subscribe_set_f)(void *, int , uint8_t);
typedef void (*subscribe_clear_f)(void *, int);
typedef void (*task_trigger_f)(void *, int);

/*
 * Given a peripheral task/event (task_event) value (A value of a nrf_<peri>_task_t or nrf_<peri>_event_t)
 * return true if it is a task, or false if it is an event
 */
static bool nrf_hack_is_task(int task_event) {
  /*
   * In the NRF peripherals:
   *  [0x000..0x080) are tasks
   *  [0x080..0x100) are subscribe registers
   *  [0x100..0x180) are events
   *  [0x180..0x200) are publish registers
   *  Note: For the nrf54 RADIO, each group is twice as big, but tasks are still < 0x100
   */
  if (task_event < 0x100) {
    return true;
  } else {
    return false;
  }
}

/*
 * Given a pointer to a task or event register in an unknown peripheral
 * Get
 *  **p_reg: A pointer to the register base address
 *  *set_f/clear_f: pointers to the HAL subscribe set and clear functions that correspond to that peripheral
 *  *trigger_f: pointer to the HAL task trigger function corresponding to that peripheral
 * The task/event *offset* corresponding to that subscribe register (input to most nrf HAL functions)
 */
static void nrf_hack_get_task_from_ptr(void *task_reg,
                                       void **p_reg,
                                       subscribe_set_f *set_f,
                                       subscribe_clear_f *clear_f,
                                       task_trigger_f *trigger_f,
                                       int *task) {

#if defined(DPPI_PRESENT)
#define IF_PER(per, nbr, post, lname)                            \
  if (IS_PERIPHERAL_REG(task_reg, per, nbr, post)) {             \
    *p_reg = PERIPHERAL_REG_BASE(per, nbr, post);                \
    *task = (intptr_t)task_reg - (intptr_t)*p_reg;               \
    *set_f = (subscribe_set_f)nrf_##lname##_subscribe_set;       \
    *clear_f = (subscribe_clear_f)nrf_##lname##_subscribe_clear; \
    *trigger_f = (task_trigger_f)nrf_##lname##_task_trigger;\
    return;
#else
#define IF_PER(per, nbr, post, lname)                            \
  if (IS_PERIPHERAL_REG(task_reg, per, nbr, post)) {             \
    *p_reg = PERIPHERAL_REG_BASE(per, nbr, post);                \
    *task = (intptr_t)task_reg - (intptr_t)*p_reg;               \
    *set_f = NULL;                                               \
    *clear_f = NULL;                                             \
    *trigger_f = (task_trigger_f)nrf_##lname##_task_trigger;\
    return;
#endif

#if defined(NRF52833_XXAA)
  IF_PER(CLOCK, , , clock)
  } else IF_PER(RADIO, , , radio)
  } else IF_PER(UART, 0, , uart)
  } else IF_PER(UARTE, 0, , uarte)
  } else IF_PER(UARTE, 1, , uarte)
  } else IF_PER(RNG, , , rng)
  } else IF_PER(GPIOTE, , , gpiote)
  } else IF_PER(TIMER, 0, , timer)
  } else IF_PER(TIMER, 1, , timer)
  } else IF_PER(TIMER, 2, , timer)
  } else IF_PER(TIMER, 3, , timer)
  } else IF_PER(TIMER, 4, , timer)
  } else IF_PER(ECB, , , ecb)
  } else IF_PER(AAR, , , aar)
  } else IF_PER(CCM, , , ccm)
  } else IF_PER(PPI, , , ppi)
  } else IF_PER(EGU, 0, , egu)
  } else IF_PER(EGU, 1, , egu)
  } else IF_PER(EGU, 2, , egu)
  } else IF_PER(EGU, 3, , egu)
  } else IF_PER(EGU, 4, , egu)
  } else IF_PER(EGU, 5, , egu)
  } else IF_PER(RTC, 0, , rtc)
  } else IF_PER(RTC, 1, , rtc)
  } else IF_PER(RTC, 2, , rtc)
  } else IF_PER(TEMP, , , temp)
  } else {
    bs_trace_error_time_line("Tried to look for a task register not known to these HW models\n");
    return; /* unreachable */
  }
#elif defined(NRF5340_XXAA_NETWORK)
  /*IF_PER(POWER, , _NS, power)
  } else*/ IF_PER(CLOCK, , _NS, clock)
  } else IF_PER(RADIO, , _NS, radio)
  } else IF_PER(RNG, , _NS, rng)
  } else IF_PER(GPIOTE, , _NS, gpiote)
  /* 11 WDT */
  } else IF_PER(TIMER, 0, _NS, timer)
  } else IF_PER(ECB, , _NS, ecb)
  } else IF_PER(AAR, , _NS, aar)
  } else IF_PER(CCM, , _NS, ccm)
  } else IF_PER(DPPIC, , _NS, dppi)
  } else IF_PER(TEMP, , _NS, temp)
  } else IF_PER(RTC, 0, _NS, rtc)
  } else IF_PER(IPC, , _NS, ipc)
  } else IF_PER(UARTE, 0, _NS, uarte)
  } else IF_PER(EGU, 0, _NS, egu)
  } else IF_PER(RTC, 1, _NS, rtc)
  } else IF_PER(TIMER, 1, _NS, timer)
  } else IF_PER(TIMER, 2, _NS, timer)
  } else {
    bs_trace_error_time_line("Tried to look for a task register not known to these HW models\n");
    return; /* unreachable */
  }
#elif defined(NRF5340_XXAA_APPLICATION)
  /*IF_PER(DCNF, , _NS, dcnf)
  } else IF_PER(FPU, , _NS, fpu)
  } else IF_PER(CACHE, , _NS, cache)
  } else IF_PER(SPU, , _NS, spu)
  } else IF_PER(OSCILLARTORS, , _NS, oscillators)
  } else IF_PER(REGULATORS, , _NS, regulators)
  } else */ IF_PER(CLOCK, , _NS, clock)
  /*} else IF_PER(POWER, , _NS, power)
  } else IF_PER(RESET, , _NS, reset)
  } else IF_PER(CTRLAP, , _NS, ctrlap)
  } else IF_PER(SPIM, 0, _NS, spi) */
  } else IF_PER(UARTE, 0, _NS, uarte)
  /*} else IF_PER(SPIM, 1, _NS, spi)*/
  } else IF_PER(UARTE, 1, _NS, uarte)
  /*} else IF_PER(SPIM, 4, _NS, spi)
  } else IF_PER(SPIM, 2, _NS, spi)*/
  } else IF_PER(UARTE, 2, _NS, uarte)
  /*} else IF_PER(SPIM, 3, _NS, spi) */
  } else IF_PER(UARTE, 3, _NS, uarte)
  } else IF_PER(GPIOTE, 0, _S, gpiote)
  /*} else IF_PER(SAADC, , _NS, saadc)*/
  } else IF_PER(TIMER, 0, _NS, timer)
  } else IF_PER(TIMER, 1, _NS, timer)
  } else IF_PER(TIMER, 2, _NS, timer)
  } else IF_PER(RTC, 0, _NS, rtc)
  } else IF_PER(RTC, 1, _NS, rtc)
  } else IF_PER(DPPIC, , _NS, dppi)
  /*} else IF_PER(WDT0, , _NS, dppi)
  } else IF_PER(WDT1, , _NS, dppi)
  } else IF_PER(COMP, , _NS, dppi)*/
  } else IF_PER(EGU, 0, _NS, egu)
  } else IF_PER(EGU, 1, _NS, egu)
  } else IF_PER(EGU, 2, _NS, egu)
  } else IF_PER(EGU, 3, _NS, egu)
  } else IF_PER(EGU, 4, _NS, egu)
  } else IF_PER(EGU, 5, _NS, egu)
  /*} else IF_PER(PWM0, , _NS, dppi)
  } else IF_PER(PWM1, , _NS, dppi)
  } else IF_PER(PWM2, , _NS, dppi)
  } else IF_PER(PWM3, , _NS, dppi)
  } else IF_PER(PDM0, , _NS, dppi)
  } else IF_PER(I2S0, , _NS, dppi)*/
  } else IF_PER(IPC, , _NS, ipc)
  /* QSPI, NFCT */
  } else {
    bs_trace_error_time_line("Tried to look for a task register not known to these HW models\n");
    return; /* unreachable */
  }
#elif defined(NRF54L15_XXAA)
  /*IF_PER(SPU, 00, _S, spu)
  } else IF_PER(MPC, 00, _S, mpc)
  } else*/ IF_PER(DPPIC, 00, _S, dppi)
  //} else IF_PER(PPIB, 00, _S, ppib)
  //} else IF_PER(KMU, , _S, kmu)
  } else IF_PER(AAR, 00, _S, aar)
  } else IF_PER(CCM, 00, _S, ccm)
  } else IF_PER(ECB, 00, _S, ecb)
  //} else IF_PER(CRCEN, , _S, cracen)
  //} else IF_PER(SPIM, 00, _S, spi)
  } else IF_PER(UARTE, 00, _S, uarte)
  //} else IF_PER(GLITCHDET, , _S, glitchdet)
  //} else IF_PER(RRAMC, , _S, rramc)
  //} else IF_PER(VPR, 00 , _S, vpr)
  //} else IF_PER(CTRLAP, , _S, ctrlap)
  } else IF_PER(TIMER, 00, _S, timer)
  //} else IF_PER(SPU, 10, _S, spu)
  } else IF_PER(DPPIC, 10, _S, dppi)
  //} else IF_PER(PPIB, 10, _S, ppib)
  //} else IF_PER(PPIB, 11, _S, ppib)
  } else IF_PER(TIMER, 10, _S, timer)
  } else IF_PER(EGU, 10, _S, egu)
  } else IF_PER(RADIO, , _S, radio)
  //} else IF_PER(SPU, 20, _S, spu)
  } else IF_PER(DPPIC, 20, _S, dppi)
  //} else IF_PER(PPIB, 20, _S, ppib)
  //} else IF_PER(PPIB, 21, _S, ppib)
  //} else IF_PER(PPIB, 22, _S, ppib)
  //} else IF_PER(SPIM, 20, _S, spi)
  //} else IF_PER(TWIM, 20, _S, twi)
  } else IF_PER(UARTE, 20, _S, uarte)
  //} else IF_PER(SPIM, 21, _S, spi)
  //} else IF_PER(TWIM, 21, _S, twi)
  } else IF_PER(UARTE, 21, _S, uarte)
  //} else IF_PER(SPIM, 22, _S, spi)
  //} else IF_PER(TWIM, 22, _S, twi)
  } else IF_PER(UARTE, 22, _S, uarte)
  } else IF_PER(EGU, 20, _S, egu)
  } else IF_PER(TIMER, 20, _S, timer)
  } else IF_PER(TIMER, 21, _S, timer)
  } else IF_PER(TIMER, 22, _S, timer)
  } else IF_PER(TIMER, 23, _S, timer)
  } else IF_PER(TIMER, 24, _S, timer)
  //} else IF_PER(MEMCONF, , _S, memconf)
  //} else IF_PER(PDM, 20, _S, pdm)
  //} else IF_PER(PDM, 21, _S, pdm)
  //} else IF_PER(PWM, 20, _S, pwm)
  //} else IF_PER(PWM, 21, _S, pwm)
  //} else IF_PER(PWM, 22, _S, pwm)
  //} else IF_PER(SAADC, , _S, saadc)
  //} else IF_PER(NFCT, , _S, nfct)
  } else IF_PER(TEMP, , _S, temp)
  } else IF_PER(GPIOTE, 20, _S, gpiote)
  //} else IF_PER(TAMPC, , _S, tamp)
  //} else IF_PER(I2S, 20, _S, i2s)
  //} else IF_PER(QDEC, 20, _S, qdec)
  //} else IF_PER(QDEC, 21, _S, qdec)
  } else IF_PER(GRTC, , _S, grtc)
  //} else IF_PER(SPU, 30, _S, spu)
  } else IF_PER(DPPIC, 30, _S, dppi)
  //} else IF_PER(PPIB, 30, _S, ppib)
  //} else IF_PER(SPIM, 30, _S, spi)
  //} else IF_PER(TWIM, 30, _S, twi)
  } else IF_PER(UARTE, 30, _S, uarte)
  //} else IF_PER(COMP, , _S, comp)
  //} else IF_PER(LPCOMP, , _S, lpcomp)
  //} else IF_PER(WDT, 30, _S, wdt)
  //} else IF_PER(WDT, 31, _S, wdt)
  } else IF_PER(GPIOTE, 30, _S, gpiote)
  } else IF_PER(CLOCK, , _NS, clock)
  //} else IF_PER(POWER, , _NS, power)
  //} else IF_PER(RESET, , _NS, reset)
  //} else IF_PER(OSCILLARTORS, , _NS, oscillators)
  //} else IF_PER(REGULATORS, , _NS, regulators)
  } else {
    bs_trace_error_time_line("Tried to look for a task register not known to these HW models\n");
    return; /* unreachable */
  }
#endif

#undef IF_PER
}

#if defined(DPPI_PRESENT)
/*
 * Given a *<subscribe>* register (NOT a task register),
 * set it to be enabled and connected to <channel>, by calling the appropriate
 * HAL set replacement function.
 *
 * This function provides the backend for the NRF_DPPI_ENDPOINT_SETUP HAL macro
 * for simulation.
 *
 * Note: This function also accepts publish registers, but ignores them
 * (so as to be callable from NRF_DPPI_ENDPOINT_SETUP() without any check)
 */
void nrf_dppi_hack_subscribe_set(void *sub_reg, unsigned int channel)
{
  void *p_reg;
  int task;
  subscribe_set_f set_f;
  subscribe_clear_f clear_f;
  task_trigger_f task_trigger_f;

  intptr_t task_reg_pr = (intptr_t)sub_reg - NRF_SUBSCRIBE_PUBLISH_OFFSET((uintptr_t)sub_reg);

  nrf_hack_get_task_from_ptr((void *)task_reg_pr, &p_reg, &set_f, &clear_f, &task_trigger_f, &task);
  if (nrf_hack_is_task(task)) {
    set_f(p_reg, task, channel);
  }
}

/*
 * Given a *<subscribe>* register (NOT a task register),
 * clear it/set it to be disabled, by calling the appropriate
 * HAL clear replacement function.
 *
 * This function provides the backend for the NRF_DPPI_ENDPOINT_CLEAR HAL macro
 * for simulation.
 *
 * Note: This function also accepts publish registers, but ignores them
 * (so as to be callable from NRF_DPPI_ENDPOINT_SETUP() without any check)
 */
void nrf_dppi_hack_subscribe_clear(void *sub_reg)
{
  void *p_reg;
  int task;
  subscribe_set_f set_f;
  subscribe_clear_f clear_f;
  task_trigger_f trigger_f;

  intptr_t task_reg_pr = (intptr_t)sub_reg - NRF_SUBSCRIBE_PUBLISH_OFFSET((uintptr_t)sub_reg);

  nrf_hack_get_task_from_ptr((void *)task_reg_pr, &p_reg, &set_f, &clear_f, &trigger_f, &task);
  if (nrf_hack_is_task(task)) {
    clear_f(p_reg, task);
  }
}

#endif /* DPPI_PRESENT */

/*
 * Given a task register address for an unspecified peripheral,
 * trigger the corresponding task in the hardware model.
 */
void nrf_hack_trigger_task_address(void *task_reg)
{
  void *p_reg;
  int task;
  subscribe_set_f set_f;
  subscribe_clear_f clear_f;
  task_trigger_f trigger_f;

  nrf_hack_get_task_from_ptr(task_reg, &p_reg, &set_f, &clear_f, &trigger_f, &task);

  if (nrf_hack_is_task(task)) {
     trigger_f(p_reg, task);
  }
}