/****************************************************************************

**

** Copyright (C) 2009 Nokia Corporation and/or its subsidiary(-ies).

** All rights reserved.

** Contact: Nokia Corporation (qt-info@nokia.com)

**

** This file is part of the QtOpenGL module of the Qt Toolkit.

**

** $QT_BEGIN_LICENSE:LGPL$

** No Commercial Usage

** This file contains pre-release code and may not be distributed.

** You may use this file in accordance with the terms and conditions

** contained in the Technology Preview License Agreement accompanying

** this package.

**

** GNU Lesser General Public License Usage

** Alternatively, this file may be used under the terms of the GNU Lesser

** General Public License version 2.1 as published by the Free Software

** Foundation and appearing in the file LICENSE.LGPL included in the

** packaging of this file.  Please review the following information to

** ensure the GNU Lesser General Public License version 2.1 requirements

** will be met: http://www.gnu.org/licenses/old-licenses/lgpl-2.1.html.

**

** In addition, as a special exception, Nokia gives you certain additional

** rights.  These rights are described in the Nokia Qt LGPL Exception

** version 1.1, included in the file LGPL_EXCEPTION.txt in this package.

**

** If you have questions regarding the use of this file, please contact

** Nokia at qt-info@nokia.com.

**

**

**

**

**

**

**

**

** $QT_END_LICENSE$

**

****************************************************************************/

#include <QFile>

#include <QList>

#include <QMap>

#include <QPair>

#include <QSet>

#include <QString>

#include <QTextStream>

#include <QtDebug>

#include <cstdlib>

QT_BEGIN_NAMESPACE

QT_USE_NAMESPACE

#define TAB "    "

typedef QPair<QString, QString> QStringPair;

QString readSourceFile(const QString &sourceFile, bool fragmentProgram = false)

{

    QFile file(sourceFile);

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

        qDebug() << "Missing source file" << sourceFile;

        exit(0);

    }

    QString source;

    QTextStream in(&file);

    while (!in.atEnd()) {

        QString line = in.readLine();

        if (fragmentProgram && line[0] == '#' && !line.startsWith("#var"))

            continue;

        if (fragmentProgram)

            source.append("    \"");

        source.append(line);

        if (fragmentProgram)

            source.append("\\n\"");

        source.append('\n');

    }

    if (fragmentProgram)

        source.append("    ;\n");

    return source;

}

QList<QStringPair> readConf(const QString &confFile)

{

    QFile file(confFile);

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

        qDebug() << "Missing file" << confFile;

        exit(0);

    }

    QList<QStringPair> result;

    QTextStream in(&file);

    while (!in.atEnd()) {

        QString line = in.readLine();

        if (line.startsWith('#'))

            continue;

        QTextStream lineStream(&line);

        QString enumerator;

        QString sourceFile;

        lineStream >> enumerator;

        if (lineStream.atEnd()) {

            qDebug() << "Error in file" << confFile << '(' << enumerator << ')';

            exit(0);

        }

        lineStream >> sourceFile;

        result << QStringPair(enumerator, readSourceFile(sourceFile));

    }

    return result;

}

QString compileSource(const QString &source)

{

    {

        QFile tempSourceFile("__tmp__.glsl");

        if (!tempSourceFile.open(QIODevice::WriteOnly | QIODevice::Text)) {

            qDebug() << "Failed opening __tmp__.glsl";

            exit(0);

        }

        QTextStream out(&tempSourceFile);

        out << source;

    }

    if (std::system("cgc -quiet -oglsl -profile arbfp1 __tmp__.glsl >__tmp__.frag") == -1) {

        qDebug() << "Failed running cgc";

        exit(0);

    }

    return readSourceFile("__tmp__.frag", true);

}

QString getWord(QString line, int word)

{

    QTextStream in(&line);

    QString result;

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

        in >> result;

    return result;

}

static int toInt(const QByteArray &str)

{

    int value = 0;

    for (int i = 0; i < str.size(); ++i) {

        if (str[i] < '0' || str[i] > '9')

            break;

        value *= 10;

        value += (str[i] - '0');

    }

    return value;

}

QList<int> getLocations(const QSet<QString> &variables, QString source)

{

    QTextStream in(&source);

    QMap<QString, int> locations;

    foreach (QString variable, variables)

        locations[variable] = -1;

    while (!in.atEnd()) {

        QString line = in.readLine().trimmed();

        line = line.right(line.size() - 1);

        if (line.startsWith("#var")) {

            QByteArray temp;

            QByteArray name;

            QTextStream lineStream(&line);

            lineStream >> temp >> temp >> name;

            int location = -1;

            while (!lineStream.atEnd()) {

                lineStream >> temp;

                if (temp.startsWith("c[")) {

                    location = toInt(temp.right(temp.size() - 2));

                    break;

                }

                if (temp == "texunit") {

                    lineStream >> temp;

                    location = toInt(temp);

                    break;

                }

            }

            locations[name] = location;

        }

    }

    QList<int> result;

    foreach (QString variable, variables)

        result << locations[variable];

    return result;

}

// remove #var statements

QString trimmed(QString source)

{

    QTextStream in(&source);

    QString result;

    while (!in.atEnd()) {

        QString line = in.readLine();

        if (!line.trimmed().startsWith("\"#"))

            result += line + '\n';

    }

    return result;

}

void writeVariablesEnum(QTextStream &out, const char *name, const QSet<QString> &s)

{

    out << "enum " << name << " {";

    QSet<QString>::const_iterator it = s.begin();

    if (it != s.end()) {

        out << "\n" TAB "VAR_" << it->toUpper();

        for (++it; it != s.end(); ++it)

            out << ",\n" TAB "VAR_" << it->toUpper();

    }

    out << "\n};\n\n";

}

void writeTypesEnum(QTextStream &out, const char *name, const QList<QStringPair> &s)

{

    out << "enum " << name << " {";

    QList<QStringPair>::const_iterator it = s.begin();

    if (it != s.end()) {

        out << "\n" TAB << it->first;

        for (++it; it != s.end(); ++it)

            out << ",\n" TAB << it->first;

    }

    out << "\n};\n\n";

}

void writeIncludeFile(const QSet<QString> &variables,

                      const QList<QStringPair> &brushes,

                      const QList<QStringPair> &compositionModes,

                      const QList<QStringPair> &masks,

                      const QMap<QString, QMap<QString, QString> > &compiled)

{

    QFile includeFile("fragmentprograms_p.h");

    if (!includeFile.open(QIODevice::WriteOnly | QIODevice::Text)) {

        qDebug() << "Failed opening fragmentprograms_p.h";

        exit(0);

    }

    QTextStream out(&includeFile);

    QLatin1String tab(TAB);

    out << "/****************************************************************************\n"

           "**\n"

           "** Copyright (C) 2009 Nokia Corporation and/or its subsidiary(-ies).\n"

           "** All rights reserved.\n"

           "** Contact: Nokia Corporation (qt-info@nokia.com)\n"

           "**\n"

           "** This file is part of the QtOpenGL module of the Qt Toolkit.\n"

           "**\n"

           "** $QT_BEGIN_LICENSE:LGPL$\n"

           "** No Commercial Usage\n"

           "** This file contains pre-release code and may not be distributed.\n"

           "** You may use this file in accordance with the terms and conditions\n"

           "** contained in the Technology Preview License Agreement accompanying\n"

           "** this package.\n"

           "**\n"

           "** GNU Lesser General Public License Usage\n"

           "** Alternatively, this file may be used under the terms of the GNU Lesser\n"

           "** General Public License version 2.1 as published by the Free Software\n"

           "** Foundation and appearing in the file LICENSE.LGPL included in the\n"

           "** packaging of this file.  Please review the following information to\n"

           "** ensure the GNU Lesser General Public License version 2.1 requirements\n"

           "** will be met: http://www.gnu.org/licenses/old-licenses/lgpl-2.1.html.\n"

           "**\n"

           "** In addition, as a special exception, Nokia gives you certain additional\n"

           "** rights.  These rights are described in the Nokia Qt LGPL Exception\n"

           "** version 1.1, included in the file LGPL_EXCEPTION.txt in this package.\n"

           "**\n"

           "** If you have questions regarding the use of this file, please contact\n"

           "** Nokia at qt-info@nokia.com.\n"

           "**\n"

           "**\n"

           "**\n"

           "**\n"

           "**\n"

           "**\n"

           "**\n"

           "**\n"

           "** $QT_END_LICENSE$\n"

           "**\n"

           "****************************************************************************/\n"

           "\n"

           "#ifndef FRAGMENTPROGRAMS_P_H\n"

           "#define FRAGMENTPROGRAMS_P_H\n"

           "\n"

           "//\n"

           "//  W A R N I N G\n"

           "//  -------------\n"

           "//\n"

           "// This file is not part of the Qt API.  It exists purely as an\n"

           "// implementation detail.  This header file may change from version to\n"

           "// version without notice, or even be removed.\n"

           "//\n"

           "// We mean it.\n"

           "//\n"

           "\n";

    writeVariablesEnum(out, "FragmentVariable", variables);

    writeTypesEnum(out, "FragmentBrushType", brushes);

    writeTypesEnum(out, "FragmentCompositionModeType", compositionModes);

    writeTypesEnum(out, "FragmentMaskType", masks);

    out << "static const unsigned int num_fragment_variables = " << variables.size() << ";\n\n";

    out << "static const unsigned int num_fragment_brushes = " << brushes.size() << ";\n";

    out << "static const unsigned int num_fragment_composition_modes = " << compositionModes.size() << ";\n";

    out << "static const unsigned int num_fragment_masks = " << masks.size() << ";\n\n";

    foreach (QStringPair mask, masks) {

        const QString compiledSource = compiled[mask.first]["MASK__"];

        out << "static const char *FragmentProgram_" << mask.first << " =\n"

            << trimmed(compiledSource)

            << '\n';

    }

    foreach (QStringPair brush, brushes) {

        foreach (QStringPair mode, compositionModes) {

            const QString compiledSource = compiled[brush.first][mode.first];

            out << "static const char *FragmentProgram_" << brush.first << '_' << mode.first << " =\n"

                << trimmed(compiledSource)

                << '\n';

        }

    }

    out << "static const char *mask_fragment_program_sources[num_fragment_masks] = {\n";

    foreach (QStringPair mask, masks)

        out << tab << "FragmentProgram_" << mask.first << ",\n";

    out << "};\n\n";

    out << "static const char *painter_fragment_program_sources[num_fragment_brushes][num_fragment_composition_modes] = {\n";

    foreach (QStringPair brush, brushes) {

        out << tab << "{\n";

        foreach (QStringPair mode, compositionModes)

            out << tab << tab << "FragmentProgram_" << brush.first << '_' << mode.first << ",\n";

        out << tab << "},\n";

    }

    out << "};\n\n";

    out << "static int painter_variable_locations[num_fragment_brushes][num_fragment_composition_modes][num_fragment_variables] = {\n";

    foreach (QStringPair brush, brushes) {

        out << tab << "{\n";

        foreach (QStringPair mode, compositionModes) {

            out << tab << tab << "{ ";

            QList<int> locations = getLocations(variables, compiled[brush.first][mode.first]);

            foreach (int location, locations)

                out << location << ", ";

            out << "},\n";

        }

        out << tab << "},\n";

    }

    out << "};\n\n";

    out << "static int mask_variable_locations[num_fragment_masks][num_fragment_variables] = {\n";

    foreach (QStringPair mask, masks) {

        out << tab << "{ ";

        QList<int> locations = getLocations(variables, compiled[mask.first]["MASK__"]);

        foreach (int location, locations)

            out << location << ", ";

        out << "},\n";

    }

    out << "};\n\n";

    out << "#endif\n";

}

QList<QString> getVariables(QString program)

{

    QList<QString> result;

    QTextStream in(&program);

    while (!in.atEnd()) {

        QString line = in.readLine();

        if (line.startsWith("uniform")) {

            QString word = getWord(line, 3);

            result << word.left(word.size() - 1);

        } else if (line.startsWith("#include")) {

            QString file = getWord(line, 2);

            result << getVariables(readSourceFile(file.mid(1, file.size() - 2)));

        }

    }

    return result;

}

int main()

{

    QList<QStringPair> brushes = readConf(QLatin1String("brushes.conf"));

    QList<QStringPair> compositionModes = readConf(QLatin1String("composition_modes.conf"));

    QList<QStringPair> masks = readConf(QLatin1String("masks.conf"));

    QString painterSource = readSourceFile("painter.glsl");

    QString painterNoMaskSource = readSourceFile("painter_nomask.glsl");

    QString fastPainterSource = readSourceFile("fast_painter.glsl");

    QString brushPainterSource = readSourceFile("brush_painter.glsl");

    QSet<QString> variables;

    QList<QStringPair> programs[3] = { brushes, compositionModes, masks };

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

        foreach (QStringPair value, programs[i])

            variables += QSet<QString>::fromList(getVariables(value.second));

    variables += QSet<QString>::fromList(getVariables(painterSource));

    variables += QSet<QString>::fromList(getVariables(fastPainterSource));

    QMap<QString, QMap<QString, QString> > compiled;

    foreach (QStringPair brush, brushes) {

        foreach (QStringPair mode, compositionModes) {

            QString combinedSource = brush.second + mode.second + painterSource;

            compiled[brush.first][mode.first] = compileSource(combinedSource);

            combinedSource = brush.second + mode.second + painterNoMaskSource;

            compiled[brush.first][mode.first + "_NOMASK"] = compileSource(combinedSource);

        }

        QString fastSource = brush.second + fastPainterSource;

        QString brushSource = brush.second + brushPainterSource;

        compiled[brush.first]["COMPOSITION_MODE_BLEND_MODE_MASK"] = compileSource(fastSource);

        compiled[brush.first]["COMPOSITION_MODE_BLEND_MODE_NOMASK"] = compileSource(brushSource);

    }

    QList<QStringPair> temp;

    foreach (QStringPair mode, compositionModes)

        temp << QStringPair(mode.first + "_NOMASK", mode.second);

    compositionModes += temp;

    compositionModes << QStringPair("COMPOSITION_MODE_BLEND_MODE_MASK", "")

                     << QStringPair("COMPOSITION_MODE_BLEND_MODE_NOMASK", "");

    foreach (QStringPair mask, masks)

        compiled[mask.first]["MASK__"] = compileSource(mask.second);

    writeIncludeFile(variables, brushes, compositionModes, masks, compiled);

    return 0;

}

QT_END_NAMESPACE