1 files changed, 88 insertions, 0 deletions
diff --git a/src/qmlcompiler/qqmljsstoragegeneralizer.cpp b/src/qmlcompiler/qqmljsstoragegeneralizer.cpp
new file mode 100644
index 0000000000..2b0938c96f
--- /dev/null
+++ b/src/qmlcompiler/qqmljsstoragegeneralizer.cpp
@@ -0,0 +1,88 @@
+/****************************************************************************
+**
+** Copyright (C) 2021 The Qt Company Ltd.
+** Contact: https://www.qt.io/licensing/
+**
+** This file is part of the tools applications of the Qt Toolkit.
+**
+** $QT_BEGIN_LICENSE:GPL-EXCEPT$
+** Commercial License Usage
+** Licensees holding valid commercial Qt licenses may use this file in
+** accordance with the commercial license agreement provided with the
+** Software or, alternatively, in accordance with the terms contained in
+** a written agreement between you and The Qt Company. For licensing terms
+** and conditions see https://www.qt.io/terms-conditions. For further
+** information use the contact form at https://www.qt.io/contact-us.
+**
+** GNU General Public License Usage
+** Alternatively, this file may be used under the terms of the GNU
+** General Public License version 3 as published by the Free Software
+** Foundation with exceptions as appearing in the file LICENSE.GPL3-EXCEPT
+** included in the packaging of this file. Please review the following
+** information to ensure the GNU General Public License requirements will
+** be met: https://www.gnu.org/licenses/gpl-3.0.html.
+**
+** $QT_END_LICENSE$
+**
+****************************************************************************/
+
+#include "qqmljsstoragegeneralizer_p.h"
+
+QT_BEGIN_NAMESPACE
+
+QQmlJSCompilePass::InstructionAnnotations QQmlJSStorageGeneralizer::run(
+        InstructionAnnotations annotations, Function *function,
+        QQmlJS::DiagnosticMessage *error)
+{
+    m_error = error;
+
+    if (QQmlJSScope::ConstPtr &returnType = function->returnType) {
+        if (QQmlJSScope::ConstPtr stored = m_typeResolver->genericType(
+                    returnType, QQmlJSTypeResolver::ComponentIsGeneric::Yes)) {
+            returnType = stored;
+        } else {
+            setError(QStringLiteral("Cannot store the return type %1.")
+                     .arg(returnType->internalName(), 0));
+            return InstructionAnnotations();
+        }
+    }
+
+    for (QQmlJSScope::ConstPtr &argument : function->argumentTypes) {
+        Q_ASSERT(argument);
+        if (QQmlJSScope::ConstPtr stored = m_typeResolver->genericType(
+                    argument, QQmlJSTypeResolver::ComponentIsGeneric::Yes)) {
+            argument = std::move(stored);
+        } else {
+            setError(QStringLiteral("Cannot store the argument type %1.")
+                     .arg(argument->internalName(), 0));
+            return InstructionAnnotations();
+        }
+    }
+
+    const auto transformRegisters = [&](QHash<int, QQmlJSRegisterContent> &registers, int offset) {
+        for (auto j = registers.begin(), jEnd = registers.end(); j != jEnd; ++j) {
+            const QQmlJSRegisterContent &content = *j;
+            if (QQmlJSScope::ConstPtr specific = content.storedType()) {
+                if (QQmlJSScope::ConstPtr generic = m_typeResolver->genericType(specific)) {
+                    *j = content.storedIn(generic);
+                } else {
+                    setError(QStringLiteral("Cannot store the register type %1.")
+                             .arg(specific->internalName()), offset);
+                    return false;
+                }
+            }
+        }
+        return true;
+    };
+
+    for (auto i = annotations.begin(), iEnd = annotations.end(); i != iEnd; ++i) {
+        if (!transformRegisters(i->registers, i.key()))
+            return InstructionAnnotations();
+        if (!transformRegisters(i->expectedTargetTypesBeforeJump, i.key()))
+            return InstructionAnnotations();
+    }
+
+    return annotations;
+}
+
+QT_END_NAMESPACE