/*   Neural Lab C link Eyebot Example */
/*   Basic XOR gate */
/*   Copyright 2000 - Zagros Robotics */
/*   by AJ Neal */

/*   This example program acts as a simple XOR gate created using
     Neural Lab.

     XOR GATE TRUTH TABLE
     INPUTS     OUTPUT
     0   0        0
     0   1        1
     1   0        1
     1   1        0
*/


#include "c:/EYEBOT/MC/INCLUDE/eyebot.h"
#include <stdio.h>
#include <stdlib.h>
#include <math.h>


int function=1;
int laynum=3;
int snum[3]={2,8,1};
float W[24]={
0.400180,
0.145211,
0.761415,
5.363467,
5.550502,
-1.531479,
0.043131,
-1.121034,
0.689130,
0.945761,
0.362040,
5.071275,
-2.524523,
3.663692,
1.094905,
2.896029,
-1.088969,
-1.253017,
-1.008614,
11.501174,
-5.406621,
-3.354352,
-1.170112,
-2.586737};

int nxtlay(int a,int b,int l,int *snum)
{
	int total=0;
	int i=0;

	/* calc layer layer total */
	while(i<l)
	{
		total=total+snum[i]*snum[i+1];
		i++;
	}
	total=total+a*snum[l+1]+b;
	return(total);
}



int answers(float *in,float *out,int function,float *W)
{
  float E[3][10];
  float O[3][10];
  int a=0;
  int b=0;
  int l=0;

  /* init E */

  a=0;
  l=0;
  while(l<laynum)
  {
    a=0;
    while(a<snum[l])
    {
	E[l][a]=0;

	a++;
    }
   l++;
   }


/* input layer */

a=0;
while(a<snum[0])
{
 O[0][a]=*in;
 in++;
 a++;
}

/* hidden and output layers */
l=1;
while(l<laynum)
{
a=0;
	while(a<snum[l])
	{
	b=0;
		while(b<snum[l-1])
		{
		E[l][a]=E[l][a]+O[l-1][b]*(W[nxtlay(b,a,l-1,snum)]);
		b++;
	}
if(function==1)
     O[l][a]=1/(1+exp(-E[l][a]));  /*unipolar activation function */
if(function==2)
      O[l][a]=2/(1+exp(-E[l][a]))-1; /* bipolar activation function */
if(function==3) /* linear */
      O[l][a]=E[l][a];
	 if(l==laynum-1)
	 {
	 /* pass back output layer */
	 *out=O[l][a];
	 out++;  /* inc pointer */

	 }

	 a++;
	 }
l++;
}
return(1);
} /* end answers */


int main ()
{

	int a=0;
	int b=0;
       	float in[2];
	float out;
        char k;
	LCDMode(SCROLLING|NOCURSOR);
        LCDPutString("Neural Lab Demo...\n");
        LCDPutString("Max Online\n");
        LCDPutString("Zagros Robotics 2000 ...\n");


      do {
      LCDMenu("T","","","END");
      KEYGetBuf(&k);
      switch (k) {
	case KEY1:
        LCDPutString("NLab Xor Test\n");
        /* run all 4 possible cases throught neural network */
 a=0;
        while(a<2)
	{
	b=0;
		while(b<2)
		{
                        in[0]=a;
                        in[1]=b;
                        answers(in,&out,1,W);
                        printf("[%d][%d]:%f\n",a,b,out);
			b++;
		}
                          
         a++;
	}

       break;
    
      } /* end switch */
    } while(KEY4 != k);
  
    

	return(0);

}  /* end main */