#include "riscv_test.h"
#include "sv_test_macros.h"

RVTEST_RV64U        # Define TVM used by program.

#define SV_PREDICATION_TEST( pred, inv, zero, expect1, expect2 ) \
                                                        \
        SV_LDD_DATA( x2, testdata   , 0);               \
        SV_LDD_DATA( x3, testdata+8 , 0);               \
        SV_LDD_DATA( x4, testdata+16, 0);               \
        SV_LDD_DATA( x5, testdata+24, 0);               \
                                                        \
        li      x6, pred;                               \
                                                        \
        SET_SV_MVL( 2);                                  \
        SET_SV_CSR( 1, 3, SV_W_8BIT, 3, 1);             \
        SET_SV_PRED_CSR( 1, 3, zero, inv, 6, 0);         \
        SET_SV_VL( 2);                                   \
        SET_SV_SUBVL( 2);                                \
                                                        \
        addi    x3, x3, 1;                              \
                                                        \
        SET_SV_SUBVL( 1);                                \
        CLR_SV_CSRS();                                  \
        SET_SV_VL( 1);                                   \
        SET_SV_MVL( 1);                                  \
                                                        \
        TEST_SV_IMM( x2, 1001);                         \
        TEST_SV_IMM( x3, expect1);                       \
        TEST_SV_IMM( x4, expect2);                       \
        TEST_SV_IMM( x5, 1002);


# SV test: vector-vector add
#
# sets up x3 and x4 with data, sets VL to 2, and carries out
# an "add 1 to x3".  which actually means "add 1 to x3 *AND* add 1 to x4"

# Test code region.
RVTEST_CODE_BEGIN   # Start of test code.

        # no zeroing, no inversion
        SV_PREDICATION_TEST( 0x1, 0, 0, 0x41424445, 0 )
        SV_PREDICATION_TEST( 0x2, 0, 0, 0x42434344, 0 )
        SV_PREDICATION_TEST( 0x3, 0, 0, 0x42434445, 0 )
        SV_PREDICATION_TEST( 0x0, 0, 0, 0x41424344, 0 )

        # zeroing, no inversion
        SV_PREDICATION_TEST( 0x1, 0, 1, 0x00004445, 0 )
        SV_PREDICATION_TEST( 0x2, 0, 1, 0x42430000, 0 )
        SV_PREDICATION_TEST( 0x3, 0, 1, 0x42434445, 0 )
        SV_PREDICATION_TEST( 0x0, 0, 1, 0x00000000, 0 )

        # no zeroing, inversion
        SV_PREDICATION_TEST( 0x1, 1, 0, 0x42434344, 0 )
        SV_PREDICATION_TEST( 0x2, 1, 0, 0x41424445, 0 )
        SV_PREDICATION_TEST( 0x3, 1, 0, 0x41424344, 0 )
        SV_PREDICATION_TEST( 0x0, 1, 0, 0x42434445, 0 )

        # zeroing, inversion
        SV_PREDICATION_TEST( 0x1, 1, 1, 0x42430000, 0 )
        SV_PREDICATION_TEST( 0x2, 1, 1, 0x00004445, 0 )
        SV_PREDICATION_TEST( 0x3, 1, 1, 0x00000000, 0 )
        SV_PREDICATION_TEST( 0x0, 1, 1, 0x42434445, 0 )

        RVTEST_PASS           # Signal success.
fail:
        RVTEST_FAIL
RVTEST_CODE_END     # End of test code.

# Input data section.
# This section is optional, and this data is NOT saved in the output.
.data
        .align 3
testdata:
        .dword 1001
        .dword 0x41424344
        .dword 0x00000000
        .dword 1002

# Output data section.
RVTEST_DATA_BEGIN   # Start of test output data region.
        .align 3
result:
        .dword -1
        .dword -1
        .dword -1
RVTEST_DATA_END     # End of test output data region.