src/opengl/util/generator.cpp
changeset 0 1918ee327afb
child 4 3b1da2848fc7
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/src/opengl/util/generator.cpp	Mon Jan 11 14:00:40 2010 +0000
@@ -0,0 +1,500 @@
+/****************************************************************************
+**
+** 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