shellimplgenerator.cpp
324 lines
| 11.9 KiB
| text/x-c
|
CppLexer
/ generator_50 / shellimplgenerator.cpp
r0 | /**************************************************************************** | |||
** | ||||
** 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> ®isteredTypeNames); | ||||
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()) { | ||||
s << shellClassName(meta_class) << "::~" << shellClassName(meta_class) << "() {" << endl; | ||||
s << " PythonQtPrivate* priv = PythonQt::priv();" << endl; | ||||
s << " if (priv) { priv->shellClassDeleted(this); }" << endl; | ||||
s << "}" << endl; | ||||
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); | ||||
bool needsWrapping = (!fun->isSlot() || fun->isVirtual()); | ||||
if (!needsWrapping) { | ||||
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; | ||||
} | ||||
} | ||||
} | ||||