/*
 * Copyright (c) 2014-2015, TAKAHASHI Tomohiro (TTRFTECH) edy555@gmail.com
 * All rights reserved.
 *
 * This is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 3, or (at your option)
 * any later version.
 *
 * The software is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with GNU Radio; see the file COPYING.  If not, write to
 * the Free Software Foundation, Inc., 51 Franklin Street,
 * Boston, MA 02110-1301, USA.
 */
#include "ch.h"
#include "hal.h"
#include "nanovna.h"
#include <string.h>

static int flash_wait_for_last_operation(void)
{
  while (FLASH->SR == FLASH_SR_BSY) {
    //WWDG->CR = WWDG_CR_T;
  }
  return FLASH->SR;
}

static void flash_erase_page0(uint32_t page_address)
{
  flash_wait_for_last_operation();
  FLASH->CR |= FLASH_CR_PER;
  FLASH->AR = page_address;
  FLASH->CR |= FLASH_CR_STRT;
  flash_wait_for_last_operation();
  FLASH->CR &= ~FLASH_CR_PER;
}

int flash_erase_page(uint32_t page_address)
{
  chSysLock();
  flash_erase_page0(page_address);
  chSysUnlock();
  return 0;
}

void flash_program_half_word(uint32_t address, uint16_t data)
{
  flash_wait_for_last_operation();
  FLASH->CR |= FLASH_CR_PG;
  *(__IO uint16_t*)address = data;
  flash_wait_for_last_operation();
  FLASH->CR &= ~FLASH_CR_PG;
}

void flash_unlock(void)
{
  // unlock sequence
  FLASH->KEYR = 0x45670123;
  FLASH->KEYR = 0xCDEF89AB;
}

static uint32_t
checksum(const void *start, size_t len)
{
  uint32_t *p = (uint32_t*)start;
  uint32_t *tail = (uint32_t*)(start + len);
  uint32_t value = 0;
  while (p < tail)
    value = __ROR(value, 31) + *p++;
  return value;
}


#define FLASH_PAGESIZE 0x800

const uint32_t save_config_area = SAVE_CONFIG_ADDR;

int
config_save(void)
{
  uint16_t *src = (uint16_t*)&config;
  uint16_t *dst = (uint16_t*)save_config_area;
  int count = sizeof(config_t) / sizeof(uint16_t);

  config.magic = CONFIG_MAGIC;
  config.checksum = checksum(&config, sizeof config - sizeof config.checksum);

  flash_unlock();

  /* erase flash pages */
  flash_erase_page((uint32_t)dst);

  /* write to flash */
  while (count-- > 0) {
    flash_program_half_word((uint32_t)dst, *src++);
    dst++;
  }

  return 0;
}

int
config_recall(void)
{
  const config_t *src = (const config_t*)save_config_area;
  void *dst = &config;

  if (src->magic != CONFIG_MAGIC)
    return -1;
  if (checksum(src, sizeof *src - sizeof src->checksum) != src->checksum)
    return -1;

  /* duplicated saved data onto sram to be able to modify marker/trace */
  memcpy(dst, src, sizeof(config_t));
  return 0;
}

const uint32_t saveareas[SAVEAREA_MAX] =
{
  SAVE_PROP_CONFIG_0_ADDR,
  SAVE_PROP_CONFIG_1_ADDR,
  SAVE_PROP_CONFIG_2_ADDR,
  SAVE_PROP_CONFIG_3_ADDR,
  SAVE_PROP_CONFIG_4_ADDR,
  SAVE_PROP_CONFIG_5_ADDR,
  SAVE_PROP_CONFIG_6_ADDR,
  SAVE_PROP_CONFIG_7_ADDR,
  SAVE_PROP_CONFIG_8_ADDR,
};

int16_t lastsaveid = 0;

int
caldata_save(int id)
{
  uint16_t *src = (uint16_t*)&setting;
  uint16_t *dst;
  int count = sizeof(setting_t) / sizeof(uint16_t);

  if (id < 0 || id >= SAVEAREA_MAX)
    return -1;
  dst = (uint16_t*)saveareas[id];

  setting.magic = CONFIG_MAGIC;
  setting.checksum = checksum(
      &setting, sizeof setting - sizeof setting.checksum);

  flash_unlock();

  /* erase flash pages */
  void *p = dst;
  void *tail = p + sizeof(setting_t);
  while (p < tail) {
    flash_erase_page((uint32_t)p);
    p += FLASH_PAGESIZE;
  }

  /* write to flash */
  while (count-- > 0) {
    flash_program_half_word((uint32_t)dst, *src++);
    dst++;
  }

  // Flash stored trace
  count = sizeof(stored_t) /  sizeof(uint16_t);
  src = (uint16_t*)&stored_t[0];
  while (count-- > 0) {
    flash_program_half_word((uint32_t)dst, *src++);
    dst++;
  }

  /* after saving data, make active configuration points to flash */
//  active_props = (setting_t*)saveareas[id];
  lastsaveid = id;

  return 0;
}

int
caldata_recall(int id)
{
  setting_t *src;
  void *dst = &setting;

  if (id < 0 || id >= SAVEAREA_MAX)
    return -1;

  // point to saved area on the flash memory
  src = (setting_t*)saveareas[id];

  if (src->magic != CONFIG_MAGIC)
    return -1;
  if (checksum(src, sizeof setting - sizeof src->checksum) != src->checksum)
    return -1;

  /* active configuration points to save data on flash memory */
//  active_props = src;
  lastsaveid = id;

  /* duplicated saved data onto sram to be able to modify marker/trace */
  memcpy(dst, src, sizeof(setting_t));
  // Restore stored trace
  memcpy(stored_t, &src[1], sizeof(stored_t));

  update_frequencies();
  set_scale(setting.scale);
  set_reflevel(setting.reflevel);
  return 0;
}
#if 0
const setting_t *
caldata_ref(int id)
{
  const setting_t *src;
  if (id < 0 || id >= SAVEAREA_MAX)
    return NULL;
  src = (const setting_t*)saveareas[id];

  if (src->magic != CONFIG_MAGIC)
    return NULL;
  if (checksum(src, sizeof *src - sizeof src->checksum) != src->checksum)
    return NULL;
  return src;
}
#endif

const uint32_t save_config_prop_area_size = SAVE_CONFIG_AREA_SIZE;

void
clear_all_config_prop_data(void)
{
  flash_unlock();

  /* erase flash pages */
  void *p = (void*)save_config_area;
  void *tail = p + save_config_prop_area_size;
  while (p < tail) {
    flash_erase_page((uint32_t)p);
    p += FLASH_PAGESIZE;
  }
}