#include "user_spi.h"

/* SPI2 Handle */
SPI_HandleTypeDef hspi2;

#define SPI_BUFFER_SIZE 255
static uint8_t spi_rx_buffer[SPI_BUFFER_SIZE] = {0};
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)
  {
    // 如果通信失败，返回 FAIL
    return FAIL;
  }
  /* 将 SPI 接收缓冲区复制到提供的数据缓冲区，以便返回给调用者 */
  memcpy(data_write, spi_rx_buffer, bytes_number);
  // 返回 TRUE 表示通信成功
  return TRUE;
}

void ad7124_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_HIGH;    // CPOL = 1
    hspi2.Init.CLKPhase = SPI_PHASE_2EDGE;         // CPHA = 1
    hspi2.Init.NSS = SPI_NSS_SOFT;
    hspi2.Init.BaudRatePrescaler = SPI_BAUDRATEPRESCALER_64;  // 降低速度以确保稳定性
    hspi2.Init.FirstBit = SPI_FIRSTBIT_MSB;
    hspi2.Init.TIMode = SPI_TIMODE_DISABLE;
    hspi2.Init.CRCCalculation = SPI_CRCCALCULATION_DISABLE;
    hspi2.Init.CRCPolynomial = 10;
    
    // 确保CS初始为高电平
    HAL_GPIO_WritePin(SPI2_CS_GPIO_Port, SPI2_CS_Pin, GPIO_PIN_SET);
    HAL_GPIO_WritePin(SPI2_CS2_GPIO_Port, SPI2_CS2_Pin, GPIO_PIN_SET);
    
    if (HAL_SPI_Init(&hspi2) != HAL_OK)
    {
        Error_Handler();
    }
}

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)
{
  // 激活AD7124 (CS低电平有效)
  HAL_GPIO_WritePin(SPI2_CS_GPIO_Port, SPI2_CS_Pin, GPIO_PIN_RESET);
  
}

void ad7124_cs_off(void)
{
  // 禁用AD7124 (CS高电平)
  HAL_GPIO_WritePin(SPI2_CS_GPIO_Port, SPI2_CS_Pin, GPIO_PIN_SET);
}
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:
    dac161s997_spi_init();
    break;
  case DAC161S997_2:
    dac161s997_spi_init();
    break;
  default:
    dac161s997_spi_init();
    break;
  }
}

void board_spi_cs_on(chip_type_e chip_type)
{
  switch (chip_type)
  {
  case AD7124:
    ad7124_cs_on();
    break;
  case DAC161S997_1:
    dac161s997_cs_on(DAC161S997_1);
    break;
  case DAC161S997_2:
    dac161s997_cs_on(DAC161S997_2);
    break;
  default:
    dac161s997_cs_on(DAC161S997_2);
    break;
  }
}

void board_spi_cs_off(chip_type_e chip_type)
{
  switch (chip_type)
  {
  case AD7124:
    ad7124_cs_off();
    break;
  case DAC161S997_1:
    dac161s997_cs_off(DAC161S997_1);
    break;
  case DAC161S997_2:
    dac161s997_cs_off(DAC161S997_2);
    break;
  default:
    dac161s997_cs_off(DAC161S997_2);
    break;
  }
}