import os

# toolchains options
ARCH        ='arm'
CPU         ='cortex-a'
CROSS_TOOL  ='gcc'

if os.getenv('RTT_ROOT'):
    RTT_ROOT = os.getenv('RTT_ROOT')
else:
    RTT_ROOT = r'../../..'

if os.getenv('RTT_CC'):
    CROSS_TOOL = os.getenv('RTT_CC')

PLATFORM    = 'gcc'
EXEC_PATH   = RTT_ROOT + '/../prebuilts/gcc/linux-x86/arm/gcc-arm-none-eabi-10-2020-q4-major-x86_64-linux/bin/'

BUILD = 'release'

if os.getenv('RTT_EXEC_PATH'):
    EXEC_PATH = os.getenv('RTT_EXEC_PATH')

if PLATFORM == 'gcc':
    # toolchains
    # PREFIX = 'arm-none-eabi-'
    PREFIX = 'arm-none-eabi-'
    CC      = PREFIX + 'gcc'
    CXX     = PREFIX + 'g++'
    AS      = PREFIX + 'gcc'
    AR      = PREFIX + 'ar'
    LINK    = PREFIX + 'gcc'
    TARGET_EXT = 'elf'
    SIZE    = PREFIX + 'size'
    OBJDUMP = PREFIX + 'objdump'
    OBJCPY  = PREFIX + 'objcopy'

    DEVICE = ' -mcpu=cortex-a35+crypto -mfloat-abi=hard -mfpu=neon-vfpv4 -marm -ffast-math'
    CFLAGS = DEVICE + ' -std=gnu99 -Wall -g -Wno-stringop-truncation'
    AFLAGS = DEVICE + ' -c -x assembler-with-cpp -D__ASSEMBLY__'
    LINK_SCRIPT = 'gcc_arm.ld.S'
    LFLAGS  = DEVICE + ' -lm -lgcc -lc' + ' -nostartfiles -Wl,--gc-sections,-Map=rtthread.map,-cref,-u,system_vectors -T gcc_arm.ld'
    CPATH   = ''
    LPATH   = ''

    if BUILD == 'debug':
        CFLAGS += ' -O0 -gdwarf-2'
        AFLAGS += ' -gdwarf-2'
    else:
        CFLAGS += ' -O2'

    if os.getenv('RTT_SRAM_BASE'):
        SRAM_BASE = os.getenv('RTT_SRAM_BASE')
    else:
        SRAM_BASE = 0xfff88000

    if os.getenv('RTT_SRAM_SIZE'):
        SRAM_SIZE = os.getenv('RTT_SRAM_SIZE')
    else:
        SRAM_SIZE = 0x00008000

    if os.getenv('RTT_PRMEM_BASE'):
        PRMEM_BASE = os.getenv('RTT_PRMEM_BASE')
    else:
        PRMEM_BASE = 0x02600000

    if os.getenv('RTT_PRMEM_SIZE'):
        PRMEM_SIZE = os.getenv('RTT_PRMEM_SIZE')
    else:
        PRMEM_SIZE = 0x00900000

    if os.getenv('RTT_SHMEM_BASE'):
        SHMEM_BASE = os.getenv('RTT_SHMEM_BASE')
    else:
        SHMEM_BASE = 0x02f00000

    if os.getenv('RTT_SHMEM_SIZE'):
        SHMEM_SIZE = os.getenv('RTT_SHMEM_SIZE')
    else:
        SHMEM_SIZE = 0x00100000

    if os.getenv('RTT_SHRPMSG_SIZE'):
        SHRPMSG_SIZE = os.getenv('RTT_SHRPMSG_SIZE')
    else:
        SHRPMSG_SIZE = 0x00080000

    if os.getenv('LINUX_RPMSG_BASE'):
        LINUX_RPMSG_BASE = os.getenv('LINUX_RPMSG_BASE')
    else:
        LINUX_RPMSG_BASE = 0x07c00000

    if os.getenv('LINUX_RPMSG_SIZE'):
        LINUX_RPMSG_SIZE = os.getenv('LINUX_RPMSG_SIZE')
    else:
        LINUX_RPMSG_SIZE = 0x00500000

    if os.getenv('RTT_SHRAMFS_SIZE'):
        SHRAMFS_SIZE = os.getenv('RTT_SHRAMFS_SIZE')
    else:
        SHRAMFS_SIZE = 0x00020000

    if os.getenv('RTT_SHLOG0_SIZE'):
        SHLOG0_SIZE = os.getenv('RTT_SHLOG0_SIZE')
    else:
        SHLOG0_SIZE = 0x00001000

    if os.getenv('RTT_SHLOG1_SIZE'):
        SHLOG1_SIZE = os.getenv('RTT_SHLOG1_SIZE')
    else:
        SHLOG1_SIZE = 0x00001000

    if os.getenv('RTT_SHLOG2_SIZE'):
        SHLOG2_SIZE = os.getenv('RTT_SHLOG2_SIZE')
    else:
        SHLOG2_SIZE = 0x00001000

    if os.getenv('RTT_SHLOG3_SIZE'):
        SHLOG3_SIZE = os.getenv('RTT_SHLOG3_SIZE')
    else:
        SHLOG3_SIZE = 0x00001000

    if os.getenv('ROOT_PART_OFFSET'):
        CFLAGS += ' -DRT_ROOT_PART_OFFSET={a}'.format(a=os.getenv('ROOT_PART_OFFSET'))

    if os.getenv('ROOT_PART_SIZE'):
        CFLAGS += ' -DRT_ROOT_PART_SIZE={a}'.format(a=os.getenv('ROOT_PART_SIZE'))

    # Define default IRQ route CPU
    if os.getenv('AMP_KERNEL_ENABLE'):
        CFLAGS += ' -DDEFAULT_IRQ_CPU={a}'.format(a=0)
        CFLAGS += ' -DAMP_LINUX_ENABLE'
    else:
        CFLAGS += ' -DDEFAULT_IRQ_CPU={a}'.format(a=1)

    # Define the primary CPU
    if os.getenv('AMP_PRIMARY_CORE'):
        AMP_PRIMARY_CPU = os.getenv('AMP_PRIMARY_CORE')
    else:
        AMP_PRIMARY_CPU = '1'

    CUR_CPU = os.getenv('CUR_CPU')
    if CUR_CPU == AMP_PRIMARY_CPU:
        CFLAGS += ' -DPRIMARY_CPU'

    if CUR_CPU == '1':
        CFLAGS += ' -DCPU1'
    if CUR_CPU == '2':
        CFLAGS += ' -DCPU2'
    if CUR_CPU == '3':
        CFLAGS += ' -DCPU3'
    if CUR_CPU == '0':
        CFLAGS += ' -DCPU0'

    CFLAGS += ' -DSRAM_BASE={a} -DSRAM_SIZE={b}'.format(a=SRAM_BASE, b=SRAM_SIZE)
    CFLAGS += ' -DFIRMWARE_BASE={a} -DDRAM_SIZE={b} -DSHMEM_BASE={c} -DSHMEM_SIZE={d} -DLINUX_RPMSG_BASE={e} -DLINUX_RPMSG_SIZE={f}'.format(a=PRMEM_BASE, b=PRMEM_SIZE, c=SHMEM_BASE, d=SHMEM_SIZE, e=LINUX_RPMSG_BASE, f=LINUX_RPMSG_SIZE)
    CFLAGS += ' -DSHRPMSG_SIZE={a} -DSHRAMFS_SIZE={b}'.format(a=SHRPMSG_SIZE, b=SHRAMFS_SIZE)
    CFLAGS += ' -DSHLOG0_SIZE={a} -DSHLOG1_SIZE={b} -DSHLOG2_SIZE={c} -DSHLOG3_SIZE={d}'.format(a=SHLOG0_SIZE, b=SHLOG1_SIZE, c=SHLOG2_SIZE, d=SHLOG3_SIZE)

    CXXFLAGS = CFLAGS

DUMP_ACTION = OBJDUMP + ' -D -S $TARGET > rtt.asm\n'
POST_ACTION = OBJCPY + ' -O binary $TARGET rtthread.bin\n' + SIZE + ' $TARGET \n'