From 20ab16178e45d07b4dc0104a7a98865d18440b62 Mon Sep 17 00:00:00 2001
From: Yong He <yonghe@outlook.com>
Date: Fri, 9 Feb 2024 22:00:15 -0800
Subject: Support link-time constants. (#3564)

* Support link-time constants.

* Fix.

* Fix.
---
 source/slang/slang-mangle.cpp | 42 +++++++++++++++++++++++++++++++-----------
 1 file changed, 31 insertions(+), 11 deletions(-)

(limited to 'source/slang/slang-mangle.cpp')

diff --git a/source/slang/slang-mangle.cpp b/source/slang/slang-mangle.cpp
index 3f701f385..9431533ac 100644
--- a/source/slang/slang-mangle.cpp
+++ b/source/slang/slang-mangle.cpp
@@ -132,7 +132,8 @@ namespace Slang
 
     void emitQualifiedName(
         ManglingContext*    context,
-        DeclRef<Decl>       declRef);
+        DeclRef<Decl>       declRef,
+        bool includeModuleName);
 
     void emitSimpleIntVal(
         ManglingContext*    context,
@@ -211,7 +212,7 @@ namespace Slang
         }
         else if( auto declRefType = dynamicCast<DeclRefType>(type) )
         {
-            emitQualifiedName(context, declRefType->getDeclRef());
+            emitQualifiedName(context, declRefType->getDeclRef(), true);
         }
         else if (auto arrType = dynamicCast<ArrayExpressionType>(type))
         {
@@ -222,7 +223,7 @@ namespace Slang
         else if( auto thisType = dynamicCast<ThisType>(type) )
         {
             emitRaw(context, "t");
-            emitQualifiedName(context, thisType->getInterfaceDeclRef());
+            emitQualifiedName(context, thisType->getInterfaceDeclRef(), true);
         }
         else if (const auto errorType = dynamicCast<ErrorType>(type))
         {
@@ -356,8 +357,26 @@ namespace Slang
 
     void emitQualifiedName(
         ManglingContext*    context,
-        DeclRef<Decl>       declRef)
+        DeclRef<Decl>       declRef,
+        bool includeModuleName)
     {
+        if (!includeModuleName)
+        {
+            if (as<ModuleDecl>(declRef))
+                return;
+        }
+        else
+        {
+            for (auto modifier : declRef.getDecl()->modifiers)
+            {
+                if (as<ExternModifier>(modifier) || as<HLSLExportModifier>(modifier))
+                {
+                    includeModuleName = false;
+                    break;
+                }
+            }
+        }
+
         auto parentDeclRef = declRef.getParent();
         if (as<FileDecl>(parentDeclRef))
             parentDeclRef = parentDeclRef.getParent();
@@ -365,7 +384,7 @@ namespace Slang
         auto parentGenericDeclRef = parentDeclRef.as<GenericDecl>();
         if( parentDeclRef )
         {
-            emitQualifiedName(context, parentDeclRef);
+            emitQualifiedName(context, parentDeclRef, includeModuleName);
         }
 
         // A generic declaration is kind of a pseudo-declaration
@@ -490,7 +509,7 @@ namespace Slang
                     for (auto type : constraint.value)
                     {
                         emitRaw(context, "C");
-                        emitQualifiedName(context, makeDeclRef(constraint.key));
+                        emitQualifiedName(context, makeDeclRef(constraint.key), true);
                             emitType(context, type);
                     }
                 }
@@ -579,7 +598,7 @@ namespace Slang
 
             auto innerDecl = getInner(genericDecl);
 
-            emitQualifiedName(context, makeDeclRef(innerDecl));
+            emitQualifiedName(context, makeDeclRef(innerDecl), true);
             return;
         }
         else if (as<ForwardDerivativeRequirementDecl>(decl))
@@ -592,7 +611,8 @@ namespace Slang
         }
 
         // Now we encode the qualified name of the decl.
-        emitQualifiedName(context, declRef);
+        
+        emitQualifiedName(context, declRef, true);
     }
 
     static String getMangledName(ASTBuilder* astBuilder, DeclRef<Decl> const& declRef)
@@ -625,8 +645,8 @@ namespace Slang
         SLANG_AST_BUILDER_RAII(astBuilder);
         ManglingContext context(astBuilder);
         emitRaw(&context, "_SW");
-        emitQualifiedName(&context, sub);
-        emitQualifiedName(&context, sup);
+        emitQualifiedName(&context, sub, true);
+        emitQualifiedName(&context, sup, true);
         return context.sb.produceString();
     }
 
@@ -643,7 +663,7 @@ namespace Slang
         //
         ManglingContext context(astBuilder);
         emitRaw(&context, "_SW");
-        emitQualifiedName(&context, sub);
+        emitQualifiedName(&context, sub, true);
         emitType(&context, sup);
         return context.sb.produceString();
     }
-- 
cgit v1.2.3