/*

program to write to CPLD registers of which channels to read.

Modify 
LOCAL_IP
SRU_IP
to proper addresses

modify
SRU_WAIT_WRITE to test the needed time between writes.

compile:
gcc -Wall -o mcmwrreg mcmwrreg.c

write and read and verify
./mcmpwrreg --fec [0-31] --register N --write --value <value> --verify


Register to use:
#define MCM_REG_ ADCCLKDIV  0x01

NOTE: DecayBias is set externally


*/

#include <getopt.h>
#include <signal.h>
#include <stdio.h>
#include <unistd.h>
#include <arpa/inet.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <string.h>
#include <errno.h>

#define __DEBUG 1

#define FEC 1
#define CHIP 0
#define CHANNEL 0

//#define LOCAL_IP "130.235.185.193"
//#define SRU_IP "130.235.185.193"

#define LOCAL_IP "192.168.4.11"
#define SRU_IP "192.168.4.4"
#define BC_IP "192.168.4.255"


// microseconds delay between SRUSC writes
// no need for this when reading, there is a timeout of 100 milliseconds
// set in the socket.
#define SRU_WAIT_WRITE 100

#define MCM_REG_ADCCLKDIV  0x01

unsigned int readout;
unsigned int pasactrl;

unsigned int  valuerd,valuewr;
int dovalue = 0;
int setvalue;
int fec;
int loop;
int verify;
int signal_stop = 0;
int dowrite;
int sruwr;
int doreset = 0;
int mcmregister = 0;
int nberr = 0;

#define PASACTRL_PREAMP 0
#define PASACTRL_GAIN 1
#define PASACTRL_GAIN2 1
#define PASACTRL_GAIN1 2
#define PASACTRL_SHAPER 3
#define PASACTRL_SHAPER3 3
#define PASACTRL_SHAPER2 4
#define PASACTRL_SHAPER1 5
#define PASACTRL_SHUTDOWN 6
#define PASACTRL_POLARITY 7
#define PASACTRL_GRST 8
#define PASACTRL_BIASCTRL 9

typedef struct {
  char controlip[64];
  int controlport;
  struct sockaddr_in si_sru;
  int scsocket; // socket for slow control
  int listenport;
  char readoutip[64];
  unsigned int readoutipnb;
  int status; // status from configfile
  int linkup; // it it answered ping
  int active; // status & linkup is set and FMVER could be read
} SRUSERVER;


typedef struct {
  struct sockaddr_in si_controlme;
  int controlsocket; // socket for control
  int listenport; // port for listen on control
  int nbrdpackets; // Number of packets expected back
  SRUSERVER sruserver;
} SRU;



/* SRU part */
int scSruConnect(SRU *srus)
{
  int s;
  struct timeval timeout;
  timeout.tv_sec = 0;
  timeout.tv_usec = 100000;

  if ((s=socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP))==-1) {
    printf("%s: socket error %s\n",__FUNCTION__,strerror(errno));
    return -1;
  }
  setsockopt(s, SOL_SOCKET, SO_RCVTIMEO,&timeout,sizeof(timeout));
  memset((char *) &srus->si_controlme, 0, sizeof(struct sockaddr));
  srus->si_controlme.sin_family = AF_INET;
  srus->si_controlme.sin_port = htons(srus->listenport);
  if (inet_aton(LOCAL_IP, &srus->si_controlme.sin_addr)==0) {
    printf("%s: inet_aton error %s\n",__FUNCTION__,strerror(errno));	
    close(s);
    return -2;
  }
  if (bind(s, (struct sockaddr *)&srus->si_controlme, sizeof(struct sockaddr)) != 0) {
    printf("%s: bind error %s\n",__FUNCTION__,strerror(errno));	
    close(s);
    return -3;
  }
  srus->controlsocket = s;

  memset((char *) &srus->sruserver.si_sru, 0, sizeof(struct sockaddr_in));
  srus->sruserver.si_sru.sin_family = AF_INET;
  srus->sruserver.si_sru.sin_port = htons(srus->sruserver.controlport);
  if (inet_aton(srus->sruserver.controlip, &srus->sruserver.si_sru.sin_addr)==0) {
    printf("%s: sru inet_aton error %s\n",__FUNCTION__,strerror(errno));	
    close(s);
    srus->controlsocket = 0;
    return -4;
  }
  return 0;
}

/* MCM part */
int mcmWriteRegister(int handle,SRUSERVER *srv,int node, int reg, unsigned int value)
{
  unsigned int buf[4];
  int s_sent,s_bytes;
  socklen_t s_length=sizeof(srv->si_sru);

  if (node >= 20) {
    buf[0] = htonl(1 << (node - 20));
    buf[1] = 0;
  }
  else {
    buf[0] = 0;
    buf[1] = htonl(1 << node);
  }
  buf[2] = htonl(reg);
  buf[3] = htonl(value);
  s_bytes = 16;
  if ( (s_sent = sendto(handle, buf, 16, 0, (struct sockaddr *)&srv->si_sru, s_length)) ==-1) {
    return -1;
  }
  if (s_sent != s_bytes) {
    return -2;
  }
  usleep(SRU_WAIT_WRITE);
  return 0;
}

int mcmReadRegister(int handle,SRUSERVER *srv,int node, int reg, unsigned int *value)
{
  unsigned int buf[20];
  int s_sent,s_bytes;
  socklen_t s_length=sizeof(srv->si_sru);
  struct sockaddr_in si_from;
  socklen_t si_fromlen=sizeof(si_from);
  unsigned int myreg,mynode;
  if (node >= 20) {
    buf[0] = htonl(1 << (node - 20));
    buf[1] = 0;
  }
  else {
    buf[0] = 0;
    buf[1] = htonl(1 << node);
 }
  buf[2] = htonl((unsigned int) reg | 0x80000000);
  buf[3] = 0;
  s_bytes = 16;
  if ( (s_sent = sendto(handle, buf, 16, 0, (struct sockaddr *)&srv->si_sru, s_length)) ==-1) {
    return -1;
  }
  if (s_sent != s_bytes) {
    return -2;
  }
  if ((s_bytes = recvfrom(handle, buf,sizeof(buf), 0, (struct sockaddr *) &si_from, &si_fromlen))==-1) {
    return -3;
  }
  if (s_bytes != 12) {
    return -4;
  }

  if (si_from.sin_addr.s_addr == srv->si_sru.sin_addr.s_addr) {
    printf("%s: Received packet length %d from %s: %d\n",__FUNCTION__,s_bytes, 
             inet_ntoa(si_from.sin_addr), ntohs(si_from.sin_port));
  }
  else {
    printf("%s: Received illegal packet length %d from %s: %d\n",__FUNCTION__,s_bytes, 
	   inet_ntoa(si_from.sin_addr), ntohs(si_from.sin_port));
    return -5;
  }
 
  mynode = ntohl(buf[0]);
  if (mynode != node) {
    printf("%s: Received not expected node %08X expected %08X\n",
	   __FUNCTION__,mynode,node);
    return -6;
  }
  myreg = ntohl(buf[1]);
  if (myreg != ((unsigned int) reg | 0x80000000)) {
    printf("%s: Received not expected register %08X expected %08X\n",
	   __FUNCTION__,myreg,(unsigned int) reg | 0x80000000);
    return -7;
  }
  *value = ntohl(buf[2]);
  printf("%s: %08X %08X %08X\n",__FUNCTION__,mynode,myreg,*value);
  return 0;
}


void signal_handler(int signum)
{
  switch(signum) {
  case SIGQUIT:
  case SIGTERM:
  case SIGINT:
    puts("\nSIGTERM");
    signal_stop = 1; break;
    /*
  case SIGTERM:
  case SIGQUIT: break;
    */
  }
}

int decodeParameters(int argc, char **argv)
{
  int c;
  int option_index = 0;
  char *short_options = "f:l:r:v:Vw";
  const struct option long_options[]=
  {
    {"fec",1,0,'f'},
    {"loop",1,0,'l'},
    {"register",1,0,'r'},
    {"write",0,0,'w'},
    {"verify",0,0,'v'},
    {"value",1,0,'V'},
    {0, 0, 0, 0}
  };

  fec = FEC;
  loop = 1;
  verify = 0;
  dowrite = 0;
  nberr = 0;
  if (argc > 1)
  {
    while ((c = getopt_long(argc, argv, short_options,
                                        long_options, &option_index)) != EOF)
    {
      switch (c)
      {
      case 'f':
	sscanf(optarg,"%d",&fec); 
	readout |= (1 << fec);
	break;
      case 'l':
	sscanf(optarg,"%X",&loop);
	break;
      case 'r':
	sscanf(optarg,"%X",&mcmregister) ;
	break;
      case 'v': verify = 1; break;
      case 'V': dovalue = 1; sscanf(optarg,"%X",&valuewr); break;
      case 'w': dowrite = 1; break;
      default: printf("Unkown paramater\n"); return 1;
      }
    }
  }
  if ((fec < 0) || (fec >= 40)) {
    printf("%s: Illegal fec value (0-39) %d\n",__FUNCTION__,fec);
    return -1;
  }
  return 0;
}

 
int main(int argc, char **argv){

  SRU srus;
  int cont,ret1,ret2;

    signal(SIGINT,signal_handler);
    signal(SIGQUIT,signal_handler);
    signal(SIGTERM,signal_handler);

    strcpy(srus.sruserver.controlip,SRU_IP);
    srus.sruserver.controlport = 4097;
    srus.listenport = 4099;

    if (decodeParameters(argc,argv)) return 0;
    printf("Parameters: FEC= %d LOOP %d VER %d WRITE %d REG %X setvalue %X:\n",
	   fec,loop,verify,dowrite,mcmregister,valuewr);
    putchar('\n');

    ret1 = scSruConnect(&srus);
    if (ret1 != 0) {
      printf("exit after scSruConnect\n");
      return(0);
    }

    
    cont = 1;
    while(cont) {
      if (dowrite != 0) {
	ret1 = mcmWriteRegister(srus.controlsocket,&srus.sruserver,fec,mcmregister,valuewr);
	ret2 = mcmReadRegister(srus.controlsocket,&srus.sruserver,fec,mcmregister,&valuerd);
 	if (verify) {
	  if ((ret1 != 0) || (ret2 != 0)) {
	    printf("ERROR WR %d RD %d\n",ret1,ret2);
	    nberr++;
	  }
	  else if (valuewr != valuerd) {
	    printf("ERROR LOOP %d valuewr %X valuerd %X\n",loop,valuewr,valuerd);
	    nberr++;
	  }
	}
	valuewr = (~valuewr & 0xFFFF);
      }
      if ((++cont % 1000) == 0) printf("LOOP %8d\r",cont);
      if (--loop == 0) cont = 0;
      if (loop < 0) loop = 0;
      if (signal_stop) cont = 0;
    } 

    close(srus.controlsocket);
    printf("NBERR %d\n",nberr);
    puts("\nEXIT");
    return 0;
}