#include "user_spi.h"
#include "stm32f4xx_hal.h"
#include "main.h"
#include <stdint.h>
#include <string.h>
#include <stdint.h>
#include <string.h>

#include "spi.h"

// 移除对platform_support.h的依赖
// 直接在此处定义需要的常量和函数
#define SUCCESS 0
#define FAILURE -1

// 声明mdelay函数
static inline void mdelay(uint32_t ms)
{
    HAL_Delay(ms);
}

/******************************************************************************/
/***************************** #defines ***************************************/
/******************************************************************************/

#define SPI_BUFFER_SIZE 255


/******************************************************************************/
/************************ Variable Declarations *******************************/
/******************************************************************************/

static uint8_t spi_rx_buffer[SPI_BUFFER_SIZE] = {0};

/******************************************************************************/
/************************ Functions Definitions *******************************/
/******************************************************************************/


#define SPI_BUFFER_SIZE 255


// SPI handle declaration
SPI_HandleTypeDef hspi2;

// Define return values
#define SUCCESS 0
#define FAILURE -1

int32_t spi_transmit_receive(SPI_HandleTypeDef *hspi, uint8_t *data_write, uint8_t bytes_number)
{
    // 使用 HAL_SPI_TransmitReceive 函数进行 SPI 通信
    if (HAL_SPI_TransmitReceive(&hspi2, data_write, (uint8_t *)spi_rx_buffer, bytes_number, 1000) != HAL_OK)
    {
        // 如果通信失败，返回 FAILURE
        return FAILURE;
    }
    /* 将 SPI 接收缓冲区复制到提供的数据缓冲区，以便返回给调用者 */
    memcpy(data_write, spi_rx_buffer, bytes_number);
    // 返回 SUCCESS 表示通信成功
    return SUCCESS;
}

void ad7124_spi_init(void)
{
    GPIO_InitTypeDef GPIO_InitStruct = {0};
    
    // Enable GPIO and SPI clocks
    __HAL_RCC_SPI2_CLK_ENABLE();
    __HAL_RCC_GPIOC_CLK_ENABLE();  // For MISO and MOSI
    __HAL_RCC_GPIOB_CLK_ENABLE();  // For SCK
    __HAL_RCC_GPIOD_CLK_ENABLE();  // For CS
    
    // Configure SPI2 pins
    // MISO (PC2) and MOSI (PC3)
    GPIO_InitStruct.Pin = GPIO_PIN_2 | GPIO_PIN_3;
    GPIO_InitStruct.Mode = GPIO_MODE_AF_PP;
    GPIO_InitStruct.Pull = GPIO_NOPULL;
    GPIO_InitStruct.Speed = GPIO_SPEED_FREQ_VERY_HIGH;
    GPIO_InitStruct.Alternate = GPIO_AF5_SPI2;  // SPI2 alternate function
    HAL_GPIO_Init(GPIOC, &GPIO_InitStruct);
    
    // SCK (PB10)
    GPIO_InitStruct.Pin = GPIO_PIN_10;
    GPIO_InitStruct.Mode = GPIO_MODE_AF_PP;
    GPIO_InitStruct.Pull = GPIO_NOPULL;
    GPIO_InitStruct.Speed = GPIO_SPEED_FREQ_VERY_HIGH;
    GPIO_InitStruct.Alternate = GPIO_AF5_SPI2;  // SPI2 alternate function
    HAL_GPIO_Init(GPIOB, &GPIO_InitStruct);
    
    // CS (PD10) - Configure as output
    GPIO_InitStruct.Pin = GPIO_PIN_10;
    GPIO_InitStruct.Mode = GPIO_MODE_OUTPUT_PP;
    GPIO_InitStruct.Pull = GPIO_PULLUP;  // Add pull-up to ensure high when not driven
    GPIO_InitStruct.Speed = GPIO_SPEED_FREQ_VERY_HIGH;
    HAL_GPIO_Init(GPIOD, &GPIO_InitStruct);
    
    // Ensure CS is high before SPI initialization
    
    // Configure SPI2 with the working configuration
    hspi2.Instance = SPI2;
    hspi2.Init.Mode = SPI_MODE_MASTER;
    hspi2.Init.Direction = SPI_DIRECTION_2LINES;
    hspi2.Init.DataSize = SPI_DATASIZE_8BIT;
    hspi2.Init.CLKPolarity = SPI_POLARITY_HIGH;     // CPOL = 1
    hspi2.Init.CLKPhase = SPI_PHASE_2EDGE;          // CPHA = 1
    hspi2.Init.NSS = SPI_NSS_SOFT;
    hspi2.Init.BaudRatePrescaler = SPI_BAUDRATEPRESCALER_256;
    hspi2.Init.FirstBit = SPI_FIRSTBIT_MSB;
    hspi2.Init.TIMode = SPI_TIMODE_DISABLE;
    hspi2.Init.CRCCalculation = SPI_CRCCALCULATION_DISABLE;
    hspi2.Init.CRCPolynomial = 10;
    
    if (HAL_SPI_Init(&hspi2) != HAL_OK)
    {
        Error_Handler();
    }
    
    // Double check CS is high after initialization
    HAL_GPIO_WritePin(SPI2_CS_GPIO_Port, SPI2_CS_Pin, GPIO_PIN_SET);//关闭AD7124_CS1
    HAL_GPIO_WritePin(SPI2_CS2_GPIO_Port, SPI2_CS2_Pin, GPIO_PIN_SET);//关闭AD7124_CS2
    HAL_GPIO_WritePin(AD7124_SYNC_GPIO_Port, AD7124_SYNC_Pin, GPIO_PIN_SET);//关闭AD7124_SYNC
    HAL_Delay(1);
}

void dac161s997_spi_init(void)
{
  /* Enable SPI2 clock */
  __HAL_RCC_SPI2_CLK_ENABLE();
  
  hspi2.Instance = SPI2;
  hspi2.Init.Mode = SPI_MODE_MASTER;
  hspi2.Init.Direction = SPI_DIRECTION_2LINES;
  hspi2.Init.DataSize = SPI_DATASIZE_8BIT;
  hspi2.Init.CLKPolarity = SPI_POLARITY_LOW;
  hspi2.Init.CLKPhase = SPI_PHASE_1EDGE;
  hspi2.Init.NSS = SPI_NSS_SOFT;
  hspi2.Init.BaudRatePrescaler = SPI_BAUDRATEPRESCALER_32;
  hspi2.Init.FirstBit = SPI_FIRSTBIT_MSB;
  hspi2.Init.TIMode = SPI_TIMODE_DISABLE;
  hspi2.Init.CRCCalculation = SPI_CRCCALCULATION_DISABLE;
  hspi2.Init.CRCPolynomial = 10;
  if (HAL_SPI_Init(&hspi2) != HAL_OK)
  {
    Error_Handler();
  }
}

void ad7124_cs_on(void)
{
    HAL_GPIO_WritePin(SPI2_CS_GPIO_Port, SPI2_CS_Pin, GPIO_PIN_RESET);  // 拉低使能AD7124_CS1
    //HAL_GPIO_WritePin(SPI2_CS2_GPIO_Port, SPI2_CS2_Pin, GPIO_PIN_SET);  // 拉高关闭AD7124_CS2
    HAL_Delay(1);//延时1ms
}

void ad7124_cs_off(void)
{
    HAL_GPIO_WritePin(SPI2_CS_GPIO_Port, SPI2_CS_Pin, GPIO_PIN_SET);    // 拉高关闭AD7124_CS1
    //HAL_GPIO_WritePin(SPI2_CS2_GPIO_Port, SPI2_CS2_Pin, GPIO_PIN_RESET);  // 拉低使能AD7124_CS2
    HAL_Delay(1);
}

void dac161s997_cs_on(chip_type_e dac_num)
{
  switch (dac_num)
  {
  case DAC161S997_1:
    HAL_GPIO_WritePin(DAC1_CS_GPIO_Port, DAC1_CS_Pin, GPIO_PIN_RESET);
    HAL_GPIO_WritePin(DAC1_CS_GPIO_Port, DAC2_CS_Pin, GPIO_PIN_SET);
    HAL_GPIO_WritePin(ADC_CS_GPIO_Port, ADC_CS_Pin, GPIO_PIN_SET);
    break;
  case DAC161S997_2:
    HAL_GPIO_WritePin(DAC1_CS_GPIO_Port, DAC2_CS_Pin, GPIO_PIN_RESET);
    HAL_GPIO_WritePin(DAC1_CS_GPIO_Port, DAC1_CS_Pin, GPIO_PIN_SET);
    HAL_GPIO_WritePin(ADC_CS_GPIO_Port, ADC_CS_Pin, GPIO_PIN_SET);
    break;
  default:
    break;
  }
}
void dac161s997_cs_off(uint8_t dac_num)
{
  switch (dac_num)
  {
  case DAC161S997_1:
    HAL_GPIO_WritePin(DAC1_CS_GPIO_Port, DAC1_CS_Pin, GPIO_PIN_SET);
    HAL_GPIO_WritePin(DAC1_CS_GPIO_Port, DAC2_CS_Pin, GPIO_PIN_SET);
    HAL_GPIO_WritePin(ADC_CS_GPIO_Port, ADC_CS_Pin, GPIO_PIN_SET);
    break;
  case DAC161S997_2:
    HAL_GPIO_WritePin(DAC1_CS_GPIO_Port, DAC1_CS_Pin, GPIO_PIN_SET);
    HAL_GPIO_WritePin(DAC1_CS_GPIO_Port, DAC2_CS_Pin, GPIO_PIN_SET);
    HAL_GPIO_WritePin(ADC_CS_GPIO_Port, ADC_CS_Pin, GPIO_PIN_SET);
    break;
  default:
    break;
  }
}
void board_spi_init(chip_type_e chip_type)
{
    switch(chip_type) {
        case AD7124:
            ad7124_spi_init();
            break;
        case DAC161S997_1:
        case DAC161S997_2:
            dac161s997_spi_init();
            break;
        default:
            break;
    }
}

void board_spi_cs_on(chip_type_e chip_type)
{
    switch(chip_type) {
        case AD7124:
            HAL_GPIO_WritePin(SPI2_CS_GPIO_Port, SPI2_CS_Pin, GPIO_PIN_RESET);  // AD7124_CS1打开
            HAL_Delay(1);
            break;
        case DAC161S997_1:
            dac161s997_cs_on(DAC161S997_1);
            break;
        case DAC161S997_2:
            dac161s997_cs_on(DAC161S997_2);
            break;
        default:
            break;
    }
}

void board_spi_cs_off(chip_type_e chip_type)
{
    switch(chip_type) {
        case AD7124:
            HAL_GPIO_WritePin(SPI2_CS_GPIO_Port, SPI2_CS_Pin, GPIO_PIN_SET);    // AD7124_CS1关闭
            HAL_Delay(1);
            break;
        case DAC161S997_1:
            dac161s997_cs_off(DAC161S997_1);
            break;
        case DAC161S997_2:
            dac161s997_cs_off(DAC161S997_2);
            break;
        default:
            break;
    }
}

// Add helper function for AD7124 SPI communication
int32_t ad7124_spi_transmit_receive(uint8_t *data, uint8_t bytes)
{
    ad7124_cs_on();
    int32_t result = spi_transmit_receive(&hspi2, data, bytes);
    ad7124_cs_off();
    return result;
}