circuitpython/tests/micropython/viper_addr.py

44 lines
774 B
Python
Raw Normal View History

# test passing addresses to viper
2021-03-15 09:57:36 -04:00
@micropython.viper
2021-03-15 09:57:36 -04:00
def get_addr(x: ptr) -> ptr:
return x
2021-03-15 09:57:36 -04:00
@micropython.viper
2021-03-15 09:57:36 -04:00
def memset(dest: ptr8, c: int, n: int):
for i in range(n):
dest[i] = c
2021-03-15 09:57:36 -04:00
@micropython.viper
def memsum(src: ptr8, n: int) -> int:
s = 0
for i in range(n):
s += src[i]
return s
# create array and get its address
2021-03-15 09:57:36 -04:00
ar = bytearray("0000")
addr = get_addr(ar)
print(type(ar))
print(type(addr))
print(ar)
# pass array as an object
2021-03-15 09:57:36 -04:00
memset(ar, ord("1"), len(ar))
print(ar)
# pass direct pointer to array buffer
2021-03-15 09:57:36 -04:00
memset(addr, ord("2"), len(ar))
print(ar)
# pass direct pointer to array buffer, with offset
2021-03-15 09:57:36 -04:00
memset(addr + 2, ord("3"), len(ar) - 2)
print(ar)
# pass a read-only bytes object in
print(memsum(b"\x01\x02\x03\x04", 4))