#include <stdio.h>
#include <string.h>
#include "time.h"
#include "libxxusb.h"


#include <fstream>
#include <iostream>
#include <vector>
#include "hvmon.h"

#define V288_BUF    0x0
#define V288_STATUS 0x2
#define V288_TX     0x4
#define V288_RST    0x8

#define V288_ADDR             0x100500
#define V288_ADDRESS_MODIFIER  0x39
#define VM_USB_SERIAL_STRING           "VM0023" 

#define TIMEOUT 100


#define V0SET                                   0
#define V1SET                                   1
#define I0SET                                   2
#define I1SET                                   3
#define VMAX                                    4
#define RUP                                     5
#define RDWN                                    6
#define TRIP                                    7
#define ISPRESENT(x) (ch_read[(x)].status&(1<<2))
#define MAKE_CODE(ch,cod)     (((ch)<<8) | (cod))

#define SY403_HV    0x08
#define SY403_PWD   0x10
#define SY403_PDWN  0x20
#define SY403_ONOFF 0x40
#define SY403_PWON  0x80


int mswap(char *x){
  char tmp;
  for (unsigned int i=0; i<strlen(x)-1;i+=2){
    tmp=x[i];
    x[i]=x[i+1];
    x[i+1]=tmp;
  } 
  return 0;

}


class Caenet {
private:
  usb_dev_handle *m_udev;
short m_am;
unsigned long m_addr;
public:
Caenet(usb_dev_handle *dev,short Address_Modifier, unsigned long Base){m_udev=dev;m_am= Address_Modifier;m_addr=Base;};

int  SingleRead(unsigned int addr,  unsigned long *data,int print){

  long dum;
  int ret;
  ret = VME_read_16 (m_udev,m_am, m_addr+addr,(long *) data);
  //  usleep(50);
  ret = VME_read_16 (m_udev,m_am, m_addr+V288_STATUS, &dum);
  if (print) printf("SingleRead Addr=0x%x Data=0x%x Status=0x%x\n",addr,(unsigned int )(*data),(unsigned int)dum);
  return dum;
};

int  SingleWrite(unsigned int  addr,  unsigned long data, int print){
  int ret;
  long dum;
  ret = VME_write_16 (m_udev,m_am, m_addr+addr, data);
  //usleep(50);
  ret = VME_read_16 (m_udev,m_am, m_addr+V288_STATUS, &dum);
  if (print) printf("SingleWrite Addr=0x%x Data=0x%x Status=0x%x\n",addr,(unsigned int ) data,(unsigned int) dum);
  return dum;
};

int WaitForReady(){
   int ret;
  unsigned long dum;
  int ncount=0;
  do {
    ret=SingleRead(V288_BUF, &dum,0 );
    ncount++;
  } while ((ret & 0x1)&ncount<TIMEOUT);
  if (ncount==TIMEOUT) printf("Wait TIMEOUT\n"); 

  return dum;
};

int Read(int max,unsigned short *data,int print){
  int ret;
  unsigned long dum;
  unsigned short *sdum=(unsigned short *)&dum;
  int ncount=0;
  do {
    ret=SingleRead(V288_BUF, &dum,0 );
    
    if (print) printf("Read %2d 0x%x 0x%x %d %c  \n",ncount+2,ret,*sdum,*sdum,*sdum);
    if (ncount<max) data[ncount]=(unsigned short) dum;
    ncount++;
  } while (!(ret & 0x1) );
  return ncount -1;
}

  int Reset(){    
    SingleWrite(V288_RST,0,1);
    usleep(5000);
    return 0;
  };


  int WriteRead(int id,int cmd,int n,unsigned short*indata, int ndata,unsigned short *data,int print){
    SingleWrite(V288_BUF, 1  , print);
    SingleWrite(V288_BUF, id , print);
    SingleWrite(V288_BUF, cmd, print);    
    for (int i=0;i<n;i++) SingleWrite( V288_BUF, indata[i]  , print );
    SingleWrite(V288_TX, 0 ,0 );
    WaitForReady();
    return Read(ndata,data,print);
  };


  int WriteRead(int id,int cmd,int ndata,unsigned short *data,int print){
    return WriteRead(id,cmd,0,(unsigned short*)NULL,ndata,data,print);
  };
  
  int WriteRead(int id,int cmd,unsigned short value, int ndata,unsigned short *data,int print){
    return WriteRead(id,cmd,1,&value,ndata,data,print);
  };



  int HvSetParameter(int id,int code, unsigned int value){
    unsigned short dum;
    int nr = WriteRead(id,code,value, 1,&dum,0);
    //    printf("%d 0x%x %d HvSetParameter IERR=0x%x\n",id, code, value,dum);
    int ncount=0;
    do { WriteRead(id,0xFF, 1,&dum,0); } while (dum!=0x2 && ncount++<TIMEOUT);	 
    //usleep(10000);
    return nr;
  };

  int HvSetChannelName(int id,int chid, char * chname){
    char name[20];
    sprintf(name,"%s",chname);
    mswap(name);
    for (int i=strlen(name);i<12;i++) name[i]=0;

    // for (int i=0;i<12;i++) printf("%d %c\n",name[i],name[i]);
    unsigned short dum;
    printf("Setting name to %s\n",name);
    
    WriteRead( id,MAKE_CODE(chid, 0x19) , 6,(unsigned short *) name, 1, &dum,0);
    return 0;
  };

  int HvSetChannelOn(int id,int chid, int state ){
    unsigned short dum;
    if (state) 
      WriteRead( id,MAKE_CODE(chid, 0x18), (SY403_HV<<8)| SY403_HV, 1,&dum,0);
    else 
      WriteRead( id,MAKE_CODE(chid, 0x18), (SY403_HV<<8)             , 1,&dum,0);
    return 0;
  };


  int HvEnableChannel(int id,int chid, int state ){
    unsigned short dum;
    if (state) 
      WriteRead( id,MAKE_CODE(chid, 0x18), (SY403_ONOFF<<8)| SY403_ONOFF, 1,&dum,0);
    else 
      WriteRead( id,MAKE_CODE(chid, 0x18), (SY403_ONOFF<<8)             , 1,&dum,0);
    return 0;
  };


  int HvSetStatusAlarm(int id, int value){
    unsigned short dum;
    WriteRead( id, 0x1A, 0, 1,&dum,0);
    return 0;
  };

  char * HvIdent(int moduleID){

    unsigned short dum[1000];
    int nr=WriteRead(moduleID,0,1000,dum,0);
    char *result = 0;
    if (nr>0) result = new char[nr];
    for (int i=0;i<nr;i++){
      result[i]=dum[i];
    }
    return result;
  };
};

Caenet *gCaenet;

usb_dev_handle *gDev=NULL;



void HvChannel::PrintFormat(){ 
  std::cout << "Format"
	    << "moduleID "
	    << "channelID " 
	    << "V0set "
	    << "IOset "
	    << "Vmax "
	    << "RUp "
	    << "RDown "  
	    << "Trip " << std::endl;
}


  void HvChannel::Print(){ 
    std::cout << name 
	      << " " << db.moduleID 
	      << " " << db.channelID
	      << " " << db.V0set
	      << " " << db.I0set
	      << " " << db.Vmax
	      << " " << db.RampUp
	      << " " << db.RampDown 
	      << " " << db.Trip << std::endl;
  }



HvChannel::HvChannel(char *cstr){

    int *data=&db.moduleID;
    char *token;
    char *delim=" ";
    token=strtok(cstr, delim);
    sprintf(name,"%s",token);
    int i=0;
    while((token=strtok(NULL, delim))!=NULL){
      // std::cout << token << std::endl;
       if (i<8) data[i++]= strtol(token,NULL,0);;
    }
    nval   =i-2;
    val    =&db.V0set;
    isvalid=i;
    db.timestamp=(int) time(NULL);
    //std::cout << "*" << std::endl;
}






HvMap::HvMap(char *fname){

  gCaenet = gCaenet;
  char cstr[500];
  std::ifstream inFile(fname);
  std::string strOneLine;

  while (inFile){
    getline(inFile, strOneLine); 
    sprintf(cstr,"%s", strOneLine.c_str());
    if (cstr[0]=='#') continue;   
    if (strOneLine.size()<10) continue;
    HvChannel * onech= new HvChannel(cstr);
    if (onech->IsValid()){
      if (onech->IsData()) hvch.push_back(*onech); 
      if (onech->IsTemplate()) hvinfo.push_back(*onech);
    }
    //std::cout << strOneLine << std::endl;
  }
  
  inFile.close();
}

int   HvMap::Print(){
  hvinfo[0].PrintFormat();


  for (unsigned int i=0; i<hvinfo.size();i++){
    hvinfo[i].Print();
  }
  for (unsigned int i=0; i<hvch.size();i++){
    hvch[i].Print();
  }
  return 0;
}

int HvMap::HvInit(){

  for (unsigned int i=0; i<hvch.size();i++){
    int moduleID = hvch[i].db.moduleID;
    int channelID= hvch[i].db.channelID;
    gCaenet->HvSetParameter(moduleID,MAKE_CODE( channelID ,0x018), 0xFF00  );
    unsigned short data=((SY403_PDWN)<<8)  | SY403_PDWN;
    gCaenet->HvSetParameter(moduleID,MAKE_CODE( channelID ,0x018), data  );

    gCaenet->HvSetChannelName(moduleID,channelID, hvch[i].name );

    for (int k=0;k< hvch[i].GetNPar();k++){
      // V1set   gCaenet->HvSetParameter(moduleID,MAKE_CODE( channelID ,0x11),  0);
      // V0set   gCaenet->HvSetParameter(moduleID,MAKE_CODE( channelID ,0x13),  0);
      unsigned int value = hvch[i].GetParameter(k);
      int code  = hvinfo[0].GetParameter(k);
      gCaenet->HvSetParameter(moduleID,MAKE_CODE( channelID ,code),  value);
    } 
    gCaenet->HvSetChannelOn(moduleID,channelID,0 );
    gCaenet->HvEnableChannel(moduleID,channelID,1 );
  }
  return 0;
 
}


int HvChannel::WriteFlag(int i,int value){
  int code = par.GetCode(i);
  int data = SY403_HV << (i-8);
  gCaenet->HvSetParameter(db.moduleID,MAKE_CODE( db.channelID ,code),  (data << 8)|data );
  
  return 0;
}
int HvChannel::WriteValue(int i,int value){
  if (par.GetType(i))  return WriteFlag(i,value);   		
  int code = par.GetCode(i);
  gCaenet->HvSetParameter(db.moduleID,MAKE_CODE( db.channelID ,code),  value*par.GetUnit(i));
  
  return 0;
}


int HvChannel::SetHvOn(int state){

  gCaenet->HvSetChannelOn(db.moduleID,db.channelID,state );
  return 0;
}





int HvMap::PrintGroupChannels(int n, unsigned short *data){
  if (n==0) {
    printf("-------------------------------------------------------\n");
    printf("ChID\tGrName\tChannel List\n");
    printf("-------------------------------------------------------\n");
    return 0;
  }
  char *cdata=(char *)data;
  printf("%s\t",cdata );
  for (int i=6;i<n;i++) {
    printf("%d\t",data[i]);
  }
  return 0;
}

int HvMap::HvReadChannel(FILE*fp,int code){

  int ndata=1000;
  unsigned short data[1000];
  for (unsigned int i=0; i<hvch.size();i++){
    int moduleID = hvch[i].db.moduleID;
    int channelID= hvch[i].db.channelID;
    
    int nr=gCaenet->WriteRead(moduleID,MAKE_CODE(channelID, code),ndata,data,0);
    fprintf(fp,"Ch. %2d |",channelID);
    switch (code) {
    case 2: 
      hvch[i].par.Fill(nr,data);
      hvch[i].par.Print(fp,nr);
      break;
    case 1: 
      hvch[i].status.Fill(nr,data);
      hvch[i].status.Print(fp,nr);
      break;
    case 0x40: if (nr>0) PrintGroupChannels(nr,data);
      break;

    default:
      for (int j=0;j<nr;j++) printf("%d\t",data[j]);
      break;
    }
    fprintf(fp,"%d\n",(int) time(NULL));

  }
  return 0;

}


int hvmon::DumpData(char *fname){

  FILE *fp=fopen(fname,"a");
  if (!fp) fprintf(stderr,"Cannot open file %s  for appending..",fname);
  int t0=0;
  HvMap *map=GetHvMap();
  for (unsigned int i=0;i< map->hvch.size();i++){
    hvstatus *s=&(map->hvch[i].status);
    fprintf(fp,"SY403\t%d\t%d\t",i,s->timestamp);
    for (int j=0;j<s->GetNVal();j++){
      fprintf(fp,"%d\t",s->GetValue(j));
    }
    fprintf(fp,"\n");
    t0=s->timestamp;
  }
  for (unsigned int i=0;i<8;i++){
    caenn470 *s=&(fMon[i]);
    fprintf(fp,"N470\t%d\t%d\t",i,t0);
    fprintf(fp,"%d\t%d\t%d\t%d\n",s->vmon,s->imon,s->vmax,s->status);
  }
  
  fclose(fp);
  return 0;
}



int hvmon::Monitor(FILE *fp){

  fHvMap->hvch[0].status.Print(fp,0);
  fHvMap->HvReadChannel(fp,0x1);
  fHvMap->hvch[0].par.Print(fp,0);
  fHvMap->HvReadChannel(fp,0x2);


  gCaenet->WriteRead(3,0x1,100,(unsigned short *)(&fMon[0].vmon), 0);
  gCaenet->WriteRead(4,0x1,100,(unsigned short *)(&fMon[4].vmon), 0);

  return 0;
}

int hvmon::Connect(){

 gDev= xxusb_serial_open(VM_USB_SERIAL_STRING);   
  if(!gDev)  {
    printf("\n\n No VM_USB present. Check if usbfs has rw permissions\n\nTEST FAILED");
    return -1; 
  } 
  return 0;
}

int hvmon::Disconnect(){

  xxusb_device_close(gDev);
  return 0;
}

int hvmon::ReloadDB(int hvinit){
  fHvMap= new HvMap("hvmon.map");
  fHvMap->Print();
  if (hvinit) fHvMap->HvInit();
  return 0;
}


int hvmon::Init( int hvinit){

  gCaenet = new Caenet(gDev, V288_ADDRESS_MODIFIER ,V288_ADDR);
  gCaenet->Reset();
  gCaenet->HvSetStatusAlarm(2,0);
  ReloadDB(hvinit);
  return 0;
}


int hvmon::Test(){

  Init(1);

  
  fHvMap->PrintGroupChannels(0,(unsigned short *)NULL);
  fHvMap->HvReadChannel(stdout,0x40);
  
  Monitor(stdout);
  DumpData("data.txt");

  unsigned short data[1000];
  gCaenet->WriteRead(2,0x6,0,10,data, 1);
  //fHvMap->HvReadChannel(0x03);
  
  return 0;
}

int hvmon::Test(int argc, char **argv){
  unsigned short data[1000];
  int print=1;
  int cmd=0;
  int mdata=-2;
  int id=2;


  if (argc>1) id   =strtol(argv[1],NULL,0);
  if (argc>2) cmd  =strtol(argv[2],NULL,0);
  if (argc>3) mdata=strtol(argv[3],NULL,0);

  int ndata=0;
  if (mdata>=0) ndata =  gCaenet->WriteRead(id,cmd,mdata,1000,data, print);
  else          ndata =  gCaenet->WriteRead(id,cmd,1000,data, print);


  
  printf("%s\n",gCaenet->HvIdent(2));

  return 0;
}

#ifdef MMAIN
int main (int argc, char **argv){
  hvmon *a=new hvmon();

  a->Connect();
  if (argc>1) a->Test(argc,argv);
  else a->Test();
  a->Disconnect();
  return 0;
}

#endif