commit acb4977dd3b6a678bfbdb80781bfe3c617be3f24
author Fariborz Jahanian <fjahanian@apple.com> Mon Dec 03 21:26:48 2007 +0000
committer Fariborz Jahanian <fjahanian@apple.com> Mon Dec 03 21:26:48 2007 +0000
tree 99bce892b35f28e60fc03d539e33abd35328b0ed
parent 80a6a5abbbf0b7c45f535d4e0b0d00f4c3f320ea

Support for rewriteing of messaging of methods which return float.


git-svn-id: https://llvm.org/svn/llvm-project/cfe/trunk@44548 91177308-0d34-0410-b5e6-96231b3b80d8

Driver/RewriteTest.cpp

diff --git a/Driver/RewriteTest.cpp b/Driver/RewriteTest.cpp
index 88ac5a4..b17f32b 100644
--- a/Driver/RewriteTest.cpp
+++ b/Driver/RewriteTest.cpp
@@ -45,6 +45,7 @@
     FunctionDecl *MsgSendSuperFunctionDecl;
     FunctionDecl *MsgSendStretFunctionDecl;
     FunctionDecl *MsgSendSuperStretFunctionDecl;
+    FunctionDecl *MsgSendFpretFunctionDecl;
     FunctionDecl *GetClassFunctionDecl;
     FunctionDecl *SelGetUidFunctionDecl;
     FunctionDecl *CFStringFunctionDecl;
@@ -66,6 +67,7 @@
       MsgSendSuperFunctionDecl = 0;
       MsgSendStretFunctionDecl = 0;
       MsgSendSuperStretFunctionDecl = 0;
+      MsgSendFpretFunctionDecl = 0;
       GetClassFunctionDecl = 0;
       SelGetUidFunctionDecl = 0;
       CFStringFunctionDecl = 0;
@@ -93,6 +95,8 @@
                       "(struct objc_object *, struct objc_selector *, ...);\n"
                       "extern struct objc_object *objc_msgSendSuper_stret"
                       "(struct objc_super *, struct objc_selector *, ...);\n"
+                      "extern struct objc_object *objc_msgSend_fpret"
+                      "(struct objc_object *, struct objc_selector *, ...);\n"
                       "extern struct objc_object *objc_getClass"
                       "(const char *);\n"
                       "extern void objc_exception_throw(struct objc_object *);\n"
@@ -149,6 +153,7 @@
     void SynthMsgSendFunctionDecl();
     void SynthMsgSendSuperFunctionDecl();
     void SynthMsgSendStretFunctionDecl();
+    void SynthMsgSendFpretFunctionDecl();
     void SynthMsgSendSuperStretFunctionDecl();
     void SynthGetClassFunctionDecl();
     void SynthCFStringFunctionDecl();
@@ -1097,6 +1102,24 @@
                                               FunctionDecl::Extern, false, 0);
 }
 
+// SynthMsgSendFpretFunctionDecl - id objc_msgSend_fpret(id self, SEL op, ...);
+void RewriteTest::SynthMsgSendFpretFunctionDecl() {
+  IdentifierInfo *msgSendIdent = &Context->Idents.get("objc_msgSend_fpret");
+  llvm::SmallVector<QualType, 16> ArgTys;
+  QualType argT = Context->getObjcIdType();
+  assert(!argT.isNull() && "Can't find 'id' type");
+  ArgTys.push_back(argT);
+  argT = Context->getObjcSelType();
+  assert(!argT.isNull() && "Can't find 'SEL' type");
+  ArgTys.push_back(argT);
+  QualType msgSendType = Context->getFunctionType(Context->getObjcIdType(),
+                                                  &ArgTys[0], ArgTys.size(),
+                                                  true /*isVariadic*/);
+  MsgSendFpretFunctionDecl = new FunctionDecl(SourceLocation(), 
+                                              msgSendIdent, msgSendType,
+                                              FunctionDecl::Extern, false, 0);
+}
+
 // SynthGetClassFunctionDecl - id objc_getClass(const char *name);
 void RewriteTest::SynthGetClassFunctionDecl() {
   IdentifierInfo *getClassIdent = &Context->Idents.get("objc_getClass");
@@ -1235,6 +1258,8 @@
     SynthMsgSendStretFunctionDecl();
   if (!MsgSendSuperStretFunctionDecl)
     SynthMsgSendSuperStretFunctionDecl();
+  if (!MsgSendFpretFunctionDecl)
+    SynthMsgSendFpretFunctionDecl();
   if (!GetClassFunctionDecl)
     SynthGetClassFunctionDecl();
   
@@ -1247,6 +1272,8 @@
     if (resultType.getCanonicalType()->isStructureType() 
         || resultType.getCanonicalType()->isUnionType())
       MsgSendStretFlavor = MsgSendStretFunctionDecl;
+    else if (resultType.getCanonicalType()->isRealFloatingType())
+      MsgSendFlavor = MsgSendFpretFunctionDecl;
   }
   
   // Synthesize a call to objc_msgSend().