diff options
Diffstat (limited to 'generator/cppgenerator.cpp')
-rw-r--r-- | generator/cppgenerator.cpp | 17 |
1 files changed, 9 insertions, 8 deletions
diff --git a/generator/cppgenerator.cpp b/generator/cppgenerator.cpp index 687a24499..44d62f4f6 100644 --- a/generator/cppgenerator.cpp +++ b/generator/cppgenerator.cpp @@ -3023,15 +3023,16 @@ void CppGenerator::writeRichCompareFunction(QTextStream& s, const AbstractMetaCl QString expression = QString("%1%2 %3 " CPP_ARG0) .arg(func->isPointerOperator() ? "&" : "") .arg(CPP_SELF_VAR).arg(op); + s << INDENT; + if (func->type()) + s << func->type()->cppSignature() << " " CPP_RETURN_VAR " = "; + s << expression << ';' << endl; s << INDENT << PYTHON_RETURN_VAR " = "; - if (!func->type()) { - s << "Py_None;" << endl; - s << INDENT << "Py_INCREF(Py_None);" << endl; - s << INDENT << expression << "; // This operator returns void." << endl; - } else { - writeToPythonConversion(s, func->type(), metaClass, expression); - s << ';' << endl; - } + if (func->type()) + writeToPythonConversion(s, func->type(), metaClass, CPP_RETURN_VAR); + else + s << "Py_None;" << endl << INDENT << "Py_INCREF(Py_None)"; + s << ';' << endl; } } s << INDENT << '}'; |