/*
 * Copyright (C) OpenTX
 *
 * Based on code named
 *   th9x - http://code.google.com/p/th9x
 *   er9x - http://code.google.com/p/er9x
 *   gruvin9x - http://code.google.com/p/gruvin9x
 *
 * License GPLv2: http://www.gnu.org/licenses/gpl-2.0.html
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License version 2 as
 * published by the Free Software Foundation.
 *
 * This program 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.
 */

#include "opentx.h"
#include "conversions.h"

#if defined(SDCARD)
#include "storage/modelslist.h"
#include "storage/sdcard_common.h"

#if defined(COLORLCD)
static void drawProgressScreen(const char* filename, int progress, int total)
{
  OpenTxTheme* l_theme = static_cast<OpenTxTheme*>(theme);

  lcd->reset();
  l_theme->drawBackground(lcd);
  lcd->drawText(LCD_W/2, LCD_H/2 - 30, STR_CONVERTING, FONT(XL) | CENTERED | ALARM_COLOR);
  lcd->drawText(LCD_W/2, LCD_H/2, filename, FONT(STD) | CENTERED | DEFAULT_COLOR);

  l_theme->drawProgressBar(lcd,
                           LCD_W / 4,
                           LCD_H / 2 + 40,
                           LCD_W / 2,
                           20,
                           progress, total);

  WDG_RESET();
  lcdRefresh();
}
#endif

void convertRadioData(int version)
{
  TRACE("convertRadioData(%d)", version);

#if defined(COLORLCD)
  // the theme has not been loaded before
  static_cast<OpenTxTheme*>(theme)->load();
#endif

  // Init backlight mode before entering alert screens
  requiredBacklightBright = BACKLIGHT_FORCED_ON;

#if defined(PCBHORUS)
  g_eeGeneral.blOffBright = 20;
#endif

  RAISE_ALERT(STR_STORAGE_WARNING, STR_SDCARD_CONVERSION_REQUIRE, NULL,
              AU_NONE);

  unsigned converted = 0;
  auto to_convert = modelslist.getModelsCount() + 1;

#if defined(COLORLCD)
  drawProgressScreen(RADIO_FILENAME, converted, to_convert);
#endif

  TRACE("converting '%s' (%d/%d)", RADIO_FILENAME, converted, to_convert);

#if STORAGE_CONVERSIONS < 220
  if (version == 219) {
    convertRadioData_219_to_220(g_eeGeneral);
  }
#endif
  converted++;

#if defined(SIMU)
  RTOS_WAIT_MS(200);
#endif
  
  const char* error = nullptr;
  for (auto category_ptr : modelslist.getCategories()) {
    for (auto model_ptr : *category_ptr) {

      uint8_t model_version;
      const char* filename = model_ptr->modelFilename;

      TRACE("converting '%s' (%d/%d)", filename, converted, to_convert);
#if defined(COLORLCD)
      drawProgressScreen(filename, converted, to_convert);
#endif

      error = readModel(filename, (uint8_t *)&g_model, sizeof(g_model), &model_version);
      if (!error) {

        convertModelData(model_version);

        char path[256];
        getModelPath(path, filename);
        error = writeFile(path, (uint8_t *)&g_model, sizeof(g_model));
        //TODO: what should be done with this error?
      }

      converted++;

#if defined(SIMU)
  RTOS_WAIT_MS(200);
#endif
    }
  }

  // reload models list
  modelslist.clear();
  modelslist.load();
}
#endif

void convertModelData(int version)
{
  TRACE("convertModelData(%d)", version);

#if STORAGE_CONVERSIONS < 220
  if (version == 219) {
    version = 219;
    convertModelData_219_to_220(g_model);
  }
#endif
}

#if defined(EEPROM)
void eeConvertModel(int id, int version)
{
  eeLoadModelData(id);
  convertModelData(version);
  uint8_t currModel = g_eeGeneral.currModel;
  g_eeGeneral.currModel = id;
  storageDirty(EE_MODEL);
  storageCheck(true);
  g_eeGeneral.currModel = currModel;
}

bool eeConvert()
{
  const char *msg = NULL;

  switch (g_eeGeneral.version) {
    case 219:
      msg = "EEprom Data v219";
      break;
    default:
      return false;
  }

  int conversionVersionStart = g_eeGeneral.version;

  // Information to the user and wait for key press
  g_eeGeneral.backlightMode = e_backlight_mode_on;
  g_eeGeneral.backlightBright = 0;
  g_eeGeneral.contrast = 25;

  ALERT(STR_STORAGE_WARNING, msg, AU_BAD_RADIODATA);

  RAISE_ALERT(STR_STORAGE_WARNING, STR_EEPROM_CONVERTING, NULL, AU_NONE);

  // General Settings conversion
  eeLoadGeneralSettingsData();
  int version = conversionVersionStart;

#if STORAGE_CONVERSIONS < 220
  if (version == 219) {
    version = 220;
    convertRadioData_219_to_220(g_eeGeneral);
  }
#endif

  storageDirty(EE_GENERAL);
  storageCheck(true);

#if defined(COLORLCD)
#elif LCD_W >= 212
  lcdDrawRect(60, 6*FH+4, 132, 3);
#else
  lcdDrawRect(10, 6*FH+4, 102, 3);
#endif

  // Models conversion
  for (uint8_t id=0; id<MAX_MODELS; id++) {
#if defined(COLORLCD)
#elif LCD_W >= 212
    lcdDrawSolidHorizontalLine(61, 6*FH+5, 10+id*2, FORCE);
#else
    lcdDrawSolidHorizontalLine(11, 6*FH+5, 10+(id*3)/2, FORCE);
#endif
    lcdRefresh();
    if (eeModelExists(id)) {
      eeConvertModel(id, conversionVersionStart);
    }
  }

  return true;
}
#endif