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

RVTEST_RV64U        # Define TVM used by program.

// TODO: add extra "code" argument and "testdata" argument,
// replace "addw" with "code"
// TODO: move SV_ELWIDTH_TEST to sv_test_macros.h
// TODO: probably remove testing of x15 and x16 (or pass in as extra args?)

#define SV_ELWIDTH_TEST( wid1, wid2, wid3, expect1, expect2, expect3 ) \
                                                        \
        SV_LDD_DATA( x12, testdata   , 0);               \
        SV_LDD_DATA( x13, testdata+8 , 0);               \
        SV_LDD_DATA( x14, testdata+16, 0);               \
        SV_LDD_DATA( x15, testdata+24, 0);               \
        SV_LDD_DATA( x16, testdata+32, 0);               \
        SV_LDD_DATA( x17, testdata+40, 0);               \
                                                        \
        li x28, 0xa5a5a5a5a5a5a5a5;                                      \
        li x29, 0xa5a5a5a5a5a5a5a5;                                      \
        li x30, 0xa5a5a5a5a5a5a5a5;                                      \
                                                        \
        SET_SV_MVL( 3);                                  \
        SET_SV_3CSRS( SV_REG_CSR( 1, 15, wid1, 15, 1),       \
                      SV_REG_CSR( 1, 12, wid2, 12, 1),        \
                      SV_REG_CSR( 1, 28, wid3, 28, 1));       \
        SET_SV_VL( 3);                                   \
                                                        \
        addw   x28, x15, x12;                              \
                                                        \
        CLR_SV_CSRS();                                  \
        SET_SV_VL( 1);                                   \
        SET_SV_MVL( 1);                                  \
                                                        \
        TEST_SV_IMM( x28, expect1 );                         \
        TEST_SV_IMM( x29, expect2 );                       \
        TEST_SV_IMM( x30, expect3 );                       \
        TEST_SV_IMM( x15, 0x0000005242322212);                       \
        TEST_SV_IMM( x16, 0x0000005141312111);


# 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.

        # TODO: add "addw" argument, add testdata argument
        SV_ELWIDTH_TEST(  0, 0, 0,
                0xffffffff8b6bab8b,  0xffffffff88684828,  0x0000000000000000 )
        SV_ELWIDTH_TEST(  0, 0, 3,
                0x886848288b6bab8b,  0xa5a5a5a500000000,  0xa5a5a5a5a5a5a5a5 )
        SV_ELWIDTH_TEST(  1, 1, 0,
                0xffffffffffffff8b,  0xffffffffffffffab,  0x000000000000006b )
        SV_ELWIDTH_TEST(  1, 1, 3,
                0xffffffabffffff8b,  0xa5a5a5a50000006b,  0xa5a5a5a5a5a5a5a5 )
        SV_ELWIDTH_TEST(  1, 1, 2,
                0xa5a5006bffabff8b,  0xa5a5a5a5a5a5a5a5,  0xa5a5a5a5a5a5a5a5 )
        SV_ELWIDTH_TEST(  1, 1, 1,
                0xa5a5a5a5a56bab8b,  0xa5a5a5a5a5a5a5a5,  0xa5a5a5a5a5a5a5a5 )

        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 0x0000005949398979
        .dword 0x0000005747372717
        .dword 0x0000000000000000
        .dword 0x0000005242322212
        .dword 0x0000005141312111
        .dword 0x0000000000000000

# 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.