commit 0b827109e3d8dea0f943bc41f2462d24c6e967a1
author Logan Chien <loganchien@google.com> Tue Dec 20 19:46:14 2011 +0800
committer Shih-wei Liao <sliao@google.com> Thu Feb 16 00:29:01 2012 -0800
tree 05e88d1d27e6120713eff4dfb3c33da05c490044
parent 7e6e33d45f96ca36803455f158d02fd1f5c21a1a

Add LLVM function declaration codegen.

Change-Id: I1ad54dd54ea73eba5ecc8ab17479fe161c308b88

src/compiler_llvm/method_compiler.cc

diff --git a/src/compiler_llvm/method_compiler.cc b/src/compiler_llvm/method_compiler.cc
index 767897c..554e5e2 100644
--- a/src/compiler_llvm/method_compiler.cc
+++ b/src/compiler_llvm/method_compiler.cc
@@ -23,6 +23,8 @@
 #include "object.h"
 #include "object_utils.h"
 #include "stl_util.h"
+#include "stringprintf.h"
+#include "utils_llvm.h"
 
 #include <iomanip>
 
@@ -58,6 +60,67 @@
 }
 
 
+void MethodCompiler::CreateFunction() {
+  // LLVM function name
+  std::string func_name(LLVMLongName(method_));
+
+  // Get function type
+  llvm::FunctionType* func_type =
+    GetFunctionType(method_idx_, method_->IsStatic());
+
+  // Create function
+  func_ = llvm::Function::Create(func_type, llvm::Function::ExternalLinkage,
+                                 func_name, module_);
+
+  // Set argument name
+  llvm::Function::arg_iterator arg_iter(func_->arg_begin());
+  llvm::Function::arg_iterator arg_end(func_->arg_end());
+
+  DCHECK_NE(arg_iter, arg_end);
+  arg_iter->setName("method");
+  ++arg_iter;
+
+  if (!method_->IsStatic()) {
+    DCHECK_NE(arg_iter, arg_end);
+    arg_iter->setName("this");
+    ++arg_iter;
+  }
+
+  for (unsigned i = 0; arg_iter != arg_end; ++i, ++arg_iter) {
+    arg_iter->setName(StringPrintf("a%u", i));
+  }
+}
+
+
+llvm::FunctionType* MethodCompiler::GetFunctionType(uint32_t method_idx,
+                                                    bool is_static) {
+  // Get method signature
+  DexFile::MethodId const& method_id = dex_file_->GetMethodId(method_idx);
+
+  int32_t shorty_size;
+  char const* shorty = dex_file_->GetMethodShorty(method_id, &shorty_size);
+  CHECK_GE(shorty_size, 1);
+
+  // Get return type
+  llvm::Type* ret_type = irb_.getJType(shorty[0], kAccurate);
+
+  // Get argument type
+  std::vector<llvm::Type*> args_type;
+
+  args_type.push_back(irb_.getJObjectTy()); // method object pointer
+
+  if (!is_static) {
+    args_type.push_back(irb_.getJType('L', kAccurate)); // "this" object pointer
+  }
+
+  for (int32_t i = 1; i < shorty_size; ++i) {
+    args_type.push_back(irb_.getJType(shorty[i], kAccurate));
+  }
+
+  return llvm::FunctionType::get(ret_type, args_type, false);
+}
+
+
 void MethodCompiler::EmitPrologue() {
   // UNIMPLEMENTED(WARNING);
 }
@@ -75,8 +138,21 @@
 
 
 CompiledMethod *MethodCompiler::Compile() {
-  // UNIMPLEMENTED(WARNING);
-  return new CompiledMethod(insn_set_, NULL);
+  // Code generation
+  CreateFunction();
+
+  EmitPrologue();
+  EmitInstructions();
+
+  // Delete the inferred register category map (won't be used anymore)
+  method_->ResetInferredRegCategoryMap();
+
+  return new CompiledMethod(insn_set_, func_);
+}
+
+
+llvm::Value* MethodCompiler::EmitLoadMethodObjectAddr() {
+  return func_->arg_begin();
 }