#include "xaxicdma_i.h"
#include "xparameters.h"
#include "xaxicdma.h"
#include "string.h"


#define BD_SPACE_BASE XPAR_DESCRIPTOR_MEMORY_AXI_BRAM_CTRL_2_S_AXI_BASEADDR
#define BD_SPACE_HIGH XPAR_DESCRIPTOR_MEMORY_AXI_BRAM_CTRL_2_S_AXI_HIGHADDR
#define NUMBER_OF_BDS_TO_TRANSFER 4


int main()
{
int n = 1024;

unsigned int *mem0_ptr = (unsigned int *) XPAR_INTERNAL_MEMORY_0_AXI_BRAM_CTRL_0_S_AXI_BASEADDR;
unsigned int *mem1_ptr = (unsigned int *) XPAR_INTERNAL_MEMORY_1_AXI_BRAM_CTRL_1_S_AXI_BASEADDR;

memset(mem0_ptr, 0, n*sizeof(unsigned int));
memset(mem1_ptr, 0, n*sizeof(unsigned int));
memset(mem1_ptr + n, 0, n*sizeof(unsigned int));

for(int i = 0; i < n; i++)
{
mem0_ptr[i] = i;
}

int status, desc_count_max;
XAxiCdma_Bd decs_template;
XAxiCdma_Config *cfg_ptr;
XAxiCdma cdma_engine;
XAxiCdma *cdma = &cdma_engine;


cfg_ptr = XAxiCdma_LookupConfig(XPAR_AXI_CDMA_0_DEVICE_ID);

status = XAxiCdma_CfgInitialize(cdma, cfg_ptr, cfg_ptr->BaseAddress);
XAxiCdma_Reset(cdma);

while(!XAxiCdma_ResetIsDone(cdma))
;

while(XAxiCdma_IsBusy(cdma))
;

//--------------------------------------------------------------------
// SG Mode
//--------------------------------------------------------------------
memset(mem1_ptr, 0, 32768);

if(XAxiCdma_IsSimpleMode(cdma))
{
XAxiCdma_SwitchMode(cdma, XAXICDMA_SG_MODE);
}

if(XAxiCdma_IsSimpleMode(cdma))
{
XAxiCdma_SwitchMode(cdma, XAXICDMA_SG_MODE);
}

desc_count_max = XAxiCdma_BdRingCntCalc(XAXICDMA_BD_MINIMUM_ALIGNMENT, BD_SPACE_HIGH - BD_SPACE_BASE + 1, BD_SPACE_BASE);
status = XAxiCdma_BdRingCreate(cdma, BD_SPACE_BASE, BD_SPACE_BASE, XAXICDMA_BD_MINIMUM_ALIGNMENT, desc_count_max);
if(status != XST_SUCCESS)
{
return 1;
}

XAxiCdma_BdClear(&decs_template);
XAxiCdma_BdRingClone(cdma, &decs_template);

XAxiCdma_Bd *desc_ptr, *curr_desc_ptr;

status = XAxiCdma_BdRingAlloc(cdma, NUMBER_OF_BDS_TO_TRANSFER, &desc_ptr);
if(status != XST_SUCCESS)
{
return 1;
}

curr_desc_ptr = desc_ptr;

for(int i = 0; i < NUMBER_OF_BDS_TO_TRANSFER; i++)
{
status = XAxiCdma_BdSetSrcBufAddr(curr_desc_ptr, (unsigned long) mem0_ptr);
if(status != XST_SUCCESS)
{
return 1;
}
status = XAxiCdma_BdSetDstBufAddr(curr_desc_ptr, (unsigned long) (mem1_ptr + i*n));
if(status != XST_SUCCESS)
{
return 1;
}
status = XAxiCdma_BdSetLength(curr_desc_ptr, n*sizeof(unsigned int));
if(status != XST_SUCCESS)
{
return 1;
}

curr_desc_ptr = XAxiCdma_BdRingNext(cdma, curr_desc_ptr);
}

XAxiCdma_BdRingToHw(cdma, NUMBER_OF_BDS_TO_TRANSFER, desc_ptr, 0, 0);

while(XAxiCdma_IsBusy(cdma))
;

while(1) {}
}