/*

 * Hedgewars, a free turn based strategy game

 * Copyright (c) 2009 Martin Minarik <ttsmj@pokec.sk>

 * Copyright (c) 2009-2011 Andrey Korotaev <unC0Rr@gmail.com>

 *

 * 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; version 2 of the License

 *

 * 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., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA

 */

#include <QFile>

#include <QTextStream>

#include <QStringList>

#include <QLineEdit>

#include "namegen.h"

#include "hwform.h"

#include "hwconsts.h"

HWNamegen::HWNamegen() :

    typesAvailable(false)

{

    loadTypes();

}

HWNamegen::~HWNamegen()

{

}

void HWNamegen::teamRandomName(HWTeam & team, const int HedgehogNumber)

{

    randomNameByHat(team, HedgehogNumber);

}

void HWNamegen::teamRandomNames(HWTeam & team, const bool changeteamname)

{

    if ((TypesHatnames.size() > 0) && typesAvailable){

        int kind = (rand()%(TypesHatnames.size()));

        if (changeteamname){

            if (TypesTeamnames[kind].size() > 0){

                team.setName(TypesTeamnames[kind][rand()%(TypesTeamnames[kind].size())]);

            }

            team.setGrave(getRandomGrave());

            team.setFort(getRandomFort());

            team.setVoicepack("Default");

        }

        //give each hedgehog a random name:

        //TODO: load the dictionary only once! (right now it's loaded once for each hedgehog)

        for(int i = 0; i < HEDGEHOGS_PER_TEAM; i++)

        {

            if ((TypesHatnames[kind].size()) > 0){

                HWHog hh = team.hedgehog(i);

                hh.Hat = TypesHatnames[kind][rand()%(TypesHatnames[kind].size())];

                team.setHedgehog(i,hh);

            }

            randomNameByHat(team,i);

        }

    }

}

void HWNamegen::randomNameByHat(HWTeam & team, const int HedgehogNumber)

{

    QStringList Dictionaries;

    hatCfgLoad(team.hedgehog(HedgehogNumber).Hat,Dictionaries);

    QStringList Dictionary;

    dictLoad(Dictionaries[rand()%(Dictionaries.size())],Dictionary);

    HWHog hh = team.hedgehog(HedgehogNumber);

    hh.Name = Dictionary[rand()%(Dictionary.size())];

    team.setHedgehog(HedgehogNumber, hh);

}

void HWNamegen::dictLoad(const QString filename, QStringList &list)

{

    list.clear();

    QFile file;

    file.setFileName(QString("%1/Data/Names/%2.txt").arg(cfgdir->absolutePath()).arg(filename));

    if (!file.exists()) file.setFileName(QString("%1/Names/%2.txt").arg(datadir->absolutePath()).arg(filename));

    if (file.open(QIODevice::ReadOnly | QIODevice::Text))

    {

        QTextStream in(&file);

        while (!in.atEnd()) {

            QString line = in.readLine();

            if(line != QString(""))

                {list.append(line);}

        }

    }

    if (list.size()==0)

         list.append(filename);

}

void HWNamegen::hatCfgLoad(const QString hatname, QStringList &list)

{

    list.clear();

    QFile file;

    file.setFileName(QString("%1/Data/Names/%2.cfg").arg(cfgdir->absolutePath()).arg(hatname));

    if (!file.exists()) file.setFileName(QString("%1/Names/%2.cfg").arg(datadir->absolutePath()).arg(hatname));

    if (file.open(QIODevice::ReadOnly | QIODevice::Text))

    {

        QTextStream in(&file);

        while (!in.atEnd()) {

            QString line = in.readLine();

            if(line != QString(""))

                {list.append(line);}

        }

    }

    if (list.size()==0)

         list.append(QString("generic"));

}

void HWNamegen::loadTypes()

{

    QFile file;

    file.setFileName(QString("%1/Data/Names/types.ini").arg(cfgdir->absolutePath()));

    if (!file.exists()) file.setFileName(QString("%1/Names/types.ini").arg(datadir->absolutePath()));

    if (!file.open(QIODevice::ReadOnly | QIODevice::Text))

        {typesAvailable = false; return;}

    int counter = 0; //counter starts with 0 (teamnames mode)

    TypesTeamnames.append(QStringList());

    TypesHatnames.append(QStringList());

    QTextStream in(&file);

    while (!in.atEnd()) {

        QString line = in.readLine();

        if (line == QString("#####")){

            counter++; //toggle mode (teamnames || hats)

            if ((counter%2) == 0){

                TypesTeamnames.append(QStringList());

                TypesHatnames.append(QStringList());

            }

        } else if ((line == QString("*****")) || (line == QString("*END*"))){

            typesAvailable = true; return; // bye bye

        } else {

            if ((counter%2) == 0){ // even => teamnames mode

                TypesTeamnames[(counter/2)].append(line);

            } else { // odd => hats mode

                TypesHatnames[((counter-1)/2)].append(line);

            }

        }

//        Types.append(line);

    }

        typesAvailable = true;

    return;

}

QString HWNamegen::getRandomGrave()

{

    QStringList Graves;

    //list all available Graves

    QDir tmpdir;

    tmpdir.cd(cfgdir->absolutePath());

    tmpdir.cd("Data/Graphics/Graves");

    tmpdir.setFilter(QDir::Files);

    Graves.append(tmpdir.entryList(QStringList("*.png")).replaceInStrings(QRegExp("^(.*)\\.png"), "\\1"));

    tmpdir.cd(datadir->absolutePath());

    tmpdir.cd("Graphics/Graves");

    tmpdir.setFilter(QDir::Files);

    QStringList tmpList = tmpdir.entryList(QStringList("*.png")).replaceInStrings(QRegExp("^(.*)\\.png"), "\\1");

    for (QStringList::Iterator it = tmpList.begin(); it != tmpList.end(); ++it) 

        if (!Graves.contains(*it,Qt::CaseInsensitive)) Graves.append(*it);

    if(Graves.size()==0)

    {

        //do some serious error handling

        return "Error";

    }

    //pick a random grave

    return Graves[rand()%(Graves.size())];

}

QString HWNamegen::getRandomFort()

{

    QStringList Forts;

    //list all available Forts

    QDir tmpdir;

    tmpdir.cd(datadir->absolutePath());

    tmpdir.cd("Forts");

    tmpdir.setFilter(QDir::Files);

    Forts.append(tmpdir.entryList(QStringList("*L.png")).replaceInStrings(QRegExp("^(.*)L\\.png"), "\\1"));

    if(Forts.size()==0)

    {

        //do some serious error handling

        return "Error";

    }

    //pick a random fort

    return Forts[rand()%(Forts.size())];

}