#include <stdio.h>
#include <stdlib.h>
#include <assert.h>

#include "ssam.h"

ssam_cnct_t *rw;
ssam_cnct_t *data;
ssam_cnct_t *addr;
ssam_cnct_t *ack;

#define MAT_SIZE 2

enum memops {
  MEM_R,
  MEM_W
};

/* MEM */
void* mem()
{
  int tab[(MAT_SIZE*2)*3] = {0, 1, 2, 3,
			     4, 5, 6, 7,
			     -1, -1, -1, -1};
  
  unsigned int op=-1;
  unsigned int at;
  unsigned int res;
  
  for(;;ssam_wait())
    {
      res = 0;      
      ssam_read(rw, &op);

      switch( op )
	{
	case MEM_R: /* Lire dans la mémoire */
	  ssam_read(addr, &at);
#ifdef DEBUG
	  printf("MEM: Read @%d (=%d)!\n", at, tab[at]);
#endif
	  ssam_write(data, &tab[at]);
	  res = 1;
	  break;
	case MEM_W: /* Ecrire dans la memoire */
	  ssam_read(addr, &at);
	  ssam_read(data, &tab[at]);
#ifdef DEBUG
	  printf("MEM: Write %d@%d!\n", tab[at], at);
#endif
	  res = 1;
	  break;
	default:
	  break;
	}

      ssam_write(ack, &res);

      	  
    }
  
  return NULL;
}

/* CPU */
void* cpu()
{

  /* Retourne la valeur à l'adresse donnée */
  int get_addr(unsigned int at)
    {
      int res;
      unsigned int my_ack=0;
      unsigned int op = MEM_R;

      ssam_write(rw, &op);
      ssam_write(addr, &at);

      do {
	ssam_wait();
	ssam_read(ack, &my_ack);
      } while ( my_ack != 1 );
      
      ssam_read(data, &res);

      return res;
    }
  
  /* Ecrit la valeur à l'adresse donnée */
  void set_addr(unsigned int at, int my_data)
    {
      unsigned int my_ack=0;
      unsigned int op = MEM_W;
      
      ssam_write(addr, &at);
      ssam_write(data, &my_data);
      ssam_write(rw, &op);

      do {
	ssam_wait();
	ssam_read(ack, &my_ack);
      } while ( my_ack != 1 );
      
    }

  
  /* Lire une ligne d'une matrice */
  void read_line(int tab[MAT_SIZE], int matno, int lineno)
    {
      int i;
      for(i=0;i<MAT_SIZE;i++)
	{
	  int ind = (matno*(MAT_SIZE*2)) + (lineno*MAT_SIZE) + i;
	  tab[i] = get_addr(ind);
	}
    }

  /* Lire une colonne d'une matrice */
  void read_col(int tab[MAT_SIZE], int matno, int colno)
    {
      int i;
      for(i=0;i<MAT_SIZE;i++)
	{
	  int ind = (matno*(MAT_SIZE*2)) + (i*MAT_SIZE) + colno;
	  tab[i] = get_addr(ind);
	}
    }

  void dump_mat(unsigned int no)
  {
    int i, j;
    int line[MAT_SIZE];
    for(i=0;i<MAT_SIZE;i++)
      {
	read_line(line, no, i);
	for(j=0;j<MAT_SIZE;j++)
	  printf("%d ", line[j]);
	printf("\n");
      }
  }


  int i, j, k;
  int line[MAT_SIZE], col[MAT_SIZE];

  for(;;)
    {
      for(i=0;i<MAT_SIZE;i++)
	{
	  read_line(line, 0, i);

	  for(j=0;j<MAT_SIZE;j++)
	    {
	      read_col(col, 1, j);

	      int total = 0;
	      for(k=0;k<MAT_SIZE;k++)
		total += line[k] + col[k];

	      set_addr(2*(MAT_SIZE*2)+(j*MAT_SIZE)+i, total);
	    }
	}

      printf("== RES ==\n");
      dump_mat(2);
      printf("\n");
    }

  return NULL;
}

int main()
{
  ssam_init(2);

  ssam_create_component(mem);
  ssam_create_component(cpu);

  rw = ssam_create_connection("rw", SSAM_BOOL_T);
  data = ssam_create_connection("data", SSAM_UINT_T);
  addr = ssam_create_connection("addr", SSAM_UINT_T);
  ack = ssam_create_connection("ack", SSAM_BOOL_T);

  ssam_start(100, "addlog");

  return ssam_exit();
}