Support mixed type registrations and enum definitions in codegen, needed for enums.h.
[qtgstreamer:qtgstreamer.git] / codegen / generator.cpp
1 /*
2     Copyright (C) 2010  George Kiagiadakis <kiagiadakis.george@gmail.com>
3
4     This library is free software; you can redistribute it and/or modify
5     it under the terms of the GNU Lesser General Public License as published
6     by the Free Software Foundation; either version 2.1 of the License, or
7     (at your option) any later version.
8
9     This program is distributed in the hope that it will be useful,
10     but WITHOUT ANY WARRANTY; without even the implied warranty of
11     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12     GNU General Public License for more details.
13
14     You should have received a copy of the GNU Lesser General Public License
15     along with this program.  If not, see <http://www.gnu.org/licenses/>.
16 */
17 #include "generator.h"
18 #include <cstdio>
19 #include <cstdlib>
20 #include <QtCore/QCoreApplication>
21 #include <QtCore/QFile>
22
23 extern int yylineno;
24 int yyparse(CodeGen *codegen);
25 void yyrestart(FILE *file);
26
27 int main(int argc, char *argv[])
28 {
29     QCoreApplication a(argc, argv);
30     QTextStream(stdout) << "// Autogenerated by the QtGstreamer helper code generator" << endl
31                         << "#include <gst/gst.h>" << endl
32                         << "#include <boost/static_assert.hpp>" << endl << endl;
33
34     for (int i=1; i<argc; ++i) {
35         QString fileName(QFile::decodeName(argv[i]));
36         if (fileName.endsWith(".h") && QFile::exists(fileName)) {
37             CodeGen::parse(fileName);
38         } else {
39             QTextStream(stderr) << "Skipping " << fileName << ": Not an existing header" << endl;
40         }
41     }
42
43     return 0;
44 }
45
46 void CodeGen::parse(const QString & fileName)
47 {
48     CodeGen codegen(fileName);
49
50     FILE *fp = std::fopen(QFile::encodeName(fileName), "r");
51     if (!fp) {
52         std::perror("fopen");
53         QTextStream(stderr) << "Could not open " << fileName << endl;
54         return;
55     }
56
57     yylineno = 1;
58     yyrestart(fp);
59     yyparse(&codegen);
60     std::fclose(fp);
61
62     codegen.generateOutput();
63 }
64
65 void CodeGen::generateOutput()
66 {
67     QTextStream outStream(stdout);
68     outStream << "#include \"" << m_fileName << "\"" << endl << endl;
69
70     foreach(const TypeRegistration & typeReg, m_typeRegistrations) {
71         printTypeRegistration(outStream, typeReg);
72         outStream << endl;
73     }
74
75     foreach(const Enum & enumDef, m_enums) {
76         if (!enumDef.options.contains("skip")) {
77             printEnumAssertions(outStream, enumDef);
78         }
79         outStream << endl;
80     }
81 }
82
83 void CodeGen::printTypeRegistration(QTextStream & outStream, const TypeRegistration & typeReg)
84 {
85     outStream << "QGLIB_REGISTER_TYPE_IMPLEMENTATION(";
86
87     outStream << typeReg["namespace"] << "::" << typeReg["class"];
88     if (!typeReg["enum"].isEmpty()) {
89         outStream << "::" << typeReg["enum"];
90     }
91
92     outStream << ',';
93
94     if (typeReg.contains("GType")) {
95         outStream << typeReg["GType"];
96     } else {
97         outStream << (typeReg["namespace"] == "QGst" ? "GST_TYPE_" : "G_TYPE_");
98         outStream << toGstStyle(typeReg["enum"].isEmpty() ? typeReg["class"] : typeReg["enum"]);
99     }
100     outStream << ")" << endl;
101 }
102
103 void CodeGen::printEnumAssertions(QTextStream& outStream, const Enum & enumDef)
104 {
105     outStream << "namespace " << enumDef.options["namespace"] << " {" << endl;
106
107     foreach(const QByteArray & value, enumDef.values) {
108         outStream << "    BOOST_STATIC_ASSERT(static_cast<int>(";
109         if (enumDef.options.contains("class") && !enumDef.options["class"].isEmpty()) {
110             outStream << enumDef.options["class"] << "::";
111         }
112         outStream << value;
113         outStream << ") == static_cast<int>(";
114
115         if (enumDef.options.contains("prefix")) {
116             outStream << enumDef.options["prefix"];
117         } else {
118             outStream << (enumDef.options["namespace"] == "QGst" ? "GST_" : "G_");
119             if (enumDef.options.contains("class") && !enumDef.options["class"].isEmpty()) {
120                 outStream << toGstStyle(enumDef.options["class"]) << "_";
121             }
122         }
123
124         if (enumDef.options.contains(value)) {
125             outStream << enumDef.options[value];
126         } else {
127             outStream << toGstStyle(value);
128         }
129         outStream << "));" << endl;
130     }
131     outStream << "}" << endl;
132 }
133
134 QByteArray CodeGen::toGstStyle(const QByteArray & str)
135 {
136     QByteArray output;
137     foreach(const char currentChar, str) {
138         if (isupper(currentChar)) {
139             if (!output.isEmpty()) { //if this is not the first char
140                 output.append('_');
141             }
142             output.append(currentChar);
143         } else {
144             output.append(toupper(currentChar));
145         }
146     }
147     return output;
148 }
149
150
151 void CodeGen::addEnum(const QList<QByteArray> & values, const QHash<QByteArray, QByteArray> & options)
152 {
153     Enum e;
154     e.values = values;
155     e.options = options;
156     e.options["namespace"] = m_currentNamespace;
157     e.options["class"] = m_currentClass;
158     m_enums.append(e);
159 }
160
161 void CodeGen::addTypeRegistration(const QByteArray& namespaceId, const QByteArray& classId,
162                                  const QByteArray& enumId, const QHash<QByteArray, QByteArray>& options)
163 {
164     TypeRegistration typeReg(options);
165     typeReg["namespace"] = namespaceId;
166     typeReg["class"] = classId;
167     typeReg["enum"] = enumId;
168     m_typeRegistrations.append(typeReg);
169 }
170
171 //called by yyerror()
172 void CodeGen::fatalError(const char* msg)
173 {
174     QTextStream(stderr) << "codegen: " << m_fileName << ":" << yylineno << ": error: " << msg << endl;
175     std::exit(1);
176 }
177