structs_14.py

from lpython import i8, dataclass, i32, f32, c32, f64, i16, i64, c64, ccallable, packed
from numpy import empty, int8, int16, int32, int64, float32, complex64, complex128, float64
from copy import deepcopy

@dataclass
class buffer_struct:
    buffer: i8[32]
    buffer1: i32[32]
    buffer2: f32[32]
    buffer3: c32[32]
    buffer4: f64[32]
    buffer5: i16[32]
    buffer6: i64[32]
    buffer7: c64[32]

@ccallable
@packed
@dataclass
class buffer_struct_clink:
    buffer: i8[32]
    buffer1: i32[32]
    buffer2: f32[32]
    buffer3: c32[32]
    buffer4: f64[32]
    buffer5: i16[32]
    buffer6: i64[32]
    buffer7: c64[32]

def f():
    i: i32
    buffer_var: i8[32] = empty(32, dtype=int8)
    buffer1_var: i32[32] = empty(32, dtype=int32)
    buffer2_var: f32[32] = empty(32, dtype=float32)
    buffer3_var: c32[32] = empty(32, dtype=complex64)
    buffer4_var: f64[32] = empty(32, dtype=float64)
    buffer5_var: i16[32] = empty(32, dtype=int16)
    buffer6_var: i64[32] = empty(32, dtype=int64)
    buffer7_var: c64[32] = empty(32, dtype=complex128)
    buffer_: buffer_struct = buffer_struct(deepcopy(buffer_var), deepcopy(buffer1_var),
                                           deepcopy(buffer2_var), deepcopy(buffer3_var),
                                           deepcopy(buffer4_var), deepcopy(buffer5_var),
                                           deepcopy(buffer6_var), deepcopy(buffer7_var))
    buffer_clink_: buffer_struct_clink = buffer_struct_clink(deepcopy(buffer_var), deepcopy(buffer1_var),
                                                             deepcopy(buffer2_var), deepcopy(buffer3_var),
                                                             deepcopy(buffer4_var), deepcopy(buffer5_var),
                                                             deepcopy(buffer6_var), deepcopy(buffer7_var))
    print(buffer_.buffer[15])
    print(buffer_.buffer1[15])
    print(buffer_.buffer2[15])
    print(buffer_.buffer3[15])
    print(buffer_.buffer4[15])
    print(buffer_.buffer5[15])
    print(buffer_.buffer6[15])
    print(buffer_.buffer7[15])
    print(buffer_clink_.buffer[15])
    print(buffer_clink_.buffer1[15])
    print(buffer_clink_.buffer2[15])
    print(buffer_clink_.buffer3[15])
    print(buffer_clink_.buffer4[15])
    print(buffer_clink_.buffer5[15])
    print(buffer_clink_.buffer6[15])
    print(buffer_clink_.buffer7[15])

    for i in range(32):
        buffer_.buffer[i] = i8(i + 1)
        buffer_clink_.buffer[i] = i8(i + 2)
        buffer_.buffer1[i] = i32(i + 3)
        buffer_clink_.buffer1[i] = i32(i + 4)
        buffer_.buffer2[i] = f32(i + 5)
        buffer_clink_.buffer2[i] = f32(i + 6)
        buffer_.buffer3[i] = c32(i + 7)
        # buffer_clink_.buffer3 is a ctypes.Array
        # of type c_float_complex (a ctypes.Structure
        # defined in lpython.py) and c32(i + 8) is a
        # Python object. Python doesn't allow assigning
        # a Python object to ctypes.Structure. Hence,
        # the following line is commented out.
        # buffer_clink_.buffer3[i] = c32(i + 8)
        buffer_.buffer4[i] = f64(i + 9)
        buffer_clink_.buffer4[i] = f64(i + 10)
        buffer_.buffer5[i] = i16(i + 11)
        buffer_clink_.buffer5[i] = i16(i + 12)
        buffer_.buffer6[i] = i64(i + 13)
        buffer_clink_.buffer6[i] = i64(i + 14)
        buffer_.buffer7[i] = c64(i + 15)
        # buffer_clink_.buffer7[i] = c64(i + 16)

    for i in range(32):
        print(i, buffer_.buffer[i], buffer_clink_.buffer[i])
        print(i, buffer_clink_.buffer1[i], buffer_.buffer1[i])
        print(i, buffer_clink_.buffer2[i], buffer_.buffer2[i])
        print(i, buffer_clink_.buffer3[i], buffer_.buffer3[i])
        print(i, buffer_clink_.buffer4[i], buffer_.buffer4[i])
        print(i, buffer_clink_.buffer5[i], buffer_.buffer5[i])
        print(i, buffer_clink_.buffer6[i], buffer_.buffer6[i])
        print(i, buffer_clink_.buffer7[i], buffer_.buffer7[i])
        assert buffer_.buffer[i] == i8(i + 1)
        assert buffer_clink_.buffer[i] == i8(i + 2)
        assert buffer_clink_.buffer[i] - buffer_.buffer[i] == i8(1)
        assert buffer_clink_.buffer1[i] - buffer_.buffer1[i] == i32(1)
        assert buffer_clink_.buffer2[i] - buffer_.buffer2[i] == f32(1)
        # assert buffer_clink_.buffer3[i] - buffer_.buffer3[i] == c32(1)
        assert buffer_clink_.buffer4[i] - buffer_.buffer4[i] == f64(1)
        assert buffer_clink_.buffer5[i] - buffer_.buffer5[i] == i16(1)
        assert buffer_clink_.buffer6[i] - buffer_.buffer6[i] == i64(1)
        # assert buffer_clink_.buffer7[i] - buffer_.buffer7[i] == c64(1)

f()