1 files changed, 61 insertions, 4 deletions
diff --git a/src/tools/tracegen/helpers.cpp b/src/tools/tracegen/helpers.cpp
index de523e74a6..0ea5848493 100644
--- a/src/tools/tracegen/helpers.cpp
+++ b/src/tools/tracegen/helpers.cpp
@@ -1,11 +1,26 @@
 // Copyright (C) 2017 Klarälvdalens Datakonsult AB, a KDAB Group company, info@kdab.com, author Rafael Roquetto <rafael.roquetto@kdab.com>
-// SPDX-License-Identifier: LicenseRef-Qt-Commercial OR LGPL-3.0-only OR GPL-2.0-only OR GPL-3.0-only
+// SPDX-License-Identifier: LicenseRef-Qt-Commercial OR GPL-3.0-only WITH Qt-GPL-exception-1.0
 
 #include "helpers.h"
 #include <qdebug.h>
 
 using namespace Qt::StringLiterals;
 
+void writeCommonPrologue(QTextStream &stream)
+{
+    stream << R"CPP(
+#ifndef Q_TRACEPOINT
+#error "Q_TRACEPOINT not set for the module, Q_TRACE not enabled."
+#endif
+)CPP";
+}
+
+QString typeToTypeName(const QString &name)
+{
+    QString ret = name;
+    return ret.replace(QStringLiteral("::"), QStringLiteral("_"));
+}
+
 QString includeGuard(const QString &filename)
 {
     QString guard = filename.toUpper();
@@ -43,14 +58,56 @@ QString formatFunctionSignature(const QList<Tracepoint::Argument> &args)
     });
 }
 
-QString formatParameterList(const QList<Tracepoint::Argument> &args, ParamType type)
+QString formatParameterList(const Provider &provider, const QList<Tracepoint::Argument> &args, const QList<Tracepoint::Field> &fields, ParamType type)
 {
     if (type == LTTNG) {
         QString ret;
 
-        for (const Tracepoint::Argument &arg : args)
-            ret += ", "_L1 + arg.name;
+        for (int i = 0; i < args.size(); i++) {
+            const Tracepoint::Argument &arg = args[i];
+            const Tracepoint::Field &field = fields[i];
+            if (field.backendType == Tracepoint::Field::FlagType)
+                ret += ", trace_convert_"_L1 + typeToTypeName(arg.type) + "("_L1 + arg.name + ")"_L1;
+            else
+                ret += ", "_L1 + arg.name;
+        }
+        return ret;
+    }
+
+    auto findEnumeration = [](const QList<TraceEnum> &enums, const QString &name) {
+        for (const auto &e : enums) {
+            if (e.name == name)
+                return e;
+        }
+        return TraceEnum();
+    };
+
+    if (type == CTF) {
+        QString ret;
 
+        for (int i = 0; i < args.size(); i++) {
+            const Tracepoint::Argument &arg = args[i];
+            const Tracepoint::Field &field = fields[i];
+            if (arg.arrayLen > 1) {
+                ret += ", trace::toByteArrayFromArray("_L1 + arg.name + ", "_L1 + QString::number(arg.arrayLen) + ") "_L1;
+            } else if (field.backendType == Tracepoint::Field::EnumeratedType) {
+                const TraceEnum &e = findEnumeration(provider.enumerations, arg.type);
+                QString integerType;
+                if (e.valueSize == 8)
+                    integerType = QStringLiteral("quint8");
+                else if (e.valueSize == 16)
+                    integerType = QStringLiteral("quint16");
+                else
+                    integerType = QStringLiteral("quint32");
+                ret += ", trace::toByteArrayFromEnum<"_L1 + integerType + ">("_L1 + arg.name + ")"_L1;
+            } else if (field.backendType == Tracepoint::Field::FlagType) {
+                ret += ", trace::toByteArrayFromFlags("_L1 + arg.name + ")"_L1;
+            } else if (field.backendType == Tracepoint::Field::String) {
+                ret += ", trace::toByteArrayFromCString("_L1 + arg.name + ")"_L1;
+            } else {
+                ret += ", "_L1 + arg.name;
+            }
+        }
         return ret;
     }