jyapayne/llvmpy
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

insert_value, extract_value can be multiple dimensional

Browse source
This commit is contained in:
Siu Kwan Lam 2013-12-16 12:43:27 -06:00
commit 091a393d1d
2 changed files with 21 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

9
llvm/core.py
View file

@ -2337,15 +2337,20 @@ class Builder(llvm.Wrapper):
# misc # misc
def extract_value(self, retval, idx, name=""): def extract_value(self, retval, idx, name=""):
return _make_value(self._ptr.CreateExtractValue(retval._ptr, [idx], name)) if not isinstance(idx, (tuple, list)):
idx = [idx]
return _make_value(self._ptr.CreateExtractValue(retval._ptr, idx,
name))
# obsolete synonym for extract_value # obsolete synonym for extract_value
getresult = extract_value getresult = extract_value
def insert_value(self, retval, rhs, idx, name=""): def insert_value(self, retval, rhs, idx, name=""):
if not isinstance(idx, (tuple, list)):
idx = [idx]
return _make_value(self._ptr.CreateInsertValue(retval._ptr, return _make_value(self._ptr.CreateInsertValue(retval._ptr,
rhs._ptr, rhs._ptr,
[idx], idx,
name)) name))
def phi(self, ty, name=""): def phi(self, ty, name=""):

15
llvm/tests/test_struct.py
View file

@ -1,5 +1,5 @@
import unittest import unittest
from llvm.core import Type from llvm.core import Type, Module, Builder, Constant
from .support import TestCase, tests from .support import TestCase, tests
class TestStruct(TestCase): class TestStruct(TestCase):
@ -8,6 +8,19 @@ class TestStruct(TestCase):
tb = Type.struct([Type.int(32), Type.float()]) tb = Type.struct([Type.int(32), Type.float()])
self.assertTrue(ta.is_layout_identical(tb)) self.assertTrue(ta.is_layout_identical(tb))
def test_struct_extract_value_2d(self):
ta = Type.struct([Type.int(32), Type.float()])
tb = Type.struct([ta, Type.float()])
m = Module.new('')
f = m.add_function(Type.function(Type.void(), []), "foo")
b = Builder.new(f.append_basic_block(''))
v = Constant.undef(tb)
ins = b.insert_value(v, Constant.real(Type.float(), 1.234), [0, 1])
ext = b.extract_value(ins, [0, 1])
b.ret_void()
m.verify()
self.assertEqual(str(ext), 'float 0x3FF3BE76C0000000')
tests.append(TestStruct) tests.append(TestStruct)
if __name__ == '__main__': if __name__ == '__main__':