Add swift stub classref get ir gen

ldc-developers · Nov 20, 2024 · 9cbd156 · 9cbd156
1 parent 3679436
commit 9cbd156
Show file tree

Hide file tree

Showing 3 changed files with 18 additions and 3 deletions.
diff --git a/gen/objcgen.cpp b/gen/objcgen.cpp
@@ -15,6 +15,7 @@
 #include "dmd/declaration.h"
 #include "dmd/identifier.h"
 #include "gen/irstate.h"
+#include "gen/runtime.h"
 #include "ir/irfunction.h"
 
 bool objc_isSupported(const llvm::Triple &triple) {
@@ -394,7 +395,14 @@ LLGlobalVariable *ObjCState::getClassRoSymbol(const ClassDeclaration& cd, bool m
   return var;
 }
 
-LLGlobalVariable *ObjCState::getClassReference(const ClassDeclaration& cd) {
+LLValue *ObjCState::getSwiftStubClassReference(const ClassDeclaration& cd) {
+  auto classref = getClassReference(cd);
+  auto toClassRefFunc = getRuntimeFunction(cd.loc, module, "objc_loadClassRef");
+  auto retv = gIR->CreateCallOrInvoke(toClassRefFunc, classref, "");
+  return retv;
+}
+
+LLConstant *ObjCState::getClassReference(const ClassDeclaration& cd) {
   llvm::StringRef name(cd.objc.identifier->toChars());
   auto it = classReferenceTable.find(name);
   if (it != classReferenceTable.end()) {

diff --git a/gen/objcgen.h b/gen/objcgen.h
@@ -66,7 +66,8 @@ class ObjCState {
   // Classes
   LLGlobalVariable *getClassSymbol(const ClassDeclaration& cd, bool meta);
   LLGlobalVariable *getClassRoSymbol(const ClassDeclaration& cd, bool meta);
-  LLGlobalVariable *getClassReference(const ClassDeclaration& cd);
+  LLConstant *getClassReference(const ClassDeclaration& cd);
+  LLValue *getSwiftStubClassReference(const ClassDeclaration& cd);
 
   // Categories
 

diff --git a/gen/toir.cpp b/gen/toir.cpp
@@ -334,14 +334,20 @@ class ToElemVisitor : public Visitor {
                          e->type->toChars());
     LOG_SCOPE;
 
-    LLGlobalVariable *decl; 
+    // Protocols
     if (auto iface = e->classDeclaration->isInterfaceDeclaration()) {
 
       auto loaded = DtoLoad(DtoType(e->type), gIR->objc.getProtocolReference(*iface));
       result = new DImValue(e->type, loaded);
       return;
     }
 
+    // Swift stub classes
+    if (e->classDeclaration->objc.isSwiftStub) {
+      result = new DImValue(e->type, gIR->objc.getSwiftStubClassReference(*e->classDeclaration));
+      return;
+    }
+
     auto loaded = DtoLoad(DtoType(e->type), gIR->objc.getClassReference(*e->classDeclaration));
     result = new DImValue(e->type, loaded);
   }