Subversion Repositories f9daq

Rev

Rev 245 | Rev 251 | Go to most recent revision | Blame | Compare with Previous | Last modification | View Log | RSS feed

  1. #include "redpitaya_gui.h"
  2. #include <ansi_c.h>
  3. #include <tcpsupp.h>
  4. #include <utility.h>
  5. #include <cvirte.h>
  6. #include <userint.h>
  7. #include "redpitaya_gui.h"
  8.  
  9. #include "H1D.h"
  10.  
  11.  
  12. #define NBEFORE 150
  13. #define MAXSAMPLES 16384
  14. #define MINTRGDELAY -8192
  15.  
  16.  
  17. static int panelHandle;
  18.  
  19. static unsigned int chandle = 0;
  20. //static int tfID;
  21. int pfreq;
  22. static int plothandle[2]= {0,0};
  23. static int tdcplothandle[2]= {0,0};
  24. static int adcplothandle[2]= {0,0};
  25. int debug ;
  26. int initialized = 0;
  27. #define MAX_THREADS 10
  28. int nsamples=0;
  29. int adctype;
  30. float athreshold, twin0,twin1;
  31. float timebins[0XFFFF];
  32.  
  33. FILE *foutput;
  34. int outwaveforms=0;
  35.  
  36. static CmtThreadPoolHandle poolHandle = 0;
  37. int ctrl_c=0;
  38.  
  39.  
  40. int CVICALLBACK SocketCB (unsigned handle, int xType, int errCode, void *callbackData);
  41.  
  42.  
  43.  
  44. static int export_data (int hid) {
  45.  
  46.   char filename[0xFF],rootcmd[0xFF];
  47.   char fname[0xFF];
  48.   int type=0;
  49.  
  50.   GetCtrlVal(panelHandle,PANEL_EXPORTNAME,fname);
  51.   GetCtrlVal(panelHandle,PANEL_FILETYPE,&type);
  52.  
  53.   FILE *fp;
  54.  
  55.   switch (type) {
  56.     case 0:
  57.     case 1:
  58.       sprintf(filename,"%s_%d.root",fname, hid);
  59.       fp  =fopen(filename,"wb");
  60.       if (fp) {
  61.         H1D_Write2File(hid,fp);
  62.         fclose(fp);
  63.         if (type) {
  64.           sprintf(rootcmd ,"thisroot.bat && root.exe H1Dload.cxx(\\\"%s\\\")", filename);
  65.           LaunchExecutable(rootcmd);
  66.  
  67.         }
  68.         printf("Histogram %d exported to %s\n", hid, filename);
  69.       }
  70.       break;
  71.     case 2:
  72.       sprintf(filename,"%s_%d.txt",fname, hid);
  73.       fp=fopen(filename,"w");
  74.       if (fp) {
  75.         for (int i=0; i<H1D_GetNbinsX(hid); i++) fprintf(fp,"%g\t%g\n", H1D_GetXBinCenter(hid,i), H1D_GetBinContent(hid,i) );
  76.         fclose(fp);
  77.         printf("Histogram %d exported to %s\n", hid, filename);
  78.       }
  79.      
  80.       break;
  81.  
  82.   }
  83.  
  84.  
  85.   return (0);
  86. }
  87.  
  88.  
  89. int main (int argc, char *argv[]) {
  90.   if (InitCVIRTE (0, argv, 0) == 0)
  91.     return -1;  /* out of memory */
  92.   if ((panelHandle = LoadPanel (0, "redpitaya_gui.uir", PANEL)) < 0)
  93.     return -1;
  94.  
  95.  
  96.   SetStdioPort (CVI_STDIO_WINDOW);
  97.   SetSleepPolicy(VAL_SLEEP_MORE);
  98.   CmtNewThreadPool (MAX_THREADS,  &poolHandle);
  99.  
  100.   //for (int i=0;i<1000;i++) H1D_Fill(1,i,i);
  101.   //H1D_Draw(1,panelHandle,PANEL_ADC1,&adcplothandle[0]);
  102.   printf("size of double = %d\n",sizeof(double));
  103.   DisplayPanel (panelHandle);
  104.   RunUserInterface ();
  105.   DiscardPanel (panelHandle);
  106.   CmtDiscardThreadPool (poolHandle);
  107.   if (chandle!=0) DisconnectFromTCPServer (chandle);
  108.   return 0;
  109. }
  110.  
  111. char strbuf[0xFF];
  112.  
  113. int gLog=0;
  114.  
  115. int printf(const char *format, ...) {
  116.   va_list aptr;
  117.   int ret;
  118.   FILE *flog;
  119.  
  120.   va_start(aptr, format);
  121.   ret = vsprintf(strbuf, format, aptr);
  122.   va_end(aptr);
  123.   SetCtrlVal(panelHandle,PANEL_STDIO,strbuf);
  124.  
  125.   if (gLog) {
  126.     flog = fopen ("stdio.log", "a");
  127.     fprintf (flog, "%s", strbuf);
  128.     fclose (flog);
  129.   }
  130.   return(ret);
  131. }
  132.  
  133. int histoinit() {
  134.   int nch;
  135.   float min,max;
  136.  
  137.   GetCtrlVal(panelHandle,PANEL_NCH_1, &nch);
  138.   GetCtrlVal(panelHandle,PANEL_MINX_1, &min);
  139.   GetCtrlVal(panelHandle,PANEL_MAXX_1, &max);
  140.  
  141.   H1D_Init(1, "ADC ch 1","Pulse height", nch, min, max );
  142.   H1D_SetTitleX(1,"ADC (V)");
  143.   H1D_SetTitleY(1,"N");
  144.  
  145.   GetCtrlVal(panelHandle,PANEL_NCH_2, &nch);
  146.   GetCtrlVal(panelHandle,PANEL_MINX_2, &min);
  147.   GetCtrlVal(panelHandle,PANEL_MAXX_2, &max);
  148.  
  149.   H1D_Init(2, "ADC ch 2","Pulse height", nch, min, max );
  150.   H1D_SetTitleX(2,"ADC (V)");
  151.   H1D_SetTitleY(2,"N");
  152.  
  153.   GetCtrlVal(panelHandle,PANEL_NCH_3, &nch);
  154.   GetCtrlVal(panelHandle,PANEL_MINX_3, &min);
  155.   GetCtrlVal(panelHandle,PANEL_MAXX_3, &max);
  156.  
  157.   H1D_Init(3, "TDC ch 1","TDC", nch, min, max  );
  158.   H1D_SetTitleX(3,"TDC (us)");
  159.   H1D_SetTitleY(3,"N");
  160.  
  161.   GetCtrlVal(panelHandle,PANEL_NCH_4, &nch);
  162.   GetCtrlVal(panelHandle,PANEL_MINX_4, &min);
  163.   GetCtrlVal(panelHandle,PANEL_MAXX_4, &max);
  164.  
  165.   H1D_Init(4, "TDC ch 2","TDC", nch, min, max   );
  166.   H1D_SetTitleX(4,"TDC (us)");
  167.   H1D_SetTitleY(4,"N");
  168.  
  169.   SetCtrlAttribute  (panelHandle, PANEL_ADC1, ATTR_XNAME, H1D_GetTitleX(1) );
  170.   SetCtrlAttribute  (panelHandle, PANEL_ADC1, ATTR_YNAME, H1D_GetTitleY(1) );
  171.   SetCtrlAttribute  (panelHandle, PANEL_ADC2, ATTR_XNAME, H1D_GetTitleX(2) );
  172.   SetCtrlAttribute  (panelHandle, PANEL_ADC2, ATTR_YNAME, H1D_GetTitleY(2) );
  173.  
  174.   SetCtrlAttribute  (panelHandle, PANEL_TDC1, ATTR_XNAME, H1D_GetTitleX(3) );
  175.   SetCtrlAttribute  (panelHandle, PANEL_TDC1, ATTR_YNAME, H1D_GetTitleY(3) );
  176.   SetCtrlAttribute  (panelHandle, PANEL_TDC2, ATTR_XNAME, H1D_GetTitleX(4) );
  177.   SetCtrlAttribute  (panelHandle, PANEL_TDC2, ATTR_YNAME, H1D_GetTitleY(4) );
  178.  
  179.   GetCtrlVal(panelHandle,PANEL_MINX_5, &min);
  180.   GetCtrlVal(panelHandle,PANEL_MAXX_5, &max);
  181.   SetAxisScalingMode (panelHandle, PANEL_GRAPH, VAL_LEFT_YAXIS, VAL_MANUAL, min, max);
  182.   SetCtrlAttribute  (panelHandle, PANEL_GRAPH, ATTR_LABEL_TEXT , "sampling adc data");
  183.   SetCtrlAttribute  (panelHandle, PANEL_GRAPH, ATTR_XNAME, "t(us)" );
  184.   SetCtrlAttribute  (panelHandle, PANEL_GRAPH, ATTR_YNAME, "U(V)" );
  185.  
  186.   GetCtrlVal(panelHandle,PANEL_TWIN0, &twin0);
  187.   GetCtrlVal(panelHandle,PANEL_TWIN1, &twin1);
  188.   GetCtrlVal(panelHandle,PANEL_ADCTYPE, &adctype);
  189.   GetCtrlVal(panelHandle,PANEL_ITRGLEVEL  , &athreshold);
  190.   return 0;
  191. }
  192.  
  193. int analyse(int nb, unsigned char *cdata, int *info, int *ninfo) {
  194.  
  195.   int   *ibuf = (int *)cdata;
  196.   float *fbuf = (float *)cdata;
  197.   float *finfo = (float *)info;
  198.   int neve=0;
  199.   int *data = (ibuf+3);
  200.   int nr=0;
  201.   static float adc = 10000;
  202.   static float qdc = 0;
  203.   *ninfo = 0;
  204.   printf("Run HDR LEN=%d NEVE=%d dt=%f adc=%f qdc=%f\n", ibuf[0],ibuf[1],fbuf[2],adc,qdc);
  205.   while (nr<nb) {
  206.  
  207.     int recid   = *data++;
  208.     int chmask  = *data++;
  209.     nr +=8;
  210.     if (recid!=0x2) continue;
  211.     for (int id=0; id<2; id++) {
  212.       if ( !(chmask & (1 << id)) ) {
  213.         if (neve % pfreq == 0)
  214.           if (plothandle[id]) {
  215.             DeleteGraphPlot (panelHandle, PANEL_GRAPH, plothandle[id], VAL_IMMEDIATE_DRAW);
  216.             plothandle[id] = 0;
  217.           }
  218.         continue;
  219.       }
  220.       if ( id != *(data++) ) printf("Error\n");
  221.  
  222.       int nsamples = *(data++);
  223.       if (nsamples<=0 || nsamples>16*1024) {
  224.         printf("Error nsamples %d\n", nsamples);
  225.         return -1;
  226.       }
  227.       float *fdata = (float *) data;
  228.       if ( nsamples>0  && neve % pfreq == 0) {
  229.         const int col[4]= {VAL_RED,VAL_GREEN,VAL_BLUE,VAL_WHITE};
  230.         if (plothandle[id]) DeleteGraphPlot (panelHandle, PANEL_GRAPH, plothandle[id], VAL_IMMEDIATE_DRAW);
  231.  
  232.         plothandle[id] = PlotXY (panelHandle, PANEL_GRAPH, timebins, fdata, nsamples, VAL_FLOAT, VAL_FLOAT, VAL_THIN_LINE, VAL_NO_POINT, VAL_SOLID, 1, col[id]);
  233.  
  234.         //plothandle[id] = PlotXY (panelHandle, PANEL_GRAPH, timebins, data, nsamples, VAL_FLOAT, VAL_INTEGER, VAL_THIN_LINE, VAL_NO_POINT, VAL_SOLID, 1, col[id]);
  235.         H1D_Draw(1,panelHandle,PANEL_ADC1,&adcplothandle[0]);
  236.         H1D_Draw(2,panelHandle,PANEL_ADC2,&adcplothandle[1]);
  237.         H1D_Draw(3,panelHandle,PANEL_TDC1,&tdcplothandle[0]);
  238.         H1D_Draw(4,panelHandle,PANEL_TDC2,&tdcplothandle[1]);
  239.  
  240.         if (debug) for (int k=0; k<10; k++) printf("%d %d (%f , %d)\n", id,k, timebins[k],data[k]);
  241.       }
  242.  
  243.       nr+=8;
  244.       adc=10000;
  245.       qdc=0;
  246.       int nqdc=0;
  247.           int ntdc=0;
  248.       for (int k=1; k<nsamples; k++) {
  249.         float t =timebins[k];
  250.         float tp=timebins[k-1];
  251.         if (fdata[k] < adc) adc = fdata[k];
  252.         if (t>twin0 && t<twin1 ) {
  253.           nqdc++;
  254.           qdc+=fdata[k];
  255.         }  
  256.         if (fdata[k]< athreshold && fdata[k-1]> athreshold) {
  257.           double t0= tp+(athreshold-fdata[k-1])/(fdata[k]-fdata[k-1])* (t-tp);
  258.           H1D_Fill(3+id, t0,1);
  259.                   finfo[*ninfo+4+ntdc]=t0;
  260.                   ntdc++;
  261.         }
  262.       }
  263.       if (nqdc) qdc/=nqdc;
  264.       //printf("adc %f qdc %f\n", adc,qdc);
  265.           info[*ninfo]=(ntdc+4)*sizeof(int); // len
  266.           info[*ninfo+1]=id | 0xF0000000;    // recid
  267.           finfo[*ninfo+2]=adc;
  268.           finfo[*ninfo+3]=qdc;
  269.           *ninfo+= ntdc+4;
  270.          
  271.       if (adctype)
  272.         H1D_Fill(1+id, -adc,1);
  273.       else
  274.         H1D_Fill(1+id, -qdc,1);
  275.      
  276.       nr+=4*nsamples;
  277.       data+=nsamples;
  278.     }
  279.     recid      = *data++;
  280.     int event  = *data++;
  281.     nr+=8;
  282.     neve++;
  283.     if (debug) printf("recid %d event %d\n",recid, event );
  284.   }
  285.  
  286.   return neve;
  287. }
  288.  
  289. const int maxlen = 100000000;
  290. unsigned char data[maxlen];
  291. int *idata = (int *) &data[0];
  292.  
  293.  
  294. int evinfo[maxlen];
  295. int CVICALLBACK SocketCB (unsigned handle, int xType, int errCode, void *callbackData) {
  296.  
  297.   int nb = 0 ;
  298.   static int event = 0;
  299.   static int ncalls = 0;
  300.   static time_t t0, t1;
  301.   time(&t0);
  302.   switch (xType) {
  303.     case TCP_DISCONNECT:
  304.       printf("TCP_DISCONNECT ErrorString %s\n",GetTCPErrorString(errCode));
  305.       printf("TCP_DISCONNECT SystemErrorString %s\n",GetTCPSystemErrorString());
  306.  
  307.       chandle = 0;
  308.       break;
  309.     case TCP_DATAREADY: {
  310.       int  hdr[2]= {0,0};
  311.       nb = ClientTCPRead(handle,&hdr[0],8,1000);
  312.  
  313.       int size = hdr[1] - 8;
  314.       if (size>maxlen) size=maxlen;
  315.       nb = 0;
  316.       while (nb < size) {
  317.         int retval = ClientTCPRead(handle,&data[nb],size-nb,1000);
  318.         if (retval<1) break;
  319.         nb += retval;
  320.       }
  321.       if (debug) printf("Received RECID %d HDRLEN %d == read %d\n", hdr[0], size, nb);
  322.  
  323.           int ninfo=0;
  324.       switch (hdr[0]) {
  325.         case 0:
  326.           data[nb]=0;
  327.           printf("%s\n",data);
  328.           break;
  329.         case 1:// read
  330.  
  331.           GetCtrlVal(panelHandle,PANEL_CEVE     , &event);
  332.           GetCtrlVal(panelHandle,PANEL_NTOTAL   , &ncalls);
  333.                  
  334.           event += analyse(nb,data, evinfo, &ninfo);
  335.           if (foutput) {
  336.                         if (outwaveforms) fwrite(data, 1, nb, foutput);
  337.                         fwrite(evinfo, 1, ninfo*sizeof(int), foutput);
  338.                         time(&t1);
  339.                         if ((t1-t0)>10000) {
  340.                            for (int i=0;i<4;i++) H1D_Write2File(1+i,foutput);
  341.                            
  342.                            time(&t0);
  343.                         }      
  344.                   }
  345.           SetCtrlVal(panelHandle,PANEL_CEVE     , event);
  346.           if (event>=ncalls) StartCB (panelHandle, PANEL_START, EVENT_COMMIT, NULL,1,0);
  347.  
  348.           break;
  349.         default:
  350.           printf("Unknown command = %d\n", hdr[0]);
  351.           break;
  352.       }
  353.       break;
  354.  
  355.     }
  356.   }
  357.   return 0;
  358. }
  359.  
  360. int rpdecimation(int i){
  361.  
  362.         switch (i){
  363.                 case 1: return 0;
  364.                 case 8: return 1;
  365.                 case 64: return 2;
  366.                 case 1024: return 3;
  367.                 case 8192: return 4;
  368.                 case 65536: return 5;
  369.         }
  370.         return 0;
  371.  
  372. }
  373.  
  374. int CVICALLBACK StartCB (int panel, int control, int event,
  375.                          void *callbackData, int eventData1, int eventData2) {
  376.   char ip[0xFF];
  377.   int hdr[0xFF];
  378.   char filename[0xFF];
  379.   unsigned short *sbuff = (unsigned short *)&hdr[5] ;
  380.   unsigned char  *cbuff = (unsigned char *)&sbuff[3] ;
  381.   int imask[2];
  382.   unsigned char mask;
  383.   unsigned char trigger;
  384.   unsigned short nsamples;
  385.   int delay;
  386.   int nbefore;
  387.   unsigned int decimation;
  388.   unsigned short neve;
  389.   int output;
  390.   switch (event) {
  391.     case EVENT_COMMIT: {
  392.       GetCtrlVal(panel,PANEL_IP, ip);
  393.       GetCtrlVal(panel,PANEL_TRIGGER, &trigger);
  394.       GetCtrlVal(panel,PANEL_SAMPLES, &nsamples);
  395.       GetCtrlVal(panel,PANEL_DECIMATION,&decimation);
  396.       GetCtrlVal(panel,PANEL_NEVE   , &neve);
  397.       GetCtrlVal(panel,PANEL_CH0    , &imask[0] );
  398.       GetCtrlVal(panel,PANEL_CH1    , &imask[1]  );
  399.       GetCtrlVal(panel,PANEL_PFREQ  , &pfreq);
  400.       GetCtrlVal(panel,PANEL_DEBUG  , &debug);
  401.       GetCtrlVal(panel,PANEL_NBEFORE  , &nbefore);
  402.       GetCtrlVal(panel,PANEL_ENABLEDOUTPUT, &output);
  403.             GetCtrlVal(panel,PANEL_OUTWAVE, &outwaveforms);  
  404.       GetCtrlVal(panel,PANEL_FILENAME, filename);
  405.  
  406.       delay= MINTRGDELAY + nsamples - nbefore + 1;
  407.  
  408.       mask = 0;
  409.       for (int i=0; i<2; i++) {
  410.         if (imask[i]) mask |= (1<<i);
  411.       }
  412.  
  413.       double level =0;
  414.       GetCtrlVal(panel,PANEL_TRGLEVEL , &level);
  415.      
  416.  
  417.       switch (control) {
  418.         case PANEL_CONNECT: {
  419.           int state;
  420.           GetCtrlVal(panel,PANEL_CONNECT, &state);
  421.           if (state) {
  422.             ConnectToTCPServerEx (&chandle, 9930, ip, SocketCB, NULL, 0, TCP_ANY_LOCAL_PORT);
  423.           } else {
  424.             if (chandle!=0) DisconnectFromTCPServer (chandle);
  425.             chandle = 0;
  426.           }
  427.         }
  428.         break;
  429.         case PANEL_START: {
  430.           int state;
  431.           GetCtrlVal(panel,PANEL_START, &state);
  432.           if (state && eventData1==0) {
  433.             histoinit();
  434.                                                 /* decimation n (=1,8,64...) : frequency = 125/n MHz*/
  435.             for (int i=0; i<nsamples; i++) timebins[i]=(i-nbefore)* (float)decimation /125.;
  436.             if (output) foutput = fopen(filename, "wb");
  437.                                                 printf("decimation %d\n", decimation);
  438.             SetCtrlVal(panel,PANEL_CEVE  , 0);
  439.             ctrl_c=0;
  440.             hdr[0] = 0;
  441.             hdr[1] = 7*sizeof(int);
  442.             hdr[2] = delay;
  443.             hdr[3] = rpdecimation(decimation);
  444.             hdr[4] = level * 1000;
  445.             sbuff[0] = neve;
  446.             sbuff[1] = nsamples;
  447.             sbuff[2] = 1000; //tout
  448.             cbuff[0] = trigger;
  449.             cbuff[1] = mask;
  450.             ClientTCPWrite(chandle,&hdr[0],hdr[1],5000);  // acquire
  451.           } else {
  452.             hdr[0] = 1;
  453.             hdr[1] = 2*sizeof(int);
  454.             ClientTCPWrite(chandle,&hdr[0],hdr[1],5000);  // stop the transfer
  455.  
  456.             if (foutput) fclose(foutput);
  457.             foutput=NULL;
  458.           }
  459.           break;
  460.         }
  461.         default:
  462.           printf("Unknown command\n");
  463.           break;
  464.       }
  465.  
  466.  
  467.  
  468.  
  469.       ProcessSystemEvents ();
  470.       break;
  471.     }
  472.   }
  473.   return 0;
  474. }
  475.  
  476.  
  477. int CVICALLBACK ExitCB (int panel, int control, int event,
  478.                         void *callbackData, int eventData1, int eventData2) {
  479.   switch (event) {
  480.     case EVENT_COMMIT:
  481.       QuitUserInterface(0);
  482.       break;
  483.   }
  484.   return 0;
  485. }
  486.  
  487.  
  488.  
  489. int CVICALLBACK ExportCB (int panel, int control, int event,
  490.                           void *callbackData, int eventData1, int eventData2) {
  491.   int hid=0;
  492.   switch (event) {
  493.     case EVENT_COMMIT:
  494.       switch (control) {
  495.         case PANEL_EXPORT_1:
  496.           hid=1;
  497.           break;
  498.         case PANEL_EXPORT_2:
  499.           hid=2;
  500.           break;
  501.         case PANEL_EXPORT_3:
  502.           hid=3;
  503.           break;
  504.         case PANEL_EXPORT_4:
  505.           hid=4;
  506.           break;
  507.  
  508.       }
  509.       export_data(hid);
  510.       break;
  511.   }
  512.   return 0;
  513. }
  514.  
  515. int CVICALLBACK ResetCB (int panel, int control, int event,
  516.                          void *callbackData, int eventData1, int eventData2) {
  517.   switch (event) {
  518.     case EVENT_COMMIT:
  519.  
  520.       for (int i=1; i<=4; i++) H1D_Clear(i);
  521.       break;
  522.   }
  523.   return 0;
  524. }
  525.