/*
 * Copyright (c) 2024 Nordic Semiconductor ASA
 *
 * SPDX-License-Identifier: Apache-2.0
 *
 * Note that the function prototypes are taken from the NRFx HAL
 */
#include "hal/nrf_aar.h"
#include "hal/nrf_ccm.h"
#include "bs_tracing.h"
#include "NHW_54_AAR_CCM_ECB.h"

static int aar_inst_from_ptr(NRF_AAR_Type * p_reg)
{
  union NRF_AARCCM_regs *p = (union NRF_AARCCM_regs *)p_reg;

  int i = ( (int)p - (int)NRF_AARCCM_regs ) / sizeof(union NRF_AARCCM_Type);
  return i;
}

void nrf_aar_int_enable(NRF_AAR_Type * p_reg, uint32_t mask)
{
  int inst = aar_inst_from_ptr(p_reg);
  p_reg->INTENSET = mask;
  nhw_AAR_regw_sideeffects_INTENSET(inst);
}

void nrf_aar_int_disable(NRF_AAR_Type * p_reg, uint32_t mask)
{
  int inst = aar_inst_from_ptr(p_reg);
  p_reg->INTENCLR = mask;
  nhw_AAR_regw_sideeffects_INTENCLR(inst);
}

void nrf_aar_task_trigger(NRF_AAR_Type * p_reg, nrf_aar_task_t task)
{
  int inst = aar_inst_from_ptr(p_reg);
  *(volatile uint32_t *)((uint8_t *)p_reg + (uint32_t)task) = 1;
  if (task == NRF_AAR_TASK_START) {
    nhw_AAR_regw_sideeffects_TASKS_START(inst);
  } else if (task == NRF_AAR_TASK_STOP) {
    nhw_AAR_regw_sideeffects_TASKS_STOP(inst);
  } else if ( (int)task == (int)NRF_CCM_TASK_RATEOVERRIDE ) {
    nhw_CCM_regw_sideeffects_TASKS_RATEOVERRIDE(inst);
  } else {
    bs_trace_error_line_time("Not supported task started in nrf_aar\n");
  }
}

void nrf_aar_enable(NRF_AAR_Type * p_reg)
{
  int inst = aar_inst_from_ptr(p_reg);
  p_reg->ENABLE = AAR_ENABLE_ENABLE_Enabled << AAR_ENABLE_ENABLE_Pos;
  nhw_AARCCM_regw_sideeffects_ENABLE(inst);
}

void nrf_aar_disable(NRF_AAR_Type * p_reg)
{
  int inst = aar_inst_from_ptr(p_reg);
  p_reg->ENABLE = AAR_ENABLE_ENABLE_Disabled << AAR_ENABLE_ENABLE_Pos;
  nhw_AARCCM_regw_sideeffects_ENABLE(inst);
}

void nrf_aar_event_clear(NRF_AAR_Type * p_reg, nrf_aar_event_t event)
{
  int inst = aar_inst_from_ptr(p_reg);
  *((volatile uint32_t *)((uint8_t *)p_reg + (uint32_t)event)) = 0x0UL;
  nhw_AAR_regw_sideeffects_EVENTS_all(inst);
}

static void nrf_aar_subscribe_common(NRF_AAR_Type * p_reg,
                                     nrf_aar_task_t task)
{
  int inst = aar_inst_from_ptr(p_reg);
  if (task == NRF_AAR_TASK_START) {
    nhw_AAR_regw_sideeffects_SUBSCRIBE_START(inst);
  } else if ( task == NRF_AAR_TASK_STOP ) {
    nhw_AAR_regw_sideeffects_SUBSCRIBE_STOP(inst);
  } else if ( (int)task == (int)NRF_CCM_TASK_RATEOVERRIDE ) {
    /* AAR and CCM share the SW register IF, and therefore we may land here also.
     * Though the START and STOP tasks are the same, RATEOVERRIDE only exists for the CCM */
    nhw_CCM_regw_sideeffects_SUBSCRIBE_RATEOVERRIDE(inst);
  } else {
    bs_trace_error_line_time("Attempted to subscribe to a not-supported task in the nrf_aar (%i)\n",
                             task);
  }
}

void nrf_aar_subscribe_set(NRF_AAR_Type * p_reg,
                           nrf_aar_task_t task,
                           uint8_t        channel)
{
  *((volatile uint32_t *) ((uint8_t *) p_reg + (uint32_t) task + 0x80uL)) =
          ((uint32_t)channel | NRF_SUBSCRIBE_PUBLISH_ENABLE);
  nrf_aar_subscribe_common(p_reg, task);
}

void nrf_aar_subscribe_clear(NRF_AAR_Type * p_reg,
                             nrf_aar_task_t task)
{
  *((volatile uint32_t *) ((uint8_t *) p_reg + (uint32_t) task + 0x80uL)) = 0;
  nrf_aar_subscribe_common(p_reg, task);
}