##// END OF EJS Templates
added sorting of polymorphic id classes to avoid changes on each code generation...
added sorting of polymorphic id classes to avoid changes on each code generation git-svn-id: svn://svn.code.sf.net/p/pythonqt/code/trunk@168 ea8d5007-eb21-0410-b261-ccb3ea6e24a9

File last commit:

r124:72837eb4c23c
r131:ed1c6f151a2c
Show More
shellimplgenerator.cpp
316 lines | 11.2 KiB | text/x-c | CppLexer
/ generator / shellimplgenerator.cpp
/****************************************************************************
**
** Copyright (C) 2008-2009 Nokia Corporation and/or its subsidiary(-ies).
** All rights reserved.
** Contact: Nokia Corporation (qt-info@nokia.com)
**
** This file is part of the Qt Script Generator project on Qt Labs.
**
** $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 "shellimplgenerator.h"
#include "reporthandler.h"
#include "fileout.h"
#include <iostream>
extern void declareFunctionMetaTypes(QTextStream &stream,
const AbstractMetaFunctionList &functions,
QSet<QString> &registeredTypeNames);
QString ShellImplGenerator::fileNameForClass(const AbstractMetaClass *meta_class) const
{
return QString("PythonQtWrapper_%1.cpp").arg(meta_class->name());
}
static bool include_less_than(const Include &a, const Include &b)
{
return a.name < b.name;
}
static void writeHelperCode(QTextStream &s, const AbstractMetaClass *)
{
}
void ShellImplGenerator::write(QTextStream &s, const AbstractMetaClass *meta_class)
{
QString builtIn = ShellGenerator::isBuiltIn(meta_class->name())?"_builtin":"";
QString pro_file_name = meta_class->package().replace(".", "_") + builtIn + "/" + meta_class->package().replace(".", "_") + builtIn + ".pri";
priGenerator->addSource(pro_file_name, fileNameForClass(meta_class));
s << "#include \"PythonQtWrapper_" << meta_class->name() << ".h\"" << endl << endl;
s << "#include <PythonQtSignalReceiver.h>" << endl;
s << "#include <PythonQtMethodInfo.h>" << endl;
s << "#include <PythonQtConversion.h>" << endl;
//if (!meta_class->generateShellClass())
// return;
IncludeList list = meta_class->typeEntry()->extraIncludes();
qSort(list.begin(), list.end());
foreach (const Include &inc, list) {
ShellGenerator::writeInclude(s, inc);
}
s << endl;
writeHelperCode(s, meta_class);
// find constructors
AbstractMetaFunctionList ctors;
ctors = meta_class->queryFunctions(AbstractMetaClass::Constructors
| AbstractMetaClass::WasVisible
| AbstractMetaClass::NotRemovedFromTargetLang);
// find member functions
AbstractMetaFunctionList functions = getFunctionsToWrap(meta_class);
// write metatype declarations
{
// QSet<QString> registeredTypeNames = m_qmetatype_declared_typenames;
// declareFunctionMetaTypes(s, functions, registeredTypeNames);
// s << endl;
}
if (meta_class->generateShellClass()) {
AbstractMetaFunctionList virtualsForShell = getVirtualFunctionsForShell(meta_class);
foreach (const AbstractMetaFunction *fun, virtualsForShell) {
bool hasReturnValue = (fun->type());
writeFunctionSignature(s, fun, meta_class, QString(),
Option(OriginalName | ShowStatic | UnderscoreSpaces),
"PythonQtShell_");
s << endl << "{" << endl;
Option typeOptions = Option(OriginalName | UnderscoreSpaces | SkipName);
AbstractMetaArgumentList args = fun->arguments();
s << "if (_wrapper) {" << endl;
s << " PyObject* obj = PyObject_GetAttrString((PyObject*)_wrapper, \"" << fun->name() << "\");" << endl;
s << " PyErr_Clear();" << endl;
s << " if (obj && !PythonQtSlotFunction_Check(obj)) {" << endl;
s << " static const char* argumentList[] ={\"";
if (hasReturnValue) {
// write the arguments, return type first
writeTypeInfo(s, fun->type(), typeOptions);
}
s << "\"";
for (int i = 0; i < args.size(); ++i) {
s << " , \"";
writeTypeInfo(s, args.at(i)->type(), typeOptions);
s << "\"";
}
s << "};" << endl;
s << " static const PythonQtMethodInfo* methodInfo = PythonQtMethodInfo::getCachedMethodInfoFromArgumentList(" << QString::number(args.size()+1) << ", argumentList);" << endl;
if (hasReturnValue) {
s << " ";
writeTypeInfo(s, fun->type(), typeOptions);
s << " returnValue;" << endl;
// TODO: POD init to default is missing...
}
s << " void* args[" << QString::number(args.size()+1) << "] = {NULL";
for (int i = 0; i < args.size(); ++i) {
s << ", (void*)&" << args.at(i)->argumentName();
}
s << "};" << endl;
s << " PyObject* result = PythonQtSignalTarget::call(obj, methodInfo, args, true);" << endl;
if (hasReturnValue) {
s << " if (result) {" << endl;
s << " args[0] = PythonQtConv::ConvertPythonToQt(methodInfo->parameters().at(0), result, false, NULL, &returnValue);" << endl;
s << " if (args[0]!=&returnValue) {" << endl;
s << " if (args[0]==NULL) {" << endl;
s << " PythonQt::priv()->handleVirtualOverloadReturnError(\"" << fun->name() << "\", methodInfo, result);" << endl;
s << " } else {" << endl;
s << " returnValue = *((";
writeTypeInfo(s, fun->type(), typeOptions);
s << "*)args[0]);" << endl;
s << " }" << endl;
s << " }" << endl;
s << " }" << endl;
}
s << " if (result) { Py_DECREF(result); } " << endl;
s << " Py_DECREF(obj);" << endl;
if (hasReturnValue) {
s << " return returnValue;" << endl;
} else {
s << " return;" << endl;
}
s << " }" << endl;
s << "}" << endl;
s << " ";
if (fun->isAbstract()) {
if (fun->type()) {
// return empty default object
s << "return ";
if (fun->type()->indirections()>0) {
s << "0;";
} else {
writeTypeInfo(s, fun->type(), typeOptions);
s << "();";
}
}
} else {
if (fun->type()) {
s << "return ";
}
s << meta_class->qualifiedCppName() << "::";
s << fun->originalName() << "(";
for (int i = 0; i < args.size(); ++i) {
if (i > 0)
s << ", ";
s << args.at(i)->argumentName();
}
s << ");";
}
s << endl << "}" << endl;
}
}
if (meta_class->generateShellClass() || !meta_class->isAbstract()) {
// write constructors
foreach (const AbstractMetaFunction *ctor, ctors) {
if (!ctor->isPublic() || ctor->isAbstract()) { continue; }
s << meta_class->qualifiedCppName() << "* ";
s << "PythonQtWrapper_" << meta_class->name() << "::";
writeFunctionSignature(s, ctor, 0, "new_", Option(OriginalName | ShowStatic));
s << endl;
s << "{ " << endl;
s << "return new " << (meta_class->generateShellClass()?shellClassName(meta_class):meta_class->qualifiedCppName()) << "(";
AbstractMetaArgumentList args = ctor->arguments();
for (int i = 0; i < args.size(); ++i) {
if (i > 0)
s << ", ";
s << args.at(i)->argumentName();
}
s << ");" << " }" << endl << endl;
}
}
QString wrappedObject = " (*theWrappedObject)";
// write member functions
for (int i = 0; i < functions.size(); ++i) {
AbstractMetaFunction *fun = functions.at(i);
if (fun->isSlot()) continue;
writeFunctionSignature(s, fun, meta_class, QString(),
Option(ConvertReferenceToPtr | FirstArgIsWrappedObject | OriginalName | ShowStatic | UnderscoreSpaces),
"PythonQtWrapper_");
s << endl << "{" << endl;
s << " ";
if (ShellGenerator::isSpecialStreamingOperator(fun)) {
s << fun->arguments().at(0)->argumentName();
if (fun->originalName().startsWith("operator>>")) {
s << " >> ";
} else {
s << " << ";
}
s << wrappedObject;
} else {
QString scriptFunctionName = fun->originalName();
AbstractMetaArgumentList args = fun->arguments();
// call the C++ implementation
if (fun->type()) {
s << "return ";
// call the C++ implementation
if (fun->type()->isReference()) {
s << "&";
}
}
s << "(";
if (scriptFunctionName.startsWith("operator>>")) {
s << wrappedObject << " >>" << args.at(0)->argumentName();
} else if (scriptFunctionName.startsWith("operator<<")) {
s << wrappedObject << " <<" << args.at(0)->argumentName();
} else if (scriptFunctionName.startsWith("operator[]")) {
s << wrappedObject << "[" << args.at(0)->argumentName() << "]";
} else if (scriptFunctionName.startsWith("operator") && args.size()==1) {
QString op = scriptFunctionName.mid(8);
s << wrappedObject << op << " " << args.at(0)->argumentName();
} else {
if (fun->isStatic()) {
s << meta_class->qualifiedCppName() << "::";
} else {
if (!fun->isPublic() || fun->isVirtual()) {
s << " ((" << promoterClassName(meta_class) << "*)theWrappedObject)->promoted_";
} else {
s << " theWrappedObject->";
}
}
s << fun->originalName() << "(";
for (int i = 0; i < args.size(); ++i) {
if (i > 0)
s << ", ";
s << args.at(i)->argumentName();
}
s << ")";
}
s << ")";
}
s << ";" << endl;
s << "}" << endl << endl;
}
if (meta_class->hasDefaultToStringFunction()) {
s << "QString PythonQtWrapper_" << meta_class->name() << "::py_toString(" << meta_class->qualifiedCppName() << "* obj) { return obj->toString(); }" << endl;
} else if (meta_class->hasToStringCapability()) {
FunctionModelItem fun = meta_class->hasToStringCapability();
int indirections = fun->arguments().at(1)->type().indirections();
QString deref = QLatin1String(indirections == 0 ? "*" : "");
s << "QString PythonQtWrapper_" << meta_class->name() << "::py_toString(" << meta_class->qualifiedCppName() << "* obj) {" << endl;
s << " QString result;" << endl;
s << " QDebug d(&result);" << endl;
s << " d << " << deref << "obj;" << endl;
s << " return result;" << endl;
s << "}" << endl << endl;
}
writeInjectedCode(s, meta_class);
}
void ShellImplGenerator::writeInjectedCode(QTextStream &s, const AbstractMetaClass *meta_class)
{
CodeSnipList code_snips = meta_class->typeEntry()->codeSnips();
foreach (const CodeSnip &cs, code_snips) {
if (cs.language == TypeSystem::PyWrapperCode) {
s << cs.code() << endl;
}
}
}