/* USER CODE BEGIN Header */
/**
  ******************************************************************************
  * @file           : main.c
  * @brief          : Main program body
  ******************************************************************************
  * @attention
  *
  * Copyright (c) 2025 STMicroelectronics.
  * All rights reserved.
  *
  * This software is licensed under terms that can be found in the LICENSE file
  * in the root directory of this software component.
  * If no LICENSE file comes with this software, it is provided AS-IS.
  *
  ******************************************************************************
  */
/* USER CODE END Header */
/* Includes ------------------------------------------------------------------*/
#include "main.h"
#include "sensor_registers.h"

/* Private includes ----------------------------------------------------------*/
/* USER CODE BEGIN Includes */

/* USER CODE END Includes */

/* Private typedef -----------------------------------------------------------*/
/* USER CODE BEGIN PTD */

/* USER CODE END PTD */

/* Private define ------------------------------------------------------------*/
/* USER CODE BEGIN PD */

/* USER CODE END PD */

/* Private macro -------------------------------------------------------------*/
/* USER CODE BEGIN PM */

/* USER CODE END PM */

/* Private variables ---------------------------------------------------------*/
I2C_HandleTypeDef hi2c1;

/* USER CODE BEGIN PV */

/* USER CODE END PV */

/* Private function prototypes -----------------------------------------------*/
void SystemClock_Config(void);

static void MX_GPIO_Init(void);

static void MX_I2C1_Init(void);

static void MX_NVIC_Init(void);

/* USER CODE BEGIN PFP */

/* USER CODE END PFP */

/* Private user code ---------------------------------------------------------*/
/* USER CODE BEGIN 0 */

#define LINE_SIZE 32
#define BLK_SIZE 128
#define SENSOR_SIZE 1024

volatile ArrayPart array_part = top_mat;
volatile StepArrayRetrieve block_step = top_block_0;
uint16_t tmp_blk0[BLK_SIZE + 1] = { 0 };
uint16_t tmp_blk1[BLK_SIZE + 1] = { 0 };
uint16_t complete_array[SENSOR_SIZE] = { 0 };

void configure_sensor() {
    uint8_t data = 1;

    HAL_I2C_Mem_Write(&hi2c1, sensor_addr, config_reg, 1, &data, 1, 6);
    HAL_Delay(6);

    data = 0x0C;
    HAL_I2C_Mem_Write(&hi2c1, sensor_addr, mbit_trim_reg, 1, &data, 1, 5);
    HAL_Delay(6);

    data = 0x0C;
    HAL_I2C_Mem_Write(&hi2c1, sensor_addr, bias_trim_top_reg, 1, &data, 1, 5);
    HAL_Delay(6);

    data = 0x0C;
    HAL_I2C_Mem_Write(&hi2c1, sensor_addr, bias_trim_bot_reg, 1, &data, 1, 5);
    HAL_Delay(6);

    data = 0x14;
    HAL_I2C_Mem_Write(&hi2c1, sensor_addr, clk_trim_reg, 1, &data, 1, 5);
    HAL_Delay(6);

    data = 0x0C;
    HAL_I2C_Mem_Write(&hi2c1, sensor_addr, bpa_trim_top_reg, 1, &data, 1, 5);
    HAL_Delay(6);

    data = 0x0C;
    HAL_I2C_Mem_Write(&hi2c1, sensor_addr, bpa_trim_bot_reg, 1, &data, 1, 5);
    HAL_Delay(6);

    data = 0x88;
    HAL_I2C_Mem_Write(&hi2c1, sensor_addr, pu_reg, 1, &data, 1, 5);
    HAL_Delay(6);
}

int8_t blk_offset_start(const StepArrayRetrieve blk_step) {
    switch (blk_step) {
        case top_block_0:
            return 0;
        case top_block_1:
            return 1;
        case top_block_2:
            return 2;
        case top_block_3:
            return 3;
        case bot_block_0:
            return 31;
        case bot_block_1:
            return 30;
        case bot_block_2:
            return 29;
        case bot_block_3:
            return 28;
        default:
            return 0;
    }
}

void copy_line(const uint16_t *restrict const blk, const int8_t j, const int8_t count) {
    for (int8_t i = 0; i < LINE_SIZE; ++i) {
        complete_array[j * LINE_SIZE + i] = blk[count * LINE_SIZE + i];
    }
}

void transfer_blk_to_array(const uint16_t *restrict const blk,
                           const ArrayPart top_bot,
                           const StepArrayRetrieve blk_step
) {
    const int8_t offset = blk_offset_start(blk_step);
    switch (top_bot) {
        case top_mat:
            for (int8_t j = offset, count = 0; count < 4; j += 4, ++count) {
                copy_line(blk, j, count);
            }
            break;
        case bot_mat:
            for (int8_t j = offset, count = 0; count < 4; j -= 4, ++count) {
                copy_line(blk, j, count);
            }
            break;
    }
}

void HAL_I2C_MasterRxCpltCallback(I2C_HandleTypeDef *hi2c) {
    switch (block_step) {
        case top_block_0:
            block_step = top_block_1;
            HAL_I2C_Mem_Read_IT(&hi2c1,
                sensor_addr,
                cam_blk_1 | data_top_reg | start_get_mat,
                1,
                (uint8_t *) tmp_blk1,
                sizeof(tmp_blk1) * sizeof(uint16_t));
            transfer_blk_to_array(tmp_blk0, top_mat, top_block_0);
            break;
        case top_block_1:
            block_step = top_block_2;
            HAL_I2C_Mem_Read_IT(&hi2c1,
                sensor_addr,
                cam_blk_2 | data_top_reg | start_get_mat,
                1,
                (uint8_t *) tmp_blk0,
                sizeof(tmp_blk0) * sizeof(uint16_t));
            transfer_blk_to_array(tmp_blk1, top_mat, top_block_1);
            break;
        case top_block_2:
            block_step = top_block_3;
            HAL_I2C_Mem_Read_IT(&hi2c1,
                sensor_addr,
                cam_blk_3 | data_top_reg | start_get_mat,
                1,
                (uint8_t *) tmp_blk1,
                sizeof(tmp_blk1) * sizeof(uint16_t));
            transfer_blk_to_array(tmp_blk0, top_mat, top_block_2);
            break;
        case top_block_3:
            block_step = bot_block_0;
            HAL_I2C_Mem_Read_IT(&hi2c1,
                sensor_addr,
                cam_blk_0 | data_bot_reg | start_get_mat,
                1,
                (uint8_t *) tmp_blk0,
                sizeof(tmp_blk0) * sizeof(uint16_t));
            transfer_blk_to_array(tmp_blk1, top_mat, top_block_3);
            break;
        case bot_block_0:
            block_step = bot_block_1;
            HAL_I2C_Mem_Read_IT(&hi2c1,
                sensor_addr,
                cam_blk_1 | data_top_reg | start_get_mat,
                1,
                (uint8_t *) tmp_blk1,
                sizeof(tmp_blk1) * sizeof(uint16_t));
            transfer_blk_to_array(tmp_blk0, bot_mat, bot_block_0);
            break;
        case bot_block_1:
            block_step = bot_block_2;
            HAL_I2C_Mem_Read_IT(&hi2c1,
                sensor_addr,
                cam_blk_2 | data_top_reg | start_get_mat,
                1,
                (uint8_t *) tmp_blk0,
                sizeof(tmp_blk0) * sizeof(uint16_t));
            transfer_blk_to_array(tmp_blk1, bot_mat, bot_block_1);
            break;
        case bot_block_2:
            block_step = bot_block_3;
            HAL_I2C_Mem_Read_IT(&hi2c1,
                sensor_addr,
                cam_blk_3 | data_top_reg | start_get_mat,
                1,
                (uint8_t *) tmp_blk1,
                sizeof(tmp_blk1) * sizeof(uint16_t));
            transfer_blk_to_array(tmp_blk0, top_mat, top_block_2);
            break;
        case bot_block_3:
            block_step = stop;
            transfer_blk_to_array(tmp_blk1, bot_mat, bot_block_3);
            break;
        default:
            block_step = stop;
            break;
    }
}

/* USER CODE END 0 */

/**
  * @brief  The application entry point.
  * @retval int
  */
int main(void) {
    /* USER CODE BEGIN 1 */

    /* USER CODE END 1 */

    /* MCU Configuration--------------------------------------------------------*/

    /* Reset of all peripherals, Initializes the Flash interface and the Systick. */
    HAL_Init();

    /* USER CODE BEGIN Init */

    /* USER CODE END Init */

    /* Configure the system clock */
    SystemClock_Config();

    /* USER CODE BEGIN SysInit */

    /* USER CODE END SysInit */

    /* Initialize all configured peripherals */
    MX_GPIO_Init();
    MX_I2C1_Init();
    MX_NVIC_Init();
    /* USER CODE BEGIN 2 */
    // TODO: configs
    if (HAL_I2C_IsDeviceReady(&hi2c1, sensor_addr, 3, 10) != HAL_OK) { Error_Handler(); }
    configure_sensor(); {
        uint8_t received_data = 0;
        HAL_I2C_Mem_Write(&hi2c1, sensor_addr, status_reg, 1, &received_data, 1, 5);
    }

    // Getting first top half array
    HAL_I2C_Mem_Read(&hi2c1,
        sensor_addr,
        data_top_reg,
        1,
        (uint8_t *) &tmp_blk0,
        sizeof(tmp_blk0) * sizeof(uint16_t),
        5);
    /* USER CODE END 2 */

    /* Infinite loop */
    /* USER CODE BEGIN WHILE */
    while (1) {
        /* USER CODE END WHILE */

        /* USER CODE BEGIN 3 */
    }
    /* USER CODE END 3 */
}

/**
  * @brief System Clock Configuration
  * @retval None
  */
void SystemClock_Config(void) {
    RCC_OscInitTypeDef RCC_OscInitStruct = { 0 };
    RCC_ClkInitTypeDef RCC_ClkInitStruct = { 0 };

    /** Configure the main internal regulator output voltage
    */
    if (HAL_PWREx_ControlVoltageScaling(PWR_REGULATOR_VOLTAGE_SCALE1) != HAL_OK) { Error_Handler(); }

    /** Initializes the RCC Oscillators according to the specified parameters
    * in the RCC_OscInitTypeDef structure.
    */
    RCC_OscInitStruct.OscillatorType = RCC_OSCILLATORTYPE_MSI;
    RCC_OscInitStruct.MSIState = RCC_MSI_ON;
    RCC_OscInitStruct.MSICalibrationValue = 0;
    RCC_OscInitStruct.MSIClockRange = RCC_MSIRANGE_6;
    RCC_OscInitStruct.PLL.PLLState = RCC_PLL_NONE;
    if (HAL_RCC_OscConfig(&RCC_OscInitStruct) != HAL_OK) { Error_Handler(); }

    /** Initializes the CPU, AHB and APB buses clocks
    */
    RCC_ClkInitStruct.ClockType = RCC_CLOCKTYPE_HCLK | RCC_CLOCKTYPE_SYSCLK
                                  | RCC_CLOCKTYPE_PCLK1 | RCC_CLOCKTYPE_PCLK2;
    RCC_ClkInitStruct.SYSCLKSource = RCC_SYSCLKSOURCE_MSI;
    RCC_ClkInitStruct.AHBCLKDivider = RCC_SYSCLK_DIV1;
    RCC_ClkInitStruct.APB1CLKDivider = RCC_HCLK_DIV1;
    RCC_ClkInitStruct.APB2CLKDivider = RCC_HCLK_DIV1;

    if (HAL_RCC_ClockConfig(&RCC_ClkInitStruct, FLASH_LATENCY_0) != HAL_OK) { Error_Handler(); }
}

/**
  * @brief NVIC Configuration.
  */
static void MX_NVIC_Init(void) {
    /* I2C1_EV_IRQn interrupt configuration */
    HAL_NVIC_SetPriority(I2C1_EV_IRQn, 0, 0);
    HAL_NVIC_EnableIRQ(I2C1_EV_IRQn);
}

/**
  * @brief I2C1 Initialization Function
  */
static void MX_I2C1_Init(void) {
    /* USER CODE BEGIN I2C1_Init 0 */

    /* USER CODE END I2C1_Init 0 */

    /* USER CODE BEGIN I2C1_Init 1 */

    /* USER CODE END I2C1_Init 1 */
    hi2c1.Instance = I2C1;
    hi2c1.Init.Timing = 0x00000003;
    hi2c1.Init.OwnAddress1 = 0;
    hi2c1.Init.AddressingMode = I2C_ADDRESSINGMODE_7BIT;
    hi2c1.Init.DualAddressMode = I2C_DUALADDRESS_DISABLE;
    hi2c1.Init.OwnAddress2 = 0;
    hi2c1.Init.OwnAddress2Masks = I2C_OA2_NOMASK;
    hi2c1.Init.GeneralCallMode = I2C_GENERALCALL_DISABLE;
    hi2c1.Init.NoStretchMode = I2C_NOSTRETCH_DISABLE;
    if (HAL_I2C_Init(&hi2c1) != HAL_OK) { Error_Handler(); }

    /** Configure Analogue filter
    */
    if (HAL_I2CEx_ConfigAnalogFilter(&hi2c1, I2C_ANALOGFILTER_ENABLE) != HAL_OK) { Error_Handler(); }

    /** Configure Digital filter
    */
    if (HAL_I2CEx_ConfigDigitalFilter(&hi2c1, 0) != HAL_OK) { Error_Handler(); }
    /* USER CODE BEGIN I2C1_Init 2 */

    /* USER CODE END I2C1_Init 2 */
}

/**
  * @brief GPIO Initialization Function
  */
static void MX_GPIO_Init(void) {
    /* USER CODE BEGIN MX_GPIO_Init_1 */
    /* USER CODE END MX_GPIO_Init_1 */

    /* GPIO Ports Clock Enable */
    __HAL_RCC_GPIOB_CLK_ENABLE();

    /* USER CODE BEGIN MX_GPIO_Init_2 */
    /* USER CODE END MX_GPIO_Init_2 */
}

/* USER CODE BEGIN 4 */

/* USER CODE END 4 */

/**
  * @brief  This function is executed in case of error occurrence.
  */
void Error_Handler(void) {
    /* USER CODE BEGIN Error_Handler_Debug */
    /* User can add his own implementation to report the HAL error return state */
    __disable_irq();
    while (1) {}
    /* USER CODE END Error_Handler_Debug */
}

#ifdef  USE_FULL_ASSERT
/**
  * @brief  Reports the name of the source file and the source line number
  *         where the assert_param error has occurred.
  * @param  file: pointer to the source file name
  * @param  line: assert_param error line source number
  * @retval None
  */
void assert_failed(uint8_t *file, uint32_t line)
{
  /* USER CODE BEGIN 6 */
  /* User can add his own implementation to report the file name and line number,
     ex: printf("Wrong parameters value: file %s on line %d\r\n", file, line) */
  /* USER CODE END 6 */
}
#endif /* USE_FULL_ASSERT */