/**************************************************************************
 *   Copyright (C) 2013 by David S. Register                               *
 *                                                                         *
 *   This program 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 2 of the License, or     *
 *   (at your option) any later version.                                   *
 *                                                                         *
 *   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.                          *
 *                                                                         *
 *   You should have received a copy of the GNU General Public License     *
 *   along with this program; if not, see <https://www.gnu.org/licenses/>. *
 **************************************************************************/

/**
 * \file
 *
 * Implement tc_data_source.h -- Tide datasource container on top of
 * TCDataSource
 */

#include <wx/log.h>
#include <wx/filename.h>

#include "tc_data_source.h"
#include "tcds_ascii_harmonic.h"
#include "tcds_binary_harmonic.h"

#include <wx/arrimpl.cpp>
WX_DEFINE_OBJARRAY(ArrayOfTCDSources);

TCDataSource::TCDataSource() {
  m_pfactory = NULL;
  pTCDS_Ascii_Harmonic = NULL;
  pTCDS_Binary_Harmonic = NULL;
}

TCDataSource::~TCDataSource() {
  wxLogMessage("UnLoading Tide/Current data source: %s",
               m_data_source_path.c_str());

  delete pTCDS_Ascii_Harmonic;
  delete pTCDS_Binary_Harmonic;
}

TC_Error_Code TCDataSource::LoadData(const wxString &data_file_path) {
  m_data_source_path = data_file_path;
  wxLogMessage("Loading Tide/Current data source: %s",
               m_data_source_path.c_str());

  wxFileName fname(data_file_path);

  if (!fname.FileExists()) return TC_FILE_NOT_FOUND;

  if (fname.GetExt() == "IDX" || fname.GetExt() == "idx") {
    TCDS_Ascii_Harmonic *pdata = new TCDS_Ascii_Harmonic;
    m_pfactory = dynamic_cast<TCDataFactory *>(pdata);
    pTCDS_Ascii_Harmonic = pdata;
  } else if (fname.GetExt() == "tcd" || fname.GetExt() == "TCD") {
    TCDS_Binary_Harmonic *pdata = new TCDS_Binary_Harmonic;
    m_pfactory = dynamic_cast<TCDataFactory *>(pdata);
    pTCDS_Binary_Harmonic = pdata;
  }

  TC_Error_Code err_code;
  if (m_pfactory) {
    err_code = m_pfactory->LoadData(data_file_path);
    if (err_code != TC_NO_ERROR) {
      wxLogMessage("Error loading tide/current data.");
      return err_code;
    }

    //  Mark the index entries individually with owner
    unsigned int max_index = GetMaxIndex();
    for (unsigned int i = 0; i < max_index; i++) {
      IDX_entry *pIDX = GetIndexEntry(i);
      if (pIDX) {
        pIDX->pDataSource = this;
        strncpy(pIDX->source_ident, m_data_source_path.mb_str(),
                MAXNAMELEN - 1);
        pIDX->source_ident[MAXNAMELEN - 1] = '\0';
      }
    }
  } else
    err_code = TC_FILE_NOT_FOUND;

  return err_code;
}

int TCDataSource::GetMaxIndex() {
  if (m_pfactory)
    return m_pfactory->GetMaxIndex();
  else
    return 0;
}

IDX_entry *TCDataSource::GetIndexEntry(int n_index) {
  if (m_pfactory) {
    if (n_index < m_pfactory->GetMaxIndex())
      return m_pfactory->GetIndexEntry(n_index);
    else
      return NULL;
  } else
    return NULL;
}

TC_Error_Code TCDataSource::LoadHarmonicData(IDX_entry *pIDX) {
  switch (pIDX->source_data_type) {
    case SOURCE_TYPE_ASCII_HARMONIC:
      return pTCDS_Ascii_Harmonic->LoadHarmonicData(pIDX);
      break;

    case SOURCE_TYPE_BINARY_HARMONIC:
      return pTCDS_Binary_Harmonic->LoadHarmonicData(pIDX);
      break;

    default:
      return TC_GENERIC_ERROR;
  }
}
