#include <xs1.h>
#include <platform.h>
#include <xs1_su.h>
#include <print.h>

#define XS1_SU_PERIPH_USB_ID 0x1

/* Note, this function is prototyped in xs1.h only from 13 tools onwards */
unsigned get_tile_id(tileref);

extern tileref tile[];

void device_reboot_aux(void) 
{
#if (XUD_SERIES_SUPPORT == 1)
    /* Disconnect from bus */
    unsigned data[] = {4};
    write_periph_32(usb_tile, XS1_SU_PERIPH_USB_ID, XS1_SU_PER_UIFM_FUNC_CONTROL_NUM, 1, data);

    /* Ideally we would reset SU1 here but then we loose power to the xcore and therefore the DFU flag */
    /* Disable USB and issue reset to xcore only - not analogue chip */
    write_node_config_reg(usb_tile, XS1_SU_CFG_RST_MISC_NUM,0b10);
#else
    unsigned int pllVal;
    unsigned int localTileId = get_local_tile_id();
    unsigned int tileId;
    unsigned int tileArrayLength;
    
    /* Find size of tile array - note in future tools versions this will be available from platform.h */
    asm volatile ("ldc %0, tile.globound":"=r"(tileArrayLength));
  
    /* Reset all remote tiles */
    for(int i = 0; i< tileArrayLength; i++) 
    {
        /* Cannot cast tileref to unsigned! */
        tileId = get_tile_id(tile[i]); 

        /* Do not reboot local tile yet! */
        if(localTileId != tileId)
        {
            read_sswitch_reg(tileId, 6, pllVal);
            write_sswitch_reg_no_ack(tileId, 6, pllVal);
        }
    }
    
    /* Finally reboot this tile! */
    read_sswitch_reg(localTileId, 6, pllVal);
    write_sswitch_reg_no_ack(localTileId, 6, pllVal);
#endif
}

/* Reboots XMOS device by writing to the PLL config register */
void device_reboot_implementation(chanend spare) 
{
#if (XUD_SERIES_SUPPORT != 1)
    outct(spare, XS1_CT_END);   // have to do this before freeing the chanend
    inct(spare);                // Receive end ct from usb_buffer to close down in both directions
    
    /* Need a spare chanend so we can talk to the pll register */
    asm("freer res[%0]"::"r"(spare));
#endif
    device_reboot_aux();
}