// ***************** overlay.c ***************************************
//                 Risc-Dsp runtime library
//
// This file implements interface to dsp overlays handling
//
// (c) multicore.ru
//
// *******************************************************************
#include <stdlib.h>
#include <string.h>

#include "mcom02.h"
#include "overlay.h"

//
// address of dsp-runtime finction exit()
// it must be implemented in dsp code
//
extern int __dsp_exit;

//
// working overlays
//
struct ovl_dsp_ctx dsp_ctx[MAX_DSP];

#include "../overlay.inc"

// variables for overlays handling
unsigned curr_dsp_num = 0xffffffff;
static unsigned ctx_is_init = 0;

//
// @name: init_dsp_entry
// @description: initializaion of dsp entries
//
void init_dsp_entry()
{
   memset(dsp_ctx,0,sizeof(struct ovl_dsp_ctx)*MAX_DSP);

  int i,j;
  for(i = 0; i < MAX_DSP; i++)
  {
          for(j = 0; j < MAX_OVERLAY_CTX; j++)
         {
            dsp_ctx[i].ctx[j].dsp_info = &dsp_ctx[i].info;
            dsp_ctx[i].ctx[j].__num = j;
         }
         dsp_ctx[i].curr_ctx_num = 0xffffffff;
  }
  ctx_is_init = 1;
}

//
// @name: set_curr_dsp
// @description: switch working dsp
//
// @in unsigned num   - dsp num (0..3)
//
void set_curr_dsp(unsigned num)
{
    curr_dsp_num = num;
}

//
// @name: calc_pram_size
// @description: calculates size of pram according to given page mode
//
// @in unsigned page_mode   - dsp num (0..3)
// @return - size !=0 if page mode is corect
//           0 - error page mode
unsigned calc_pram_size(unsigned page_mode)
{
   unsigned memory_size = 0;
   switch(page_mode)
   {
       case  0 :
                memory_size = 0x8000;
                break;
       case  2 :
                memory_size = 0x8000*3;
                break;
       case  3 :
                memory_size = 0x8000*4;
                break;
        default:
                break;
   }
   return memory_size;
}

//
// @name: calc_xyram_size
// @description: calculates size of xram according to given page mode
//
// @in unsigned page_mode   - dsp num (0..3)
// @return - size !=0 if page mode is corect
//           0 - error page mode
//
unsigned calc_xyram_size(unsigned page_mode)
{
   unsigned memory_size = 0;
   switch(page_mode)
   {
        case 0 :
                memory_size = 0x8000*4;
                break;
        case 2 :
                memory_size = 0x8000*2;
                break;
        case 3 :
                memory_size = 0x8000;
                break;
        default:
                break;
   }
   return memory_size;
}

//
// @name: calc_xyram_size
// @description: calculates size of xram according to given page mode
//
// @in unsigned page_mode   - dsp num (0..3)
// @return - size !=0 if page mode is corect
//           0 - error page mode
//
void add_dsp_entry(
        unsigned  num,
        unsigned  page_mode,
        unsigned  qstr_stop,
        unsigned  pram_base,
        unsigned  xyram_base)
{
     if(!ctx_is_init)
        init_dsp_entry();

     if(num >= MAX_DSP) return;

     dsp_ctx[num].info.num = num;
     dsp_ctx[num].info.page_mode = page_mode;
     dsp_ctx[num].info.qstr_stop = qstr_stop;
     dsp_ctx[num].info.pram_base = pram_base;
     dsp_ctx[num].info.pram_size = calc_pram_size(page_mode);
     dsp_ctx[num].info.xyram_base = xyram_base;
     dsp_ctx[num].info.xyram_size = calc_xyram_size(page_mode);
     dsp_ctx[num].info.__present = 1;
}

//
// @name: find_overlay_ctx
// @description: select overlay at given address
//
// @in unsigned address - address of program
// @return - struct ovl_ctx* != NULL - overlay section
//			 NULL	- address is not valid or overlay for given address was not created
//
struct ovl_ctx* find_overlay_ctx(unsigned address)
{
	struct ovl_ctx* find_ctx = 0;
	struct ovl_ctx* curr_ctx = 0;
	if(curr_dsp_num >= MAX_DSP) return 0;

	struct ovl_dsp_ctx * curr_dsp_ctx = &dsp_ctx[curr_dsp_num];
	int i;
	for(i = 0; i < MAX_OVERLAY_CTX; i++)
	{
		curr_ctx = &curr_dsp_ctx->ctx[i];
		if(!curr_ctx->__present)
		    break;                  // end of ctx table

		unsigned bottom = *((int*)&curr_ctx->__text_src);
		unsigned top = bottom + curr_ctx->__text_size;
		if(address >= bottom && address < top)
		{
		    find_ctx = curr_ctx;
		    break;
		}
	}

	return find_ctx;
}

//
// @name: init_overlay
// @description: select active overlay and load it into memory
//
// @in unsigned num - core id [0..MAX_DSP]
//
void init_overlay(unsigned num)
{

	if(num >= MAX_DSP)   // num overflow MAX_DSP number
	{
	        return;
	}
	if(!dsp_ctx[num].info.__present)
	{
	        return;
	}

	// set overlay table
	curr_dsp_num = num;
	struct ovl_dsp_ctx * curr_dsp_ctx = &dsp_ctx[curr_dsp_num];

	__overlay_ctx_init_table(curr_dsp_ctx->ctx);

	unsigned exit_address = (unsigned int)&__dsp_exit;
	unsigned __dsp_exit_address;

	struct ovl_ctx* find_ctx = find_overlay_ctx(exit_address);

	if(find_ctx)
	{
		__dsp_exit_address = (exit_address - (unsigned int)find_ctx->__text_src)>>2;
	}
	else
	{
		return;
	}

	int i;
	for(i = 0; i < MAX_OVERLAY_CTX; i++)
	{
	   if(!curr_dsp_ctx->ctx[i].__present)
	       break;                  // end of ctx table
	   curr_dsp_ctx->ctx[i].__exit_dsp =  __dsp_exit_address;
	}

	print_overlay(find_ctx);
	load_overlay(find_ctx);
}

//
// @name: load_overlay
// @description: load overlay it into memory
//
// @in struct ovl_ctx* ctx - ovrelay to be loaded
// @return 1 - overlay is loaded
//         0 - memory boundary error
//
int load_overlay(struct ovl_ctx* ctx)
{
	// chech need load

	if(ctx->__num == dsp_ctx[curr_dsp_num].curr_ctx_num
	&& ctx->dsp_info->num == curr_dsp_num)
	{
		return 1; // already loaded
	}

	unsigned *src, *dst;
	unsigned size;

	// load program

	src  = ctx->__text_src;
	dst  = ctx->__text_dst;
	size = (ctx->__text_size + 3) & 0xfffffffc;

	// verify address
	if(((unsigned)(dst) >= ctx->dsp_info->pram_base)
	  && (((unsigned)(dst) + size) <= (ctx->dsp_info->pram_base + ctx->dsp_info->pram_size)))
	{
		if(src != dst)
		while(size) *dst++ = *src++,size-=4;
	}
	else
	{
		asm("sizepramerror: .globl sizepramerror");
		return 0;
	}

	src  = ctx->__data_src;
	dst  = ctx->__data_dst;
	size = (ctx->__data_size + 3) & 0xfffffffc;

	// verify address
	if(((unsigned)(dst) >= ctx->dsp_info->xyram_base)
	  && (((unsigned)(dst) + size) <= (ctx->dsp_info->xyram_base + ctx->dsp_info->xyram_size)))
	{
		if(src != dst)
			while(size) *dst++ = *src++,size-=4;
	}
	else
	{
		asm("sizexramerror: .globl sizexramerror");
		return 0;
	}

	src  = ctx->__bss_src;
	dst  = ctx->__bss_dst;
	size = (ctx->__bss_size + 3) & 0xfffffffc;

	// verify address
	if(((unsigned)(dst) >= ctx->dsp_info->xyram_base)
	  && (((unsigned)(dst) + size) <= (ctx->dsp_info->xyram_base + ctx->dsp_info->xyram_size)))
	{
		if(src != dst)
			while(size) *dst++ = *src++,size-=4;
	}
	else
	{
		return 0;
	}

	dsp_ctx[curr_dsp_num].curr_ctx_num = ctx->__num;

	return 1;
}

// @name: print_overlay
// @description: reports overlay content, (uncomment code in source to use it)
//
// @in struct ovl_ctx* ctx - ovrelay to be reported
//
void print_overlay(struct ovl_ctx* ctx)
{
      //debug_printf("overlay_info: \n");
      //debug_printf("text : %08x -> %08x, (0x%x) \n", ctx->__text_src, ctx->__text_dst, ctx->__text_size);
      //debug_printf("data : %08x -> %08x, (0x%x) \n", ctx->__data_src, ctx->__data_dst, ctx->__data_size);
      //debug_printf("bss  : %08x -> %08x, (0x%x) \n", ctx->__bss_src, ctx->__bss_dst, ctx->__bss_size);
}

// @name: print_dsp_memory_info
// @description: reports dsp_memory settings (uncomment code in source to use it)
//
void print_dsp_memory_info()
{
      int i;
      //debug_printf("dsp memory info: \n");
      for(i = 0; i < MAX_DSP; i++)
      {
        if(!dsp_ctx[i].info.__present) break;
        //debug_printf("dsp[%d] pram  - [%08x : %08x], (0x%05x)\n",i, dsp_ctx[i].info.pram_base,
                //dsp_ctx[i].info.pram_base + dsp_ctx[i].info.pram_size, dsp_ctx[i].info.pram_size);
        //debug_printf("dsp[%d] xyram - [%08x : %08x], (0x%05x)\n",i, dsp_ctx[i].info.xyram_base,
                //dsp_ctx[i].info.xyram_base + dsp_ctx[i].info.xyram_size, dsp_ctx[i].info.xyram_size);
      }
}