/*
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you 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
 *
 *  http://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.
 */
#include <assert.h>

#include "bsp/bsp.h"
#include "os/mynewt.h"
#include "mynewt_cm.h"
#include "spiflash/spiflash.h"

#include <hal/hal_bsp.h>
#include <hal/hal_flash_int.h>
#include <hal/hal_system.h>

#include <stm32g0xx.h>
#include <stm32_common/stm32_hal.h>
#if MYNEWT_PKG_apache_mynewt_core__bus_drivers
#include <bus/drivers/spi_common.h>
#endif

#if MYNEWT_PKG_apache_mynewt_core__hw_drivers_flash_fs_flash
#include <fs_flash/fs_flash.h>
#endif

#if MYNEWT_VAL(PWM_0) || MYNEWT_VAL(PWM_1) || MYNEWT_VAL(PWM_2)
#include <pwm_stm32/pwm_stm32.h>
#endif

#if MYNEWT_VAL(PWM_0)
struct stm32_pwm_conf os_bsp_pwm0_cfg = {
    .tim = TIM2,
    .irq = TIM2_IRQn,
};
#endif
#if MYNEWT_VAL(PWM_1)
struct stm32_pwm_conf os_bsp_pwm1_cfg = {
    .tim = TIM4,
    .irq = TIM3_TIM4_IRQn,
};
#endif
#if MYNEWT_VAL(PWM_2)
struct stm32_pwm_conf os_bsp_pwm2_cfg = {
    .tim = TIM1,
    .irq = TIM1_CC_IRQn,
};
#endif

#if MYNEWT_VAL(UART_0)
const struct stm32_uart_cfg os_bsp_uart0_cfg = {
    .suc_uart = USART1,
    .suc_rcc_reg = &RCC->APBENR2,
    .suc_rcc_dev = RCC_APBENR2_USART1EN,
    .suc_pin_tx = MYNEWT_VAL(UART_0_PIN_TX),
    .suc_pin_rx = MYNEWT_VAL(UART_0_PIN_RX),
    .suc_pin_rts = MYNEWT_VAL(UART_0_PIN_RTS),
    .suc_pin_cts = MYNEWT_VAL(UART_0_PIN_CTS),
    .suc_pin_af = GPIO_AF0_USART1,
    .suc_irqn = USART1_IRQn,
};
#endif

#if MYNEWT_VAL(UART_1)
const struct stm32_uart_cfg os_bsp_uart1_cfg = {
    .suc_uart = USART2,
    .suc_rcc_reg = &RCC->APBENR1,
    .suc_rcc_dev = RCC_APBENR1_USART2EN,
    .suc_pin_tx = MYNEWT_VAL(UART_1_PIN_TX),
    .suc_pin_rx = MYNEWT_VAL(UART_1_PIN_RX),
    .suc_pin_rts = MYNEWT_VAL(UART_1_PIN_RTS),
    .suc_pin_cts = MYNEWT_VAL(UART_1_PIN_CTS),
    .suc_pin_af = GPIO_AF1_USART2,
    .suc_irqn = USART2_LPUART2_IRQn,
};
#endif

#if MYNEWT_VAL(UART_2)
const struct stm32_uart_cfg os_bsp_uart1_cfg = {
    .suc_uart = USART3,
    .suc_rcc_reg = &RCC->APBENR1,
    .suc_rcc_dev = RCC_APBENR1_USART3EN,
    .suc_pin_tx = MYNEWT_VAL(UART_2_PIN_TX),
    .suc_pin_rx = MYNEWT_VAL(UART_2_PIN_RX),
    .suc_pin_rts = MYNEWT_VAL(UART_2_PIN_RTS),
    .suc_pin_cts = MYNEWT_VAL(UART_2_PIN_CTS),
    .suc_pin_af = GPIO_AF0_USART3,
    .suc_irqn = USART3_4_5_6_LPUART1_IRQn,
};
#endif

#if MYNEWT_VAL(UART_3)
const struct stm32_uart_cfg os_bsp_uart1_cfg = {
    .suc_uart = USART4,
    .suc_rcc_reg = &RCC->APBENR1,
    .suc_rcc_dev = RCC_APBENR1_USART4EN,
    .suc_pin_tx = MYNEWT_VAL(UART_4_PIN_TX),
    .suc_pin_rx = MYNEWT_VAL(UART_4_PIN_RX),
    .suc_pin_rts = MYNEWT_VAL(UART_4_PIN_RTS),
    .suc_pin_cts = MYNEWT_VAL(UART_4_PIN_CTS),
    .suc_pin_af = GPIO_AF0_USART4,
    .suc_irqn = USART3_4_5_6_LPUART1_IRQn,
};
#endif

#if MYNEWT_VAL(UART_4)
const struct stm32_uart_cfg os_bsp_uart1_cfg = {
    .suc_uart = USART5,
    .suc_rcc_reg = &RCC->APBENR1,
    .suc_rcc_dev = RCC_APBENR1_USART5EN,
    .suc_pin_tx = MYNEWT_VAL(UART_4_PIN_TX),
    .suc_pin_rx = MYNEWT_VAL(UART_4_PIN_RX),
    .suc_pin_rts = MYNEWT_VAL(UART_4_PIN_RTS),
    .suc_pin_cts = MYNEWT_VAL(UART_4_PIN_CTS),
    .suc_pin_af = GPIO_AF3_USART5,
    .suc_irqn = USART3_4_5_6_LPUART1_IRQn,
};
#endif

#if MYNEWT_VAL(UART_5)
const struct stm32_uart_cfg os_bsp_uart1_cfg = {
    .suc_uart = USART6,
    .suc_rcc_reg = &RCC->APBENR1,
    .suc_rcc_dev = RCC_APBENR1_USART6EN,
    .suc_pin_tx = MYNEWT_VAL(UART_5_PIN_TX),
    .suc_pin_rx = MYNEWT_VAL(UART_5_PIN_RX),
    .suc_pin_rts = MYNEWT_VAL(UART_5_PIN_RTS),
    .suc_pin_cts = MYNEWT_VAL(UART_5_PIN_CTS),
    .suc_pin_af = GPIO_AF3_USART6,
    .suc_irqn = USART3_4_5_6_LPUART1_IRQn,
};
#endif

#if MYNEWT_VAL(I2C_0)
/*
 * The PB8 and PB9 pins are connected through jumpers in the board to
 * both ADC_IN and I2C pins. To enable I2C functionality SB147/SB157 need
 * to be removed (they are the default connections) and SB138/SB143 need
 * to be shorted.
 */
const struct stm32_hal_i2c_cfg os_bsp_i2c0_cfg = {
    .hic_i2c = I2C1,
    .hic_rcc_reg = &RCC->APBENR1,
    .hic_rcc_dev = RCC_APBENR1_I2C1EN,
    .hic_pin_sda = MYNEWT_VAL(I2C_0_PIN_SDA),
    .hic_pin_scl = MYNEWT_VAL(I2C_0_PIN_SCL),
    .hic_pin_af = GPIO_AF6_I2C1,
    .hic_10bit = 0,
    .hic_timingr = 0x20404768,  /* 100kHz at 216 MHz system clock */
};
#endif

#if MYNEWT_VAL(I2C_1)
const struct stm32_hal_i2c_cfg os_bsp_i2c1_cfg = {
    .hic_i2c = I2C2,
    .hic_rcc_reg = &RCC->APBENR1,
    .hic_rcc_dev = RCC_APBENR1_I2C2EN,
    .hic_pin_sda = MYNEWT_VAL(I2C_1_PIN_SDA),
    .hic_pin_scl = MYNEWT_VAL(I2C_1_PIN_SCL),
    .hic_pin_af = GPIO_AF6_I2C2,
    .hic_10bit = 0,
    .hic_timingr = 0x20404768,  /* 100kHz at 216 MHz system clock */
};
#endif

#if MYNEWT_VAL(I2C_2)
const struct stm32_hal_i2c_cfg os_bsp_i2c2_cfg = {
    .hic_i2c = I2C3,
    .hic_rcc_reg = &RCC->APBENR1,
    .hic_rcc_dev = RCC_APBENR1_I2C3EN,
    .hic_pin_sda = MYNEWT_VAL(I2C_2_PIN_SDA),
    .hic_pin_scl = MYNEWT_VAL(I2C_2_PIN_SCL),
    .hic_pin_af = GPIO_AF6_I2C3,
    .hic_10bit = 0,
    .hic_timingr = 0x20404768,  /* 100kHz at 216 MHz system clock */
};
#endif

static const struct hal_bsp_mem_dump dump_cfg[] = {
    [0] = {
        .hbmd_start = &_ram_start,
        .hbmd_size = RAM_SIZE,
    },
};

extern const struct hal_flash stm32_flash_dev;

static const struct hal_flash *flash_devs[] = {
    [0] = &stm32_flash_dev,
#if MYNEWT_VAL(SPIFLASH)
    [1] = &spiflash_dev.hal,
#endif
#if MYNEWT_PKG_apache_mynewt_core__hw_drivers_flash_fs_flash
    [2] = &fs_flash_dev.hal,
#endif
};

const struct hal_flash *
hal_bsp_flash_dev(uint8_t id)
{
    if (id >= ARRAY_SIZE(flash_devs)) {
        return NULL;
    }

    return flash_devs[id];
}

const struct hal_bsp_mem_dump *
hal_bsp_core_dump(int *area_cnt)
{
    *area_cnt = sizeof(dump_cfg) / sizeof(dump_cfg[0]);
    return dump_cfg;
}

void
hal_bsp_init(void)
{
    stm32_periph_create();
}

void
hal_bsp_deinit(void)
{
    Cortex_DisableAll();

    RCC->AHBENR = 0x00000100;
    RCC->APBENR1 = 0x00000000;
    RCC->APBENR2 = 0x00000000;

    RCC->AHBRSTR = 0x00051103;
    RCC->APBRSTR1 = 0xFFFFFFB7;
    RCC->APBRSTR2 = 0x0017D801;

    RCC->AHBRSTR = 0;
    RCC->APBRSTR1 = 0;
    RCC->APBRSTR2 = 0;
}

/**
 * Returns the configured priority for the given interrupt. If no priority
 * configured, return the priority passed in
 *
 * @param irq_num
 * @param pri
 *
 * @return uint32_t
 */
uint32_t
hal_bsp_get_nvic_priority(int irq_num, uint32_t pri)
{
    /* Add any interrupt priorities configured by the bsp here */
    return pri;
}
