Subversion Repositories f9daq

#include "H1D.h"
#include "H2D.h"
#include <formatio.h>
#include <utility.h>
#include <ansi_c.h>
#include <cvirte.h>
#include <userint.h>
#include "drs4.h"
#include "drsread.h"

#define MIKRO

#ifdef MIKRO
#include "MIKRO.h"
#endif

static int daq_on;
static int ph, p2,plothandle[4]= {0,0,0,0};
static int scanplothandle;
static int tfID;
static int controlID;

#define MAX_THREADS 10

static CmtThreadPoolHandle poolHandle = 0;

static float gSum[4]={0,0,0,0};
int main (int argc, char *argv[]) {
        short port;
        if (InitCVIRTE (0, argv, 0) == 0)
                return -1;  /* out of memory */
        if ((ph = LoadPanel (0, "drs4.uir", PANEL)) < 0)
                return -1;
        if ((p2 = LoadPanel (0, "drs4.uir", SCAN)) < 0)
                return -1;
        SetStdioPort (CVI_STDIO_WINDOW);
        SetSleepPolicy(VAL_SLEEP_MORE);
        CmtNewThreadPool (MAX_THREADS,  &poolHandle);

        DisplayPanel (ph);
        DisplayPanel (p2);
       
#ifdef MIKRO
        GetCtrlVal(p2, SCAN_PORT, &port);
        if (MIKRO_Open (port)) MessagePopup ("Error", "Mikro Port Not found !\n Change in the GUI") ;
        MIKRO_Init(1,0);
        MIKRO_Init(2,0);
#endif
       
        RunUserInterface ();
        DiscardPanel (ph);
        DiscardPanel (p2);
        CmtDiscardThreadPool (poolHandle);
        MIKRO_Close ();
        return 0;
}


char strbuf[0xFF];

int gLog=0;

int printf(const char *format, ...) {
  va_list aptr;
  int ret;
  FILE *flog;

  va_start(aptr, format);
  ret = vsprintf(strbuf, format, aptr);
  va_end(aptr);
  SetCtrlVal(ph,PANEL_STDIO,strbuf);

  if (gLog) {
    flog = fopen ("stdio.log", "a");
    fprintf (flog, "%s", strbuf);
    fclose (flog);
  }
  return(ret);
}



static void start_timer (double tout) {
        SetCtrlAttribute (ph, PANEL_TIMER, ATTR_INTERVAL, tout);
        SetCtrlAttribute (ph, PANEL_TIMER, ATTR_ENABLED, 1);
}

static void stop_timer ( void ) {
        SetCtrlAttribute (ph, PANEL_TIMER, ATTR_ENABLED, 0);
        DRSSetTimeout();
}




void CVICALLBACK EndOfThread ( CmtThreadPoolHandle poolhandle,
                                                                                                                         CmtThreadFunctionID functionID, unsigned int event,
                                                                                                                         int value, void *callbackData  ) {

        daq_on=0;
        //SetDimming(0);
        printf("End of Thread \n");
        return ;

}


static float xs[4][1024];      
int CVICALLBACK daq(void *functionData) {


        int neve;
        char filename[0xff];
        int imask[4];
        unsigned long mask;
        int frequency;
        double trgdelay;
        double trglevel;
        int trgtype;
        int trgchannel;
        int trgpolarity;
        int verbose;
        double range;
        int pfreq;
        int enabledoutput;
        int neveold = 0;
        double rate;
        int *args =  (int *) functionData;
        double twin[2];

       

        GetCtrlVal(ph, PANEL_CH0, &imask[0] );
        GetCtrlVal(ph, PANEL_CH1, &imask[1]  );
        GetCtrlVal(ph, PANEL_CH2, &imask[2]  );
        GetCtrlVal(ph, PANEL_CH3, &imask[3]  );
        mask = 0;
        for (int i=0; i<4; i++) {
                if (imask[i]) mask |= (1<<i);
        }


               
        GetCtrlVal(ph,PANEL_NEVE, &neve);
        GetCtrlVal(ph,PANEL_DEBUG, &verbose);
        GetCtrlVal(ph,PANEL_PFREQ, &pfreq);
        GetCtrlVal(ph,PANEL_ENABLEDOUTPUT, &enabledoutput);

        GetCtrlVal(ph,PANEL_FREQUENCY, &frequency);
        GetCtrlVal(ph,PANEL_TRGDELAY, &trgdelay);
        GetCtrlVal(ph,PANEL_TRGCHANNEL, &trgchannel);

       
        GetCtrlVal(ph,PANEL_TRGTYPE, &trgtype);
        GetCtrlVal(ph,PANEL_TRGLEVEL, &trglevel);
        GetCtrlVal(ph,PANEL_TRGPOLARITY, &trgpolarity);
        GetCtrlVal(ph,PANEL_RANGE, &range);

        GetCtrlVal(ph,PANEL_TWIN0, &twin[0]);
  GetCtrlVal(ph,PANEL_TWIN1, &twin[1]);
        //printf("mask=0x%x\n",mask);

        DRSSetMask( (unsigned char)( mask & 0xF ) );

        DRSSetFrequency( frequency  );
        DRSSetTriggerDelay(trgdelay );
        DRSSetTriggerChannel(trgchannel );
        DRSSetTriggerType( trgtype );
        DRSSetTriggerLevel(trglevel);
        DRSSetTriggerPolarity(trgpolarity);

        DRSSetRange (  range );



        FILE *fp= NULL;

        if (enabledoutput) {
                if (args[0]) {
                        GetCtrlVal(ph, PANEL_FILENAME, filename );  
                        fp =  fopen(filename,"wb");
          } else {
                        GetCtrlVal(p2, SCAN_FILENAME, filename );  
                        fp =  fopen(filename,"ab");
                }      
        }

        static unsigned char *buffer;

        int buffer_size = 0;
        const int nBoards=1;
        const int waveDepth=1024;

        if (buffer_size == 0) {
         buffer_size =   8; // file header + time header
         buffer_size +=  nBoards * (4 + 4*(4+waveDepth*4)); // bin widths
         buffer_size += 24 + nBoards * (8 + 4*(8+waveDepth*2));
         buffer = (unsigned char *)malloc(buffer_size);
  }
                       
        time_t t=0,told=0, tstart=0;

        if (!DRSInit()) {

                time(&tstart);
                told=tstart;
                int nev=0;
                for (int k = 0;k<4;k++){
                         gSum[k]=0;
                }
                for (int i=0; i<neve; i++) {
                        start_timer(1);// 1 s timeout
                        int retval = DRSRead(0);
                        stop_timer();
                        int nb =  ( retval == 0 && fp ) ? DRSToBuffer( buffer , i ) : 0;
                        SetCtrlVal(ph,PANEL_CEVE,i);
                        if (retval) i--;
                        nev++;
                        if (!daq_on) break;
                        time(&t);
                        if (t!=told ) {
                                rate =  (i-neveold);
                                printf("%d events in %2.2f min (%d s) Rate %f Hz %s   ",i+1, (double)(t-tstart)/60.,(t-tstart), rate , ctime(&t));
                                GetCtrlVal(ph,PANEL_PFREQ, &pfreq);
                                neveold = i;
                        }
                        told=t;
// Save data
                        if (nb>0 && fp) fwrite(buffer, 1,nb ,fp);
// Plot Data
                         for (int k=0; k<4; k++) {
                                        if ( (mask & ( 0x1<<k ))  ){
                                          float *t=DRSGetTime(k);
                                          float *x=DRSGetWave(k);
                                                                 ;
                                          for (int j=0 ; j<1024 ; j++) {
                                                  xs[k][j]= x[j]*1e-3;
                                               
                                                  if (t[j]> twin[0] && t[j] < twin[1]) gSum[k]+= fabs(xs[k][j]);
                                                  if (verbose) printf("[%d] %d. x= %3.2f  y=%3.2f\n", k, i, t[j], x[j] );
                                                  //h[k]->Fill( t[i], x[i]*1e-3);
                                          }

                                          if (i % pfreq == 0) {
                                                       
                                            const int col[4]= {VAL_WHITE,VAL_RED,VAL_GREEN,VAL_BLUE};
                                          if (plothandle[k])    DeleteGraphPlot (ph, PANEL_GRAPH, plothandle[k], VAL_IMMEDIATE_DRAW);
                                            plothandle[k] = PlotXY (ph, PANEL_GRAPH, t, xs[k], 1024, VAL_FLOAT, VAL_FLOAT, VAL_THIN_LINE, VAL_NO_POINT, VAL_SOLID, 1, col[k]);
                                          }    
                                        }
                                       
                                }


                }
                time(&t);
                printf("%d events in %2.2f min (%d s) %s",nev, (double)(t-tstart)/60.,t-tstart, ctime(&t));
                DRSEnd();
        }

        if (fp) fclose(fp);

        free(buffer);

        return 0;

}



int CVICALLBACK scan(void *functionData) {

        int dx[3]={0,0,0};
        int nx[3]={0,0,0};
        int x0[3]={0,0,0};
        int ix[3]={0,0,0};
        int idx[3]={0,0,0};  
        int size;
        char posrec[4]="POSR"; 
        char runbuf[4]="PRUN";
       
        int  n[3];
        char filename[0xFF];
        int enabledoutput;
       
        FILE *fp;
        GetCtrlVal(p2, SCAN_FILENAME, filename );
       
        GetCtrlVal(ph,PANEL_ENABLEDOUTPUT, &enabledoutput);
       
        if ( GetFileInfo(filename,&size) ) {
                MessagePopup ("Warning","File exist. Remove it first or choose another file");
          return 0;
        }        
        GetCtrlVal(p2, SCAN_STEPX, &dx[0]);
        GetCtrlVal(p2, SCAN_STEPY, &dx[1]);
        GetCtrlVal(p2, SCAN_NSTEPSX, &nx[0]);
        GetCtrlVal(p2, SCAN_NSTEPSY, &nx[1]);
        GetCtrlVal(p2, SCAN_STARTX, &x0[0]);
        GetCtrlVal(p2, SCAN_STARTY, &x0[1]);
        for (int k=0;k<4;k++)  {
                H2D_Init(k, "charge","Induced charge",
                                                nx[0], x0[0] - dx[0]*0.5 ,x0[0] + dx[0] * ( nx[0] - 0.5) ,
                                                nx[1], x0[1] - dx[1]*0.5 ,x0[1] + dx[1] * ( nx[1] - 0.5));  
          H1D_Init(k, "charge","Induced charge projection x",
                                                nx[0], x0[0] - dx[0]*0.5 ,x0[0] + dx[0] * ( nx[0] - 0.5) );  
          H1D_Init(100 + k, "charge","Induced charge projection y",
                                                nx[1], x0[1] - dx[1]*0.5 ,x0[1] + dx[1] * ( nx[1] - 0.5) );  
        }
        if (enabledoutput) {
                                fp =  fopen(filename,"ab");
                                if (fp) {
                                         size=36;
                                         fwrite(runbuf, 1,4 ,fp);
                                         fwrite(&size  , 1,4 ,fp);
                                         fwrite(x0    , 1,4*3 ,fp);
                                         fwrite(dx    , 1,4*3 ,fp);
                                         fwrite(nx    , 1,4*3 ,fp);
                                         fclose(fp);
                                }    
        }      
       
        for (int i=0; i<nx[0]; i++) {

                ix[0]= x0[0]+i*dx[0];
#ifdef MIKRO
                MIKRO_MoveTo(1,ix[0]);
#endif
                SetCtrlVal (p2, SCAN_IX, i);
                for (int j=0; j<nx[1]; j++) {
                       
                        SetCtrlVal (p2, SCAN_IY, j);

                        ix[1]= x0[1]+j*dx[1];
#ifdef MIKRO
                        MIKRO_MoveTo(2,ix[1]);

                        MIKRO_GetPosition(1,&n[0]);
                        SetCtrlVal (p2, SCAN_XP, n[0]);
                        MIKRO_GetPosition(2,&n[1]);
                        SetCtrlVal (p2, SCAN_YP, n[1]);
       
                        if (enabledoutput) {
                                fp =  fopen(filename,"ab");
                                if (fp) {
                                         idx[0]=i;
                                         idx[1]=j;
                                         size=24;
                                         fwrite(posrec, 1,4 ,fp);
                                         fwrite(&size , 1,4 ,fp);
                                         fwrite(n     , 1,4*3 ,fp);
                                         fwrite(idx    , 1,4*3 ,fp);
                                         fclose(fp);
                                }    
                        }      
                               
#endif
                        daq(functionData);
                       
                        for (int k=0;k<4;k++) {
                                H2D_Fill(k, ix[0], ix[1] , gSum[k] );
                                H1D_Fill(k, ix[0] , gSum[k] );
                                H1D_Fill(100+ k, ix[1] , gSum[k] );  
                        }      
                        PlotScanHistogramCB(0,0,EVENT_COMMIT,NULL, 0,0);
                        if (!daq_on) break;
                }
                if (!daq_on) break;
        }

        return 0;
}


int CVICALLBACK StartCB (int panel, int control, int event,
                                                                                                 void *callbackData, int eventData1, int eventData2) {
        ThreadFunctionPtr mythread = NULL;
        switch (event) {

                case EVENT_COMMIT:

                        controlID=0;
                        if (panel == ph && control == PANEL_START) {
                                mythread = daq;
                                controlID= control;
                        }
                        if (panel == p2 && control == SCAN_SCAN)   mythread = scan;
                        if (mythread!=NULL) {
                                printf("New Thread panel=%d button=%d\n", panel, control);

                                // SetDimming(1);

                                daq_on=1;
                                CmtScheduleThreadPoolFunctionAdv (poolHandle, mythread, &controlID,
                                                                                                                                                                        DEFAULT_THREAD_PRIORITY,
                                                                                                                                                                        EndOfThread,
                                                                                                                                                                        EVENT_TP_THREAD_FUNCTION_END,
                                                                                                                                                                        NULL, RUN_IN_SCHEDULED_THREAD,
                                                                                                                                                                        &tfID);
                        }
                        break;
        }
        return 0;
}

int CVICALLBACK StopCB (int panel, int control, int event,
                                                                                                void *callbackData, int eventData1, int eventData2) {
        switch (event) {
                case EVENT_COMMIT:
                        daq_on=0;
                        break;
        }
        return 0;
}

int CVICALLBACK ExitCB (int panel, int control, int event,
                                                                                                void *callbackData, int eventData1, int eventData2) {
        switch (event) {
                case EVENT_COMMIT:
                        QuitUserInterface (0);
                        break;
        }
        return 0;
}

int CVICALLBACK MoveStageCB (int panel, int control, int event,
                                                                                                                 void *callbackData, int eventData1, int eventData2) {
        int axis=0, step=1000, direction=1, n;
        switch (event) {
                case EVENT_COMMIT:

                        if (panel == p2) {
                                switch (control) {
                                        case SCAN_BR :
                                                axis = 1;
                                                direction = 1;
                                                GetCtrlVal(p2, SCAN_STEPX, &step);
                                                break;
                                        case SCAN_BL :
                                                axis = 1;
                                                direction = -1;
                                                GetCtrlVal(p2, SCAN_STEPX, &step);
                                                break;
                                        case SCAN_BU :
                                                axis = 2;
                                                direction = 1;
                                                GetCtrlVal(p2, SCAN_STEPY, &step);
                                                break;
                                        case SCAN_BD :
                                                axis = 2;
                                                direction = -1;
                                                GetCtrlVal(p2, SCAN_STEPY, &step);
                                                break;
                                }
#ifdef MIKRO
                                MIKRO_MoveFor(axis,  direction*step );
                                MIKRO_GetPosition(axis,&n);
                                if (axis == 1) SetCtrlVal (p2, SCAN_XP, n);
                                if (axis == 2) SetCtrlVal (p2, SCAN_YP, n);
#endif // MIKRO                                
                        }

                        break;
        }
        return 0;
}





int CVICALLBACK GoXCB (int panel, int control, int event,
                                                                                         void *callbackData, int eventData1, int eventData2) {
        int n2;
        switch (event) {
                case EVENT_COMMIT:
                        GetCtrlVal (p2, SCAN_XG, &n2);
#ifdef MIKRO
                        MIKRO_MoveTo(1,n2);
                        MIKRO_GetPosition(1,&n2);
                       
#endif
                        SetCtrlVal (p2, SCAN_XP, n2);
                        break;
        }
        return 0;
}

int CVICALLBACK GoYCB (int panel, int control, int event,
                                                                                         void *callbackData, int eventData1, int eventData2) {
        int n2;
        switch (event) {
                case EVENT_COMMIT:
                        GetCtrlVal (p2, SCAN_YG, &n2);
#ifdef MIKRO
                        MIKRO_MoveTo(2,n2);
                        MIKRO_GetPosition(2,&n2);
                       
#endif
                        SetCtrlVal (p2, SCAN_YP, n2);
                        break;
        }
        return 0;
}

int CVICALLBACK GetCurrentPositionCB (int panel, int control, int event,
                                                                                                                                        void *callbackData, int eventData1, int eventData2) {
       
        int n[2];      
        switch (event) {
                case EVENT_COMMIT:
#ifdef MIKRO                   
                        MIKRO_GetPosition(1,&n[0]);
                        SetCtrlVal (p2, SCAN_XP, n[0]);
                        MIKRO_GetPosition(2,&n[1]);
                        SetCtrlVal (p2, SCAN_YP, n[1]);
#endif                 
                        break;
        }
        return 0;
}

int CVICALLBACK HomeCB (int panel, int control, int event,
                                                                                                void *callbackData, int eventData1, int eventData2) {
        switch (event) {
                case EVENT_COMMIT:
#ifdef MIKRO                   
                        MIKRO_ReferenceMove(1);
                        MIKRO_ReferenceMove(2);
                        GetCurrentPositionCB(panel, control, event, NULL, 0, 0);
#endif                 
                        break;
        }
        return 0;
}

int CVICALLBACK PlotScanHistogramCB (int panel, int control, int event,
                                                                                                                                                 void *callbackData, int eventData1, int eventData2) {
        int hid=0;
        int nx;
        int ny;
        switch (event) {
                case EVENT_COMMIT:
                        GetCtrlVal (p2, SCAN_CHANNEL, &hid);
                        GetCtrlVal(p2, SCAN_NSTEPSX, &nx);
            GetCtrlVal(p2, SCAN_NSTEPSY, &ny);
                        if (nx>1 && ny>1) {
                                H2D_Draw(hid,p2,SCAN_GRAPH,&scanplothandle);
                                printf("redraw 2d\n");
                        } else {
                                if (nx>1)  {
                                        printf("redraw 1d x\n");
                                        H1D_Draw(hid,p2,SCAN_GRAPH,&scanplothandle);
                                }      
                          if (ny>1) {
                                        printf("redraw 1d x\n");
                                        H1D_Draw(100+hid,p2,SCAN_GRAPH,&scanplothandle);
                                }
                        }      
                        break;
        }
        return 0;
}

int CVICALLBACK OpenGuiCB (int panel, int control, int event,
                                                                                                         void *callbackData, int eventData1, int eventData2) {
        switch (event) {
                case EVENT_COMMIT:
                        DisplayPanel (ph);
            DisplayPanel (p2);
                        break;
        }
        return 0;
}

int CVICALLBACK CalibrateCB (int panel, int control, int event,
                                                                                                                 void *callbackData, int eventData1, int eventData2) {
        int status = 0;
        switch (event) {
                case EVENT_COMMIT:
                  status = DRSCalibrateTiming();
                        printf("DRSCalibrateTiming() status = %d\n", status);
                        break;
        }
        return 0;
}