/*
    Copyright (C) 2023 Westberry Technology Corp., Ltd

    Licensed 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 "ch.h"
#include "hal.h"
#include "debug.h"
#include "chprintf.h"

#define PORTAB_LINE_LED1      PAL_LINE(GPIOB, 14U)
#define PORTAB_LINE_LED2      PAL_LINE(GPIOB, 13U)
#define PORTAB_LED_OFF        PAL_HIGH
#define PORTAB_LED_ON         AL_LOW
#define RTC_ALARMPERIOD       10
#define TEST_ALARM_WAKEUP     TRUE

RTCDateTime timespec;
RTCAlarm alarmspec;
 
extern void __early_init(void);

#if TEST_ALARM_WAKEUP

static const uint32_t pre_lp_code[] = {553863175u,554459777u,1208378049u,4026624001u,688390415u,554227969u,3204472833u,1198571264u,1073807360u,1073808388u};
#define PRE_LP()  ((void(*)(void))((unsigned int)(pre_lp_code) | 0x01))()

static const uint32_t post_lp_code[] = {553863177u,554459777u,1208509121u,51443856u,4026550535u,1745485839u,3489677954u,536895496u,673389632u,1198578684u,1073807360u,536866816u,1073808388u};
#define POST_LP()  ((void(*)(void))((unsigned int)(post_lp_code) | 0x01))()
  
static void stop_mode_entry(void) {

  EXTI->PR = 0x7FFFF;
  for (uint8_t i = 0; i < 8; i++) {
    for (uint8_t j = 0; j < 32; j++) {
      if (NVIC->ISPR[i] & (0x01UL < j)) {
        NVIC->ICPR[i] = (0x01UL < j);
      }
    }
  }
  SCB->ICSR |= SCB_ICSR_PENDSTCLR_Msk; // Clear Systick IRQ Pending

  /* Clear all bits except DBP and FCLKSD bit */
  PWR->CR0 &= 0x09U;

  /* STOP LP4 MODE S32KON */
  PWR->CR0 |= 0x3B004U;
  PWR->CFGR = 0x3B3;

  PRE_LP();

  /* Set SLEEPDEEP bit of Cortex System Control Register */
  SCB->SCR |= SCB_SCR_SLEEPDEEP_Msk;

  /* Request Wait For Interrupt */
  __WFI();

  POST_LP();

  /* Clear SLEEPDEEP bit of Cortex System Control Register */
  SCB->SCR &= (~SCB_SCR_SLEEPDEEP_Msk);
}

/*
 * Running indicator thread.
 */
static THD_WORKING_AREA(blinkWA, 128);
static THD_FUNCTION(blink_thd, arg) {
  (void)arg;
  while (true) {
    chThdSleepMilliseconds(100);
    palToggleLine(PORTAB_LINE_LED2);
  }
}

static void my_cb(RTCDriver *rtcp, rtcevent_t event) {

  (void)rtcp;

  switch (event) {
    case RTC_EVENT_OVERFLOW: break;
    case RTC_EVENT_SECOND: {
      palToggleLine(PORTAB_LINE_LED1);
    } break;
    case RTC_EVENT_ALARM: {
      osalSysLockFromISR();
      NVIC_DisableIRQ(RTCAlarm_IRQn);
      osalSysUnlockFromISR();
    } break;
  }
}

int main(void) {

  uint32_t tv_sec;
  
  halInit();
  chSysInit();
  
  serial_debug_init();
 
  /* 
   * Init LED port and pad.
   */
  palSetPadMode(PAL_PORT(PORTAB_LINE_LED1), PAL_PAD(PORTAB_LINE_LED1), PAL_WB32_MODE_OUTPUT | PAL_WB32_OTYPE_PUSHPULL);
  palSetPadMode(PAL_PORT(PORTAB_LINE_LED2), PAL_PAD(PORTAB_LINE_LED2), PAL_WB32_MODE_OUTPUT | PAL_WB32_OTYPE_PUSHPULL);
  
  chThdCreateStatic(blinkWA, sizeof(blinkWA), NORMALPRIO, blink_thd, NULL);

  while (true){
    chThdSleepSeconds(2);
    rtcGetTime(&RTCD1, &timespec);
    chThdSleepSeconds(3);

    chSysDisable();
    wb32_set_main_clock_to_mhsi();

    rtclp_lld_init();
    rtcSetCallback(&RTCD1, my_cb);
    rtcWB32GetSecMsec(&RTCD1, &tv_sec, NULL);
    alarmspec.tv_sec = tv_sec + RTC_ALARMPERIOD;
    rtcSetAlarm(&RTCD1, 0, &alarmspec);
    NVIC_EnableIRQ(RTCAlarm_IRQn);
    
    stop_mode_entry();
 
    /* Recovery clock */
    __early_init();
    rtc_lld_init();
    rccEnableEXTI();
    rccEnableUART1();
    rtcSetCallback(&RTCD1, my_cb);

    chSysEnable();
  }
}

#else /* TEST_ALARM_WAKEUP */

binary_semaphore_t alarm_sem;

/*
 * Alarm callback.
 */
static void my_cb(RTCDriver *rtcp, rtcevent_t event) {

  (void)rtcp;

  switch (event) {
    case RTC_EVENT_OVERFLOW: break;
    case RTC_EVENT_SECOND: {
      palToggleLine(PORTAB_LINE_LED2);
    } break;
    case RTC_EVENT_ALARM: {
      palToggleLine(PORTAB_LINE_LED1);
      osalSysLockFromISR();
      chBSemSignalI(&alarm_sem);
      osalSysUnlockFromISR();
    } break;
  }
}

static time_measurement_t sett, gett;

int main(void) {

  msg_t status = MSG_TIMEOUT;
  uint32_t tv_sec;

  halInit();
  chSysInit();

  /* 
  * Init LED port and pad.
  */
  palSetPadMode(PAL_PORT(PORTAB_LINE_LED1), PAL_PAD(PORTAB_LINE_LED1), PAL_WB32_MODE_OUTPUT | PAL_WB32_OTYPE_PUSHPULL);
  palSetPadMode(PAL_PORT(PORTAB_LINE_LED2), PAL_PAD(PORTAB_LINE_LED2), PAL_WB32_MODE_OUTPUT | PAL_WB32_OTYPE_PUSHPULL);

  chBSemObjectInit(&alarm_sem, TRUE);
  chTMObjectInit(&sett);
  chTMObjectInit(&gett);

  /* compile ability test */
  chTMStartMeasurementX(&gett);
  rtcGetTime(&RTCD1, &timespec);
  chTMStopMeasurementX(&gett);

  rtcWB32SetSec(&RTCD1, 1414845464);
  osalThreadSleepMilliseconds(10);
  rtcGetTime(&RTCD1, &timespec);
  timespec.month -= 1;

  chTMStartMeasurementX(&sett);
  rtcSetTime(&RTCD1, &timespec);
  chTMStopMeasurementX(&sett);
  osalThreadSleepMilliseconds(10);

  rtcGetTime(&RTCD1, &timespec);

  rtcWB32GetSecMsec(&RTCD1, &tv_sec, NULL);
  alarmspec.tv_sec = tv_sec + RTC_ALARMPERIOD;
  rtcSetAlarm(&RTCD1, 0, &alarmspec);

  rtcSetCallback(&RTCD1, my_cb);

  while (true){
    /* Wait until alarm callback signaled semaphore.*/
    status = chBSemWaitTimeout(&alarm_sem, TIME_S2I(RTC_ALARMPERIOD + 5));

    if (status == MSG_TIMEOUT){
      osalSysHalt("time is out");
    } else {
      rtcWB32GetSecMsec(&RTCD1, &tv_sec, NULL);
      alarmspec.tv_sec = tv_sec + RTC_ALARMPERIOD;
      rtcSetAlarm(&RTCD1, 0, &alarmspec);
    }
  }
}
#endif /* TEST_ALARM_WAKEUP */