Extending Ptr, Array, and UInt classes

[marioroy]

The casting for simple things makes the language less fun e.g. buf[int(index)], arr[int(index)]. For example, using a 32-bit index value to enhance performance on the GPU.

@extend
class Ptr:
    @pure
    @derives
    @llvm
    def __getitem__(self, index: u32) -> T:
        %0 = getelementptr {=T}, ptr %self, i32 %index
        %1 = load {=T}, ptr %0
        ret {=T} %1

    @pure
    @derives
    @llvm
    def __getitem__(self, index: u64) -> T:
        %0 = getelementptr {=T}, ptr %self, i64 %index
        %1 = load {=T}, ptr %0
        ret {=T} %1

    @self_captures
    @llvm
    def __setitem__(self, index: u32, what: T) -> None:
        %0 = getelementptr {=T}, ptr %self, i32 %index
        store {=T} %what, ptr %0
        ret {} {}

    @self_captures
    @llvm
    def __setitem__(self, index: u64, what: T) -> None:
        %0 = getelementptr {=T}, ptr %self, i64 %index
        store {=T} %what, ptr %0
        ret {} {}

@extend
class Array:
    def __getitem__(self, index: u32) -> T:
        return self.ptr[index]

    def __getitem__(self, index: u64) -> T:
        return self.ptr[index]

    def __setitem__(self, index: u32, what: T):
        self.ptr[index] = what

    def __setitem__(self, index: u64, what: T):
        self.ptr[index] = what

buf = Ptr[u8](3)
buf[0] = u8(1)
buf[1] = u8(2)
buf[2] = u8(3)

arr = Array[u8](3)
arr[0] = u8(1)
arr[1] = u8(2)
arr[2] = u8(3)

print( buf[0] )
print( buf[u32(1)] )
print( buf[u64(2)] )

print( arr[0] )
print( arr[u32(1)] )
print( arr[u64(2)] )

Should Codon at least know how to handle this simple use-case?

[marioroy]

For Uint, I wrote a helper file to auto-cast "int" to "Uint" automatically. All the casting made the code less readable.

marioroy/mce-sandbox@62e0c83