/*
 *   Copyright (C) 2008 Peter Grasch <peter.grasch@bedahr.org>
 *
 *   This program is free software; you can redistribute it and/or modify
 *   it under the terms of the GNU General Public License version 2,
 *   or (at your option) any later version, 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
 *
 *   You should have received a copy of the GNU General Public
 *   License along with this program; if not, write to the
 *   Free Software Foundation, Inc.,
 *   51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
 */

#include "bompdict.h"
#include <QString>
#include <QFile>
#include <QTextStream>
#include <QRegExp>
#include <QFileInfo>
#include <KDebug>
#include <KMimeType>
#include <KFilterDev>
#include <QTextCodec>

/**
 * \brief Constructor
 * \author Peter Grasch
 * @param path Sets the path of the dictionary (member)
 * @param parent Parent of the object
 */
BOMPDict::BOMPDict(QObject* parent): Dict(parent)
{
}


/**
 * \brief Loads the file from the given path
 * \author Peter Grasch
 *
 * @fixme This function assumes the system-charset to be ISO-8859-1 and WILL destroy special characters if it isn't
 *
 * @param path If the path is empty the path set by the constructor is used
 */
void BOMPDict::load(QString path, QString encodingName)
{
  emit progress(0);

  QIODevice *dict = KFilterDev::deviceForFile(path,
    KMimeType::findByFileContent(path)->name());
  if (!dict)
    return;
  if (!dict->open(QIODevice::ReadOnly)) {
    dict->deleteLater();
    return;
  }

  int maxProg=0;

  KMimeType::Ptr mimeType = KMimeType::findByFileContent(path);
  if (mimeType->is("text/plain")) {               //not compressed
    QFileInfo info;
    QFile f(path);
    info.setFile(f);
    maxProg = info.size();
  }

  int currentProg = 0;

  QTextStream *dictStream = new QTextStream(dict);
  dictStream->setCodec(QTextCodec::codecForName(encodingName.toAscii()));
  emit loaded();

  QString line, xsp, terminal;
  int wordend, termend;
  line = dictStream->readLine(1000);

  QString filteredXsp;
  QString xspFertig;
  QString currentPhoneme;
  QString currentTerminal;
  QString currentFinalXsp;
  QString currentWord;
  while (!line.isNull()) {
    wordend = line.indexOf("\t");
    termend = line.indexOf("\t", wordend+1);

    xsp = line.mid(termend).trimmed();

    //		xsp.remove(QRegExp("^'*?*"));
    xsp.remove('\'');
    xsp.remove('|');
    xsp.remove(',');
    currentFinalXsp = segmentSampa(adaptToSimonPhonemeSet(xsp));

    currentWord = line.left(wordend);
    currentTerminal = line.mid(wordend,
      termend-wordend).trimmed();

    QStringList currentTerminals = currentTerminal.split(':', QString::SkipEmptyParts);
    QStringList currentTerminalsUnique;

    if (currentTerminals.isEmpty()) {
      line = dictStream->readLine(1000);
      continue;
    }

    QString currentTerminalStr = currentTerminals[0];
    currentTerminalsUnique << currentTerminalStr;
    words << currentWord;
    terminals << currentTerminalStr;
    pronunciations << currentFinalXsp;

    for (int k=1; k < currentTerminals.count(); k++) {
      currentTerminalStr = currentTerminals[k];
      if (!currentTerminalsUnique.contains(currentTerminalStr)) {
        currentTerminalsUnique << currentTerminalStr;

        words << currentWord;
        terminals << currentTerminalStr;
        pronunciations << currentFinalXsp;
      }
    }

    currentProg += line.length();

    if (maxProg != 0)
      emit progress((int) (((((double)currentProg) /
        ((double)maxProg)))*1000));
    else emit progress(-1);
    line = dictStream->readLine(1000);
  }
  delete dictStream;
  dict->close();
  delete dict;
}


BOMPDict::~BOMPDict()
{
}
