@@ -7,10 +7,52 @@ llvm_value_wrap(Value* v) { } VALUE -llvm_value_get_name(VALUE self) { - Value *v = LLVM_VAL(self); - std::string name = v->getName(); - return rb_str_new2(name.c_str()); +llvm_value_name(VALUE self) { + Value *v; + Data_Get_Struct(self, Value, v); + + if(v->hasName()) { + const char *name = v->getNameStart(); + int len = v->getNameLen(); + return rb_str_new(name, len); + } else { + return Qnil; + } +} + +VALUE +llvm_value_set_name(VALUE self, VALUE rname) { + Value *v; + Data_Get_Struct(self, Value, v); + v->setName(RSTRING_PTR(rname), RSTRING_LEN(rname)); + return rname; +} + +VALUE +llvm_value_num_uses(VALUE self) { + Value *v; + Data_Get_Struct(self, Value, v); + return INT2FIX(v->getNumUses()); +} + +VALUE +llvm_value_used_in_basic_block(VALUE self, VALUE rbb) { + Value *v; + Data_Get_Struct(self, Value, v); + + BasicBlock *bb; + Data_Get_Struct(rbb, BasicBlock, bb); + + return v->isUsedInBasicBlock(bb) ? Qtrue : Qfalse; +} + +VALUE +llvm_value_replace_all_uses_with(VALUE self, VALUE rv2) { + Value *v1, *v2; + Data_Get_Struct(self, Value, v1); + Data_Get_Struct(rv2, Value, v2); + v1->replaceAllUsesWith(v2); + return rv2; } VALUE ext/llvm_value.cpp @@ -106,8 +106,11 @@ VALUE llvm_value_get_float_constant(VALUE); VALUE llvm_value_get_double_constant(VALUE); VALUE llvm_value_get_immediate_constant(VALUE); VALUE llvm_value_get_struct_constant(int, VALUE*, VALUE); -VALUE llvm_value_get_name(VALUE); - +VALUE llvm_value_name(VALUE); +VALUE llvm_value_set_name(VALUE, VALUE); +VALUE llvm_value_num_uses(VALUE); +VALUE llvm_value_used_in_basic_block(VALUE, VALUE); +VALUE llvm_value_replace_all_uses_with(VALUE, VALUE); VALUE llvm_phi_add_incoming(VALUE, VALUE, VALUE); @@ -158,7 +161,11 @@ void Init_llvmruby() { rb_define_module_function(cLLVMValue, "get_double_constant", llvm_value_get_double_constant, 1); rb_define_module_function(cLLVMValue, "get_immediate_constant", llvm_value_get_immediate_constant, 1); rb_define_module_function(cLLVMValue, "get_struct_constant", llvm_value_get_struct_constant, -1); - rb_define_method(cLLVMValue, "get_name", llvm_value_get_name, 0); + rb_define_method(cLLVMValue, "name", llvm_value_name, 0); + rb_define_method(cLLVMValue, "name=", llvm_value_set_name, 1); + rb_define_method(cLLVMValue, "num_uses", llvm_value_num_uses, 0); + rb_define_method(cLLVMValue, "used_in_basic_block?", llvm_value_used_in_basic_block, 1); + rb_define_method(cLLVMValue, "replace_all_uses_with", llvm_value_replace_all_uses_with, 1); init_instructions(); ext/llvmruby.c @@ -57,9 +57,24 @@ class BasicBlockTests < Test::Unit::TestCase expected = { 'entry' => 2, 'bb5' => 6, 'bb11' => 2, 'bb15.split' => 2, 'bb19' => 4, 'bb26' => 2 } res = Hash.new bbs.each { |b| - res[b.get_name] = b.size + res[b.name] = b.size } assert_equal(expected,res) end + def test_manipulate_values + m = LLVM::Module.read_assembly(@assembly_gcd) + gcd = m.get_function("gcd") + + bbs = gcd.get_basic_block_list + bb = bbs.first + assert_equal('entry', bb.name) + bb.name = 'first_block' + assert_equal('first_block', bb.name) + assert_equal(3, bb.num_uses) + bb2 = bbs.find {|x| x.name == 'bb19'} + assert(bb.used_in_basic_block?(bb2)) + bb3 = bbs.find {|x| x.name = 'bb26'} + assert(!bb.used_in_basic_block?(bb3)) + end end test/test_basic_block.rb e0eb33ec85b71d3761cb648e1fdc89a724b5bb9e Tom Bagby tomatobagby@gmail.com http://github.com/tombagby/llvmruby/commit/d5243b0f46107e05471f884a95fe723ec6aaee23 d5243b0f46107e05471f884a95fe723ec6aaee23 2008-10-19T16:20:12-07:00 2008-10-19T16:20:12-07:00 fix value name handling, more value methods 78be5acd5b09cc8659fb2f01a64032e50ee4015b Tom Bagby tomatobagby@gmail.com